From 961597c764419d97c4ac586fa971e3a0d2cffa22 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Mon, 25 Nov 2019 13:58:31 +0100
Subject: codegen/generate-dfa-benchmark: Support measurement of getEnergy call

---
 bin/generate-dfa-benchmark.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'bin')

diff --git a/bin/generate-dfa-benchmark.py b/bin/generate-dfa-benchmark.py
index 429703b..76c1f77 100755
--- a/bin/generate-dfa-benchmark.py
+++ b/bin/generate-dfa-benchmark.py
@@ -381,6 +381,10 @@ if __name__ == '__main__':
 
         repo = Repo('/home/derf/var/projects/multipass/build/repo.acp')
 
+        if 'accounting' in opt and 'getEnergy' not in map(lambda x: x.name, pta.transitions):
+            for state in pta.get_state_names():
+                pta.add_transition(state, state, 'getEnergy')
+
         pta.set_random_energy_model()
 
         if 'accounting' in opt:
-- 
cgit v1.2.3