JETAN/AliAnalysisTaskJets.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17 #include <Riostream.h>
  18 #include <TROOT.h>
  19 #include <TSystem.h>
  20 #include <TInterpreter.h>
  21 #include <TChain.h>
  22 #include <TFile.h>
  23 #include <TList.h>
  24
  25 #include "AliAnalysisTaskJets.h"
  26 #include "AliAnalysisManager.h"
  27 #include "AliJetFinder.h"
  28 #include "AliJetHeader.h"
  29 #include "AliJetReader.h"
  30 #include "AliJetReaderHeader.h"
  31 #include "AliJetHistos.h"
  32 #include "AliESDEvent.h"
  33 #include "AliESD.h"
  34 #include "AliAODEvent.h"
  35 #include "AliAODJetEventBackground.h"
  36 #include "AliAODHandler.h"
  37 #include "AliMCEventHandler.h"
  38 #include "AliESDInputHandler.h"
  39 #include "AliMCEvent.h"
  40 #include "AliStack.h"
  41
  42
  43 ClassImp(AliAnalysisTaskJets)
  44
  45 ////////////////////////////////////////////////////////////////////////
  46
  47 AliAnalysisTaskJets::AliAnalysisTaskJets():
  48   AliAnalysisTaskSE(),
  49   fConfigFile("ConfigJetAnalysis.C"),
  50   fNonStdBranch(""),
  51   fNonStdFile(""),
  52   fJetFinder(0x0),
  53   fHistos(0x0),
  54   fAODExtension(0x0),
  55   fListOfHistos(0x0),
  56   fChain(0x0),
  57   fOpt(0),
  58   fReadAODFromOutput(0)
  59 {
  60   // Default constructor
  61 }
  62
  63 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
  64   AliAnalysisTaskSE(name),
  65   fConfigFile("ConfigJetAnalysis.C"),
  66   fNonStdBranch(""),
  67   fNonStdFile(""),
  68   fJetFinder(0x0),
  69   fHistos(0x0),
  70   fAODExtension(0x0),
  71   fListOfHistos(0x0),
  72   fChain(0x0),
  73   fOpt(0),
  74   fReadAODFromOutput(0)
  75 {
  76   // Default constructor
  77   DefineOutput(1, TList::Class());
  78 }
  79
  80 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
  81   AliAnalysisTaskSE(name),
  82   fConfigFile("ConfigJetAnalysis.C"),
  83   fNonStdBranch(""),
  84   fNonStdFile(""),
  85   fJetFinder(0x0),
  86   fHistos(0x0),
  87   fAODExtension(0x0),
  88   fListOfHistos(0x0),
  89   fChain(chain),
  90   fOpt(0),
  91   fReadAODFromOutput(0)
  92 {
  93   // Default constructor
  94   DefineOutput(1, TList::Class());
  95 }
  96
  97 //----------------------------------------------------------------
  98 void AliAnalysisTaskJets::UserCreateOutputObjects()
  99 {
 100   // Create the output container
 101   //
 102   if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
 103
 104   // we already have a non standard output, need not to book a new file...
 105   if (!IsStandardAOD())fNonStdFile  = "";
 106
 107   if(fNonStdBranch.Length()==0)
 108     {
 109       //  Connec default AOD to jet finder
 110       // create a new branch for the background
 111       if(!AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())){
 112         AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
 113         evBkg->SetName(AliAODJetEventBackground::StdBranchName());
 114         AddAODBranch("AliAODJetEventBackground",&evBkg);
 115       }
 116       fJetFinder->ConnectAOD(AODEvent());
 117     }
 118   else
 119     {
 120       // Create a new branch for jets...
 121       // how is this reset? -> cleared in the UserExec....
 122       // Can this be handled by the framework?
 123       // here we can also have the case that the brnaches are written to a separate file
 124
 125       TClonesArray *tca = new TClonesArray("AliAODJet", 0);
 126       tca->SetName(fNonStdBranch.Data());
 127       AddAODBranch("TClonesArray",&tca,fNonStdFile.Data());
 128       if(!AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){
 129         AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
 130         evBkg->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()));
 131         AddAODBranch("AliAODJetEventBackground",&evBkg,fNonStdFile.Data());
 132       }
 133
 134       if(fNonStdFile.Length()!=0){
 135         //
 136         // case that we have an AOD extension we need to fetch the jets from the extended output
 137         // we identifay the extension aod event by looking for the branchname
 138         AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
 139         TObjArray* extArray = aodH->GetExtensions();
 140         if (extArray) {
 141           TIter next(extArray);
 142           while ((fAODExtension=(AliAODExtension*)next())){
 143             TObject *obj = fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data());
 144             if(fDebug>10){
 145               Printf("%s:%d Dumping..",(char*)__FILE__,__LINE__);
 146               fAODExtension->GetAOD()->Dump();
 147             }
 148             if(obj){
 149               if(fDebug>1)Printf("AODExtension found for %s",fNonStdBranch.Data());
 150               break;
 151             }
 152             fAODExtension = 0;
 153           }
 154         }
 155         if(fAODExtension)fJetFinder->ConnectAODNonStd(fAODExtension->GetAOD(), fNonStdBranch.Data());
 156       }
 157       else{
 158         fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data());
 159       }
 160     }
 161
 162
 163   // Histograms
 164   OpenFile(1);
 165   fListOfHistos = new TList();
 166   fHistos       = new AliJetHistos();
 167   fHistos->AddHistosToList(fListOfHistos);
 168
 169   // Add the JetFinderInformation to the Outputlist
 170   AliJetHeader *fH = fJetFinder->GetHeader();
 171
 172   // Compose a characteristic output name
 173   // with the name of the output branch
 174   if(fH) {
 175     if(fNonStdBranch.Length()==0) {
 176       fH->SetName("AliJetHeader_jets");
 177     }
 178     else {
 179       fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
 180     }
 181   }
 182   if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH);
 183   else fAODExtension->GetTree()->GetUserInfo()->Add(fH);
 184 }
 185
 186 //----------------------------------------------------------------
 187 void AliAnalysisTaskJets::Init()
 188 {
 189   // Initialization
 190   if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
 191
 192   // Call configuration file
 193   if (fConfigFile.Length()) {
 194     gROOT->LoadMacro(fConfigFile);
 195     fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
 196   }
 197   AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
 198   fOpt = header->GetDetector();
 199
 200   // Initialise Jet Analysis
 201   if(fOpt == 0) fJetFinder->Init();
 202   else fJetFinder->InitTask(fChain); // V2
 203 }
 204
 205
 206 //----------------------------------------------------------------
 207 void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
 208 {
 209   // Execute analysis for current event
 210   //
 211   // Fill control histos
 212   TClonesArray* jarray = 0;
 213   AliAODJetEventBackground*  evBkg;
 214
 215   if(fNonStdBranch.Length()==0) {
 216     jarray = AODEvent()->GetJets();
 217     evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName()));
 218     evBkg->Reset();
 219   }
 220   else {
 221     jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data()));
 222     if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()));
 223     jarray->Delete();    // this is our responsibility, clear before filling again
 224     evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
 225     if(!evBkg)  evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
 226     evBkg->Reset();
 227   }
 228
 229   if (dynamic_cast<AliAODEvent*>(InputEvent()) !=  0 && !fReadAODFromOutput) {
 230 // AOD is input event..........................................V
 231       fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
 232   } else {
 233 // AOD is read from output ....................................V
 234       fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
 235   }
 236
 237
 238
 239   if(fOpt==0) fJetFinder->ProcessEvent();
 240   else fJetFinder->ProcessEvent2();    // V2
 241
 242   // Fill control histos
 243   fHistos->FillHistos(jarray);
 244
 245   // Post the data
 246   PostData(1, fListOfHistos);
 247   return;
 248 }
 249
 250
 251 //*************************************************************
 252
 253 void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
 254 {
 255 // Terminate analysis
 256 //
 257     if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
 258 //    if (fJetFinder) fJetFinder->FinishRun();
 259 }
 260