# THE SOFTWARE.
from bt2 import native_bt, object, utils
-import bt2.field_types
+import bt2.field_class
import collections.abc
import functools
import numbers
def _create_from_ptr(ptr):
- # recreate the field type wrapper of this field's type (the identity
+ # recreate the field class wrapper of this field's type (the identity
# could be different, but the underlying address should be the
# same)
- field_type_ptr = native_bt.ctf_field_get_type(ptr)
- utils._handle_ptr(field_type_ptr, "cannot get field object's type")
- field_type = bt2.field_types._create_from_ptr(field_type_ptr)
- typeid = native_bt.ctf_field_type_get_type_id(field_type._ptr)
+ field_class_ptr = native_bt.field_get_type(ptr)
+ utils._handle_ptr(field_class_ptr, "cannot get field object's type")
+ field_class = bt2.field_class._create_from_ptr(field_class_ptr)
+ typeid = native_bt.field_class_get_type_id(field_class._ptr)
field = _TYPE_ID_TO_OBJ[typeid]._create_from_ptr(ptr)
- field._field_type = field_type
+ field._field_class = field_class
return field
class _Field(object._Object, metaclass=abc.ABCMeta):
def __copy__(self):
- ptr = native_bt.ctf_field_copy(self._ptr)
+ ptr = native_bt.field_copy(self._ptr)
utils._handle_ptr(ptr, 'cannot copy {} field object'.format(self._NAME.lower()))
return _create_from_ptr(ptr)
return cpy
def __eq__(self, other):
- # special case: two unset fields with the same field type are equal
+ # special case: two unset fields with the same field class are equal
if isinstance(other, _Field):
if not self.is_set or not other.is_set:
- if not self.is_set and not other.is_set and self.field_type == other.field_type:
+ if not self.is_set and not other.is_set and self.field_class == other.field_class:
return True
return False
return self._spec_eq(other)
@property
- def field_type(self):
- return self._field_type
+ def field_class(self):
+ return self._field_class
@property
def is_set(self):
- is_set = native_bt.ctf_field_is_set(self._ptr)
+ is_set = native_bt.field_is_set(self._ptr)
return is_set > 0
def reset(self):
- ret = native_bt.ctf_field_reset(self._ptr)
+ ret = native_bt.field_reset(self._ptr)
utils._handle_ret(ret, "cannot reset field object's value")
+ def _repr(self):
+ raise NotImplementedError
+
+ def __repr__(self):
+ return self._repr() if self.is_set else 'Unset'
+
@functools.total_ordering
class _NumericField(_Field):
def __float__(self):
return float(self._value)
- def __str__(self):
- return str(self._value)
+ def _repr(self):
+ return repr(self._value)
def __lt__(self, other):
if not isinstance(other, numbers.Number):
value = int(value)
- if self.field_type.is_signed:
+ if self.field_class.is_signed:
utils._check_int64(value)
else:
utils._check_uint64(value)
@property
def _value(self):
- if self.field_type.is_signed:
- ret, value = native_bt.ctf_field_signed_integer_get_value(self._ptr)
+ if self.field_class.is_signed:
+ ret, value = native_bt.field_signed_integer_get_value(self._ptr)
else:
- ret, value = native_bt.ctf_field_unsigned_integer_get_value(self._ptr)
+ ret, value = native_bt.field_unsigned_integer_get_value(self._ptr)
if ret < 0:
if not self.is_set:
def _set_value(self, value):
value = self._value_to_int(value)
- if self.field_type.is_signed:
- ret = native_bt.ctf_field_signed_integer_set_value(self._ptr, value)
+ if self.field_class.is_signed:
+ ret = native_bt.field_signed_integer_set_value(self._ptr, value)
else:
- ret = native_bt.ctf_field_unsigned_integer_set_value(self._ptr, value)
+ ret = native_bt.field_unsigned_integer_set_value(self._ptr, value)
utils._handle_ret(ret, "cannot set integer field object's value")
value = property(fset=_set_value)
+
class _FloatingPointNumberField(_RealField):
_NAME = 'Floating point number'
@property
def _value(self):
- ret, value = native_bt.ctf_field_floating_point_get_value(self._ptr)
+ ret, value = native_bt.field_floating_point_get_value(self._ptr)
if ret < 0:
if not self.is_set:
def _set_value(self, value):
value = self._value_to_float(value)
- ret = native_bt.ctf_field_floating_point_set_value(self._ptr, value)
+ ret = native_bt.field_floating_point_set_value(self._ptr, value)
utils._handle_ret(ret, "cannot set floating point number field object's value")
value = property(fset=_set_value)
+
class _EnumerationField(_IntegerField):
_NAME = 'Enumeration'
@property
def integer_field(self):
- int_field_ptr = native_bt.ctf_field_enumeration_get_container(self._ptr)
+ int_field_ptr = native_bt.field_enumeration_get_container(self._ptr)
assert(int_field_ptr)
return _create_from_ptr(int_field_ptr)
def _set_value(self, value):
self.integer_field.value = value
+ def _repr(self):
+ labels = [repr(v.name) for v in self.mappings]
+ return '{} ({})'.format(self._value, ', '.join(labels))
+
value = property(fset=_set_value)
@property
@property
def mappings(self):
- iter_ptr = native_bt.ctf_field_enumeration_get_mappings(self._ptr)
+ iter_ptr = native_bt.field_enumeration_get_mappings(self._ptr)
assert(iter_ptr)
- return bt2.field_types._EnumerationFieldTypeMappingIterator(iter_ptr,
- self.field_type.is_signed)
+ return bt2.field_class._EnumerationFieldClassMappingIterator(iter_ptr,
+ self.field_class.is_signed)
@functools.total_ordering
@property
def _value(self):
- value = native_bt.ctf_field_string_get_value(self._ptr)
+ value = native_bt.field_string_get_value(self._ptr)
return value
def _set_value(self, value):
value = self._value_to_str(value)
- ret = native_bt.ctf_field_string_set_value(self._ptr, value)
+ ret = native_bt.field_string_set_value(self._ptr, value)
utils._handle_ret(ret, "cannot set string field object's value")
value = property(fset=_set_value)
def __bool__(self):
return bool(self._value)
+ def _repr(self):
+ return repr(self._value)
+
def __str__(self):
- return self._value
+ return self._value if self.is_set else repr(self)
def __getitem__(self, index):
return self._value[index]
def __iadd__(self, value):
value = self._value_to_str(value)
- ret = native_bt.ctf_field_string_append(self._ptr, value)
+ ret = native_bt.field_string_append(self._ptr, value)
utils._handle_ret(ret, "cannot append to string field object's value")
return self
_NAME = 'Structure'
def _count(self):
- return len(self.field_type)
+ return len(self.field_class)
def __getitem__(self, key):
utils._check_str(key)
- ptr = native_bt.ctf_field_structure_get_field_by_name(self._ptr, key)
+ ptr = native_bt.field_structure_get_field_by_name(self._ptr, key)
if ptr is None:
raise KeyError(key)
if index >= len(self):
raise IndexError
- field_ptr = native_bt.ctf_field_structure_get_field_by_index(self._ptr, index)
+ field_ptr = native_bt.field_structure_get_field_by_index(self._ptr, index)
assert(field_ptr)
return _create_from_ptr(field_ptr)
def __iter__(self):
# same name iterator
- return iter(self.field_type)
+ return iter(self.field_class)
def _spec_eq(self, other):
try:
value = property(fset=_set_value)
+ def _repr(self):
+ items = ['{}: {}'.format(repr(k), repr(v)) for k, v in self.items()]
+ return '{{{}}}'.format(', '.join(items))
+
+
class _VariantField(_Field):
_NAME = 'Variant'
@property
def tag_field(self):
- field_ptr = native_bt.ctf_field_variant_get_tag(self._ptr)
+ field_ptr = native_bt.field_variant_get_tag(self._ptr)
if field_ptr is None:
return
def field(self, tag_field=None):
if tag_field is None:
- field_ptr = native_bt.ctf_field_variant_get_current_field(self._ptr)
+ field_ptr = native_bt.field_variant_get_current_field(self._ptr)
if field_ptr is None:
return
else:
utils._check_type(tag_field, _EnumerationField)
- field_ptr = native_bt.ctf_field_variant_get_field(self._ptr, tag_field._ptr)
+ field_ptr = native_bt.field_variant_get_field(self._ptr, tag_field._ptr)
utils._handle_ptr(field_ptr, "cannot select variant field object's field")
return _create_from_ptr(field_ptr)
def __bool__(self):
return bool(self.selected_field)
+ def __str__(self):
+ return str(self.selected_field) if self.is_set else repr(self)
+
+ def _repr(self):
+ return repr(self.selected_field)
+
@property
def _value(self):
if self.selected_field is not None:
value = property(fset=_set_value)
+
class _ArraySequenceField(_ContainerField, collections.abc.MutableSequence):
def __getitem__(self, index):
if not isinstance(index, numbers.Integral):
def _value(self):
return [field._value for field in self]
+ def _repr(self):
+ return '[{}]'.format(', '.join([repr(v) for v in self]))
+
class _ArrayField(_ArraySequenceField):
_NAME = 'Array'
def _count(self):
- return self.field_type.length
+ return self.field_class.length
def _get_field_ptr_at_index(self, index):
- return native_bt.ctf_field_array_get_field(self._ptr, index)
+ return native_bt.field_array_get_field(self._ptr, index)
def _set_value(self, values):
if len(self) != len(values):
@property
def length_field(self):
- field_ptr = native_bt.ctf_field_sequence_get_length(self._ptr)
+ field_ptr = native_bt.field_sequence_get_length(self._ptr)
if field_ptr is None:
return
return _create_from_ptr(field_ptr)
@length_field.setter
def length_field(self, length_field):
utils._check_type(length_field, _IntegerField)
- ret = native_bt.ctf_field_sequence_set_length(self._ptr, length_field._ptr)
+ ret = native_bt.field_sequence_set_length(self._ptr, length_field._ptr)
utils._handle_ret(ret, "cannot set sequence field object's length field")
def _get_field_ptr_at_index(self, index):
- return native_bt.ctf_field_sequence_get_field(self._ptr, index)
+ return native_bt.field_sequence_get_field(self._ptr, index)
def _set_value(self, values):
original_length_field = self.length_field
if len(values) != self.length_field:
if self.length_field is not None:
- length_ft = self.length_field.field_type
+ length_fc = self.length_field.field_class
else:
- length_ft = bt2.IntegerFieldType(size=64, is_signed=False)
- self.length_field = length_ft(len(values))
+ length_fc = bt2.IntegerFieldClass(size=64, is_signed=False)
+ self.length_field = length_fc(len(values))
try:
for index, value in enumerate(values):
value = property(fset=_set_value)
+
_TYPE_ID_TO_OBJ = {
- native_bt.CTF_FIELD_TYPE_ID_INTEGER: _IntegerField,
- native_bt.CTF_FIELD_TYPE_ID_FLOAT: _FloatingPointNumberField,
- native_bt.CTF_FIELD_TYPE_ID_ENUM: _EnumerationField,
- native_bt.CTF_FIELD_TYPE_ID_STRING: _StringField,
- native_bt.CTF_FIELD_TYPE_ID_STRUCT: _StructureField,
- native_bt.CTF_FIELD_TYPE_ID_ARRAY: _ArrayField,
- native_bt.CTF_FIELD_TYPE_ID_SEQUENCE: _SequenceField,
- native_bt.CTF_FIELD_TYPE_ID_VARIANT: _VariantField,
}