attempt figure animation

.ipynb_checkpoints/app-checkpoint.py

@@ -458,11 +458,11 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     fig, ax = plt.subplots()
     plt.ylim((0,220))
     plt.xlim((0,250))
-    line = ax.plot(volumes[(N-2)*60000], pressures[(N-2)*60000], lw=1)
-    #line = ax.plot(volumes[(N-2)*60000:(N)*60000], pressures[(N-2)*60000:(N)*60000], lw=1)
-    #print(line)
-    line = line[0]
-    #print(line)
+    start = (N-2)*60000
+    end = (N-2)*60000+50000
+    line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
+    point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
+
     
     fig.suptitle('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
@@ -474,14 +474,13 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     # fig.subplots_adjust(left=0.25, bottom=0.25)
 
     def update(frame):
-        # for each frame, update the data stored on each artist.
-        x = volumes[:(N-2)*60000+frame]
-        y = pressures[:(N-2)*60000:(N)+frame]
-        line.set_xdata(x)
-        line.set_ydata(y)
-        return line
-
-    anim = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=30)
+        # update to add more of the loop
+        end = (N-2)*60000+1000 * frame
+        x = volumes[start:end]
+        y = pressures[start:end]
+        ax.plot(x, y, lw=1, c='b')
+
+    anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
     
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
@@ -551,13 +550,13 @@ with gr.Blocks() as demo:
             
             
             with gr.Row():
-                sl1 = gr.Slider(0.005, 0.1, value=Rm, label="Rm")
-                sl2 = gr.Slider(0.0001, 0.25, value=Ra, label="Ra")
-                sl3 = gr.Slider(0.5, 3.5, value=Emax, label="Emax")
-                sl4 = gr.Slider(0.02, 0.1, value= Emin, label="Emin")
-                sl5 = gr.Slider(4.0, 25.0, value=Vd, label="Vd")
-                sl6 = gr.Slider(0.4, 1.7, value=Tc, label="Tc") 
-                sl7 = gr.Slider(0.0, 280.0, value=start_v, label="start_v")
+                sl1 = gr.Slider(0.005, 0.1, value=.005, label="Rm")
+                sl2 = gr.Slider(0.0001, 0.25, value=.0001, label="Ra")
+                sl3 = gr.Slider(0.5, 3.5, value=.5, label="Emax")
+                sl4 = gr.Slider(0.02, 0.1, value= .02, label="Emin")
+                sl5 = gr.Slider(4.0, 25.0, value= 4.0, label="Vd")
+                sl6 = gr.Slider(0.4, 1.7, value= 0.4, label="Tc") 
+                sl7 = gr.Slider(0.0, 280.0, value= 140., label="start_v")
     
     
     generate_button.click(fn=generate_example, outputs = [video,plot,Rm,Ra,Emax,Emin,Vd,Tc,start_v])

app.py

@@ -458,11 +458,11 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     fig, ax = plt.subplots()
     plt.ylim((0,220))
     plt.xlim((0,250))
-    line = ax.plot(volumes[(N-2)*60000], pressures[(N-2)*60000], lw=1)
-    #line = ax.plot(volumes[(N-2)*60000:(N)*60000], pressures[(N-2)*60000:(N)*60000], lw=1)
-    #print(line)
-    line = line[0]
-    #print(line)
+    start = (N-2)*60000
+    end = (N-2)*60000+50000
+    line = ax.plot(volumes[start:(start+1)], pressures[start:(start+1)], lw=1, color='b')
+    point = ax.scatter(volumes[start:(start+1)], pressures[start:(start+1)], c="b", s=5)
+
     
     fig.suptitle('Predicted PI-SSL LV Pressure Volume Loop', fontsize=16)
     #plt.rcParams['fig.suptitle'] = -2.0
@@ -474,14 +474,13 @@ def pvloop_simulator(Rm, Ra, Emax, Emin, Vd, Tc, start_v):
     # fig.subplots_adjust(left=0.25, bottom=0.25)
 
     def update(frame):
-        # for each frame, update the data stored on each artist.
-        x = volumes[:(N-2)*60000+frame]
-        y = pressures[:(N-2)*60000:(N)+frame]
-        line.set_xdata(x)
-        line.set_ydata(y)
-        return line
-
-    anim = animation.FuncAnimation(fig=fig, func=update, frames=40, interval=30)
+        # update to add more of the loop
+        end = (N-2)*60000+1000 * frame
+        x = volumes[start:end]
+        y = pressures[start:end]
+        ax.plot(x, y, lw=1, c='b')
+
+    anim = animation.FuncAnimation(fig, partial(update), frames=43, interval=1)
     
     return plt, Rm, Ra, Emax, Emin, Vd, Tc, start_v
 
@@ -551,13 +550,13 @@ with gr.Blocks() as demo:
             
             
             with gr.Row():
-                sl1 = gr.Slider(0.005, 0.1, value=Rm, label="Rm")
-                sl2 = gr.Slider(0.0001, 0.25, value=Ra, label="Ra")
-                sl3 = gr.Slider(0.5, 3.5, value=Emax, label="Emax")
-                sl4 = gr.Slider(0.02, 0.1, value= Emin, label="Emin")
-                sl5 = gr.Slider(4.0, 25.0, value=Vd, label="Vd")
-                sl6 = gr.Slider(0.4, 1.7, value=Tc, label="Tc") 
-                sl7 = gr.Slider(0.0, 280.0, value=start_v, label="start_v")
+                sl1 = gr.Slider(0.005, 0.1, value=.005, label="Rm")
+                sl2 = gr.Slider(0.0001, 0.25, value=.0001, label="Ra")
+                sl3 = gr.Slider(0.5, 3.5, value=.5, label="Emax")
+                sl4 = gr.Slider(0.02, 0.1, value= .02, label="Emin")
+                sl5 = gr.Slider(4.0, 25.0, value= 4.0, label="Vd")
+                sl6 = gr.Slider(0.4, 1.7, value= 0.4, label="Tc") 
+                sl7 = gr.Slider(0.0, 280.0, value= 140., label="start_v")
     
     
     generate_button.click(fn=generate_example, outputs = [video,plot,Rm,Ra,Emax,Emin,Vd,Tc,start_v])