29 lines
1.8 KiB
Python
29 lines
1.8 KiB
Python
|
import pymumble_py3
|
||
|
|
import numpy as np
|
||
|
|
import threading
|
||
|
|
|
||
|
|
|
||
|
|
class CustomMumble(pymumble_py3.Mumble):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.audio_handler = None
|
||
|
|
self.is_running = False
|
||
|
|
|
||
|
|
def set_audio_handler(self, audio_handler):
|
||
|
|
self.audio_handler = audio_handler
|
||
|
|
|
||
|
|
def sound_received_handler(self, user, soundchunk):
|
||
|
|
if self.audio_handler is not None:
|
||
|
|
sound_array = np.frombuffer(soundchunk.pcm, dtype=np.int16)
|
||
|
|
num_channels = self.audio_handler.channels
|
||
|
|
sound_array = sound_array.reshape((-1, num_channels))
|
||
|
|
sound_float = sound_array.astype(np.float32) / 32767.0
|
||
|
|
self.audio_handler.write(sound_float)
|
||
|
|
|
||
|
|
def start(self):
|
||
|
|
self.is_running = True
|
||
|
|
threading.Thread(target=self.run).start()
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.is_running = False
|