#include "R3BPspDigitizer.h"
#include "TClonesArray.h"
#include "FairRootManager.h"
#include "FairRunAna.h"
#include "FairRuntimeDb.h"


// includes for modeling
#include "TGeoManager.h"
#include "TParticle.h"
#include "TVirtualMC.h"
#include "TGeoMatrix.h"
#include "TGeoMaterial.h"
#include "TGeoMedium.h"
#include "TGeoBBox.h"
#include "TGeoCompositeShape.h"
#include "TGeoShapeAssembly.h"


#include "TVector3.h"
#include "TMath.h"
#include "TRandom.h"
#include "TH1F.h"
#include "TH2F.h"
#include <string>
#include <iostream>


#include "R3BPspPoint.h"
#include "R3BMCTrack.h"
		
using std::cout;
using std::endl;
		

R3BPspDigitizer::R3BPspDigitizer() :
  FairTask("R3B Psp Digitization scheme ") { 
}


R3BPspDigitizer::~R3BPspDigitizer() {
}


void R3BPspDigitizer::SetParContainers() {

  // Get run and runtime database
  FairRunAna* run = FairRunAna::Instance();
  if ( ! run ) Fatal("SetParContainers", "No analysis run");

  FairRuntimeDb* rtdb = run->GetRuntimeDb();
  if ( ! rtdb ) Fatal("SetParContainers", "No runtime database");

  fPspDigiPar = (R3BPspDigiPar*)(rtdb->getContainer("R3BPspDigiPar"));

  if ( fPspDigiPar ) {
      cout << "-I- R3BPspDigitizer::SetParContainers() "<< endl;
      cout << "-I- Container R3BPspDigiPar  loaded " << endl;
  }

}




InitStatus R3BPspDigitizer::Init() {

//  cout<<"Init "<<endl;
  // Get input array 
  FairRootManager* ioman = FairRootManager::Instance();
  if ( ! ioman ) Fatal("Init", "No FairRootManager");
  fPspPoints = (TClonesArray*) ioman->GetObject("PSPPoint");
  fPspMCTrack = (TClonesArray*) ioman->GetObject("MCTrack");
   
  // Register output array DchDigi
  fPspDigi = new TClonesArray("R3BPspDigi",1000);
  ioman->Register("PspDigi", "Digital response in Psp", fPspDigi, kTRUE);
  
  eventNoPsp=0;
  
  // Initialise control histograms
//      PspXhis = new TH1F("PspXhis","PspXhis",600,-3.,3.);
//      PspXhis->GetXaxis()->SetTitle("Position");
//      PspXhis->GetYaxis()->SetTitle("Counts");

  return kSUCCESS;

}


void R3BPspDigitizer::Exec(Option_t* opt) {

   Reset();
   eventNoPsp+=1;
//     if(eventNoPsp/1000. == (int)eventNoPsp/1000.) cout<<"Event #: "<<eventNoPsp-1<<endl;
  
  Int_t nentriesPsp = fPspPoints->GetEntries();
  Int_t TrackIdPsp=0;
     
  Int_t psp3mul;
  Double_t psp3x, psp3y, psp3e;
  

//******************** PSP3 ********************//
  psp3mul=0;
  psp3x=0.0;
  psp3y=0.0;
  psp3e=0.0;
  
   for (Int_t l=0;l<nentriesPsp;l++){
//   cout<<"entries-Psp "<<l<<endl;
    
     R3BPspPoint *psp_obj = (R3BPspPoint*) fPspPoints->At(l);     
          
     TrackIdPsp = psp_obj->GetTrackID();
     R3BMCTrack *aTrack = (R3BMCTrack*) fPspMCTrack->At(TrackIdPsp);   
     Int_t PID = aTrack->GetPdgCode();
//     Int_t mother = aTrack->GetMotherId();
     

     Double_t fX_in = psp_obj->GetXIn();
     Double_t fY_in = psp_obj->GetYIn();
     Double_t fZ_in = psp_obj->GetZIn();
     Double_t fX_out = psp_obj->GetXOut();
     Double_t fY_out = psp_obj->GetYOut();
     Double_t fZ_out = psp_obj->GetZOut();
     Double_t PSPeloss = psp_obj->GetEnergyLoss()*1000;
     
     Double_t fX = ((fX_in + fX_out)/2);
     Double_t fY = ((fY_in + fY_out)/2);
     Double_t fZ = ((fZ_in + fZ_out)/2);
     
     
     
    
//     if (PID==1000080150 && mother<0){
    if (PID>1000501000){
     if(fZ>0.0){ //i.e. psp3

//      psp3x=(((fX + 0.0) * 1.0) - ((fZ - 94.1) * (0.0)));
     psp3x=-fX;  //to be consistent with mfi, gfi, tof-walls, right-handed?
//      psp3x=fX;  //+ for s412?
     psp3y=fY;
     psp3e += PSPeloss;
     
     
//      PspXhis->Fill(psp3x);
     psp3mul++;
     } //psp3
    } //PID
     
   }

AddHit( psp3mul, psp3x, psp3y, psp3e);


}
// -------------------------------------------------------------------------

void R3BPspDigitizer::Reset(){
// Clear the structure
//   cout << " -I- Digit Reset() called " << endl;

   
 if (fPspDigi ) fPspDigi->Clear();

}   

void R3BPspDigitizer::Finish()
{
// Write control histograms

//    PspXhis->Write();

}

R3BPspDigi* R3BPspDigitizer::AddHit(Int_t psp3mul,Double_t psp3x,Double_t psp3y,Double_t psp3e){   
  TClonesArray& clref = *fPspDigi;
  Int_t size = clref.GetEntriesFast();
  return new(clref[size]) R3BPspDigi( psp3mul, psp3x, psp3y, psp3e);
 
}



//R3BDchDigi* R3BDchDigitizer::AddHit(
 // return new(clref[size]) R3BDchDigi();
//}

ClassImp(R3BPspDigitizer)