HLT/JET/AliHLTJETReader.h

   1 //-*- Mode: C++ -*-
   2
   3 // $Id:  $
   4
   5 #ifndef ALIHLTJETREADER_H
   6 #define ALIHLTJETREADER_H
   7
   8 /* This file is property of and copyright by the ALICE HLT Project        *
   9  * ALICE Experiment at CERN, All rights reserved.                         *
  10  * See cxx source for full Copyright notice                               */
  11
  12 /** @file   AliHLTJETReader.h
  13     @author Jochen Thaeder
  14     @date
  15     @brief  Reader for jet finder
  16 */
  17
  18 #include "AliHLTLogging.h"
  19
  20 #include "AliJetReader.h"
  21 #include "AliJetReaderHeader.h"
  22
  23 #include "AliHLTJETReaderHeader.h"
  24
  25 #include "AliHLTMCEvent.h"
  26
  27 #include "AliESDEvent.h"
  28 #include "AliAODEvent.h"
  29
  30
  31 /**
  32  * @class  AliHLTJETReader
  33  * Reader for jet finder
  34  *
  35  * @ingroup alihlt_jet
  36  */
  37
  38 class AliHLTJETReader : public AliJetReader, public AliHLTLogging {
  39
  40 public:
  41
  42   /*
  43    * ---------------------------------------------------------------------------------
  44    *                            Constructor / Destructor
  45    * ---------------------------------------------------------------------------------
  46    */
  47
  48   /** standard constructor */
  49   AliHLTJETReader();
  50
  51   /** destructor */
  52   virtual ~AliHLTJETReader();
  53
  54   /*
  55    * ---------------------------------------------------------------------------------
  56    *                               Reader functionality
  57    * ---------------------------------------------------------------------------------
  58    */
  59
  60   /** Fill tracks in momentum array
  61    *  @return kTRUE on success, otherwise kFALSE
  62    */
  63   Bool_t FillMomentumArray();
  64
  65   /** Fill MC tracks in momentum array
  66    *  @return kTRUE on success, otherwise kFALSE
  67    */
  68   Bool_t FillMomentumArrayMC();
  69
  70   /** Fill ESD tracks in momentum array
  71    *  @return kTRUE on success, otherwise kFALSE
  72    */
  73   Bool_t FillMomentumArrayESD();
  74
  75   /** Fill AOD tracks in momentum array
  76    *  @return kTRUE on success, otherwise kFALSE
  77    */
  78   Bool_t FillMomentumArrayAOD();
  79
  80   /*
  81    * ---------------------------------------------------------------------------------
  82    *                                     Setter
  83    * ---------------------------------------------------------------------------------
  84    */
  85
  86   /** Set pointer to input event
  87    *  @param esd an AliESDEvent
  88    *  @param aod an AliAODEvent
  89    *  @param mc an AliHLTMCEvent
  90    */
  91   void SetInputEvent(TObject* esd, TObject* aod, TObject* mc);
  92
  93   /*
  94    * ---------------------------------------------------------------------------------
  95    *                                     Getter
  96    * ---------------------------------------------------------------------------------
  97    */
  98
  99   /** Get Ptr to AliHLTJETReaderHeader
 100    *  @return ptr to AliHLTJETReaderHeader
 101    */
 102   AliHLTJETReaderHeader* GetReaderHeader() {return dynamic_cast<AliHLTJETReaderHeader*>(fReaderHeader);}
 103
 104   ///////////////////////////////////////////////////////////////////////////////////
 105
 106 private:
 107
 108   /** copy constructor prohibited */
 109   AliHLTJETReader (const AliHLTJETReader&);
 110
 111   /** assignment operator prohibited */
 112   AliHLTJETReader& operator= (const AliHLTJETReader&);
 113
 114   /*
 115    * ---------------------------------------------------------------------------------
 116    *                             Members - private
 117    * ---------------------------------------------------------------------------------
 118    */
 119
 120   /** ESD event */
 121   AliESDEvent*       fESD;                     //! transient
 122
 123   /** MC event */
 124   AliHLTMCEvent*     fMC;                      //! transient
 125
 126   /** AOD event */
 127   AliAODEvent*       fAOD;                     //! transient
 128
 129   ClassDef(AliHLTJETReader, 1)
 130
 131 };
 132 #endif
 133