mouse5.py

from direct.showbase.ShowBase import ShowBase
from panda3d.core import CollisionTraverser, CollisionNode
from panda3d.core import CollisionHandlerQueue, CollisionRay
from panda3d.core import AmbientLight, DirectionalLight, LightAttrib
from panda3d.core import TextNode, GeomNode
from panda3d.core import LPoint3, LVector3, BitMask32
from direct.gui.OnscreenText import OnscreenText
from direct.showbase.DirectObject import DirectObject
from direct.task.Task import Task
from panda3d.core import Lens, OrthographicLens
import sys

def color(r,g,b):
    return((r/255,g/255,b/255))


class MouseTask(ShowBase):
    def __init__(self):
        
        super().__init__()
        
        self.disableMouse()
        
        self.squares = []
        
        count = 0
        for index in range(1, 65):
            self.squares += [index]
        
        index = -1
        for x in range(-3,5):
            # going to next row
            for y in range(13,5,-1):
                index += 1
#                 print(index)
                # going throug a constant row
                if count % 2 == 0:
                    self.squares[index] = loader.loadModel("models/square.egg")
                    self.squares[index].reparentTo(render)
                    self.squares[index].setPos(x,y,0)
                    self.squares[index].setColor(0,0,0)
                    self.squares[index].setTag('squares', str(index))
                    count+= 1
                else:
                    self.squares[index] = loader.loadModel("models/square.egg")
                    self.squares[index].reparentTo(render)
                    self.squares[index].setPos(x,y,0)
                    self.squares[index].setColor(255,255,255)
                    self.squares[index].setTag('squares', str(index))
                    count+= 1
            count+= 1
        
        self.Handler = CollisionHandlerQueue()
        self.picker = CollisionTraverser()
        self.pickerNode = CollisionNode('mouseRay')
        self.pickerNP = camera.attachNewNode(self.pickerNode)
        self.pickerNode.setFromCollideMask(GeomNode.getDefaultCollideMask())
        self.pickerRay = CollisionRay()
        self.pickerNode.addSolid(self.pickerRay)
        self.picker.addCollider(self.pickerNP, self.Handler)
        

        
        self.setBackgroundColor(0.5,0.5,0)
        self.camera.setPos(0.5,-3,6)
        self.camera.setHpr(0,-30,0)
        
        self.accept("mouse1", self.mousePressed)
        self.taskMgr.add(self.update, "update")
        
    def mousePressed(self):
#         mousePos = self.win.getPointer(0)
#         print(mousePos.getX(), mousePos.getY())
#         pass
        if base.mouseWatcherNode.hasMouse():
            x = base.mouseWatcherNode.getMouseX()
            y = base.mouseWatcherNode.getMouseY()
        self.pickerRay.setFromLens(self.camNode, x, y)
        print(self.pickerRay)
        self.picker.traverse(render)
        # Assume for simplicity's sake that myHandler is a CollisionHandlerQueue.
        if self.Handler.getNumEntries() > 0:
            # This is so we get the closest object
            self.Handler.sortEntries()
            self.pickedObj = self.Handler.getEntry(0).getIntoNodePath()
            self.pickedObj = self.pickedObj.findNetTag('squares')
            if not self.pickedObj.isEmpty():
                a = self.pickedObj.getColor()
                print(a)
                if a == (0,0,0,1):
                    self.pickedObj.setColor(1,1,1)
                else:
                    self.pickedObj.setColor(0,0,0)
                self.pickedObj = None
    def update(self, task):
        if self.mouseWatcherNode.hasMouse():
            x = self.mouseWatcherNode.getMouseX()
            y = self.mouseWatcherNode.getMouseY()
#             print(x, y)
            self.pickerRay.setFromLens(self.camNode, x, y)
            print(self.pickerRay)
            self.picker.traverse(render)
            # Assume for simplicity's sake that myHandler is a CollisionHandlerQueue.
            if self.Handler.getNumEntries() > 0:
                # This is so we get the closest object
                self.Handler.sortEntries()
                self.pickedObj = self.Handler.getEntry(0).getIntoNodePath()
                self.pickedObj = self.pickedObj.findNetTag('squares')
                if not self.pickedObj.isEmpty():
                    a = self.pickedObj.setColor(color(0, 255, 255)[0],color(0, 255, 255)[1],color(0, 255, 255)[2])
                    print(a)
#                     if a == (0,0,0,1):
#                         self.pickedObj.setColor(1,1,1)
#                     else:
#                         self.pickedObj.setColor(0,0,0)
                    self.pickedObj = None
        return task.cont
        
game = MouseTask()
game.run()