Derive from AliAnalysisTaskSE.
authormorsch <morsch@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 28 Feb 2008 14:37:05 +0000 (14:37 +0000)
committermorsch <morsch@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 28 Feb 2008 14:37:05 +0000 (14:37 +0000)
JETAN/AliAnalysisTaskJets.cxx
JETAN/AliAnalysisTaskJets.h
JETAN/AliJetESDReader.cxx
JETAN/AliJetESDReader.h
JETAN/AliJetFinder.cxx
JETAN/AliJetFinder.h
JETAN/AliJetKineReader.cxx
JETAN/AliJetKineReader.h
JETAN/AliJetReader.h

index 20e7c09..d41d27b 100644 (file)
@@ -41,12 +41,10 @@ ClassImp(AliAnalysisTaskJets)
 ////////////////////////////////////////////////////////////////////////
 
 AliAnalysisTaskJets::AliAnalysisTaskJets():
+    AliAnalysisTaskSE(),
     fDebug(0),
     fJetFinder(0x0),
     fTree(0x0),
-    fESD(0x0),
-    fAOD(0x0),
-    fTreeA(0x0),
     fHistos(0x0),
     fListOfHistos(0x0)
 {
@@ -54,33 +52,25 @@ AliAnalysisTaskJets::AliAnalysisTaskJets():
 }
 
 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
-    AliAnalysisTask(name, "AnalysisTaskJets"),
+    AliAnalysisTaskSE(name),
     fDebug(0),
     fJetFinder(0x0),
     fTree(0x0),
-    fESD(0x0),
-    fAOD(0x0),
-    fTreeA(0x0),
     fHistos(0x0),
     fListOfHistos(0x0)
 {
   // Default constructor
-    DefineInput (0, TChain::Class());
-    DefineOutput(0, TTree::Class());
     DefineOutput(1, TList::Class());
 }
 
-void AliAnalysisTaskJets::CreateOutputObjects()
+void AliAnalysisTaskJets::UserCreateOutputObjects()
 {
 // Create the output container
 //
-//  Default AOD
     if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
-    AliAODHandler* handler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
-    
-    fAOD   = handler->GetAOD();
-    fTreeA = handler->GetTree();
-    fJetFinder->ConnectAOD(fAOD);
+//  Connec default AOD to jet finder
+
+    fJetFinder->ConnectAOD(AODEvent());
 //
 //  Histograms
     OpenFile(1);
@@ -103,42 +93,24 @@ void AliAnalysisTaskJets::Init()
     fJetFinder->WriteHeaders();
 }
 
-void AliAnalysisTaskJets::ConnectInputData(Option_t */*option*/)
-{
-// Connect the input data
-    if (fDebug > 1) printf("AnalysisTaskJets::ConnectInputData() \n");
-    AliESDInputHandler* esdH = (AliESDInputHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
-    fESD = esdH->GetEvent();
-    fTree = esdH->GetTree();
-    
     
 
-    AliMCEventHandler*    mcTruth = (AliMCEventHandler*) 
-       ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler());
-    
-    fJetFinder->GetReader()->SetInputEvent(fESD, fAOD, mcTruth);
-}
 
-void AliAnalysisTaskJets::Exec(Option_t */*option*/)
+
+void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
 {
 // Execute analysis for current event
 //
-    AliMCEventHandler*    mctruth = (AliMCEventHandler*) 
-       ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler());
-    if (mctruth) {
-       AliStack* stack = mctruth->MCEvent()->Stack();
-       //printf("AliAnalysisTaskJets: Number of tracks on stack %5d\n", stack->GetNtrack());
-    }
     
-    Long64_t ientry = fTree->GetReadEntry();
-    if (fDebug > 1) printf("Analysing event # %5d\n", (Int_t) ientry);
-    fJetFinder->ProcessEvent(ientry);
+    if (MCEvent()) {
+       AliStack* stack = MCEvent()->Stack();
+    }
+    fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
+    fJetFinder->ProcessEvent();
 
     // Fill control histos
-    fHistos->FillHistos(fAOD->GetJets());
-    
+    fHistos->FillHistos(AODEvent()->GetJets());
     // Post the data
-    PostData(0, fTreeA);
     PostData(1, fListOfHistos);
 }
 
index 38e46a3..c2f900f 100644 (file)
@@ -4,7 +4,7 @@
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
  
-#include "AliAnalysisTask.h"
+#include "AliAnalysisTaskSE.h"
 class AliJetFinder;
 class AliESDEvent;
 class TTree;
@@ -12,18 +12,17 @@ class AliAODEvent;
 class AliJetHistos;
 
 
