codegen/generate-dfa-benchmark: Support measurement of getEnergy call

1 files changed, 5 insertions, 0 deletions

diff --git a/lib/codegen.py b/lib/codegen.py
index 45ae8f7..45a6871 100644
--- a/lib/codegen.py
+++ b/lib/codegen.py
@@ -7,6 +7,8 @@ header_template = """
 #ifndef DFATOOL_{name}_H
 #define DFATOOL_{name}_H
 
+#include "stdint.h"
+
 {includes}
 
 class {name}
@@ -486,6 +488,9 @@ class MultipassDriver:
 
         for transition in self.pta.get_unique_transitions():
 
+            if transition.name == 'getEnergy':
+                continue
+
             # XXX right now we only verify whether both functions have the
             # same number of arguments. This breaks in many overloading cases.
             function_info = self.class_info.function[transition.name]