Implémentation du bouton « Home » de la télécommande.

[minwii.git] / src / minwii / app.py

diff --git a/src/minwii/app.py b/src/minwii/app.py
index 55ea722..da2bcb8 100755 (executable)
--- a/src/minwii/app.py
+++ b/src/minwii/app.py
@@ -7,71 +7,108 @@ $URL$
 """
 
 import pygame
 """
 
 import pygame

+from pgu.gui import Theme

 from pgu.gui import Desktop
 from pgu.gui import QUIT
 from pgu.gui import Desktop
 from pgu.gui import QUIT

-from widgets.launch import LaunchScreen
-from widgets.home import Home
-from widgets.playingscreen import SongPlayingScreen, PlayingScreen
-from widgets.instrumentselector import InstrumentSelector
-from synth import Synth
-from eventutils import EventDispatcher
-from musicxml import musicXml2Song
-from config import SONG_FILE_PATH
-from globals import PLAYING_MODES_DICT
-from log import console, LOG_FORMAT_VERSION, envLogger
+from minwii.widgets.launch import LaunchScreen
+from minwii.widgets.home import Home
+from minwii.widgets.playingscreen import SongPlayingScreen, PlayingScreen
+from minwii.widgets.instrumentselector import InstrumentSelector
+from minwii.synth import Synth
+from minwii.eventutils import EventDispatcher
+from minwii.musicxml import musicXml2Song
+from minwii.config import SONG_FILE_PATH
+from minwii.config import SCREEN_RESOLUTION
+from minwii.globals import PLAYING_MODES_DICT
+from minwii.log import console, LOG_FORMAT_VERSION, envLogger
+import os.path
+
+wiiuse = None
+pygame_wiimouse = None
+
+SCREEN_HOME         = 0
+SCREEN_INSTRUMENTS  = 1
+SCREEN_PLAY         = 2
+

 
 
 class MinWii(object):
     
 
 
 class MinWii(object):
     

-    def __init__(self, wiimoteSupport=True) :
+    def __init__(self, wiimoteSupport=True, fullscreen=False) :

         envLogger.info('winwii log format version : %s', LOG_FORMAT_VERSION)
         self.wiimoteSupport = wiimoteSupport
         envLogger.info('winwii log format version : %s', LOG_FORMAT_VERSION)
         self.wiimoteSupport = wiimoteSupport

+        self.fullscreen = fullscreen

         LaunchScreen()
         LaunchScreen()

-        self.app = Desktop()
+        themedir = __file__.split(os.path.sep)[:-1] + ['widgets', 'data', 'minwii_theme']
+        themedir = os.path.sep.join(themedir)
+        theme = Theme(themedir)
+        self.app = Desktop(theme=theme)

         self.synth = Synth()
         self.synth = Synth()

-        self.screenResolution = (1024,768)
+        self.screenResolution = SCREEN_RESOLUTION

         envLogger.info('résolution écran : %s', self.screenResolution)
         self.nwiimotes = 0
         self.initWiimotes()
         self.firstSong = True
         envLogger.info('résolution écran : %s', self.screenResolution)
         self.nwiimotes = 0
         self.initWiimotes()
         self.firstSong = True

+        self.screen = SCREEN_HOME

     
     def initWiimotes(self) :
         if self.wiimoteSupport :
     
     def initWiimotes(self) :
         if self.wiimoteSupport :

+            global wiiuse
+            from pywiiuse import PyWiiUse as wiiuse
+            global pygame_wiimouse

             from pywiiuse import pygame_wiimouse
             from pywiiuse import pygame_wiimouse

-            pygame_wiimouse.init(4, 5, self.screenResolution) # look for 4, wait 5 seconds
+            from minwii.config import IR_POSITION
+            pygame_wiimouse.init(2, 5, self.screenResolution, IR_POSITION) # look for 4, wait 5 seconds

             self.nwiimotes = nwiimotes = pygame_wiimouse.get_count()
             console.debug('wiimotes found : %d', nwiimotes)
             self.WT = WT = pygame_wiimouse.WT
             self.nwiimotes = nwiimotes = pygame_wiimouse.get_count()
             console.debug('wiimotes found : %d', nwiimotes)
             self.WT = WT = pygame_wiimouse.WT

+            WT.setEventCallBack(self._wiimotesEventCallBack)

             WT.pause()
         else :
             self.WT = _WTFacade()
             WT.pause()
         else :
             self.WT = _WTFacade()

