refactor model generation from Analytic/PTAModel into ModelAttribute class

Iteration over states/transitions and model attributes is no longer hardcoded
into most model generation code. This should make support for decision trees
and sub-states much easier.

1 files changed, 10 insertions, 0 deletions

diff --git a/lib/utils.py b/lib/utils.py
index 560ab79..b38a359 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -173,6 +173,16 @@ def match_parameter_values(input_param: dict, match_param: dict):
     return True
 
 
+def partition_by_param(data, param_values):
+    ret = dict()
+    for i, parameters in enumerate(param_values):
+        param_key = tuple(parameters)
+        if param_key not in ret:
+            ret[param_key] = list()
+        ret[param_key].append(data[i])
+    return ret
+
+
 def by_name_to_by_param(by_name: dict):
     """
     Convert aggregation by name to aggregation by name and parameter values.