audio_engine.py

from audiostream import get_output
from audio_source_one_shot import AudioSourceOneShot
from audio_source_track import AudioSourceTrack
from audio_source_mixer import AudioSourceMixer

class AudioEngine:
    CHANNELS = 1
    SAMPLES_RATE = 44100
    BUFFER_SIZE = 1024

    def __init__(self):
        self.output_stream = get_output(channels=self.CHANNELS,
                                    rate=self.SAMPLES_RATE, 
                                    buffersize=self.BUFFER_SIZE)
        

        self.audio_source_one_shot = AudioSourceOneShot(self.output_stream)
        self.audio_source_one_shot.start()

        
    
    def play_sound(self, wav_samples):
       self.audio_source_one_shot.set_wav_samples(wav_samples)

    
    def create_track(self, wav_samples, bpm, min_bpm):
        source_track = AudioSourceTrack(self.output_stream, wav_samples, bpm, self.SAMPLES_RATE, min_bpm)
        # source_track.set_steps((1, 0, 0, 0))
        source_track.start()
        return source_track
    
    def create_mixer(self, all_wav_samples, bpm, nb_steps, on_current_step_changed, min_bpm):
        source_mixer = AudioSourceMixer(self.output_stream, all_wav_samples, bpm, self.SAMPLES_RATE, nb_steps, on_current_step_changed, min_bpm)
        source_mixer.start()
        return source_mixer