PWG/EMCAL/AliClusterContainer.h

   1 #ifndef AliClusterContainer_H
   2 #define AliClusterContainer_H
   3
   4 //
   5 // container with name, TClonesArray and cuts for particles
   6 //
   7
   8 class TLorentzVector;
   9
  10 class AliVEvent;
  11 class AliVCluster;
  12
  13 #include "AliEmcalContainer.h"
  14
  15 class AliClusterContainer : public AliEmcalContainer {
  16  public:
  17   AliClusterContainer();
  18   AliClusterContainer(const char *name);
  19   virtual ~AliClusterContainer(){;}
  20
  21   void SetClusterArray(AliVEvent *event);
  22
  23   void SetClusPtCut(Double_t cut)                  { fClusPtCut      = cut ; }
  24   void SetClusTimeCut(Double_t min, Double_t max)  { fClusTimeCutLow = min ; fClusTimeCutUp = max ; }
  25   void SetClusterBitMap(UInt_t m)                  { fClusterBitMap     = m ; }
  26   void SetMCClusterBitMap(UInt_t m)                { fMCClusterBitMap   = m ; }
  27   void SetMinMCLabel(Int_t s)                      { fMinMCLabel        = s ; }
  28
  29   AliVCluster                *GetLeadingCluster(const char* opt="")  const;
  30   AliVCluster                *GetCluster(Int_t i)                    const;
  31   AliVCluster                *GetAcceptCluster(Int_t i)              const;
  32   AliVCluster                *GetNextAcceptCluster(Int_t i=-1)       const;
  33   void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
  34   Bool_t                      AcceptCluster(AliVCluster         *vp) const;
  35   Int_t                       GetNClusters()                         const   {return GetNEntries();}
  36
  37  protected:
  38   Double_t                    fClusPtCut;                  // cut on cluster pt
  39   Double_t                    fClusTimeCutLow;             // low time cut for clusters
  40   Double_t                    fClusTimeCutUp;              // up time cut for clusters
  41   UInt_t                      fClusterBitMap;              // bit map of accepted clusters (non MC)
  42   UInt_t                      fMCClusterBitMap;            // bit map of accepted MC clusters
  43   Int_t                       fMinMCLabel;                 // minimum MC label value for the tracks/clusters being considered MC particles
  44
  45  private:
  46   AliClusterContainer(const AliClusterContainer& obj); // copy constructor
  47   AliClusterContainer& operator=(const AliClusterContainer& other); // assignment
  48
  49   ClassDef(AliClusterContainer,1);
  50
  51 };
  52
  53 #endif
  54