mouse cursor

[pmachines.git] / pmachines / scene.py

diff --git a/pmachines/scene.py b/pmachines/scene.py
index dbc6282a6dcf22a8764b0b142ecd7d1aa3443a7a..75886b4c5d3213e8148b739c2ac16e0fc2dcccd1 100644 (file)
--- a/pmachines/scene.py
+++ b/pmachines/scene.py
@@ -1,8 +1,12 @@
 from panda3d.core import AmbientLight, DirectionalLight, Point3
-from panda3d.bullet import BulletPlaneShape, BulletGhostNode# BulletRigidBodyNode
+from panda3d.bullet import BulletPlaneShape, BulletGhostNode
+from direct.gui.OnscreenImage import OnscreenImage
+from direct.gui.DirectGui import DirectButton
+from direct.gui.DirectGuiGlobals import FLAT, DISABLED, NORMAL
 from direct.showbase.DirectObject import DirectObject
 from pmachines.items.background import Background
 from pmachines.items.box import Box
+from lib.engine.gui.cursor import MouseCursor
 
 
 class Scene(DirectObject):
@@ -11,6 +15,10 @@ class Scene(DirectObject):
         super().__init__()
         self._world = world
         self._set_camera()
+        self._cursor = MouseCursor(
+            'assets/buttons/arrowUpLeft.png', (.04, 1, .04), (.5, .5, .5, 1),
+            (.01, .01))
+        self._set_gui()
         self._set_lights()
         self._set_input()
         self._set_mouse_plane()
@@ -22,6 +30,50 @@ class Scene(DirectObject):
         base.camera.set_pos(0, -20, 0)
         base.camera.look_at(0, 0, 0)
 
+    def _set_gui(self):
+        def load_img_btn(path, col):
+            img = OnscreenImage('assets/buttons/%s.png' % path)
+            img.set_transparency(True)
+            img.set_color(col)
+            img.detach_node()
+            return img
+        def load_images_btn(path, col):
+            colors = {
+                'gray': [
+                    (.6, .6, .6, 1),  # ready
+                    (1, 1, 1, 1), # press
+                    (.8, .8, .8, 1), # rollover
+                    (.4, .4, .4, .4)],
+                'green': [
+                    (.1, .68, .1, 1),
+                    (.1, 1, .1, 1),
+                    (.1, .84, .1, 1),
+                    (.4, .1, .1, .4)]}[col]
+            return [load_img_btn(path, col) for col in colors]
+        abl, abr = base.a2dBottomLeft, base.a2dBottomRight
+        btn_info = [
+            ('home', self.on_home, DISABLED, abl, 'gray'),
+            ('information', self.on_information, DISABLED, abl, 'gray'),
+            ('right', self.on_play, NORMAL, abr, 'green'),
+            ('next', self.on_next, DISABLED, abr, 'gray'),
+            ('previous', self.on_prev, DISABLED, abr, 'gray'),
+            ('rewind', self.on_rewind, DISABLED, abr, 'gray')]
+        num_l = num_r = 0
+        for binfo in btn_info:
+            imgs = load_images_btn(binfo[0], binfo[4])
+            if binfo[3] == base.a2dBottomLeft:
+                sign, num = 1, num_l
+                num_l += 1
+            else:
+                sign, num = -1, num_r
+                num_r += 1
+            fcols = (.4, .4, .4, .14), (.3, .3, .3, .05)
+            btn = DirectButton(
+                image=imgs, scale=.05, pos=(sign * (.06 + .11 * num), 1, .06),
+                parent=binfo[3], command=binfo[1], state=binfo[2], relief=FLAT,
+                frameColor=fcols[0] if binfo[2] == NORMAL else fcols[1])
+            btn.set_transparency(True)
+
     def _set_directional_light(self, name, hpr, color):
         light = DirectionalLight(name)
         light_np = render.attach_new_node(light)
@@ -45,7 +97,6 @@ class Scene(DirectObject):
         self.accept('mouse1-up', self.on_release)
         self.accept('mouse3', self.on_click_r)
         self.accept('mouse3-up', self.on_release)
-        self.accept('p-up', self.on_play)
 
     def _set_mouse_plane(self):
         shape = BulletPlaneShape((0, -1, 0), 1)
@@ -72,6 +123,8 @@ class Scene(DirectObject):
         for hit in self._get_hits():
             for item in [i for i in self.items if hit.get_node() == i.node]:
                 getattr(item, method)(pos)
+                img = 'move' if method == 'on_click_l' else 'rotate'
+                self._cursor.set_image('assets/buttons/%s.png' % img)
 
     def on_click_l(self):
         self._on_click('on_click_l')
@@ -81,6 +134,7 @@ class Scene(DirectObject):
 
     def on_release(self):
         [item.on_release() for item in self.items]
+        self._cursor.set_image('assets/buttons/arrowUpLeft.png')
 
     def on_frame(self, task):
         hits = self._get_hits()
@@ -99,3 +153,18 @@ class Scene(DirectObject):
 
     def on_play(self):
         [itm.play() for itm in self.items]
+
+    def on_next(self):
+        print('on_next')
+
+    def on_prev(self):
+        print('on_prev')
+
+    def on_rewind(self):
+        print('on_rewind')
+
+    def on_home(self):
+        print('on_home')
+
+    def on_information(self):
+        print('on_information')