PWGLF/totEt/AliAnalysisEtSelector.h

   1 #ifndef ALIANALYSISETSELECTOR_H
   2 #define ALIANALYSISETSELECTOR_H
   3 //_________________________________________________________________________
   4 //  Utility Class for transverse energy studies
   5 //  Selector Base class
   6 //  -
   7 //
   8 //*-- Authors: Oystein Djuvsland (Bergen)
   9 //_________________________________________________________________________
  10 #include <Rtypes.h>
  11 #include "AliAnalysisEtCommon.h"
  12 #include "AliESDEvent.h"
  13
  14 class AliESDCaloCluster;
  15 class AliVTrack;
  16 class TRefArray;
  17 class AliAnalysisEtCuts;
  18 class TParticle;
  19 class TParticlePDG;
  20 class AliStack;
  21
  22 class AliAnalysisEtSelector : public AliAnalysisEtCommon
  23 {
  24
  25 public:
  26
  27     // Constructor takes cuts object
  28     AliAnalysisEtSelector(AliAnalysisEtCuts *cuts);
  29     AliAnalysisEtSelector();
  30
  31     // Destructor
  32     virtual ~AliAnalysisEtSelector();
  33
  34     // Set the current event
  35     virtual void SetEvent(const AliESDEvent *event);
  36
  37     // Init
  38     virtual void Init() {}
  39
  40     // Init with event
  41     virtual Int_t Init(const AliESDEvent *event) { fRunNumber = event->GetRunNumber(); return 0; }
  42
  43     // Return CaloClusters for the detector
  44     virtual TRefArray* GetClusters() { return 0; }
  45
  46     // Return true if cluster has energy > cut
  47     virtual Bool_t PassMinEnergyCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  48
  49     // Return true if cluster has energy > cut
  50     virtual Bool_t PassMinEnergyCut(const TParticle &/*part*/) const { return true; }
  51
  52     // Cut on distance to bad channel
  53     virtual Bool_t PassDistanceToBadChannelCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  54
  55     // Cut on track matching
  56     virtual Bool_t PassTrackMatchingCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  57
  58     // Cut on neutral monte carlo particle
  59     virtual Bool_t IsNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const;
  60
  61     // Is it an EM E_T particle
  62     virtual Bool_t IsEmEtParticle(const Int_t pdgCode) const;
  63
  64     // Does the particle come from an EM E_T primary ?
  65     virtual Bool_t PrimaryIsEmEtParticle(const Int_t pIdx, AliStack &stack) const;
  66
  67     // Get the index of primary particle for the particle
  68     Int_t GetPrimary(const Int_t partIdx, AliStack &stack) const;
  69
  70     // Cut on geometrical acceptance
  71     virtual Bool_t CutGeometricalAcceptance(const TParticle &/*part*/) const { return true; }
  72
  73     // Cut on geometrical acceptance
  74     virtual Bool_t CutGeometricalAcceptance(const AliVTrack &/*part*/) const { return true; }
  75
  76     // Cut on geometrical acceptance
  77     virtual Bool_t CutGeometricalAcceptance(const AliESDCaloCluster &/*part*/) const { return true; }
  78
  79     // From secondary vertex?
  80     virtual Bool_t FromSecondaryInteraction(const TParticle& part, AliStack& stack) const;
  81
  82     // Cluster is in correct detector
  83     virtual Bool_t IsDetectorCluster(const AliESDCaloCluster &cluster) const = 0;
  84
  85 protected:
  86
  87     const AliVEvent *fEvent; // Pointer to current event
  88
  89     TRefArray *fClusterArray; // Array of clusters
  90
  91     AliAnalysisEtCuts *fCuts; // Pointer to the cuts object; DS: also in base class?
  92
  93     Bool_t SuspeciousDecayInChain(const UInt_t suspectMotherPdg, const UInt_t suspectDaughterPdg, const TParticle& part, AliStack& stack) const;
  94
  95     Int_t fRunNumber;
  96
  97     Bool_t fInitialized; // matrix initialized
  98
  99 private:
 100
 101     //AliAnalysisEtSelector(); // Prohibited
 102     AliAnalysisEtSelector(const AliAnalysisEtSelector& other);// Prohibited
 103     AliAnalysisEtSelector& operator=(const AliAnalysisEtSelector& other);// Prohibited
 104     bool operator==(const AliAnalysisEtSelector& other) const;// Prohibited
 105
 106     ClassDef(AliAnalysisEtSelector, 1);
 107 };
 108
 109 #endif // ALIANALYSISETSELECTOR_H