STEER/AliAODv0.h

   1 #ifndef AliAODv0_H
   2 #define AliAODv0_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 //-------------------------------------------------------------------------
   7 //     Analysis Oriented Data (AOD) V0 vertex class
   8 //     Authors: B.Hippolyte, IPHC, hippolyt@in2p3.fr
   9 //              G.Van Buren, BNL,  gene@bnl.gov      (original STAR MuDsts)
  10 //-------------------------------------------------------------------------
  11
  12 #include "AliAODRecoDecay.h"
  13
  14 //class AliESDEvent;
  15 //class AliESDv0;
  16
  17 class AliAODv0 : public AliAODRecoDecay {
  18
  19 public:
  20
  21   AliAODv0();
  22   //  AliAODv0(AliESDv0 *rV0Vertex, AliESDEvent *rEvent);
  23   AliAODv0(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
  24            Double_t *rMomPos, Double_t *rMomNeg, Double_t *rDcaDaughterToPrimVertex);
  25   virtual ~AliAODv0();
  26
  27   AliAODv0(const AliAODv0& rAliAODv0);
  28   AliAODv0& operator=(const AliAODv0& rAliAODv0);
  29
  30   //  void     Fill(AliESDv0 *rV0Vertex, AliESDEvent *rEvent);
  31   void     Fill(AliAODVertex *rAODVertex, Double_t rDcaV0Daughters, Double_t rDcaV0ToPrimVertex,
  32                 Double_t *rMomPos, Double_t *rMomNeg, Double_t *rDcaDaughterToPrimVertex);
  33   void     ResetV0();
  34   void     Print(Option_t* option = "") const;
  35
  36   void     SetOnFlyStatus(Bool_t status){fOnFlyStatus=status;}
  37   Bool_t   GetOnFlyStatus() const {return fOnFlyStatus;}
  38
  39   Double_t DecayVertexV0X() const;
  40   Double_t DecayVertexV0Y() const;
  41   Double_t DecayVertexV0Z() const;
  42
  43   Double_t DecayLengthV0(double*) const;
  44
  45   Double_t DcaV0Daughters() const;
  46   Double_t DcaV0ToPrimVertex() const;
  47   Double_t DcaPosToPrimVertex() const;
  48   Double_t DcaNegToPrimVertex() const;
  49   Double_t RadiusV0()           const;
  50   Double_t OpenAngleV0()        const;
  51
  52   Double_t MomPosX() const;
  53   Double_t MomPosY() const;
  54   Double_t MomPosZ() const;
  55   Double_t MomNegX() const;
  56   Double_t MomNegY() const;
  57   Double_t MomNegZ() const;
  58
  59   //  Int_t    KeyPos()  const;
  60   //  Int_t    KeyNeg()  const;
  61
  62   Double_t Chi2V0()  const;
  63
  64   Double_t MomV0X()  const;
  65   Double_t MomV0Y()  const;
  66   Double_t MomV0Z()  const;
  67
  68   Double_t Ptot2Pos() const;
  69   Double_t Ptot2Neg() const;
  70   Double_t Ptot2V0()  const;
  71   Double_t Pt2V0()    const;
  72   Double_t MomPosAlongV0() const;
  73   Double_t MomNegAlongV0() const;
  74   Double_t AlphaV0() const;
  75   Double_t PtArmV0() const;
  76   Double_t EPosProton() const;
  77   Double_t ENegProton() const;
  78   Double_t EPosPion() const;
  79   Double_t ENegPion() const;
  80   Double_t ELambda() const;
  81   Double_t EK0Short() const;
  82   Double_t MassLambda() const;
  83   Double_t MassAntiLambda() const;
  84   Double_t MassK0Short() const;
  85   Double_t RapK0Short() const;
  86   Double_t RapLambda() const;
  87   Double_t PseudoRapV0()    const;
  88   Double_t PseudoRapPos()   const;
  89   Double_t PseudoRapNeg()   const;
  90
  91
  92 protected:
  93   Double32_t fDcaV0ToPrimVertex;    // dca of V0 to primary vertex
  94   Bool_t     fOnFlyStatus;          // if kTRUE, then this V0 is recontructed
  95                                     // "on fly" during the tracking
  96   ClassDef(AliAODv0,2)
  97 };
  98
  99 inline Double_t AliAODv0::DecayVertexV0X() const {return this->GetSecVtxX();}
 100 inline Double_t AliAODv0::DecayVertexV0Y() const {return this->GetSecVtxY();}
 101 inline Double_t AliAODv0::DecayVertexV0Z() const {return this->GetSecVtxZ();}
 102
 103 inline Double_t AliAODv0::DecayLengthV0(double *tParentVertexPosition) const {
 104   return ::sqrt(::pow(DecayVertexV0X() - tParentVertexPosition[0],2) +
 105                 ::pow(DecayVertexV0Y() - tParentVertexPosition[1],2) +
 106                 ::pow(DecayVertexV0Z() - tParentVertexPosition[2],2));
 107 }
 108
 109 inline Double_t AliAODv0::DcaV0Daughters() const {return fDCA[0];}
 110 inline Double_t AliAODv0::DcaV0ToPrimVertex() const {return fDcaV0ToPrimVertex;}
 111 inline Double_t AliAODv0::DcaPosToPrimVertex() const {return fd0[0];}
 112 inline Double_t AliAODv0::DcaNegToPrimVertex() const {return fd0[1];}
 113
 114 inline Double_t AliAODv0::RadiusV0() const {
 115   return RadiusSecVtx();
 116 }
 117
 118 inline Double_t AliAODv0::OpenAngleV0() const {
 119   Double_t lPtot1xPtot2 = PxProng(0)*PxProng(1)+PyProng(0)*PyProng(1)+PzProng(0)*PzProng(1);
 120   Double_t lPtot1Ptot2_2 = Ptot2Pos()*Ptot2Neg();
 121   return ::acos(lPtot1xPtot2/::sqrt(lPtot1Ptot2_2) );
 122 }
 123
 124 inline Double_t AliAODv0::MomPosX() const {return fPx[0];}
 125 inline Double_t AliAODv0::MomPosY() const {return fPy[0];}
 126 inline Double_t AliAODv0::MomPosZ() const {return fPz[0];}
 127 inline Double_t AliAODv0::MomNegX() const {return fPx[1];}
 128 inline Double_t AliAODv0::MomNegY() const {return fPy[1];}
 129 inline Double_t AliAODv0::MomNegZ() const {return fPz[1];}
 130
 131 // inline Int_t AliAODv0::KeyPos() const {return fKeyPos;}
 132 // inline Int_t AliAODv0::KeyNeg() const {return fKeyNeg;}
 133
 134 inline Double_t AliAODv0::Chi2V0() const {return GetSecondaryVtx()->GetChi2perNDF();}
 135
 136 // Compare eventually AliAODv0::MomV0X() and AliAODRecoDecay::Px()
 137 inline Double_t AliAODv0::MomV0X() const {return MomPosX()+MomNegX();}
 138 inline Double_t AliAODv0::MomV0Y() const {return MomPosY()+MomNegY();}
 139 inline Double_t AliAODv0::MomV0Z() const {return MomPosZ()+MomNegZ();}
 140
 141 inline Double_t AliAODv0::Ptot2Pos() const {
 142   return (::pow(MomPosX(),2) + ::pow(MomPosY(),2) + ::pow(MomPosZ(),2) );
 143 }
 144 inline Double_t AliAODv0::Ptot2Neg() const {
 145   return (::pow(MomNegX(),2) + ::pow(MomNegY(),2) + ::pow(MomNegZ(),2) );
 146 }
 147 inline Double_t AliAODv0::Ptot2V0() const {return ( Pt2V0() + ::pow(MomV0Z(),2) );}
 148 inline Double_t AliAODv0::Pt2V0() const {
 149   return (::pow(MomV0X(),2) + ::pow(MomV0Y(),2) );
 150 }
 151
 152 inline Double_t AliAODv0::MomPosAlongV0() const {
 153   Double_t lPtot2V0 = Ptot2V0();
 154   if (lPtot2V0)
 155     return (MomPosX()*MomV0X() +
 156             MomPosY()*MomV0Y() +
 157             MomPosZ()*MomV0Z()) / ::sqrt(lPtot2V0);
 158   return 0.;
 159 }
 160
 161 inline Double_t AliAODv0::MomNegAlongV0() const {
 162   Double_t lPtot2V0 = Ptot2V0();
 163   if (lPtot2V0)
 164     return (MomNegX()*MomV0X() +
 165             MomNegY()*MomV0Y() +
 166             MomNegZ()*MomV0Z()) / ::sqrt(lPtot2V0);
 167   return 0.;
 168 }
 169
 170 inline Double_t AliAODv0::AlphaV0() const {
 171   return Alpha();
 172 }
 173 inline Double_t AliAODv0::PtArmV0() const {
 174   return QtProng(0);
 175 }
 176
 177 inline Double_t AliAODv0::EPosProton() const {
 178   return EProng(0,2212);
 179 }
 180
 181 inline Double_t AliAODv0::ENegProton() const {
 182   return EProng(1,2212);
 183 }
 184
 185 inline Double_t AliAODv0::EPosPion() const {
 186   return EProng(0,211);
 187 }
 188
 189 inline Double_t AliAODv0::ENegPion() const {
 190   return EProng(1,211);
 191 }
 192
 193 inline Double_t AliAODv0::ELambda() const {
 194   return E(3122);
 195 }
 196
 197 inline Double_t AliAODv0::EK0Short() const {
 198   return E(310);
 199 }
 200
 201 inline Double_t AliAODv0::MassLambda() const {
 202   return InvMass2Prongs(0,1,2212,211);
 203 }
 204
 205 inline Double_t AliAODv0::MassAntiLambda() const {
 206   return InvMass2Prongs(0,1,211,2212);
 207 }
 208
 209 inline Double_t AliAODv0::MassK0Short() const {
 210   return InvMass2Prongs(0,1,211,211);
 211 }
 212
 213 inline Double_t AliAODv0::RapK0Short() const {
 214   return Y(310);
 215 }
 216
 217 inline Double_t AliAODv0::RapLambda() const {
 218   return Y(3122);
 219 }
 220
 221 inline Double_t AliAODv0::PseudoRapV0() const {
 222   return Eta();
 223 }
 224
 225 inline Double_t AliAODv0::PseudoRapPos()   const {
 226   return EtaProng(0);
 227 }
 228
 229 inline Double_t AliAODv0::PseudoRapNeg()   const {
 230   return EtaProng(1);
 231 }
 232 //----------------------------------------------------------------------------
 233
 234 #endif