PWG4/PartCorrBase/AliCaloTrackAODReader.cxx

   1
   2 /**************************************************************************
   3  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  *                                                                        *
   5  * Author: The ALICE Off-line Project.                                    *
   6  * Contributors are mentioned in the code where appropriate.              *
   7  *                                                                        *
   8  * Permission to use, copy, modify and distribute this software and its   *
   9  * documentation strictly for non-commercial purposes is hereby granted   *
  10  * without fee, provided that the above copyright notice appears in all   *
  11  * copies and that both the copyright notice and this permission notice   *
  12  * appear in the supporting documentation. The authors make no claims     *
  13  * about the suitability of this software for any purpose. It is          *
  14  * provided "as is" without express or implied warranty.                  *
  15  **************************************************************************/
  16 /* $Id: $ */
  17
  18 //_________________________________________________________________________
  19 // Class for reading data (AODs) in order to do prompt gamma
  20 //  or other particle identification and correlations.
  21 // This part is commented: Mixing analysis can be done, input AOD with events
  22 // is opened in the AliCaloTrackReader::Init()
  23 //
  24 //
  25 //*-- Author: Gustavo Conesa (LNF-INFN)
  26 //////////////////////////////////////////////////////////////////////////////
  27
  28
  29 // --- ROOT system ---
  30 //#include "Riostream.h"
  31
  32 //---- ANALYSIS system ----
  33 #include "AliCaloTrackAODReader.h"
  34 #include "AliAODInputHandler.h"
  35 #include "AliMultiEventInputHandler.h"
  36 #include "AliAnalysisManager.h"
  37 #include "AliMixedEvent.h"
  38
  39 ClassImp(AliCaloTrackAODReader)
  40
  41 //____________________________________________________________________________
  42 AliCaloTrackAODReader::AliCaloTrackAODReader() :
  43   AliCaloTrackReader()
  44 {
  45   //Default Ctor
  46
  47   //Initialize parameters
  48   fDataType=kAOD;
  49   fReadStack          = kTRUE;
  50   fReadAODMCParticles = kFALSE;
  51
  52 }
  53
  54 //____________________________________________________________________________
  55 //void AliCaloTrackAODReader::GetSecondInputAODVertex(Double_t  v[3]) const {
  56 //      //Return vertex position of second AOD input
  57 //
  58 //      fSecondInputAODEvent->GetPrimaryVertex()->GetXYZ(v);
  59 //
  60 //}
  61
  62 //____________________________________________________________________________
  63 Double_t AliCaloTrackAODReader::GetBField() const {
  64   //Return magnetic field
  65
  66   Double_t bfield = fInputEvent->GetMagneticField();
  67
  68   return bfield;
  69
  70 }
  71
  72 //____________________________________________________________________________
  73 void AliCaloTrackAODReader::SetInputOutputMCEvent(AliVEvent* input, AliAODEvent* aod, AliMCEvent* mc) {
  74   // Connect the data pointers
  75   // If input is AOD, do analysis with input, if not, do analysis with the output aod.
  76
  77           //printf("AODInputHandler %p, MergeEvents %d \n",aodIH, aodIH->GetMergeEvents());
  78
  79   Bool_t tesd = kFALSE ;
  80   Bool_t taod = kTRUE ;
  81   if ( strcmp(input->GetName(), "AliMixedEvent") == 0 ) {
  82     AliMultiEventInputHandler* multiEH = dynamic_cast<AliMultiEventInputHandler*>((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
  83     if (multiEH->GetFormat() == 0 ) {
  84       tesd = kTRUE ;
  85     } else if (multiEH->GetFormat() == 1) {
  86       taod = kTRUE ;
  87     }
  88   }
  89   if (strcmp(input->GetName(),"AliESDEvent") == 0) {
  90     tesd = kTRUE ;
  91   } else if (strcmp(input->GetName(),"AliAODEvent") == 0) {
  92     taod = kTRUE ;
  93   }
  94
  95
  96   if(tesd)   {
  97     SetInputEvent(aod);
  98     SetOutputEvent(aod);
  99   }
 100   else if(taod){
 101     AliAODInputHandler* aodIH = dynamic_cast<AliAODInputHandler*>((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
 102           if (aodIH && aodIH->GetMergeEvents()) {
 103                   //Merged events, use output AOD.
 104                   SetInputEvent(aod);
 105                   SetOutputEvent(aod);
 106           }
 107           else{
 108                   SetInputEvent(input);
 109                   SetOutputEvent(aod);
 110           }
 111   }
 112   else{
 113     AliFatal(Form("AliCaloTrackAODReader::SetInputOutputMCEvent() - STOP : Wrong data format: %s\n",input->GetName()));
 114   }
 115
 116   SetMC(mc);
 117
 118 }
 119