PWG/CaloTrackCorrBase/AliCaloTrackESDReader.h

   1 #ifndef ALICALOTRACKESDREADER_H
   2 #define ALICALOTRACKESDREADER_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice     */
   5
   6 //_________________________________________________________________________
   7 // Class for reading data (ESDs) in order to do prompt gamma
   8 //  or other particle identification and correlations
   9 //
  10 //
  11 //
  12 //
  13 //*-- Author: Gustavo Conesa (INFN-LNF)
  14
  15 class AliESDEvent;
  16
  17 #include "AliCaloTrackReader.h"
  18
  19 class AliCaloTrackESDReader : public AliCaloTrackReader {
  20
  21 public:
  22
  23                    AliCaloTrackESDReader() ; // ctor
  24
  25   virtual         ~AliCaloTrackESDReader() ; // virtual dtor
  26
  27   Bool_t           CheckForPrimaryVertex() const ;
  28
  29   void             Init();
  30
  31   Bool_t           SelectTrack(AliVTrack* track, Double_t* pTrack);
  32
  33   AliESDtrackCuts* GetTrackCuts()                    const { return fESDtrackCuts     ; }
  34   void             SetTrackCuts(AliESDtrackCuts * cuts) ;
  35
  36   AliESDtrackCuts* GetTrackComplementaryCuts()       const { return fESDtrackComplementaryCuts ; }
  37   void             SetTrackComplementaryCuts(AliESDtrackCuts * cuts)  ;
  38
  39   void             SwitchOnConstrainTrackToVertex()        { fConstrainTrack = kTRUE  ; }
  40   void             SwitchOffConstrainTrackToVertex()       { fConstrainTrack = kFALSE ; }
  41
  42   void             SetInputOutputMCEvent(AliVEvent* esd, AliAODEvent* aod, AliMCEvent* mc) ;
  43
  44 private:
  45
  46   Bool_t           fConstrainTrack;            // Constrain Track to vertex
  47   AliESDtrackCuts* fESDtrackCuts ;             // Track cut
  48   AliESDtrackCuts* fESDtrackComplementaryCuts; // Track cut, complementary cuts for hybrids
  49
  50   AliCaloTrackESDReader(              const AliCaloTrackESDReader & r) ; // cpy ctor
  51   AliCaloTrackESDReader & operator = (const AliCaloTrackESDReader & r) ; // cpy assignment
  52
  53   ClassDef(AliCaloTrackESDReader,2)
  54
  55 } ;
  56
  57
  58 #endif //ALICALOTRACKESDREADER_H
  59
  60
  61