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