feat: Stops ground from moving when paused

CogitoNTNU · Nov 5, 2024 · cc0d52a · cc0d52a
1 parent 6e58b88
commit cc0d52a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/agent_parts_main.py b/agent_parts_main.py
@@ -173,7 +173,8 @@ def make_limb():
             creature.update_creature_position(dragged_limb, new_position)
 
         screen.fill((135, 206, 235))
-        environment.update()
+        if(physics_on):
+            environment.update()
         environment.render()
 
         #creature.set_joint_rates([random.random()*2, random.random()*2])

diff --git a/src/agent_parts/creature.py b/src/agent_parts/creature.py
@@ -1,7 +1,7 @@
 import pymunk
 import pygame
-from agent_parts.limb import Limb
-from agent_parts.motorjoint import MotorJoint  
+from src.agent_parts.limb import Limb
+from src.agent_parts.motorjoint import MotorJoint  
 
 
 class Creature:

diff --git a/src/interface.py b/src/interface.py
@@ -48,8 +48,9 @@ def add_button(self, button: Button) -> Button:
         return button
 
     def remove_button(self, button: Button) -> Button: 
-        self.elements.remove(button)
-        return button
+        if button in self.elements:
+            self.elements.remove(button)
+            return button
 
     def render(self, screen):
         """Render all UI elements."""