plots heatmap for final length, added length, relative change in length in the…

plots heatmap for final length, added length, relative change in length in the controlled phase, fixed minor bugs

basic_model_only_density/heatmap_length.py

@@ -79,7 +79,8 @@ def length_heatmap(DIR):
                    extent = [np.min(retardationtime)-rt_step/2, np.max(retardationtime)+rt_step/2, 
                              np.min(peclet)-peclet_step/2, np.max(peclet)+peclet_step/2],
                     interpolation ='nearest', origin ='lower') 
-    plt.title('$L_f$ in the controlled growth phase for $L_i = 1$')
+    plt.colorbar(fig1)
+    plt.title('$L_f$ in the controlled growth phase for $L_i = %s$' % p['system_size'])
     plt.show()
 
     plt.figure()
@@ -89,7 +90,8 @@ def length_heatmap(DIR):
                    extent = [np.min(retardationtime)-rt_step/2, np.max(retardationtime)+rt_step/2, 
                              np.min(peclet)-peclet_step/2, np.max(peclet)+peclet_step/2],
                     interpolation ='nearest', origin ='lower') 
-    plt.title('$\Delta L$ in the controlled growth phase for $L_i = 1$')
+    plt.colorbar(fig2)
+    plt.title('$\Delta L$ in the controlled growth phase for $L_i = %s$' % p['system_size'])
     plt.show()
 
     plt.figure()
@@ -99,7 +101,8 @@ def length_heatmap(DIR):
                    extent = [np.min(retardationtime)-rt_step/2, np.max(retardationtime)+rt_step/2, 
                              np.min(peclet)-peclet_step/2, np.max(peclet)+peclet_step/2],
                     interpolation ='nearest', origin ='lower') 
-    plt.title('$\Delta L/L_i$ in the controlled growth phase for $L_i = 1$')
+    plt.colorbar(fig3)
+    plt.title('$\Delta L/L_i$ in the controlled growth phase for $L_i = %s$' % p['system_size'])
     plt.show()