**************************************************************************/
/* $Id$ */
-
+#include <Riostream.h>
#include <TROOT.h>
#include <TSystem.h>
#include <TInterpreter.h>
#include <TChain.h>
#include <TFile.h>
#include <TList.h>
+#include <TH1.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 "AliAODExtension.h"
#include "AliMCEventHandler.h"
#include "AliESDInputHandler.h"
#include "AliMCEvent.h"
////////////////////////////////////////////////////////////////////////
AliAnalysisTaskJets::AliAnalysisTaskJets():
- AliAnalysisTaskSE(),
- fConfigFile("ConfigJetAnalysis.C"),
- fNonStdBranch(""),
+ AliAnalysisTaskSE(),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fNonStdFile(""),
fJetFinder(0x0),
fHistos(0x0),
- fListOfHistos(0x0)
+ fAODExtension(0x0),
+ fListOfHistos(0x0),
+ fChain(0x0),
+ fOpt(0),
+ fReadAODFromOutput(0),
+ fUseAODBackground(kFALSE),
+ fFilterPt(0.)
{
// Default constructor
}
AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
- AliAnalysisTaskSE(name),
- fConfigFile("ConfigJetAnalysis.C"),
- fNonStdBranch(""),
- fJetFinder(0x0),
- fHistos(0x0),
- fListOfHistos(0x0)
+ AliAnalysisTaskSE(name),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fNonStdFile(""),
+ fJetFinder(0x0),
+ fHistos(0x0),
+ fAODExtension(0x0),
+ fListOfHistos(0x0),
+ fChain(0x0),
+ fOpt(0),
+ fReadAODFromOutput(0),
+ fUseAODBackground(kFALSE),
+ fFilterPt(0.)
+{
+ // Default constructor
+ DefineOutput(1, TList::Class());
+}
+
+AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
+ AliAnalysisTaskSE(name),
+ fConfigFile("ConfigJetAnalysis.C"),
+ fNonStdBranch(""),
+ fNonStdFile(""),
+ fJetFinder(0x0),
+ fHistos(0x0),
+ fAODExtension(0x0),
+ fListOfHistos(0x0),
+ fChain(chain),
+ fOpt(0),
+ fReadAODFromOutput(0),
+ fUseAODBackground(kFALSE),
+ fFilterPt(0.)
{
// Default constructor
- DefineOutput(1, TList::Class());
+ DefineOutput(1, TList::Class());
}
+//----------------------------------------------------------------
void AliAnalysisTaskJets::UserCreateOutputObjects()
{
-// Create the output container
-//
- if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
+ // Create the output container
+ //
+ if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
+ AliJetHeader *fH = fJetFinder->GetHeader();
- if(fNonStdBranch.Length()==0){
+ if(fNonStdBranch.Length()==0)
+ {
// Connec default AOD to jet finder
+ // create a new branch for the background
+
+ if(fUseAODBackground){
+ if(!AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())){
+ AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
+ evBkg->SetName(AliAODJetEventBackground::StdBranchName());
+ AddAODBranch("AliAODJetEventBackground",&evBkg);
+ }
+ }
fJetFinder->ConnectAOD(AODEvent());
}
- else{
+ else
+ {
// Create a new branch for jets...
- // how is this is reset cleared in the UserExec....
+ // how is this reset? -> cleared in the UserExec....
// Can this be handled by the framework?
+ // here we can also have the case that the brnaches are written to a separate file
+
TClonesArray *tca = new TClonesArray("AliAODJet", 0);
- tca->SetName(fNonStdBranch);
- AddAODBranch("TClonesArray",&tca);
- fJetFinder->ConnectAODNonStd(AODEvent(),fNonStdBranch.Data());
+ tca->SetName(fNonStdBranch.Data());
+ AddAODBranch("TClonesArray",&tca,fNonStdFile.Data());
+ if(fUseAODBackground){
+ if(!AODEvent() || !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,fNonStdFile.Data());
+ }
+ }
+ if(fNonStdFile.Length()!=0){
+ //
+ // case that we have an AOD extension we need to fetch the jets from the extended output
+ // we identifay the extension aod event by looking for the branchname
+ AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
+ fAODExtension = 0;
+ if(aodH){
+ TObjArray* extArray = aodH->GetExtensions();
+ if (extArray) {
+ TIter next(extArray);
+ while ((fAODExtension=(AliAODExtension*)next())){
+ TObject *obj = fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data());
+ if(fDebug>10){
+ Printf("%s:%d Dumping..",(char*)__FILE__,__LINE__);
+ fAODExtension->GetAOD()->Dump();
+ }
+ if(obj){
+ if(fDebug>1)Printf("AODExtension found for %s",fNonStdBranch.Data());
+ break;
+ }
+ fAODExtension = 0;
+ }
+ }
+ }
+ if(fAODExtension)fJetFinder->ConnectAODNonStd(fAODExtension->GetAOD(), fNonStdBranch.Data());
+ }
+ else{
+ if (fDebug > 1) printf("Connecting Non Std Branch AOD %p %s \n",AODEvent(),fNonStdBranch.Data());
+ fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data());
+ }
+ }
+
+ // do not add the histograms in the directory
+ // all handled by the list
+ Bool_t oldStatus = TH1::AddDirectoryStatus();
+ TH1::AddDirectory(kFALSE);
+
+
+ // Histograms
+ fListOfHistos = new TList();
+ fListOfHistos->SetOwner();
+ fHistos = new AliJetHistos();
+ fHistos->AddHistosToList(fListOfHistos);
+
+ // Add the JetFinderInformation to the Outputlist
+
+ // Compose a characteristic output name
+ // with the name of the output branch
+ if(fH) {
+ if(fNonStdBranch.Length()==0) {
+ fH->SetName("AliJetHeader_jets");
+ }
+ else {
+ fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
}
- // Histograms
- OpenFile(1);
- fListOfHistos = new TList();
- fHistos = new AliJetHistos();
- fHistos->AddHistosToList(fListOfHistos);
-
+ }
+
+ TH1::AddDirectory(oldStatus);
+
+
+ if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH);
+ else fAODExtension->GetTree()->GetUserInfo()->Add(fH);
+
+ // post
+ PostData(1, fListOfHistos);
+
}
+//----------------------------------------------------------------
void AliAnalysisTaskJets::Init()
{
- // Initialization
- if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
-
- // Call configuration file
+ // Initialization
+ if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
+
+ // Call configuration file
+ if (fConfigFile.Length()) {
gROOT->LoadMacro(fConfigFile);
fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
- // Initialise Jet Analysis
- fJetFinder->Init();
- // Write header information to local file
- fJetFinder->WriteHeaders();
-}
-
-
+ }
+ AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
+ fOpt = header->GetDetector();
+// AODB path for Jetan Analysis... if not alread defined, use the standard one
+if ((!(((AliJetReader*) fJetFinder->GetReader())->GetJetanOADBPath()).Length())) {
+((AliJetReader*) fJetFinder->GetReader())->SetJetanOADBPath(AliAnalysisManager::GetOADBPath());
+}
+else Info( "Init"," OADBPath for Jetan was already defined (config file?) as %s",(char*)((fJetFinder->GetReader()->GetJetanOADBPath()).Data()));
+
+ // Initialise Jet Analysis
+ if(fOpt == 0) fJetFinder->Init();
+ else fJetFinder->InitTask(fChain); // V2
+}
+//----------------------------------------------------------------
void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
{
// Execute analysis for current event
//
-
-
// Fill control histos
TClonesArray* jarray = 0;
- if(fNonStdBranch.Length()==0){
+ AliAODJetEventBackground* evBkg = 0;
+
+ // only need this once
+ static AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
+
+ if(fNonStdBranch.Length()==0) {
jarray = AODEvent()->GetJets();
+ if(fUseAODBackground){
+ 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
+ else {
+ if(AODEvent())jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data()));
+ if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()));
+ if(jarray)jarray->Delete(); // this is our responsibility, clear before filling again
+ if(fUseAODBackground){
+ if(AODEvent())evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
+ if(!evBkg) evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
+ if(evBkg)evBkg->Reset();
+ }
+ }
+
+ if (dynamic_cast<AliAODEvent*>(InputEvent()) != 0 && !fReadAODFromOutput) {
+// AOD is input event..........................................V
+ fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
+ } else {
+// AOD is read from output ....................................V
+ fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
}
+
+
+
+ if(fOpt==0) fJetFinder->ProcessEvent();
+ else fJetFinder->ProcessEvent2(); // V2
+
+ // Fill control histos
+ if(jarray)fHistos->FillHistos(jarray);
- fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
- fJetFinder->ProcessEvent();
- fHistos->FillHistos(jarray);
+ if(jarray&&aodH&&fFilterPt>0){
+ if(jarray->GetEntries()>0){
+ AliAODJet *jet = (AliAODJet*)jarray->At(0);
+ if(jet->Pt()>fFilterPt){
+ // aodH->EnableFillAODforEvent();
+ aodH->SetFillAOD(kTRUE);
+ }
+ }
+ }
+
// Post the data
PostData(1, fListOfHistos);
+ return;
}
+
+//*************************************************************
+
void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
{
// Terminate analysis