bakeup

tracking/utils/read_data.py

@@ -9,7 +9,8 @@ func: extract_data()
 import numpy as np
 import re
 import os
-
+from collections import OrderedDict
+import matplotlib.pyplot as plt
 
 
 
@@ -206,19 +207,130 @@ def read_deletedBarcode_file(filePth):
     return all_list
 
 
+def read_weight_timeConsuming(filePth):
+    WeightDict, SensorDict, ProcessTimeDict = OrderedDict(), OrderedDict(), OrderedDict()
+    
+    with open(filePth, 'r', encoding='utf-8') as f:
+        lines = f.readlines()
+        for i, line in enumerate(lines):
+            line = line.strip()
+            
+            if line.find(':') < 0: continue
+            if line.find("Weight") >= 0: 
+                label = "Weight"
+                continue
+            if line.find("Sensor") >= 0: 
+                label = "Sensor"
+                continue
+            if line.find("processTime") >= 0: 
+                label = "ProcessTime"
+                continue
+            
+            keyword = line.split(':')[0]
+            value = line.split(':')[1]
+            
+            if label == "Weight":
+                WeightDict[keyword] = float(value.strip(','))
+            if label == "Sensor":
+                SensorDict[keyword] = [float(s) for s in value.split(',') if len(s)]
+            if label == "ProcessTime":
+                ProcessTimeDict[keyword] = float(value.strip(','))
+  
+    # print("Done!")
+    return WeightDict, SensorDict, ProcessTimeDict
+            
+                
+def plot_sensor_curve(WeightDict, SensorDict, ProcessTimeDict):
+    
+    wtime, wdata = [], []
+    stime, sdata = [], []
+    for key, value in WeightDict.items():
+        wtime.append(int(key))
+        wdata.append(value)
+    
+    for key, value in SensorDict.items():
+        if len(value) != 9: continue
+        
+        stime.append(int(key))
+        sdata.append(np.array(value))
+        
+    static_range = []
+    dynamic_range = []
+    windth = 8
+    nw = len(wdata)
+    assert(nw) >= 8, "The num of weight data is less than 8!"
+    
+    i1, i2 = 0, 7
+    while i2 < nw:
+        data = wdata[i1:(i2+1)]
+        max(data) - min(data)
 
+        if i2<7:
+            i1 = 0
+        else:
+            i1 = i2-windth
+
+    min_t = min(wtime + stime)
+    wtime = [t-min_t for t in wtime]
+    stime = [t-min_t for t in stime]
+    
+    max_t = max(wtime + stime)
+    
+    fig = plt.figure(figsize=(16, 12))
+    gs = fig.add_gridspec(2, 1,  left=0.1, right=0.9, bottom=0.1, top=0.9,
+                      wspace=0.05, hspace=0.15)
+    # ax1, ax2 = axs
+    
+    ax1 = fig.add_subplot(gs[0,0])
+    ax2 = fig.add_subplot(gs[1,0])
+    
+    ax1.plot(wtime, wdata, 'b--', linewidth=2 )
+    for i in range(9):
+        ydata = [s[i] for s in sdata]
+        ax2.plot(stime, ydata, linewidth=2 )
+    
+    ax1.grid(True), ax1.set_xlim(0, max_t), ax1.set_title('Weight')
+    ax1.set_label("(Time: ms)")
+    # ax1.legend()
+    
+    ax2.grid(True), ax2.set_xlim(0, max_t), ax2.set_title('IMU')
+    # ax2.legend()
+    
+    plt.show()
+    
+    
+    
+        
+    
+    
+def main(file_path):
+    WeightDict, SensorDict, ProcessTimeDict = read_weight_timeConsuming(file_path)
+    plot_sensor_curve(WeightDict, SensorDict, ProcessTimeDict)
+
+           
+            
 
 if __name__ == "__main__":    
-    files_path = 'D:/contrast/dataset/1_to_n/709/20240709-112658_6903148351833/'  
     
-    # 遍历目录下的所有文件和目录  
+    files_path = r'\\192.168.1.28\share\测试_202406\0814\0814\20240814-102227-62264578-a720-4eb9-b95e-cb8be009aa98_null'  
+    k = 0
     for filename in os.listdir(files_path):    
-        filename = '1_track.data'
+        filename = 'process.data'
+        
         file_path = os.path.join(files_path, filename)
         if os.path.isfile(file_path) and filename.find("track.data")>0:
             extract_data(file_path)
- 
-            print("Done")
+        
+        if os.path.isfile(file_path) and filename.find("process.data")>=0:
+            main(file_path)
+        
+        k += 1
+        if k == 1:
+            break
+            
+            
+            
+        # print("Done")