add parameter normalization support to YAML DFA models

1 files changed, 4 insertions, 1 deletions

diff --git a/lib/automata.py b/lib/automata.py
index 6437839..3220d81 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -1,6 +1,6 @@
 """Classes and helper functions for PTA and other automata."""
 
-from functions import AnalyticFunction
+from functions import AnalyticFunction, NormalizationFunction
 import itertools
 
 def _dict_to_list(input_dict: dict) -> list:
@@ -310,6 +310,9 @@ class PTA:
             if parameter in self.parameter_normalization:
                 if 'enum' in self.parameter_normalization[parameter] and value in self.parameter_normalization[parameter]['enum']:
                     normalized_param[parameter] = self.parameter_normalization[parameter]['enum'][value]
+                if 'formula' in self.parameter_normalization[parameter]:
+                    normalization_formula = NormalizationFunction(self.parameter_normalization[parameter]['formula'])
+                    normalized_param[parameter] = normalization_formula.eval(value)
         return normalized_param
 
     @classmethod