+class _EventClockSnapshotsIterator(collections.abc.Iterator):
+ def __init__(self, event_clock_snapshots):
+ self._event_clock_snapshots = event_clock_snapshots
+ self._clock_classes = event_clock_snapshots._event._clock_classes
+ self._at = 0
+
+ def __next__(self):
+ if self._at == len(self._clock_classes):
+ raise StopIteration
+
+ self._at += 1
+ return self._clock_classes[at]
+
+
+class _EventClockSnapshots(collections.abc.Mapping):
+ def __init__(self, event):
+ self._event = event
+
+ def __getitem__(self, clock_class):
+ utils._check_type(clock_class, bt2.ClockClass)
+ clock_snapshot_ptr = native_bt.event_get_clock_snapshot(self._event._ptr,
+ clock_class._ptr)
+
+ if clock_snapshot_ptr is None:
+ return
+
+ clock_snapshot = bt2.clock_snapshot._create_clock_snapshot_from_ptr(clock_snapshot_ptr)
+ return clock_snapshot
+
+ def add(self, clock_snapshot):
+ utils._check_type(clock_snapshot, bt2.clock_snapshot._ClockSnapshot)
+ ret = native_bt.event_set_clock_snapshot(self._ptr,
+ clock_snapshot._ptr)
+ utils._handle_ret(ret, "cannot set event object's clock value")
+
+ def __len__(self):
+ count = len(self._event._clock_classes)
+ assert(count >= 0)
+ return count
+
+ def __iter__(self):
+ return _EventClockSnapshotsIterator(self)
+
+