Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnMiniOutput.h

diff --git a/PWGLF/RESONANCES/AliRsnMiniOutput.h b/PWGLF/RESONANCES/AliRsnMiniOutput.h
index a801a9b17071f6b898536bc01efc6636ad8d14d2..d95a4a4517bdabe4ef8a149c309ca8c15cbc9a6e 100644 (file)
--- a/PWGLF/RESONANCES/AliRsnMiniOutput.h
+++ b/PWGLF/RESONANCES/AliRsnMiniOutput.h
@@ -43,6 +43,7 @@ public:
       kTrackPairRotated2,
       kTruePair,
       kMother,
+      kMotherInAcc,
       kComputations
    };
 
@@ -57,6 +58,7 @@ public:
    Bool_t          IsTrackPairMix()     const {return (fComputation == kTrackPairMix);}
    Bool_t          IsTruePair()         const {return (fComputation == kTruePair);}
    Bool_t          IsMother()           const {return (fComputation == kMother);}
+   Bool_t          IsMotherInAcc()      const {return (fComputation == kMotherInAcc);}
    Bool_t          IsDefined()          const {return (IsEventOnly() || IsTrackPair() || IsTrackPairMix() || IsTruePair() || IsMother());}
    Bool_t          IsLikeSign()         const {return (fCharge[0] == fCharge[1]);}
    Bool_t          IsSameCut()          const {return (fCutID[0] == fCutID[1]);}
@@ -87,6 +89,9 @@ public:
    void            SetFillHistogramOnlyInRange(Bool_t fillInRangeOnly) { fCheckHistRange = fillInRangeOnly; }
    void            SetMaxNSisters(Short_t n)          {fMaxNSisters = n;}
    void            SetCheckMomentumConservation(Bool_t checkP) {fCheckP = checkP;}
+   void            SetCheckFeedDown(Bool_t checkFeedDown)      {fCheckFeedDown = checkFeedDown;}
+   void            SetDselection(UShort_t originDselection);
+   void           SetRejectCandidateIfNotFromQuark(Bool_t opt){fRejectIfNoQuark=opt;}
 
    void            AddAxis(Int_t id, Int_t nbins, Double_t min, Double_t max);
    void            AddAxis(Int_t id, Double_t min, Double_t max, Double_t step);
@@ -97,6 +102,7 @@ public:
    AliRsnMiniPair &Pair() {return fPair;}
    Bool_t          Init(const char *prefix, TList *list);
    Bool_t          FillMother(const AliRsnMiniPair *pair, AliRsnMiniEvent *event, TClonesArray *valueList);
+   Bool_t          FillMotherInAcceptance(const AliRsnMiniPair *pair, AliRsnMiniEvent *event, TClonesArray *valueList);
    Bool_t          FillEvent(AliRsnMiniEvent *event, TClonesArray *valueList);
    Int_t           FillPair(AliRsnMiniEvent *event1, AliRsnMiniEvent *event2, TClonesArray *valueList, Bool_t refFirst = kTRUE);
 
@@ -125,9 +131,14 @@ private:
    TArrayI          fSel2;             //! list of selected particles for definition 2
    Short_t          fMaxNSisters;      // maximum number of allowed mother's daughter
    Bool_t           fCheckP;           // flag to set in order to check the momentum conservation for daughters
+   Bool_t           fCheckFeedDown;    // flag to set in order to check the particle feed down (specific for D meson analysis)
+   UShort_t        fOriginDselection; // flag to select D0 origins. 0 Only from charm 1 only from beauty 2 both from charm and beauty (specific for D meson analysis)
+   Bool_t          fKeepDfromB;       // flag for the feed down from b quark decay (specific for D meson analysis)                       
+   Bool_t          fKeepDfromBOnly;   // flag to keep only the charm particles that comes from beauty decays (specific for D meson analysis)
+   Bool_t          fRejectIfNoQuark;  // flag to remove events not generated with PYTHIA
    Bool_t           fCheckHistRange;   //  check if values is in histogram range
 
-   ClassDef(AliRsnMiniOutput, 3)  // AliRsnMiniOutput class
+   ClassDef(AliRsnMiniOutput, 4)  // AliRsnMiniOutput class
 };
 
 #endif