Das deutsche Python-Forum

Code: Alles auswählen

import pygame
from pygame.locals import *

class EventHandler(object):

    def __init__(self, main):
        self.main = main
        self.userEvents = {}
        self.userEventsNum = 0
        # SONG_END = pygame.USEREVENT + 1
        self.KEYDOWN_event = {}
        self.KEYUP_event = {}
        self.KEYPRESSED_event = {}
        self.MOUSEBUTTONDOWN_event = {}
        self.MOUSEBUTTONUP_event = {}
        self.JOYBUTTONDOWN_event = {}
        self.KEYDOWN_listener = []
        self.JOYBUTTONDOWN_listener = []

    def tick(self):
        pygame.joystick.init()
        for event in pygame.event.get():               
                          
            if event.type == QUIT:
                self.main.quit()


            

            for i in self.userEvents:
                if event.type == i:
                    self.userEvents[i](event)

            self.checkForEvent(event, KEYDOWN, self.KEYDOWN_event, True)
            self.checkForEvent(event, KEYUP, self.KEYUP_event, True)
            self.checkForEvent(event, MOUSEBUTTONDOWN,
                               self.MOUSEBUTTONDOWN_event, False)
            self.checkForEvent(event, MOUSEBUTTONUP,
                               self.MOUSEBUTTONUP_event, False)
##


             
            
            

            if event.type == KEYDOWN:
                for i in self.KEYDOWN_listener:
                    i(event.key)


        if self.main.loop:
            keys_presed = pygame.key.get_pressed()
            joy_presed = pygame.button.get_numbuttons()

            for i in self.KEYPRESSED_event:
                if keys_presed[i]:
                    for func in self.KEYPRESSED_event[i]:
                        func()
            for f in self.pygame.button.get_numbuttons():
                if joy_presed[f]:
                    for func in self.pygame.button.get_numbuttons[f]:
                        func()
                


    def checkForEvent(self, event, checkEvent, functions, key):
        if event.type == checkEvent:
            try:
                if key:
                    functionsFIN = functions[event.key]
                else:
                    functionsFIN = functions[event.button]
                for i in functionsFIN:
                        i()
            except:
                pass
    def registerUserEvent(self, function):
        self.userEventsNum += 1
        event = pygame.USEREVENT + self.userEventsNum
        self.userEvents[event] = function
        return event

    def unregisterUserEvent(self, event):
        del(self.userEvents[event])

    def unregisterUserEventsAll(self):
        self.userEvents = {}


    def registerKEYDOWNevent(self, key, function):
        try:
            temp = self.KEYDOWN_event[key]
            temp.append(function)
            self.KEYDOWN_event[key] = temp
        except:
            self.KEYDOWN_event[key] = [function]

    def unregisterKEYDOWNeventAll(self, key):
        del(self.KEYDOWN_event[key])

    def unregisterKEYDOWNevent(self, key, function):
        try:
            self.KEYDOWN_event[key].remove(function)
        except:
            pass

    def unregisterKEYDOWNeventAllKeys(self):
        self.KEYDOWN_event = {}


##    def registerJOYBUTTONDOWNevent(self, joybutton, function):
##        try:
##            temp = self.JOYBUTTONDOWN_event[button]



##            temp.append(function)
##            self.JOYBUTTONDOWN[joybutton] = temp
##        except:
##            self.JOYBUTTONDOWN_event[joybutton] = [function]
##
##    def unregisterJOYBUTTONDOWNeventAll(self, joybutton):
##        del( self.JOYBUTTONDOWN_event[joybutton])
##
##    def unregisterJOYBUTTONDOWNevent(self, joybutton, function):
##        try:
##            self. self.JOYBUTTONDOWN_event[joybutton].remove(function)
##        except:
##            pass
##    def unregisterJOYBUTTONDOWNeventAllKeys(self):
##        self.JOYBUTTONDOWN_event = {}

        




    def registerKEYUPevent(self, key, function):
        try:
            temp = self.KEYUP_event[key]
            temp.append(function)
            self.KEYUP_event[key] = temp
        except:
            self.KEYUP_event[key] = [function]

    def unregisterKEYUPeventAll(self, key):
        del(self.KEYUP_event[key])

    def unregisterKEYUPevent(self, key, function):
        try:
            self.KEYUP_event[key].remove(function)
        except:
            pass

    def unregisterKEYUPeventAllKeys(self):
        self.KEYUP_event = {}

    def registerKEYPRESSEDevent(self, key, function):
        try:
            temp = self.KEYPRESSED_event[key]
            temp.append(function)
            self.KEYPRESSED_event[key] = temp
        except:
            self.KEYPRESSED_event[key] = [function]

    def unregisterKEYPRESSEDeventAll(self, key):
        del(self.KEYPRESSED_event[key])

    def unregisterKEYPRESSEDevent(self, key, function):
        try:
            self.KEYPRESSED_event[key].remove(function)
        except:
            pass

    def unregisterKEYPRESSEDeventAllKeys(self):
        self.KEYPRESSED_event = {}

    def registerMOUSEBUTTONDOWNevent(self, button, function):
        try:
            temp = self.MOUSEBUTTONDOWN_event[button]
            temp.append(function)
            self.MOUSEBUTTONDOWN_event[button] = temp

        except KeyError:
            self.MOUSEBUTTONDOWN_event[button] = [function]

    def unregisterMOUSEBUTTONDOWNeventAll(self, button):
        del(self.MOUSEBUTTONDOWN_event[button])


    def unregisterMOUSEBUTTONDOWNevent(self, button, function):
        try:
            self.MOUSEBUTTONDOWN_event[button].remove(function)
        except:
            pass

    def unregisterJOYBUTTONDOWNeventAllKEys(self):
        self.JOYBUTTONDOWN_event = {}

    def registerJOYBUTTONDOWNevent(self, button, function):  
        try:
            temp = self.JOYBUTTONDOWN_event[button]
            temp.append(function)
            self.JOYBUTTONDOWN_event[button] = temp

        except KeyError:
            self.JOYBUTTONDOWN_event[button] = [button]

    def unregisterJOYBUTTONDOWNeventAll(self, button):
        del(self.JOYBUTTONDOWN_event[button])


    def unregisterJOYBUTTONDOWNDOWNevent(self, button, function):
        try:
            self.JOYBUTTONDOWNDOWN_event[button].remove(function)
        except:
            pass

    def unregisteJOYBUTTONDOWNDOWNeventAllKEys(self):
        self.JOYBUTTONDOWN_event = {}



    def registerMOUSEBUTTONUPevent(self, button, function):
        try:
            temp = self.MOUSEBUTTONUP_event[button]
            temp.append(function)
            self.MOUSEBUTTONUP_event[button] = temp
        except KeyError:
            self.MOUSEBUTTONUP_event[button] = [function]

    def unregisterMOUSEBUTTONUPeventAll(self, button):
        del(self.MOUSEBUTTONUP_event[button])

    def unregisterMOUSEBUTTONUPevent(self, button, function):
        try:
            self.MOUSEBUTTONUP_event[button].remove(function)
        except:
            pass

    def unregisterMOUSEBUTTONUPeventAllKEys(self):
        self.MOUSEBUTTONUP_event = {}


    def registerKEYDOWNlistener(self, function):
        self.KEYDOWN_listener.append(function)

    def unregisterKEYDOWNlistener(self, function):
        del self.KEYDOWN_listener[self.KEYDOWN_listener.index(function)]

    def unregisterAllKEYDOWNlisteners(self):
        self.KEYDOWN_listener = []