pymca 4.5.0-4 / usr / share / pyshared / PyMca / Object3D / setup.py

import sys,os
import glob
import platform
from distutils.core import Extension, setup
try:
    import numpy
except ImportError:
    text  = "You must have numpy installed.\n"
    text += "See http://sourceforge.net/project/showfiles.php?group_id=1369&package_id=175103\n"
    raise ImportError, text
import distutils.sysconfig
global OBJECT3D_INSTALL_DIR
global OBJECT3D_SCRIPTS_DIR
import string

__version__ = '1.0'
for line in file('SceneGLWindow.py').readlines():
    if line[:11] == '__version__':
        exec(line)
        # Append cvs tag if working from cvs tree
        if os.path.isdir('.svn') and os.path.isfile(os.sep.join(['.svn', 'entries'])):
            import re
            revision = 0
            revre = re.compile('committed-rev="(\d+)"')
            for match in revre.finditer(open(os.sep.join(['.svn', 'entries'])).read()):
                revision = max(revision, int(match.group(1)))
            __version__ += 'dev_r%i' % revision
        break

print "Object3D Toolkit %s" % __version__
print 
print "Type 'L' to view the license."
print "Type 'yes' to accept the terms of the license."
print "Type 'no' to decline the terms of the license."
print

while 1:
    try:
        resp = raw_input("Do you accept the terms of the license? ")
    except KeyboardInterrupt:
        raise SystemExit
    except:
        resp = ""

    resp = string.lower(string.strip(resp))

    if resp == "yes":
        break

    if resp == "no":
        sys.exit(1)

    if resp == "l":
        os.system("more LICENSE.GPL")


# Specify all the required Object3D data
data_files = [('Object3D', ['LICENSE.GPL',])]

packages = ['Object3D']
package_dir = {'Object3D':os.path.dirname(__file__)}

"""
#py_modules = []
py_modules = []
for python_file in glob.glob('*.py'):
    if python_file not in ['setup.py', 'cx_setup.py']:
        continue
    m = "Object3D.%s" % (os.path.basename(python_file)[:-3])
    py_modules.append(m)
"""

py_modules = []
for python_file in glob.glob('Object3DPlugins/*.py'):
    m = "Object3D.Object3DPlugins.%s" % (os.path.basename(python_file)[:-3])
    py_modules.append(m)

sources = glob.glob('*.c')
if sys.platform == "win32":
    libraries = ['opengl32', 'glu32']
    define_macros = [('WIN32',None)]
    script_files = []
    script_files.append('scripts/object3d_win_post_install.py')
else:
    libraries = ['GL', 'GLU']
    if sys.platform == 'darwin':
        libraries=[]
    define_macros = []
    script_files = []
    for f in glob.glob('scripts/*'):
        if f.endswith('.py'):
            continue
        script_files.append(f)
            

def build_Object3DCTools(ext_modules):
    includes = [numpy.get_include()]
    if sys.platform == 'windows':
        WindowsSDK = os.getenv('WindowsSdkDir')
        #if WindowsSDK is not None:
        #    includes.append(WindowsSDK)
    module  = Extension(name = 'Object3D.Object3DCTools',
                        sources = glob.glob('Object3DCTools/*.c'),
                        define_macros = define_macros,
                        libraries  = libraries,
                        include_dirs = includes)
    ext_modules.append(module)

def build_Object3DQhull(ext_modules):
    module  = Extension(name = 'Object3D.Object3DQhull',
                        sources = glob.glob('Object3DQhull/src/*.c'),
                        define_macros = define_macros,
                        include_dirs = [numpy.get_include()])

    ext_modules.append(module)

ext_modules = []
build_Object3DCTools(ext_modules)
build_Object3DQhull(ext_modules)

# data_files fix from http://wiki.python.org/moin/DistutilsInstallDataScattered
from distutils.command.install_data import install_data
class smart_install_data(install_data):
    def run(self):
        global OBJECT3D_INSTALL_DIR
        #need to change self.install_dir to the library dir
        install_cmd = self.get_finalized_command('install')
        self.install_dir = getattr(install_cmd, 'install_lib')
        OBJECT3D_INSTALL_DIR = self.install_dir
        print "Object3D to be installed in %s" %  self.install_dir
        return install_data.run(self)

