//roarfctest.c:

/*
 *      Copyright (C) Philipp 'ph3-der-loewe' Schafft - 2008-2013
 *
 *  This file is part of roarclients a part of RoarAudio,
 *  a cross-platform sound system for both, home and professional use.
 *  See README for details.
 *
 *  This file is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 3
 *  as published by the Free Software Foundation.
 *
 *  RoarAudio is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this software; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 51 Franklin Street, Fifth Floor,
 *  Boston, MA 02110-1301, USA.
 *
 */

#include <roaraudio.h>  /* libroar */

#ifdef ROAR_HAVE_LIBM

#include <math.h>       /* sin() */
#include <stdio.h>      /* *printf*() */
#include <libroardsp/libroardsp.h>

void usage (void) {
 printf("roarfctest [OPTIONS]...\n");

 printf("\nOptions:\n\n");

 printf(
        "  --rate      RATE      - Set sample rate\n"
        "  --sfreq     SFREQ     - Start frequency \n"
        "  --efreq     EFREQ     - End frequency\n"
        "  --engage    ENGAGE    - This option is not documented\n"
        "  --tmax      TMAX      - This option is not documented\n"
        "  --gnuplot             - Output data using gnuplot(1)\n"
        "  --rmsout    RMSOUT    - Set output file for RMS data\n"
        "  --help   -h           - Show this help\n"
       );
}

