import bt2
import collections.abc
+
# Run callable `func` in the context of a component's __init__ method. The
# callable is passed the Component being instantiated.
#
return run_in_component_init(f)
+# Create a pair of list, one containing non-const messages and the other
+# containing const messages
+def _get_all_message_types(with_packet=True):
+ _msgs = None
+
+ class MyIter(bt2._UserMessageIterator):
+ def __init__(self, self_output_port):
+
+ nonlocal _msgs
+ self._at = 0
+ self._msgs = [
+ self._create_stream_beginning_message(
+ self_output_port.user_data['stream']
+ )
+ ]
+
+ if with_packet:
+ assert self_output_port.user_data['packet']
+ self._msgs.append(
+ self._create_packet_beginning_message(
+ self_output_port.user_data['packet']
+ )
+ )
+
+ default_clock_snapshot = 789
+
+ if with_packet:
+ assert self_output_port.user_data['packet']
+ ev_parent = self_output_port.user_data['packet']
+ else:
+ assert self_output_port.user_data['stream']
+ ev_parent = self_output_port.user_data['stream']
+
+ msg = self._create_event_message(
+ self_output_port.user_data['event_class'],
+ ev_parent,
+ default_clock_snapshot,
+ )
+
+ msg.event.payload_field['giraffe'] = 1
+ msg.event.specific_context_field['ant'] = -1
+ msg.event.common_context_field['cpu_id'] = 1
+ self._msgs.append(msg)
+
+ if with_packet:
+ self._msgs.append(
+ self._create_packet_end_message(
+ self_output_port.user_data['packet']
+ )
+ )
+
+ self._msgs.append(
+ self._create_stream_end_message(self_output_port.user_data['stream'])
+ )
+
+ _msgs = self._msgs
+
+ def __next__(self):
+ if self._at == len(self._msgs):
+ raise bt2.Stop
+
+ msg = self._msgs[self._at]
+ self._at += 1
+ return msg
+
+ class MySrc(bt2._UserSourceComponent, message_iterator_class=MyIter):
+ def __init__(self, params, obj):
+ tc = self._create_trace_class()
+ clock_class = self._create_clock_class(frequency=1000)
+
+ # event common context (stream-class-defined)
+ cc = tc.create_structure_field_class()
+ cc += [('cpu_id', tc.create_signed_integer_field_class(8))]
+
+ # packet context (stream-class-defined)
+ pc = None
+
+ if with_packet:
+ pc = tc.create_structure_field_class()
+ pc += [('something', tc.create_unsigned_integer_field_class(8))]
+
+ stream_class = tc.create_stream_class(
+ default_clock_class=clock_class,
+ event_common_context_field_class=cc,
+ packet_context_field_class=pc,
+ supports_packets=with_packet,
+ )
+
+ # specific context (event-class-defined)
+ sc = tc.create_structure_field_class()
+ sc += [('ant', tc.create_signed_integer_field_class(16))]
+
+ # event payload
+ ep = tc.create_structure_field_class()
+ ep += [('giraffe', tc.create_signed_integer_field_class(32))]
+
+ event_class = stream_class.create_event_class(
+ name='garou', specific_context_field_class=sc, payload_field_class=ep
+ )
+
+ trace = tc(environment={'patate': 12})
+ stream = trace.create_stream(stream_class, user_attributes={'salut': 23})
+
+ if with_packet:
+ packet = stream.create_packet()
+ packet.context_field['something'] = 154
+ else:
+ packet = None
+
+ self._add_output_port(
+ 'out',
+ {
+ 'tc': tc,
+ 'stream': stream,
+ 'event_class': event_class,
+ 'trace': trace,
+ 'packet': packet,
+ },
+ )
+
+ _graph = bt2.Graph()
+ _src_comp = _graph.add_component(MySrc, 'my_source')
+ _msg_iter = TestOutputPortMessageIterator(_graph, _src_comp.output_ports['out'])
+
+ const_msgs = list(_msg_iter)
+
+ return _msgs, const_msgs
+
+
+def get_stream_beginning_message():
+ msgs, _ = _get_all_message_types()
+ for m in msgs:
+ if type(m) is bt2._StreamBeginningMessage:
+ return m
+
+
+def get_const_stream_beginning_message():
+ _, const_msgs = _get_all_message_types()
+ for m in const_msgs:
+ if type(m) is bt2._StreamBeginningMessageConst:
+ return m
+
+
+def get_stream_end_message():
+ msgs, _ = _get_all_message_types()
+ for m in msgs:
+ if type(m) is bt2._StreamEndMessage:
+ return m
+
+
+def get_packet_beginning_message():
+ msgs, _ = _get_all_message_types(with_packet=True)
+ for m in msgs:
+ if type(m) is bt2._PacketBeginningMessage:
+ return m
+
+
+def get_const_packet_beginning_message():
+ _, const_msgs = _get_all_message_types(with_packet=True)
+ for m in const_msgs:
+ if type(m) is bt2._PacketBeginningMessageConst:
+ return m
+
+
+def get_packet_end_message():
+ msgs, _ = _get_all_message_types(with_packet=True)
+ for m in msgs:
+ if type(m) is bt2._PacketEndMessage:
+ return m
+
+
+def get_event_message():
+ msgs, _ = _get_all_message_types()
+ for m in msgs:
+ if type(m) is bt2._EventMessage:
+ return m
+
+
+def get_const_event_message():
+ _, const_msgs = _get_all_message_types()
+ for m in const_msgs:
+ if type(m) is bt2._EventMessageConst:
+ return m
+
+
# Proxy sink component class.
#
# This sink accepts a list of a single item as its initialization