#include <PndPidBremCorrector.h>

Inheritance diagram for PndPidBremCorrector:

Public Member Functions
	PndPidBremCorrector ()

virtual	~PndPidBremCorrector ()

virtual InitStatus	Init ()

virtual void	Exec (Option_t *opt)

void	SetStorageOfData (Bool_t p=kTRUE)

virtual void	FinishTask ()

void	SetOutputBranch (TString branch)

Private Member Functions
PndPidBremCorrected4Mom *	AddBremCorrected4Mom ()

double	GetSepPhotonE (PndPidCandidate *, std::vector< Int_t > &)

double	GetMergPhotonE (PndPidCandidate *, std::vector< Int_t > &)

void	GetEmcPhiBumpList (int iClust)

	PndPidBremCorrector (const PndPidBremCorrector &L)

PndPidBremCorrector &	operator= (const PndPidBremCorrector &)

	ClassDef (PndPidBremCorrector, 1)

Private Attributes
TClonesArray *	fBumpArray

TClonesArray *	fClusterArray

TClonesArray *	fPhiBumpArray

TClonesArray *	fChargedCandidateArray

TClonesArray *	fNeutralCandidateArray

TClonesArray *	fBremCorrected4MomArray

double	fRecMomOfEle

double	fRecThetaOfEle

double	fRecPhiOfEle

int	fCharge

Double_t	fSepPhotonE

Double_t	fMergPhotonE

std::vector< PndEmcBump * >	fEmcPhiBumpList

TString	fTrackBranchNamePidHypo

Bool_t	fPersistance

Detailed Description

Definition at line 23 of file PndPidBremCorrector.h.

Constructor & Destructor Documentation

PndPidBremCorrector::PndPidBremCorrector ( )

Definition at line 36 of file PndPidBremCorrector.cxx.

                                         :
    fBumpArray(0), fClusterArray(0), fPhiBumpArray(0), fChargedCandidateArray(0), fNeutralCandidateArray(0), fBremCorrected4MomArray(0),fRecMomOfEle(0), fRecThetaOfEle(0), fRecPhiOfEle(0), fCharge(0), fSepPhotonE(0.), fMergPhotonE(0.), fEmcPhiBumpList(),  fTrackBranchNamePidHypo(""), fPersistance(kTRUE)
 {
 
 }

PndPidBremCorrector::~PndPidBremCorrector ( )

virtual

Definition at line 42 of file PndPidBremCorrector.cxx.

 {
 
 }

PndPidBremCorrector::PndPidBremCorrector ( const PndPidBremCorrector & L )

private

Member Function Documentation

PndPidBremCorrected4Mom * PndPidBremCorrector::AddBremCorrected4Mom ( )

private

Definition at line 144 of file PndPidBremCorrector.cxx.

References fBremCorrected4MomArray.

Referenced by Exec().

                                                                   {
   TClonesArray& clref = *fBremCorrected4MomArray;
   Int_t size = clref.GetEntriesFast();
   return new(clref[size]) PndPidBremCorrected4Mom();
 }

PndPidBremCorrector::ClassDef	(	PndPidBremCorrector	,
		1
	)

private

void PndPidBremCorrector::Exec ( Option_t * opt )

virtual

Virtual method Exec

Definition at line 102 of file PndPidBremCorrector.cxx.

References AddBremCorrected4Mom(), fBremCorrected4MomArray, fCharge, fChargedCandidateArray, fMergPhotonE, fRecMomOfEle, fRecPhiOfEle, fRecThetaOfEle, fSepPhotonE, PndPidCandidate::GetCharge(), PndPidCandidate::GetEnergy(), GetMergPhotonE(), PndPidCandidate::GetMomentum(), GetSepPhotonE(), i, and mom.

 {
 
   // Reset output array
   if ( ! fBremCorrected4MomArray ) Fatal("Exec", "No BremCorrected4Mom Array");
   fBremCorrected4MomArray->Delete();
 
   int nChargedCand = fChargedCandidateArray->GetEntriesFast();
 
   for (int iCand = 0; iCand<nChargedCand; ++iCand){
 
     PndPidCandidate* theChargedCand = (PndPidCandidate*) fChargedCandidateArray->At(iCand);
     fRecMomOfEle = theChargedCand->GetMomentum().Mag();
     fRecThetaOfEle = theChargedCand->GetMomentum().Theta()*TMath::RadToDeg();
     fRecPhiOfEle = theChargedCand->GetMomentum().Phi()*TMath::RadToDeg();
     fCharge = theChargedCand->GetCharge();
 
     TVector3 mom = theChargedCand->GetMomentum();
     double ene = theChargedCand->GetEnergy();
     TLorentzVector m4 = TLorentzVector(mom,ene);
     //double mass = m4.M();
     double mass = 0.000511; // Electron mass hypothesis (GeV)
 
     std::vector<int> sep_bumps, phi_bumps;
     fSepPhotonE = GetSepPhotonE(theChargedCand, sep_bumps);
     fMergPhotonE = GetMergPhotonE(theChargedCand, phi_bumps);
     double energy_gamma = fSepPhotonE + fMergPhotonE;
 
     TVector3 momCorr = ((energy_gamma+fRecMomOfEle)/fRecMomOfEle) * mom;
     double eneCorr = TMath::Hypot(mass, momCorr.Mag());
 
     PndPidBremCorrected4Mom *bremCorr = AddBremCorrected4Mom();
     bremCorr->SetMomentum(momCorr);
     bremCorr->SetEnergy(eneCorr);
     for (size_t i=0; i < sep_bumps.size(); ++i) bremCorr->AddToSepBumpList(sep_bumps[i]);
     for (size_t i=0; i < phi_bumps.size(); ++i) bremCorr->AddToPhiBumpList(phi_bumps[i]);
     bremCorr->SetPidCandIdx(iCand);
 
   }
 
 }

