Correction de typage.
[minwii.git] / src / minwii / app.py
1 # -*- coding: utf-8 -*-
2 """
3 l'application winwii
4
5 $Id$
6 $URL$
7 """
8
9 import pygame
10 from pgu.gui import Theme
11 from pgu.gui import Desktop
12 from pgu.gui import QUIT
13 from minwii.widgets.launch import LaunchScreen
14 from minwii.widgets.home import Home
15 from minwii.widgets.playingscreen import SongPlayingScreen, PlayingScreen
16 from minwii.widgets.instrumentselector import InstrumentSelector
17 from minwii.synth import Synth
18 from minwii.eventutils import EventDispatcher
19 from minwii.musicxml import musicXml2Song
20 from minwii.config import SONG_FILE_PATH
21 from minwii.config import SCREEN_RESOLUTION
22 from minwii.globals import PLAYING_MODES_DICT
23 from minwii.log import console, LOG_FORMAT_VERSION, envLogger
24 import os.path
25
26 wiiuse = None
27 pygame_wiimouse = None
28
29 SCREEN_HOME = 0
30 SCREEN_INSTRUMENTS = 1
31 SCREEN_PLAY = 2
32
33
34
35 class MinWii(object):
36
37 def __init__(self, wiimoteSupport=True, fullscreen=False) :
38 envLogger.info('winwii log format version : %s', LOG_FORMAT_VERSION)
39 self.wiimoteSupport = wiimoteSupport
40 self.fullscreen = fullscreen
41 LaunchScreen()
42 themedir = __file__.split(os.path.sep)[:-1] + ['widgets', 'data', 'minwii_theme']
43 themedir = os.path.sep.join(themedir)
44 theme = Theme(themedir)
45 self.app = Desktop(theme=theme)
46 self.synth = Synth()
47 self.screenResolution = SCREEN_RESOLUTION
48 envLogger.info('résolution écran : %s', self.screenResolution)
49 self.nwiimotes = 0
50 self.initWiimotes()
51 self.firstSong = True
52 self.screen = SCREEN_HOME
53 self.playingScreen = None
54
55 def initWiimotes(self) :
56 if self.wiimoteSupport :
57 global wiiuse
58 from pywiiuse import PyWiiUse as wiiuse
59 global pygame_wiimouse
60 from pywiiuse import pygame_wiimouse
61 from minwii.config import IR_POSITION
62 pygame_wiimouse.init(5, 5, self.screenResolution, IR_POSITION) # look for 5, wait 5 seconds
63 self.nwiimotes = nwiimotes = pygame_wiimouse.get_count()
64 console.debug('wiimotes found : %d', nwiimotes)
65 self.WT = WT = pygame_wiimouse.WT
66 WT.setEventCallBack(self._wiimotesEventCallBack)
67 WT.pause()
68 else :
69 self.WT = _WTFacade()
70
71 def _wiimotesEventCallBack(self, wt, id, wmp) :
72 if self.screen == SCREEN_PLAY :
73 pygame_wiimouse._default_event_cb(wt, id, wmp)
74
75 wm = wmp[0]
76 if id == self.nwiimotes - 1 :
77 # le bouton Home de la télécommande permet de sortir
78 # (envoi d'un événement clavier « q »)
79 if wiiuse.is_just_pressed(wm, wiiuse.button['Home']) :
80 event = pygame.event.Event(pygame.KEYDOWN,
81 key = pygame.K_q,
82 unicode = u'q')
83 pygame.event.post(event)
84 elif wiiuse.is_just_pressed(wm, wiiuse.button['+']) :
85 self.synth.inc_gain()
86 elif wiiuse.is_just_pressed(wm, wiiuse.button['-']) :
87 self.synth.dec_gain()
88 elif wiiuse.is_just_pressed(wm, wiiuse.button['1']) and self.playingScreen :
89 self.playingScreen.tempoTrimUp()
90 elif wiiuse.is_just_pressed(wm, wiiuse.button['2']) and self.playingScreen :
91 self.playingScreen.tempoTrimDown()
92
93 elif self.screen in (SCREEN_HOME, SCREEN_INSTRUMENTS) :
94 pygame_wiimouse._full_mouse_event_cb(wt, id, wmp)
95
96 def run(self) :
97 "séquençage de l'affichage des écrans"
98 displayFlags = 0
99 if self.fullscreen :
100 displayFlags = displayFlags | pygame.FULLSCREEN
101 pygame.display.set_mode(self.screenResolution, displayFlags)
102 pygame.display.set_caption('MINWii')
103 WT = self.WT
104
105 songFile, playMode, displayNotes, wiimoteIndex = '', 'NORMAL', True, 0
106
107 while True :
108 WT.resume()
109 WT.selectWiimote(self.nwiimotes - 1) # la télécommande est la souris
110 self.screen = SCREEN_HOME
111 exit, songFile, playMode, displayNotes, wiimoteIndex = \
112 self.selectSongAndOptions(songFile, playMode, displayNotes, wiimoteIndex)
113 if exit : break
114
115
116 self.screen = SCREEN_INSTRUMENTS
117 instrumentDescription = self.selectInstrument()
118 if not instrumentDescription :
119 continue
120
121 self.screen = SCREEN_PLAY
122 WT.selectWiimote(wiimoteIndex)
123 self.runPlayingScreen(songFile, playMode, displayNotes, instrumentDescription)
124 WT.pause()
125
126
127 def selectSongAndOptions(self, songFile, playMode, displayNotes, wiimoteIndex) :
128 """ lance l'écran de paramétrage et retourne un tuple comportant :
129 - drapeau de sortie de l'application (booléen)
130 - chemin du fichier de la chanson
131 - mode (entier)
132 - affichage des notes (booléen)
133 - wiimote sélectionnée (entier)
134 """
135 home = Home(songPath=SONG_FILE_PATH,
136 songFile=songFile,
137 playMode=playMode,
138 displayNotes=displayNotes,
139 wiimoteIndex=wiimoteIndex,
140 nwiimotes=self.nwiimotes)
141 app = self.app
142 home.connect(QUIT, app.quit)
143 app.run(home)
144 app.close(home)
145
146 #logging
147 if home.exitApp :
148 console.debug("sortie de l'application")
149 else :
150 actual_wiimotes = self.WT.get_count()
151 if self.firstSong :
152 self.firstSong = False
153 else :
154 envLogger.info('NEW_LOG_FILE')
155 console.info('chanson : %s', home.songFile)
156 console.info('mode : %s', home.modeSelect.value)
157 if actual_wiimotes is None :
158 console.info('HID : souris')
159 elif actual_wiimotes == 0 :
160 console.info('HID : souris (pas de wiimote trouvée)')
161 else :
162 console.info('HID : wiimote %d', home.selectedWiimote.value + 1)
163 #---
164
165 return (home.exitApp,
166 home.songFile,
167 home.selectedPlayMode,
168 home.displayNotes,
169 home.selectedWiimoteIndex)
170
171 def selectInstrument(self) :
172 """ lance l'écran de sélection de l'instrument et retourne
173 un dictionnaire comportant la description de l'instrument
174 """
175 selector = InstrumentSelector()
176 selector.run()
177 selector.stop()
178 pygame.event.clear()
179 EventDispatcher.reset()
180 instru = selector.selectedInstrument
181 if instru :
182 console.info('instrument : %s', instru['name'])
183 console.info('preset : %d', instru['preset'])
184 console.info('bank : %d', instru['bank'])
185 console.info('ajustement octave : %d', instru['octave'])
186 return instru
187
188 def runPlayingScreen(self, songFile, playMode, displayNotes, instrumentDescription) :
189 """ Lance l'écran de jeu principal avec la chanson 'songFile' dans le mode 'playMode'
190 avec l'instrument midi 'instrumentDescription'.
191 """
192 playMode = PLAYING_MODES_DICT[playMode]
193 bank, preset = instrumentDescription['bank'], instrumentDescription['preset']
194 octave = instrumentDescription['octave']
195 self.synth.adjust_octave(0, octave)
196 self.synth.program_select(0, bank, preset)
197 if playMode == PLAYING_MODES_DICT['IMPRO'] :
198 playingScreen = PlayingScreen(self.synth, displayNotes=displayNotes)
199 else :
200 song = musicXml2Song(songFile)
201 self.playingScreen = playingScreen = SongPlayingScreen(self.synth, song, mode=playMode, displayNotes=displayNotes)
202 playingScreen.run()
203 pygame.event.clear()
204 EventDispatcher.reset()
205 self.playingScreen = None
206
207
208 class _WTFacade :
209 """ Classe utilitaire pour singer l'api
210 de pygame_wiimouse en cas d'abscence de wiimote.
211 """
212 selectWimoteIndex = 0
213 def pause(self):
214 pass
215 def resume(self):
216 pass
217 def selectWiimote(self, i):
218 pass
219 def get_count(self) :
220 return None