''' Created on 15 juil. 2009 @author: Samuel Benveniste ''' import pygame import pygame.midi import sys import time import pickle import random from numpy import array from numpy.linalg import norm from math import floor from gui.constants import * from PlayingScreen import PlayingScreen from instruments.Instrument import Instrument from cursor.WarpingCursor import * from controllers.Wiimote import Wiimote from logging.EventLog import EventLog from logging.PickleableEvent import PickleableEvent joyNames = ["PPJoy Virtual joystick 1", "PPJoy Virtual joystick 2", "PPJoy Virtual joystick 3"] portNames = ["Out To MIDI Yoke: 1"] majorScale = [55, 57, 59, 60, 62, 64, 65, 67, 69, 71, 72] minorScale = [55, 56, 58, 60, 62, 63, 65, 67, 68, 70, 72] myxolydianScale = [55, 57, 58, 60, 62, 64, 65, 67, 69, 70, 72] dorianScale = [55, 57, 58, 60, 62, 63, 65, 67, 69, 70, 72] instrumentImagePathList = ["piano", "guitare", "accordeon", "violon", "flute", "tuba", "orgue", "violoncelle", "celesta"] octaves = [0, -1, 0, 1, 1, -2, 0, -1, 0] class Familiarizer: ''' The screen for choosing instruments instruments: The available instruments wiimotes: The wiimotes used in this session window: The main display window screen: The main display surface clock: The clock used to animate the screen savedScreen: The background that is painted every time playerScreen: The buffer for painting everything before bliting width: The width of the window in pixels height: The height of the window in pixels done: Goes to True when all instruments have been selected cursorPositions: The positions of the cursors on the screen, in pixels imageRects: The rectangles where the images of the instruments are located focus: The numbers of the instruments currently in focus ''' def __init__(self, instruments, wiimotes, window, screen, clock, joys, portOffset, eventLog=None, replay = False, logFilePath = None, scaleFactor = 1, level = 1): ''' Constructor instruments: The instruments for this session wiimotes: The wiimotes used in this session ''' self.scaleFactor = scaleFactor self.instruments = instruments self.wiimotes = wiimotes self.window = window self.screen = screen self.clock = clock self.width = int(floor(screen.get_width()*self.scaleFactor)) self.height = int(floor(screen.get_height()*self.scaleFactor)) self.blitOrigin = ((self.screen.get_width()-self.width)/2,(self.screen.get_height()-self.height)/2) self.joys = joys self.portOffset = portOffset self.currentWiimote = 0 self.done = False self.cursorPositions = [] self.imageRects = [] self.displayedInstruments = [] self.savedImageRects = [] self.focus = [] self.level = level if eventLog == None: self.eventLog = EventLog() self.replay = False else: self.eventLog = eventLog self.replay = replay self.savedScreen = pygame.Surface(self.screen.get_size()) self.displayedInstruments.append(0) if level == 1 : self.displayedInstruments.append(2) self.savedScreen.fill((255, 255, 255)) for i in range(len(self.displayedInstruments)): self.imageRects.append(self.drawInstrument(self.displayedInstruments[i])) self.savedImageRects = self.imageRects[:] #Draw the initial cursor on the buffer self.playerScreen = pygame.Surface(self.savedScreen.get_size()) self.playerScreen.blit(self.savedScreen, (0, 0)) for i in range(len(self.wiimotes)): #Create the list of instrument focus (one focus per wiimote) self.focus.append(None) #Set the screen for the cursors (it can't be set before) self.wiimotes[i].cursor.screen = self.playerScreen self.cursorPositions.append(self.wiimotes[i].cursor.centerPosition) self.wiimotes[self.currentWiimote].cursor.blit(self.playerScreen) #The main loop while self.done == False : #Clear the cursors from the screen self.drawBackground() self.playerScreen.blit(self.savedScreen, (0, 0)) # Limit frame speed to 50 FPS # timePassed = self.clock.tick(50) if self.replay: self.eventLog.update(timePassed) pickledEventsToPost = self.eventLog.getPickledEvents() for pickledEvent in pickledEventsToPost: pygame.event.post(pickledEvent.event) events = pygame.event.get() if not self.replay: pickledEvents = [PickleableEvent(event.type,event.dict) for event in events if self.eventFilter(event)] if pickledEvents != [] : self.eventLog.appendEventGroup(pickledEvents) for event in events: self.input(event) for i in range(len(self.wiimotes)): self.wiimotes[i].cursor.update(timePassed, self.cursorPositions[i]) self.wiimotes[self.currentWiimote].cursor.blit(self.playerScreen) self.focus[self.currentWiimote] = None for i in range(len(self.imageRects)) : if self.imageRects[i].collidepoint(self.cursorPositions[self.currentWiimote]): self.focus[self.currentWiimote] = i self.screen.blit(self.playerScreen, (0, 0)) pygame.display.flip() def input(self, event): if event.type == pygame.QUIT: pygame.midi.quit() sys.exit() if event.type == pygame.JOYAXISMOTION: self.updateCursorPositionFromJoy(event) if event.type == pygame.JOYBUTTONDOWN : self.joyClicked(event) if event.type == pygame.MOUSEBUTTONDOWN: self.mouseClicked(event) if event.type == pygame.MOUSEMOTION: self.updateCursorPositionFromMouse(event) if event.type == pygame.KEYDOWN: if event.key == pygame.K_q: self.done = True def updateCursorPositionFromJoy(self, joyEvent): joyName = pygame.joystick.Joystick(joyEvent.joy).get_name() correctedJoyId = joyNames.index(joyName) if correctedJoyId < len(self.cursorPositions): if joyEvent.axis == 0 : self.cursorPositions[correctedJoyId] = (int((joyEvent.value + 1) / 2 * self.screen.get_width()), self.cursorPositions[correctedJoyId][1]) if joyEvent.axis == 1 : self.cursorPositions[correctedJoyId] = (self.cursorPositions[correctedJoyId][0], int((joyEvent.value + 1) / 2 * self.screen.get_height())) def assignInstrumentToWiimote(self, joyEvent): joyName = pygame.joystick.Joystick(joyEvent.joy).get_name() correctedJoyId = joyNames.index(joyName) if self.zoomed[correctedJoyId] == self.focus[correctedJoyId]: self.wiimotes[correctedJoyId].instrument = self.instruments[self.focus[correctedJoyId]] self.zoomed[correctedJoyId] = None self.imageRects = self.savedImageRects[:] if self.currentWiimote