cleaning up streamers

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

diff --git a/PWGLF/totEt/AliAnalysisEtSelector.cxx b/PWGLF/totEt/AliAnalysisEtSelector.cxx
index 79264adb2c52e6eaa1ebaa4ae43d2d2d4486b292..9bf9a33a562f3fd3cd10f17ee70f38a7eae0bbc5 100644 (file)
--- a/PWGLF/totEt/AliAnalysisEtSelector.cxx
+++ b/PWGLF/totEt/AliAnalysisEtSelector.cxx
@@ -19,10 +19,20 @@ 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()
@@ -32,8 +42,13 @@ AliAnalysisEtSelector::~AliAnalysisEtSelector()
     delete fClusterArray;
   }
 }
+void AliAnalysisEtSelector::SetEvent(const AliESDEvent* event)
+{
+    fEvent = event;
+    if(!fInitialized) Init(event);
+}
 
-Bool_t AliAnalysisEtSelector::CutNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const
+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);
 }
@@ -68,34 +83,72 @@ Int_t AliAnalysisEtSelector::GetPrimary(const Int_t partIdx, AliStack& stack) co
   }
   return -1;
 }
-Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(const TParticle& part, AliStack &stack) const
+//Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(const TParticle& part, AliStack &stack) const
+Bool_t AliAnalysisEtSelector::FromSecondaryInteraction(Int_t partID, AliStack &stack) const
 {
-  if(0)
-  {
-  // Let's ignore the gamma<->e+/- channel
-  UInt_t pdgCode = TMath::Abs(stack.Particle(part.GetFirstMother())->GetPdgCode());
+//   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);
   
-  if(pdgCode == fgGammaCode || fgEMinusCode)
-  {
-      std::cout << "EM channel" << std::endl;
-  }
-  }
-  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);
-  
-  //Let's find suspect decay (typical for secondary interaction)...
-  
-  return SuspeciousDecayInChain(211, 111, part, stack);
+
+  //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;
+//   }
   
                            
   
   
 }
 
-Bool_t AliAnalysisEtSelector::SuspeciousDecayInChain(const Int_t suspectMotherPdg, const Int_t suspectDaughterPdg, const TParticle &part, AliStack& stack) const
+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)
@@ -108,5 +161,21 @@ Bool_t AliAnalysisEtSelector::SuspeciousDecayInChain(const Int_t suspectMotherPd
   {
     return kTRUE;
   }
-  return SuspeciousDecayInChain(suspectMotherPdg, suspectDaughterPdg, *mother, stack);
+  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();
+}
+