// ------------------------------------------------------------------------------
// -----                PndDchPreFitterTRQATask2 source file                -----
// -----            Created 03.11.2008  by A. Wronska, P.Hawranek           -----
// ------------------------------------------------------------------------------

// Panda Headers ----------------------
#include "PndDchPreFitterTRQATask.h"
#include "CbmRootManager.h"
#include "CbmMCTrack.h"
#include "FitterExceptions.h"
#include "PndDchTrackMatch.h"
#include "PndDchTrack.h"

// ROOT Headers -----------------------
#include "TClonesArray.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TFile.h"
#include "TGeoManager.h"
#include "TCanvas.h"
#include "TMath.h"
#include "TStyle.h"

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


PndDchPreFitterTRQATask::PndDchPreFitterTRQATask()
  : CbmTask("QA Task for Kalman od DCH"){
  fhP = NULL;
  fhChi2 = NULL;
  fThetaH = NULL;
  fPhiH = NULL;
  fCanvas = NULL;
}


PndDchPreFitterTRQATask::~PndDchPreFitterTRQATask() {
  
  WriteHistograms();

  if(fhP!=NULL)    delete fhP;
  if(fhChi2!=NULL) delete fhChi2;
  if(0!=fThetaH)   delete fThetaH;
  if(0!=fPhiH)     delete fPhiH;
  if(0!=fCanvas)   delete fCanvas;
}

InitStatus PndDchPreFitterTRQATask::Init(){
  //Get ROOT Manager
  CbmRootManager* ioman= CbmRootManager::Instance();
  if(ioman==0){
      Error("PndDchPreFitterTRQATask::Init","RootManager not instantiated!");
      return kERROR;
    }
  
  fTrackArray=(TClonesArray*) ioman->GetObject("PndDchTrack");
  if(fTrackArray==0) {
    Error("PndDchPreFitterTRQATask::Init","PndDchTrack not found!");
    return kERROR;
  }

  fMCTrackArray=(TClonesArray*) ioman->GetObject("MCTrack");
  if(fMCTrackArray==0) {
    Error("PndDchPreFitterTRTask2::Init","MCTrack array not found!");
    return kERROR;
  }
  
  fDchTrackMatchArray = (TClonesArray*) ioman->GetObject("PndDchTrackMatch"); 
  if(fDchTrackMatchArray==0){
    Error("PndDchPrepareKalmanTracks2::Init","PndDchTrackMatch array not found!");
    return kERROR;
  }
  
  // // setup histograms
  fhP    = new TH1D("pullP","(p_{Rec}-p_{MC})/p_{MC}",60,-0.15,0.15);
  fhP->SetFillColor(9);
  fhChi2 = new TH1D("chi2","chi2",500,0,100);
  fhChi2->SetFillColor(2);
  fThetaH =new TH2D("theta","#theta_{rec}-#theta_{MC} vs #theta_{MC}",25,-5,25,100,-1,1.);
  fThetaH->SetFillColor(9);
  fPhiH   =new TH2D("phi",  "#phi_{rec}-#phi_{MC} vs #phi_{MC}",45,-180,180,100,-10,10);
  fPhiH->SetFillColor(9);
  
  return kSUCCESS;
}


void PndDchPreFitterTRQATask::Exec(Option_t* opt) {
  if(fVerbose>0)
    std::cout<<"Entering PndDchPreFitterTRQATask::Exec"<<std::endl;
  Int_t ntracks=fTrackArray->GetEntries();
  
  for(Int_t itr=0;itr<ntracks;++itr){
    PndDchTrack* trk=new PndDchTrack(*(PndDchTrack*) fTrackArray->At(itr));
    if(trk!=0){
      TVector3 mom0;
      trk->GetParamFirst()->Momentum(mom0);
      
      Int_t mcTrid = -1;
      Int_t id = 0;
      while(id<fDchTrackMatchArray->GetEntriesFast()){
	PndDchTrackMatch* dchtrmatch = (PndDchTrackMatch*) fDchTrackMatchArray->At(id);
	if(dchtrmatch->GetRecTrackID()==itr){
	  mcTrid = dchtrmatch->GetMCTrackID();
	  break;
	}
	id++;
      }
      if(mcTrid<0){
	if(fVerbose>0) std::cout<<"No MCTrack to compare the PndDchTrack with!\n";
	return;
      }
      
      CbmMCTrack* mc=(CbmMCTrack*)fMCTrackArray->At(mcTrid);
      if(mc==0){
	Error("PndDchPrepareKalmanTracks::Exec","MCTrack Id=0 not found!");
	continue;
      }
      TVector3 mcmom = mc->GetMomentum();
      if(fhP) fhP->Fill((mom0.Mag()-mcmom.Mag())/mcmom.Mag());
      if(fThetaH) fThetaH->Fill(mcmom.Theta()*TMath::RadToDeg(),
				(-mcmom.Theta()+mom0.Theta())*TMath::RadToDeg());
      if(fPhiH) fPhiH->Fill(mcmom.Phi()*TMath::RadToDeg(),
			    (-mcmom.Phi()+mom0.Phi())*TMath::RadToDeg());
    }
    delete trk;
  }
  return;
}
  
Bool_t  PndDchPreFitterTRQATask::WriteHistograms(){
  TFile* file = CbmRootManager::Instance()->GetOutFile();
  file->cd();
  file->mkdir("DchPreFitterTRQA");
  file->cd("DchPreFitterTRQA");


  if(0!=fhP) fhP->Write();
  if(0!=fhChi2) fhChi2->Write();
  if(0!=fThetaH) fThetaH->Write();
  if(0!=fPhiH) fPhiH->Write();
  if(0!=fCanvas)  fCanvas->Write();
  return kTRUE;
}

void PndDchPreFitterTRQATask::PlotHistograms(){
  gStyle->SetOptStat(111111);
  gStyle->SetPalette(1,0);
  fCanvas = new TCanvas("DchPreFitterQACanvas","Results from the DchPreFitterTRQA",1000,700);
  fCanvas->Divide(2,2);
  
  if(fhP!=0){
    fCanvas->cd(1);
    gPad->SetGrid(1,0);
    fhP->Draw();
  }
  if(0!=fPhiH){
    fCanvas->cd(2);
    gPad->SetGrid(1,1);
    fPhiH->Draw("colz");
  }
  if(0!=fThetaH){
    fCanvas->cd(3);
    gPad->SetGrid(1,1);
    fThetaH->Draw("colz");
  }
  if(0!=fhChi2){
    fCanvas->cd(4);
    gPad->SetGrid(1,0);
    fhChi2->Draw();
  }
}

 
void PndDchPreFitterTRQATask::Finish(){
  WriteHistograms();
}
  

ClassImp(PndDchPreFitterTRQATask)