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     virtual Float_t ShiftAngle(Float_t phi);//Always returns an angle in radians between -pi<phi<pi
  47
  48     // Return true if cluster has energy > cut
  49     virtual Bool_t PassMinEnergyCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  50
  51     // Return true if cluster has energy > cut
  52     virtual Bool_t PassMinEnergyCut(const TParticle &/*part*/) const { return true; }
  53
  54     // Cut on distance to bad channel
  55     virtual Bool_t PassDistanceToBadChannelCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  56
  57     // Cut on track matching
  58     virtual Bool_t PassTrackMatchingCut(const AliESDCaloCluster &/*cluster*/) const { return true; }
  59
  60     // Cut on neutral monte carlo particle
  61     virtual Bool_t IsNeutralMcParticle(Int_t pIdx, AliStack& s, const TParticlePDG& pdg) const;
  62
  63     // Is it an EM E_T particle
  64     virtual Bool_t IsEmEtParticle(const Int_t pdgCode) const;
  65
  66     // Does the particle come from an EM E_T primary ?
  67     virtual Bool_t PrimaryIsEmEtParticle(const Int_t pIdx, AliStack &stack) const;
  68
  69     // Get the index of primary particle for the particle
  70     Int_t GetPrimary(const Int_t partIdx, AliStack &stack) const;
  71
  72     // Cut on geometrical acceptance
  73     virtual Bool_t CutGeometricalAcceptance(const TParticle &/*part*/) { return true; }
  74
  75     // Cut on geometrical acceptance
  76     virtual Bool_t CutGeometricalAcceptance(const AliVTrack &/*part*/) { return true; }
  77
  78     // Cut on geometrical acceptance
  79     virtual Bool_t CutGeometricalAcceptance(const AliESDCaloCluster &/*part*/) { return true; }
  80
  81     // From secondary vertex?
  82     virtual Bool_t FromSecondaryInteraction(const TParticle& part, AliStack& stack) const;
  83
  84     // Cluster is in correct detector
  85     virtual Bool_t IsDetectorCluster(const AliESDCaloCluster &cluster) const = 0;
  86
  87 protected:
  88
  89     const AliVEvent *fEvent; // Pointer to current event
  90
  91     TRefArray *fClusterArray; // Array of clusters
  92
  93     AliAnalysisEtCuts *fCuts; // Pointer to the cuts object; DS: also in base class?
  94
  95     Bool_t SuspeciousDecayInChain(const UInt_t suspectMotherPdg, const UInt_t suspectDaughterPdg, const TParticle& part, AliStack& stack) const;
  96
  97     Int_t fRunNumber;
  98
  99     Bool_t fInitialized; // matrix initialized
 100
 101 private:
 102
 103     //AliAnalysisEtSelector(); // Prohibited
 104     AliAnalysisEtSelector(const AliAnalysisEtSelector& other);// Prohibited
 105     AliAnalysisEtSelector& operator=(const AliAnalysisEtSelector& other);// Prohibited
 106     bool operator==(const AliAnalysisEtSelector& other) const;// Prohibited
 107
 108     ClassDef(AliAnalysisEtSelector, 1);
 109 };
 110
 111 #endif // ALIANALYSISETSELECTOR_H