from bt2 import utils
import bt2
import itertools
-import bt2.message_iterator
+from bt2 import message_iterator as bt2_message_iterator
+from bt2 import logging as bt2_logging
+from bt2 import port as bt2_port
import datetime
from collections import namedtuple
import numbers
plugin_name,
class_name,
params=None,
- logging_level=bt2.logging.LoggingLevel.NONE,
+ obj=None,
+ logging_level=bt2_logging.LoggingLevel.NONE,
):
utils._check_str(plugin_name)
utils._check_str(class_name)
self._plugin_name = plugin_name
self._class_name = class_name
self._logging_level = logging_level
+ self._obj = obj
if type(params) is str:
self._params = bt2.create_value({'inputs': [params]})
def params(self):
return self._params
+ @property
+ def obj(self):
+ return self._obj
+
# datetime.datetime or integral to nanoseconds
def _get_ns(obj):
FILTER = 1
-class TraceCollectionMessageIterator(bt2.message_iterator._MessageIterator):
+class TraceCollectionMessageIterator(bt2_message_iterator._MessageIterator):
def __init__(
self,
source_component_specs,
params = {'inputs': inputs}
- # query the port's component for the `trace-info` object which
- # contains the stream intersection range for each exposed
- # trace
- query_exec = bt2.QueryExecutor()
- trace_info_res = query_exec.query(
- src_comp_and_spec.comp.cls, 'trace-info', params
+ # query the port's component for the `babeltrace.trace-info`
+ # object which contains the stream intersection range for each
+ # exposed trace
+ query_exec = bt2.QueryExecutor(
+ src_comp_and_spec.comp.cls, 'babeltrace.trace-info', params
)
+ trace_info_res = query_exec.query()
begin = None
end = None
comp_cls = comp_classes[comp_spec.class_name]
name = self._get_unique_comp_name(comp_spec)
comp = self._graph.add_component(
- comp_cls, name, comp_spec.params, comp_spec.logging_level
+ comp_cls, name, comp_spec.params, comp_spec.obj, comp_spec.logging_level
)
return comp
if not self._connect_ports:
return
- if type(port) is bt2.port._InputPort:
+ if type(port) is bt2_port._InputPort:
return
if component not in [comp.comp for comp in self._src_comps_and_specs]: