Update simulation.py

simulation.py

@@ -168,7 +168,7 @@ def callback(self):
             self.population[0][10] = 1
 
 
-    def run(self):
+    def run(self, simulation_ad):
         '''run simulation'''
 
         i = 0
@@ -200,13 +200,26 @@ def run(self):
         print('total infectious: %i' %len(self.population[(self.population[:,6] == 1) |
                                                           (self.population[:,6] == 4)]))
         print('total unaffected: %i' %len(self.population[self.population[:,6] == 0]))
+
+        simulation_ad.runad()
 
 
     def plot_sir(self, size=(6,3), include_fatalities=False,
                  title='S-I-R plot of simulation'):
         plot_sir(self.Config, self.pop_tracker, size, include_fatalities,
                  title)
 
+class simulation_():
+    def __init__(self):
+        pass
+    def run_(self):
+        pass
+
+class adapter:
+    def __init__(self):
+        self.simulation_ = simulation_()
+    def runad(self):
+        self.simulation_.run_()
 
 
 if __name__ == '__main__':
@@ -239,4 +252,5 @@ def plot_sir(self, size=(6,3), include_fatalities=False,
     #sim.population_init() #reinitialize population to enforce new roaming bounds
 
     #run, hold CTRL+C in terminal to end scenario early
-    sim.run()
+    adapter = adapter()
+    sim.run(adapter)