/** Test for phase vocoder in jack
 *
 * This program should start correctly, when jackd is started or when
 * using JACK_START_SERVER=true and reconstruct each audio input channel
 * on the corresponding output channel with some strange effects and a
 * delay equal to the hop size (hop_s).
 *
 */

#include <stdio.h>
#include <unistd.h>  /* sleep() */
#include <aubio.h>
#ifdef HAVE_JACK
#include "jackio.h"
#endif /* HAVE_JACK */

uint_t testing  = 0;  /* change this to 1 to listen        */

uint_t win_s    = 512;/* window size                       */
uint_t hop_s    = 128;/* hop size                          */
uint_t channels = 2;  /* number of audio channels          */
uint_t midiin   = 4;  /* number of midi input channels     */
uint_t midiout  = 2;  /* number of midi output channels    */
uint_t pos      = 0;  /* frames%dspblocksize for jack loop */

fvec_t * in[2];
cvec_t * fftgrain[2];
fvec_t * out[2];

aubio_pvoc_t * pv[2];

int aubio_process(float **input, float **output, int nframes);

int main(){
        /* allocate some memory */
  uint_t i;
    for (i=0;i<channels;i++) {
        in[i]       = new_fvec (hop_s); /* input buffer       */
        fftgrain[i] = new_cvec (win_s); /* fft norm and phase */
        out[i]      = new_fvec (hop_s); /* output buffer      */
        /* allocate fft and other memory space */
        pv[i] = new_aubio_pvoc(win_s,hop_s);
    }

#ifdef HAVE_JACK
        aubio_jack_t * jack_setup;
        jack_setup  = new_aubio_jack(channels, channels, 
            midiin, midiout,
            (aubio_process_func_t)aubio_process);
        aubio_jack_activate(jack_setup);
        /* stay in main jack loop for 1 seconds only */
        do {
          sleep(1);
        } while(testing);
        aubio_jack_close(jack_setup);
#else
        fprintf(stderr, "WARNING: no jack support\n");
#endif
        
    for (i=0;i<channels;i++) {
        del_aubio_pvoc(pv[i]);
        del_cvec(fftgrain[i]);
        del_fvec(in[i]);
        del_fvec(out[i]);
    }
        aubio_cleanup();
        return 0;
}

int aubio_process(float **input, float **output, int nframes) {
  uint_t i;       /*channels*/
  uint_t j;       /*frames*/
  for (j=0;j<(unsigned)nframes;j++) {
    for (i=0;i<channels;i++) {
      /* write input to datanew */
      fvec_write_sample(in[i], input[i][j], pos);
      /* put synthnew in output */
      output[i][j] = fvec_read_sample(out[i], pos);
    }
    /*time for fft*/
    if (pos == hop_s-1) {
      /* block loop */
    for (i=0;i<channels;i++) {
      aubio_pvoc_do (pv[i], in[i], fftgrain[i]);
      // zero phases of first channel
      for (i=0;i<fftgrain[i]->length;i++) fftgrain[0]->phas[i] = 0.; 
      // double phases of second channel
      for (i=0;i<fftgrain[i]->length;i++) {
        fftgrain[1]->phas[i] = 
          aubio_unwrap2pi (fftgrain[1]->phas[i] * 2.); 
      }
      // copy second channel to third one
      aubio_pvoc_rdo(pv[i], fftgrain[i], out[i]);
      pos = -1;
    }
    }
    pos++;
  }
  return 0;
}