Change the cells to be plotted when using the butterfly plot

bergolho · Apr 2, 2019 · 8a3c2a1 · 8a3c2a1
1 parent bd68722
commit 8a3c2a1
Show file tree

Hide file tree

Showing 6 changed files with 39 additions and 38 deletions.
diff --git a/scripts/butterfly_plot/butterfly_plot.py b/scripts/butterfly_plot/butterfly_plot.py
@@ -195,7 +195,7 @@ def main ():
     for k in range(2):
 
         # Iterate over each cell
-        for i in range(100):
+        for i in range(0,10000,100):
 
             # Get the transmembrane potential from the current simulation for the current cell
             if (k == 0):
@@ -209,8 +209,9 @@ def main ():
             else:
                 plt.plot(t, vms, label="Vm", c="red", linewidth=1.0)
 
-    plt.show()
-    #plt.savefig("output/butterfly.pdf")
+    #plt.show()
+    plt.savefig("output/butterfly.pdf")
+    print("[+] Output file saved at 'output/butterfly.pdf'")
 
 
 if __name__ == "__main__":

diff --git a/scripts/butterfly_plot/output/butterfly.pdf b/scripts/butterfly_plot/output/butterfly.pdf
diff --git a/scripts/butterfly_plot/output/butterfly.png b/scripts/butterfly_plot/output/butterfly.png
diff --git a/scripts/butterfly_plot/output/cells_positions_100.vtp b/scripts/butterfly_plot/output/cells_positions_100.vtp
diff --git a/scripts/butterfly_plot/output/cells_positions_butterfly.vtp b/scripts/butterfly_plot/output/cells_positions_butterfly.vtp
diff --git a/scripts/cell_positions_calculator/main.cpp b/scripts/cell_positions_calculator/main.cpp
@@ -175,7 +175,7 @@ void write_control_volumes_middle_positions_to_vtp (struct tissue *the_tissue)
     vtkSmartPointer<vtkAppendPolyData> appendFilter = vtkSmartPointer<vtkAppendPolyData>::New();
 
     // For each control volume create a sphere
-    for (uint32_t i = 0; i < 100; i++)
+    for (uint32_t i = 0; i < nv; i += 100)
     //for (uint32_t i = 0; i < nv; i++)
     {
         printf("[main] Working on control volume %u\n",i);