Explore a bit with the example file

example_retrieve_battery_data.py

@@ -3,6 +3,11 @@ from glob import glob
 
 from extraction.retrieve_battery_data import extract_charge_discharge_impedance
 
+def print_length(items, key):
+    mesures = [x[key] for x in items]
+    len_measures = [len(x) for x in mesures]
+    print(sum(len_measures))
+
 
 def example_how_to_use_this_module():
     all_folders_containing_batteries_mat_files = [
@@ -14,18 +19,43 @@ def example_how_to_use_this_module():
         "./data/BatteryAgingARC-FY08Q4/",
     ]
 
-    filepaths = [y for x in os.walk("./data/") for y in glob(os.path.join(x[0], '*.mat'))]
-    filepaths = sorted(filepaths)
-
-    # TODO warning for for the [:1]
-    for filepath in filepaths[:]:
-        print(filepath)
-        charge_items, discharge_items, impedance_items = extract_charge_discharge_impedance(filepath)
-        print(len(charge_items[0]["voltage_measured"]) == len(charge_items[1]["voltage_measured"]))
-        print(len(charge_items[0]["voltage_measured"]) == len(discharge_items[0]["voltage_measured"]))
-        print(len(charge_items[0]["voltage_measured"]))
-        print(len(charge_items[1]["voltage_measured"]))
-        print(len(discharge_items[0]["voltage_measured"]))
+    for root in all_folders_containing_batteries_mat_files:
+        print("----------------------------")
+        # filepaths = [y for x in os.walk("./data/") for y in glob(os.path.join(x[0], '*.mat'))]
+        filepaths = [y for x in os.walk(root) for y in glob(os.path.join(x[0], '*.mat'))]
+        filepaths = sorted(filepaths)
+
+        for filepath in filepaths[:]:
+            print(filepath)
+            charge_items, discharge_items, impedance_items = extract_charge_discharge_impedance(filepath)
+
+            ## EXPLORATION PART
+            # print("charges")
+            # print_length(charge_items, "voltage_measured")
+            # print_length(charge_items, "current_measured")
+            # print_length(charge_items, "temperature_measured")
+            # print_length(charge_items, "current_charge")
+            # print_length(charge_items, "voltage_charge")
+
+            # print("discharges")
+            # print_length(discharge_items, "voltage_measured")
+            # print_length(discharge_items, "current_measured")
+            # print_length(discharge_items, "temperature_measured")
+            # print_length(discharge_items, "current_charge")
+            # print_length(discharge_items, "voltage_charge")
+
+            # print("impedance")
+            print_length(impedance_items, "sense_current")
+            print_length(impedance_items, "battery_current")
+            print_length(impedance_items, "current_ratio")
+            print_length(impedance_items, "battery_impedance")
+            print_length(impedance_items, "rectified_impedance")
+
+            # print(len() == len(charge_items[1]["voltage_measured"]))
+            # print(len(charge_items[0]["voltage_measured"]) == len(discharge_items[0]["voltage_measured"]))
+            # print(len(charge_items[0]["voltage_measured"]))
+            # print(len(charge_items[1]["voltage_measured"]))
+            # print(len(discharge_items[0]["voltage_measured"]))
 
 
 if __name__ == "__main__":