virtual void PndPidBremCorrector::FinishTask ( )

inlinevirtual

Definition at line 43 of file PndPidBremCorrector.h.

43 {};

void PndPidBremCorrector::GetEmcPhiBumpList ( int iClust )

private

Definition at line 241 of file PndPidBremCorrector.cxx.

References fEmcPhiBumpList, fPhiBumpArray, and PndEmcBump::GetClusterIndex().

Referenced by GetMergPhotonE().

                                                       {
   fEmcPhiBumpList.clear();
   int nPhiBump = fPhiBumpArray->GetEntriesFast();
   for (int ipb=0; ipb<nPhiBump; ++ipb) {
     PndEmcBump *phibump = (PndEmcBump*) fPhiBumpArray->At(ipb);
     if ( phibump->GetClusterIndex() == iClust ) {
       fEmcPhiBumpList.push_back(phibump);
     }
   }
 }

double PndPidBremCorrector::GetMergPhotonE	(	PndPidCandidate *	,
		std::vector< Int_t > &
	)

private

Definition at line 202 of file PndPidBremCorrector.cxx.

References Double_t, energy, fBumpArray, fCharge, fEmcPhiBumpList, fRecMomOfEle, PndEmcBump::GetClusterIndex(), PndPidCandidate::GetEmcIndex(), GetEmcPhiBumpList(), and r.

Referenced by Exec().

                                                                                                  {
 
   Double_t PhotonTotEnergyMerg = 0.0;
 
   // no EMcal cluster associated with track ...
   if (ChargedCand->GetEmcIndex() < 0) return 0.0;
 
   PndEmcBump *EleBump = (PndEmcBump *) fBumpArray->At(ChargedCand->GetEmcIndex());
   Int_t EleRefCluster = EleBump->GetClusterIndex();
 
   if (EleRefCluster < 0) return 0.0;
   GetEmcPhiBumpList(EleRefCluster);
 
   //Double_t EnergyCut = 0.15/TMath::Sin(fRecThetaOfEle*TMath::DegToRad()); //[R.K. 01/2017] unused variable
   //Double_t EleEnergy = 0; //[R.K. 01/2017] unused variable
 
   int iMax = 0;
   Float_t eMax = -1e9;
   for (size_t ib = 0; ib < fEmcPhiBumpList.size(); ++ib) {
     if( fEmcPhiBumpList[ib]->energy() > eMax) {
       iMax = ib;
       eMax = fEmcPhiBumpList[ib]->energy();
     }
   }
   Int_t iS = fCharge<0?0:iMax+1;
   Int_t iE = fCharge<0?iMax-1:fEmcPhiBumpList.size()-1;
   for(Int_t r = iS; r<=iE; r++) {
     PhotonTotEnergyMerg += fEmcPhiBumpList[r]->energy();
     phi_bumps.push_back(r);
   }
 
   if(PhotonTotEnergyMerg > fRecMomOfEle/100.)  {
     return PhotonTotEnergyMerg;
   } else {
     return 0.0;
   }
 
 }

double PndPidBremCorrector::GetSepPhotonE	(	PndPidCandidate *	,
		std::vector< Int_t > &
	)

private

Definition at line 150 of file PndPidBremCorrector.cxx.

References CAMath::ASin(), Bool_t, Double_t, fBumpArray, fCharge, fRecMomOfEle, fRecPhiOfEle, fRecThetaOfEle, PndEmcBump::GetClusterIndex(), PndPidCandidate::GetEmcIndex(), PndEmcCluster::GetEnergyCorrected(), PndEmcCluster::position(), CAMath::Sin(), and CAMath::Tan().

