Ajout des boutons de tri.
[minwii.git] / src / minwii / widgets / playingscreen.py
index 0735825..9c1ebfb 100755 (executable)
@@ -7,22 +7,23 @@ $Id$
 $URL$
 """
 import pygame
 $URL$
 """
 import pygame
-from cursors import WarpingCursor
-from column import Column
-import events
-from eventutils import event_handler, EventDispatcher, EventHandlerMixin
 import types
 import types
-from musicxml import Tone
 
 
-from config import FRAMERATE
-from config import FIRST_HUE
-from config import MIDI_VELOCITY_RANGE
-from config import MIDI_PAN_RANGE
-from config import MIDI_VELOCITY_WRONG_NOTE_ATTN
+import minwii.events as events
+from minwii.log import eventLogger
+from minwii.eventutils import event_handler, EventDispatcher, EventHandlerMixin
+from minwii.musicxml import Tone
+from minwii.config import FRAMERATE
+from minwii.config import FIRST_HUE
+from minwii.config import MIDI_VELOCITY_RANGE
+from minwii.config import MIDI_PAN_RANGE
+from minwii.config import MIDI_VELOCITY_WRONG_NOTE_ATTN
+from minwii.globals import BACKGROUND_LAYER
+from minwii.globals import CURSOR_LAYER
+from minwii.globals import PLAYING_MODES_DICT
 
 
-from globals import BACKGROUND_LAYER
-from globals import CURSOR_LAYER
-from globals import PLAYING_MODES_DICT
+from cursors import WarpingCursor
+from column import Column
 
 class PlayingScreenBase(pygame.sprite.LayeredDirty, EventHandlerMixin) :
 
 
 class PlayingScreenBase(pygame.sprite.LayeredDirty, EventHandlerMixin) :
 
@@ -99,7 +100,9 @@ class PlayingScreenBase(pygame.sprite.LayeredDirty, EventHandlerMixin) :
 
     @event_handler(pygame.KEYDOWN)       
     def handleKeyDown(self, event) :
 
     @event_handler(pygame.KEYDOWN)       
     def handleKeyDown(self, event) :
-        if event.key == pygame.K_q or event.unicode == u'q':
+        if event.key == pygame.K_q or \
+            event.unicode == u'q' or \
+            pygame.K_ESCAPE:
             self.stop()
     
     @event_handler(pygame.MOUSEBUTTONDOWN)
             self.stop()
     
     @event_handler(pygame.MOUSEBUTTONDOWN)
@@ -160,22 +163,26 @@ class PlayingScreen(PlayingScreenBase) :
 
     def __init__(self, synth) :
         distinctNotes = []
 
     def __init__(self, synth) :
         distinctNotes = []
+        self.currentColumn = None
         for midi in self.scale :
             tone = Tone(midi)
             distinctNotes.append(tone)
         
         super(PlayingScreen, self).__init__(synth, distinctNotes)
             
         for midi in self.scale :
             tone = Tone(midi)
             distinctNotes.append(tone)
         
         super(PlayingScreen, self).__init__(synth, distinctNotes)
             
-    @event_handler(events.NOTEON)
-    def noteon(self, evt) :
-        tone = evt.tone
-        self.synth.noteon(0, tone.midi, 96)
-
-    @event_handler(events.NOTEOFF)
-    def noteoff(self, evt) :
-        tone = evt.tone
-        self.synth.noteoff(0, tone.midi)
+    @event_handler(events.COLDOWN)
+    def noteon(self, event) :
+        col = event.column
+        col.update(True)
+        self.currentColumn = col
+        self.playnote(col, event.pos)
 
 
+    @event_handler(events.COLUP)
+    def noteoff(self, event) :
+        if self.currentColumn :
+            self.currentColumn.update(False)
+            self.synth.noteoff(0, self.currentColumn.tone.midi)
+    
 
 class SongPlayingScreen(PlayingScreenBase) :
     
 
 class SongPlayingScreen(PlayingScreenBase) :
     
@@ -226,7 +233,7 @@ class SongPlayingScreen(PlayingScreenBase) :
     def handleEasyColumnOver(self, event) :
         col = event.column
         if col.state and \
     def handleEasyColumnOver(self, event) :
         col = event.column
         if col.state and \
-           any(event.mouseEvent.buttons) and \
+           self.cursor.pressed and \
            not self.currentNotePlayed :
             self.playnote(col, event.pos)
             SongPlayingScreen.setNoteTimeout(
            not self.currentNotePlayed :
             self.playnote(col, event.pos)
             SongPlayingScreen.setNoteTimeout(
@@ -239,7 +246,7 @@ class SongPlayingScreen(PlayingScreenBase) :
     def handleNormalColumnOver(self, event) :
         col = event.column
         if col.state and \
     def handleNormalColumnOver(self, event) :
         col = event.column
         if col.state and \
-           any(event.mouseEvent.buttons) and \
+           self.cursor.pressed and \
            not self.currentNotePlayed :
             self.playnote(col, event.pos)
             self.currentNotePlayed = True
            not self.currentNotePlayed :
             self.playnote(col, event.pos)
             self.currentNotePlayed = True
@@ -292,8 +299,17 @@ class SongPlayingScreen(PlayingScreenBase) :
     def displayNext(self, event=None) :
         if self.currentColumn:
             self.currentColumn.update(False)
     def displayNext(self, event=None) :
         if self.currentColumn:
             self.currentColumn.update(False)
-        note, verseIndex = self.noteIterator.next()
-        syllabus = note.lyrics[verseIndex].syllabus()
+        try :
+            note, verseIndex = self.noteIterator.next()
+        except StopIteration :
+            self.noteIterator = self.song.iterNotes()
+            note, verseIndex = self.noteIterator.next()
+            eventLogger.info(pygame.event.Event(events.SONGEND))
+        try :
+            syllabus = note.lyrics[verseIndex].syllabus()
+        except IndexError :
+            syllabus = u'…'
+        
         column = self.columns[note.midi]
         column.update(True, syllabus)
         self.currentColumn = column
         column = self.columns[note.midi]
         column.update(True, syllabus)
         self.currentColumn = column