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