Setting for HF jets, fix
authorsalapoin <s.lapointe@cern.ch>
Mon, 21 Apr 2014 18:56:21 +0000 (20:56 +0200)
committermvl <marco.van.leeuwen@cern.ch>
Tue, 22 Apr 2014 10:47:37 +0000 (12:47 +0200)
JETAN/AliAnalysisTaskJetCluster.cxx
JETAN/AliAnalysisTaskJetCluster.h

index e663917..8173f51 100644 (file)
@@ -1901,12 +1901,9 @@ Int_t  AliAnalysisTaskJetCluster::GetListOfTracks(TList *list,Int_t type){
          if (trPt <= 20. && (ntpcClus < NTPCClsCut.Eval(trPt))) continue;
          else if (trPt > 20. && ntpcClus < 100) continue;
 
-         Int_t idtr1 = tr->GetID(); 
-         Bool_t flagSame = kFALSE;
-         AvoidDoubleCountingHF(aod, idtr1, flagSame);
-         if (flagSame) continue; 
+         if (AvoidDoubleCountingHF(aod,tr)) continue;
        }
-       // 
+       //
 
         if(fRequireITSRefit){if((tr->GetStatus()&AliESDtrack::kITSrefit)==0)continue;}
         if (fApplySharedClusterCut) {
@@ -2109,23 +2106,25 @@ Int_t AliAnalysisTaskJetCluster::AddDaughters(TList * list, AliAODMCParticle *pa
        }                       
 return count;  
 }
-void AliAnalysisTaskJetCluster::AvoidDoubleCountingHF(AliAODEvent *aod, Int_t idtr1,  Bool_t &fFlagSameTr){
+
+
+Bool_t AliAnalysisTaskJetCluster::AvoidDoubleCountingHF(AliAODEvent *aod, AliAODTrack *tr1){
   
-       Bool_t sametr = kFALSE;
+  if(!(tr1->TestFilterBit(BIT(9)))) return kFALSE;
 
-     for(int jt = 0;jt < aod->GetNumberOfTracks();++jt){
+  Int_t idtr1 = tr1->GetID();
 
-       AliAODTrack *tr2 = aod->GetTrack(jt);
-       Int_t idtr2 = tr2->GetID();
-       if (idtr2>-1) continue;
+  for(int jt = 0;jt < aod->GetNumberOfTracks();++jt){
 
-       idtr2=-1*idtr2;
-       if (idtr1==idtr2-1) sametr = kTRUE;
+    const AliAODTrack *tr2 = aod->GetTrack(jt);
+    Int_t idtr2 = tr2->GetID();
+       
+    if (!(tr2->TestFilterBit(BIT(4)))) continue;
+    if (idtr1==(idtr2+1)*-1.) return kTRUE;
        
-     }
-     fFlagSameTr = sametr;
+  }
+  return kFALSE;
 }
-
 void AliAnalysisTaskJetCluster::LoadTrPtResolutionRootFileFromOADB() {
 
    if (!gGrid) {
index 8424bd5..042de73 100644 (file)
@@ -27,6 +27,7 @@ namespace fastjet {
 class AliJetHeader;
 class AliESDEvent;
 class AliAODEvent;
+class AliAODTrack;
 class AliAODExtension;
 class AliAODJet;
 class AliGenPythiaEventHeader;
@@ -158,7 +159,7 @@ class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
 
     Int_t GetListOfTracks(TList *list,Int_t type);
        Int_t AddDaughters(TList * list, AliAODMCParticle *part, TClonesArray * tca);
-    void AvoidDoubleCountingHF(AliAODEvent *aod, Int_t idtr1, Bool_t &fFlagSameTr);
+    Bool_t AvoidDoubleCountingHF(AliAODEvent *aod, AliAODTrack *tr1);
        
     AliAODEvent     *fAOD;                // ! where we take the jets from can be input or output AOD
     AliAODExtension *fAODExtension;       // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard