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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
import os
import re
import shutil
import tempfile
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
def generate_c_utilities(module_node):
UtilityCode = Code.UtilityCode
match_special = UtilityCode.get_special_comment_matcher('/')
for c_utility_file in os.listdir(Code.get_utility_dir()):
if not c_utility_file.endswith('.c'):
continue
for line in Code.read_utilities_hook(c_utility_file):
if not ((m := match_special(line)) and (name := m.group('name'))):
continue
if not (section_title := UtilityCode.match_section_title(name)):
continue
name, section_type = section_title.groups()
if section_type == 'export':
module_node.scope.use_utility_code(UtilityCode.load_cached(name, c_utility_file))
return module_node
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=()),
generate_c_utilities,
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
|
