!7

import static x30_pkg.x30_util.VF1;

module LiveFFT > DynImageSurface {
  volatile bool enabled = true;
  int windowSize = 1; // seconds
  transient Q q;
  transient L<short[]> buffer = synchroList();
  transient volatile int bufferSize;
  transient int neededSamples;
  transient Clip clip;
  
  enhanceFrame {
    dm_intFieldMenuItem(f, 'windowSize, formTitle := "Window Size (Seconds)", onSet := r onNewWindowSize);
  }

  visualize {
    ret centerAndSouthWithMargins(super.visualize(),
      jrightalignedline(dm_fieldCheckBox('enabled)));
  }
  
  void onNewWindowSize {
    neededSamples = windowSize*44100;
    buffer.clear();
    bufferSize = 0;
  }

  start {
    onNewWindowSize();
    
    q = dm_startQ();
    dm_addAudioListener(voidfunc(short[] data) {
      if (!enabled) ret;
      buffer.add(mixStereoShortArrayToMono(data));
      bufferSize += l(data)/2;
      
      if (bufferSize >= neededSamples) {
        bufferSize -= neededSamples;
        final L<short[]> salvaged = cloneList(buffer);
        buffer.clear();
        if (bufferSize > 0) {
          buffer.add(lastNShorts(bufferSize, last(salvaged)));
          replaceLastElement(salvaged, dropLastNShorts(bufferSize, last(salvaged)));
        }
        q.add(r {
          MultiShortArrayInputStream_resettable stream = new(salvaged);
          stream.bigEndian = true;
          clip = spectro_clipFromMonoInputStream(stream);
          vmBus_send newAudioFrequencyClip(clip);
          BufferedImage img = clipToFrequencyImage(clip);
          vmBus_send newAudioFrequencyImage(img);
          setImage(img);
        });
      }
    });
  }  
}