diff options
Diffstat (limited to 'lib')
| -rwxr-xr-x | lib/automata.py | 9 | ||||
| -rw-r--r-- | lib/harness.py | 2 |
2 files changed, 8 insertions, 3 deletions
diff --git a/lib/automata.py b/lib/automata.py index d36acd7..a80796c 100755 --- a/lib/automata.py +++ b/lib/automata.py @@ -575,6 +575,8 @@ class PTA: kwargs['is_interrupt'] = transition['is_interrupt'] if 'return_value' in transition: kwargs['return_value_handlers'] = transition['return_value'] + if 'codegen' in transition: + kwargs['codegen'] = transition['codegen'] if 'loop' in transition: for state_name in transition['loop']: pta.add_transition(state_name, state_name, trans_name, @@ -784,7 +786,7 @@ class PTA: ret.append((transition, arguments, param)) yield ret - def bfs(self, depth: int = 10, orig_state: str = 'UNINITIALIZED', transition_filter=None, state_filter=None): + def bfs(self, depth: int = 10, orig_state: str = 'UNINITIALIZED', param_dict: dict = None, with_parameters: bool = False, transition_filter=None, state_filter=None): """ Return a generator object for breadth-first search of traces starting at orig_state. @@ -804,6 +806,9 @@ class PTA: :param orig_state: initial state for breadth-first search :param depth: search depth, default 10 + :param param_dict: initial parameter values + :param with_arguments: perform dfs with argument values + :param with_parameters: include parameters in trace? :param transition_filter: If set, only follow a transition if transition_filter(transition object) returns true. Default None. :param state_iflter: If set, only follow a state if state_filter(state_object) returns true. Default None. """ @@ -935,4 +940,4 @@ class PTA: transition.timeout_function = param_model(transition.name, 'timeout')['function'] except KeyError: print('[W] skipping model update of transition {} due to missing data'.format(state.name)) - pass
\ No newline at end of file + pass diff --git a/lib/harness.py b/lib/harness.py index 7a336e5..59b5fa0 100644 --- a/lib/harness.py +++ b/lib/harness.py @@ -330,7 +330,7 @@ class OnboardTimerHarness(TransitionHarness): log_data_target['parameter'][parameter_name].append(parameter_value) def parser_cb(self, line): - #print('[HARNESS] got line {}'.format(line)) + # print('[HARNESS] got line {}'.format(line)) res = re.match(r'\[PTA\] nop=(\S+)/(\S+)', line) if res: self.nop_cycles = int(res.group(1)) |