PWG1/AliGenV0Info.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16
  17 ///////////////////////////////////////////////////////////////////////////
  18 /*
  19
  20 Origin: marian.ivanov@cern.ch
  21 Container classes with MC infomation for V0
  22
  23
  24 */
  25
  26 #if !defined(__CINT__) || defined(__MAKECINT__)
  27 #include <stdio.h>
  28 #include <string.h>
  29 //ROOT includes
  30 #include "TROOT.h"
  31 #include "Rtypes.h"
  32 #include "TFile.h"
  33 #include "TTree.h"
  34 #include "TChain.h"
  35 #include "TCut.h"
  36 #include "TString.h"
  37 #include "TStopwatch.h"
  38 #include "TParticle.h"
  39 #include "TSystem.h"
  40 #include "TCanvas.h"
  41 #include "TGeometry.h"
  42 #include "TPolyLine3D.h"
  43
  44 //ALIROOT includes
  45 #include "AliRun.h"
  46 #include "AliStack.h"
  47 #include "AliSimDigits.h"
  48 #include "AliTPCParam.h"
  49 #include "AliTPC.h"
  50 #include "AliTPCLoader.h"
  51 #include "AliDetector.h"
  52 #include "AliTrackReference.h"
  53 #include "AliTPCParamSR.h"
  54 #include "AliTracker.h"
  55 #include "AliMagF.h"
  56 #include "AliHelix.h"
  57 #include "AliTrackPointArray.h"
  58
  59 #endif
  60 #include "AliGenV0Info.h"
  61 //
  62 //
  63
  64 ClassImp(AliGenV0Info)
  65
  66
  67
  68
  69
  70 /////////////////////////////////////////////////////////////////////////////////
  71 AliGenV0Info::AliGenV0Info():
  72   fMCd(),      //info about daughter particle -
  73   fMCm(),      //info about mother particle   - first particle for V0
  74   fMotherP(),   //particle info about mother particle
  75   fMCDist1(0), //info about closest distance according closest MC - linear DCA
  76   fMCDist2(0),    //info about closest distance parabolic DCA
  77   fMCRr(0),       // rec position of the vertex
  78   fMCR(0),        //exact r position of the vertex
  79   fInvMass(0),  //reconstructed invariant mass -
  80   fPointAngleFi(0), //point angle fi
  81   fPointAngleTh(0), //point angle theta
  82   fPointAngle(0)   //point angle full
  83 {
  84   for (Int_t i=0;i<3; i++){
  85    fMCPdr[i]=0;
  86    fMCX[i]=0;
  87    fMCXr[i]=0;
  88    fMCPm[i]=0;
  89    fMCAngle[i]=0;
  90    fMCPd[i]=0;
  91   }
  92   fMCPd[3]=0;
  93   for (Int_t i=0; i<2;i++){
  94     fPdg[i]=0;
  95     fLab[i]=0;
  96   }
  97 }
  98
  99 void AliGenV0Info::Update(Float_t vertex[3])
 100 {
 101   //
 102   // Update information - calculates derived variables
 103   //
 104
 105   fMCPd[0] = fMCd.GetParticle().Px();
 106   fMCPd[1] = fMCd.GetParticle().Py();
 107   fMCPd[2] = fMCd.GetParticle().Pz();
 108   fMCPd[3] = fMCd.GetParticle().P();
 109   //
 110   fMCX[0]  = fMCd.GetParticle().Vx();
 111   fMCX[1]  = fMCd.GetParticle().Vy();
 112   fMCX[2]  = fMCd.GetParticle().Vz();
 113   fMCR       = TMath::Sqrt( fMCX[0]*fMCX[0]+fMCX[1]*fMCX[1]);
 114   //
 115   fPdg[0]    = fMCd.GetParticle().GetPdgCode();
 116   fPdg[1]    = fMCm.GetParticle().GetPdgCode();
 117   //
 118   fLab[0]    = fMCd.GetParticle().GetUniqueID();
 119   fLab[1]    = fMCm.GetParticle().GetUniqueID();
 120   //
 121   //
 122   //
 123   Double_t x1[3],p1[3];
 124   TParticle& pdaughter = fMCd.GetParticle();
 125   x1[0] = pdaughter.Vx();
 126   x1[1] = pdaughter.Vy();
 127   x1[2] = pdaughter.Vz();
 128   p1[0] = pdaughter.Px();
 129   p1[1] = pdaughter.Py();
 130   p1[2] = pdaughter.Pz();
 131   Double_t sign = (pdaughter.GetPDG()->Charge()>0)? -1:1;
 132   AliHelix dhelix1(x1,p1,sign);
 133   //
 134   //
 135   Double_t x2[3],p2[3];
 136   //
 137   TParticle& pmother = fMCm.GetParticle();
 138   x2[0] = pmother.Vx();
 139   x2[1] = pmother.Vy();
 140   x2[2] = pmother.Vz();
 141   p2[0] = pmother.Px();
 142   p2[1] = pmother.Py();
 143   p2[2] = pmother.Pz();
 144   //
 145   //
 146   sign = (pmother.GetPDG()->Charge()>0) ? -1:1;
 147   AliHelix mhelix(x2,p2,sign);
 148
 149   //
 150   //
 151   //
 152   //find intersection linear
 153   //
 154   Double_t distance1, distance2;
 155   Double_t phase[2][2],radius[2];
 156   Int_t  points = dhelix1.GetRPHIintersections(mhelix, phase, radius);
 157   Double_t delta1=10000,delta2=10000;
 158   if (points>0){
 159     dhelix1.LinearDCA(mhelix,phase[0][0],phase[0][1],radius[0],delta1);
 160     dhelix1.LinearDCA(mhelix,phase[0][0],phase[0][1],radius[0],delta1);
 161     dhelix1.LinearDCA(mhelix,phase[0][0],phase[0][1],radius[0],delta1);
 162   }
 163   else{
 164     fInvMass=-1;
 165     return;
 166   }
 167   if (points==2){
 168     dhelix1.LinearDCA(mhelix,phase[1][0],phase[1][1],radius[1],delta2);
 169     dhelix1.LinearDCA(mhelix,phase[1][0],phase[1][1],radius[1],delta2);
 170     dhelix1.LinearDCA(mhelix,phase[1][0],phase[1][1],radius[1],delta2);
 171   }
 172   distance1 = TMath::Min(delta1,delta2);
 173   //
 174   //find intersection parabolic
 175   //
 176   points = dhelix1.GetRPHIintersections(mhelix, phase, radius);
 177   delta1=10000,delta2=10000;
 178
 179   if (points>0){
 180     dhelix1.ParabolicDCA(mhelix,phase[0][0],phase[0][1],radius[0],delta1);
 181   }
 182   if (points==2){
 183     dhelix1.ParabolicDCA(mhelix,phase[1][0],phase[1][1],radius[1],delta2);
 184   }
 185
 186   distance2 = TMath::Min(delta1,delta2);
 187   //
 188   if (delta1<delta2){
 189     //get V0 info
 190     dhelix1.Evaluate(phase[0][0],fMCXr);
 191     dhelix1.GetMomentum(phase[0][0],fMCPdr);
 192     mhelix.GetMomentum(phase[0][1],fMCPm);
 193     dhelix1.GetAngle(phase[0][0],mhelix,phase[0][1],fMCAngle);
 194     fMCRr = TMath::Sqrt(radius[0]);
 195   }
 196   else{
 197     dhelix1.Evaluate(phase[1][0],fMCXr);
 198     dhelix1.GetMomentum(phase[1][0],fMCPdr);
 199     mhelix.GetMomentum(phase[1][1],fMCPm);
 200     dhelix1.GetAngle(phase[1][0],mhelix,phase[1][1],fMCAngle);
 201     fMCRr = TMath::Sqrt(radius[1]);
 202   }
 203   //
 204   //
 205   fMCDist1 = TMath::Sqrt(distance1);
 206   fMCDist2 = TMath::Sqrt(distance2);
 207   //
 208   //
 209   //
 210   Float_t v[3] = {fMCXr[0]-vertex[0],fMCXr[1]-vertex[1],fMCXr[2]-vertex[2]};
 211   //Float_t v[3] = {fMCXr[0],fMCXr[1],fMCXr[2]};
 212   Float_t p[3] = {fMCPdr[0]+fMCPm[0], fMCPdr[1]+fMCPm[1],fMCPdr[2]+fMCPm[2]};
 213   Float_t vnorm2 = v[0]*v[0]+v[1]*v[1];
 214   Float_t vnorm3 = TMath::Sqrt(v[2]*v[2]+vnorm2);
 215   vnorm2 = TMath::Sqrt(vnorm2);
 216   Float_t pnorm2 = p[0]*p[0]+p[1]*p[1];
 217   Float_t pnorm3 = TMath::Sqrt(p[2]*p[2]+pnorm2);
 218   pnorm2 = TMath::Sqrt(pnorm2);
 219   //
 220   fPointAngleFi = (v[0]*p[0]+v[1]*p[1])/(vnorm2*pnorm2);
 221   fPointAngleTh = (v[2]*p[2]+vnorm2*pnorm2)/(vnorm3*pnorm3);
 222   fPointAngle   = (v[0]*p[0]+v[1]*p[1]+v[2]*p[2])/(vnorm3*pnorm3);
 223   Double_t mass1 = fMCd.GetMass();
 224   Double_t mass2 = fMCm.GetMass();
 225
 226
 227   Double_t e1    = TMath::Sqrt(mass1*mass1+
 228                               fMCPd[0]*fMCPd[0]+
 229                               fMCPd[1]*fMCPd[1]+
 230                               fMCPd[2]*fMCPd[2]);
 231   Double_t e2    = TMath::Sqrt(mass2*mass2+
 232                               fMCPm[0]*fMCPm[0]+
 233                               fMCPm[1]*fMCPm[1]+
 234                               fMCPm[2]*fMCPm[2]);
 235
 236   fInvMass =
 237     (fMCPm[0]+fMCPd[0])*(fMCPm[0]+fMCPd[0])+
 238     (fMCPm[1]+fMCPd[1])*(fMCPm[1]+fMCPd[1])+
 239     (fMCPm[2]+fMCPd[2])*(fMCPm[2]+fMCPd[2]);
 240
 241   //  fInvMass = TMath::Sqrt((e1+e2)*(e1+e2)-fInvMass);
 242   fInvMass = (e1+e2)*(e1+e2)-fInvMass;
 243   if (fInvMass>0) fInvMass = TMath::Sqrt(fInvMass);
 244 }
 245
 246
 247
 248
 249
 250
 251
 252