Fix frame rate when importing music

[perso/Immae/Projets/Python/MusicSampler.git] / music_sampler.py

diff --git a/music_sampler.py b/music_sampler.py
index 9797c918073f4baa8033f85634b9c5163efc34e9..c17f7bff98d1764704ee542ef9a1004aa1151f8b 100644 (file)
--- a/music_sampler.py
+++ b/music_sampler.py
@@ -1,34 +1,26 @@
 import sys
-
-if getattr(sys, 'frozen', False):
-    os.chdir(sys._MEIPASS)
-
 import pygame
-import pydub
 import helpers
+import threading
 
 pygame.mixer.pre_init(frequency = 44100)
 pygame.init()
 
-size = width, height = 1024, 600
-
+size = width, height = 913, 563
 screen = pygame.display.set_mode(size)
-background = pygame.Surface(screen.get_size())
-background = background.convert()
-background.fill((250, 250, 250))
+screen.fill((229, 228, 226))
 
-action_surface = pygame.Surface((600, 250)).convert()
-action_surface.fill((0,0,0))
-helpers.parse_config()
+draw_lock = helpers.Lock("draw")
 
-for key_name in helpers.Mapping.KEYS:
-    key = helpers.Mapping.KEYS[key_name]
-    key.draw(background)
+mapping = helpers.Mapping(screen, draw_lock)
+channel_number, open_files = helpers.parse_config(mapping)
+pygame.mixer.set_num_channels(channel_number)
 
-screen.blit(background, (0, 0))
-screen.blit(action_surface, (10, 330))
+mapping.draw()
 
+draw_lock.acquire()
 pygame.display.flip()
+draw_lock.release()
 
 contexts = [
     'normal'
@@ -36,24 +28,59 @@ contexts = [
 
 context = 'normal'
 
+#### Normal workflow ####
 while 1:
     event = pygame.event.wait()
+
     if event.type == pygame.QUIT or (
             event.type == pygame.KEYDOWN and 
             event.mod  == 4160 and
             event.key  == pygame.K_c):
+        for thread in threading.enumerate():
+            if thread.getName()[0:2] != "MS":
+                continue
+            thread.join()
+
         pygame.quit()
         sys.exit()
 
     if context == 'normal':
         if event.type == pygame.KEYDOWN:
-            key = helpers.Key.find_by_key_num(event.key)
-            if key is not None:
-                key.do_actions()
+            key = mapping.find_by_key_num(event.key)
+            if key is not None and not key.disabled:
+                threading.Thread(name = "MSKeyAction", target=key.do_actions).start()
+                threading.Thread(name = "MSClic", target=key.list_actions, args = [screen]).start()
         elif event.type == pygame.MOUSEBUTTONUP:
-            key = helpers.Key.find_by_collidepoint(pygame.mouse.get_pos())
+            key = mapping.find_by_collidepoint(pygame.mouse.get_pos())
             if key is not None:
-                key.list_actions(action_surface)
+                threading.Thread(name = "MSClic", target=key.list_actions, args = [screen]).start()
+
+    draw_lock.acquire()
+    police = helpers.font(14)
+    icon_police = helpers.font(14, font = "Symbola")
+
+    surface = pygame.Surface((208, 250)).convert()
+    surface.fill((250, 250, 250))
+    offset = 0
+    for music_file in open_files.values():
+        police.set_bold(False)
+        if music_file.is_playing():
+            if music_file.is_paused():
+                icon = icon_police.render("⏸", True, (0,0,0))
+            else:
+                icon = icon_police.render("⏵", True, (0,0,0))
+                police.set_bold(True)
+            text = police.render(music_file.name, True, (0,0,0))
+            surface.blit(icon, (0, offset))
+            surface.blit(text, (20, offset))
+            offset += police.get_linesize()
+    screen.blit(surface, (700, 308))
 
     pygame.display.flip()
+    draw_lock.release()
 
+#### In Ipython ####
+# for thread in threading.enumerate():
+#     if thread.getName()[0:2] != "MS":
+#         continue
+#     thread.join()