Added levers.

robotgame/lever.py

@@ -0,0 +1,66 @@
+# This file is part of ROBOTGAME
+#
+# ROBOTGAME is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# ROBOTGAME is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# ROBOTGAME.  If not, see <http://www.gnu.org/licenses/>.
+#
+# ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' '
+#
+#                          lever.py
+#                     --------------------
+#       date created : Wed Aug 8 2012
+#          copyright : (C) 2012 Sakse Dalum
+#      maintained by : Sakse Dalum <don_s@hongabar.org>
+
+"""
+A generic lever.
+"""
+
+import pygame
+
+import worldobject
+
+class Lever(worldobject.WorldObject):
+    def __init__(self, level, x, y, links,
+                 anim='lever_leftright', toggling=False):
+        self.__dict__.update(locals())
+        worldobject.WorldObject.__init__(self, level, x, y)
+
+        self.frame = 0
+        self.anim_speed = 15
+        self.setting = False
+
+    def use(self, obj):
+        self.setting = not self.setting
+
+        for link in self.links:
+            link(self.setting)
+
+    def update(self, e, t, dt):
+        # Update the animation
+        if self.setting:
+            self.frame = min(self.frame + self.anim_speed * dt,
+                             len(self.level.imgs[self.anim]) - 1)
+        else:
+            self.frame = max(self.frame - self.anim_speed * dt,
+                             0)
+        if not self.toggling:
+            if (self.frame + self.anim_speed * dt
+                >= len(self.level.imgs[self.anim]) - 1):
+                self.setting = 0
+
+        worldobject.WorldObject.update(self, e, t, dt)
+
+    def draw(self, window):
+        self.img = self.level.imgs[self.anim][int(self.frame)]
+        window.blit(self.img, (self.x - 32 - self.level.camera_x,
+                               self.y - self.img.get_size()[1] + 24
+                               - self.level.camera_y))