analyze-log: support --function-override

1 files changed, 7 insertions, 0 deletions

diff --git a/bin/analyze-log.py b/bin/analyze-log.py
index 1e866b7..41aa9af 100755
--- a/bin/analyze-log.py
+++ b/bin/analyze-log.py
@@ -146,10 +146,17 @@ def main():
         norm = dfatool.cli.parse_nfp_normalization(args.normalize_nfp)
         dfatool.utils.normalize_nfp_in_aggregate(by_name, norm)
 
+    function_override = dict()
+    if args.function_override:
+        for function_desc in args.function_override.split(";"):
+            state_or_tran, attribute, *function_str = function_desc.split(" ")
+            function_override[(state_or_tran, attribute)] = " ".join(function_str)
+
     model = AnalyticModel(
         by_name,
         parameter_names,
         force_tree=args.force_tree,
+        function_override=function_override,
     )
 
     if args.info: