#include "PndMvdTrackFinderAnaTask.h"

#include <iostream>

// Root includes
#include "TROOT.h"
#include "TString.h"
#include "TClonesArray.h"
#include "TParticlePDG.h"

// framework includes
#include "CbmRootManager.h"
#include "CbmRunAna.h"
#include "CbmRuntimeDb.h"


// PndMvd includes
// #include "PndMvdTrackCand.h"
#include "TrackCand.h"


PndMvdTrackFinderAnaTask::PndMvdTrackFinderAnaTask() : CbmTask("MVD Track Finder Ana")
{
	fTrackCandBranch = "MVDTrackCand";
}

PndMvdTrackFinderAnaTask::~PndMvdTrackFinderAnaTask()
{
}

void PndMvdTrackFinderAnaTask::SetParContainers()
{
}

InitStatus PndMvdTrackFinderAnaTask::ReInit()
{
  
  InitStatus stat=kERROR;
  return stat;
}

// -----   Public method Init   --------------------------------------------
InitStatus PndMvdTrackFinderAnaTask::Init() 
{
  
  CbmRootManager* ioman = CbmRootManager::Instance();

  if ( ! ioman ) 
    {
      std::cout << "-E- PndMvdTrackFinderAna::Init: "
     << "RootManager not instantiated!" << std::endl;
      return kFATAL;
    }
    
  // Get input array
  fTrackCandArray = (TClonesArray*) ioman->GetObject(fTrackCandBranch);
  if ( !fTrackCandArray){
    std::cout << "-W- PndMvdTrackFinderAnaTask::Init: " << "No TrackCandArray!" << std::endl;
    return kERROR;
  }
  
  std::cout << "-I- PndMvdTrackFinderAnaTask: Initialisation successfull" << std::endl;
  return kSUCCESS;
}

// -----   Public method Exec   --------------------------------------------
void PndMvdTrackFinderAnaTask::Exec(Option_t* opt) 
{

  // Reset output array
  unsigned int detID, hitID;
  if ( ! fTrackCandArray )
    Fatal("Exec", "No trackCandArray");
  std::cout << "<<<<<<<<<<<< newEvent >>>>>>>>>>>>>>" << std::endl;
  for (int i = 0; i < fTrackCandArray->GetEntriesFast(); i++){
	  TrackCand* myTrackCand = (TrackCand*)(fTrackCandArray->At(i));
	  std::cout << "Track" << i << ": ";
	  for (int j = 0; j < myTrackCand->getNHits(); j++){
		  myTrackCand->getHit(j, detID, hitID);
		  std::cout << detID << "/" << hitID << " ";
	  }
	  std::cout << " Curv: " << myTrackCand->getCurv() << " Dip: " << myTrackCand->getDip()
	  		    << " Inv: " << myTrackCand->inverted() << std::endl;
  }
}

ClassImp(PndMvdTrackFinderAnaTask);