Referenced by Exec().

                                                                                                 {
 
   Float_t PhotonTotEnergySepWtd = 0;
 
   const int iTrkEmcIdx = ChargedCand->GetEmcIndex();
   if (iTrkEmcIdx<0) return 0;
 
   const int nBump = fBumpArray->GetEntriesFast();
   for(Int_t iBump = 0; iBump<nBump; ++iBump)
     {
       PndEmcBump *PhotonBump = (PndEmcBump *) fBumpArray->At(iBump);
       const Float_t PhotonEnergySep = PhotonBump->GetEnergyCorrected();
 
       const Int_t iSepBump = PhotonBump->GetClusterIndex();
       if ( iSepBump == iTrkEmcIdx ) continue;
 
       const Double_t PhotonThetaSep = PhotonBump->position().Theta()*TMath::RadToDeg();
       const Double_t PhotonPhiSep = PhotonBump->position().Phi()*TMath::RadToDeg();
 
       const Bool_t fwd = fRecThetaOfEle <= 23.;
       const Float_t Pt = fRecMomOfEle*TMath::Sin(fRecThetaOfEle/TMath::RadToDeg());
       const Float_t DeltaPhiBarrel = TMath::ASin(0.12/Pt)*2.*TMath::RadToDeg();
       const Float_t DeltaPhiForward = (0.6*2.0/Pt)*TMath::Tan(fRecThetaOfEle/57.3)*57.3;
 
       const Float_t RealDeltaPhi = fCharge<0?PhotonPhiSep-fRecPhiOfEle:fRecPhiOfEle-PhotonPhiSep;
       const Float_t RealDeltaTheta = fCharge<0?PhotonThetaSep-fRecThetaOfEle:fRecThetaOfEle-PhotonThetaSep;
 
       const Float_t rad_calc = 100.*TMath::Sin(RealDeltaPhi*TMath::DegToRad()/2.)*2.0*Pt/0.3/2.0; // B=2T
       const Float_t zed_calc = rad_calc/TMath::Tan(TMath::DegToRad()*fRecThetaOfEle);
 
       const Float_t wt = fwd ? 1.0/(1.+TMath::Exp((zed_calc-90.)/25.)) : 1.0/(1.+TMath::Exp((rad_calc-21.)/5.));
       const Float_t ThetaCutUp = 2.;
       const Float_t ThetaCutDown = -2.;
       const Float_t PhiCutUp = fwd ? DeltaPhiForward : DeltaPhiBarrel;
       const Float_t PhiCutDown = -1;
 
       const Bool_t PhiCut = RealDeltaPhi <= PhiCutUp && RealDeltaPhi >= PhiCutDown;
       const Bool_t ThetaCut = RealDeltaTheta <= ThetaCutUp && RealDeltaTheta >= ThetaCutDown;
 
       if (PhiCut && ThetaCut) {
         PhotonTotEnergySepWtd += wt*PhotonEnergySep;
         sep_bumps.push_back(iSepBump);
       }
 
     }//loop neutralcand
 
   if (PhotonTotEnergySepWtd < fRecMomOfEle/100.) PhotonTotEnergySepWtd = 0;
 
   return PhotonTotEnergySepWtd;
 
 }

InitStatus PndPidBremCorrector::Init ( )

virtual

Virtual method Init

Definition at line 47 of file PndPidBremCorrector.cxx.

References fBremCorrected4MomArray, fBumpArray, fChargedCandidateArray, fClusterArray, fNeutralCandidateArray, fPersistance, fPhiBumpArray, and fTrackBranchNamePidHypo.

                                      {
 
   // Get RootManager
   FairRootManager* ioman = FairRootManager::Instance();
   if ( ! ioman ){
     cout << "-E- PndPidBremCorrector::Init: "
          << "RootManager not instantiated!" << endl;
     return kFATAL;
   }
 
 
  fClusterArray = dynamic_cast<TClonesArray *> (ioman->GetObject("EmcCluster"));
   if ( ! fClusterArray ) {
     cout << "-W- PndEmcMakeBump::Init: "
          << "No PndEmcCluster array!" << endl;
     return kERROR;
   }
 
  fPhiBumpArray = dynamic_cast<TClonesArray *> (ioman->GetObject("EmcPhiBump"));
   if ( ! fPhiBumpArray ) {
     cout << "-W- PndEmcMakePhiBump::Init: "
          << "No PhiBumpArray array!" << endl;
     return kERROR;
   }
 
  fBumpArray = dynamic_cast<TClonesArray *> (ioman->GetObject("EmcBump"));
   if ( ! fBumpArray ) {
     cout << "-W- PndEmcMakeBump::Init: "
          << "No PndEmcBump array!" << endl;
     return kERROR;
   }
 
  fChargedCandidateArray = dynamic_cast<TClonesArray *> (ioman->GetObject("PidChargedCand"+fTrackBranchNamePidHypo));
   if ( ! fChargedCandidateArray ) {
     cout << "-W- PndEmcMakeBump::Init: "
          << "No PidChargedCand array!" << endl;
     return kERROR;
   }
 
 
  fNeutralCandidateArray = dynamic_cast<TClonesArray *> (ioman->GetObject("PidNeutralCand"+fTrackBranchNamePidHypo));
   if ( ! fNeutralCandidateArray ) {
     cout << "-W- PndEmcMakeBump::Init: "
          << "No PidNeutralCand array!" << endl;
     return kERROR;
   }
 
 
  fBremCorrected4MomArray = new TClonesArray("PndPidBremCorrected4Mom");
   ioman->Register("BremCorrected4Mom"+fTrackBranchNamePidHypo,"Pid",fBremCorrected4MomArray,fPersistance);
 
   return kSUCCESS;
 
 }