// file: $SPEECH_HOMEWORK4/hw4/class/snr_ratio/v1.0/snr_proc_1.cc
//

// isip include files
//
#include "signal_noise_ratio.h"
#include "signal_noise_ratio_constants.h"

// method: process_plots_cc
//
// arguments:
//  float_4** energy (output) the energy signal for each sample frame
//                            at each channel
//  int_4 num_energy (output) the total number of energy signals
//
// return: a logical_1 indicating status
//
logical_1 Signal::process_plots_cc (float_4** energy_a, int_4 num_energy_a) {

  // check arguments
  //
  if (energy_a == (float_4**)NULL) {
    error_handler_cc((char_1*)"process_plots_cc",
		     (char_1*)"energy information is empty");
  }

  if (num_energy_a <= 0) {
    error_handler_cc((char_1*)"process_plots_cc",
		     (char_1*)"invalid number of energy values");
  }

  // set up min and max value for pdf
  //
  float_4* min = new float_4[num_chans_d];
  float_4* max = new float_4[num_chans_d];

  // find the min and max of the energy for pdf
  //
  find_range_cc(energy_a, num_energy_a, min, max);
  
  // set up variable to perform pdf
  //
  int_4** pdf = new int_4*[num_bins_d];
  float_4** mean = new float_4*[num_bins_d];

  for (int_4 i=0;i<num_bins_d;i++) {
    pdf[i] = new int_4[num_chans_d];
    mean[i] = new float_4[num_chans_d];

    for (int_4 j=0;j<num_chans_d;j++) {
      pdf[i][j] = 0;
      mean[i][j] = 0;
    }
  }

  calculate_pdf_cc(energy_a, num_energy_a, pdf, mean, min, max);

  int_4** cdf = new int_4*[num_bins_d];

  for (int_4 i=0;i<num_bins_d;i++) {
    cdf[i] = new int_4[num_chans_d];
  }

  calculate_cdf_cc(pdf, cdf);

  if (debug_level_d > 1) {
    for (int_4 i=0;i<num_bins_d; i++ ){
      for (int_4 j=0;j<num_chans_d;j++) {
	fprintf(stdout, "chan %ld :: cdf = %ld\n", j, cdf[i][j]);
      }
    }
  }

  float_4* snr = new float_4[num_chans_d];
  calculate_snr_cc(cdf, energy_a, num_energy_a, max, min, mean, snr);

  // output the results to output file
  //
  if (output_file_d != (char_1*)NULL) {
    FILE* fp = fopen ((char*)output_file_d, "w");
    for (int_4 i=0;i<num_chans_d;i++) {
      fprintf(fp, "The SNR for channel %ld is %.6f dB\n", i, snr[i]);
    }
    fclose(fp);
  }

  // output the results to screen
  //
  for (int_4 i=0;i<num_chans_d;i++) {
    fprintf(stdout, "The SNR for channel %ld is %.6f dB\n", i, snr[i]);
  }
  
  // clean up
  //
  delete [] min;
  delete [] max;

  for (int_4 i=0;i<num_bins_d;i++) {
    delete [] pdf[i];
  }
  delete [] pdf;

  for (int_4 i=0;i<num_bins_d;i++) {
    delete [] cdf[i];
  }
  delete [] cdf;

 
  // exit gracefully
  //
  return ISIP_TRUE;
}