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
|
from rpython.rtyper.llannotation import lltype_to_annotation
from rpython.rlib.objectmodel import specialize
from rpython.rlib import jit, rgc
from rpython.rtyper.lltypesystem import lltype, llmemory
from rpython.rtyper.extregistry import ExtRegistryEntry
from rpython.tool.pairtype import pair
def copy_list_to_raw_array(lst, array):
for i, item in enumerate(lst):
array[i] = item
def populate_list_from_raw_array(lst, array, length):
lst[:] = [array[i] for i in range(length)]
class Entry(ExtRegistryEntry):
_about_ = copy_list_to_raw_array
def compute_result_annotation(self, *s_args):
pass
def specialize_call(self, hop):
hop.exception_cannot_occur()
v_list, v_buf = hop.inputargs(*hop.args_r)
return hop.gendirectcall(ll_copy_list_to_raw_array, v_list, v_buf)
class Entry(ExtRegistryEntry):
_about_ = populate_list_from_raw_array
def compute_result_annotation(self, s_list, s_array, s_length):
s_item = lltype_to_annotation(s_array.ll_ptrtype.TO.OF)
s_newlist = self.bookkeeper.newlist(s_item)
s_newlist.listdef.resize()
pair(s_list, s_newlist).union()
def specialize_call(self, hop):
v_list, v_buf, v_length = hop.inputargs(*hop.args_r)
hop.exception_is_here()
return hop.gendirectcall(ll_populate_list_from_raw_array, v_list, v_buf, v_length)
@specialize.ll()
def get_raw_buf(ptr):
ofs = llmemory.itemoffsetof(lltype.typeOf(ptr).TO, 0)
return llmemory.cast_ptr_to_adr(ptr) + ofs
get_raw_buf._always_inline_ = True
@jit.dont_look_inside
def ll_copy_list_to_raw_array(ll_list, dst_ptr):
if rgc.must_split_gc_address_space():
for i in range(ll_list.ll_length()):
dst_ptr[i] = ll_list.ll_getitem_fast(i)
return
# this code is delicate: we must ensure that there are no GC operations
# around the call to raw_memcopy
#
ITEM = lltype.typeOf(dst_ptr).TO.OF
size = llmemory.sizeof(ITEM) * ll_list.ll_length()
# start of no-GC section
src_adr = get_raw_buf(ll_list.ll_items())
dst_adr = get_raw_buf(dst_ptr)
llmemory.raw_memcopy(src_adr, dst_adr, size)
# end of no-GC section
@jit.dont_look_inside
def ll_populate_list_from_raw_array(ll_list, src_ptr, length):
ll_list._ll_resize(length)
if rgc.must_split_gc_address_space():
for i in range(length):
ll_list.ll_setitem_fast(i, src_ptr[i])
return
ITEM = lltype.typeOf(src_ptr).TO.OF
size = llmemory.sizeof(ITEM) * length
# start of no-GC section
src_adr = get_raw_buf(src_ptr)
dst_adr = get_raw_buf(ll_list.ll_items())
llmemory.raw_memcopy(src_adr, dst_adr, size)
# end of no-GC section
|
