cf1a670f8a2dcc9062b2a2683f68b4a8107b35de
[minwii.git] / src / logging / LogPGUPlayer.py
1 '''
2 Created on 25 janv. 2010
3
4 @author: Samuel Benveniste
5 '''
6
7 import pygame
8 import pickle
9
10 from pgu import gui as pguGui
11
12 from gui import constants
13
14 from instruments.Instrument import Instrument
15 from songs.Song import Song
16 from cursor.WarpingCursor import *
17 from gui.StaticFamiliarizer import StaticFamiliarizer
18 from gui.SongFamiliarizer import SongFamiliarizer
19 from gui.SongPlayingScreen import SongPlayingScreen
20 from gui.DummyInstrumentChoice import DummyInstrumentChoice
21 from controllers.Wiimote import Wiimote
22
23 class LogPGUPlayer(pguGui.Desktop):
24 '''
25 classdocs
26 '''
27
28
29 def __init__(self,rate):
30 '''
31 Constructor
32 '''
33 self.firstStepDurations = []
34 self.firstStepClicks = []
35 self.firstStepClicksIn = []
36
37 self.secondStepDurations = []
38 self.secondStepClicks = []
39 self.secondStepClicksIn = []
40
41 self.thirdStepDurations = []
42 self.thirdStepClicks = []
43 self.thirdStepClicksIn = []
44
45 self.songDurations = []
46 self.songClicks = []
47 self.songClicksIn = []
48 self.songClicksPerMinute = []
49 self.songClicksInPerMinute = []
50 self.songTotalDurations = []
51
52 self.meanTimeBetweenNotes = []
53
54 pguGui.Desktop.__init__(self)
55
56 self.replayRate = rate
57 #pguGui.theme.load('../data/themes/default')
58
59 self.connect(pguGui.QUIT,self.quit,None)
60
61 main = pguGui.Container(width=500, height=400) #, background=(220, 220, 220) )
62
63
64 main.add(pguGui.Label("File Dialog Example", cls="h1"), 20, 20)
65
66
67 td_style = {'padding_right': 10}
68 t = pguGui.Table()
69 t.tr()
70 t.td( pguGui.Label('File Name:') , style=td_style )
71 self.input_file = pguGui.Input()
72 t.td( self.input_file, style=td_style )
73 self.browseButton = pguGui.Button("Browse...")
74 t.td( self.browseButton, style=td_style )
75 self.browseButton.connect(pguGui.CLICK, self.open_file_browser, None)
76
77 self.goButton = pguGui.Button("Go")
78
79 self.goButton.connect(pguGui.CLICK, self.goButtonClicked,None)
80
81 self.quitButton = pguGui.Button("Fin")
82 self.quitButton.connect(pguGui.CLICK,self.quit,None)
83
84 t.td( self.browseButton, style=td_style )
85 t.td( self.goButton, style=td_style )
86 t.td( self.quitButton, style=td_style )
87
88 main.add(t, 20, 100)
89
90 self.run(main)
91 #import profile
92 #profile.run('app.run(main)')
93
94 def open_file_browser(self,data=None):
95 d = pguGui.FileDialog(path = "../../../saves")
96 d.connect(pguGui.CHANGE, self.handle_file_browser_closed, d)
97 d.open()
98
99
100 def handle_file_browser_closed(self,dlg):
101 if dlg.value: self.input_file.value = dlg.value
102
103 def goButtonClicked(self,data=None):
104 if self.input_file.value.endswith(".fmwi"):
105 f = file(self.input_file.value, 'r')
106 unpickler = pickle.Unpickler(f)
107 log = unpickler.load()
108 f.close()
109
110 log.eventLog.setReplayRate(self.replayRate)
111
112 pygame.midi.init()
113 instruments = [Instrument(log.scale, i + 1, "".join(["../instruments/instrumentImages/", constants.instrumentImagePathList[i], ".jpg"]), constants.octaves[i]) for i in range(9)]
114
115 joys = [[id,pygame.joystick.Joystick(id).get_name()] for id in range(pygame.joystick.get_count())]
116 for joy in joys:
117 if joy[1] in constants.joyNames:
118 pygame.joystick.Joystick(joy[0]).init()
119
120 ports = [pygame.midi.get_device_info(id)[1] for id in range(pygame.midi.get_count())]
121 portOffset = ports.index(constants.portNames[0])
122 print(portOffset)
123
124 screen = pygame.display.get_surface()
125 clock = pygame.time.Clock()
126 cursorImages=[['../cursor/cursorImages/black/10.png'],['../cursor/cursorImages/red/10.png'],['../cursor/cursorImages/blue/10.png'],['../cursor/cursorImages/green/10.png']]
127 durations = [75 for i in range(len(cursorImages))]
128
129 extsc = True
130 casc = False
131 easyMode = True
132
133 song = Song(constants.scaleDict["majorScale"],[3,9,6,4,1,8,5,7,2,10],True)
134
135 wiimoteCount = 4
136
137 cursors = [WarpingCursor(None, cursorImages[i], durations, (300 * i, 300 * i),'../cursor/cursorImages/black/flash.png') for i in range(wiimoteCount)]
138 wiimotes = [Wiimote(i, i + portOffset, None, instruments[0], cursors[i]) for i in range(wiimoteCount)]
139 dummyInstrumentChoice = DummyInstrumentChoice(wiimotes, window, screen, clock, joys, portOffset, log.activeWiimotes)
140 if log.level < 2 :
141 familiarize = StaticFamiliarizer(wiimotes, window, screen, clock, joys, portOffset,log.activeWiimotes,level = log.level,eventLog = log.eventLog,replay = True)
142 self.fillStaticFamiliarizerStats(familiarize)
143 elif log.level == 2 :
144 familiarize = SongFamiliarizer(wiimotes, window, screen, clock, joys, portOffset,song,log.activeWiimotes,casc,extsc,easyMode,eventLog = log.eventLog,replay = True)
145 self.fillSongFamiliarizerStats(familiarize)
146 else :
147 familiarize = SongPlayingScreen(dummyInstrumentChoice,constants.songDict["clairdelalune"],easyMode = True,eventLog = log.eventLog,replay = True)
148 self.fillSongStats(familiarize)
149
150 while familiarize.nextLevel != None :
151 if familiarize.nextLevel < 2 :
152 familiarize = StaticFamiliarizer(wiimotes, window, screen, clock, joys, portOffset,log.activeWiimotes,level = familiarize.nextLevel,eventLog = familiarize.eventLog,replay = True)
153 self.fillStaticFamiliarizerStats(familiarize)
154 elif familiarize.nextLevel == 2 :
155 familiarize = SongFamiliarizer(wiimotes, window, screen, clock, joys, portOffset,song,log.activeWiimotes,casc,extsc,easyMode,eventLog = familiarize.eventLog,replay = True)
156 self.fillSongFamiliarizerStats(familiarize)
157 else :
158 familiarize = SongPlayingScreen(dummyInstrumentChoice,constants.songDict["clairdelalune"],easyMode = True,eventLog = familiarize.eventLog,replay = True)
159 self.fillSongStats(familiarize)
160
161 for wiimote in wiimotes:
162 del wiimote.port
163
164 pygame.midi.quit()
165 self.printStatsToFile()
166
167 def fillStaticFamiliarizerStats(self,familiarizer):
168 if familiarizer.level == 0 :
169 self.firstStepClicks.append(familiarizer.clicks)
170 self.firstStepClicksIn.append(familiarizer.clicksIn)
171 self.firstStepDurations.append(familiarizer.duration)
172
173 if familiarizer.level == 1 :
174 self.secondStepClicks.append(familiarizer.clicks)
175 self.secondStepClicksIn.append(familiarizer.clicksIn)
176 self.secondStepDurations.append(familiarizer.duration)
177
178 def fillSongFamiliarizerStats(self,familiarizer):
179 self.thirdStepClicks.append(familiarizer.clicks)
180 self.thirdStepClicksIn.append(familiarizer.clicksIn)
181 self.thirdStepDurations.append(familiarizer.duration)
182
183 def fillSongStats(self,familiarizer):
184 self.songClicks.append(familiarizer.clicks)
185 self.songClicksIn.append(familiarizer.clicksIn)
186 self.songClicksPerMinute.append(familiarizer.clicksPerMinute)
187 self.songClicksInPerMinute.append(familiarizer.clicksInPerMinute)
188 self.songDurations.append(familiarizer.songDurations)
189 self.songTotalDurations.append(familiarizer.totalDuration)
190
191 def statsToFormattedString(self):
192 return("First step durations :\n"+"\n"+
193 str(self.firstStepDurations)+"\n"+"\n"+
194 "First step clicks :\n"+"\n"+
195 str(self.firstStepClicks)+"\n"+"\n"+
196 "First step clicksIn :\n"+"\n"+
197 str(self.firstStepClicksIn)+"\n"+"\n"+
198 "Second step durations :\n"+"\n"+
199 str(self.secondStepDurations)+"\n"+"\n"+
200 "Second step clicks :\n"+"\n"+
201 str(self.secondStepClicks)+"\n"+"\n"+
202 "Second step clicksIn :\n"+"\n"+
203 str(self.secondStepClicksIn)+"\n"+"\n"+
204 "Third step durations :\n"+"\n"+
205 str(self.thirdStepDurations)+"\n"+"\n"+
206 "Third step clicks :\n"+"\n"+
207 str(self.thirdStepClicks)+"\n"+"\n"+
208 "Third step clicksIn :\n"+"\n"+
209 str(self.thirdStepClicksIn)+"\n"+"\n"+
210 "song durations :\n"+"\n"+
211 str(self.songDurations)+"\n"+"\n"+
212 "song clicks :\n"+"\n"+
213 str(self.songClicks)+"\n"+"\n"+
214 "song clicksIn :\n"+"\n"+
215 str(self.songClicksIn)+"\n"+"\n"+
216 "song clicks per minute:\n"+"\n"+
217 str(self.songClicksPerMinute)+"\n"+"\n"+
218 "song clicksIn per minute :\n"+"\n"+
219 str(self.songClicksInPerMinute)+"\n"+"\n"+
220 "song total durations :\n"+"\n"+
221 str(self.songTotalDurations)+"\n"+"\n")
222
223 def printStatsToFile(self,path=None):
224 if path == None :
225 path = self.input_file.value.replace(".fmwi",".txt")
226 file = open(path,"w")
227 file.write("Log ID : "+self.input_file.value+"\n"+"\n")
228 file.write(self.statsToFormattedString())
229 file.close()
230
231 if __name__ == "__main__":
232 pygame.init()
233 modeResolution = (1024,768)
234 window = pygame.display.set_mode(modeResolution,pygame.FULLSCREEN)
235 logConfig = LogPGUPlayer(10000)
236 pygame.quit()
237