1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import tempfile
import os
import shutil
from Cython.Compiler import (
MemoryView, Code, Options, Pipeline, Errors, Main, Symtab
)
from Cython.Compiler.StringEncoding import EncodedString
from Cython.Compiler.Scanning import FileSourceDescriptor
def create_shared_library_pipeline(context, scope, options, result):
parse = Pipeline.parse_stage_factory(context)
def generate_tree_factory(context):
def generate_tree(compsrc):
tree = parse(compsrc)
tree.scope.use_utility_code(
MemoryView.get_view_utility_code(options.shared_utility_qualified_name))
tree.scope.use_utility_code(MemoryView._get_memviewslice_declare_code())
tree.scope.use_utility_code(MemoryView._get_typeinfo_to_format_code())
context.include_directories.append(Code.get_utility_dir())
return tree
return generate_tree
orig_cimport_from_pyx = Options.cimport_from_pyx
def set_cimport_from_pyx(cimport_from_pyx):
def inner(node):
Options.cimport_from_pyx = cimport_from_pyx
return node
return inner
return [
# "cimport_from_pyx=True" to force generating __Pyx_ExportFunction
set_cimport_from_pyx(True),
generate_tree_factory(context),
*Pipeline.create_pipeline(context, 'pyx', exclude_classes=()),
Pipeline.inject_pxd_code_stage_factory(context),
Pipeline.inject_utility_code_stage_factory(context, internalise_c_class_entries=False),
Pipeline.inject_utility_pxd_code_stage_factory(context),
Pipeline.abort_on_errors,
Pipeline.generate_pyx_code_stage_factory(options, result),
set_cimport_from_pyx(orig_cimport_from_pyx),
]
def generate_shared_module(options):
Errors.init_thread()
Errors.open_listing_file(None)
dest_c_file = options.shared_c_file_path
module_name = os.path.splitext(os.path.basename(dest_c_file))[0]
context = Main.Context.from_options(options)
scope = Symtab.ModuleScope('MemoryView', parent_module = None, context = context, is_package=False)
with tempfile.TemporaryDirectory() as tmpdirname:
pyx_file = os.path.join(tmpdirname, f'{module_name}.pyx')
c_file = os.path.join(tmpdirname, f'{module_name}.c')
with open(pyx_file, 'w'):
pass
source_desc = FileSourceDescriptor(pyx_file)
comp_src = Main.CompilationSource(source_desc, EncodedString(module_name), os.getcwd())
result = Main.create_default_resultobj(comp_src, options)
pipeline = create_shared_library_pipeline(context, scope, options, result)
err, enddata = Pipeline.run_pipeline(pipeline, comp_src)
if err is None:
shutil.copy(c_file, dest_c_file)
return err, enddata
|
