Fix in the computetion of <pt>

[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliCFVertexingHF3Prong.h

diff --git a/PWGHF/vertexingHF/AliCFVertexingHF3Prong.h b/PWGHF/vertexingHF/AliCFVertexingHF3Prong.h
index 9cd76df9332c00ff884b6d452179d7bb7410a6aa..3c7464812785f9db7141e3cab76be6c0ecab192f 100644 (file)
--- a/PWGHF/vertexingHF/AliCFVertexingHF3Prong.h
+++ b/PWGHF/vertexingHF/AliCFVertexingHF3Prong.h
@@ -29,8 +29,10 @@ class TDatabasePDG;
 class AliCFVertexingHF3Prong : public AliCFVertexingHF{\r
        public:\r
                \r
+  AliCFVertexingHF3Prong(Int_t decay,UInt_t resonantDecay);\r
   AliCFVertexingHF3Prong(Int_t decay);\r
   AliCFVertexingHF3Prong(TClonesArray *mcArray, UShort_t originDselection, Int_t decay);  \r
+  AliCFVertexingHF3Prong(TClonesArray *mcArray, UShort_t originDselection, Int_t decay,UInt_t resonantDecay);  \r
   virtual ~AliCFVertexingHF3Prong(){};\r
   \r
   \r
@@ -41,10 +43,15 @@ class AliCFVertexingHF3Prong : public AliCFVertexingHF{
   \r
   Bool_t SetRecoCandidateParam(AliAODRecoDecayHF *recoCand);\r
   virtual void SetDecay3Prong(Int_t decay){fDecay=decay;}\r
+  Bool_t CheckLc3Prong() const;\r
+  void SetResonantDecay(UInt_t resonantDecay) {fResonantDecay = resonantDecay;}\r
+  UInt_t GetResonantDecay() const {return fResonantDecay;}\r
  \r
   void SetGeneratedDsOption(Int_t opt) {fGenDsOption=opt;}\r
   Int_t GetGeneratedDsOption() const {return fGenDsOption;}\r
-  \r
+\r
+  enum EDsSel {kCountAllDsKKpi=0, kCountPhipi=1, kCountK0stK=2, kCountResonant=3, kCountNonResonant=4};\r
+\r
  protected:\r
   \r
   \r
@@ -54,9 +61,10 @@ class AliCFVertexingHF3Prong : public AliCFVertexingHF{
   AliCFVertexingHF3Prong& operator= (const AliCFVertexingHF3Prong& other);\r
   \r
   Int_t fDecay;   // decay mode id\r
-  Int_t fGenDsOption;  // 0(default) --> count all Ds->KKpi, 1 --> count Ds->phipi 2--> count Ds->K0*K\r
+  Int_t fGenDsOption;  // option for selection Ds (see enum)\r
+  UInt_t fResonantDecay; // resonant decay for which to run the CF\r
 \r
-  ClassDef(AliCFVertexingHF3Prong, 2);\r
+  ClassDef(AliCFVertexingHF3Prong, 3);\r
   \r
 };\r
 \r