Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGLF / totEt / AliAnalysisEtSelector.cxx

//_________________________________________________________________________
//  Utility Class for transverse energy studies
//  Selector Base class 
//  -  
// implementation file
//
//*-- Authors: Oystein Djuvsland (Bergen)
//_________________________________________________________________________
#include "AliAnalysisEtSelector.h"
#include "AliAnalysisEtCuts.h"
#include "AliESDCaloCluster.h"
#include "AliStack.h"
#include "TParticle.h"
#include "TParticlePDG.h"
#include "AliAnalysisEtCommon.h"
#include <iostream>

ClassImp(AliAnalysisEtSelector);

AliAnalysisEtSelector::AliAnalysisEtSelector(AliAnalysisEtCuts *cuts) : AliAnalysisEtCommon()
,fEvent(0)
,fClusterArray(0)
,fCuts(cuts)
,fRunNumber(0)
,fInitialized(kFALSE)
{
}
AliAnalysisEtSelector::AliAnalysisEtSelector() : AliAnalysisEtCommon()
,fEvent(0)
,fClusterArray(0)
,fCuts(0)
,fRunNumber(0)
,fInitialized(kFALSE)
{
  fCuts = new AliAnalysisEtCuts();
}

AliAnalysisEtSelector::~AliAnalysisEtSelector()
{ // dtor
  if(fClusterArray)
  {
    delete fClusterArray;
  }
}
void AliAnalysisEtSelector::SetEvent(const AliESDEvent* event)
{
    fEvent = event;
    if(!fInitialized) Init(event);
}

Bool_t AliAnalysisEtSelector::IsNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const
{
  return s.IsPhysicalPrimary(pIdx) &&(TMath::Abs(TMath::Abs(pdg.Charge()) - fCuts->GetMonteCarloNeutralParticle())<1e-3);
}

Bool_t AliAnalysisEtSelector::IsEmEtParticle(const Int_t pdgCode) const
{
  return pdgCode == fgGammaCode || pdgCode == fgPi0Code || pdgCode == fgEtaCode || pdgCode == fgEPlusCode || pdgCode == fgEMinusCode;
}


Bool_t AliAnalysisEtSelector::PrimaryIsEmEtParticle(const Int_t pIdx, AliStack& stack) const
{
  return IsEmEtParticle(stack.Particle(GetPrimary(pIdx, stack))->GetPdgCode());
}
Int_t AliAnalysisEtSelector::GetPrimary(const Int_t partIdx, AliStack& stack) const
{ // get primary
  if(partIdx >= 0) 
  {
    Int_t mothIdx = stack.Particle(partIdx)->GetMother(0);
    if(mothIdx < 0) return -1;
    TParticle *mother = stack.Particle(mothIdx);
    if(mother)
    {
      if(stack.IsPhysicalPrimary(mothIdx)) return mothIdx;
      else return GetPrimary(mothIdx, stack);
    }
    else 
    {
      std::cout << "WAT!" << std::endl;
      return -1;
    }
  }
  return -1;
}
//Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(const TParticle& part, AliStack &stack) const
Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(Int_t partID, AliStack &stack) const
{
//   Bool_t partVtxSecondary = (
// 			      TMath::Sqrt(part.Vx()*part.Vx() + part.Vy()*part.Vy()) > fCuts->GetPrimaryVertexCutXY() 
// 			      || TMath::Abs(part.Vz()) > fCuts->GetPrimaryVertexCutZ()
// 			    )
// 			    && TMath::Sqrt(part.Vx()*part.Vx()+part.Vy()*part.Vy() + part.Vz()*part.Vz())<(fCuts->GetGeometryPhosDetectorRadius()-10);
  

  //Bool_t partVtxSecondary = (TMath::Sqrt(part.Vx()*part.Vx() + part.Vy()*part.Vy()) <420);
  //if(partVtxSecondary) return kFalse;
//   //Let's find suspect decay (typical for secondary interaction)...
//   if(partVtxSecondary){
  // return SuspiciousDecayInChain(211, 111, part, stack);
  //return stack.IsSecondaryFromMaterial(part.GetUniqueID());
  return stack.IsSecondaryFromMaterial(partID);
//   }
//   else{
//     return kFALSE;
//   }
  
			    
  
  
}

Int_t AliAnalysisEtSelector::GetMother(Int_t partID, AliStack& stack) const {
  if(partID>0){
    TParticle *particle = stack.Particle(partID);
      if(particle){
	
	return particle->GetMother(0);
      }
  }
  return -1;
}
Bool_t AliAnalysisEtSelector::IsFromDetectorCover(Int_t partID, AliStack& stack) const{
  if(partID>0){
    TParticle *particle = stack.Particle(partID);
    if(particle){//particle exists
      if(stack.IsSecondaryFromMaterial(partID)){//particle is from an interaction with the material
	//say that it's from the detector cover if its vertex is larger than 400 cm because this is where the cover is
	return (TMath::Sqrt(particle->Vx()*particle->Vx() + particle->Vy()*particle->Vy()) >400);
	
      }
    }
  }
  return kFALSE;
}

Int_t AliAnalysisEtSelector::GetFirstMotherNotFromDetectorCover(Int_t partID, AliStack& stack) const{
  Int_t targetParticle = partID;
  Int_t testParticle = partID;
  Int_t iteration  = 0;
  while(IsFromDetectorCover(targetParticle,stack) && testParticle>0){//if the particle is from the detector cover
    //cout<<"Particle "<<targetParticle<<" is from detector cover.  Getting mother ";
    testParticle = GetMother(targetParticle,stack);
    if(testParticle>0) targetParticle = testParticle;
    //cout<<targetParticle<<" iteration "<<iteration<<endl;
    iteration++;
  }
  //if(iteration>0) cout<<"iterations "<<iteration<<endl;
  return targetParticle;
}
Bool_t AliAnalysisEtSelector::SuspiciousDecayInChain(const UInt_t suspectMotherPdg, const UInt_t suspectDaughterPdg, const TParticle &part, AliStack& stack) const
{
  UInt_t partPdg = TMath::Abs(part.GetPdgCode());
  if(part.GetFirstMother() == -1)
  {
    return kFALSE;
  }
  TParticle *mother = stack.Particle(part.GetFirstMother()); 
  UInt_t motherPdg = TMath::Abs(mother->GetPdgCode());
  if((suspectDaughterPdg==partPdg || 2112 == partPdg )&& suspectMotherPdg == motherPdg)
  {
    return kTRUE;
  }
  return SuspiciousDecayInChain(suspectMotherPdg, suspectDaughterPdg, *mother, stack);
}

Float_t AliAnalysisEtSelector::ShiftAngle(Float_t phi){//Always returns an angle in radians between -pi<phi<pi
  float myphi = phi;
  while(myphi>TMath::Pi()){//angle is too high, decrease the angle
    myphi = myphi - 2*TMath::Pi();
  }
  while(myphi<-TMath::Pi()){//angle is too low, increase the angle
    myphi = myphi + 2*TMath::Pi();
  }
  return myphi;
}
Bool_t AliAnalysisEtSelector::PassMinEnergyCut(Double_t e) const
{
  return e > fCuts->GetReconstructedEmcalClusterEnergyCut();
}