from distutils.command.install_scripts import install_scripts
class smart_install_scripts(install_scripts):
    def run (self):
        global OBJECT3D_SCRIPTS_DIR
        #I prefer not to translate the python used during the build
        #process for the case of having an installation on a disk shared
        #by different machines and starting python from a shell script
        #that positions the environment
        from distutils import log
        from stat import ST_MODE
        install_cmd = self.get_finalized_command('install')
        #This is to ignore the --install-scripts keyword
        #I do not know if to leave it optional ...
        if False:
            self.install_dir = os.path.join(getattr(install_cmd, 'install_lib'), 'Object3D')
            self.install_dir = os.path.join(self.install_dir, 'bin')
        else:
            self.install_dir = getattr(install_cmd, 'install_scripts')
        OBJECT3D_SCRIPTS_DIR = self.install_dir        
        if sys.platform != "win32":
            print "Object3D scripts to be installed in %s" %  self.install_dir
        self.outfiles = self.copy_tree(self.build_dir, self.install_dir)
        self.outfiles = []
        for filein in glob.glob('scripts/*'):
            filedest = os.path.join(self.install_dir, os.path.basename(filein))
            if os.path.exists(filedest):
                os.remove(filedest)
            moddir = os.path.join(getattr(install_cmd,'install_lib'), "Object3D")
            f = open(filein, 'r')
            modfile = f.readline().replace("\n","")
            f.close()
            text  = "#!/bin/bash\n"
            text += "export PYTHONPATH=%s:${PYTHONPATH}\n" % moddir
            text += "exec python %s $*\n" %  os.path.join(moddir, modfile)
            f=open(filedest, 'w')
            f.write(text)
            f.close()
            #self.copy_file(filein, filedest)
            self.outfiles.append(filedest)
        if os.name == 'posix':
            # Set the executable bits (owner, group, and world) on
            # all the scripts we just installed.
            for ffile in self.get_outputs():
                if self.dry_run:
                    log.info("changing mode of %s", ffile)
                else:
                    mode = ((os.stat(ffile)[ST_MODE]) | 0555) & 07777
                    log.info("changing mode of %s to %o", ffile, mode)
                    os.chmod(ffile, mode)
   
description = "GPL License unless a commercial license is bought. Please contact industry@esrf.fr if needed."
long_description = """Stand-alone python application and tools for multidimensional data visualization"""

distrib = setup(name="Object3D",
                version= __version__,
                description = description,
                author = "V. Armando Sole",
                author_email="sole@esrf.fr",
                license= "GPL - Please read LICENSE.GPL for details",
                url = "http://pymca.sourceforge.net",
                long_description = long_description,
                packages = packages,
                package_dir=package_dir,
                platforms='any',
                ext_modules = ext_modules,
                data_files = data_files,
                cmdclass = {'install_data':smart_install_data, 
                            'install_scripts':smart_install_scripts},
                scripts=script_files,
                py_modules=py_modules,
                )
#cleanup files
for fname in ['setup.py', 'cx_setup.py']:
    file_to_remove = os.path.join(OBJECT3D_INSTALL_DIR, 'Object3D',fname)
    if os.path.exists(file_to_remove):
        os.remove(file_to_remove)
    if os.path.exists(file_to_remove+'c'):
        os.remove(file_to_remove+'c')
    if os.path.exists(file_to_remove+'o'):
        os.remove(file_to_remove+'o')


#post installation checks
try:
    import sip
    SIP = True
except ImportError:
    SIP = False
    print "sip must be installed for full Object3D functionality."

badtext = "No valid PyQt4 and PyOpenGL installation found.\n"

try:
    print "Object3D is installed in %s " % OBJECT3D_INSTALL_DIR
except NameError:
    #I really do not see how this may happen but ...
    pass
    
if SIP:
    try:
        import PyQt4.QtCore
        QT4 = True
    except ImportError:
        QT4 = False

    try:
        import OpenGL
        GL = True
    except:
        GL = False        
    if QT4 and GL:
        print "You have PyQt4 and PyOpenGL installed."
        print "Object3D is fully functional under PyQt4 with PyOpenGL (ctypes)."
        print "You can easily embed Object3D in your Qt4 graphical "
        print "applications using SceneGLWindow.py"
    else:
        print badtext
else:
    print "No valid PyQt4 installation found."