#include <fstream>
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "architecture.h"
#include "configuration.h"

using namespace std;

//variables
const int MAX_CHANNELS=4096;
static const u32 SYNC = MAX_U32;
double dumptime, freq, inttimems;
float miny, maxy;
int dsindex, nchan, bandindex, threadindex, coreindex, nthreads, configindex;
string pol, antennaname;
float data[MAX_CHANNELS];
ifstream * input;
Configuration * config;
Model * model;

//prototypes
bool getData(ifstream * input);
void getMinMax(float * data, int length, float * min, float * max);

int main(int argc, char *argv[])
{
  //check for correct invocation
  if(argc != 3) {
    cout << "Error - invoke with printfilterbank <dump file> <correlator config file>" << endl;
    return EXIT_FAILURE;
  }

  //open the file containing the filterbank data
  input = new ifstream(argv[1], ios::binary);
  config = new Configuration(argv[2], -1);
  model = config->getModel();

  //loop through inspecting the data
  while (!input->eof()) {
    if(getData(input)) {
      getMinMax(data, nchan, &miny, &maxy);
    }
    else {
      cout << "Skipping data since its all empty..." << endl;
    }
  }
}

void getMinMax(float * data, int length, float * min, float * max)
{
  *min = data[0];
  *max = data[0];
  for (int i=1;i<length;i++) {
    if(data[i] > *max)
      *max = data[i];
    if(data[i] < *min)
      *min = data[i];
  }
  printf("Data min was %f, max was %f\n", *min, *max);
}

bool getData(ifstream * input)
{
  u32 sync;
  int configindex, scan, sec, ns, nswidth, extrachan = 0;

  input->read((char*)&sync, sizeof(int));
  while(sync != SYNC && !input->eof()) {
    cout << "Searching for lost SYNC!" << endl;
    input->read((char*)&sync, sizeof(int));
  }
  if (input->eof())
    return false;
  input->read((char*)&dsindex, sizeof(int));
  input->read((char*)&scan, sizeof(int));
  input->read((char*)&sec, sizeof(int));
  input->read((char*)&ns, sizeof(int));
  input->read((char*)&nswidth, sizeof(int));
  input->read((char*)&bandindex, sizeof(int));
  input->read((char*)&nchan, sizeof(int));
  input->read((char*)&coreindex, sizeof(int));
  input->read((char*)&threadindex, sizeof(int));

  //figure out the MJD (including fractional component) for this dump and the freq/pol/antname
  dumptime = model->getScanStartMJD(scan) + ((double)sec)/86400.0 + ((double)ns)/86400000000000.0;
  inttimems = ((double)nswidth)/1000000.0;
  configindex = config->getScanConfigIndex(scan);
  freq = config->getFreqTableFreq(config->getDRecordedFreqIndex(configindex, dsindex, bandindex));
  pol = config->getDRecordedBandPol(configindex, dsindex, bandindex);
  antennaname = config->getDStationName(configindex, dsindex);

  printf("Read %d channels of data from %s, scan %d, sec %04d, ns %09d, nswidth %09d, coreindex %d, threadindex %d\n", nchan, antennaname.c_str(), scan, sec, ns, nswidth, coreindex, threadindex);
  //cout << "Read data from " << antennaname << ", scan " << scan << ", sec " << sec << ", ns " << ns << ", nswidth is " << nswidth << ", coreindex is " << coreindex << ", threadindex is " << threadindex << endl;

  if(nchan > MAX_CHANNELS) {
    cout << "nchan=" << nchan << " is greater than MAX_CHANNELS=" << MAX_CHANNELS;
    cout << "Will only read the first MAX_CHANNELS channels" << endl;
    extrachan = nchan-MAX_CHANNELS;
    nchan = MAX_CHANNELS;
  }
  input->read((char*)data, sizeof(int)*nchan);
  if(extrachan > 0)
    input->ignore(extrachan);
  return true;
}