* about the suitability of this software for any purpose. It is *
* provided "as is" without express or implied warranty. *
**************************************************************************/
-
+
+/* $Id$ */
+#include <Riostream.h>
#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 "AliJetHeader.h"
+#include "AliJetReader.h"
+#include "AliJetReaderHeader.h"
+#include "AliJetHistos.h"
+#include "AliESDEvent.h"
#include "AliESD.h"
+#include "AliAODEvent.h"
+#include "AliAODJetEventBackground.h"
+#include "AliAODHandler.h"
+#include "AliMCEventHandler.h"
+#include "AliESDInputHandler.h"
+#include "AliMCEvent.h"
+#include "AliStack.h"
+
ClassImp(AliAnalysisTaskJets)
////////////////////////////////////////////////////////////////////////
AliAnalysisTaskJets::AliAnalysisTaskJets():
- fDebug(0),
- fJetFinder(0x0),
- fChain(0x0),
- fESD(0x0),
- fTreeJ(0x0)
+ AliAnalysisTaskSE(),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fJetFinder(0x0),
+ fHistos(0x0),
+ fListOfHistos(0x0),
+ fChain(0x0),
+ fOpt(0)
{
// Default constructor
}
AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
- AliAnalysisTask(name, "AnalysisTaskJets"),
- fDebug(0),
- fJetFinder(0x0),
- fChain(0x0),
- fESD(0x0),
- fTreeJ(0x0)
+ AliAnalysisTaskSE(name),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fJetFinder(0x0),
+ fHistos(0x0),
+ fListOfHistos(0x0),
+ fChain(0x0),
+ fOpt(0)
{
// Default constructor
- DefineInput (0, TChain::Class());
- DefineOutput(0, TTree::Class());
+ DefineOutput(1, TList::Class());
}
-void AliAnalysisTaskJets::CreateOutputObjects()
+AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
+ AliAnalysisTaskSE(name),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fJetFinder(0x0),
+ fHistos(0x0),
+ fListOfHistos(0x0),
+ fChain(chain),
+ fOpt(0)
{
-// Create the output container
- OpenFile(0);
- fTreeJ = fJetFinder->MakeTreeJ("TreeJ");
+ // Default constructor
+ DefineOutput(1, TList::Class());
}
-void AliAnalysisTaskJets::Init()
+//----------------------------------------------------------------
+void AliAnalysisTaskJets::UserCreateOutputObjects()
{
- // Initialization
- if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
+ // Create the output container
+ //
+ if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
+
+ if(fNonStdBranch.Length()==0)
+ {
+ // Connec default AOD to jet finder
+ // create a new branch for the background
+ if(!AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())){
+ AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
+ evBkg->SetName(AliAODJetEventBackground::StdBranchName());
+ AddAODBranch("AliAODJetEventBackground",&evBkg);
+ }
+ fJetFinder->ConnectAOD(AODEvent());
+ }
+ else
+ {
+ // Create a new branch for jets...
+ // how is this reset? -> cleared in the UserExec....
+ // Can this be handled by the framework?
+ TClonesArray *tca = new TClonesArray("AliAODJet", 0);
+ tca->SetName(fNonStdBranch);
+ AddAODBranch("TClonesArray",&tca);
+ if(!AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){
+ AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
+ evBkg->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()));
+ AddAODBranch("AliAODJetEventBackground",&evBkg);
+ }
+ fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data());
+ }
- // Call configuration file
- gROOT->LoadMacro("ConfigJetAnalysis.C");
- fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
- // Initialise Jet Analysis
- fJetFinder->Init();
- // Write header information to local file
- fJetFinder->WriteHeaders();
-}
-void AliAnalysisTaskJets::ConnectInputData(Option_t */*option*/)
-{
-// Connect the input data
-//
- if (fDebug > 1) printf("AnalysisTaskJets::ConnectInputData() \n");
- fChain = (TChain*)GetInputData(0);
-
- char ** address = (char **)GetBranchAddress(0, "ESD");
- if (address) {
-
-// Branch has been already connected
- fESD = (AliESD*)(*address);
+ // Histograms
+ OpenFile(1);
+ fListOfHistos = new TList();
+ fHistos = new AliJetHistos();
+ fHistos->AddHistosToList(fListOfHistos);
+
+ // Add the JetFinderInformation to the Outputlist
+ AliJetHeader *fH = fJetFinder->GetHeader();
+
+ // Compose a characteristic output name
+ // with the name of the output branch
+ if(fH) {
+ if(fNonStdBranch.Length()==0) {
+ fH->SetName("AliJetHeader_jets");
}
- else {
-// First task taking the branch enables it
- fESD = new AliESD();
- SetBranchAddress(0, "ESD", &fESD);
+ else {
+ fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
}
-
- fJetFinder->ConnectTree(fChain, fESD);
+ }
+ OutputTree()->GetUserInfo()->Add(fH);
+}
+
+//----------------------------------------------------------------
+void AliAnalysisTaskJets::Init()
+{
+ // Initialization
+ if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
+
+ // Call configuration file
+ if (fConfigFile.Length()) {
+ gROOT->LoadMacro(fConfigFile);
+ fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
+ }
+ AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
+ fOpt = header->GetDetector();
+
+ // Initialise Jet Analysis
+ if(fOpt == 0) fJetFinder->Init();
+ else fJetFinder->InitTask(fChain); // V2
+
+ // Write header information to local file
+ fJetFinder->WriteHeaders();
}
-void AliAnalysisTaskJets::Exec(Option_t */*option*/)
+//----------------------------------------------------------------
+void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
{
-// Execute analysis for current event
-//
- Long64_t ientry = fChain->GetReadEntry();
- if (fDebug > 1) printf("Analysing event # %5d\n", (Int_t) ientry);
- fJetFinder->ProcessEvent(ientry);
- PostData(0, fTreeJ);
+ // Execute analysis for current event
+ //
+ // Fill control histos
+ TClonesArray* jarray = 0;
+ AliAODJetEventBackground* evBkg;
+
+ if(fNonStdBranch.Length()==0) {
+ jarray = AODEvent()->GetJets();
+ evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName()));
+ evBkg->Reset();
+ }
+ else {
+ jarray = dynamic_cast<TClonesArray*>(AODEvent()->FindListObject(fNonStdBranch.Data()));
+ jarray->Delete(); // this is our responsibility, clear before filling again
+ evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
+ evBkg->Reset();
+ }
+ if (dynamic_cast<AliAODEvent*>(InputEvent()) != 0) {
+ fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
+ } else {
+ fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
+ }
+
+
+
+ if(fOpt==0) fJetFinder->ProcessEvent();
+ else fJetFinder->ProcessEvent2(); // V2
+
+ // Fill control histos
+ fHistos->FillHistos(jarray);
+
+ // Post the data
+ PostData(1, fListOfHistos);
}
+
+//*************************************************************
+
void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
{
// Terminate analysis
//
if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
- // if (fJetFinder) fJetFinder->FinishRun();
+// if (fJetFinder) fJetFinder->FinishRun();
}