def setUp(self): self._trace = bt2.Trace() self._sc = bt2.StreamClass() self._ec = bt2.EventClass('salut') self._my_int_ft = bt2.IntegerFieldType(32) self._ec.payload_field_type = bt2.StructureFieldType() self._ec.payload_field_type += collections.OrderedDict([ ('my_int', self._my_int_ft), ]) self._sc.add_event_class(self._ec) self._clock_class = bt2.ClockClass('allo', 1000) self._trace.add_clock_class(self._clock_class) self._trace.packet_header_field_type = bt2.StructureFieldType() self._trace.packet_header_field_type += collections.OrderedDict([ ('hello', self._my_int_ft), ]) self._trace.add_stream_class(self._sc) self._cc_prio_map = bt2.ClockClassPriorityMap() self._cc_prio_map[self._clock_class] = 231 self._stream = self._sc() self._packet = self._stream.create_packet() self._packet.header_field['hello'] = 19487 self._event = self._ec() self._event.add_clock_value(self._clock_class(1772)) self._event.payload_field['my_int'] = 23 self._event.packet = self._packet
def _complete_sc_event_simple_ts(self): ehft = bt2.StructureFieldType() ehft.append_field('id', bt2.IntegerFieldType(32)) ehft.append_field('timestamp', bt2.IntegerFieldType(16, mapped_clock_class=self._cc)) self._sc.event_header_field_type = ehft self._ec = bt2.EventClass('evt') payloadft = bt2.StructureFieldType() payloadft.append_field('value', bt2.IntegerFieldType(32)) self._ec.payload_field_type = payloadft self._complete_sc()
def setUp(self): self._context_ft = bt2.StructureFieldType() self._context_ft.append_field('allo', bt2.StringFieldType()) self._context_ft.append_field('zola', bt2.IntegerFieldType(18)) self._payload_ft = bt2.StructureFieldType() self._payload_ft.append_field('zoom', bt2.StringFieldType()) self._ec = bt2.EventClass('my_event') self._ec.id = 18 self._ec.context_field_type = self._context_ft self._ec.payload_field_type = self._payload_ft
def _create_event_classes(self): context_ft = bt2.StructureFieldType() context_ft.append_field('allo', bt2.StringFieldType()) context_ft.append_field('zola', bt2.IntegerFieldType(18)) payload_ft = bt2.StructureFieldType() payload_ft.append_field('zoom', bt2.StringFieldType()) ec1 = bt2.EventClass('event23', id=23, context_field_type=context_ft, payload_field_type=payload_ft) ec2 = bt2.EventClass('event17', id=17, context_field_type=payload_ft, payload_field_type=context_ft) return ec1, ec2
def setUp(self): self._packet_context_ft = bt2.StructureFieldType() self._packet_context_ft.append_field('menu', bt2.FloatingPointNumberFieldType()) self._packet_context_ft.append_field('sticker', bt2.StringFieldType()) self._event_header_ft = bt2.StructureFieldType() self._event_header_ft.append_field('id', bt2.IntegerFieldType(19)) self._event_context_ft = bt2.StructureFieldType() self._event_context_ft.append_field('msg', bt2.StringFieldType()) self._ec1, self._ec2 = self._create_event_classes() self._sc = bt2.StreamClass(name='my_stream_class', id=12, packet_context_field_type=self._packet_context_ft, event_header_field_type=self._event_header_ft, event_context_field_type=self._event_context_ft, event_classes=(self._ec1, self._ec2))
def _create_stream_class(self, name, id): ec1, ec2 = self._create_event_classes() packet_context_ft = bt2.StructureFieldType() packet_context_ft.append_field('menu', bt2.FloatingPointNumberFieldType()) packet_context_ft.append_field('sticker', bt2.StringFieldType()) event_header_ft = bt2.StructureFieldType() event_header_ft.append_field('id', bt2.IntegerFieldType(19)) event_context_ft = bt2.StructureFieldType() event_context_ft.append_field('msg', bt2.StringFieldType()) return bt2.StreamClass(name=name, id=id, packet_context_field_type=packet_context_ft, event_header_field_type=event_header_ft, event_context_field_type=event_context_ft, event_classes=(ec1, ec2))
def test_iadd(self): struct_ft = bt2.StructureFieldType() c_field_type = bt2.StringFieldType() d_field_type = bt2.EnumerationFieldType(size=32) e_field_type = bt2.StructureFieldType() struct_ft.append_field('c_string', c_field_type) struct_ft.append_field('d_enum', d_field_type) struct_ft.append_field('e_struct', e_field_type) a_field_type = bt2.FloatingPointNumberFieldType() b_field_type = bt2.IntegerFieldType(17) self._ft.append_field('a_float', a_field_type) self._ft.append_field('b_int', b_field_type) self._ft += struct_ft self.assertEqual(self._ft['a_float'], a_field_type) self.assertEqual(self._ft['b_int'], b_field_type) self.assertEqual(self._ft['c_string'], c_field_type) self.assertEqual(self._ft['d_enum'], d_field_type) self.assertEqual(self._ft['e_struct'], e_field_type)
def _create_ec(): # clock class cc = bt2.ClockClass('salut_clock') # event header eh = bt2.StructureFieldType() eh += OrderedDict(( ('id', bt2.IntegerFieldType(8)), ('ts', bt2.IntegerFieldType(32, mapped_clock_class=cc)), )) # packet context pc = bt2.StructureFieldType() pc += OrderedDict((('something', bt2.IntegerFieldType(8)), )) # stream class sc = bt2.StreamClass() sc.packet_context_field_type = pc sc.event_header_field_type = eh sc.event_context_field_type = None # event payload ep = bt2.StructureFieldType() ep += OrderedDict((('mosquito', bt2.StringFieldType()), )) # event class event_class = bt2.EventClass('ec', id=0) event_class.context_field_type = None event_class.payload_field_type = ep sc.add_event_class(event_class) # packet header ph = bt2.StructureFieldType() ph += OrderedDict(( ('magic', bt2.IntegerFieldType(32)), ('stream_id', bt2.IntegerFieldType(16)), )) # trace trace = bt2.Trace() trace.packet_header_field_type = ph trace.add_clock_class(cc) trace.add_stream_class(sc) return event_class
def _test_eq_create_objects(self): cc1_uuid = uuid.UUID('bc7f2f2d-2ee4-4e03-ab1f-2e0e1304e94f') cc1 = bt2.ClockClass('cc1', uuid=cc1_uuid) cc2_uuid = uuid.UUID('da7d6b6f-3108-4706-89bd-ab554732611b') cc2 = bt2.ClockClass('cc2', uuid=cc2_uuid) sc1 = self._create_stream_class('sc1', 3) sc2 = self._create_stream_class('sc2', 9) header_ft = bt2.StructureFieldType() header_ft.append_field('magic', bt2.IntegerFieldType(32)) return cc1, cc2, sc1, sc2, header_ft
def setUp(self): self._trace_path = tempfile.mkdtemp() self._writer = bt2.CtfWriter(self._trace_path) self._cc = bt2.ClockClass('default', int(1e9)) self._writer.trace.add_clock_class(self._cc) self._sc = bt2.StreamClass() pcft = bt2.StructureFieldType() pcft.append_field('packet_size', bt2.IntegerFieldType(32)) pcft.append_field('content_size', bt2.IntegerFieldType(32)) pcft.append_field('timestamp_begin', bt2.IntegerFieldType(64, mapped_clock_class=self._cc)) pcft.append_field('timestamp_end', bt2.IntegerFieldType(64, mapped_clock_class=self._cc)) self._sc.packet_context_field_type = pcft
def _build_meta(self): self._trace = bt2.Trace() self._sc = bt2.StreamClass() self._ec = bt2.EventClass('salut') self._my_int_ft = bt2.IntegerFieldType(32) self._ec.payload_field_type = bt2.StructureFieldType() self._ec.payload_field_type += collections.OrderedDict([ ('my_int', self._my_int_ft), ]) self._sc.add_event_class(self._ec) self._trace.add_stream_class(self._sc) self._stream = self._sc() self._packet = self._stream.create_packet()
def setUp(self): self._trace = bt2.Trace() self._sc = bt2.StreamClass() self._ec = bt2.EventClass('salut') self._clock_class = bt2.ClockClass('yo', 1000) self._uint64_int_ft = bt2.IntegerFieldType( 64, mapped_clock_class=self._clock_class) self._my_int_ft = bt2.IntegerFieldType(32) self._ec.payload_field_type = bt2.StructureFieldType() self._ec.payload_field_type += collections.OrderedDict([ ('my_int', self._my_int_ft), ]) self._sc.add_event_class(self._ec) self._sc.packet_context_field_type = bt2.StructureFieldType() self._sc.packet_context_field_type += collections.OrderedDict([ ('events_discarded', self._my_int_ft), ('timestamp_begin', self._uint64_int_ft), ('timestamp_end', self._uint64_int_ft), ]) self._trace.add_clock_class(self._clock_class) self._trace.add_stream_class(self._sc) self._stream = self._sc()
def test_type(self): int_ft = bt2.IntegerFieldType(32) enum_ft = bt2.EnumerationFieldType(int_ft) enum_ft.append_mapping('corner', 23) enum_ft.append_mapping('zoom', 17, 20) enum_ft.append_mapping('mellotron', 1001) enum_ft.append_mapping('giorgio', 2000, 3000) array_ft = bt2.ArrayFieldType(int_ft, 5) seq_ft = bt2.SequenceFieldType(int_ft, 'the_len_field') float_ft = bt2.FloatingPointNumberFieldType() struct_ft = bt2.StructureFieldType() struct_ft.append_field('a', int_ft) struct_ft.append_field('b', int_ft) struct_ft.append_field('c', int_ft) _string_ft = bt2.StringFieldType() variant_ft = bt2.VariantFieldType('tag', enum_ft) variant_ft.append_field('corner', int_ft) variant_ft.append_field('zoom', array_ft) variant_ft.append_field('mellotron', float_ft) variant_ft.append_field('giorgio', struct_ft) expected_types = { babeltrace.CTFTypeId.INTEGER: int_ft, babeltrace.CTFTypeId.FLOAT: float_ft, babeltrace.CTFTypeId.ENUM: enum_ft, babeltrace.CTFTypeId.STRING: _string_ft, babeltrace.CTFTypeId.STRUCT: struct_ft, babeltrace.CTFTypeId.VARIANT: variant_ft, babeltrace.CTFTypeId.ARRAY: array_ft, babeltrace.CTFTypeId.SEQUENCE: seq_ft, } for type_id, ft in expected_types.items(): declaration = self._get_declaration(ft) self.assertIsNotNone(declaration) self.assertEqual(declaration.type, type_id)
def test_create_full(self): header_ft = bt2.StructureFieldType() header_ft.append_field('magic', bt2.IntegerFieldType(32)) clock_classes = bt2.ClockClass('cc1'), bt2.ClockClass('cc2') sc = self._create_stream_class('sc1', 3) tc = bt2.Trace(name='my name', native_byte_order=bt2.ByteOrder.LITTLE_ENDIAN, env={ 'the_string': 'value', 'the_int': 23 }, packet_header_field_type=header_ft, clock_classes=clock_classes, stream_classes=(sc, )) self.assertEqual(tc.name, 'my name') self.assertEqual(tc.native_byte_order, bt2.ByteOrder.LITTLE_ENDIAN) self.assertEqual(tc.env['the_string'], 'value') self.assertEqual(tc.env['the_int'], 23) self.assertEqual(tc.packet_header_field_type, header_ft) self.assertEqual(tc.clock_classes['cc1'], clock_classes[0]) self.assertEqual(tc.clock_classes['cc2'], clock_classes[1]) self.assertEqual(tc[3], sc)
def setUp(self): self._values = { 'ph_field_1': 42, 'ph_field_2': 'bla bla', 'spc_field': 'some string', 'seh_field': 'another string', 'sec_field': 68752, 'ec_field': 89, 'ef_field': 8476, } self._int_ft = bt2.IntegerFieldType(32) self._str_ft = bt2.StringFieldType() self._trace = bt2.Trace() self._trace.packet_header_field_type = bt2.StructureFieldType() self._trace.packet_header_field_type += collections.OrderedDict([ ('ph_field_1', self._int_ft), ('ph_field_2', self._str_ft), ]) self._sc = bt2.StreamClass() self._sc.packet_context_field_type = bt2.StructureFieldType() self._sc.packet_context_field_type += collections.OrderedDict([ ('spc_field', self._str_ft), ]) self._sc.event_header_field_type = bt2.StructureFieldType() self._sc.event_header_field_type += collections.OrderedDict([ ('seh_field', self._str_ft), ]) self._sc.event_context_field_type = bt2.StructureFieldType() self._sc.event_context_field_type += collections.OrderedDict([ ('sec_field', self._int_ft), ]) self._clock_class = bt2.ClockClass('allo', 1000) self._trace.add_clock_class(self._clock_class) self._ec = bt2.EventClass('event_class_name') self._ec.context_field_type = bt2.StructureFieldType() self._ec.context_field_type += collections.OrderedDict([ ('ec_field', self._int_ft), ]) self._ec.payload_field_type = bt2.StructureFieldType() self._ec.payload_field_type += collections.OrderedDict([ ('ef_field', self._int_ft), ]) self._sc.add_event_class(self._ec) self._trace.add_stream_class(self._sc) self._cc_prio_map = bt2.ClockClassPriorityMap() self._cc_prio_map[self._clock_class] = 231 self._stream = self._sc() self._packet = self._stream.create_packet() self._packet.header_field['ph_field_1'] = self._values['ph_field_1'] self._packet.header_field['ph_field_2'] = self._values['ph_field_2'] self._packet.context_field['spc_field'] = self._values['spc_field'] self._event = self._ec() self._event.add_clock_value(self._clock_class(1772)) self._event.header_field['seh_field'] = self._values['seh_field'] self._event.stream_event_context_field['sec_field'] = self._values[ 'sec_field'] self._event.context_field['ec_field'] = self._values['ec_field'] self._event.payload_field['ef_field'] = self._values['ef_field'] self._event.packet = self._packet
def test_create_with_min_alignment(self): ft = bt2.StructureFieldType(8) self.assertEqual(ft.alignment, 8)
def setUp(self): self._ft = bt2.StructureFieldType()
def _get_std_header(self): header_ft = bt2.StructureFieldType() header_ft.append_field('magic', bt2.IntegerFieldType(32)) header_ft.append_field('stream_id', bt2.IntegerFieldType(32)) return header_ft
def test_assign_packet_header_field_type(self): header_ft = bt2.StructureFieldType() header_ft.append_field('magic', bt2.IntegerFieldType(32)) self._tc.packet_header_field_type = header_ft self.assertEqual(self._tc.packet_header_field_type, header_ft)
def _create_packet(self, with_ph=True, with_pc=True): # event header eh = bt2.StructureFieldType() eh += OrderedDict(( ('id', bt2.IntegerFieldType(8)), ('ts', bt2.IntegerFieldType(32)), )) # stream event context sec = bt2.StructureFieldType() sec += OrderedDict(( ('cpu_id', bt2.IntegerFieldType(8)), ('stuff', bt2.FloatingPointNumberFieldType()), )) # packet context if with_pc: pc = bt2.StructureFieldType() pc += OrderedDict(( ('something', bt2.IntegerFieldType(8)), ('something_else', bt2.FloatingPointNumberFieldType()), )) else: pc = None # stream class sc = bt2.StreamClass() sc.packet_context_field_type = pc sc.event_header_field_type = eh sc.event_context_field_type = sec # event context ec = bt2.StructureFieldType() ec += OrderedDict(( ('ant', bt2.IntegerFieldType(16, is_signed=True)), ('msg', bt2.StringFieldType()), )) # event payload ep = bt2.StructureFieldType() ep += OrderedDict(( ('giraffe', bt2.IntegerFieldType(32)), ('gnu', bt2.IntegerFieldType(8)), ('mosquito', bt2.IntegerFieldType(8)), )) # event class event_class = bt2.EventClass('ec') event_class.context_field_type = ec event_class.payload_field_type = ep sc.add_event_class(event_class) # packet header if with_ph: ph = bt2.StructureFieldType() ph += OrderedDict(( ('magic', bt2.IntegerFieldType(32)), ('stream_id', bt2.IntegerFieldType(16)), )) else: ph = None # trace c;ass tc = bt2.Trace() tc.packet_header_field_type = ph tc.add_stream_class(sc) # stream stream = sc() # packet return stream.create_packet()