-
+    
+    def _wiimotesEventCallBack(self, wt, id, wmp) :
+        if self.screen == SCREEN_PLAY :
+            pygame_wiimouse._default_event_cb(wt, id, wmp)
+            wm = wmp[0]
+            if id == self.nwiimotes - 1 and \
+                wiiuse.is_just_pressed(wm, wiiuse.button['Home']) :
+                    event = pygame.event.Event(pygame.KEYDOWN,
+                                               key = pygame.K_q,
+                                               unicode = u'q')
+                    pygame.event.post(event)
+    

     def run(self) :
         "séquençage de l'affichage des écrans"
     def run(self) :
         "séquençage de l'affichage des écrans"

-
-        pygame.display.set_mode(self.screenResolution)
-        pygame.display.set_caption('MinWii')
+        displayFlags = 0
+        if self.fullscreen :
+            displayFlags = displayFlags | pygame.FULLSCREEN
+        pygame.display.set_mode(self.screenResolution, displayFlags)
+        pygame.display.set_caption('MINWii')

         WT = self.WT
         WT = self.WT

+        
+        songFile, playMode, wiimoteIndex = '', 'NORMAL', 0

 
         while True :
 
         while True :

-
-            exit, songFile, playMode, selectedWiimoteIndex = self.selectSongAndOptions()
+            WT.resume()
+            self.screen = SCREEN_HOME
+            exit, songFile, playMode, wiimoteIndex = \
+                self.selectSongAndOptions(songFile, playMode, wiimoteIndex)

             if exit : break
             
             if exit : break
             

-            WT.selectWiimote(selectedWiimoteIndex)
-            WT.resume()
-
+            WT.selectWiimote(wiimoteIndex)
+            
+            self.screen = SCREEN_INSTRUMENTS

             instrumentDescription = self.selectInstrument()
             if not instrumentDescription :
             instrumentDescription = self.selectInstrument()
             if not instrumentDescription :

-                WT.pause()

                 continue
             
                 continue
             

+            self.screen = SCREEN_PLAY

             self.runPlayingScreen(songFile, playMode, instrumentDescription)
             self.runPlayingScreen(songFile, playMode, instrumentDescription)

-            

             WT.pause()
     
     
             WT.pause()
     
     

-    def selectSongAndOptions(self) :
+    def selectSongAndOptions(self, songFile, playMode, wiimoteIndex) :

         """ lance l'écran de paramétrage et retourne un tuple comportant :
             - drapeau de sortie de l'application (booléen)
             - chemin du fichier de la chanson
         """ lance l'écran de paramétrage et retourne un tuple comportant :
             - drapeau de sortie de l'application (booléen)
             - chemin du fichier de la chanson


@@ -79,6 +116,9 @@ class MinWii(object):
             - wiimote sélectionnée (entier)
         """
         home = Home(songPath=SONG_FILE_PATH,
             - wiimote sélectionnée (entier)
         """
         home = Home(songPath=SONG_FILE_PATH,

+                    songFile=songFile,
+                    playMode=playMode,
+                    wiimoteIndex=wiimoteIndex,

                     nwiimotes=self.nwiimotes)
         app = self.app
         home.connect(QUIT, app.quit)
                     nwiimotes=self.nwiimotes)
         app = self.app
         home.connect(QUIT, app.quit)


@@ -106,8 +146,8 @@ class MinWii(object):
         
         return (home.exitApp,
                 home.songFile,
         
         return (home.exitApp,
                 home.songFile,

-                home.modeSelect.value,
-                home.selectedWiimote.value)
+                home.selectedPlayMode,
+                home.selectedWiimoteIndex)

     
     def selectInstrument(self) :
         """ lance l'écran de sélection de l'instrument et retourne
     
     def selectInstrument(self) :
         """ lance l'écran de sélection de l'instrument et retourne


@@ -131,12 +171,15 @@ class MinWii(object):
             avec l'instrument midi 'instrumentDescription'.
         """
         playMode = PLAYING_MODES_DICT[playMode]
             avec l'instrument midi 'instrumentDescription'.
         """
         playMode = PLAYING_MODES_DICT[playMode]

-        song = musicXml2Song(songFile)

         bank, preset = instrumentDescription['bank'], instrumentDescription['preset']
         octave = instrumentDescription['octave']
         self.synth.adjust_octave(0, octave)
         self.synth.program_select(0, bank, preset)
         bank, preset = instrumentDescription['bank'], instrumentDescription['preset']
         octave = instrumentDescription['octave']
         self.synth.adjust_octave(0, octave)
         self.synth.program_select(0, bank, preset)

-        playingScreen = SongPlayingScreen(self.synth, song, mode=playMode)
+        if playMode == PLAYING_MODES_DICT['IMPRO'] :
+            playingScreen = PlayingScreen(self.synth)
+        else :
+            song = musicXml2Song(songFile)
+            playingScreen = SongPlayingScreen(self.synth, song, mode=playMode)

         playingScreen.run()
         pygame.event.clear()
         EventDispatcher.reset()
         playingScreen.run()
         pygame.event.clear()
         EventDispatcher.reset()