]>
Commit | Line | Data |
---|---|---|
be27763f IB |
1 | import threading |
2 | import pydub | |
3 | import pygame | |
87f211fb | 4 | import math |
98ff4305 | 5 | import time |
be27763f IB |
6 | |
7 | class MusicFile: | |
87f211fb | 8 | def __init__(self, filename, lock, channel_id, name = None, gain = 1): |
be27763f | 9 | self.filename = filename |
d479af33 | 10 | self.channel_id = channel_id |
9de92b6d | 11 | self.name = name or filename |
be27763f | 12 | self.raw_data = None |
87f211fb | 13 | self.gain = gain |
be27763f IB |
14 | |
15 | self.loaded = False | |
9de92b6d | 16 | self.flag_paused = False |
23b7e0e5 | 17 | threading.Thread(name = "MSMusicLoad", target = self.load_sound, args = [lock]).start() |
be27763f IB |
18 | |
19 | def load_sound(self, lock): | |
20 | lock.acquire() | |
9de92b6d | 21 | print("Loading « {} »".format(self.name)) |
87f211fb IB |
22 | volume_factor = 20 * math.log10(self.gain) |
23 | audio_segment = pydub.AudioSegment.from_file(self.filename).set_frame_rate(44100).apply_gain(volume_factor) | |
b58b8220 IB |
24 | self.sound_duration = audio_segment.duration_seconds |
25 | self.raw_data = audio_segment.raw_data | |
9de92b6d | 26 | print("Loaded « {} »".format(self.name)) |
be27763f IB |
27 | self.loaded = True |
28 | lock.release() | |
29 | ||
0e5d59f7 | 30 | def is_playing(self): |
d479af33 | 31 | return self.channel().get_busy() |
0e5d59f7 | 32 | |
9de92b6d IB |
33 | def is_paused(self): |
34 | return self.flag_paused | |
35 | ||
98ff4305 IB |
36 | @property |
37 | def sound_position(self): | |
38 | if self.is_playing() and not self.is_paused(): | |
39 | return min(time.time() - self.started_at, self.sound_duration) | |
40 | elif self.is_playing() and self.is_paused(): | |
41 | return min(self.paused_at - self.started_at, self.sound_duration) | |
42 | else: | |
43 | return 0 | |
44 | ||
0e5d59f7 | 45 | def play(self, fade_in = 0, volume = 100, start_at = 0): |
9de92b6d IB |
46 | self.channel().set_endevent() |
47 | self.channel().set_endevent(pygame.USEREVENT) | |
0e5d59f7 IB |
48 | self.set_volume(volume) |
49 | ||
50 | if start_at > 0: | |
0e5d59f7 | 51 | raw_data_length = len(self.raw_data) |
b58b8220 | 52 | start_offset = int((raw_data_length / self.sound_duration) * start_at) |
0e5d59f7 | 53 | start_offset = start_offset - (start_offset % 2) |
b58b8220 | 54 | sound = pygame.mixer.Sound(self.raw_data[start_offset:]) |
0e5d59f7 | 55 | else: |
b58b8220 | 56 | sound = pygame.mixer.Sound(self.raw_data) |
0e5d59f7 | 57 | |
98ff4305 | 58 | self.started_at = time.time() |
92cc4ce2 | 59 | self.channel().play(sound, fade_ms = int(fade_in * 1000)) |
9de92b6d | 60 | self.flag_paused = False |
be27763f IB |
61 | |
62 | def pause(self): | |
98ff4305 | 63 | self.paused_at = time.time() |
d479af33 | 64 | self.channel().pause() |
9de92b6d IB |
65 | self.flag_paused = True |
66 | ||
67 | def unpause(self): | |
98ff4305 | 68 | self.started_at += (time.time() - self.paused_at) |
9de92b6d IB |
69 | self.channel().unpause() |
70 | self.flag_paused = False | |
be27763f | 71 | |
0e5d59f7 | 72 | def stop(self, fade_out = 0): |
0e5d59f7 | 73 | if fade_out > 0: |
92cc4ce2 | 74 | self.channel().fadeout(int(fade_out * 1000)) |
0e5d59f7 | 75 | else: |
d479af33 | 76 | self.channel().stop() |
0e5d59f7 IB |
77 | |
78 | def set_volume(self, value): | |
79 | if value < 0: | |
80 | value = 0 | |
81 | if value > 100: | |
82 | value = 100 | |
b58b8220 | 83 | self.channel().set_volume(value / 100) |
be27763f | 84 | |
b86db9f1 IB |
85 | def wait_end(self): |
86 | pass | |
d479af33 IB |
87 | |
88 | def channel(self): | |
89 | return pygame.mixer.Channel(self.channel_id) |