super().__init__(ptr)
- def _handle_status(self, status, gen_error_msg):
- if status == native_bt.GRAPH_STATUS_CANCELED:
- raise bt2.GraphCanceled
- elif status == native_bt.GRAPH_STATUS_END:
- raise bt2.Stop
- elif status == native_bt.GRAPH_STATUS_AGAIN:
- raise bt2.TryAgain
- elif status < 0:
- raise bt2.Error(gen_error_msg)
def add_component(self, component_class, name, params=None,
logging_level=bt2.logging.LoggingLevel.NONE):
add_fn = native_bt.graph_add_sink_component
cc_type = native_bt.COMPONENT_CLASS_TYPE_SINK
elif issubclass(component_class, bt2.component._UserSourceComponent):
- cc_ptr = component_class._cc_ptr
+ cc_ptr = component_class._bt_cc_ptr
add_fn = native_bt.graph_add_source_component
cc_type = native_bt.COMPONENT_CLASS_TYPE_SOURCE
elif issubclass(component_class, bt2.component._UserSinkComponent):
- cc_ptr = component_class._cc_ptr
+ cc_ptr = component_class._bt_cc_ptr
add_fn = native_bt.graph_add_sink_component
cc_type = native_bt.COMPONENT_CLASS_TYPE_SINK
elif issubclass(component_class, bt2.component._UserFilterComponent):
- cc_ptr = component_class._cc_ptr
+ cc_ptr = component_class._bt_cc_ptr
add_fn = native_bt.graph_add_filter_component
cc_type = native_bt.COMPONENT_CLASS_TYPE_FILTER
else:
status, comp_ptr = add_fn(self._ptr, cc_ptr, name,
params_ptr, logging_level)
- self._handle_status(status, 'cannot add component to graph')
+ utils._handle_func_status(status, 'cannot add component to graph')
assert comp_ptr
return bt2.component._create_component_from_ptr(comp_ptr, cc_type)
status, conn_ptr = native_bt.graph_connect_ports(self._ptr,
upstream_port._ptr,
downstream_port._ptr)
- self._handle_status(status, 'cannot connect component ports within graph')
+ utils._handle_func_status(status,
+ 'cannot connect component ports within graph')
assert(conn_ptr)
return bt2.connection._Connection._create_from_ptr(conn_ptr)
if not callable(listener):
raise TypeError("'listener' parameter is not callable")
- fn = native_bt.py3_graph_add_port_added_listener
+ fn = native_bt.bt2_graph_add_port_added_listener
listener_from_native = functools.partial(_graph_port_added_listener_from_native,
listener)
if not callable(listener):
raise TypeError("'listener' parameter is not callable")
- fn = native_bt.py3_graph_add_ports_connected_listener
+ fn = native_bt.bt2_graph_add_ports_connected_listener
listener_from_native = functools.partial(_graph_ports_connected_listener_from_native,
listener)
def run(self):
status = native_bt.graph_run(self._ptr)
- if status == native_bt.GRAPH_STATUS_END:
+ try:
+ utils._handle_func_status(status,
+ 'graph object stopped running because of an unexpected error')
+ except bt2.Stop:
+ # done
return
-
- self._handle_status(status, 'graph object stopped running because of an unexpected error')
+ except Exception:
+ raise
def cancel(self):
status = native_bt.graph_cancel(self._ptr)
- self._handle_status(status, 'cannot cancel graph object')
+ utils._handle_func_status(status, 'cannot cancel graph object')
@property
def is_canceled(self):