def __init__(self, traces=None, attr=None, templates=None):
self._value_parser = Group(
IDENTIFIER +
COLON +
IDENTIFIER).setParseAction(
self._parse_value)
self._event_map = {}
self._attr = attr if attr else {}
self.traces = traces
self.templates = templates
class AbstractDataPlotter(object):
"""This is an abstract data plotting Class defining an interface
for the various Plotting Classes"""
__metaclass__ = ABCMeta
def __init__(self, traces=None, attr=None, templates=None):
self._value_parser = Group(
IDENTIFIER +
COLON +
IDENTIFIER).setParseAction(
self._parse_value)
self._event_map = {}
self._attr = attr if attr else {}
self.traces = traces
self.templates = templates
@abstractmethod
def view(self):
"""View the graph"""
raise NotImplementedError("Method Not Implemented")
@abstractmethod
def savefig(self, path):
"""Save the image as a file
:param path: Location of the Saved File
:type path: str
"""
raise NotImplementedError("Method Not Implemented")
def _check_data(self):
"""Internal function to check the received data"""
data = listify(self.traces)
if len(data):
mask = map(lambda x: isinstance(x, DataFrame), data)
data_frame = reduce(lambda x, y: x and y, mask)
sig_or_template = self.templates or "signals" in self._attr
if not data_frame and not sig_or_template:
raise ValueError(
"Cannot understand data. Accepted DataFormats are pandas.DataFrame or trappy.FTrace/BareTrace/SysTrace (with templates)")
elif data_frame and not self._attr["column"]:
raise ValueError("Column not specified for DataFrame input")
else:
raise ValueError("Empty Data received")
def _parse_value(self, tokens):
"""Grammar parser function to parse a signal"""
event, column = tokens[0]
try:
return self._event_map[event], column
except KeyError:
for trace in listify(self.traces):
if event in trace.class_definitions:
self._event_map[event] = trace.class_definitions[event]
return self._event_map[event], column
raise ValueError(
"Event: " +
event +
" not found in Trace Object")
def _describe_signals(self):
"""Internal Function for populating templates and columns
from signals
"""
if "column" in self._attr or self.templates:
raise ValueError("column/templates specified with values")
self._attr["column"] = []
if self.templates is None:
self.templates = []
for value in listify(self._attr["signals"]):
template, column = self._value_parser.parseString(value)[0]
self.templates.append(template)
self._attr["column"].append(column)