PWG2/RESONANCES/AliRsnPIDIndex.h

   1 //
   2 // Class AliRsnPIDIndex
   3 //
   4 // It sorts the indexes of all tracks in an AliRsnEvent
   5 // for a fast retrieval of them according to charge and PID.
   6 //
   7 // author: M. Vala (email: martin.vala@cern.ch)
   8 //
   9
  10 #ifndef AliRsnPIDIndex_h
  11 #define AliRsnPIDIndex_h
  12
  13 #include <TArrayI.h>
  14
  15 #include "AliRsnEvent.h"
  16 #include "AliAODTrack.h"
  17 #include "AliRsnDaughter.h"
  18
  19 class AliESDtrackCuts;
  20
  21 class AliRsnPIDIndex : public TObject
  22 {
  23   public:
  24
  25     AliRsnPIDIndex(Int_t num = 1000);
  26     AliRsnPIDIndex(const AliRsnPIDIndex &copy);
  27     AliRsnPIDIndex& operator= (const AliRsnPIDIndex& copy);
  28
  29     virtual  ~AliRsnPIDIndex();
  30
  31     void      Print(Option_t *option = "") const;
  32     void      ResetAll(Int_t num=1000);
  33
  34     void      FillFromEvent(AliRsnEvent *event = 0, AliESDtrackCuts *cuts = 0);
  35
  36     void      AddIndex(const Int_t index, AliRsnDaughter::EPIDMethod meth, Char_t sign, AliPID::EParticleType type);
  37     void      SetCorrectIndexSize();
  38
  39     TArrayI*  GetTracksArray(AliRsnDaughter::EPIDMethod meth, Char_t sign, AliPID::EParticleType type);
  40     TArrayI*  GetCharged(Char_t sign);
  41
  42     // Prior probs
  43     void            SetPriorProbability(AliPID::EParticleType type, Double_t p);
  44     void            SetPriorProbability(Double_t *out);
  45     void            DumpPriors();
  46     void            GetPriorProbability(Double_t *out);
  47
  48   private:
  49
  50     Int_t     ChargeIndex(Char_t sign) const;
  51     Char_t    IndexCharge(Short_t sign) const;
  52
  53     TArrayI   fIndex[AliRsnDaughter::kMethods][2][AliPID::kSPECIES + 1];       // index arrays of pos/neg particles of each PID
  54     Int_t     fNumOfIndex[AliRsnDaughter::kMethods][2][AliPID::kSPECIES + 1];  //! array size
  55
  56     Double_t  fPrior[AliPID::kSPECIES]; // prior probabilities
  57
  58     ClassDef(AliRsnPIDIndex, 1);
  59 };
  60
  61 #endif