From d8911da92f20e4d9f51a53326a207b794aa167ca Mon Sep 17 00:00:00 2001
From: "a.ott" <a.ott@student.uni-tuebingen.de>
Date: Sat, 1 Aug 2020 12:04:10 +0200
Subject: [PATCH] add function to recalculate saved values

---
 cell_overview.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/cell_overview.py b/cell_overview.py
index c10be43..666c3eb 100644
--- a/cell_overview.py
+++ b/cell_overview.py
@@ -1,19 +1,32 @@
 
-from CellData import icelldata_of_dir
+from CellData import icelldata_of_dir, CellData
 from Baseline import BaselineCellData
 from FiCurve import FICurveCellData
 import os
-
+import numpy as np
 
 def main():
 
     # plot_visualizations("cells/")
     # full_overview("cells/master_table.csv", "cells/")
 
-    for cell_data in icelldata_of_dir("data/invivo_bursty/", test_for_v1_trace=False):
+    recalculate_saved_preanalysis("data/test_data/")
+
+
+def recalculate_saved_preanalysis(data_folder):
+    for cell_data in icelldata_of_dir(data_folder, test_for_v1_trace=True):
+        print(cell_data.get_data_path())
         baseline = BaselineCellData(cell_data)
         baseline.save_values(cell_data.get_data_path())
 
+        contrasts = cell_data.get_fi_contrasts()
+        fi_curve = FICurveCellData(cell_data, contrasts)
+
+        if fi_curve.get_f_inf_slope() < 0:
+            contrasts = np.array(cell_data.get_fi_contrasts()) * -1
+        print(contrasts)
+        fi_curve = FICurveCellData(cell_data, contrasts, save_dir=cell_data.get_data_path(), recalculate=True)
+        # fi_curve.plot_fi_curve()
 
 
 def move_rejected_cell_data():