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()
