39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
|
|
Import('envGPUgeo2rdr')
|
|
package = envGPUgeo2rdr['PACKAGE']
|
|
project = envGPUgeo2rdr['PROJECT']
|
|
install = envGPUgeo2rdr['PRJ_SCONS_INSTALL'] + '/' + package + '/' + project
|
|
build = envGPUgeo2rdr['PRJ_SCONS_BUILD'] + '/' + package + '/' + project + '/src'
|
|
|
|
if envGPUgeo2rdr['GPU_ACC_ENABLED']:
|
|
envGPUgeo2rdr.AppendUnique(CPPPATH=envGPUgeo2rdr['CUDACPPPATH'])
|
|
envGPUgeo2rdr.AppendUnique(LIBPATH=envGPUgeo2rdr['CUDALIBPATH'])
|
|
envGPUgeo2rdr.AppendUnique(LIBS=['cuda','cudart'])
|
|
|
|
|
|
###Custom cython builder
|
|
cythonBuilder = Builder(action = 'cython3 $SOURCE --cplus',
|
|
suffix = '.cpp',
|
|
src_suffix = '.pyx')
|
|
envGPUgeo2rdr.Append(BUILDERS = {'Pyx2Cpp':cythonBuilder})
|
|
|
|
def cythonPseudoBuilder(env,source,bld,inst):
|
|
cppCode = env.Pyx2Cpp(source)
|
|
listFiles = [source+'.cpp', 'Ellipsoid.cpp', 'Geo2rdr.cpp', 'GeoController.cpp', 'LinAlg.cpp', 'Orbit.cpp', 'Poly1d.cpp']
|
|
env.MergeFlags('-fopenmp -O3 -std=c++11 -fPIC -pthread -Wno-unused-result -Wsign-compare -DNDEBUG -g -fwrapv -Wall -Wstrict-prototypes')
|
|
|
|
if (env['GPU_ACC_ENABLED']):
|
|
listFiles.append('GPUgeo.cu')
|
|
lib = env.LoadableModule(target = 'GPUgeo2rdr.abi3.so', source = listFiles, CPPDEFINES = 'GPU_ACC_ENABLED')
|
|
else:
|
|
lib = env.LoadableModule(target = 'GPUgeo2rdr.abi3.so', source = listFiles)
|
|
env.Install(inst,lib)
|
|
env.Alias('install',inst)
|
|
|
|
|
|
envGPUgeo2rdr.AddMethod(cythonPseudoBuilder,'Cython')
|
|
envGPUgeo2rdr.Cython('GPUgeo2rdr',build,install)
|