int main (int argc, char * argv[]) {
 int rate     = ROAR_RATE_DEFAULT;
 int bits     = 16;
 int channels = 1; /* mono */
 int codec    = ROAR_CODEC_DEFAULT;
 float freq   = 523.2;            /* middle C */
 float step; //   = M_PI*2*freq/rate; /* how much time per sample we have to encode ... */
 float t      = 0; /* current time */
 float length;
 int16_t amp  = 32767;
 int      i;
 char   * k;
 int32_t  tmp;
 int16_t  out[1024];
 int16_t  maxval, maxval_in;
 uint32_t sc, sc_tot = 0;
 int      count = 0;
 float    rms, rms_in = amp/sqrt(2);
 float    worklen;
 float    sfreq  = 1, efreq = -1;
 float    engage = 0;
 float    tmax   = 1024;
 struct roardsp_filterchain fc[1];
 struct roardsp_filter      filt[3];
 struct roar_stream         stream[1];
 FILE * rmsout = NULL;
 int kill_var;
#ifdef ROAR_HAVE_BIN_GNUPLOT
 int do_gnuplot = 0;
#endif

 for (i = 1; i < argc; i++) {
  k = argv[i];
  kill_var = 1;

  if ( !strcmp(k, "--rate") ) {
   rate = atoi(argv[++i]);
  } else if ( !strcmp(k, "--sfreq") ) {
   sfreq = atof(argv[++i]);
  } else if ( !strcmp(k, "--efreq") ) {
   efreq = atof(argv[++i]);
  } else if ( !strcmp(k, "--engage") ) {
   engage = atof(argv[++i]);
  } else if ( !strcmp(k, "--tmax") ) {
   tmax = atof(argv[++i]);
#ifdef ROAR_HAVE_BIN_GNUPLOT
  } else if ( !strcmp(k, "--gnuplot") ) {
   do_gnuplot = 1;
#endif
  } else if ( !strcmp(k, "--rmsout") ) {
   rmsout = fopen(argv[++i], "w");
  } else if ( !strcmp(k, "--help") || !strcmp(k, "-h") ) {
   usage();
   return 0;
  } else {
   kill_var = 0;
  }

  if ( kill_var )
   *k = 0;
 }

#ifdef ROAR_HAVE_BIN_GNUPLOT
 if ( do_gnuplot && rmsout == NULL ) {
  if ( (rmsout = popen(ROAR_HAVE_BIN_GNUPLOT, "w")) != NULL ) {
   fprintf(rmsout, "set grid\n");
   fprintf(rmsout, "set log x 10\n");
   fprintf(rmsout, "set log y 10\n");
   fprintf(rmsout, "plot \"-\" with lines title \"filter amplification\"\n");
  }
 }
#endif

 if ( roar_stream_new(stream, rate, channels, bits, codec) == -1 ) {
  pclose(rmsout);
  return 2;
 }

 if ( roardsp_filter_init(filt, stream, ROARDSP_FILTER_DCBLOCK) == -1 ) {
  ROAR_ERR("main(*): roardsp_filter_init() failed: errno=%s(%i)", strerror(errno), errno);
  pclose(rmsout);
  return 1;
 }

/*
 freq = 1000;
 roardsp_filter_ctl(filt, ROARDSP_FCTL_FREQ, &freq);
*/

 if ( roardsp_filter_init(filt+1, stream, ROARDSP_FILTER_HIGHP) == -1 ) {
  ROAR_ERR("main(*): roardsp_filter_init() failed: errno=%s(%i)", strerror(errno), errno);
  pclose(rmsout);
  return 1;
 }

 freq = 1000;
 roardsp_filter_ctl(filt+1, ROARDSP_FCTL_FREQ, &freq);

 if ( roardsp_filter_init(filt+2, stream, ROARDSP_FILTER_AMP) == -1 ) {
  ROAR_ERR("main(*): roardsp_filter_init() failed: errno=%s(%i)", strerror(errno), errno);
  pclose(rmsout);
  return 1;
 }

 tmp = 2;
 roardsp_filter_ctl(filt+2, ROARDSP_FCTL_DIV, &tmp);
 tmp = 3;
 roardsp_filter_ctl(filt+2, ROARDSP_FCTL_MUL, &tmp);


 roardsp_fchain_init(fc);
 roardsp_fchain_add(fc, filt);
// roardsp_fchain_add(fc, filt+1);
// roardsp_fchain_add(fc, filt+2);

 fprintf(stderr, "Starting analysis in frequency domain...\n");

 if ( efreq == -1 )
  efreq = (float)rate/2;

 length = 5/sfreq;

 for (freq = sfreq; freq < efreq; freq *= (1+exp(1)/100), length /= (1+exp(1)/100)) {
  step      = M_PI*2*freq/rate;
  maxval    = -amp;
  maxval_in = -amp;
  sc        = 0;
  rms       = 0;
  t         = -engage;

  if ( roardsp_fchain_reset(fc, ROARDSP_RESET_STATE) == -1 ) {
   ROAR_ERR("Can not reset filterchain.");
   return 8;
  }


  worklen = length;

  if ( worklen > tmax )
   worklen = tmax;

  worklen = 2*M_PI*freq*worklen;


  while (t < worklen) {
   for (i = 0; i < 1024; i++) {
    out[i] = amp*sin(t);
    if ( out[i] > maxval_in )
     maxval_in = out[i];
    t += step;
   }

   roardsp_fchain_calc(fc, out, 1024);

   if ( t >= 0 ) {
    for (i = 0; i < 1024; i++) {
     rms += out[i] * out[i];
     if ( out[i] > maxval )
      maxval = out[i];
    }
    sc += 1024;
   } else {
    sc_tot += 1024;
   }
  }

  count++;
  sc_tot += sc;

  rms /= sc;
  rms  = sqrt(rms);
//  rms /= amp;
  rms /= rms_in;

  printf("%f: %f %u %f\n", freq, (float)maxval/maxval_in, sc, rms);

  if ( rmsout != NULL )
   fprintf(rmsout, "%f: %f\n", freq, rms);
 }


 fprintf(stderr, "Finished analysis in frequency domain: done tests on a total of %u samples on %i frequencies\n",
           sc_tot, count);

 roardsp_fchain_uninit(fc);

 if ( rmsout != NULL )
  pclose(rmsout);

 return 0;
}

#else
int main (void) {
 fprintf(stderr, "Error: No Math library support compiled in.\n");
 return 1;
}
#endif

//ll