enabling usage of several jet finders within one train
authorkleinb <kleinb@f7af4fe6-9843-0410-8265-dc069ae4e863>
Fri, 18 Dec 2009 07:15:40 +0000 (07:15 +0000)
committerkleinb <kleinb@f7af4fe6-9843-0410-8265-dc069ae4e863>
Fri, 18 Dec 2009 07:15:40 +0000 (07:15 +0000)
PWG4/JetTasks/AliAnalysisTaskUE.cxx
PWG4/JetTasks/AliAnalysisTaskUE.h

index addbd84..e31757a 100644 (file)
@@ -83,6 +83,7 @@ fTrigger(0),
 fDebug(0),
 fDeltaAOD(kFALSE),
 fDeltaAODBranch(""),
+fAODBranch("jets"),
 fArrayJets(0x0),           
 fAOD(0x0),            
 fAODjets(0x0),
@@ -130,9 +131,9 @@ fhRegionSumPtMaxVsEt(0x0),
 fhRegionMultMax(0x0),         
 fhRegionMultMaxVsEt(0x0),     
 fhRegionSumPtMinVsEt(0x0), //fhRegionMultMin(0x0),         
-fhRegionMultMinVsEt(0x0),     
-fhRegionAveSumPtVsEt(0x0),    
-fhRegionDiffSumPtVsEt(0x0),   
+fhRegionMultMinVsEt(0x0),
+fhRegionAveSumPtVsEt(0x0),
+fhRegionDiffSumPtVsEt(0x0),
 fhRegionAvePartPtMaxVsEt(0x0),
 fhRegionAvePartPtMinVsEt(0x0),
 fhRegionMaxPartPtMaxVsEt(0x0),
@@ -325,17 +326,19 @@ void  AliAnalysisTaskUE::AnalyseUE()
        }
        nJets=fArrayJets->GetEntries();
     }else{
-      if (fDebug > 1) AliInfo(" ==== Read Standard-AODs !");
-      nJets = fAODjets->GetNJets();
+      if (fDebug > 1) AliInfo(" ==== Read Standard-AODs  !");
+      if (fDebug > 1) AliInfo(Form(" ====  Reading Branch: %s  ",fAODBranch.Data()));
+      
+      nJets = ((TClonesArray*)fAODjets->FindListObject(fAODBranch.Data()))->GetEntries();
     }
     //printf("AOD %d jets \n", nJets);
 
     for( Int_t i=0; i<nJets; ++i ) {
        AliAODJet* jet;
       if (fDeltaAOD){
-      jet =(AliAODJet*)fArrayJets->At(i);
+       jet =(AliAODJet*)fArrayJets->At(i);
       }else{
-      jet = fAODjets->GetJet(i);
+       jet = (AliAODJet*)((TClonesArray*)fAODjets->FindListObject(fAODBranch.Data()))->At(i);
       }
       Double_t jetPt = jet->Pt();//*1.666; // FIXME Jet Pt Correction ?????!!!
  
@@ -356,7 +359,7 @@ void  AliAnalysisTaskUE::AnalyseUE()
       if (fDeltaAOD) {
        jet =(AliAODJet*) fArrayJets->At(index1);
       }else{
-       jet = fAODjets->GetJet(index1);
+       jet = (AliAODJet*)((TClonesArray*)fAODjets->FindListObject(fAODBranch.Data()))->At(index1);
       }
       if(jet)jetVect[0].SetXYZ(jet->Px(), jet->Py(), jet->Pz());
     }
@@ -365,7 +368,7 @@ void  AliAnalysisTaskUE::AnalyseUE()
       if (fDeltaAOD) {
        jet= (AliAODJet*) fArrayJets->At(index2);
        }else{
-       jet= fAODjets->GetJet(index2);
+       jet=(AliAODJet*) ((TClonesArray*)fAODjets->FindListObject(fAODBranch.Data()))->At(index2);
        }
       if(jet)jetVect[1].SetXYZ(jet->Px(), jet->Py(), jet->Pz());
     }
@@ -374,7 +377,7 @@ void  AliAnalysisTaskUE::AnalyseUE()
       if (fDeltaAOD) {
        jet= (AliAODJet*) fArrayJets->At(index3);
       }else{
-       fAODjets->GetJet(index3);
+       ((TClonesArray*)fAODjets->FindListObject(fAODBranch.Data()))->At(index3);
       }
       if(jet)jetVect[2].SetXYZ(jet->Px(), jet->Py(), jet->Pz());
     }
@@ -1184,6 +1187,8 @@ void  AliAnalysisTaskUE::CreateHistos()
    */
 }
 
+
+
 //____________________________________________________________________
 void  AliAnalysisTaskUE::Terminate(Option_t */*option*/)
 {
index bbd2b94..2f6fb00 100644 (file)
@@ -42,6 +42,7 @@ class  AliAnalysisTaskUE : public AliAnalysisTask
     // Read deltaAODs
     void   ReadDeltaAOD()                   { fDeltaAOD = kTRUE; }
     void   SelectDeltaAODBranch(const char* val)     { fDeltaAODBranch = val;   }
+    void   SelectAODBranch(const char* val)     { fAODBranch = val;   }
 
     // Setters for MC
     void  SetUseMCBranch(){fUseMCParticleBranch = kTRUE;}
@@ -53,7 +54,7 @@ class  AliAnalysisTaskUE : public AliAnalysisTask
     void   SetRegionType( Int_t val )     { fRegionType = val; }
     void   SetUseChPartJet( Int_t val )   { fUseChPartJet = val; }
     void   SetUseChargeHadrons( Bool_t val ){ fUseChargeHadrons = val; }
-    void   SetPtSumOrdering( Bool_t val ) { fOrdering = val;   }
+    void   SetPtSumOrdering( Int_t val ) { fOrdering = val;   }
     void   SetFilterBit( UInt_t val )     { fFilterBit = val;  }
     void   SetJetsOnFly( Bool_t val )     { fJetsOnFly = val;  }
     void   SetConeRadius( Double_t val )  { fConeRadius = val; }
@@ -73,7 +74,6 @@ class  AliAnalysisTaskUE : public AliAnalysisTask
   private:
     AliAnalysisTaskUE(const  AliAnalysisTaskUE &det);
     AliAnalysisTaskUE&   operator=(const  AliAnalysisTaskUE &det);
-    
     void   AnalyseUE();
     Int_t   IsTrackInsideRegion(TVector3 *jetVect, TVector3 *partVect);
     void   CreateHistos();
@@ -89,6 +89,7 @@ class  AliAnalysisTaskUE : public AliAnalysisTask
     Int_t      fDebug;           //  Debug flag
     Bool_t      fDeltaAOD;        //  Read jets from delta AOD 
     TString     fDeltaAODBranch;  //  Jet branch name from delta AOD
+    TString     fAODBranch;       //  Jet branch name from standard AOD
     TClonesArray*  fArrayJets;       //  Array of Jets from delta AOD
 
     AliAODEvent*  fAOD;             //! AOD Event