Clusterize all chambers when the combined tracking is disabled.

[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJets.cxx

diff --git a/JETAN/AliAnalysisTaskJets.cxx b/JETAN/AliAnalysisTaskJets.cxx
index c3228780326259feb1f3ea8e7547abb12aaeb9c9..20e7c098caf78fde9b89b4752b5323de1ec1e7b9 100644 (file)
--- a/JETAN/AliAnalysisTaskJets.cxx
+++ b/JETAN/AliAnalysisTaskJets.cxx
@@ -12,20 +12,28 @@
  * about the suitability of this software for any purpose. It is          *
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
+
+/* $Id$ */
  
 #include <TROOT.h>
 #include <TSystem.h>
 #include <TInterpreter.h>
 #include <TChain.h>
 #include <TFile.h>
-#include <TH1.h>
+#include <TList.h>
 
 #include "AliAnalysisTaskJets.h"
 #include "AliAnalysisManager.h"
 #include "AliJetFinder.h"
+#include "AliJetHistos.h"
+#include "AliESDEvent.h"
 #include "AliESD.h"
 #include "AliAODEvent.h"
 #include "AliAODHandler.h"
+#include "AliMCEventHandler.h"
+#include "AliESDInputHandler.h"
+#include "AliMCEvent.h"
+#include "AliStack.h"
 
 
 ClassImp(AliAnalysisTaskJets)
@@ -35,10 +43,12 @@ ClassImp(AliAnalysisTaskJets)
 AliAnalysisTaskJets::AliAnalysisTaskJets():
     fDebug(0),
     fJetFinder(0x0),
-    fChain(0x0),
+    fTree(0x0),
     fESD(0x0),
     fAOD(0x0),
-    fTreeA(0x0)
+    fTreeA(0x0),
+    fHistos(0x0),
+    fListOfHistos(0x0)
 {
   // Default constructor
 }
@@ -47,25 +57,36 @@ AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
     AliAnalysisTask(name, "AnalysisTaskJets"),
     fDebug(0),
     fJetFinder(0x0),
-    fChain(0x0),
+    fTree(0x0),
     fESD(0x0),
     fAOD(0x0),
-    fTreeA(0x0)
+    fTreeA(0x0),
+    fHistos(0x0),
+    fListOfHistos(0x0)
 {
   // Default constructor
     DefineInput (0, TChain::Class());
     DefineOutput(0, TTree::Class());
+    DefineOutput(1, TList::Class());
 }
 
 void AliAnalysisTaskJets::CreateOutputObjects()
 {
 // Create the output container
-    OpenFile(0);
-    AliAODHandler* handler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetEventHandler());
+//
+//  Default AOD
+    if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
+    AliAODHandler* handler = (AliAODHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
     
     fAOD   = handler->GetAOD();
     fTreeA = handler->GetTree();
     fJetFinder->ConnectAOD(fAOD);
+//
+//  Histograms
+    OpenFile(1);
+    fListOfHistos = new TList();
+    fHistos       = new AliJetHistos();
+    fHistos->AddHistosToList(fListOfHistos);
 }
 
 void AliAnalysisTaskJets::Init()
@@ -85,33 +106,40 @@ void AliAnalysisTaskJets::Init()
 void AliAnalysisTaskJets::ConnectInputData(Option_t */*option*/)
 {
 // Connect the input data
-//
     if (fDebug > 1) printf("AnalysisTaskJets::ConnectInputData() \n");
-    fChain = (TChain*)GetInputData(0);
+    AliESDInputHandler* esdH = (AliESDInputHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
+    fESD = esdH->GetEvent();
+    fTree = esdH->GetTree();
+    
+    
 
-    char ** address = (char **)GetBranchAddress(0, "ESD");
-    if (address)     {
-       
-// Branch has been already connected
-       fESD = (AliESD*)(*address);
-    }
-    else     {
-// First task taking the branch enables it
-       fESD = new AliESD();
-       SetBranchAddress(0, "ESD", &fESD);
-    }
+    AliMCEventHandler*    mcTruth = (AliMCEventHandler*) 
+       ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler());
     
-    fJetFinder->ConnectTree(fChain, fESD);
+    fJetFinder->GetReader()->SetInputEvent(fESD, fAOD, mcTruth);
 }
 
 void AliAnalysisTaskJets::Exec(Option_t */*option*/)
 {
 // Execute analysis for current event
 //
-    Long64_t ientry = fChain->GetReadEntry();
+    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);
+
+    // Fill control histos
+    fHistos->FillHistos(fAOD->GetJets());
+    
+    // Post the data
     PostData(0, fTreeA);
+    PostData(1, fListOfHistos);
 }
 
 void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
@@ -119,6 +147,6 @@ void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
 // Terminate analysis
 //
     if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
-    // if (fJetFinder) fJetFinder->FinishRun();
+//    if (fJetFinder) fJetFinder->FinishRun();
 }