155 lines
5.2 KiB
Python
155 lines
5.2 KiB
Python
from distutils.core import setup, Extension
|
|
from distutils.command.build import build
|
|
from distutils.command.clean import clean
|
|
from distutils.command.build_ext import build_ext
|
|
from distutils.dir_util import remove_tree
|
|
from distutils.sysconfig import get_python_inc
|
|
from distutils import log
|
|
import os, os.path
|
|
import glob
|
|
import platform
|
|
import time
|
|
|
|
try:
|
|
import py2exe
|
|
except ImportError:
|
|
py2exe = None
|
|
|
|
# now, setup the keyword arguments for setup
|
|
# (because we don't know until runtime if py2exe is available)
|
|
setup_kwargs = {}
|
|
setup_kwargs['options'] = {}
|
|
setup_kwargs['ext_modules'] = []
|
|
setup_kwargs['cmdclass'] = {}
|
|
|
|
# helper to create a 'data_files'-type sequence recursively for a given dir
|
|
def recursive_data_files(src, dest=None):
|
|
if dest is None:
|
|
dest = src
|
|
|
|
ret = []
|
|
for dirpath, dirnames, filenames in os.walk(src):
|
|
current_dest = os.path.relpath(dirpath, src)
|
|
if current_dest == '.':
|
|
current_dest = dest
|
|
else:
|
|
current_dest = os.path.join(dest, current_dest)
|
|
|
|
current_sources = map(lambda p: os.path.join(dirpath, p), filenames)
|
|
|
|
ret.append((current_dest, current_sources))
|
|
return ret
|
|
|
|
#
|
|
# py2exe options
|
|
#
|
|
|
|
if py2exe is not None:
|
|
setup_kwargs['console'] = ['overviewer.py']
|
|
setup_kwargs['data_files'] = [('textures', ['textures/lava.png', 'textures/water.png', 'textures/fire.png', 'textures/portal.png']),
|
|
('', ['COPYING.txt', 'README.rst'])]
|
|
setup_kwargs['data_files'] += recursive_data_files('web_assets')
|
|
setup_kwargs['zipfile'] = None
|
|
if platform.system() == 'Windows' and '64bit' in platform.architecture():
|
|
b = 3
|
|
else:
|
|
b = 1
|
|
setup_kwargs['options']['py2exe'] = {'bundle_files' : b, 'excludes': 'Tkinter'}
|
|
|
|
#
|
|
# c_overviewer extension
|
|
#
|
|
|
|
# Third-party modules - we depend on numpy for everything
|
|
import numpy
|
|
# Obtain the numpy include directory. This logic works across numpy versions.
|
|
try:
|
|
numpy_include = numpy.get_include()
|
|
except AttributeError:
|
|
numpy_include = numpy.get_numpy_include()
|
|
|
|
try:
|
|
pil_include = os.environ['PIL_INCLUDE_DIR'].split(os.pathsep)
|
|
except:
|
|
pil_include = [ os.path.join(get_python_inc(plat_specific=1), 'Imaging') ]
|
|
if not os.path.exists(pil_include[0]):
|
|
pil_include = [ ]
|
|
|
|
|
|
# used to figure out what files to compile
|
|
render_modes = ['normal', 'overlay', 'lighting', 'night', 'spawn', 'cave']
|
|
|
|
c_overviewer_files = ['src/main.c', 'src/composite.c', 'src/iterate.c', 'src/endian.c', 'src/rendermodes.c']
|
|
c_overviewer_files += map(lambda mode: 'src/rendermode-%s.c' % (mode,), render_modes)
|
|
c_overviewer_files += ['src/Draw.c']
|
|
c_overviewer_includes = ['src/overviewer.h', 'src/rendermodes.h']
|
|
|
|
setup_kwargs['ext_modules'].append(Extension('c_overviewer', c_overviewer_files, include_dirs=['.', numpy_include] + pil_include, depends=c_overviewer_includes, extra_link_args=[]))
|
|
|
|
# tell build_ext to build the extension in-place
|
|
# (NOT in build/)
|
|
setup_kwargs['options']['build_ext'] = {'inplace' : 1}
|
|
# tell the build command to only run build_ext
|
|
build.sub_commands = [('build_ext', None)]
|
|
|
|
# custom clean command to remove in-place extension
|
|
class CustomClean(clean):
|
|
def run(self):
|
|
# do the normal cleanup
|
|
clean.run(self)
|
|
|
|
# try to remove '_composite.{so,pyd,...}' extension,
|
|
# regardless of the current system's extension name convention
|
|
build_ext = self.get_finalized_command('build_ext')
|
|
pretty_fname = build_ext.get_ext_filename('c_overviewer')
|
|
fname = pretty_fname
|
|
if os.path.exists(fname):
|
|
try:
|
|
if not self.dry_run:
|
|
os.remove(fname)
|
|
log.info("removing '%s'", pretty_fname)
|
|
except OSError:
|
|
log.warn("'%s' could not be cleaned -- permission denied",
|
|
pretty_fname)
|
|
else:
|
|
log.debug("'%s' does not exist -- can't clean it",
|
|
pretty_fname)
|
|
|
|
class CustomBuild(build_ext):
|
|
def build_extensions(self):
|
|
c = self.compiler.compiler_type
|
|
if c == "msvc":
|
|
# customize the build options for this compilier
|
|
for e in self.extensions:
|
|
e.extra_link_args.append("/MANIFEST")
|
|
|
|
build_ext.build_extensions(self)
|
|
|
|
|
|
if py2exe is not None:
|
|
# define a subclass of py2exe to build our version file on the fly
|
|
class CustomPy2exe(py2exe.build_exe.py2exe):
|
|
def run(self):
|
|
try:
|
|
import util
|
|
f = open("overviewer_version.py", "w")
|
|
f.write("VERSION=%r\n" % util.findGitVersion())
|
|
f.write("BUILD_DATE=%r\n" % time.asctime())
|
|
f.write("BUILD_PLATFORM=%r\n" % platform.processor())
|
|
f.write("BUILD_OS=%r\n" % platform.platform())
|
|
f.close()
|
|
setup_kwargs['data_files'].append(('.', ['overviewer_version.py']))
|
|
except:
|
|
print "WARNING: failed to build overview_version file"
|
|
py2exe.build_exe.py2exe.run(self)
|
|
setup_kwargs['cmdclass']['py2exe'] = CustomPy2exe
|
|
|
|
setup_kwargs['cmdclass']['clean'] = CustomClean
|
|
setup_kwargs['cmdclass']['build_ext'] = CustomBuild
|
|
###
|
|
|
|
setup(**setup_kwargs)
|
|
|
|
|
|
print "\nBuild Complete"
|