PWG/CaloTrackCorrBase/AliAnalysisTaskCaloTrackCorrelation.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 //_________________________________________________________________________
  17 // Analysis task that executes the analysis classes
  18 // that depend on the CaloTrackCorr frame, frame for Particle identification
  19 // with calorimeters and tracks and correlations.
  20 // Specially designed for calorimeters but also can be used for charged tracks
  21 // Input of this task is a configuration file that contains all the settings
  22 // of the analysis
  23 //
  24 // -- Author: Gustavo Conesa (INFN-LNF, LPSC-Grenoble)
  25
  26 #include <cstdlib>
  27
  28 // --- Root ---
  29 #include <TROOT.h>
  30 #include <TInterpreter.h>
  31 #include <TClonesArray.h>
  32 //#include <Riostream.h>
  33 //#include <TObjectTable.h>
  34
  35 // --- Analysis ---
  36 #include "AliAnalysisTaskCaloTrackCorrelation.h"
  37 #include "AliAnaCaloTrackCorrMaker.h"
  38 #include "AliCaloTrackReader.h"
  39 #include "AliPDG.h"
  40 #include "AliAnalysisManager.h"
  41 #include "AliInputEventHandler.h"
  42
  43 ClassImp(AliAnalysisTaskCaloTrackCorrelation)
  44
  45 //________________________________________________________________________
  46 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation() :
  47   AliAnalysisTaskSE(),
  48   fAna(0x0),
  49   fOutputContainer(0x0),
  50   fConfigName(""),
  51   fCuts(0x0)
  52 {
  53   // Default constructor
  54 }
  55
  56 //________________________________________________________________________________________
  57 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation(const char* name) :
  58   AliAnalysisTaskSE(name),
  59   fAna(0x0),
  60   fOutputContainer(0x0),
  61   fConfigName(""),
  62   fCuts(0x0)
  63 {
  64   // Default constructor
  65
  66   DefineOutput(1, TList::Class());
  67   DefineOutput(2, TList::Class());  // will contain cuts or local params
  68 }
  69
  70 //_________________________________________________________________________
  71 AliAnalysisTaskCaloTrackCorrelation::~AliAnalysisTaskCaloTrackCorrelation()
  72 {
  73   // Remove all pointers
  74
  75   if (AliAnalysisManager::GetAnalysisManager()->IsProofMode()) return;
  76
  77   if (fOutputContainer)
  78   {
  79     fOutputContainer->Clear() ;
  80     delete fOutputContainer ;
  81   }
  82
  83   if (fAna) delete fAna;
  84
  85 }
  86
  87 //_________________________________________________________________
  88 void AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects()
  89 {
  90   // Create the output container
  91   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - Begin\n");
  92
  93   //Get list of aod arrays, add each aod array to analysis frame
  94   TList * list = fAna->FillAndGetAODBranchList(); //Loop the analysis and create the list of branches
  95
  96   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
  97
  98   //Put the delta AODs in output file, std or delta
  99   if((fAna->GetReader())->WriteDeltaAODToFile())
 100   {
 101     TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
 102     for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++)
 103     {
 104       TClonesArray * array = (TClonesArray*) list->At(iaod);
 105       if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
 106       else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
 107     }
 108   }
 109
 110   //Histograms container
 111   OpenFile(1);
 112   fOutputContainer = fAna->GetOutputContainer();
 113
 114   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
 115
 116   fOutputContainer->SetOwner(kTRUE);
 117
 118   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - End\n");
 119
 120   PostData(1,fOutputContainer);
 121
 122 }
 123
 124 //___________________________________________________
 125 void AliAnalysisTaskCaloTrackCorrelation::LocalInit()
 126 {
 127   // Local Initialization
 128
 129   //Call the Init to initialize the configuration of the analysis
 130   Init();
 131
 132 }
 133
 134 //______________________________________________
 135 void AliAnalysisTaskCaloTrackCorrelation::Init()
 136 {
 137   // Initialization
 138
 139   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Begin\n");
 140
 141   // Call configuration file if specified
 142
 143   if (fConfigName.Length())
 144   {
 145     printf("AliAnalysisTaskCaloTrackCorrelation::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
 146     gROOT->LoadMacro(fConfigName+".C");
 147     fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
 148   }
 149
 150   if(!fAna)
 151   {
 152     printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
 153     abort();
 154   }
 155
 156   // Add different generator particles to PDG Data Base
 157   // to avoid problems when reading MC generator particles
 158   AliPDG::AddParticlesToPdgDataBase();
 159
 160   //Set in the reader the name of the task in case is needed
 161   (fAna->GetReader())->SetTaskName(GetName());
 162
 163   // Initialise analysis
 164   fAna->Init();
 165
 166   //Delta AOD
 167   if((fAna->GetReader())->GetDeltaAODFileName()!="")
 168     AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
 169
 170   // Selected Trigger
 171   if(fAna->GetReader()->IsEventTriggerAtSEOn()) fAna->GetReader()->SetEventTriggerMask(GetCollisionCandidates());
 172
 173   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - End\n");
 174
 175 }
 176
 177
 178 //______________________________________________________________________
 179 void AliAnalysisTaskCaloTrackCorrelation::UserExec(Option_t */*option*/)
 180 {
 181   // Execute analysis for current event
 182
 183   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Begin\n");
 184
 185   //Get the type of data, check if type is correct
 186   Int_t  datatype = fAna->GetReader()->GetDataType();
 187   if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
 188      datatype != AliCaloTrackReader::kMC)
 189   {
 190     printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Wrong type of data\n");
 191     return ;
 192   }
 193
 194   fAna->GetReader()->SetInputOutputMCEvent(InputEvent(), AODEvent(), MCEvent());
 195
 196   //Process event
 197   fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
 198
 199   //printf("AliAnalysisTaskCaloTrackCorrelation::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
 200   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - End\n");
 201
 202   PostData(1, fOutputContainer);
 203
 204   //gObjectTable->Print();
 205
 206 }
 207
 208 //_______________________________________________________________________
 209 void AliAnalysisTaskCaloTrackCorrelation::Terminate(Option_t */*option*/)
 210 {
 211   // Terminate analysis
 212   // Do some plots
 213
 214   // Get merged histograms from the output container
 215   // Propagate histagrams to maker
 216   fAna->Terminate((TList*)GetOutputData(1));
 217
 218   // Create cuts/param objects and publish to slot
 219   fCuts = fAna->GetListOfAnalysisCuts();
 220   fCuts ->SetOwner(kTRUE);
 221   // Post Data
 222   PostData(2, fCuts);
 223
 224 }
 225
 226 //__________________________________________________________
 227 void AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput()
 228 {
 229   // Put in the output some event summary histograms
 230   AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
 231
 232   AliInputEventHandler *inputH = dynamic_cast<AliInputEventHandler*>(am->GetInputEventHandler());
 233
 234   if (!inputH) return;
 235
 236   TH2F *histStat = dynamic_cast<TH2F*>(inputH->GetStatistics());
 237   TH2F *histBin0 = dynamic_cast<TH2F*>(inputH->GetStatistics("BIN0"));
 238
 239   if(histStat)fOutputContainer->Add(histStat);
 240   else if(DebugLevel()>0)
 241     printf("AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput() - Stat histogram not available check, \n if ESDs, that AliPhysicsSelection was on, \n if AODs, if EventStat_temp.root exists \n");
 242
 243   if(histBin0)fOutputContainer->Add(histBin0);
 244
 245 }
 246