add ptalog to dfa benchmark generation

2 files changed, 11 insertions, 0 deletions

diff --git a/bin/generate-dfa-benchmark.py b/bin/generate-dfa-benchmark.py
index 8bb8943..1eb5073 100755
--- a/bin/generate-dfa-benchmark.py
+++ b/bin/generate-dfa-benchmark.py
@@ -39,20 +39,28 @@ if __name__ == '__main__':
     with open(modelfile, 'r') as f:
         pta = PTA.from_json(json.load(f))
 
+    print('ptalog.startBenchmark(0);')
+
     for run in pta.dfs(opt['depth'], with_arguments = True):
+        print('ptalog.reset();')
         for transition, arguments in run:
             print('// {} -> {}'.format(transition.origin.name, transition.destination.name))
+            print('ptalog.passTransition({:d});'.format(pta.get_transition_id(transition)))
             if transition.is_interrupt:
                 print('// wait for {} interrupt'.format(transition.name))
+                print('// TODO add startTransition / stopTransition calls to interrupt routine')
             else:
+                print('ptalog.startTransition();')
                 if 'instance' in opt:
                     print('{}.{}({});'.format(opt['instance'], transition.name, ', '.join(arguments)))
                 else:
                     print('{}({});'.format(transition.name, ', '.join(arguments)))
+                print('ptalog.stopTransition();')
 
             if 'sleep' in opt:
                 print('arch.delay_ms({:d});'.format(opt['sleep']))
 
+        print('ptalog.dump();')
         print()
 
     sys.exit(0)
diff --git a/lib/automata.py b/lib/automata.py
index 37b52e1..7db1ebf 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -411,6 +411,9 @@ class PTA:
         self.transitions.append(new_transition)
         orig_state.add_outgoing_transition(new_transition)
 
+    def get_transition_id(self, transition: Transition) -> int:
+        return self.transitions.index(transition)
+
     def dfs(self, depth: int = 10, orig_state: str = 'UNINITIALIZED', **kwargs):
         """
         Return a generator object for depth-first search starting at orig_state.