generate-dfa-benchmark: Add trace filter

1 files changed, 29 insertions, 5 deletions

diff --git a/bin/generate-dfa-benchmark.py b/bin/generate-dfa-benchmark.py
index 2503097..b9790ed 100755
--- a/bin/generate-dfa-benchmark.py
+++ b/bin/generate-dfa-benchmark.py
@@ -35,6 +35,16 @@ from harness import OnboardTimerHarness
 
 opt = {}
 
+def trace_matches_filter(trace: list, trace_filter: list) -> bool:
+    for allowed_trace in trace_filter:
+        if len(trace) < len(allowed_trace):
+            continue
+        different_element_count = len(list(filter(None, map(lambda x,y: x[0].name != y, trace, allowed_trace))))
+        if different_element_count == 0:
+            return True
+    return False
+
+
 if __name__ == '__main__':
 
     try:
@@ -46,6 +56,7 @@ if __name__ == '__main__':
             'run= '
             'sleep= '
             'timer-pin= '
+            'trace-filter= '
         )
         raw_opts, args = getopt.getopt(sys.argv[1:], "", optspec.split(' '))
 
@@ -61,6 +72,12 @@ if __name__ == '__main__':
         if 'sleep' in opt:
             opt['sleep'] = int(opt['sleep'])
 
+        if 'trace-filter' in opt:
+            trace_filter = []
+            for trace in opt['trace-filter'].split():
+                trace_filter.append(trace.split(','))
+            opt['trace-filter'] = trace_filter
+
     except getopt.GetoptError as err:
         print(err)
         sys.exit(2)
@@ -106,14 +123,18 @@ if __name__ == '__main__':
         class_prefix = '{}.'.format(pta.codegen['instance'])
 
     num_transitions = 0
+    num_traces = 0
     for run in pta.dfs(opt['depth'], with_arguments = True, with_parameters = True):
+        if 'trace-filter' in opt and not trace_matches_filter(run, opt['trace-filter']):
+            continue
         outbuf.write(harness.start_run())
         harness.start_trace()
         param = pta.get_initial_param_dict()
         for transition, arguments, parameter in run:
             num_transitions += 1
             harness.append_state(transition.origin.name, param)
-            harness.append_transition(transition.name, param)
+            harness.append_transition(transition.name, param, arguments)
+            param = transition.get_params_after_transition(param, arguments)
             outbuf.write('// {} -> {}\n'.format(transition.origin.name, transition.destination.name))
             if transition.is_interrupt:
                 outbuf.write('// wait for {} interrupt\n'.format(transition.name))
@@ -127,11 +148,15 @@ if __name__ == '__main__':
             if 'sleep' in opt:
                 outbuf.write('arch.delay_ms({:d});\n'.format(opt['sleep']))
 
-        outbuf.write(harness.stop_run())
+        outbuf.write(harness.stop_run(num_traces))
         outbuf.write('\n')
+        num_traces += 1
+
+    if num_transitions == 0:
+        print('DFS returned no traces -- perhaps your trace-filter is too restrictive?', file=sys.stderr)
+        sys.exit(1)
 
     outbuf.write(harness.stop_benchmark())
-    print(harness.traces)
     outbuf.write('}\n')
     outbuf.write('return 0;\n')
     outbuf.write('}\n')
@@ -156,8 +181,7 @@ if __name__ == '__main__':
                 while True:
                     time.sleep(5)
                     slept += 5
-                    if slept < run_timeout:
-                        print('[MON] approx. {:.0f}% done'.format(slept * 100 / run_timeout))
+                    print('[MON] approx. {:.0f}% done'.format(slept * 100 / run_timeout))
             except KeyboardInterrupt:
                 pass
             lines = monitor.get_lines()