coverity

[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliCaloTrackMCReader.h

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

//_________________________________________________________________________
// Class for reading data (Kinematics) in order to do prompt gamma 
// or other particle  correlations
// Separates generated particles into charged (CTS) 
// and neutral (PHOS or EMCAL acceptance)
//
// Now, it only works with data stored in Kinematics.root and 
// not in filtered Kinematics branch in AODs
//
//*-- Author: Gustavo Conesa (LNF-INFN) 
//////////////////////////////////////////////////////////////////////////////

//-- Author: Gustavo Conesa (INFN-LNF)

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

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

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

  void   InitParameters();
  
  void   Print(const Option_t * opt) const; 
  
  void   SwitchOnPi0Decay()           { fDecayPi0 = kTRUE  ; } 
  void   SwitchOffPi0Decay()          { fDecayPi0 = kFALSE ; } 
  Int_t  IsPi0DecaySwitchedOn() const { return fDecayPi0   ; } 
  
  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 ; }
  
  void      GetVertex(Double_t v[3]) const ;
  Double_t* GetVertex(const Int_t evtIndex) const {return fVertex[evtIndex];}
  void      GetVertex(Double_t vertex[3], const Int_t evtIndex) const 
    {vertex[0]=fVertex[evtIndex][0];  vertex[1]=fVertex[evtIndex][1];  vertex[2]=fVertex[evtIndex][2];}   
  
  Bool_t FillInputEvent(const Int_t iEntry, const char * currentFileName) ;
  AliVEvent*  GetInputEvent() const {return (AliVEvent *) GetMC();}
  void   SetInputOutputMCEvent(AliVEvent* esd, AliAODEvent* aod, AliMCEvent* mc) ;
  
  void   SetCaloClusterPID(const Int_t pdgCode, AliVCluster *calo)     const ;
  void   SetTrackChargeAndPID(const Int_t pdgCode, AliAODTrack *track) const ;
  
  void   SwitchOnOverlapCheck()  {fCheckOverlap = kTRUE  ; }
  void   SwitchOffOverlapCheck() {fCheckOverlap = kFALSE ; }

  Float_t GetEMCALOverlapAngle() const {return fEMCALOverlapAngle ; }
  Float_t GetPHOSOverlapAngle()  const {return fPHOSOverlapAngle  ; }
  void    SetEMCALOverlapAngle(Float_t angle)  {fEMCALOverlapAngle = angle ; }
  void    SetPHOSOverlapAngle(Float_t angle)   {fPHOSOverlapAngle  = angle ; }
    
 private:
  
  void   CheckOverlap(const Float_t anglethres, const Int_t imom, Int_t & iPrimary, Int_t & index, TLorentzVector & mom, Int_t & pdg);
  void   MakePi0Decay(const TLorentzVector p0, TLorentzVector &p1, TLorentzVector &p2) const ;//, Double_t &angle); 
  void   FillCalorimeters(Int_t & iParticle, TParticle* particle, TLorentzVector & momentum) ;
  
  private:
  Bool_t    fDecayPi0 ;              // If not decayed, decay pi0 by hand
  TArrayI * fNeutralParticlesArray ; // Do not keep neutral particles of this list in calorimeter.
  TArrayI * fChargedParticlesArray ; // Keep charged particles of this list in calorimeter.
  TArrayI * fStatusArray ;           // Keep particles with status of the list.
  Bool_t    fKeepAllStatus ;         // Do or do not select particles depending on their status code.
  Bool_t    fCheckOverlap;           // Check of overlapped photons from pi0 enter the calorimeter
  Float_t   fEMCALOverlapAngle;      // Aperture angle of photons from decay that is not resolved by EMCAL, in radians
  Float_t   fPHOSOverlapAngle;       // Aperture angle of photons from decay that is not resolved by PHOS, in radians
  Int_t     fIndex2ndPhoton;         // Check overlap of first decay photon already done, internal use.
  Bool_t    fOnlyGeneratorParticles; // Use particles only generated by PYTHIA/HERWIG/... and not by the MC tranport G3/G4/FLUKA ...
  
  AliCaloTrackMCReader(const AliCaloTrackMCReader & g) ; // cpy ctor	
  AliCaloTrackMCReader & operator = (const AliCaloTrackMCReader & g) ;//cpy assignment
  
  ClassDef(AliCaloTrackMCReader,4)
} ;


#endif //ALICALOTRACKMCREADER_H