/**
* @file SEightChan_swTrig.cpp
* @author Dan Saunders, on behalf of the SoLid collaboration.
* @date 19 Feb 2016
*/

#include "SEightChan_swTrig.h"

//==============================================================================

//! Constructor setting up default values.
SEightChan_swTrig::SEightChan_swTrig(SDetector * dtr, SClipboard * cb) :
  ISAlgorithm(dtr, cb, "SEightChan_swTrig"),
  m_nBytesIn(0)
{
  // This is ROOTs fault - have tried a few different ways, but ROOT does not seem
  // to like putting these into a vector to begin with.
  br_waveforms.push_back(&br_waveform0);
  br_waveforms.push_back(&br_waveform1);
  br_waveforms.push_back(&br_waveform2);
  br_waveforms.push_back(&br_waveform3);
  br_waveforms.push_back(&br_waveform4);
  br_waveforms.push_back(&br_waveform5);
  br_waveforms.push_back(&br_waveform6);
  br_waveforms.push_back(&br_waveform7);
}


//==============================================================================

//! Setup branches to intput tree and load files on the clipboard.
void SEightChan_swTrig::initialize() 
{
  m_chain = new TChain("waveforms");

  for (uint iC = 0; iC<br_waveforms.size(); iC++) {
    (*br_waveforms[iC]) =  new std::vector<float>;
    (*br_waveforms[iC]) = 0;
    std::string name = "wf_chan" + std::to_string(iC);
    m_chain->SetBranchAddress(name.c_str(), br_waveforms[iC]);
  }

  if (cb()->inputDataFileNames()->size()==0) {
    std::cout<<"[Error]: Exiting since no input files specified"<<std::endl;
    exit(0);
  }
  for (auto f : (*cb()->inputDataFileNames())) {
    std::cout<<"[Note]: Adding input data file: "<<f<<std::endl;
    int status = m_chain->Add(f.c_str());
    if (status < 1) {
      std::cout<<"[Error]: Exiting due to problem adding file: "<<f<<std::endl;
      exit(1);
    } 
  }
  std::cout<<"[Note]: Attempting to get zeroth entry..."<<std::endl;
  m_chain->GetEntry(0);
  std::cout<<"[Note]: ...done"<<std::endl;

  // Since this is the first file, update conditions.
  updateConditions();
}


//==============================================================================

//! Loads the waveforms for this cycle.
/*! Very easy data format to work with - each entry corresponds to a saffron cycle.
 */
void SEightChan_swTrig::execute() 
{
  // Have to keep track of when a file changes - use the filename. If changed, 
  // update conditions.
  m_nBytesIn += m_chain->GetEntry(cb()->cycle());
  std::string currentFileName(m_chain->GetFile()->GetName());
  if (currentFileName != cb()->currentFileName()) updateConditions();

  // Loop over each channel.
  for (uint iC=0; iC<br_waveforms.size(); iC++) {
    if (!dtr()->channel(iC)->masked()) {
    	uint64_t time = 0;
      SWaveform * w = new SWaveform(time, 0, dtr()->channel(iC));
      for (uint i=0; i<(*br_waveforms[iC])->size(); i++) 
        w->samples()->emplace_back(uint16_t((*br_waveforms[iC])->at(i)));

      cb()->waveforms()->push_back(w);
      dtr()->channel(iC)->addWaveform(w);
    }
  }

  // See if the last cycle.
  if (cb()->cycle() == m_chain->GetEntries()-1)  setRunStopped(true);

  // Status print out.
  if (cb()->cycle() % 500 == 0) {
    std::cout<<"[Status Note]: Cycle: "<<cb()->cycle()<<"\t"<<"Fraction data processed: "<<
      cb()->cycle()/(1.*m_chain->GetEntries())<<std::endl;
  }
}


//==============================================================================

//! Update conditions on clipboard for the current input file.
void SEightChan_swTrig::updateConditions() 
{
	std::cout<<"[Note]: Updating conditions..."<<std::endl;
	auto cdnsMap = cb()->cdns();

  // Get the current tfile.
  TFile * currentFile = m_chain->GetFile();

  // Get the conditions tree and setup branchs.
  TTree * t = (TTree*)currentFile->Get("conditions");
  t->Print();
  std::vector<std::string> branchNames = {"HVbias", "initialtemp", "finaltemp", "LVtrims"};
  std::vector<std::string> branchTypes = {"double", "vector_doubles", "vector_doubles", "vector_doubles"};

  uint nConditions = branchNames.size();
  // Loop over branchs specified.
  for (uint i=0; i<nConditions; i++) {
  	SCondition * cdn;

  	/* Check if the condition already exists. If so, clear it for updating.
  	 * Else, create a new one.
  	 */
  	if ((*cdnsMap).count(branchNames[i]) == 0) {
			cdn = new SCondition(branchNames[i]);
			(*cdnsMap)[branchNames[i]] = cdn;
  	}
		else {
			cdn = (*cdnsMap)[branchNames[i]];
			cdn->clear();
		}

  	// Set the cycle of update time.
  	cdn->updateCycle = cb()->cycle();

  	if (branchTypes[i] == "double") {
			double val = -999;
			int status = t->SetBranchAddress(branchNames[i].c_str(), &val);
			if (status == -5) std::cout<<"[Warning]: Branch not found in conditions tree: "<<branchNames[i]<<std::endl;
			else {
				t->GetEntry(0);
				cdn->vals.push_back(val);
				t->ResetBranchAddresses();
			}
  	}

  	else if (branchTypes[i] == "vector_doubles") {
  		std::vector<double> * vals = 0;
  		int status = t->SetBranchAddress(branchNames[i].c_str(), &vals);
  		if (status == -5) std::cout<<"[Warning]: Branch not found in conditions tree: "<<branchNames[i]<<std::endl;
			else {
				t->GetEntry(0);
				cdn->vals = (*vals);
				t->ResetBranchAddresses();
			}
  	}
  }
  cb()->setCurrentFileName(std::string(currentFile->GetName()));
  cb()->printConditions();
  std::cout<<"[Note]: ... updating conditions done."<<std::endl;
}


//==============================================================================

void SEightChan_swTrig::finalize() 
{
	std::cout<<"[Note]: Fraction data read in: "<<cb()->cycle()/float(m_chain->GetEntries())<<std::endl;
	std::cout<<"[Note]: Amount of data read-in (MB): "<<m_nBytesIn/1000000.<<std::endl;
}


//==============================================================================