PWGLF/RESONANCES/AliRsnLoopPair.h

   1 #ifndef ALIRSNLOOPPAIR_H
   2 #define ALIRSNLOOPPAIR_H
   3
   4 //
   5 // Class to combine pairs of daughters.
   6 // See general definition inf mother class AliRsnLoop
   7 // This is the only implementation which takes into account mixing objects.
   8 //
   9
  10 #include "AliRsnLoop.h"
  11
  12 class TList;
  13 class AliRsnEvent;
  14 class AliRsnPairDef;
  15
  16 class AliRsnLoopPair : public AliRsnLoop {
  17 public:
  18
  19    AliRsnLoopPair(const char *name = "default", AliRsnPairDef *def = 0, Bool_t isMixed = kFALSE);
  20    AliRsnLoopPair(const AliRsnLoopPair &copy);
  21    AliRsnLoopPair &operator=(const AliRsnLoopPair &copy);
  22    ~AliRsnLoopPair();
  23
  24    // getters
  25    Bool_t         IsTrueMC()     const {return fTrueMC;}
  26    Bool_t         IsOnlyTrue()   const {return fOnlyTrue;}
  27    Bool_t         IsCheckDecay() const {return fCheckDecay;}
  28    AliRsnPairDef *GetPairDef()         {return fPairDef;}
  29    AliRsnCutSet  *GetPairCuts()        {return fPairCuts;}
  30    AliRsnMother  *GetMother()          {return &fMother;}
  31
  32    // setters (not for all members)
  33    void           SetPairCuts(AliRsnCutSet *cuts)      {fPairCuts = cuts;}
  34    void           SetTrueMC(Bool_t yn = kTRUE)         {fTrueMC = yn;}
  35    void           SetOnlyTrue(Bool_t onlyTrue = kTRUE) {fOnlyTrue = onlyTrue;}
  36    void           SetMCRefInfo(Bool_t b = kTRUE)       {fUseMCRef = b;}
  37    void           SetCheckDecay(Bool_t check = kTRUE)  {fCheckDecay = check;}
  38    void           SetListID(Int_t i, Int_t val)        {if (i==0||i==1) fListID[i] = val;}
  39    void           SetRangeY(Double_t range)            {fRangeY = range;}
  40
  41    // methods
  42    Bool_t         IsTrueMother();
  43    virtual void   Print(Option_t *opt = "") const;
  44    virtual Bool_t Init(const char *prefix, TList *list);
  45    virtual Int_t  DoLoop(AliRsnEvent *main, AliRsnDaughterSelector *smain, AliRsnEvent *mix = 0, AliRsnDaughterSelector *smix = 0);
  46
  47 protected:
  48
  49    Bool_t AssignMotherAndDaughters   (AliRsnEvent *event, Int_t ipart);
  50    Bool_t AssignMotherAndDaughtersESD(AliRsnEvent *event, Int_t ipart);
  51    Bool_t AssignMotherAndDaughtersAOD(AliRsnEvent *event, Int_t ipart);
  52    Int_t  LoopTrueMC(AliRsnEvent *event);
  53
  54    Bool_t            fTrueMC;          //  if this flag is TRUE, scan the MC for all true resonances from MC
  55    Bool_t            fOnlyTrue;        //  select true pairs only?
  56    Bool_t            fUseMCRef;        //  uses MC ref instead of REC
  57    Bool_t            fCheckDecay;      //  is the decay channel correct in a true pair?
  58    Int_t             fListID[2];       //  indexes of the two entry lists to be used
  59    Double_t          fRangeY;          //  range in rapidity (serves always)
  60
  61    AliRsnPairDef    *fPairDef;         //  pair definition
  62    AliRsnCutSet     *fPairCuts;        //  collection of all cuts
  63    AliRsnMother      fMother;          //! mother candidate (to avoid creating it continuously)
  64    AliRsnDaughter    fDaughter[2];     //! daughter candidates
  65
  66 private:
  67
  68    ClassDef(AliRsnLoopPair, 4)
  69 };
  70
  71 #endif
  72