X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=JETAN%2FAliAnalysisTaskJets.cxx;h=d92731c9b3116cc9dc39ccdada52301bededf48b;hb=b3849ff32a9559a0639dd295e964d37c857f3692;hp=d185bb47878078bdb3f3e5e5b67c33d995745b3a;hpb=1f234aced66aca9ce49046e902a145991b116c97;p=u%2Fmrichter%2FAliRoot.git diff --git a/JETAN/AliAnalysisTaskJets.cxx b/JETAN/AliAnalysisTaskJets.cxx index d185bb47878..d92731c9b31 100644 --- a/JETAN/AliAnalysisTaskJets.cxx +++ b/JETAN/AliAnalysisTaskJets.cxx @@ -14,22 +14,28 @@ **************************************************************************/ /* $Id$ */ - +#include #include #include #include #include #include #include +#include #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" @@ -41,74 +47,260 @@ ClassImp(AliAnalysisTaskJets) //////////////////////////////////////////////////////////////////////// AliAnalysisTaskJets::AliAnalysisTaskJets(): - AliAnalysisTaskSE(), - fJetFinder(0x0), - fTree(0x0), - fHistos(0x0), - fListOfHistos(0x0) + AliAnalysisTaskSE(), + 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 } AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name): - AliAnalysisTaskSE(name), - fJetFinder(0x0), - fTree(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"); -// Connec default AOD to jet finder + // Create the output container + // + if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n"); + + AliJetHeader *fH = fJetFinder->GetHeader(); + + 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 + { + // Create a new branch for jets... + // 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.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(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())); + } + } + + TH1::AddDirectory(oldStatus); + + + if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH); + else fAODExtension->GetTree()->GetUserInfo()->Add(fH); + + // post + PostData(1, fListOfHistos); - fJetFinder->ConnectAOD(AODEvent()); -// -// Histograms - OpenFile(1); - fListOfHistos = new TList(); - fHistos = new AliJetHistos(); - fHistos->AddHistosToList(fListOfHistos); } +//---------------------------------------------------------------- void AliAnalysisTaskJets::Init() { - // Initialization - if (fDebug > 1) printf("AnalysisTaskJets::Init() \n"); - - // Call configuration file - gROOT->LoadMacro("ConfigJetAnalysis.C"); + // 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 -// - - fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent()); - fJetFinder->ProcessEvent(); - - // Fill control histos - fHistos->FillHistos(AODEvent()->GetJets()); - // Post the data - PostData(1, fListOfHistos); + // Execute analysis for current event + // + // Fill control histos + TClonesArray* jarray = 0; + AliAODJetEventBackground* evBkg = 0; + + // only need this once + static AliAODHandler *aodH = dynamic_cast(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()); + + if(fNonStdBranch.Length()==0) { + jarray = AODEvent()->GetJets(); + if(fUseAODBackground){ + evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())); + evBkg->Reset(); + } + } + 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(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); + + + 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