ménage.
[minwii.git] / src / gui / InstrumentChoice.py
1 '''
2 Created on 15 juil. 2009
3
4 @author: Samuel Benveniste
5 '''
6
7 import pygame
8 import pygame.midi
9 import sys
10 import time
11 import pickle
12
13 from numpy import array
14 from numpy.linalg import norm
15
16 from math import floor
17
18 from gui.constants import *
19 from PlayingScreen import PlayingScreen
20 from instruments.Instrument import Instrument
21 from cursor.WarpingCursor import *
22 from controllers.Wiimote import Wiimote
23 from logging.EventLog import EventLog
24 from logging.PickleableEvent import PickleableEvent
25
26 class InstrumentChoice:
27 '''
28 The screen for choosing instruments
29
30 instruments:
31 The available instruments
32 wiimotes:
33 The wiimotes used in this session
34 window:
35 The main display window
36 screen:
37 The main display surface
38 clock:
39 The clock used to animate the screen
40 savedScreen:
41 The background that is painted every time
42 playerScreen:
43 The buffer for painting everything before bliting
44 width:
45 The width of the window in pixels
46 height:
47 The height of the window in pixels
48 done:
49 Goes to True when all instruments have been selected
50 cursorPositions:
51 The positions of the cursors on the screen, in pixels
52 imageRects:
53 The rectangles where the images of the instruments are located
54 focus:
55 The numbers of the instruments currently in focus
56 '''
57
58 def __init__(self, instruments, wiimotes, window, screen, clock, joys, portOffset, activeWiimotes, eventLog=None, replay = False, logFilePath = None, scaleFactor = 1):
59 '''
60 Constructor
61
62 instruments:
63 The instruments for this session
64 wiimotes:
65 The wiimotes used in this session
66 '''
67 self.scaleFactor = scaleFactor
68
69 self.instruments = instruments
70 self.wiimotes = wiimotes
71 self.window = window
72 self.screen = screen
73 self.clock = clock
74 self.width = int(floor(screen.get_width()*self.scaleFactor))
75 self.height = int(floor(screen.get_height()*self.scaleFactor))
76 self.blitOrigin = ((self.screen.get_width()-self.width)/2,(self.screen.get_height()-self.height)/2)
77 self.joys = joys
78 self.portOffset = portOffset
79
80 self.activeWiimotes = activeWiimotes
81
82 self.currentWiimote = 0
83 while not self.activeWiimotes[self.currentWiimote] :
84 self.currentWiimote += 1
85 self.done = False
86
87 self.cursorPositions = []
88 self.imageRects = []
89 self.savedImageRects = []
90 self.focus = []
91 self.zoomed = []
92
93 if eventLog == None:
94 self.eventLog = EventLog()
95 self.replay = False
96 else:
97 self.eventLog = eventLog
98 self.replay = replay
99 print self.replay
100
101 #There are 3 instruments per row, up to 9 instruments
102 #Draw their images on the screen
103 self.savedScreen = pygame.Surface(self.screen.get_size())
104 self.savedScreen.fill((255, 255, 255))
105 for i in range(len(self.instruments)) :
106 drawPos = array([(self.width / 3) * (i % 3), (self.height / 3) * (i / 3)])
107 curImage = pygame.image.load(self.instruments[i].image).convert_alpha()
108 scaledImage = pygame.transform.smoothscale(curImage, (self.width / 3, self.height / 3))
109 self.imageRects.append(self.savedScreen.blit(scaledImage, drawPos + self.blitOrigin))
110 self.savedImageRects = self.imageRects[:]
111 #Draw the initial cursor on the buffer
112 self.playerScreen = pygame.Surface(self.savedScreen.get_size())
113 self.playerScreen.blit(self.savedScreen, (0, 0))
114
115 for i in range(len(self.wiimotes)):
116 #Create the list of instrument focus (one focus per wiimote)
117 self.focus.append(0)
118 self.zoomed.append(None)
119 #Set the screen for the cursors (it can't be set before)
120 self.wiimotes[i].cursor.screen = self.playerScreen
121 self.cursorPositions.append(self.wiimotes[i].cursor.centerPosition)
122
123 self.wiimotes[self.currentWiimote].cursor.blit(self.playerScreen)
124
125 #The main loop
126 while self.done == False :
127
128 #Clear the cursors from the screen
129 self.playerScreen.blit(self.savedScreen, (0, 0))
130
131 # Limit frame speed to 50 FPS
132 #
133 timePassed = self.clock.tick(50)
134
135 if self.replay:
136 self.eventLog.update(timePassed)
137 pickledEventsToPost = self.eventLog.getPickledEvents()
138 for pickledEvent in pickledEventsToPost:
139 pygame.event.post(pickledEvent.event)
140
141 events = pygame.event.get()
142
143 if not self.replay:
144 pickledEvents = [PickleableEvent(event.type,event.dict) for event in events if self.eventFilter(event)]
145 if pickledEvents != [] :
146 self.eventLog.appendEventGroup(pickledEvents)
147
148 for event in events:
149 self.input(event)
150
151
152 if self.zoomed[self.currentWiimote] != None :
153 self.imageRects = self.savedImageRects[:]
154 #inflate the chosen rect
155 zoomedRectNumber = self.zoomed[self.currentWiimote]
156 newRect = zoomRect(self.imageRects[zoomedRectNumber], 1.3)
157 self.imageRects[zoomedRectNumber] = newRect
158 curImage = pygame.image.load(self.instruments[zoomedRectNumber].image).convert_alpha()
159 self.scaledImage = pygame.transform.smoothscale(curImage, newRect.size)
160 self.playerScreen.blit(self.scaledImage, newRect.topleft)
161
162 for i in range(len(self.wiimotes)):
163 self.wiimotes[i].cursor.update(timePassed, self.cursorPositions[i])
164
165 self.wiimotes[self.currentWiimote].cursor.blit(self.playerScreen)
166
167 if self.zoomed[self.currentWiimote] != None and self.imageRects[self.zoomed[self.currentWiimote]].collidepoint(self.cursorPositions[self.currentWiimote]):
168 self.focus[self.currentWiimote] = self.zoomed[self.currentWiimote]
169 pygame.draw.rect(self.playerScreen, pygame.Color(0, 255, 0, 255), self.imageRects[self.zoomed[self.currentWiimote]], 10)
170 else:
171 for i in range(len(self.imageRects)) :
172 if self.imageRects[i].collidepoint(self.cursorPositions[self.currentWiimote]):
173 self.focus[self.currentWiimote] = i
174 pygame.draw.rect(self.playerScreen, pygame.Color(0, 255, 0, 255), self.imageRects[i], 10)
175 if self.zoomed[self.currentWiimote] != None:
176 self.playerScreen.blit(self.scaledImage, self.imageRects[self.zoomed[self.currentWiimote]].topleft)
177
178 self.screen.blit(self.playerScreen, (0, 0))
179
180 pygame.display.flip()
181
182 def input(self, event):
183 if event.type == pygame.QUIT:
184 pygame.midi.quit()
185 sys.exit()
186 if event.type == pygame.JOYAXISMOTION:
187 self.updateCursorPositionFromJoy(event)
188 if event.type == pygame.JOYBUTTONDOWN :
189 self.assignInstrumentToWiimote(event)
190 if event.type == pygame.MOUSEBUTTONDOWN:
191 if self.zoomed[self.currentWiimote] == self.focus[self.currentWiimote]:
192 self.assignInstrumentToMouse(event)
193 else:
194 self.zoomed[self.currentWiimote] = self.focus[self.currentWiimote]
195 if event.type == pygame.MOUSEMOTION:
196 self.updateCursorPositionFromMouse(event)
197
198 def updateCursorPositionFromJoy(self, joyEvent):
199 joyName = pygame.joystick.Joystick(joyEvent.joy).get_name()
200 print joyName
201 correctedJoyId = joyNames.index(joyName)
202 if self.activeWiimotes[correctedJoyId]:
203 if correctedJoyId < len(self.cursorPositions):
204 if joyEvent.axis == 0 :
205 self.cursorPositions[correctedJoyId] = (int((joyEvent.value + 1) / 2 * self.screen.get_width()), self.cursorPositions[correctedJoyId][1])
206 if joyEvent.axis == 1 :
207 self.cursorPositions[correctedJoyId] = (self.cursorPositions[correctedJoyId][0], int((joyEvent.value + 1) / 2 * self.screen.get_height()))
208
209 def assignInstrumentToWiimote(self, joyEvent):
210 joyName = pygame.joystick.Joystick(joyEvent.joy).get_name()
211 correctedJoyId = joyNames.index(joyName)
212 if self.activeWiimotes[correctedJoyId]:
213 if self.zoomed[correctedJoyId] == self.focus[correctedJoyId]:
214 self.wiimotes[correctedJoyId].instrument = self.instruments[self.focus[correctedJoyId]]
215 self.zoomed[correctedJoyId] = None
216 self.imageRects = self.savedImageRects[:]
217 if self.currentWiimote<len(self.wiimotes)-1:
218 self.currentWiimote = self.currentWiimote+1
219 else:
220 self.zoomed[correctedJoyId] = self.focus[correctedJoyId]
221 if self.hasFinished():
222 self.done = True
223
224 def updateCursorPositionFromMouse(self, mouseEvent):
225 correctedJoyId = 0
226 while not self.activeWiimotes[correctedJoyId] :
227 correctedJoyId += 1
228 self.cursorPositions[correctedJoyId] = mouseEvent.pos
229
230 def assignInstrumentToMouse(self, mouseEvent):
231 correctedJoyId = 0
232 while not self.activeWiimotes[correctedJoyId] :
233 correctedJoyId += 1
234 self.wiimotes[correctedJoyId].instrument = self.instruments[self.focus[correctedJoyId]]
235 if self.hasFinished():
236 self.done = True
237
238 def hasFinished(self):
239 finished = True
240 for i in range(len(self.wiimotes)):
241 if self.wiimotes[i].instrument == None and self.activeWiimotes[i]:
242 finished = False
243 return(finished)
244
245 def eventFilter(self, event):
246 c = event.type
247 if c == 17:
248 return False
249 elif c == pygame.MOUSEMOTION or pygame.MOUSEBUTTONDOWN or pygame.MOUSEBUTTONUP or pygame.JOYAXISMOTION or pygame.JOYBUTTONDOWN or pygame.JOYBUTTONUP or pygame.KEYDOWN:
250 return True
251 else:
252 return False
253
254 def zoomRect(rect, ratio):
255 zoomedRect = rect.inflate(int(floor((ratio - 1) * rect.width)), int(floor((ratio - 1) * rect.height)))
256 return(zoomedRect)
257
258 if __name__ == "__main__":
259 pygame.init()
260 #pygame.event.set_blocked([pygame.MOUSEBUTTONDOWN,pygame.MOUSEBUTTONUP,pygame.MOUSEMOTION])
261
262 pygame.midi.init()
263 instruments = [Instrument(majorScale, i + 1, "".join(["../instruments/instrumentImages/", instrumentImagePathList[i], ".jpg"]), octaves[i]) for i in range(9)]
264
265 joys = [pygame.joystick.Joystick(id).get_name() for id in range(pygame.joystick.get_count())]
266 joyOffset = joys.index(joyNames[0])
267 pygame.joystick.Joystick(joyOffset).init()
268 print(joyOffset)
269
270 ports = [pygame.midi.get_device_info(id)[1] for id in range(pygame.midi.get_count())]
271 portOffset = ports.index(portNames[0])
272 print(portOffset)
273
274 window = pygame.display.set_mode((1280, 1024),pygame.FULLSCREEN)
275 screen = pygame.display.get_surface()
276 clock = pygame.time.Clock()
277 cursorImages = createImageListFromPath('../cursor/cursorImages/black', 11)
278 durations = [75 for i in range(len(cursorImages))]
279
280 extsc = False
281 casc = False
282
283 jadbt = [3, 4, 5, 3, 4, 4, 5, 6, 6, 5, 3, 3, 4, 5, 3, 4, 4, 5, 6, 7, 3]
284 song = None
285
286 cursors = [WarpingCursor(None, cursorImages, durations, (300 * i, 300 * i)) for i in range(1)]
287 wiimotes = [Wiimote(i, i + portOffset, None, None, cursors[i]) for i in range(1)]
288 choice = InstrumentChoice(instruments, wiimotes, window, screen, clock, joyOffset, portOffset)
289 play = PlayingScreen(choice, song, casc, extsc)
290 for wiimote in wiimotes:
291 del wiimote.port
292
293 f = file('temp.pkl', 'w')
294 pickler = pickle.Pickler(f)
295 pickler.dump(play.eventLog.eventGroups)
296 pickler.dump(play.eventLog.times)
297 f.close()
298
299 f = file('temp.pkl', 'r')
300 unpickler = pickle.Unpickler(f)
301 eventGroups = unpickler.load()
302 times = unpickler.load()
303 f.close()
304 eventLog = EventLog(eventGroups,times)
305
306 cursors = [WarpingCursor(None, cursorImages, durations, (300 * i, 300 * i)) for i in range(1)]
307 wiimotes = [Wiimote(i, i + portOffset, None, None, cursors[i]) for i in range(1)]
308 choice2 = InstrumentChoice(instruments, wiimotes, window, screen, clock, joyOffset, portOffset, eventLog)
309 play = PlayingScreen(choice2, song, casc, extsc)
310
311 for wiimote in wiimotes:
312 del wiimote.port
313
314 pygame.midi.quit()
315 sys.exit()