hotfix for paulvangentcom#3 until refactor is done

Author: paulvangentcom

path_planning_demo.py

@@ -1,3 +1,4 @@
+import os
 import sys
 
 import numpy as np
@@ -336,7 +337,7 @@ def update(frame, population, destinations, pop_size, infection_range=0.01,
 
             #ax2.legend(loc = 1, fontsize = 6)
 
-        plt.savefig('render/%i.png' %frame)
+        #plt.savefig('render/%i.png' %frame)
 
     return population
 
@@ -403,9 +404,9 @@ def update(frame, population, destinations, pop_size, infection_range=0.01,
     #initialize destination matrix
     destinations = initialize_destination_matrix(pop_size, 1)
 
-   
-
-    #place hospital on map
+    #create render folder if doesn't exist
+    if not os.path.exists('render/'):
+        os.makedirs('render/')
 
     #define figure
     fig = plt.figure(figsize=(5,7))

simple_simulation.py

@@ -1,3 +1,4 @@
+import os
 import sys
 
 import numpy as np
@@ -347,7 +348,7 @@ def update(frame, population, infection_range=0.01, infection_chance=0.03,
         ax2.set_ylim(0, pop_size + 100)
         ax2.plot(infected_plot, color='gray')
 
-        plt.savefig('render/%s.png' %frame)
+        #plt.savefig('render/%s.png' %frame)
 
 
 
@@ -375,7 +376,10 @@ def update(frame, population, infection_range=0.01, infection_chance=0.03,
     ax2.set_xlim(0, simulation_steps)
     ax2.set_ylim(0, pop_size + 100)
 
-    
+    #create render folder if doesn't exist
+    if not os.path.exists('render/'):
+        os.makedirs('render/')
+           
 
     #start animation loop through matplotlib visualisation
     animation = FuncAnimation(fig, update, fargs = (population,), frames = simulation_steps, interval = 33)

simulation.py

@@ -1,3 +1,4 @@
+import os
 import sys
 
 import numpy as np
@@ -240,6 +241,10 @@ def update(frame, population, destinations, pop_size, infection_range=0.01,
     ##### END OF SETTABLE PARAMETERS #####
     ######################################
 
+    #create render folder if doesn't exist
+    if not os.path.exists('render/'):
+        os.makedirs('render/')
+
     #initialise population
     population = initialize_population(pop_size, mean_age, max_age, xbounds, ybounds)