lib: standardize variant field option function names
[babeltrace.git] / src / bindings / python / bt2 / bt2 / field.py
CommitLineData
81447b5b
PP
1# The MIT License (MIT)
2#
811644b8 3# Copyright (c) 2017 Philippe Proulx <pproulx@efficios.com>
81447b5b
PP
4#
5# Permission is hereby granted, free of charge, to any person obtaining a copy
6# of this software and associated documentation files (the "Software"), to deal
7# in the Software without restriction, including without limitation the rights
8# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9# copies of the Software, and to permit persons to whom the Software is
10# furnished to do so, subject to the following conditions:
11#
12# The above copyright notice and this permission notice shall be included in
13# all copies or substantial portions of the Software.
14#
15# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
21# THE SOFTWARE.
22
23from bt2 import native_bt, object, utils
3fb99a22 24from bt2 import field_class as bt2_field_class
81447b5b
PP
25import collections.abc
26import functools
27import numbers
28import math
81447b5b
PP
29
30
f0a42b33
FD
31def _create_field_from_ptr_template(
32 object_map, ptr, owner_ptr, owner_get_ref, owner_put_ref
33):
34
2ae9f48c 35 field_class_ptr = native_bt.field_borrow_class_const(ptr)
2ae9f48c 36 typeid = native_bt.field_class_get_type(field_class_ptr)
f0a42b33 37 field = object_map[typeid]._create_from_ptr_and_get_ref(
cfbd7cf3
FD
38 ptr, owner_ptr, owner_get_ref, owner_put_ref
39 )
81447b5b
PP
40 return field
41
42
f0a42b33
FD
43def _create_field_from_ptr(ptr, owner_ptr, owner_get_ref, owner_put_ref):
44 return _create_field_from_ptr_template(
45 _TYPE_ID_TO_OBJ, ptr, owner_ptr, owner_get_ref, owner_put_ref
46 )
47
48
49def _create_field_from_const_ptr(ptr, owner_ptr, owner_get_ref, owner_put_ref):
50 return _create_field_from_ptr_template(
51 _TYPE_ID_TO_CONST_OBJ, ptr, owner_ptr, owner_get_ref, owner_put_ref
52 )
53
54
cec0261d
PP
55# Get the "effective" field of `field`. If `field` is a variant, return
56# the currently selected field. If `field` is an option, return the
57# content field. If `field` is of any other type, return `field`
1eccc498 58# directly.
81447b5b 59
cfbd7cf3 60
1eccc498 61def _get_leaf_field(field):
f0a42b33 62 if isinstance(field, _VariantFieldConst):
cec0261d 63 return _get_leaf_field(field.selected_option)
1eccc498 64
f0a42b33 65 if isinstance(field, _OptionFieldConst):
cec0261d
PP
66 return _get_leaf_field(field.field)
67
68 return field
81447b5b 69
e1c6bebd 70
f0a42b33
FD
71class _FieldConst(object._UniqueObject):
72 _create_field_from_ptr = staticmethod(_create_field_from_const_ptr)
73 _create_field_class_from_ptr_and_get_ref = staticmethod(
74 bt2_field_class._create_field_class_from_const_ptr_and_get_ref
75 )
76 _borrow_class_ptr = staticmethod(native_bt.field_borrow_class_const)
77
1eccc498 78 def __eq__(self, other):
e1c6bebd
JG
79 other = _get_leaf_field(other)
80 return self._spec_eq(other)
81
81447b5b 82 @property
d8e2073c 83 def cls(self):
f0a42b33 84 field_class_ptr = self._borrow_class_ptr(self._ptr)
838a5a52 85 assert field_class_ptr is not None
f0a42b33 86 return self._create_field_class_from_ptr_and_get_ref(field_class_ptr)
81447b5b 87
12bf0d88
JG
88 def _repr(self):
89 raise NotImplementedError
90
91 def __repr__(self):
1eccc498 92 return self._repr()
12bf0d88 93
81447b5b 94
f0a42b33
FD
95class _Field(_FieldConst):
96 _create_field_from_ptr = staticmethod(_create_field_from_ptr)
97 _create_field_class_from_ptr_and_get_ref = staticmethod(
98 bt2_field_class._create_field_class_from_ptr_and_get_ref
99 )
100 _borrow_class_ptr = staticmethod(native_bt.field_borrow_class)
101
102
103class _BitArrayFieldConst(_FieldConst):
104 _NAME = 'Const bit array'
ead8c3d4
PP
105
106 @property
107 def value_as_integer(self):
108 return native_bt.field_bit_array_get_value_as_integer(self._ptr)
109
ead8c3d4
PP
110 def _spec_eq(self, other):
111 if type(other) is not type(self):
112 return False
113
114 return self.value_as_integer == other.value_as_integer
115
116 def _repr(self):
117 return repr(self.value_as_integer)
118
119 def __str__(self):
120 return str(self.value_as_integer)
121
122 def __len__(self):
d8e2073c 123 return self.cls.length
ead8c3d4
PP
124
125
f0a42b33
FD
126class _BitArrayField(_BitArrayFieldConst, _Field):
127 _NAME = 'Bit array'
128
129 def _value_as_integer(self, value):
130 utils._check_uint64(value)
131 native_bt.field_bit_array_set_value_as_integer(self._ptr, value)
132
133 value_as_integer = property(
134 fget=_BitArrayFieldConst.value_as_integer.fget, fset=_value_as_integer
135 )
136
137
81447b5b 138@functools.total_ordering
f0a42b33 139class _NumericFieldConst(_FieldConst):
81447b5b
PP
140 @staticmethod
141 def _extract_value(other):
f0a42b33 142 if isinstance(other, _BoolFieldConst) or isinstance(other, bool):
aae30e61 143 return bool(other)
81447b5b
PP
144
145 if isinstance(other, numbers.Integral):
146 return int(other)
147
148 if isinstance(other, numbers.Real):
149 return float(other)
150
151 if isinstance(other, numbers.Complex):
152 return complex(other)
153
cfbd7cf3
FD
154 raise TypeError(
155 "'{}' object is not a number object".format(other.__class__.__name__)
156 )
81447b5b
PP
157
158 def __int__(self):
e1c6bebd 159 return int(self._value)
81447b5b
PP
160
161 def __float__(self):
e1c6bebd 162 return float(self._value)
81447b5b 163
12bf0d88 164 def _repr(self):
5abb9e33 165 return repr(self._value)
81447b5b
PP
166
167 def __lt__(self, other):
168 if not isinstance(other, numbers.Number):
cfbd7cf3
FD
169 raise TypeError(
170 'unorderable types: {}() < {}()'.format(
171 self.__class__.__name__, other.__class__.__name__
172 )
173 )
81447b5b 174
09a926c1 175 return self._value < self._extract_value(other)
81447b5b 176
e1c6bebd 177 def _spec_eq(self, other):
f11ed062
PP
178 try:
179 return self._value == self._extract_value(other)
4c4935bf 180 except Exception:
f11ed062 181 return False
81447b5b 182
dda659b3
FD
183 def __hash__(self):
184 return hash(self._value)
185
81447b5b 186 def __rmod__(self, other):
e1c6bebd 187 return self._extract_value(other) % self._value
81447b5b
PP
188
189 def __mod__(self, other):
e1c6bebd 190 return self._value % self._extract_value(other)
81447b5b
PP
191
192 def __rfloordiv__(self, other):
e1c6bebd 193 return self._extract_value(other) // self._value
81447b5b
PP
194
195 def __floordiv__(self, other):
e1c6bebd 196 return self._value // self._extract_value(other)
81447b5b
PP
197
198 def __round__(self, ndigits=None):
199 if ndigits is None:
e1c6bebd 200 return round(self._value)
81447b5b 201 else:
e1c6bebd 202 return round(self._value, ndigits)
81447b5b
PP
203
204 def __ceil__(self):
e1c6bebd 205 return math.ceil(self._value)
81447b5b
PP
206
207 def __floor__(self):
e1c6bebd 208 return math.floor(self._value)
81447b5b
PP
209
210 def __trunc__(self):
e1c6bebd 211 return int(self._value)
81447b5b
PP
212
213 def __abs__(self):
e1c6bebd 214 return abs(self._value)
81447b5b
PP
215
216 def __add__(self, other):
e1c6bebd 217 return self._value + self._extract_value(other)
81447b5b
PP
218
219 def __radd__(self, other):
220 return self.__add__(other)
221
222 def __neg__(self):
e1c6bebd 223 return -self._value
81447b5b
PP
224
225 def __pos__(self):
e1c6bebd 226 return +self._value
81447b5b
PP
227
228 def __mul__(self, other):
e1c6bebd 229 return self._value * self._extract_value(other)
81447b5b
PP
230
231 def __rmul__(self, other):
232 return self.__mul__(other)
233
234 def __truediv__(self, other):
e1c6bebd 235 return self._value / self._extract_value(other)
81447b5b
PP
236
237 def __rtruediv__(self, other):
e1c6bebd 238 return self._extract_value(other) / self._value
81447b5b
PP
239
240 def __pow__(self, exponent):
e1c6bebd 241 return self._value ** self._extract_value(exponent)
81447b5b
PP
242
243 def __rpow__(self, base):
e1c6bebd 244 return self._extract_value(base) ** self._value
81447b5b 245
81447b5b 246
f0a42b33 247class _NumericField(_NumericFieldConst, _Field):
dda659b3
FD
248 def __hash__(self):
249 # Non const field are not hashable as their value may be modified
250 # without changing the underlying Python object.
251 raise TypeError('unhashable type: \'{}\''.format(self._NAME))
f0a42b33
FD
252
253
254class _IntegralFieldConst(_NumericFieldConst, numbers.Integral):
81447b5b 255 def __lshift__(self, other):
e1c6bebd 256 return self._value << self._extract_value(other)
81447b5b
PP
257
258 def __rlshift__(self, other):
e1c6bebd 259 return self._extract_value(other) << self._value
81447b5b
PP
260
261 def __rshift__(self, other):
e1c6bebd 262 return self._value >> self._extract_value(other)
81447b5b
PP
263
264 def __rrshift__(self, other):
e1c6bebd 265 return self._extract_value(other) >> self._value
81447b5b
PP
266
267 def __and__(self, other):
e1c6bebd 268 return self._value & self._extract_value(other)
81447b5b
PP
269
270 def __rand__(self, other):
e1c6bebd 271 return self._extract_value(other) & self._value
81447b5b
PP
272
273 def __xor__(self, other):
e1c6bebd 274 return self._value ^ self._extract_value(other)
81447b5b
PP
275
276 def __rxor__(self, other):
e1c6bebd 277 return self._extract_value(other) ^ self._value
81447b5b
PP
278
279 def __or__(self, other):
e1c6bebd 280 return self._value | self._extract_value(other)
81447b5b
PP
281
282 def __ror__(self, other):
e1c6bebd 283 return self._extract_value(other) | self._value
81447b5b
PP
284
285 def __invert__(self):
e1c6bebd 286 return ~self._value
81447b5b 287
81447b5b 288
f0a42b33
FD
289class _IntegralField(_IntegralFieldConst, _NumericField):
290 pass
291
292
293class _BoolFieldConst(_IntegralFieldConst, _FieldConst):
294 _NAME = 'Const boolean'
aae30e61
PP
295
296 def __bool__(self):
297 return self._value
298
f0a42b33
FD
299 @classmethod
300 def _value_to_bool(cls, value):
301 if isinstance(value, _BoolFieldConst):
aae30e61
PP
302 value = value._value
303
304 if not isinstance(value, bool):
305 raise TypeError(
f0a42b33 306 "'{}' object is not a 'bool', '_BoolFieldConst', or '_BoolField' object".format(
aae30e61
PP
307 value.__class__
308 )
309 )
310
311 return value
312
313 @property
314 def _value(self):
315 return bool(native_bt.field_bool_get_value(self._ptr))
316
f0a42b33
FD
317
318class _BoolField(_BoolFieldConst, _IntegralField, _Field):
319 _NAME = 'Boolean'
320
aae30e61
PP
321 def _set_value(self, value):
322 value = self._value_to_bool(value)
323 native_bt.field_bool_set_value(self._ptr, value)
324
325 value = property(fset=_set_value)
326
327
f0a42b33 328class _IntegerFieldConst(_IntegralFieldConst, _FieldConst):
81447b5b
PP
329 pass
330
331
f0a42b33 332class _IntegerField(_IntegerFieldConst, _IntegralField, _Field):
1198c635
FD
333 def _check_range(self, value):
334 if not (value >= self._lower_bound and value <= self._upper_bound):
335 raise ValueError(
336 "Value {} is outside valid range [{}, {}]".format(
337 value, self._lower_bound, self._upper_bound
338 )
339 )
f0a42b33
FD
340
341
342class _UnsignedIntegerFieldConst(_IntegerFieldConst, _FieldConst):
343 _NAME = 'Const unsigned integer'
1eccc498 344
f0a42b33
FD
345 @classmethod
346 def _value_to_int(cls, value):
25bb9bab
PP
347 if not isinstance(value, numbers.Integral):
348 raise TypeError('expecting an integral number object')
81447b5b 349
1198c635 350 return int(value)
81447b5b
PP
351
352 @property
e1c6bebd 353 def _value(self):
9c08c816 354 return native_bt.field_integer_unsigned_get_value(self._ptr)
81447b5b 355
f0a42b33
FD
356
357class _UnsignedIntegerField(_UnsignedIntegerFieldConst, _IntegerField, _Field):
358 _NAME = 'Unsigned integer'
359
2ae9f48c
SM
360 def _set_value(self, value):
361 value = self._value_to_int(value)
1198c635
FD
362
363 self._check_range(value)
364
9c08c816 365 native_bt.field_integer_unsigned_set_value(self._ptr, value)
2ae9f48c
SM
366
367 value = property(fset=_set_value)
368
1198c635
FD
369 @property
370 def _lower_bound(self):
371 return 0
372
373 @property
374 def _upper_bound(self):
375 return (2 ** self.cls.field_value_range) - 1
376
2ae9f48c 377
f0a42b33
FD
378class _SignedIntegerFieldConst(_IntegerFieldConst, _FieldConst):
379 _NAME = 'Const signed integer'
1eccc498 380
f0a42b33
FD
381 @classmethod
382 def _value_to_int(cls, value):
25bb9bab
PP
383 if not isinstance(value, numbers.Integral):
384 raise TypeError('expecting an integral number object')
e1c6bebd 385
1198c635 386 return int(value)
81447b5b 387
2ae9f48c
SM
388 @property
389 def _value(self):
9c08c816 390 return native_bt.field_integer_signed_get_value(self._ptr)
2ae9f48c 391
f0a42b33
FD
392
393class _SignedIntegerField(_SignedIntegerFieldConst, _IntegerField, _Field):
394 _NAME = 'Signed integer'
395
e1c6bebd 396 def _set_value(self, value):
81447b5b 397 value = self._value_to_int(value)
1198c635
FD
398
399 self._check_range(value)
400
9c08c816 401 native_bt.field_integer_signed_set_value(self._ptr, value)
81447b5b 402
e1c6bebd 403 value = property(fset=_set_value)
81447b5b 404
1198c635
FD
405 @property
406 def _lower_bound(self):
407 return -1 * (2 ** (self.cls.field_value_range - 1))
408
409 @property
410 def _upper_bound(self):
411 return (2 ** (self.cls.field_value_range - 1)) - 1
412
0b03f63e 413
f0a42b33
FD
414class _RealFieldConst(_NumericFieldConst, numbers.Real):
415 _NAME = 'Const real'
81447b5b 416
f0a42b33
FD
417 @classmethod
418 def _value_to_float(cls, value):
81447b5b
PP
419 if not isinstance(value, numbers.Real):
420 raise TypeError("expecting a real number object")
421
422 return float(value)
423
fe4df857
FD
424
425class _SinglePrecisionRealFieldConst(_RealFieldConst):
426 _NAME = 'Const single-precision real'
427
428 @property
429 def _value(self):
430 return native_bt.field_real_single_precision_get_value(self._ptr)
431
432
433class _DoublePrecisionRealFieldConst(_RealFieldConst):
434 _NAME = 'Const double-precision real'
435
81447b5b 436 @property
e1c6bebd 437 def _value(self):
fe4df857 438 return native_bt.field_real_double_precision_get_value(self._ptr)
81447b5b 439
f0a42b33
FD
440
441class _RealField(_RealFieldConst, _NumericField):
442 _NAME = 'Real'
443
fe4df857
FD
444
445class _SinglePrecisionRealField(_SinglePrecisionRealFieldConst, _RealField):
446 _NAME = 'Single-precision real'
447
448 def _set_value(self, value):
449 value = self._value_to_float(value)
450 native_bt.field_real_single_precision_set_value(self._ptr, value)
451
452 value = property(fset=_set_value)
453
454
455class _DoublePrecisionRealField(_DoublePrecisionRealFieldConst, _RealField):
456 _NAME = 'Double-precision real'
457
e1c6bebd 458 def _set_value(self, value):
81447b5b 459 value = self._value_to_float(value)
fe4df857 460 native_bt.field_real_double_precision_set_value(self._ptr, value)
81447b5b 461
e1c6bebd 462 value = property(fset=_set_value)
81447b5b 463
0b03f63e 464
f0a42b33 465class _EnumerationFieldConst(_IntegerFieldConst):
1eccc498
SM
466 def _repr(self):
467 return '{} ({})'.format(self._value, ', '.join(self.labels))
81447b5b
PP
468
469 @property
1eccc498 470 def labels(self):
d24d5663 471 status, labels = self._get_mapping_labels(self._ptr)
cfbd7cf3 472 utils._handle_func_status(status, "cannot get label for enumeration field")
81447b5b 473
1eccc498
SM
474 assert labels is not None
475 return labels
81447b5b 476
4addd228 477
f0a42b33
FD
478class _EnumerationField(_EnumerationFieldConst, _IntegerField):
479 pass
480
481
482class _UnsignedEnumerationFieldConst(
483 _EnumerationFieldConst, _UnsignedIntegerFieldConst
484):
485 _NAME = 'Const unsigned Enumeration'
cfbd7cf3 486 _get_mapping_labels = staticmethod(
9c08c816 487 native_bt.field_enumeration_unsigned_get_mapping_labels
cfbd7cf3 488 )
e1c6bebd 489
e1c6bebd 490
f0a42b33
FD
491class _UnsignedEnumerationField(
492 _UnsignedEnumerationFieldConst, _EnumerationField, _UnsignedIntegerField
493):
494 _NAME = 'Unsigned enumeration'
495
496
497class _SignedEnumerationFieldConst(_EnumerationFieldConst, _SignedIntegerFieldConst):
498 _NAME = 'Const signed Enumeration'
cfbd7cf3 499 _get_mapping_labels = staticmethod(
9c08c816 500 native_bt.field_enumeration_signed_get_mapping_labels
cfbd7cf3 501 )
81447b5b
PP
502
503
f0a42b33
FD
504class _SignedEnumerationField(
505 _SignedEnumerationFieldConst, _EnumerationField, _SignedIntegerField
506):
507 _NAME = 'Signed enumeration'
508
509
81447b5b 510@functools.total_ordering
f0a42b33
FD
511class _StringFieldConst(_FieldConst):
512 _NAME = 'Const string'
81447b5b 513
f0a42b33
FD
514 @classmethod
515 def _value_to_str(cls, value):
516 if isinstance(value, _StringFieldConst):
e1c6bebd 517 value = value._value
81447b5b
PP
518
519 if not isinstance(value, str):
520 raise TypeError("expecting a 'str' object")
521
522 return value
523
524 @property
e1c6bebd 525 def _value(self):
1eccc498 526 return native_bt.field_string_get_value(self._ptr)
81447b5b 527
e1c6bebd 528 def _spec_eq(self, other):
81447b5b 529 try:
f11ed062 530 return self._value == self._value_to_str(other)
4c4935bf 531 except Exception:
81447b5b
PP
532 return False
533
81447b5b 534 def __lt__(self, other):
e1c6bebd 535 return self._value < self._value_to_str(other)
81447b5b
PP
536
537 def __bool__(self):
e1c6bebd 538 return bool(self._value)
81447b5b 539
dda659b3
FD
540 def __hash__(self):
541 return hash(self._value)
542
12bf0d88 543 def _repr(self):
d623d2e9
JG
544 return repr(self._value)
545
81447b5b 546 def __str__(self):
1eccc498 547 return str(self._value)
81447b5b
PP
548
549 def __getitem__(self, index):
e1c6bebd 550 return self._value[index]
81447b5b
PP
551
552 def __len__(self):
1eccc498 553 return native_bt.field_string_get_length(self._ptr)
81447b5b 554
f0a42b33
FD
555
556class _StringField(_StringFieldConst, _Field):
557 _NAME = 'String'
558
559 def _set_value(self, value):
560 value = self._value_to_str(value)
561 native_bt.field_string_set_value(self._ptr, value)
562
563 value = property(fset=_set_value)
564
81447b5b
PP
565 def __iadd__(self, value):
566 value = self._value_to_str(value)
d24d5663 567 status = native_bt.field_string_append(self._ptr, value)
cfbd7cf3
FD
568 utils._handle_func_status(
569 status, "cannot append to string field object's value"
570 )
81447b5b
PP
571 return self
572
dda659b3
FD
573 def __hash__(self):
574 # Non const field are not hashable as their value may be modified
575 # without changing the underlying Python object.
576 raise TypeError('unhashable type: \'{}\''.format(self._NAME))
577
81447b5b 578
f0a42b33 579class _ContainerFieldConst(_FieldConst):
81447b5b
PP
580 def __bool__(self):
581 return len(self) != 0
582
f0a42b33
FD
583 def _count(self):
584 return len(self.cls)
585
81447b5b
PP
586 def __len__(self):
587 count = self._count()
1eccc498 588 assert count >= 0
81447b5b
PP
589 return count
590
591 def __delitem__(self, index):
592 raise NotImplementedError
593
f0a42b33
FD
594 def __setitem__(self, index, value):
595 raise TypeError(
596 '\'{}\' object does not support item assignment'.format(self.__class__)
597 )
81447b5b 598
81447b5b 599
f0a42b33
FD
600class _ContainerField(_ContainerFieldConst, _Field):
601 pass
81447b5b 602
81447b5b 603
f0a42b33
FD
604class _StructureFieldConst(_ContainerFieldConst, collections.abc.Mapping):
605 _NAME = 'Const structure'
606 _borrow_member_field_ptr_by_index = staticmethod(
607 native_bt.field_structure_borrow_member_field_by_index_const
608 )
609 _borrow_member_field_ptr_by_name = staticmethod(
610 native_bt.field_structure_borrow_member_field_by_name_const
611 )
612
613 def _count(self):
614 return len(self.cls)
81447b5b 615
81447b5b
PP
616 def __iter__(self):
617 # same name iterator
d8e2073c 618 return iter(self.cls)
81447b5b 619
e1c6bebd 620 def _spec_eq(self, other):
f11ed062
PP
621 if not isinstance(other, collections.abc.Mapping):
622 return False
81447b5b 623
f11ed062
PP
624 if len(self) != len(other):
625 # early mismatch
626 return False
81447b5b 627
f11ed062
PP
628 for self_key in self:
629 if self_key not in other:
630 return False
81447b5b 631
f11ed062
PP
632 if self[self_key] != other[self_key]:
633 return False
e1c6bebd 634
f11ed062 635 return True
81447b5b 636
12bf0d88 637 def _repr(self):
ac7e2dc6
JG
638 items = ['{}: {}'.format(repr(k), repr(v)) for k, v in self.items()]
639 return '{{{}}}'.format(', '.join(items))
640
1eccc498
SM
641 def __getitem__(self, key):
642 utils._check_str(key)
f0a42b33 643 field_ptr = self._borrow_member_field_ptr_by_name(self._ptr, key)
0b03f63e 644
1eccc498
SM
645 if field_ptr is None:
646 raise KeyError(key)
81447b5b 647
f0a42b33 648 return self._create_field_from_ptr(
cfbd7cf3
FD
649 field_ptr, self._owner_ptr, self._owner_get_ref, self._owner_put_ref
650 )
811644b8 651
1eccc498
SM
652 def member_at_index(self, index):
653 utils._check_uint64(index)
811644b8 654
1eccc498
SM
655 if index >= len(self):
656 raise IndexError
f0a42b33 657 field_ptr = self._borrow_member_field_ptr_by_index(self._ptr, index)
1eccc498 658 assert field_ptr is not None
f0a42b33 659 return self._create_field_from_ptr(
cfbd7cf3
FD
660 field_ptr, self._owner_ptr, self._owner_get_ref, self._owner_put_ref
661 )
1eccc498
SM
662
663
f0a42b33
FD
664class _StructureField(
665 _StructureFieldConst, _ContainerField, collections.abc.MutableMapping
666):
667 _NAME = 'Structure'
668 _borrow_member_field_ptr_by_index = staticmethod(
669 native_bt.field_structure_borrow_member_field_by_index
670 )
671 _borrow_member_field_ptr_by_name = staticmethod(
672 native_bt.field_structure_borrow_member_field_by_name
673 )
674
675 def __setitem__(self, key, value):
676 # raises if key is somehow invalid
677 field = self[key]
678
679 # the field's property does the appropriate conversion or raises
680 # the appropriate exception
681 field.value = value
682
683 def _set_value(self, values):
684 try:
685 for key, value in values.items():
686 self[key].value = value
687 except Exception:
688 raise
689
690 value = property(fset=_set_value)
691
692
693class _OptionFieldConst(_FieldConst):
694 _NAME = 'Const option'
695 _borrow_field_ptr = staticmethod(native_bt.field_option_borrow_field_const)
cec0261d
PP
696
697 @property
698 def field(self):
f0a42b33 699 field_ptr = self._borrow_field_ptr(self._ptr)
cec0261d
PP
700
701 if field_ptr is None:
702 return
703
f0a42b33 704 return self._create_field_from_ptr(
cec0261d
PP
705 field_ptr, self._owner_ptr, self._owner_get_ref, self._owner_put_ref
706 )
707
708 @property
709 def has_field(self):
710 return self.field is not None
711
cec0261d
PP
712 def _spec_eq(self, other):
713 return _get_leaf_field(self) == other
714
715 def __bool__(self):
716 return self.has_field
717
718 def __str__(self):
719 return str(self.field)
720
721 def _repr(self):
722 return repr(self.field)
723
f0a42b33
FD
724
725class _OptionField(_OptionFieldConst, _Field):
726 _NAME = 'Option'
727 _borrow_field_ptr = staticmethod(native_bt.field_option_borrow_field)
728
729 def _has_field(self, value):
730 utils._check_bool(value)
731 native_bt.field_option_set_has_field(self._ptr, value)
732
733 has_field = property(fget=_OptionFieldConst.has_field.fget, fset=_has_field)
734
cec0261d
PP
735 def _set_value(self, value):
736 self.has_field = True
737 field = self.field
738 assert field is not None
739 field.value = value
740
741 value = property(fset=_set_value)
742
743
f0a42b33
FD
744class _VariantFieldConst(_ContainerFieldConst, _FieldConst):
745 _NAME = 'Const variant'
746 _borrow_selected_option_field_ptr = staticmethod(
747 native_bt.field_variant_borrow_selected_option_field_const
748 )
81447b5b 749
2b9aa00b
FD
750 def _count(self):
751 return len(self.cls)
752
81447b5b 753 @property
1eccc498 754 def selected_option_index(self):
7b4311c1 755 return native_bt.field_variant_get_selected_option_index(self._ptr)
81447b5b 756
1eccc498
SM
757 @property
758 def selected_option(self):
5ae9f1bf
SM
759 # TODO: Is there a way to check if the variant field has a selected_option,
760 # so we can raise an exception instead of hitting a pre-condition check?
761 # If there is something, that check should be added to selected_option_index too.
f0a42b33 762 field_ptr = self._borrow_selected_option_field_ptr(self._ptr)
81447b5b 763
f0a42b33 764 return self._create_field_from_ptr(
cfbd7cf3
FD
765 field_ptr, self._owner_ptr, self._owner_get_ref, self._owner_put_ref
766 )
81447b5b 767
e1c6bebd 768 def _spec_eq(self, other):
f11ed062 769 return _get_leaf_field(self) == other
811644b8
PP
770
771 def __bool__(self):
1eccc498 772 raise NotImplementedError
81447b5b 773
12bf0d88 774 def __str__(self):
1eccc498 775 return str(self.selected_option)
12bf0d88
JG
776
777 def _repr(self):
1eccc498 778 return repr(self.selected_option)
e1c6bebd 779
f0a42b33
FD
780
781class _VariantField(_VariantFieldConst, _ContainerField, _Field):
782 _NAME = 'Variant'
783 _borrow_selected_option_field_ptr = staticmethod(
784 native_bt.field_variant_borrow_selected_option_field
785 )
786
787 def _selected_option_index(self, index):
788 if index < 0 or index >= len(self):
789 raise IndexError('{} field object index is out of range'.format(self._NAME))
790
7b4311c1 791 native_bt.field_variant_select_option_by_index(self._ptr, index)
f0a42b33
FD
792
793 selected_option_index = property(
794 fget=_VariantFieldConst.selected_option_index.fget, fset=_selected_option_index
795 )
796
e1c6bebd 797 def _set_value(self, value):
1eccc498 798 self.selected_option.value = value
e1c6bebd
JG
799
800 value = property(fset=_set_value)
81447b5b 801
0b03f63e 802
f0a42b33
FD
803class _ArrayFieldConst(_ContainerFieldConst, _FieldConst, collections.abc.Sequence):
804 _borrow_element_field_ptr_by_index = staticmethod(
805 native_bt.field_array_borrow_element_field_by_index_const
806 )
807
1eccc498
SM
808 def _get_length(self):
809 return native_bt.field_array_get_length(self._ptr)
810
811 length = property(fget=_get_length)
812
81447b5b
PP
813 def __getitem__(self, index):
814 if not isinstance(index, numbers.Integral):
cfbd7cf3
FD
815 raise TypeError(
816 "'{}' is not an integral number object: invalid index".format(
817 index.__class__.__name__
818 )
819 )
81447b5b
PP
820
821 index = int(index)
822
823 if index < 0 or index >= len(self):
824 raise IndexError('{} field object index is out of range'.format(self._NAME))
825
f0a42b33 826 field_ptr = self._borrow_element_field_ptr_by_index(self._ptr, index)
cfbd7cf3 827 assert field_ptr
f0a42b33 828 return self._create_field_from_ptr(
cfbd7cf3
FD
829 field_ptr, self._owner_ptr, self._owner_get_ref, self._owner_put_ref
830 )
81447b5b 831
81447b5b
PP
832 def insert(self, index, value):
833 raise NotImplementedError
834
e1c6bebd 835 def _spec_eq(self, other):
f11ed062
PP
836 if not isinstance(other, collections.abc.Sequence):
837 return False
7c54e2e7 838
f11ed062
PP
839 if len(self) != len(other):
840 # early mismatch
e1c6bebd 841 return False
7c54e2e7 842
f11ed062
PP
843 for self_elem, other_elem in zip(self, other):
844 if self_elem != other_elem:
845 return False
846
847 return True
848
12bf0d88 849 def _repr(self):
2bc21382
JG
850 return '[{}]'.format(', '.join([repr(v) for v in self]))
851
81447b5b 852
f0a42b33
FD
853class _ArrayField(
854 _ArrayFieldConst, _ContainerField, _Field, collections.abc.MutableSequence
855):
856 _borrow_element_field_ptr_by_index = staticmethod(
857 native_bt.field_array_borrow_element_field_by_index
858 )
859
860 def __setitem__(self, index, value):
861 # raises if index is somehow invalid
862 field = self[index]
863
864 if not isinstance(field, (_NumericField, _StringField)):
865 raise TypeError('can only set the value of a number or string field')
866
867 # the field's property does the appropriate conversion or raises
868 # the appropriate exception
869 field.value = value
870
871
872class _StaticArrayFieldConst(_ArrayFieldConst, _FieldConst):
873 _NAME = 'Const static array'
81447b5b
PP
874
875 def _count(self):
1eccc498 876 return native_bt.field_array_get_length(self._ptr)
81447b5b 877
f0a42b33
FD
878
879class _StaticArrayField(_StaticArrayFieldConst, _ArrayField, _Field):
880 _NAME = 'Static array'
881
e1c6bebd
JG
882 def _set_value(self, values):
883 if len(self) != len(values):
74695c0d
SM
884 raise ValueError(
885 'expected length of value ({}) and array field ({}) to match'.format(
886 len(values), len(self)
887 )
888 )
e1c6bebd 889
1eccc498
SM
890 for index, value in enumerate(values):
891 if value is not None:
892 self[index].value = value
e1c6bebd
JG
893
894 value = property(fset=_set_value)
895
81447b5b 896
f0a42b33
FD
897class _DynamicArrayFieldConst(_ArrayFieldConst, _FieldConst):
898 _NAME = 'Const dynamic array'
81447b5b
PP
899
900 def _count(self):
1eccc498 901 return self.length
81447b5b 902
f0a42b33
FD
903
904class _DynamicArrayField(_DynamicArrayFieldConst, _ArrayField, _Field):
905 _NAME = 'Dynamic array'
906
1eccc498
SM
907 def _set_length(self, length):
908 utils._check_uint64(length)
9c08c816 909 status = native_bt.field_array_dynamic_set_length(self._ptr, length)
d24d5663 910 utils._handle_func_status(status, "cannot set dynamic array length")
81447b5b 911
1eccc498 912 length = property(fget=_ArrayField._get_length, fset=_set_length)
81447b5b 913
e1c6bebd 914 def _set_value(self, values):
1eccc498
SM
915 if len(values) != self.length:
916 self.length = len(values)
e1c6bebd 917
1eccc498
SM
918 for index, value in enumerate(values):
919 if value is not None:
920 self[index].value = value
e1c6bebd
JG
921
922 value = property(fset=_set_value)
81447b5b 923
0b03f63e 924
f0a42b33
FD
925_TYPE_ID_TO_CONST_OBJ = {
926 native_bt.FIELD_CLASS_TYPE_BOOL: _BoolFieldConst,
927 native_bt.FIELD_CLASS_TYPE_BIT_ARRAY: _BitArrayFieldConst,
928 native_bt.FIELD_CLASS_TYPE_UNSIGNED_INTEGER: _UnsignedIntegerFieldConst,
929 native_bt.FIELD_CLASS_TYPE_SIGNED_INTEGER: _SignedIntegerFieldConst,
fe4df857
FD
930 native_bt.FIELD_CLASS_TYPE_SINGLE_PRECISION_REAL: _SinglePrecisionRealFieldConst,
931 native_bt.FIELD_CLASS_TYPE_DOUBLE_PRECISION_REAL: _DoublePrecisionRealFieldConst,
f0a42b33
FD
932 native_bt.FIELD_CLASS_TYPE_UNSIGNED_ENUMERATION: _UnsignedEnumerationFieldConst,
933 native_bt.FIELD_CLASS_TYPE_SIGNED_ENUMERATION: _SignedEnumerationFieldConst,
934 native_bt.FIELD_CLASS_TYPE_STRING: _StringFieldConst,
935 native_bt.FIELD_CLASS_TYPE_STRUCTURE: _StructureFieldConst,
936 native_bt.FIELD_CLASS_TYPE_STATIC_ARRAY: _StaticArrayFieldConst,
81b8fa44
PP
937 native_bt.FIELD_CLASS_TYPE_DYNAMIC_ARRAY_WITHOUT_LENGTH_FIELD: _DynamicArrayFieldConst,
938 native_bt.FIELD_CLASS_TYPE_DYNAMIC_ARRAY_WITH_LENGTH_FIELD: _DynamicArrayFieldConst,
de821fe5
PP
939 native_bt.FIELD_CLASS_TYPE_OPTION_WITHOUT_SELECTOR_FIELD: _OptionFieldConst,
940 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_BOOL_SELECTOR_FIELD: _OptionFieldConst,
941 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_UNSIGNED_INTEGER_SELECTOR_FIELD: _OptionFieldConst,
942 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_SIGNED_INTEGER_SELECTOR_FIELD: _OptionFieldConst,
943 native_bt.FIELD_CLASS_TYPE_VARIANT_WITHOUT_SELECTOR_FIELD: _VariantFieldConst,
944 native_bt.FIELD_CLASS_TYPE_VARIANT_WITH_UNSIGNED_INTEGER_SELECTOR_FIELD: _VariantFieldConst,
945 native_bt.FIELD_CLASS_TYPE_VARIANT_WITH_SIGNED_INTEGER_SELECTOR_FIELD: _VariantFieldConst,
f0a42b33
FD
946}
947
81447b5b 948_TYPE_ID_TO_OBJ = {
aae30e61 949 native_bt.FIELD_CLASS_TYPE_BOOL: _BoolField,
ead8c3d4 950 native_bt.FIELD_CLASS_TYPE_BIT_ARRAY: _BitArrayField,
2ae9f48c
SM
951 native_bt.FIELD_CLASS_TYPE_UNSIGNED_INTEGER: _UnsignedIntegerField,
952 native_bt.FIELD_CLASS_TYPE_SIGNED_INTEGER: _SignedIntegerField,
fe4df857
FD
953 native_bt.FIELD_CLASS_TYPE_SINGLE_PRECISION_REAL: _SinglePrecisionRealField,
954 native_bt.FIELD_CLASS_TYPE_DOUBLE_PRECISION_REAL: _DoublePrecisionRealField,
1eccc498
SM
955 native_bt.FIELD_CLASS_TYPE_UNSIGNED_ENUMERATION: _UnsignedEnumerationField,
956 native_bt.FIELD_CLASS_TYPE_SIGNED_ENUMERATION: _SignedEnumerationField,
2ae9f48c
SM
957 native_bt.FIELD_CLASS_TYPE_STRING: _StringField,
958 native_bt.FIELD_CLASS_TYPE_STRUCTURE: _StructureField,
1eccc498 959 native_bt.FIELD_CLASS_TYPE_STATIC_ARRAY: _StaticArrayField,
81b8fa44
PP
960 native_bt.FIELD_CLASS_TYPE_DYNAMIC_ARRAY_WITHOUT_LENGTH_FIELD: _DynamicArrayField,
961 native_bt.FIELD_CLASS_TYPE_DYNAMIC_ARRAY_WITH_LENGTH_FIELD: _DynamicArrayField,
de821fe5
PP
962 native_bt.FIELD_CLASS_TYPE_OPTION_WITHOUT_SELECTOR_FIELD: _OptionField,
963 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_BOOL_SELECTOR_FIELD: _OptionField,
964 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_UNSIGNED_INTEGER_SELECTOR_FIELD: _OptionField,
965 native_bt.FIELD_CLASS_TYPE_OPTION_WITH_SIGNED_INTEGER_SELECTOR_FIELD: _OptionField,
966 native_bt.FIELD_CLASS_TYPE_VARIANT_WITHOUT_SELECTOR_FIELD: _VariantField,
967 native_bt.FIELD_CLASS_TYPE_VARIANT_WITH_UNSIGNED_INTEGER_SELECTOR_FIELD: _VariantField,
968 native_bt.FIELD_CLASS_TYPE_VARIANT_WITH_SIGNED_INTEGER_SELECTOR_FIELD: _VariantField,
81447b5b 969}
This page took 0.107009 seconds and 4 git commands to generate.