PWG/CaloTrackCorrBase/AliAnalysisTaskCaloTrackCorrelationM.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 "AliAnalysisTaskCaloTrackCorrelationM.h"
  37 #include "AliAnaCaloTrackCorrMaker.h"
  38 #include "AliCaloTrackReader.h"
  39 #include "AliPDG.h"
  40 #include "AliAnalysisManager.h"
  41 #include "AliMultiEventInputHandler.h"
  42 #include "AliMixedEvent.h"
  43 #include "AliAODEvent.h"
  44 #include "AliESDEvent.h"
  45 #include "AliAnalysisDataSlot.h"
  46
  47 ClassImp(AliAnalysisTaskCaloTrackCorrelationM)
  48
  49 //__________________________________________________________________________
  50 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM() :
  51   AliAnalysisTaskME(),
  52   fAna(0x0),
  53   fOutputContainer(0x0),
  54   fConfigName(""),
  55   fCuts(0x0),
  56   fInputEvent(NULL)
  57 {
  58   // Default constructor
  59 }
  60
  61 //__________________________________________________________________________________________
  62 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM(const char* name) :
  63   AliAnalysisTaskME(name),
  64   fAna(0x0),
  65   fOutputContainer(0x0),
  66   fConfigName(""),
  67   fCuts(0x0),
  68   fInputEvent(NULL)
  69 {
  70   // Default constructor
  71
  72   DefineOutput(1, TList::Class());
  73   DefineOutput(2, TList::Class());  // will contain cuts or local params
  74 }
  75
  76 //___________________________________________________________________________
  77 AliAnalysisTaskCaloTrackCorrelationM::~AliAnalysisTaskCaloTrackCorrelationM()
  78 {
  79   // Remove all pointers
  80
  81   //  // Do not delete it here, already done somewhere else, need to understand where.
  82   //  if(fOutputContainer && !AliAnalysisManager::GetAnalysisManager()->IsProofMode()){
  83   //    fOutputContainer->Clear() ;
  84   //    delete fOutputContainer ;
  85   //  }
  86
  87   delete fInputEvent ;
  88   if(fAna) delete fAna;
  89
  90 }
  91
  92 //__________________________________________________________________
  93 void AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects()
  94 {
  95   // Create the output container
  96   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - Begin\n");
  97
  98   //Get list of aod arrays, add each aod array to analysis frame
  99   TClonesArray *array = 0;
 100   TList * list = fAna->FillAndGetAODBranchList();
 101   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
 102
 103   //Put the delta AODs in output file, std or delta
 104   if((fAna->GetReader())->WriteDeltaAODToFile())
 105   {
 106     TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
 107     for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++)
 108     {
 109       array = (TClonesArray*) list->At(iaod);
 110       if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
 111       else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
 112     }
 113         }
 114
 115   //Histograms container
 116   OpenFile(1);
 117   fOutputContainer = fAna->GetOutputContainer();
 118
 119   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
 120
 121   fOutputContainer->SetOwner(kTRUE);
 122
 123   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - End\n");
 124
 125   PostData(1,fOutputContainer);
 126
 127 }
 128
 129 //____________________________________________________
 130 void AliAnalysisTaskCaloTrackCorrelationM::LocalInit()
 131 {
 132         // Local Initialization
 133
 134         //Call the Init to initialize the configuration of the analysis
 135         Init();
 136
 137 }
 138
 139 //_______________________________________________
 140 void AliAnalysisTaskCaloTrackCorrelationM::Init()
 141 {
 142   // Initialization
 143
 144   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Begin\n");
 145
 146   fInputEvent = new AliMixedEvent() ;
 147
 148   // Call configuration file if specified
 149
 150   if (fConfigName.Length())
 151   {
 152     printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
 153     gROOT->LoadMacro(fConfigName+".C");
 154     fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
 155   }
 156
 157   if(!fAna)
 158   {
 159     printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
 160     abort();
 161   }
 162
 163   // Add different generator particles to PDG Data Base
 164   // to avoid problems when reading MC generator particles
 165   AliPDG::AddParticlesToPdgDataBase();
 166
 167   //Set in the reader the name of the task in case is needed
 168   (fAna->GetReader())->SetTaskName(GetName());
 169
 170   // Initialise analysis
 171   fAna->Init();
 172
 173   //Delta AOD
 174   if((fAna->GetReader())->GetDeltaAODFileName()!="")
 175           AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
 176
 177   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - End\n");
 178
 179 }
 180
 181
 182 //_______________________________________________________________________
 183 void AliAnalysisTaskCaloTrackCorrelationM::UserExec(Option_t */*option*/)
 184 {
 185   // Execute analysis for current event
 186
 187   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Begin\n");
 188
 189   //Get the type of data, check if type is correct
 190   Int_t  datatype = fAna->GetReader()->GetDataType();
 191   if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
 192      datatype != AliCaloTrackReader::kMC){
 193     printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Wrong type of data\n");
 194     return ;
 195   }
 196
 197   Int_t nev = fInputHandler->GetBufferSize();
 198   fInputEvent->Reset();
 199
 200   for (Int_t iev = 0; iev < nev; iev++)
 201   {
 202     if (datatype == AliCaloTrackReader::kESD)
 203     {
 204       AliESDEvent* esd = dynamic_cast<AliESDEvent*>(GetEvent(iev));
 205       fInputEvent->AddEvent(esd);
 206     } else if (datatype == AliCaloTrackReader::kAOD)
 207     {
 208       AliAODEvent* aod = dynamic_cast<AliAODEvent*>(GetEvent(iev));
 209       fInputEvent->AddEvent(aod);
 210     } else
 211     {
 212       AliFatal("need to implement mixed event for MC") ;
 213     }
 214   }
 215
 216   fInputEvent->Init();
 217
 218   fAna->GetReader()->SetInputOutputMCEvent(fInputEvent, AODEvent(), MCEvent());
 219
 220   //Process event
 221   fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
 222
 223   //printf("AliAnalysisTaskCaloTrackCorrelationM::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
 224   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - End\n");
 225
 226   PostData(1, fOutputContainer);
 227
 228   AliAnalysisDataSlot *out0 = GetOutputSlot(0);
 229   if (out0 && out0->IsConnected()) PostData(0, fTreeA);
 230
 231   //gObjectTable->Print();
 232
 233 }
 234
 235 //________________________________________________________________________
 236 void AliAnalysisTaskCaloTrackCorrelationM::Terminate(Option_t */*option*/)
 237 {
 238   // Terminate analysis
 239   // Do some plots
 240
 241   // Get merged histograms from the output container
 242   // Propagate histagrams to maker
 243   fAna->Terminate((TList*)GetOutputData(1));
 244
 245   // Create cuts/param objects and publish to slot
 246         fCuts = fAna->GetListOfAnalysisCuts();
 247   fCuts ->SetOwner(kTRUE);
 248
 249         // Post Data
 250         PostData(2, fCuts);
 251
 252 }
 253