PWG2/RESONANCES/AliRsnValue.h

   1 //
   2 // Class AliRsnValue
   3 //
   4 // Definition of a single value which can be computed
   5 // from any of the defined input objects implemented
   6 // in the resonance package.
   7 //
   8
   9 #ifndef ALIRSNVALUE_H
  10 #define ALIRSNVALUE_H
  11
  12 #include "TNamed.h"
  13 #include "TArrayD.h"
  14
  15 class AliRsnPairDef;
  16 class AliRsnMother;
  17
  18 class AliRsnValue : public TNamed
  19 {
  20   public:
  21
  22     enum EValueType
  23     {
  24       kTrack1P,
  25       kTrack2P,
  26       kTrack1Pt,
  27       kTrack2Pt,
  28       kTrack1Px,
  29       kTrack1Py,
  30       kTrack1Pz,
  31       kTrack2Px,
  32       kTrack2Py,
  33       kTrack2Pz,
  34       kPairInvMass,
  35       kPairInvMassMC,
  36       kPairInvMassRes,
  37       kPairPt,
  38       kPairEta,
  39       kPairMt,
  40       kPairY,
  41       kPairCosThetaStar,
  42       kPairCosThetaStar1,
  43       kPairCosThetaStar2,
  44       kPairCosThetaStarMC1,
  45       kPairCosThetaStarMC2,
  46       kAngleToLeading,
  47       kLeadingPt,
  48       kQInv,
  49       kEventMult,
  50       kValueTypes
  51     };
  52
  53     AliRsnValue();
  54     AliRsnValue(const char *name, EValueType type, Int_t n = 0, Double_t min = 0.0, Double_t max = 0.0);
  55     AliRsnValue(const char *name, EValueType type, Double_t min, Double_t max, Double_t step);
  56     AliRsnValue(const char *name, EValueType type, Int_t n, Double_t *array);
  57     AliRsnValue(const AliRsnValue& copy) : TNamed(copy),fValue(copy.fValue),fType(copy.fType),fArray(copy.fArray) {}
  58     AliRsnValue& operator=(const AliRsnValue& copy) {SetName(copy.GetName());fType=copy.fType;fValue=copy.fValue;fArray=copy.fArray;return (*this);}
  59     virtual ~AliRsnValue() { }
  60
  61     TArrayD     GetArray() const {return fArray;}
  62     Double_t    GetValue() const {return fValue;}
  63     EValueType  GetValueType() {return fType;}
  64
  65     void        SetValueType(EValueType type) {fType = type;}
  66     void        SetBins(Int_t n, Double_t min, Double_t max);
  67     void        SetBins(Double_t min, Double_t max, Double_t step);
  68     void        SetBins(Int_t n, Double_t *array);
  69     void        Set(EValueType type, Int_t n = 0, Double_t min = 0.0, Double_t max = 0.0) {fType = type; SetBins(n, min, max);}
  70     void        Set(EValueType type, Double_t min, Double_t max, Double_t step) {fType = type; SetBins(min, max, step);}
  71     void        Set(EValueType type, Int_t n, Double_t *array) {fType = type; SetBins(n, array);}
  72
  73     virtual Bool_t  Eval(AliRsnMother * const mother, AliRsnPairDef * const pairDef, AliRsnEvent * const event);
  74     virtual Bool_t  Eval(AliRsnDaughter * const daughter, AliRsnEvent * const event);
  75     virtual void    Print(Option_t *option = "") const;
  76
  77   protected:
  78
  79     Double_t   fValue;   // computed value
  80
  81   private:
  82
  83     EValueType fType;    // value type
  84     TArrayD    fArray;   // array of bins (when necessary)
  85
  86     // ROOT dictionary
  87     ClassDef(AliRsnValue, 1)
  88 };
  89
  90 #endif