add tab switching results

1 files changed, 7 insertions, 2 deletions

diff --git a/bjoern/videoanalyse/post_processing.py b/bjoern/videoanalyse/post_processing.py
index b5e3c65..445b72f 100644
--- a/bjoern/videoanalyse/post_processing.py
+++ b/bjoern/videoanalyse/post_processing.py
@@ -8,6 +8,7 @@ import pandas as pd
 
 import LogParser
 import utils
+import tab_switch
 
 argparser = argparse.ArgumentParser(description="OCR-Logfile evaluation")
 argparser.add_argument("vp_dir", help="Directory with all VPs")
@@ -18,6 +19,7 @@ data_path = Path(args.vp_dir)
 all_vp = [x for x in data_path.iterdir() if x.is_dir()]
 
 vp_results = []
+tab_results = []
 for vp_path in all_vp:
     log = LogParser.extract_activities(LogParser.get_log_data(vp_path))
     log = LogParser.generate_log(log)
@@ -43,8 +45,11 @@ for vp_path in all_vp:
 
     df = pd.read_csv(f"{vp_path}/metrics_grps.csv")
 
+    tab_df = tab_switch.tab_switches_per_type(df)
+    tab_results.append(tab_df)
+    tab_df.to_csv(f"{vp_path}/tabswitches.csv")
+
     vp_results.append(df)
 
-evals = utils.evaluate_results(vp_results)
-pprint(evals)
+evals = utils.evaluate_results(tab_results)
 evals.to_csv(f"{data_path}/evaluation.csv")