!7

import static x30_pkg.x30_util.VF1;
import javax.sound.sampled.AudioFormat;

cmodule APlayAudioOutput > DynPrintLog {
  switchable int bufSizeMS = 200;
  switchable int sampleRate = 44100;
  switchable int channels = 2; // only 2 is tested
  switchable int bits = 16;
  switchable float gain = 1;
  switchable bool registerService;
  
  transient L<VF1<double[]>> sources = notifyingList(r setNumSources);
  transient SimpleLiveValue<Int> numSources = new(Int, 0);
  transient StreamSoundToAPlay loop;
  transient int bufSizeInFrames;
  transient short[] buffer;

  visualize {
    ret centerAndSouth(super.visualize(),
      withLeftAndRightMargin(vstackWithSpacing(
        withLabel("Volume (0-200%):", liveSliderZeroToOne(gain/2, voidfunc(float f) { setField(gain := f*2) })),
        jrightalignedline(withLabel("Sound sources:", jLiveValueLabel(numSources)))
      )));
  }
  
  start {
    ownResource(loop = new StreamSoundToAPlay);
    copyFields(module(), loop, 'sampleRate, 'channels, 'bits, 'bufSizeMS);
    bufSizeInFrames = msToSamples_int(bufSizeMS, sampleRate);
    buffer = new short[bufSizeInFrames*2];
    fillBuffer();
    loop.start();
    loop.writeSamples(buffer);
    doEveryAndNow(bufSizeMS, r {
      fillBuffer();
      loop.writeSamples(buffer);
    });

    dm_registerAs audioOutputModule();
    
    if (registerService)
      dm_registerService('playWAVAndWait, func(S method, O[] args) -> bool {
        temp enter();
        callMC(f dm_playWAV, args);
        true;
      });
  }
  
  void fillBuffer enter {
    VF1<VF1<double[]>> vfRemoveSource = voidfunc(VF1<double[]> source) {
      removeSource(source)
    };
    
    double[] pair = new double[2];
    for i over buffer: {
      audio_makeSampleFromSources(sources, pair, gain, vfRemoveSource);
      buffer[i++] = clipShort(pair[0]);
      buffer[i] = clipShort(pair[1]);
    }
  }
  
  void setNumSources { numSources.set(l(sources)); }
  
  // hold audio sources during reload
  
  O _getReloadData() { ret sources; }
  void _setReloadData(L<VF1<double[]>> sources) { addAll(this.sources, sources); }

  // API
  
  void addSource(VF1<double[]> source) {
    sources.add(source);
    print("Audio source added.");
  }
  
  void removeSource(VF1<double[]> source) {
    sources.remove(source);
    print("Audio source removed.");
  }
  
  AudioFormat getAudioFormat() { ret new AudioFormat(AudioFormat.Encoding.PCM_SIGNED, sampleRate, bits, channels, channels*bits/2, sampleRate, false); }
  
  long samplesPlayed() { ret loop.frameCounter; }
}