bt2: add bit array field class and field support

[babeltrace.git] / src / bindings / python / bt2 / bt2 / trace_class.py

diff --git a/src/bindings/python/bt2/bt2/trace_class.py b/src/bindings/python/bt2/bt2/trace_class.py
index 3ae3addefbcb3126ee21d4baed643632d791e0f7..b9ac172fc0a19de3bd6268e485e4d8313cc73038 100644 (file)
--- a/src/bindings/python/bt2/bt2/trace_class.py
+++ b/src/bindings/python/bt2/bt2/trace_class.py
@@ -194,6 +194,21 @@ class _TraceClass(object._SharedObject, collections.abc.Mapping):
 
         return bt2_field_class._BoolFieldClass._create_from_ptr(field_class_ptr)
 
+    def create_bit_array_field_class(self, length):
+        utils._check_uint64(length)
+
+        if length < 1 or length > 64:
+            raise ValueError(
+                'invalid length {}: expecting a value in the [1, 64] range'.format(
+                    length
+                )
+            )
+
+        field_class_ptr = native_bt.field_class_bit_array_create(self._ptr, length)
+        self._check_field_class_create_status(field_class_ptr, 'bit array')
+
+        return bt2_field_class._BitArrayFieldClass._create_from_ptr(field_class_ptr)
+
     def _create_integer_field_class(
         self, create_func, py_cls, type_name, field_value_range, preferred_display_base
     ):