66import hpat
77import numpy as np
88
9- def read_ros_images ():
10- return 0
9+ def read_ros_images (f_name ):
10+ # implementation to enable regular python
11+ def f (file_name ): # pragma: no cover
12+ bag = hpat .ros .open_bag (file_name )
13+ num_msgs = hpat .ros .get_msg_count (bag )
14+ m ,n = hpat .ros .get_image_dims (bag )
15+ # hpat.cprint(num_msgs, m, n)
16+ A = np .empty ((num_msgs , m , n , 3 ), dtype = np .uint8 )
17+ s = hpat .ros .read_ros_images_inner (A , bag )
18+ return A
19+
20+ return hpat .jit (f )(f_name )
1121
1222# inner functions
1323
14- def open_bag ():
24+ def open_bag (file_name ):
1525 return 0
1626
17- def get_msg_count ():
27+ def get_msg_count (bag ):
1828 return 0
1929
20- def get_image_dims ():
30+ def get_image_dims (bag ):
2131 return 0
2232
23- def read_ros_images_inner ():
33+ def read_ros_images_inner (A , bag ):
2434 return 0
2535
2636def read_ros_images_inner_parallel ():
@@ -31,11 +41,11 @@ def _handle_read_images(lhs, rhs):
3141 fname = rhs .args [0 ]
3242 def f (file_name ): # pragma: no cover
3343 bag = hpat .ros .open_bag (file_name )
34- num_msgs = hpat .ros .get_msg_count (bag )
35- m , n = hpat .ros .get_image_dims (bag )
44+ _num_msgs = hpat .ros .get_msg_count (bag )
45+ _ros_m , _ros_n = hpat .ros .get_image_dims (bag )
3646 # hpat.cprint(num_msgs, m, n)
37- A = np .empty ((num_msgs , m , n , 3 ), dtype = np .uint8 )
38- s = hpat .ros .read_ros_images_inner (A , bag )
47+ _in_ros_arr = np .empty ((_num_msgs , _ros_m , _ros_n , 3 ), dtype = np .uint8 )
48+ _ret = hpat .ros .read_ros_images_inner (_in_ros_arr , bag )
3949
4050 f_block = compile_to_numba_ir (f , {'np' : np , 'hpat' : hpat }).blocks .popitem ()[1 ]
4151 replace_arg_nodes (f_block , [fname ])
@@ -155,4 +165,3 @@ def lower_read_images_inner(context, builder, sig, args):
155165 lir .IntType (64 )])
156166 fn = builder .module .get_or_insert_function (fnty , name = "read_images_parallel" )
157167 return builder .call (fn , [ builder .bitcast (out .data , lir .IntType (8 ).as_pointer ()), bag , args [2 ], args [3 ]])
158-
0 commit comments