AliCaloTrackMCReader.h

Go to the documentation of this file.

#ifndef ALICALOTRACKMCREADER_H
#define ALICALOTRACKMCREADER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

//_________________________________________________________________________
//_________________________________________________________________________

// --- ROOT system ---
class TArrayI   ;

// --- AliRoot system ---
#include "AliCaloTrackReader.h" 
class AliVCluster ;
class AliAODTrack ;
class AliAODEvent ;
#include "AliMCEvent.h"
class AliVEvent   ;

class AliCaloTrackMCReader : public AliCaloTrackReader {
  
 public: 
  
  AliCaloTrackMCReader() ;          // ctor
  virtual ~AliCaloTrackMCReader() ; // virtual dtor

  // Main methods in source file
  
  void   CheckOverlap(Float_t anglethres, Int_t imom, Int_t & iPrimary, Int_t & index, Int_t & pdg);

  void   FillCalorimeters(Int_t & iParticle, Int_t motherIndex, Int_t pdg) ;

  Bool_t FillInputEvent(Int_t iEntry, const char * currentFileName) ;
  
  void   InitParameters();
  
  void   MakePi0Decay() ;//, Double_t &angle);

  void   Print(const Option_t * opt) const; 
  
  void   SetCaloClusterPID   (Int_t pdgCode, AliVCluster *calo ) const ;
  
  void   SetTrackChargeAndPID(Int_t pdgCode, AliAODTrack *track) const ;
  
  void   SetInputOutputMCEvent(AliVEvent* esd, AliAODEvent* aod, AliMCEvent* mc) ;

  // Data members setters and getters
  
  AliVEvent*  GetInputEvent()               const { return (AliVEvent *) GetMC()   ; }
  AliGenEventHeader* GetGenEventHeader()    const 
  {
    if ( GetMC() ) return GetMC()->GenEventHeader();
    else           return 0x0;
  }

  void      GetVertex(Double_t v[3]) const ;
  Double_t* GetVertex(Int_t evtIndex) const {return fVertex[evtIndex];}
  void      GetVertex(Double_t vertex[3], Int_t evtIndex) const 
  { vertex[0]=fVertex[evtIndex][0];  vertex[1]=fVertex[evtIndex][1];  vertex[2]=fVertex[evtIndex][2]; }  
  
  // Particle type, status, arrays 
  void      AddNeutralParticlesArray(TArrayI & array)  
      { fNeutralParticlesArray   = new TArrayI(array) ; }
  TArrayI * GetNeutralParticlesArray()      const { return  fNeutralParticlesArray ; }
  Bool_t    SkipNeutralParticles(Int_t pdg) const ;
  
  void      AddChargedParticlesArray(TArrayI & array)  
      { fChargedParticlesArray   = new TArrayI(array) ; }
  TArrayI * GetChargedParticlesArray()      const { return  fChargedParticlesArray ; }
  Bool_t    KeepChargedParticles(Int_t pdg) const ;

  void      AddStatusArray(TArrayI & array)  
      { fStatusArray   = new TArrayI(array) ; }
  TArrayI * GetStatusArray()                const { return  fStatusArray           ; }
  
  void      SwitchOnStatusSelection()             { fKeepAllStatus = kFALSE        ; }
  void      SwitchOffStatusSelection()            { fKeepAllStatus = kTRUE         ; }
  Bool_t    KeepParticleWithStatus(Int_t status) const ;
  
  void      SwitchOnOnlyGeneratorParticles()      { fOnlyGeneratorParticles = kTRUE  ; }
  void      SwitchOffOnlyGeneratorParticles()     { fOnlyGeneratorParticles = kFALSE ; }
  
  // Pi0 Overlapps, decays 
  
  void      SwitchOnPi0Decay()                    { fDecayPi0 = kTRUE              ; } 
  void      SwitchOffPi0Decay()                   { fDecayPi0 = kFALSE             ; } 
  Int_t     IsPi0DecaySwitchedOn()          const { return fDecayPi0               ; }   
  
  void      SwitchOnOverlapCheck()                { fCheckOverlap = kTRUE          ; }
  void      SwitchOffOverlapCheck()               { fCheckOverlap = kFALSE         ; }

  Float_t   GetEMCALOverlapAngle()          const { return fEMCALOverlapAngle      ; }
  void      SetEMCALOverlapAngle(Float_t angle)   { fEMCALOverlapAngle = angle     ; }
  
  Float_t   GetPHOSOverlapAngle()           const { return fPHOSOverlapAngle       ; }
  void      SetPHOSOverlapAngle (Float_t angle)   { fPHOSOverlapAngle  = angle     ; }
    
private:

  Bool_t    fDecayPi0 ;              
  TArrayI * fNeutralParticlesArray ; 
  TArrayI * fChargedParticlesArray ; 
  TArrayI * fStatusArray ;           
  Bool_t    fKeepAllStatus ;         
  Bool_t    fCheckOverlap;           
  Float_t   fEMCALOverlapAngle;      
  Float_t   fPHOSOverlapAngle;       
  Int_t     fIndex2ndPhoton;         
  Bool_t    fOnlyGeneratorParticles; 
  
  TLorentzVector fMomentum;          
  TLorentzVector fPi0Momentum;       
  TLorentzVector fGamDecayMom1;      
  TLorentzVector fGamDecayMom2;      
  
  AliCaloTrackMCReader(              const AliCaloTrackMCReader & r) ; 
  
  AliCaloTrackMCReader & operator = (const AliCaloTrackMCReader & r) ;
  
  ClassDef(AliCaloTrackMCReader,5) ; 
  
} ;

#endif //ALICALOTRACKMCREADER_H