//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Implementation of class PndTpcAdcTask
//      see PndTpcAdcTask.hh for details
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//
//
//-----------------------------------------------------------

// Panda Headers ----------------------

// This Class' Header ------------------
#include "PndTpcAdcTask.h"

// C/C++ Headers ----------------------
#include <algorithm>

// Collaborating Class Headers --------
#include "CbmRootManager.h"
#include "TClonesArray.h"
#include "TRandom.h"
#include "PndTpcSignal.h"
#include "PndTpcSignalAge.h"
#include "PndTpcSample.h"
#include "PndTpcFrontend.h"
#include "PndTpcCRRCPulseshape.h"
#include "PndTpcDigitizationPolicy.h"

#include <iostream>

// Class Member definitions -----------


PndTpcAdcTask::PndTpcAdcTask()
  : CbmTask("TPC Adc"), _persistence(kFALSE)
{
  _signalBranchName = "PndTpcSignal";
}


PndTpcAdcTask::~PndTpcAdcTask()
{
  if(_frontend!=0)delete _frontend;
  if(_pulseshape!=0)delete _pulseshape;
}

InitStatus
PndTpcAdcTask::Init()
{
  //Get ROOT Manager
  CbmRootManager* ioman= CbmRootManager::Instance();

  if(ioman==0)
    {
      Error("PndTpcAdcTask::Init","RootManager not instantiated!");
      return kERROR;
    }
  
  // Get input collection
  _signalArray=(TClonesArray*) ioman->GetObject(_signalBranchName);
  
  if(_signalArray==0)
    {
      Error("PndTpcAdcTask::Init","Signal-array not found!");
      return kERROR;
    }
  
  // create and register output array
  _sampleArray = new TClonesArray("PndTpcSample"); 
  ioman->Register("PndTpcSample","PndTpc",_sampleArray,_persistence);

  
  //TODO: Get this from Database!
  _frontend= new PndTpcFrontend(100,    // AdcThreshold
			     100000, // adcmax
			     10,     // adcbits
			     40,     // SamplingFreq_Mhz
			     -60000,      // t0
			     30,     // timebits
			     00);     // PSAthreshold

  _pulseshape= new PndTpcCRRCPulseshape(50,50,0.01); // tdiff,tint,tsig

  return kSUCCESS;
}


void
PndTpcAdcTask::Exec(Option_t* opt)
{
  // Reset output Array
  if(_sampleArray==0) Fatal("PndTpcAdc::Exec)","No SampleArray");
  _sampleArray->Delete();
  std::vector <PndTpcSignal*> sv;
  std::vector <PndTpcSample*> samplev;
  
  Int_t ns=_signalArray->GetEntriesFast();
  if(ns>0){
    for(Int_t is=0;is<ns;++is){
      // move signals into vector
      PndTpcSignal* sig=(PndTpcSignal*)_signalArray->At(is);
      sv.push_back(sig);
      //_signalArray->RemoveAt(is);
      //delete sig;
    } // end loop over signals
    //_signalArray->Clear();
    //sort in time
    std::cout<<"begin sorting signals in time"<<std::endl;
    sort(sv.begin(),sv.end(),PndTpcSignalAge());
    std::cout<<"end sorting signals in time"<<std::endl;

    PndTpcDigitizationPolicy().Digitize(&sv,&samplev,_frontend,_pulseshape);
  }

  //copy data into sample_array (TClonesvector)
  int nsamp=samplev.size();
  for(int isamp=0;isamp<nsamp;++isamp) {
    new((*_sampleArray)[isamp]) PndTpcSample(*(samplev[isamp]));
    delete samplev[isamp]; // clean up temporay store
  }

  std::cout<<_sampleArray->GetEntriesFast()<<" Samples created"<<std::endl;
  return;
}

ClassImp(PndTpcAdcTask)