-class AliAnalysisTaskJets : public AliAnalysisTask
+class AliAnalysisTaskJets : public AliAnalysisTaskSE
 {
  public:
     AliAnalysisTaskJets();
     AliAnalysisTaskJets(const char* name);
     virtual ~AliAnalysisTaskJets() {;}
     // Implementation of interface methods
-    virtual void ConnectInputData(Option_t *option = "");
-    virtual void CreateOutputObjects();
+    virtual void UserCreateOutputObjects();
     virtual void Init();
     virtual void LocalInit() {Init();}
-    virtual void Exec(Option_t *option);
+    virtual void UserExec(Option_t *option);
     virtual void Terminate(Option_t *option);
     virtual void SetDebugLevel(Int_t level) {fDebug = level;}
     
@@ -31,12 +30,8 @@ class AliAnalysisTaskJets : public AliAnalysisTask
     Int_t         fDebug;        //  Debug flag
     AliJetFinder* fJetFinder;    //  Pointer to the jet finder 
     TTree*        fTree;         //! The input tree
-    AliESDEvent*  fESD;          //! ESD
-    AliAODEvent*  fAOD;          //! AOD
-    TTree*        fTreeA;        //  tree of reconstructed jets
     AliJetHistos* fHistos;       //  Histogram manager class
     TList*        fListOfHistos; //  Output list of histograms
-    
     ClassDef(AliAnalysisTaskJets, 2); // Analysis task for standard jet analysis
 };
  
index 10af1cd..dd6d682 100755 (executable)
@@ -149,7 +149,7 @@ void AliJetESDReader::SetInputEvent(TObject* esd, TObject* /*aod*/, TObject* /*m
 }
 
 //____________________________________________________________________________
-Bool_t AliJetESDReader::FillMomentumArray(Int_t /*event*/)
+Bool_t AliJetESDReader::FillMomentumArray()
 {
   // Fill momentum array
 
index 55d41cb..fb4e9bb 100755 (executable)
@@ -35,7 +35,7 @@ class AliJetESDReader : public AliJetReader
   Int_t   GetTrackSign() const {return fSign;}  // returns sign of the track
 
   // Setters
-  Bool_t FillMomentumArray(Int_t event); 
+  Bool_t FillMomentumArray(); 
   void   OpenInputFiles();
   void   InitUnitArray();
   void   CreateTasks();
index 02416cc..f5df125 100644 (file)
@@ -149,7 +149,7 @@ void AliJetFinder::Run()
 
   // loop over events
   for (Int_t i=nFirst;i<nLast;i++) {
-      fReader->FillMomentumArray(i);
+      fReader->FillMomentumArray();
       fLeading->FindLeading(fReader);
       fReader->GetGenJets(fGenJets);
 
@@ -204,15 +204,13 @@ void AliJetFinder::WriteHeaders()
 }
 
 
-Bool_t AliJetFinder::ProcessEvent(Long64_t entry)
+Bool_t AliJetFinder::ProcessEvent()
 {
 //
 // Process one event
 //
     Int_t debug  = fReader->GetReaderHeader()->GetDebug();
-    if (debug > 0) printf("<<<<< Processing Event %5d >>>>> \n", (Int_t) entry);
-    
-    Bool_t ok = fReader->FillMomentumArray(entry);
+    Bool_t ok = fReader->FillMomentumArray();
     if (!ok) return kFALSE;
 
     // Leading particles
index 74286c1..0261d00 100755 (executable)
@@ -53,7 +53,7 @@ class AliJetFinder : public TObject
   virtual void FindJetsTPC(){}
   virtual void WriteJHeaderToFile() { }
   // some methods to allow steering from the outside
-  virtual Bool_t ProcessEvent(Long64_t entry);
+  virtual Bool_t ProcessEvent();
   virtual void   FinishRun();
   virtual void   ConnectTree(TTree* tree, TObject* data);
   virtual void   ConnectAOD(AliAODEvent* aod);
index 548c7e0..318f4d5 100644 (file)
@@ -68,7 +68,7 @@ void AliJetKineReader::OpenInputFiles()
 
 //____________________________________________________________________________
 
-Bool_t AliJetKineReader::FillMomentumArray(Int_t event)
+Bool_t AliJetKineReader::FillMomentumArray()
 {
   // Fill momentum array for event
     Int_t goodTrack = 0;
@@ -156,7 +156,7 @@ Bool_t AliJetKineReader::FillMomentumArray(Int_t event)
 // set the signal flags
     fSignalFlag.Set(goodTrack, sflag);
     fCutFlag.Set(goodTrack, cflag);
-    AliInfo(Form("\n In event %d, number of good tracks %d \n", event, goodTrack));
+    AliInfo(Form("\n Number of good tracks %d \n", goodTrack));
     
     delete[] sflag;
     delete[] cflag;
index aa0e028..d9501cc 100644 (file)
@@ -22,7 +22,7 @@ class AliJetKineReader : public AliJetReader
   AliJetKineReader();
   virtual ~AliJetKineReader();
   // Setters
-  Bool_t  FillMomentumArray(Int_t event);
+  Bool_t  FillMomentumArray();
   void    OpenInputFiles();
   void    SetInputEvent(TObject* esd, TObject* aod, TObject* mc);
   // Fast Simulation
index a104c34..4758544 100755 (executable)
@@ -48,7 +48,7 @@ class AliJetReader : public TObject
   virtual Int_t               GetNumCandidateCut()   const {return fNumCandidateCut;}
   
   // Setters
-  virtual Bool_t FillMomentumArray(Int_t) {return kTRUE;}
+  virtual Bool_t FillMomentumArray() {return kTRUE;}
   virtual void   FillUnitArrayFromTPCTracks(Int_t) {}     // temporarily not used
   virtual void   FillUnitArrayFromEMCALHits() {}          // temporarily not used
   virtual void   FillUnitArrayFromEMCALDigits(Int_t) {}   // temporarily not used