def export_pythonmodules(binding):
tmpdir = temporarydir()
from shutil import copy
#copy python sources to temp dir
pysources = []
for module in binding.python_sources:
copy(module, tmpdir)
pysources.append(os.path.basename(module))
continue
#make a Make.mm and local.def in the temp dir
makemm = pythonpackage_mk.Generator(binding.python_package, pysources)
makemm.generate(tmpdir)
localdef = local_def.Generator(includes=binding.c_includes)
localdef.generate(tmpdir)
#run mm
callmm(tmpdir)
#clean up
from mcstas2 import DEBUG
if not DEBUG:
rmtree(tmpdir)
return
def build_binding(binding):
tmpdir = temporarydir()
from shutil import copy
#copy c souces to temp dir
csources = []
for csrc in binding.c_sources + binding.c_headers:
copy(csrc, tmpdir)
csources.append(os.path.basename(csrc))
continue
#make a Make.mm and local.def there
makemm = binding_mk.Generator(binding.python_package,
binding.binding_module, csources,
binding.c_libs, binding.c_libdirs)
makemm.generate(tmpdir)
localdef = local_def.Generator(includes=binding.c_includes,
defines=binding.c_defines)
localdef.generate(tmpdir)
#run mm
callmm(tmpdir)
#clean up
from mcstas2 import DEBUG
if not DEBUG:
rmtree(tmpdir)
return
def createBindingObject(
componentfilename, componentcategory,
pythonpackage = None,
path = None, componentname = None,
bindingname = None, bindingtype = 'boostpython'):
"""Create the binding object of a McStas component in python
componentfilename: component file path. Eg. /.../monitors/E_monitor.comp
componentcategory: category of component. Eg. monitors
pythonpackage: the python package where this component will be export to. Eg. mcstas.components.monitors
path: temporary path where binding sources go
componentname: name of the component. Eg. E_monitor
bindingname: name of the binding. Eg. E_monitorboostpython
bindingtype: type of binding. Eg. boostpython (currently only this is supported)
"""
debug.log( 'pythonpackage=%s' % pythonpackage )
if pythonpackage is None:
pythonpackage = 'mcstas2.components.%s' % componentcategory
pass
if not path:
import temporaryfiles
path = temporaryfiles.temporarydir()
debug.log('generateing wrapper in %s' % path)
pass
if not componentname:
componentname = os.path.splitext( os.path.basename( componentfilename ) )[0]
pass
if not bindingname: bindingname = '%s%s' % (componentname, bindingtype)
# read component info
from mcstas2.utils.parsers import parseComponent
compInfo = parseComponent( componentfilename )
# generate c++ sources for the component
from component2cppClass.component2cppClass import componentInfo2cppClass
klass = componentInfo2cppClass( compInfo )
from mcstas2.utils.mills.cxx.factory import createHHandCC
hh, cc = createHHandCC( klass, path )
# generate bindings for the c++ class
from binding import binding
binding = binding( bindingtype )
bindingsources = binding.generate_binding_sources(
bindingname, klass, path )
# genearte python code to wrap the binding into a factory method
from pymodule import generate
pysources = generate( compInfo, bindingname, path )
if bindingsources.get( 'python' ) is None:
bindingsources['python'] = pysources
else:
bindingsources['python'] += pysources
# build binding
from mcstas2.release import mcvinedir as export_root
export_include = os.path.join(export_root, 'include')
danse_dir = os.environ.get('DANSE_DIR') or ''
danse_ins_include = os.path.join(danse_dir, 'include')
export_lib = os.path.join(export_root, 'lib')
danse_ins_libs = [
os.path.join(danse_dir, 'lib64'),
os.path.join(danse_dir, 'lib'),
]
from binding_builder import binding as bindingdataobject
bindingobj = bindingdataobject(
python_package = pythonpackage, binding_module = bindingname,
c_headers = [ hh ],
c_sources = bindingsources['c'] + [cc],
python_sources = bindingsources['python'],
c_libs = ['mcstas2', 'mcstas2_share', 'mcni' ] + binding.libstolink,
c_libdirs = [export_lib] + danse_ins_libs,
c_includes = [export_include, danse_ins_include],
c_defines = binding.define_macros,
dependencies = [ bindingtype, 'caltech-config', 'mcstas2', 'mcni' ],
)
return bindingobj, klass.name, componentcategory, componentname
def wrap(componentfilename,
componentcategory,
pythonpackage=None,
path=None,
componentname=None,
bindingname=None,
bindingtype='boostpython',
buildername='mm',
pythonexportroot=None):
"""wrap a McStas component in python
componentfilename: component file path. Eg. /.../monitors/E_monitor.comp
componentcategory: category of component. Eg. monitors
pythonpackage: the python package where this component will be export to. Eg. mcstas.components.monitors
path: temporary path where binding sources go
componentname: name of the component. Eg. E_monitor
bindingname: name of the binding. Eg. E_monitorboostpython
bindingtype: type of binding. Eg. boostpython (currently only this is supported)
buildername: binding builder. choices: mm and distutils
pythonexportroot: directory where python modules are exported. Eg. $EXPORT_ROOT. None means pyton modules will be exported wherever the binding builder's default export path.
"""
debug.log('pythonpackage=%s' % pythonpackage)
if pythonpackage is None:
pythonpackage = 'mcstas2.components.%s' % componentcategory
pass
if not path:
import temporaryfiles
path = temporaryfiles.temporarydir()
debug.log('generateing wrapper in %s' % path)
pass
if not componentname:
componentname = os.path.splitext(
os.path.basename(componentfilename))[0]
pass
if not bindingname: bindingname = '%s%s' % (componentname, bindingtype)
# read component info
from mcstas2.utils.parsers import parseComponent
compInfo = parseComponent(componentfilename)
# generate c++ sources for the component
from component2cppClass.component2cppClass import componentInfo2cppClass
klass = componentInfo2cppClass(compInfo)
from mcstas2.utils.mills.cxx.factory import createHHandCC
hh, cc = createHHandCC(klass, path)
# generate bindings for the c++ class
from binding import binding
binding = binding(bindingtype)
bindingsources = binding.generate_binding_sources(bindingname, klass, path)
# genearte python code to wrap the binding into a factory method
from pymodule import generate
pysources = generate(compInfo, bindingname, path)
if bindingsources.get('python') is None:
bindingsources['python'] = pysources
else:
bindingsources['python'] += pysources
# build binding
from mcstas2.release import mcvinedir as export_root
export_include = os.path.join(export_root, 'include')
export_lib = os.path.join(export_root, 'lib')
from binding_builder import binding as bindingdataobject
bindingobj = bindingdataobject(
python_package=pythonpackage,
binding_module=bindingname,
c_headers=[hh],
c_sources=bindingsources['c'] + [cc],
python_sources=bindingsources['python'],
c_libs=['mcstas2', 'mcstas2_share', 'mcni'] + binding.libstolink,
c_libdirs=[export_lib],
c_includes=[export_include],
c_defines=binding.define_macros,
dependencies=[bindingtype, 'caltech-config', 'mcstas2', 'mcni'],
)
from binding_builder import builder
builder(buildername).build(bindingobj, pythonexportroot)
# register the new factory
from mcstas2.components import registercomponent
m = __import__('%s.%s' % (pythonpackage, klass.name), {}, {}, [''])
registercomponent(componentcategory, componentname, m)
return
