Move setup code to a function.

1 files changed, 16 insertions, 13 deletions

diff --git a/physics.py b/physics.py
index 748366a..452341c 100644
--- a/physics.py
+++ b/physics.py
@@ -37,6 +37,21 @@ class PhysicsGame:
              self.toolList[c.name] = c(self)
         self.currentTool = self.toolList[tools.allTools[0].name]
 
+    def setup(self):
+        self.screen = pygame.display.get_surface()
+
+        # get everything set up
+        self.clock = pygame.time.Clock()
+        self.font = pygame.font.Font(None, 24) # font object
+
+        # set up the world (instance of Elements)
+        self.box2d = box2d
+        self.world = elements.Elements(self.screen.get_size())
+        self.world.renderer.set_surface(self.screen)
+        
+        # set up static environment
+        self.world.add.ground()
+        
     def stop_start_toggle(self):
         self.world.run_physics = not self.world.run_physics
     
@@ -51,20 +66,8 @@ class PhysicsGame:
         self.world.json_load(file_path)
         
     def run(self):
-        self.screen = pygame.display.get_surface()
-
-        # get everything set up
-        self.clock = pygame.time.Clock()
-        self.font = pygame.font.Font(None, 24) # font object
-
-        # set up the world (instance of Elements)
-        self.box2d = box2d
-        self.world = elements.Elements(self.screen.get_size())
-        self.world.renderer.set_surface(self.screen)
+        self.setup()
         
-        # set up static environment
-        self.world.add.ground()
-
         self.running = True    
         while self.running:
             # Pump GTK messages.