Added flow rate plots at each outlet + Q_avg

functions.py

@@ -7,6 +7,7 @@ from tkinter import Tk
 from tkinter.filedialog import askopenfilename, asksaveasfile, askdirectory
 import pandas as pd
 import tkinter as tk
+from statistics import mean 
 from scipy.signal import find_peaks
 
 
@@ -81,15 +82,42 @@ def pressure(folder,N_ts,T_cyc,dt,n_cyc):
     Nc = round(T_cyc/dt)
     time = np.linspace(0,T_cyc,Nc)
     fig, ax = plt.subplots()
+    SBP = np.empty(pressure.shape[1])
+    DBP = np.empty(pressure.shape[1])
+    MBP = np.empty(pressure.shape[1])
     for i in range(0,pressure.shape[1]):
-        ax.plot(time,pressure[N_ts-Nc:N_ts,-i]/1333.22,label='ROI-'+str(i+2))
+        ax.plot(time,pressure[N_ts-Nc:N_ts,i]/1333.22,label='ROI-'+str(i+2)) 
-        
+        SBP[i] = (np.amax(pressure[N_ts-Nc:N_ts,i]/1333.22))
+        DBP[i] = (np.amin(pressure[N_ts-Nc:N_ts,i]/1333.22))
+        MBP[i] = (mean(pressure[N_ts-Nc:N_ts,i]/1333.22)) 
+    PP = SBP-DBP    
     ax.set(xlabel='time [s]', ylabel='Pressure [mmHg]',
         title='Pressure @ each outlet')
     ax.spines['right'].set_visible(False)
     ax.spines['top'].set_visible(False)
     ax.legend(loc=0)
     plt.show()
+    return (DBP,MBP,SBP,PP)
+
+
+def flow(folder,N_ts,T_cyc,dt,n_cyc):
+    flow = np.loadtxt(folder+'/QHistRCR.dat',skiprows=2,)
+    Nc = round(T_cyc/dt)
+    time = np.linspace(0,T_cyc,Nc)
+    fig, ax = plt.subplots()
+    Q = np.empty(flow.shape[1])
+    for i in range(0,flow.shape[1]):
+        ax.plot(time,flow[N_ts-Nc:N_ts,i],label='ROI-'+str(i+2))
+        Q[i] = (mean(flow[N_ts-Nc:N_ts,i]))
+   
+    ax.set(xlabel='time [s]', ylabel='Flow [mL/s]',
+        title='Flow @ each outlet')
+    ax.spines['right'].set_visible(False)
+    ax.spines['top'].set_visible(False)
+    ax.legend(loc=0)
+    plt.show()
+    return Q
+    
         
         
         

main.py

@@ -13,7 +13,7 @@ from scipy import signal
 import statistics
 
 
-from functions import error_plot, periodicity, pressure
+from functions import error_plot, periodicity, pressure, flow
   
 # Selct dir 
 Tk().withdraw()
@@ -49,5 +49,7 @@ error_plot(folder,dt,rc,False)
 periodicity(project,folder,dt,T_cyc,n_cyc)
 
 # Pressure
-pressure(folder,N_ts,T_cyc,dt,n_cyc)
+(DBP,MBP,SBP,PP) = pressure(folder,N_ts,T_cyc,dt,n_cyc)
  
+# Flow Rate
+(Q_avg) = flow(folder,N_ts,T_cyc,dt,n_cyc)