1 files changed, 106 insertions, 0 deletions
diff --git a/buttons.py b/buttons.py
new file mode 100755
index 0000000..9542796
--- /dev/null
+++ b/buttons.py
@@ -0,0 +1,106 @@
+# buttons.py
+import g,utils,pygame
+
+class Button:
+    _instances=[]
+    
+    def __init__(self,name,(x1,y1),centre=True): # eg ('plus',(30,40))
+        self._instances.append(self)
+        up=utils.load_image(name+"_up.png",True)
+        down=utils.load_image(name+"_down.png",True)
+        w=up.get_width();h=up.get_height();x=x1;y=y1
+        if centre:
+            self.cx=x; self.cy=y
+            x=x-w/2; y=y-h/2
+        else:
+            self.cx=x+w/2; self.cy=y+h/2
+        self.rect=pygame.Rect(x,y,w,h)
+        self.name=name; self.x=x; self.y=y; self.active=True
+        self.up=up; self.down=down; self.stay_down=False
+
+    def mouse_on(self):
+        mx,my=g.pos
+        return self.rect.collidepoint(mx,my)
+
+    def draw_up(self):
+        g.screen.blit(self.up,(self.x,self.y))
+
+    def draw_down(self):
+        g.screen.blit(self.down,(self.x,self.y))
+
+    def on(self):
+        self.active=True
+
+    def off(self):
+        self.active=False
+
+def draw():
+    for b in Button._instances:
+        if b.active:
+            if b.stay_down:
+                b.draw_down()
+            else:
+                b.draw_up()
+
+def check():
+    clear()
+    for b in Button._instances:
+        if b.active:
+            if b.mouse_on():
+                if b.name in ('xyz1','xyz2'):
+                    b.stay_down=True
+                else:
+                    b.draw_down()
+                    g.screen.blit(g.pointer,g.pos)
+                    pygame.display.flip()
+                    pygame.time.wait(300)
+                return b.name #****
+    return '' # no button pressed
+
+def clear():
+    for b in Button._instances: b.stay_down=False
+    
+def active(name):
+    for b in Button._instances:
+        if b.name==name: return b.active #****
+    return False # not found
+
+def stay_down(name):
+    for b in Button._instances:
+        if b.name==name: b.stay_down=True; return #****
+    return
+
+def mouse_on(name):
+    mx,my=g.pos
+    for b in Button._instances:
+        if b.name==name:
+            return b.rect.collidepoint(mx,my)
+    return False
+
+def set_mouse(name):
+    p2=g.pointer.get_height()/2
+    for b in Button._instances:
+        if b.name==name:
+            x=b.cx; y=b.cy+p2
+            pygame.mouse.set_pos(x,y); g.pos=(x,y)
+
+# eg1 buttons.on('plus')
+# eg2 buttons.on(['plus','times'])
+def on(name):
+    if type(name)==type('a'):
+        list1=[]; list1.append(name)
+    else:
+        list1=name
+    for b in Button._instances:
+        if b.name in list1: b.active=True
+
+# eg1 buttons.off('plus')
+# eg2 buttons.off(['plus','times'])
+def off(name):
+    if type(name)==type('a'):
+        list1=[]; list1.append(name)
+    else:
+        list1=name
+    for b in Button._instances:
+        if b.name in list1: b.active=False
+