Fix threading problems with ipython

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

import sys
import pygame
import helpers
import threading

pygame.mixer.pre_init(frequency = 44100)
pygame.init()

size = width, height = 1024, 600
screen = pygame.display.set_mode(size)
screen.fill((250, 250, 250))

draw_lock = helpers.Lock("draw")

mapping = helpers.Mapping(screen, draw_lock)
helpers.parse_config(mapping)

mapping.draw()

draw_lock.acquire()
pygame.display.flip()
draw_lock.release()

contexts = [
    'normal'
]

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 = mapping.find_by_key_num(event.key)
            if key is not None:
                threading.Thread(name = "MSKeyAction", target=key.do_actions).start()
        elif event.type == pygame.MOUSEBUTTONUP:
            key = mapping.find_by_collidepoint(pygame.mouse.get_pos())
            if key is not None:
                threading.Thread(name = "MSClic", target=key.list_actions, args = [screen]).start()

    draw_lock.acquire()
    pygame.display.flip()
    draw_lock.release()

#### In Ipython ####
# for thread in threading.enumerate():
#     if thread.getName()[0:2] != "MS":
#         continue
#     thread.join()