From bcd014c15b5d9e68f3764a8ae09fa5f20a22dfe8 Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Mon, 2 Sep 2019 15:51:13 +0200 Subject: Add basic auto-generated energy accounting code for online model eval --- lib/automata.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'lib/automata.py') diff --git a/lib/automata.py b/lib/automata.py index d26c0bc..4cd694d 100755 --- a/lib/automata.py +++ b/lib/automata.py @@ -522,6 +522,14 @@ class PTA: """Return PTA-specific ID of transition.""" return self.transitions.index(transition) + def get_state_names(self): + """Return lexically sorted list of PTA state names.""" + return sorted(self.state.keys()) + + def get_state_id(self, state: State) -> int: + """Return PTA-specific ID of state.""" + return self.get_state_names().index(state.name) + def get_initial_param_dict(self): return dict([[self.parameters[i], self.initial_param_values[i]] for i in range(len(self.parameters))]) -- cgit v1.2.3