Updates D* cuts for peripheral PbPb events

[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAODRecoDecayHF3Prong.h

diff --git a/PWG3/vertexingHF/AliAODRecoDecayHF3Prong.h b/PWG3/vertexingHF/AliAODRecoDecayHF3Prong.h
index 61e0ce9a0a0e18333a0bf6fc0de7143044c98b47..24ee38fe1a432fc71abb86772742c0b80cafc0a7 100644 (file)
--- a/PWG3/vertexingHF/AliAODRecoDecayHF3Prong.h
+++ b/PWG3/vertexingHF/AliAODRecoDecayHF3Prong.h
@@ -3,6 +3,8 @@
 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
+/* $Id$ */ 
+
 //***********************************************************
 // Class AliAODRecoDecayHF3Prong
 // base class for AOD reconstructed 3-prong heavy-flavour decays
@@ -16,7 +18,7 @@
 class AliAODRecoDecayHF3Prong : public AliAODRecoDecayHF {
 
  public:
-
+  
   AliAODRecoDecayHF3Prong();
    AliAODRecoDecayHF3Prong(AliAODVertex *vtx2,
                           Double_t *px,Double_t *py,Double_t *pz,
@@ -30,14 +32,13 @@ class AliAODRecoDecayHF3Prong : public AliAODRecoDecayHF {
 
   AliAODRecoDecayHF3Prong(const AliAODRecoDecayHF3Prong& source);
   AliAODRecoDecayHF3Prong& operator=(const AliAODRecoDecayHF3Prong& source); 
-
-  virtual ~AliAODRecoDecayHF3Prong() {}  
- 
   void GetDCAs(Double_t dca[3]) const 
     {for(Int_t i=0;i<3;i++) dca[i]=GetDCA(i);}
   Double_t GetSigmaVert() const { return fSigmaVert; }
   Double_t GetDist12toPrim() const { return fDist12toPrim; }
   Double_t GetDist23toPrim() const { return fDist23toPrim; }
+  void SetDist12toPrim(Double_t d) { fDist12toPrim=d; }
+  void SetDist23toPrim(Double_t d) { fDist23toPrim=d; }
 
 
   // D+->Kpipi
@@ -57,8 +58,16 @@ class AliAODRecoDecayHF3Prong : public AliAODRecoDecayHF {
   Double_t CtDs(AliAODVertex *vtx1) const {return AliAODRecoDecay::Ct(431,vtx1);}
   Double_t InvMassDsKKpi() const {UInt_t pdg[3]={321,321,211};return InvMass(3,pdg);}
   Double_t InvMassDspiKK() const {UInt_t pdg[3]={211,321,321};return InvMass(3,pdg);}
-  Bool_t   SelectDs(const Double_t* cuts,Int_t &okDsKKpi,Int_t &okDspiKK) 
-    const; // only mass cut
+  
+  Double_t CosPiKPhiRFrameKKpi() const {return CosPiKPhiRFrame(0);}
+  Double_t CosPiKPhiRFramepiKK() const {return CosPiKPhiRFrame(1);}
+  Double_t CosPiDsLabFrameKKpi() const {return CosPiDsLabFrame(0);}
+  Double_t CosPiDsLabFramepiKK() const {return CosPiDsLabFrame(1);}
+  
+  Double_t CosPiKPhiRFrame(Int_t option) const;
+  Double_t CosPiDsLabFrame(Int_t option) const;
+  Bool_t   SelectDs(const Double_t* cuts,Int_t &okDsKKpi,Int_t &okDspiKK, Int_t &okMassPhi, Int_t &okMassK0star) 
+    const; // same variables as D+, for now
 
   // Lambdac+->pKpi
   Double_t ELc() const {return E(4122);} 
@@ -69,7 +78,7 @@ class AliAODRecoDecayHF3Prong : public AliAODRecoDecayHF {
   Double_t InvMassLcpKpi() const {UInt_t pdg[3]={2212,321,211};return InvMass(3,pdg);}
   Double_t InvMassLcpiKp() const {UInt_t pdg[3]={211,321,2212};return InvMass(3,pdg);}
   Bool_t   SelectLc(const Double_t* cuts,Int_t &okLcpKpi,Int_t &okLcpiKp) 
-    const;// only mass cut
+    const; // same variables as D+, for now
 
  private: