pigame.py

import pygame,pitft_touchscreen,os
defaultrot = os.getenv('PIGAME_ROT') or '90'
support_gpio = True
envmk = ['PIGAME_V2','PIGAME_INVERTX','PIGAME_INVERTY','PIGAME_SWAPXY','PIGAME_BTN1','PIGAME_BTN2','PIGAME_BTN3','PIGAME_BTN4']
env = {}
for i in envmk:
    env[i] = os.getenv(i)
try:
    import RPi.GPIO as GPIO
except ImportError:
    support_gpio = False
from pygame.locals import *
class PiTft:
    def __init__(self,rotation:int=-1,v2:bool=False if env['PIGAME_V2']=='off' else True,allow_gpio:bool=True,invertx:bool=True if env['PIGAME_INVERTX']=='on' else False,inverty:bool=True if env['PIGAME_INVERTY']=='on' else False,swapxy:bool=True if env['PIGAME_SWAPXY']=='on' else False,buttons=[False if env['PIGAME_BTN1']=='off' else True,False if env['PIGAME_BTN2']=='off' else True,False if env['PIGAME_BTN3']=='off' else True,False if env['PIGAME_BTN4']=='off' else True]):
        self.use_gpio = support_gpio and allow_gpio and not (os.getenv('PIGAME_GPIO') == 'off')
        if not self.use_gpio:
            buttons=[False,False,False,False]
        if rotation == -1:
            rotation = int(defaultrot)
        self.pitft=pitft_touchscreen.pitft_touchscreen()
        self.pitft.button_down=False
        self.pitft.pigameapi=2
        self.pitft.pigamerotr=rotation
        self.invertx = invertx
        self.inverty = inverty
        self.swapxy = swapxy
        self.cachedpos = [0,0]
        self.__b1 = False
        self.__b2 = False
        self.__b3 = False
        self.__b4 = False
        self.__pin1 = 17
        self.__pin2 = 22
        self.__pin3 = 23
        self.__pin4 = 27
        if self.use_gpio:
            GPIO.setmode(GPIO.BCM)
        if buttons[0]:
            GPIO.setup(self.__pin1, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            self.__b1 = True
        if buttons[1]:
            GPIO.setup(self.__pin2, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            self.__b2 = True
        if buttons[2]:
            GPIO.setup(self.__pin3, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            self.__b3 = True
        if buttons[3]:
            if not v2:
                self.__pin4 = 21
            GPIO.setup(self.__pin4, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            self.__b4 = True
        self.pitft.start()
    def update(self):
        """Add Touchscreen Events to PyGame event queue."""
        while not self.pitft.queue_empty():
            for r in self.pitft.get_event():
                e={"y":(r["x"] if r["x"] else self.cachedpos[0]),"x":(r["y"] if r["y"] else self.cachedpos[1])}
                rel=(e["x"] - self.cachedpos[0],e["y"] - self.cachedpos[1])
                self.cachedpos=(e["x"],e["y"])
                if self.pitft.pigamerotr==90:
                    e={"x":e["x"],"y":240-e["y"]}
                    rel=(rel[0],240-rel[1])
                elif self.pitft.pigamerotr==270:
                    e={"x":320-e["x"],"y":e["y"]}
                    rel=(320-rel[0],rel[1])
                else:
                    raise(Exception("PiTft rotation is unsupported"))
                d={}
                t=MOUSEBUTTONUP if r["touch"]==0 else (MOUSEMOTION if self.pitft.button_down else MOUSEBUTTONDOWN)
                if self.invertx:
                    e={"x":320-e["x"],"y":e["y"]}
                    rel=(320-rel[0],rel[1])
                if self.inverty:
                    rel=(rel[0],240-rel[1])
                    e={"y":240-e["y"],"x":e["x"]}
                if self.swapxy:
                    rel=(rel[1],rel[0])
                    e={"x":e["y"],"y":e["x"]}
                if t==MOUSEBUTTONDOWN:
                    d["button"]=1
                    d["pos"]=(e["x"],e["y"])
                    self.pitft.button_down = True
                    pygame.mouse.set_pos(e["x"],e["y"])
                elif t==MOUSEBUTTONUP:
                    self.pitft.button_down = False
                    d["button"]=1
                    d["pos"]=(e["x"],e["y"])
                else:
                    d["buttons"]=(True,False,False)
                    d["rel"]=rel
                    d["pos"]=(e["x"],e["y"])
                    pygame.mouse.set_pos(e["x"],e["y"])
                pe=pygame.event.Event(t,d)
                pygame.event.post(pe)
    def __del__(self):
        """Cleaning up Touchscreen events and Threads when the Object destroyed."""
        self.pitft.stop()
        if self.use_gpio:
            GPIO.cleanup()
    def Button1Interrupt(self,callback=None,bouncetime=200):
        """Calls callback if Button1 pressed."""
        if self.__b1: 
            GPIO.add_event_detect(self.__pin1,GPIO.FALLING,callback=callback,bouncetime=bouncetime)
    def Button2Interrupt(self,callback=None,bouncetime=200):
        """Calls callback if Button2 pressed."""
        if self.__b2: 
            GPIO.add_event_detect(self.__pin2,GPIO.FALLING,callback=callback,bouncetime=bouncetime)
    def Button3Interrupt(self,callback=None,bouncetime=200):
        """Calls callback if Button3 pressed."""
        if self.__b3: 
            GPIO.add_event_detect(self.__pin3,GPIO.FALLING,callback=callback,bouncetime=bouncetime)
    def Button4Interrupt(self,callback=None,bouncetime=200):
        """Calls callback if Button4 pressed."""
        if self.__b4: 
            GPIO.add_event_detect(self.__pin4,GPIO.FALLING,callback=callback,bouncetime=bouncetime)
    @property
    def Button1(self):
        """Equals True if Button 1 is pressed."""
        if self.__b1:
            return not GPIO.input(self.__pin1)
    @property
    def Button2(self):
        """Equals True if Button 2 is pressed."""
        if self.__b2:
            return not GPIO.input(self.__pin2)
    @property
    def Button3(self):
        """Equals True if Button 3 is pressed."""
        if self.__b3:
            return not GPIO.input(self.__pin3)
    @property
    def Button4(self):
        """Equals True if Button 4 is pressed."""
        if self.__b4:
            return not GPIO.input(self.__pin4)