ITS/AliITSV0Finder.cxx

   1 /**************************************************************************\r
   2  * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *\r
   3  *                                                                        *\r
   4  * Author: The ALICE Off-line Project.                                    *\r
   5  * Contributors are mentioned in the code where appropriate.              *\r
   6  *                                                                        *\r
   7  * Permission to use, copy, modify and distribute this software and its   *\r
   8  * documentation strictly for non-commercial purposes is hereby granted   *\r
   9  * without fee, provided that the above copyright notice appears in all   *\r
  10  * copies and that both the copyright notice and this permission notice   *\r
  11  * appear in the supporting documentation. The authors make no claims     *\r
  12  * about the suitability of this software for any purpose. It is          *\r
  13  * provided "as is" without express or implied warranty.                  *\r
  14  **************************************************************************/\r
  15 /* $Id$ */\r
  16 \r
  17 //-------------------------------------------------------------------------\r
  18 //      Implementation of the on-the-fly v0 finder for ITS tracker MI\r
  19 //          Origin: Marian Ivanov, CERN, Marian.Ivanov@cern.ch \r
  20 //          Extraction from AliITStrackerMI: Andrea Dainese\r
  21 //          Current support and development: \r
  22 //-------------------------------------------------------------------------\r
  23 \r
  24 #include <TMatrixD.h>\r
  25 #include <TTree.h>\r
  26 #include <TString.h>\r
  27 #include <TRandom.h>\r
  28 #include <TTreeStream.h>\r
  29 \r
  30 \r
  31 #include "AliLog.h"\r
  32 #include "AliESDVertex.h"\r
  33 #include "AliESDEvent.h"\r
  34 #include "AliESDtrack.h"\r
  35 #include "AliV0.h"\r
  36 #include "AliHelix.h"\r
  37 #include "AliITSRecPoint.h"\r
  38 #include "AliITSReconstructor.h"\r
  39 #include "AliITStrackerMI.h"\r
  40 #include "AliITSV0Finder.h"\r
  41 #include "AliKFParticle.h"\r
  42 #include "AliKFVertex.h"\r
  43 \r
  44 ClassImp(AliITSV0Finder)\r
  45 \r
  46 \r
  47 AliITSV0Finder::AliITSV0Finder():\r
  48 fDebugStreamer(0)\r
  49  {\r
  50   //Default constructor\r
  51 \r
  52    if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0)\r
  53      fDebugStreamer = new TTreeSRedirector("ITSdebug.root");\r
  54 } \r
  55 \r
  56 //------------------------------------------------------------------------\r
  57 void AliITSV0Finder::UpdateTPCV0(const AliESDEvent *event,\r
  58                                    AliITStrackerMI *tracker)\r
  59 {\r
  60   //\r
  61   //try to update, or reject TPC  V0s\r
  62   //\r
  63   const Float_t kMinTgl2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl2();\r
  64 \r
  65   TObjArray *trackHypothesys = tracker->GetTrackHypothesys();\r
  66 \r
  67   Int_t nv0s = event->GetNumberOfV0s();\r
  68   Int_t nitstracks = trackHypothesys->GetEntriesFast();\r
  69 \r
  70   for (Int_t i=0;i<nv0s;i++){\r
  71     AliESDv0 * vertex = event->GetV0(i);\r
  72     Int_t ip = vertex->GetIndex(0);\r
  73     Int_t im = vertex->GetIndex(1);\r
  74     //\r
  75     TObjArray * arrayp = (ip<nitstracks) ? (TObjArray*)trackHypothesys->At(ip):0;\r
  76     TObjArray * arraym = (im<nitstracks) ? (TObjArray*)trackHypothesys->At(im):0;\r
  77     AliITStrackMI * trackp = (arrayp!=0) ? (AliITStrackMI*)arrayp->At(0):0;\r
  78     AliITStrackMI * trackm = (arraym!=0) ? (AliITStrackMI*)arraym->At(0):0;\r
  79     //\r
  80     //\r
  81     if (trackp){\r
  82       if (trackp->GetNumberOfClusters()+trackp->GetNDeadZone()>5.5){\r
  83         if (trackp->GetConstrain()&&trackp->GetChi2MIP(0)<3) vertex->SetStatus(-100);\r
  84         if (!trackp->GetConstrain()&&trackp->GetChi2MIP(0)<2) vertex->SetStatus(-100); \r
  85       }\r
  86     }\r
  87 \r
  88     if (trackm){\r
  89       if (trackm->GetNumberOfClusters()+trackm->GetNDeadZone()>5.5){\r
  90         if (trackm->GetConstrain()&&trackm->GetChi2MIP(0)<3) vertex->SetStatus(-100);\r
  91         if (!trackm->GetConstrain()&&trackm->GetChi2MIP(0)<2) vertex->SetStatus(-100); \r
  92       }\r
  93     }\r
  94     if (vertex->GetStatus()==-100) continue;\r
  95     //\r
  96     Double_t xrp[3]; vertex->GetXYZ(xrp[0],xrp[1],xrp[2]);  //I.B.\r
  97     Int_t clayer = tracker->GetNearestLayer(xrp);                    //I.B.\r
  98     vertex->SetNBefore(clayer);        //\r
  99     vertex->SetChi2Before(9*clayer);   //\r
 100     vertex->SetNAfter(6-clayer);       //\r
 101     vertex->SetChi2After(0);           //\r
 102     //\r
 103     if (clayer >1 ){ // calculate chi2 before vertex\r
 104       Float_t chi2p = 0, chi2m=0;  \r
 105       //\r
 106       if (trackp){\r
 107         for (Int_t ilayer=0;ilayer<clayer;ilayer++){\r
 108           if (trackp->GetClIndex(ilayer)>=0){\r
 109             chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+\r
 110               trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));\r
 111           }\r
 112           else{\r
 113             chi2p+=9;\r
 114           }\r
 115         }\r
 116       }else{\r
 117         chi2p = 9*clayer;\r
 118       }\r
 119       //\r
 120       if (trackm){\r
 121         for (Int_t ilayer=0;ilayer<clayer;ilayer++){\r
 122           if (trackm->GetClIndex(ilayer)>=0){\r
 123             chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+\r
 124               trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));\r
 125           }\r
 126           else{\r
 127             chi2m+=9;\r
 128           }\r
 129         }\r
 130       }else{\r
 131         chi2m = 9*clayer;\r
 132       }\r
 133       vertex->SetChi2Before(TMath::Min(chi2p,chi2m));\r
 134       if (TMath::Min(chi2p,chi2m)/Float_t(clayer)<4) vertex->SetStatus(-10);  // track exist before vertex\r
 135     }\r
 136     \r
 137     if (clayer < 5 ){ // calculate chi2 after vertex\r
 138       Float_t chi2p = 0, chi2m=0;  \r
 139       //\r
 140       if (trackp&&TMath::Abs(trackp->GetTgl())<kMinTgl2){\r
 141         for (Int_t ilayer=clayer;ilayer<6;ilayer++){\r
 142           if (trackp->GetClIndex(ilayer)>=0){\r
 143             chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+\r
 144               trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));\r
 145           }\r
 146           else{\r
 147             chi2p+=9;\r
 148           }\r
 149         }\r
 150       }else{\r
 151         chi2p = 0;\r
 152       }\r
 153       //\r
 154       if (trackm&&TMath::Abs(trackm->GetTgl())<kMinTgl2){\r
 155         for (Int_t ilayer=clayer;ilayer<6;ilayer++){\r
 156           if (trackm->GetClIndex(ilayer)>=0){\r
 157             chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+\r
 158               trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));\r
 159           }\r
 160           else{\r
 161             chi2m+=9;\r
 162           }\r
 163         }\r
 164       }else{\r
 165         chi2m = 0;\r
 166       }\r
 167       vertex->SetChi2After(TMath::Max(chi2p,chi2m));\r
 168       if (TMath::Max(chi2m,chi2p)/Float_t(6-clayer)>9) vertex->SetStatus(-20);  // track not found in ITS\r
 169     }\r
 170   }\r
 171   //\r
 172 }\r
 173 //------------------------------------------------------------------------\r
 174 void AliITSV0Finder::FindV02(AliESDEvent *event,\r
 175                                AliITStrackerMI *tracker) \r
 176 {\r
 177   //\r
 178   // V0 finder\r
 179   //\r
 180   //  Cuts on DCA -  R dependent\r
 181   //          max distance DCA between 2 tracks cut \r
 182   //          maxDist = TMath::Min(kMaxDist,kMaxDist0+pvertex->GetRr()*kMaxDist);\r
 183   //\r
 184   const Bool_t kCheckPropagate = kFALSE;\r
 185   const Float_t kMaxDist0 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist0();\r
 186   const Float_t kMaxDist1 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist1();\r
 187   const Float_t kMaxDist = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist();\r
 188   const Float_t kMinPointAngle = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle();\r
 189   const Float_t kMinPointAngle2 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle2();\r
 190   const Float_t kMinR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinR();\r
 191   const Float_t kMaxR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxR();\r
 192   const Float_t kMinPABestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPABestConst();\r
 193   const Float_t kMaxRBestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRBestConst();\r
 194   const Float_t kCausality0Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCausality0Cut();\r
 195   const Float_t kLikelihood01Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood01Cut();\r
 196   const Float_t kLikelihood1Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood1Cut();\r
 197   const Float_t kCombinedCut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCombinedCut();\r
 198   const Float_t kMinClFullTrk= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClFullTrk();\r
 199   const Float_t kMinTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl0();\r
 200   const Float_t kMinRTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTgl0();\r
 201 \r
 202   const Float_t kMinNormDistForbTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbTgl0();\r
 203   const Float_t kMinClForb0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClForb0();\r
 204   const Float_t kMinNormDistForb1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb1();\r
 205   const Float_t kMinNormDistForb2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb2();\r
 206   const Float_t kMinNormDistForb3= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb3();\r
 207   const Float_t kMinNormDistForb4= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb4();\r
 208   const Float_t kMinNormDistForb5= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb5();\r
 209   const Float_t kMinNormDistForbProt= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbProt();\r
 210   const Float_t kMaxPidProbPionForb= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxPidProbPionForb();\r
 211 \r
 212   const Float_t kMinRTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTPCdensity();\r
 213   const Float_t kMaxRTPCdensity0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity0();\r
 214   const Float_t kMaxRTPCdensity10= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity10();\r
 215   const Float_t kMaxRTPCdensity20= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity20();\r
 216   const Float_t kMaxRTPCdensity30= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity30();\r
 217 \r
 218 \r
 219   const Float_t kMinTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTPCdensity();\r
 220   const Float_t kMinTgl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl1();\r
 221 \r
 222   const Float_t kMinchi2before0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before0();\r
 223   const Float_t kMinchi2before1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before1();\r
 224   const Float_t kMinchi2after0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after0();\r
 225   const Float_t kMinchi2after1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after1();\r
 226   const Float_t kAddchi2SharedCl= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2SharedCl();\r
 227   const Float_t kAddchi2NegCl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl0();\r
 228   const Float_t kAddchi2NegCl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl1();\r
 229 \r
 230   const Float_t kSigp0Par0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par0();\r
 231   const Float_t kSigp0Par1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par1();\r
 232   const Float_t kSigp0Par2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par2();\r
 233 \r
 234   const Float_t kSigpPar0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar0();\r
 235   const Float_t kSigpPar1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar1();\r
 236   const Float_t kSigpPar2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar2();\r
 237   const Float_t kMaxDcaLh0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDcaLh0();\r
 238 \r
 239 \r
 240   TObjArray *trackHypothesys = tracker->GetTrackHypothesys();\r
 241   TObjArray *bestHypothesys  = tracker->GetBestHypothesys();\r
 242   TObjArray *originalTracks  = tracker->GetOriginal();\r
 243   //\r
 244   //\r
 245   TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());\r
 246   Int_t ntracks    = event->GetNumberOfTracks(); \r
 247   Int_t nitstracks = trackHypothesys->GetEntriesFast();\r
 248   originalTracks->Expand(ntracks);\r
 249   trackHypothesys->Expand(ntracks);\r
 250   bestHypothesys->Expand(ntracks);\r
 251   //\r
 252   AliHelix * helixes   = new AliHelix[ntracks+2];\r
 253   TObjArray trackarray(ntracks+2);     //array with tracks - with vertex constrain\r
 254   TObjArray trackarrayc(ntracks+2);    //array of "best    tracks" - without vertex constrain\r
 255   TObjArray trackarrayl(ntracks+2);    //array of "longest tracks" - without vertex constrain\r
 256   //Change from Bool_t to Int_t for optimization\r
 257   //  Int_t forbN=0;\r
 258   //  Int_t * forbidden   = new Int_t [ntracks+2];\r
 259   Bool_t * forbidden   = new Bool_t [ntracks+2];\r
 260   Int_t   *itsmap      = new Int_t  [ntracks+2];\r
 261   Float_t *dist        = new Float_t[ntracks+2];\r
 262   Float_t *normdist0   = new Float_t[ntracks+2];\r
 263   Float_t *normdist1   = new Float_t[ntracks+2];\r
 264   Float_t *normdist    = new Float_t[ntracks+2];\r
 265   Float_t *norm        = new Float_t[ntracks+2];\r
 266   Float_t *maxr        = new Float_t[ntracks+2];\r
 267   Float_t *minr        = new Float_t[ntracks+2];\r
 268   Float_t *minPointAngle= new Float_t[ntracks+2];\r
 269   //\r
 270   AliV0 *pvertex      = new AliV0;\r
 271   AliITStrackMI * dummy= new AliITStrackMI;\r
 272   dummy->SetLabel(0);\r
 273   AliITStrackMI  trackat0;    //temporary track for DCA calculation\r
 274   //\r
 275   Float_t primvertex[3]={tracker->GetX(),tracker->GetY(),tracker->GetZ()};\r
 276   //\r
 277   // make ITS -  ESD map\r
 278   //\r
 279   for (Int_t itrack=0;itrack<ntracks+2;itrack++) {\r
 280     itsmap[itrack]        = -1;\r
 281     //    forbidden[itrack]     = 0;\r
 282     forbidden[itrack]     = kFALSE;\r
 283     maxr[itrack]          = kMaxR;\r
 284     minr[itrack]          = kMinR;\r
 285     minPointAngle[itrack] = kMinPointAngle;\r
 286   }\r
 287   for (Int_t itrack=0;itrack<nitstracks;itrack++){\r
 288     AliITStrackMI * original =   (AliITStrackMI*)(originalTracks->At(itrack));\r
 289     Int_t           esdindex =   original->GetESDtrack()->GetID();\r
 290     itsmap[esdindex]         =   itrack;\r
 291   }\r
 292   //\r
 293   // create ITS tracks from ESD tracks if not done before\r
 294   //\r
 295   for (Int_t itrack=0;itrack<ntracks;itrack++){\r
 296     if (itsmap[itrack]>=0) continue;\r
 297     AliITStrackMI * tpctrack = new AliITStrackMI(*(event->GetTrack(itrack)));\r
 298     //tpctrack->fD[0] = tpctrack->GetD(GetX(),GetY());\r
 299     //tpctrack->fD[1] = tpctrack->GetZat(GetX())-GetZ(); \r
 300     tpctrack->GetDZ(tracker->GetX(),tracker->GetY(),tracker->GetZ(),tpctrack->GetDP());   //I.B.\r
 301     if (tpctrack->GetD(0)<20 && tpctrack->GetD(1)<20){\r
 302       // tracks which can reach inner part of ITS\r
 303       // propagate track to outer its volume - with correction for material\r
 304       AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(tpctrack);  \r
 305     }\r
 306     itsmap[itrack] = nitstracks;\r
 307     originalTracks->AddAt(tpctrack,nitstracks);\r
 308     nitstracks++;\r
 309   }\r
 310   //\r
 311   // fill temporary arrays\r
 312   //\r
 313   for (Int_t itrack=0;itrack<ntracks;itrack++){\r
 314     AliESDtrack *  esdtrack = event->GetTrack(itrack);\r
 315     Int_t          itsindex = itsmap[itrack];\r
 316     AliITStrackMI *original = (AliITStrackMI*)originalTracks->At(itsmap[itrack]);\r
 317     if (!original) continue;\r
 318     AliITStrackMI *bestConst  = 0;\r
 319     AliITStrackMI *bestLong   = 0;\r
 320     AliITStrackMI *best       = 0;    \r
 321     //\r
 322     //\r
 323     TObjArray * array    = (TObjArray*)  trackHypothesys->At(itsindex);\r
 324     Int_t       hentries = (array==0) ?  0 : array->GetEntriesFast();\r
 325     // Get best track with vertex constrain\r
 326     for (Int_t ih=0;ih<hentries;ih++){\r
 327       AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);\r
 328       if (!trackh->GetConstrain()) continue;\r
 329       if (!bestConst) bestConst = trackh;\r
 330       if (trackh->GetNumberOfClusters()>kMinClFullTrk ){\r
 331         bestConst  = trackh;                         // full track -  with minimal chi2\r
 332         break;\r
 333       }\r
 334       if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone())  continue;      \r
 335       bestConst = trackh;\r
 336       break;\r
 337     }\r
 338     // Get best long track without vertex constrain and best track without vertex constrain\r
 339     for (Int_t ih=0;ih<hentries;ih++){\r
 340       AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);\r
 341       if (trackh->GetConstrain()) continue;\r
 342       if (!best)     best     = trackh;\r
 343       if (!bestLong) bestLong = trackh;\r
 344       if (trackh->GetNumberOfClusters()>kMinClFullTrk){\r
 345         bestLong  = trackh;                         // full track -  with minimal chi2\r
 346         break;\r
 347       }\r
 348       if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone())  continue;      \r
 349       bestLong = trackh;        \r
 350     }\r
 351     if (!best) {\r
 352       best     = original;\r
 353       bestLong = original;\r
 354     }\r
 355     //I.B. trackat0 = *bestLong;\r
 356     new (&trackat0) AliITStrackMI(*bestLong);\r
 357     Double_t xx,yy,zz,alpha; \r
 358     if (!bestLong->GetGlobalXYZat(bestLong->GetX(),xx,yy,zz)) continue;\r
 359     \r
 360 \r
 361     alpha = TMath::ATan2(yy,xx);    \r
 362     //    if (!trackat0.Propagate(alpha,0)) continue;    \r
 363     //    trackat0.Propagate(alpha,0); //PH The check on the return value is temporarily disabled (bug 45751) \r
 364     if(!trackat0.Propagate(alpha,0) && kCheckPropagate)continue;\r
 365     // calculate normalized distances to the vertex \r
 366     //\r
 367     Float_t ptfac  = (1.+100.*TMath::Abs(trackat0.GetC()));\r
 368     if ( bestLong->GetNumberOfClusters()>3 ){      \r
 369       dist[itsindex]      = trackat0.GetY();\r
 370       norm[itsindex]      = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());\r
 371       normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);\r
 372       normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));\r
 373       normdist[itsindex]  = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
 374       if (!bestConst){\r
 375         if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<6) normdist[itsindex]*=2.;\r
 376         if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<5) normdist[itsindex]*=2.;\r
 377         if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<4) normdist[itsindex]*=2.;\r
 378       }else{\r
 379         if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<6) normdist[itsindex]*=1.5;\r
 380         if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<5) normdist[itsindex]*=1.5;\r
 381       }\r
 382     }\r
 383     else{      \r
 384       if (bestConst&&bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>4.5){\r
 385         dist[itsindex] = bestConst->GetD(0);\r
 386         norm[itsindex] = bestConst->GetDnorm(0);\r
 387         normdist0[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);\r
 388         normdist1[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);\r
 389         normdist[itsindex]  = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
 390       }else{\r
 391         dist[itsindex]      = trackat0.GetY();\r
 392         norm[itsindex]      = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());\r
 393         normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);\r
 394         normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));\r
 395         normdist[itsindex]  = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
 396         if (TMath::Abs(trackat0.GetTgl())>kMinTgl0){\r
 397           if (normdist[itsindex]<kMinNormDistForbTgl0){\r
 398             //      forbN=1;\r
 399             //      forbidden[itsindex]+=1<<forbN;\r
 400             forbidden[itsindex]=kTRUE;\r
 401           }\r
 402           if (normdist[itsindex]>kMinNormDistForbTgl0) {\r
 403             minr[itsindex] = TMath::Max(kMinRTgl0,minr[itsindex]);\r
 404           }\r
 405         }\r
 406       }\r
 407     }\r
 408 \r
 409 \r
 410     //\r
 411     //-----------------------------------------------------------\r
 412     //Forbid primary track candidates - \r
 413     //\r
 414     //treetr->SetAlias("forbidden0","Tr0.fN<4&&Tr1.fN+Tr1.fNDeadZone>4.5");\r
 415     //treetr->SetAlias("forbidden1","ND<3&&Tr1.fN+Tr1.fNDeadZone>5.5");\r
 416     //treetr->SetAlias("forbidden2","ND<2&&Tr1.fClIndex[0]>0&&Tr1.fClIndex[0]>0");\r
 417     //treetr->SetAlias("forbidden3","ND<1&&Tr1.fClIndex[0]>0");\r
 418     //treetr->SetAlias("forbidden4","ND<4&&Tr1.fNormChi2[0]<2");\r
 419     //treetr->SetAlias("forbidden5","ND<5&&Tr1.fNormChi2[0]<1");\r
 420     //-----------------------------------------------------------\r
 421     if (bestConst){\r
 422       if (bestLong->GetNumberOfClusters()<4 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>kMinClForb0){\r
 423         //      forbN=2;\r
 424         //      forbidden[itsindex]+=1<<forbN;\r
 425         forbidden[itsindex]=kTRUE;\r
 426       }\r
 427       if (normdist[itsindex]<kMinNormDistForb1 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>5.5){\r
 428         //      forbN=3;\r
 429         //      forbidden[itsindex]+=1<<forbN;\r
 430         forbidden[itsindex]=kTRUE;\r
 431       }\r
 432       if (normdist[itsindex]<kMinNormDistForb2 && bestConst->GetClIndex(0)>=0 && bestConst->GetClIndex(1)>=0 ){\r
 433         //      forbN=4;\r
 434         //      forbidden[itsindex]+=1<<forbN;\r
 435         forbidden[itsindex]=kTRUE;\r
 436       }\r
 437       if (normdist[itsindex]<kMinNormDistForb3 && bestConst->GetClIndex(0)>=0){\r
 438         //      forbN=5;\r
 439         //      forbidden[itsindex]+=1<<forbN;\r
 440         forbidden[itsindex]=kTRUE;\r
 441       }\r
 442       if (normdist[itsindex]<kMinNormDistForb4 && bestConst->GetNormChi2(0)<2){\r
 443         //      forbN=6;\r
 444         //      forbidden[itsindex]+=1<<forbN;\r
 445         forbidden[itsindex]=kTRUE;\r
 446       }\r
 447       if (normdist[itsindex]<kMinNormDistForb5 && bestConst->GetNormChi2(0)<1){\r
 448         //      forbN=7;\r
 449         //      forbidden[itsindex]+=1<<forbN;\r
 450         forbidden[itsindex]=kTRUE;\r
 451       }\r
 452       if (bestConst->GetNormChi2(0)<2.5) {\r
 453         minPointAngle[itsindex]= kMinPABestConst;\r
 454         maxr[itsindex]         = kMaxRBestConst;\r
 455       }\r
 456     }\r
 457     //\r
 458     //forbid daughter kink candidates\r
 459     //\r
 460     if (esdtrack->GetKinkIndex(0)>0) forbidden[itsindex] = kTRUE;\r
 461     Bool_t isElectron = kTRUE;\r
 462     Bool_t isProton   = kTRUE;\r
 463     Double_t pid[5];\r
 464     esdtrack->GetESDpid(pid);\r
 465     for (Int_t i=1;i<5;i++){\r
 466       if (pid[0]<pid[i]) isElectron= kFALSE;\r
 467       if (pid[4]<pid[i]) isProton= kFALSE;\r
 468     }\r
 469     if (isElectron){\r
 470       forbidden[itsindex]=kFALSE;       \r
 471       normdist[itsindex]*=-1;\r
 472     }\r
 473     if (isProton){\r
 474       if (normdist[itsindex]>kMinNormDistForbProt) forbidden[itsindex]=kFALSE;  \r
 475       normdist[itsindex]*=-1;\r
 476     }\r
 477 \r
 478     // We allow all tracks that are not pions\r
 479     if( (pid[1]+pid[2])< kMaxPidProbPionForb ){\r
 480       forbidden[itsindex]=kFALSE;\r
 481     }\r
 482 \r
 483     //\r
 484     // Causality cuts in TPC volume\r
 485     //\r
 486     if (esdtrack->GetTPCdensity(0,10) >kMinTPCdensity)  maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity0),maxr[itsindex]);\r
 487     if (esdtrack->GetTPCdensity(10,30)>kMinTPCdensity)  maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity10),maxr[itsindex]);\r
 488     if (esdtrack->GetTPCdensity(20,40)>kMinTPCdensity)  maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity20),maxr[itsindex]);\r
 489     if (esdtrack->GetTPCdensity(30,50)>kMinTPCdensity)  maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity30),maxr[itsindex]);\r
 490     //\r
 491     if (esdtrack->GetTPCdensity(0,60)<0.4&&bestLong->GetNumberOfClusters()<3) minr[itsindex]=kMinRTPCdensity;    \r
 492     //\r
 493     //\r
 494 \r
 495     if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){\r
 496       cstream<<"Track"<<\r
 497         "Tr0.="<<best<<\r
 498         "Tr1.="<<((bestConst)? bestConst:dummy)<<\r
 499         "Tr2.="<<bestLong<<\r
 500         "Tr3.="<<&trackat0<<\r
 501         "Esd.="<<esdtrack<<\r
 502         "Dist="<<dist[itsindex]<<\r
 503         "ND0="<<normdist0[itsindex]<<\r
 504         "ND1="<<normdist1[itsindex]<<\r
 505         "ND="<<normdist[itsindex]<<\r
 506         "Pz="<<primvertex[2]<<\r
 507         "Forbid="<<forbidden[itsindex]<<\r
 508         "\n";\r
 509       //\r
 510     }\r
 511     trackarray.AddAt(best,itsindex);\r
 512     trackarrayc.AddAt(bestConst,itsindex);\r
 513     trackarrayl.AddAt(bestLong,itsindex);\r
 514     new (&helixes[itsindex]) AliHelix(*best);\r
 515   }\r
 516   //\r
 517   //\r
 518   //\r
 519   // first iterration of V0 finder\r
 520   //\r
 521   // AM Comment out for optimization\r
 522   //  Int_t rejectBase=0;\r
 523   //  Int_t cutN=0;\r
 524 \r
 525   for (Int_t iesd0=0;iesd0<ntracks;iesd0++){\r
 526     Int_t itrack0 = itsmap[iesd0];\r
 527     //-AM comment for optimization and store the forbidden value in the debug streamer\r
 528     if (forbidden[itrack0]) continue;\r
 529     AliITStrackMI * btrack0 = (AliITStrackMI*)trackarray.At(itrack0);\r
 530     if (!btrack0) continue;    \r
 531     AliITStrackMI *trackc0 = (AliITStrackMI*)trackarrayc.At(itrack0);\r
 532     //\r
 533     for (Int_t iesd1=iesd0+1;iesd1<ntracks;iesd1++){\r
 534       Int_t itrack1 = itsmap[iesd1];\r
 535       //-AM comment for optimization and store the forbidden value in the debug streamer\r
 536       if (forbidden[itrack1]) continue;\r
 537 \r
 538       AliITStrackMI * btrack1 = (AliITStrackMI*)trackarray.At(itrack1); \r
 539       if (!btrack1) continue;\r
 540 \r
 541       if ( (btrack0->GetSign()==btrack1->GetSign()) && !AliITSReconstructor::GetRecoParam()->GetStoreLikeSignV0s()) continue;\r
 542 \r
 543       Bool_t isGold = kFALSE;\r
 544       if (TMath::Abs(TMath::Abs(btrack0->GetLabel())-TMath::Abs(btrack1->GetLabel()))==1){\r
 545         isGold = kTRUE;\r
 546       }\r
 547       //      rejectBase=0;\r
 548       AliITStrackMI *trackc1 = (AliITStrackMI*)trackarrayc.At(itrack1);\r
 549       AliHelix &h1 = helixes[itrack0];\r
 550       AliHelix &h2 = helixes[itrack1];\r
 551       //\r
 552       // find linear distance\r
 553       Double_t rmin =0;\r
 554       //\r
 555       //\r
 556       //\r
 557       Double_t phase[2][2],radius[2];\r
 558       Int_t  points = h1.GetRPHIintersections(h2, phase, radius);\r
 559       if    (points==0) {\r
 560         //      cutN=1;\r
 561         //      rejectBase+=1<<cutN;\r
 562         continue;\r
 563       }\r
 564       Double_t delta[2]={1000000,1000000};        \r
 565       rmin = radius[0];\r
 566       h1.ParabolicDCA(h2,phase[0][0],phase[0][1],radius[0],delta[0]);\r
 567       if (points==2){    \r
 568         if (radius[1]<rmin) rmin = radius[1];\r
 569         h1.ParabolicDCA(h2,phase[1][0],phase[1][1],radius[1],delta[1]);\r
 570       }\r
 571       rmin = TMath::Sqrt(rmin);\r
 572       Double_t distance = 0;\r
 573       Double_t radiusC  = 0;\r
 574       Int_t    iphase   = 0;\r
 575       if (points==1 || delta[0]<delta[1]){\r
 576         distance = TMath::Sqrt(delta[0]);\r
 577         radiusC  = TMath::Sqrt(radius[0]);\r
 578       }else{\r
 579         distance = TMath::Sqrt(delta[1]);\r
 580         radiusC  = TMath::Sqrt(radius[1]);\r
 581         iphase=1;\r
 582       }\r
 583       if (radiusC<TMath::Max(minr[itrack0],minr[itrack1])){\r
 584         //      cutN=2;\r
 585         //rejectBase+=1<<cutN;\r
 586         continue;\r
 587       }\r
 588       if (radiusC>TMath::Min(maxr[itrack0],maxr[itrack1])){\r
 589         //      cutN=3;\r
 590         //      rejectBase+=1<<cutN;\r
 591         continue;\r
 592       } \r
 593       Float_t maxDist  = TMath::Min(kMaxDist,Float_t(kMaxDist0+radiusC*kMaxDist1));      \r
 594       if (distance>maxDist){\r
 595         //      cutN=4;\r
 596         //      rejectBase+=1<<cutN;\r
 597         continue;\r
 598       }\r
 599       Float_t pointAngle = h1.GetPointAngle(h2,phase[iphase],primvertex);\r
 600       if (pointAngle<TMath::Max(minPointAngle[itrack0],minPointAngle[itrack1])) {\r
 601         //      cutN=5;\r
 602         //      rejectBase+=1<<cutN;\r
 603         continue;\r
 604       }\r
 605       //\r
 606       //\r
 607       //      Double_t distance = TestV0(h1,h2,pvertex,rmin);\r
 608       //\r
 609       //       if (distance>maxDist)           continue;\r
 610       //       if (pvertex->GetRr()<kMinR)     continue;\r
 611       //       if (pvertex->GetRr()>kMaxR)     continue;\r
 612       AliITStrackMI * track0=btrack0;\r
 613       AliITStrackMI * track1=btrack1;\r
 614       //      if (pvertex->GetRr()<3.5){  \r
 615       if (radiusC<3.5){  \r
 616         //use longest tracks inside the pipe\r
 617         track0 = (AliITStrackMI*)trackarrayl.At(itrack0);\r
 618         track1 = (AliITStrackMI*)trackarrayl.At(itrack1);\r
 619       }      \r
 620       //\r
 621       //\r
 622       \r
 623 \r
 624  \r
 625       pvertex->SetParamN(*track0);\r
 626       pvertex->SetParamP(*track1);\r
 627       pvertex->Update(primvertex);\r
 628       pvertex->SetClusters(track0->ClIndex(),track1->ClIndex());  // register clusters\r
 629 \r
 630       // Define gamma, K0, lambda and lambda_bar from the decay particles and calculate the chi2      \r
 631       AliKFVertex vertexKF;\r
 632 \r
 633       vertexKF.X() = tracker->GetX();\r
 634       vertexKF.Y() = tracker->GetY();\r
 635       vertexKF.Z() = tracker->GetZ();\r
 636       vertexKF.Covariance(0,0) = tracker->GetSigmaX()*tracker->GetSigmaX();\r
 637       vertexKF.Covariance(1,2) = tracker->GetSigmaY()*tracker->GetSigmaY();\r
 638       vertexKF.Covariance(2,2) = tracker->GetSigmaZ()*tracker->GetSigmaZ();\r
 639       \r
 640       AliKFParticle elecKF( *(pvertex->GetParamN()) ,11);\r
 641       AliKFParticle posiKF( *(pvertex->GetParamP()) ,-11);\r
 642       AliKFParticle pipKF( *(pvertex->GetParamN()) , 211);\r
 643       AliKFParticle pinKF( *(pvertex->GetParamP()) ,-211);\r
 644       AliKFParticle protKF( *(pvertex->GetParamP()) ,2212);\r
 645       AliKFParticle aproKF ( *(pvertex->GetParamN()) ,-2212);\r
 646 \r
 647 \r
 648       // Gamma\r
 649       AliKFParticle gamKF(elecKF,posiKF);\r
 650       gamKF.SetProductionVertex(vertexKF);\r
 651 \r
 652       Float_t gamKFchi2 = 1000;\r
 653       if ( gamKF.GetNDF()!=0 ){\r
 654         gamKFchi2 = gamKF.GetChi2()/gamKF.GetNDF();\r
 655       }\r
 656 \r
 657       Float_t massG=0.;\r
 658       Float_t sigmaMG=0.001;\r
 659       gamKF.SetMassConstraint(massG,sigmaMG);\r
 660 \r
 661       Float_t gamKFchi2C = 1000;\r
 662       if ( gamKF.GetNDF()!=0 ){\r
 663         gamKFchi2C = gamKF.GetChi2()/gamKF.GetNDF();\r
 664       }\r
 665 \r
 666       //K0 short\r
 667       AliKFParticle k0KF(pipKF,pinKF);\r
 668       k0KF.SetProductionVertex(vertexKF);\r
 669 \r
 670       Float_t k0KFchi2 = 1000;\r
 671       if ( k0KF.GetNDF()!=0 ){\r
 672         k0KFchi2 = k0KF.GetChi2()/k0KF.GetNDF();\r
 673       }\r
 674 \r
 675       //Lambda\r
 676       AliKFParticle lambdaKF(protKF,pinKF);\r
 677       lambdaKF.SetProductionVertex(vertexKF);\r
 678 \r
 679       Float_t lambdaKFchi2 = 1000;\r
 680       if ( lambdaKF.GetNDF()!=0 ){\r
 681         lambdaKFchi2 = lambdaKF.GetChi2()/lambdaKF.GetNDF();\r
 682       }\r
 683 \r
 684       //Lambda_bar\r
 685       AliKFParticle alambKF(aproKF,pipKF);\r
 686       alambKF.SetProductionVertex(vertexKF);\r
 687 \r
 688       Float_t alambKFchi2 = 1000;\r
 689       if ( alambKF.GetNDF()!=0 ){\r
 690         alambKFchi2 = alambKF.GetChi2()/alambKF.GetNDF();\r
 691       }\r
 692 \r
 693 \r
 694 \r
 695 \r
 696 \r
 697       if (pvertex->GetRr()<kMinR){\r
 698         //      cutN=6;\r
 699         //      rejectBase+=1<<cutN;\r
 700         continue;\r
 701       }\r
 702       if (pvertex->GetRr()>kMaxR){\r
 703         //      cutN=7;\r
 704         //      rejectBase+=1<<cutN;\r
 705         continue;\r
 706       }\r
 707       if (pvertex->GetV0CosineOfPointingAngle()<kMinPointAngle){\r
 708         //      cutN=8;\r
 709         //      rejectBase+=1<<cutN;\r
 710         continue;\r
 711       }\r
 712 //Bo:      if (pvertex->GetDist2()>maxDist) continue;\r
 713 \r
 714       if (pvertex->GetDcaV0Daughters()>maxDist){\r
 715         //      cutN=9;\r
 716         //      rejectBase+=1<<cutN;\r
 717         continue;\r
 718       }\r
 719 //Bo:        pvertex->SetLab(0,track0->GetLabel());\r
 720 //Bo:        pvertex->SetLab(1,track1->GetLabel());\r
 721       pvertex->SetIndex(0,track0->GetESDtrack()->GetID());\r
 722       pvertex->SetIndex(1,track1->GetESDtrack()->GetID());\r
 723       //      \r
 724       AliITStrackMI * htrackc0 = trackc0 ? trackc0:dummy;      \r
 725       AliITStrackMI * htrackc1 = trackc1 ? trackc1:dummy;\r
 726 \r
 727       //\r
 728       //\r
 729       TObjArray * array0b     = (TObjArray*)bestHypothesys->At(itrack0);\r
 730       if (!array0b&&pvertex->GetRr()<40 && TMath::Abs(track0->GetTgl())<kMinTgl1) {\r
 731         tracker->SetCurrentEsdTrack(itrack0);\r
 732         tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack0),itrack0, kFALSE);\r
 733       }\r
 734       TObjArray * array1b    = (TObjArray*)bestHypothesys->At(itrack1);\r
 735       if (!array1b&&pvertex->GetRr()<40 && TMath::Abs(track1->GetTgl())<kMinTgl1) { \r
 736         tracker->SetCurrentEsdTrack(itrack1);\r
 737         tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack1),itrack1, kFALSE);\r
 738       }\r
 739       //\r
 740       AliITStrackMI * track0b = (AliITStrackMI*)originalTracks->At(itrack0);\r
 741       AliITStrackMI * track1b = (AliITStrackMI*)originalTracks->At(itrack1);\r
 742       AliITStrackMI * track0l = (AliITStrackMI*)originalTracks->At(itrack0);\r
 743       AliITStrackMI * track1l = (AliITStrackMI*)originalTracks->At(itrack1);\r
 744       \r
 745       Float_t minchi2before0=kMinchi2before0;\r
 746       Float_t minchi2before1=kMinchi2before1;\r
 747       Float_t minchi2after0 =kMinchi2after0;\r
 748       Float_t minchi2after1 =kMinchi2after1;\r
 749       Double_t xrp[3]; pvertex->GetXYZ(xrp[0],xrp[1],xrp[2]);  //I.B.\r
 750       Int_t maxLayer = tracker->GetNearestLayer(xrp);                   //I.B.\r
 751       \r
 752       if (array0b) for (Int_t i=0;i<5;i++){\r
 753         // best track after vertex\r
 754         AliITStrackMI * btrack = (AliITStrackMI*)array0b->At(i);\r
 755         if (!btrack) continue;\r
 756         if (btrack->GetNumberOfClusters()>track0l->GetNumberOfClusters()) track0l = btrack;     \r
 757         //      if (btrack->fX<pvertex->GetRr()-2.-0.5/(0.1+pvertex->GetAnglep()[2])) {\r
 758         if (btrack->GetX()<pvertex->GetRr()-2.) {\r
 759           if ( (maxLayer>i+2|| (i==0)) && btrack->GetNumberOfClusters()==(6-i)&&i<3){\r
 760             Float_t sumchi2= 0;\r
 761             Float_t sumn   = 0;\r
 762             if (maxLayer<3){   // take prim vertex as additional measurement\r
 763               if (normdist[itrack0]>0 && htrackc0){\r
 764                 sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack0]*normdist[itrack0],16.);\r
 765               }else{\r
 766                 sumchi2 +=  TMath::Min((3.-maxLayer)*(3*normdist[itrack0]*normdist[itrack0]+3.),16.);\r
 767               }\r
 768               sumn    +=  3-maxLayer;\r
 769             }\r
 770             for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){\r
 771               sumn+=1.;       \r
 772               if (btrack->GetClIndex(ilayer)<0){\r
 773                 sumchi2+=kAddchi2NegCl0;\r
 774                 continue;\r
 775               }else{\r
 776                 Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);\r
 777                 for (Int_t itrack=0;itrack<4;itrack++){\r
 778                   AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);\r
 779                   if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack0){\r
 780                     sumchi2+=kAddchi2SharedCl;  //shared cluster\r
 781                     break;\r
 782                   }\r
 783                 }\r
 784                 sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));\r
 785                 sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer));            \r
 786               }\r
 787             }\r
 788             sumchi2/=sumn;\r
 789             if (sumchi2<minchi2before0) minchi2before0=sumchi2; \r
 790           }\r
 791           continue;   //safety space - Geo manager will give exact layer\r
 792         }\r
 793         track0b       = btrack;\r
 794         minchi2after0 = btrack->GetNormChi2(i);\r
 795         break;\r
 796       }\r
 797       if (array1b) for (Int_t i=0;i<5;i++){\r
 798         // best track after vertex\r
 799         AliITStrackMI * btrack = (AliITStrackMI*)array1b->At(i);\r
 800         if (!btrack) continue;\r
 801         if (btrack->GetNumberOfClusters()>track1l->GetNumberOfClusters()) track1l = btrack;     \r
 802         //      if (btrack->fX<pvertex->GetRr()-2-0.5/(0.1+pvertex->GetAnglep()[2])){\r
 803         if (btrack->GetX()<pvertex->GetRr()-2){\r
 804           if ((maxLayer>i+2 || (i==0))&&btrack->GetNumberOfClusters()==(6-i)&&(i<3)){\r
 805             Float_t sumchi2= 0;\r
 806             Float_t sumn   = 0;\r
 807             if (maxLayer<3){   // take prim vertex as additional measurement\r
 808               if (normdist[itrack1]>0 && htrackc1){\r
 809                 sumchi2 +=  TMath::Min((3.-maxLayer)*normdist[itrack1]*normdist[itrack1],16.);\r
 810               }else{\r
 811                 sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack1]*normdist[itrack1]+3.),16.);\r
 812               }\r
 813               sumn    +=  3-maxLayer;\r
 814             }\r
 815             for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){\r
 816               sumn+=1.;\r
 817               if (btrack->GetClIndex(ilayer)<0){\r
 818                 sumchi2+=kAddchi2NegCl1;\r
 819                 continue;\r
 820               }else{\r
 821                 Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);\r
 822                 for (Int_t itrack=0;itrack<4;itrack++){\r
 823                   AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);\r
 824                   if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack1){\r
 825                     sumchi2+=kAddchi2SharedCl;  //shared cluster\r
 826                     break;\r
 827                   }\r
 828                 }\r
 829                 sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));\r
 830                 sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer));            \r
 831               }\r
 832             }\r
 833             sumchi2/=sumn;\r
 834             if (sumchi2<minchi2before1) minchi2before1=sumchi2; \r
 835           }\r
 836           continue;   //safety space - Geo manager will give exact layer           \r
 837         }\r
 838         track1b = btrack;\r
 839         minchi2after1 = btrack->GetNormChi2(i);\r
 840         break;\r
 841       }\r
 842       //\r
 843       // position resolution - used for DCA cut\r
 844       Float_t sigmad = track0b->GetSigmaY2()+track0b->GetSigmaZ2()+track1b->GetSigmaY2()+track1b->GetSigmaZ2()+\r
 845         (track0b->GetX()-pvertex->GetRr())*(track0b->GetX()-pvertex->GetRr())*(track0b->GetSigmaSnp2()+track0b->GetSigmaTgl2())+\r
 846         (track1b->GetX()-pvertex->GetRr())*(track1b->GetX()-pvertex->GetRr())*(track1b->GetSigmaSnp2()+track1b->GetSigmaTgl2());\r
 847       sigmad =TMath::Sqrt(sigmad)+0.04;\r
 848       if (pvertex->GetRr()>50){\r
 849         Double_t cov0[15],cov1[15];\r
 850         track0b->GetESDtrack()->GetInnerExternalCovariance(cov0);\r
 851         track1b->GetESDtrack()->GetInnerExternalCovariance(cov1);\r
 852         sigmad = cov0[0]+cov0[2]+cov1[0]+cov1[2]+\r
 853           (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov0[5]+cov0[9])+\r
 854           (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov1[5]+cov1[9]);\r
 855         sigmad =TMath::Sqrt(sigmad)+0.05;\r
 856       }\r
 857       //       \r
 858       AliV0 vertex2;\r
 859       vertex2.SetParamN(*track0b);\r
 860       vertex2.SetParamP(*track1b);\r
 861       vertex2.Update(primvertex);\r
 862       //Bo:      if (vertex2.GetDist2()<=pvertex->GetDist2()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){\r
 863       if (vertex2.GetDcaV0Daughters()<=pvertex->GetDcaV0Daughters()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){\r
 864         pvertex->SetParamN(*track0b);\r
 865         pvertex->SetParamP(*track1b);\r
 866         pvertex->Update(primvertex);\r
 867         pvertex->SetClusters(track0b->ClIndex(),track1b->ClIndex());  // register clusters\r
 868         pvertex->SetIndex(0,track0->GetESDtrack()->GetID());\r
 869         pvertex->SetIndex(1,track1->GetESDtrack()->GetID());\r
 870       }\r
 871       pvertex->SetDistSigma(sigmad);\r
 872       //Bo:      pvertex->SetDistNorm(pvertex->GetDist2()/sigmad);       \r
 873       pvertex->SetNormDCAPrim(normdist[itrack0],normdist[itrack1]);\r
 874       //\r
 875       // define likelihhod and causalities\r
 876       //\r
 877       Float_t pa0=1, pa1=1, pb0=0.26, pb1=0.26;      \r
 878       if (maxLayer<1){\r
 879         Float_t fnorm0 = normdist[itrack0];\r
 880         if (fnorm0<0) fnorm0*=-3;\r
 881         Float_t fnorm1 = normdist[itrack1];\r
 882         if (fnorm1<0) fnorm1*=-3;\r
 883         if ((pvertex->GetAnglep()[2]>0.1) || ( (pvertex->GetRr()<10.5)&& pvertex->GetAnglep()[2]>0.05 ) || (pvertex->GetRr()<3)){\r
 884           pb0    =  TMath::Exp(-TMath::Min(fnorm0,Float_t(16.))/12.);\r
 885           pb1    =  TMath::Exp(-TMath::Min(fnorm1,Float_t(16.))/12.);\r
 886         }\r
 887         pvertex->SetChi2Before(normdist[itrack0]);\r
 888         pvertex->SetChi2After(normdist[itrack1]);       \r
 889         pvertex->SetNAfter(0);\r
 890         pvertex->SetNBefore(0);\r
 891       }else{\r
 892         pvertex->SetChi2Before(minchi2before0);\r
 893         pvertex->SetChi2After(minchi2before1);\r
 894          if (pvertex->GetAnglep()[2]>0.1 || ( pvertex->GetRr()<10.5 && pvertex->GetAnglep()[2]>0.05) || pvertex->GetRr()<3){\r
 895            pb0    =  TMath::Exp(-TMath::Min(minchi2before0,Float_t(16))/12.);\r
 896            pb1    =  TMath::Exp(-TMath::Min(minchi2before1,Float_t(16))/12.);\r
 897          }\r
 898          pvertex->SetNAfter(maxLayer);\r
 899          pvertex->SetNBefore(maxLayer);      \r
 900       }\r
 901       if (pvertex->GetRr()<90){\r
 902         pa0  *= TMath::Min(track0->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));\r
 903         pa1  *= TMath::Min(track1->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));\r
 904       }\r
 905       if (pvertex->GetRr()<20){\r
 906         pa0  *= (0.2+TMath::Exp(-TMath::Min(minchi2after0,Float_t(16))/8.))/1.2;\r
 907         pa1  *= (0.2+TMath::Exp(-TMath::Min(minchi2after1,Float_t(16))/8.))/1.2;\r
 908       }\r
 909       //\r
 910       pvertex->SetCausality(pb0,pb1,pa0,pa1);\r
 911       //\r
 912       //  Likelihood calculations  - apply cuts\r
 913       //         \r
 914  \r
 915       // AM - Comment out for optimization and store the v0OK value\r
 916       //      Int_t v0OK = 0;\r
 917       //      Int_t cutOKN=0;\r
 918       Bool_t v0OK = kTRUE;\r
 919 \r
 920 \r
 921       Float_t p12 = pvertex->GetParamP()->GetParameter()[4]*pvertex->GetParamP()->GetParameter()[4];\r
 922       p12        += pvertex->GetParamN()->GetParameter()[4]*pvertex->GetParamN()->GetParameter()[4];\r
 923       p12         = TMath::Sqrt(p12);                             // "mean" momenta\r
 924 \r
 925       Float_t    sigmap0   = kSigp0Par0+kSigp0Par1/(kSigp0Par2+pvertex->GetRr()); \r
 926       Float_t    sigmap    = kSigpPar0*sigmap0*(kSigpPar1+kSigpPar2*p12);           // "resolution: of point angle - as a function of radius and momenta\r
 927 \r
 928 \r
 929       Float_t causalityA  = (1.0-pvertex->GetCausalityP()[0])*(1.0-pvertex->GetCausalityP()[1]);\r
 930       Float_t causalityB  = TMath::Sqrt(TMath::Min(pvertex->GetCausalityP()[2],Double_t(0.7))*\r
 931                                         TMath::Min(pvertex->GetCausalityP()[3],Double_t(0.7)));\r
 932       //\r
 933       //Bo:      Float_t likelihood0 = (TMath::Exp(-pvertex->GetDistNorm())+0.1) *(pvertex->GetDist2()<0.5)*(pvertex->GetDistNorm()<5);\r
 934       Float_t lDistNorm = pvertex->GetDcaV0Daughters()/pvertex->GetDistSigma();\r
 935       Float_t likelihood0 = (TMath::Exp(-lDistNorm)+0.1) *(pvertex->GetDcaV0Daughters()<kMaxDcaLh0)*(lDistNorm<5);\r
 936 \r
 937       Float_t likelihood1 = TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/sigmap)+\r
 938         0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(4.*sigmap))+\r
 939         0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(8.*sigmap))+\r
 940         0.1*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/0.01);\r
 941       //\r
 942       if (causalityA<kCausality0Cut){\r
 943         //      cutOKN=1;\r
 944         //      v0OK += 1<<cutOKN;\r
 945         v0OK = kFALSE;\r
 946       }\r
 947       if (TMath::Sqrt(likelihood0*likelihood1)<kLikelihood01Cut){\r
 948         //      cutOKN=2;\r
 949         //      v0OK += 1<<cutOKN;\r
 950         v0OK = kFALSE;\r
 951       }\r
 952       if (likelihood1<kLikelihood1Cut){\r
 953         //      cutOKN=3;\r
 954         //      v0OK += 1<<cutOKN;\r
 955         v0OK = kFALSE;\r
 956       }\r
 957       if (TMath::Power(likelihood0*likelihood1*causalityB,0.33)<kCombinedCut){\r
 958         //      cutOKN=4;\r
 959         //      v0OK += 1<<cutOKN;\r
 960         v0OK = kFALSE;\r
 961       }\r
 962       //\r
 963       //\r
 964       if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){      \r
 965         Bool_t gold = TMath::Abs(TMath::Abs(track0->GetLabel())-TMath::Abs(track1->GetLabel()))==1;\r
 966         cstream<<"It0"<<\r
 967           "Tr0.="<<track0<<                       //best without constrain\r
 968           "Tr1.="<<track1<<                       //best without constrain  \r
 969           "posiKF.="<<&posiKF<<                       //KF from track0\r
 970           "elecKF.="<<&elecKF<<                       //KF from track1\r
 971           "Tr0B.="<<track0b<<                     //best without constrain  after vertex\r
 972           "Tr1B.="<<track1b<<                     //best without constrain  after vertex \r
 973           "Tr0C.="<<htrackc0<<                    //best with constrain     if exist\r
 974           "Tr1C.="<<htrackc1<<                    //best with constrain     if exist\r
 975           "Tr0L.="<<track0l<<                     //longest best           \r
 976           "Tr1L.="<<track1l<<                     //longest best\r
 977           "Esd0.="<<track0->GetESDtrack()<<           // esd track0 params\r
 978           "Esd1.="<<track1->GetESDtrack()<<           // esd track1 params\r
 979           "V0.="<<pvertex<<                       //vertex properties\r
 980           "V0b.="<<&vertex2<<                       //vertex properties at "best" track\r
 981           "gamKF.="<<&gamKF<<                          //KF from pvertex\r
 982           "k0KF.="<<&k0KF<<                          //KF from pvertex\r
 983           "lambdaKF.="<<&lambdaKF<<                          //KF from pvertex\r
 984           "alambKF.="<<&lambdaKF<<                          //KF from pvertex\r
 985           "gamKFchi2="<<gamKFchi2<<    //Normalized chi2 from KF assuming gamma momther\r
 986           "gamKFchi2C="<<gamKFchi2C<<    //Normalized chi2 from KF assuming gamma mother+mass constraint\r
 987           "k0KFchi2="<<k0KFchi2<<    //Normalized chi2 from KF assuming K0 mother\r
 988           "lambdaKFchi2="<<lambdaKFchi2<<    //Normalized chi2 from KF assuming Lambda mother\r
 989           "alambKFchi2="<<alambKFchi2<<    //Normalized chi2 from KF assuming lambda_bar mother\r
 990           "ND0="<<normdist[itrack0]<<             //normalize distance for track0\r
 991           "ND1="<<normdist[itrack1]<<             //normalize distance for track1\r
 992           "Gold="<<gold<<                         //\r
 993           // "RejectBase="<<rejectBase<<             //rejection in First itteration\r
 994           "OK="<<v0OK<<\r
 995           "rmin="<<rmin<<\r
 996           "sigmad="<<sigmad<<\r
 997           "Forbid0="<<forbidden[itrack0]<<\r
 998           "Forbid1="<<forbidden[itrack1]<<\r
 999           "\n";\r
1000       }      \r
1001       //      if (rejectBase>0) continue;\r
1002       //if (forbidden[itrack0]>0 ||forbidden[itrack1]>0) continue; \r
1003       //\r
1004       pvertex->SetStatus(0);\r
1005       //      if (rejectBase) {\r
1006       //        pvertex->SetStatus(-100);\r
1007       //}\r
1008       if (pvertex->GetV0CosineOfPointingAngle()>kMinPointAngle2) {\r
1009         //Bo:     pvertex->SetESDindexes(track0->GetESDtrack()->GetID(),track1->GetESDtrack()->GetID());\r
1010         pvertex->SetIndex(0,track0->GetESDtrack()->GetID());//Bo: consistency 0 for neg\r
1011         pvertex->SetIndex(1,track1->GetESDtrack()->GetID());//Bo: consistency 1 for pos\r
1012         //      if (v0OK==0){\r
1013         if (v0OK){\r
1014           //      AliV0vertex vertexjuri(*track0,*track1);\r
1015           //      vertexjuri.SetESDindexes(track0->fESDtrack->GetID(),track1->fESDtrack->GetID());\r
1016           //      event->AddV0(&vertexjuri);\r
1017           pvertex->SetStatus(100);\r
1018         }\r
1019         pvertex->SetOnFlyStatus(kTRUE);\r
1020         pvertex->ChangeMassHypothesis(kK0Short);\r
1021         event->AddV0(pvertex);\r
1022       }\r
1023     }\r
1024   }\r
1025   //\r
1026   //\r
1027   // delete temporary arrays\r
1028   //  \r
1029   delete[] forbidden;\r
1030   delete[] minPointAngle;\r
1031   delete[] maxr;\r
1032   delete[] minr;\r
1033   delete[] norm;\r
1034   delete[] normdist;\r
1035   delete[] normdist1;\r
1036   delete[] normdist0;\r
1037   delete[] dist;\r
1038   delete[] itsmap;\r
1039   delete[] helixes;\r
1040   delete   pvertex;\r
1041 }\r
1042 //------------------------------------------------------------------------\r
1043 void AliITSV0Finder::RefitV02(const AliESDEvent *event,\r
1044                                 AliITStrackerMI *tracker) \r
1045 {\r
1046   //\r
1047   //try to refit  V0s in the third path of the reconstruction\r
1048   //\r
1049   TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());\r
1050   //\r
1051   Int_t  nv0s = event->GetNumberOfV0s();\r
1052   Float_t primvertex[3]={tracker->GetX(),tracker->GetY(),tracker->GetZ()};\r
1053   AliV0 v0temp;\r
1054   for (Int_t iv0 = 0; iv0<nv0s;iv0++){\r
1055     AliV0 * v0mi = (AliV0*)event->GetV0(iv0);\r
1056     if (!v0mi) continue;\r
1057     Int_t     itrack0   = v0mi->GetIndex(0);\r
1058     Int_t     itrack1   = v0mi->GetIndex(1);\r
1059     AliESDtrack *esd0   = event->GetTrack(itrack0);\r
1060     AliESDtrack *esd1   = event->GetTrack(itrack1);\r
1061     if (!esd0||!esd1) continue;\r
1062     AliITStrackMI tpc0(*esd0);  \r
1063     AliITStrackMI tpc1(*esd1);\r
1064     Double_t x,y,z; v0mi->GetXYZ(x,y,z); //I.B. \r
1065     Double_t alpha =TMath::ATan2(y,x);   //I.B.\r
1066     if (v0mi->GetRr()>85){\r
1067       if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
1068         v0temp.SetParamN(tpc0);\r
1069         v0temp.SetParamP(tpc1);\r
1070         v0temp.Update(primvertex);\r
1071         if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
1072           "V0.="<<v0mi<<\r
1073           "V0refit.="<<&v0temp<<\r
1074           "Tr0.="<<&tpc0<<\r
1075           "Tr1.="<<&tpc1<<\r
1076           "\n";\r
1077         //Bo:   if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1078         if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1079           v0mi->SetParamN(tpc0);\r
1080           v0mi->SetParamP(tpc1);\r
1081           v0mi->Update(primvertex);\r
1082         }\r
1083       }\r
1084       continue;\r
1085     }\r
1086     if (v0mi->GetRr()>35){\r
1087       AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);\r
1088       AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1);\r
1089       if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
1090         v0temp.SetParamN(tpc0);\r
1091         v0temp.SetParamP(tpc1);\r
1092         v0temp.Update(primvertex);\r
1093         if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
1094           "V0.="<<v0mi<<\r
1095           "V0refit.="<<&v0temp<<\r
1096           "Tr0.="<<&tpc0<<\r
1097           "Tr1.="<<&tpc1<<\r
1098           "\n";\r
1099         //Bo:   if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1100         if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1101           v0mi->SetParamN(tpc0);\r
1102           v0mi->SetParamP(tpc1);\r
1103           v0mi->Update(primvertex);\r
1104         }       \r
1105       }\r
1106       continue;\r
1107     }\r
1108     AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);\r
1109     AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1);    \r
1110     //    if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
1111     if (tracker->RefitAt(v0mi->GetRr(),&tpc0, v0mi->GetClusters(0)) && \r
1112         tracker->RefitAt(v0mi->GetRr(),&tpc1, v0mi->GetClusters(1))){\r
1113       v0temp.SetParamN(tpc0);\r
1114       v0temp.SetParamP(tpc1);\r
1115       v0temp.Update(primvertex);\r
1116       if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
1117         "V0.="<<v0mi<<\r
1118         "V0refit.="<<&v0temp<<\r
1119         "Tr0.="<<&tpc0<<\r
1120         "Tr1.="<<&tpc1<<\r
1121         "\n";\r
1122       //Bo:      if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1123       if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
1124         v0mi->SetParamN(tpc0);\r
1125         v0mi->SetParamP(tpc1);\r
1126         v0mi->Update(primvertex);\r
1127       } \r
1128     }    \r
1129   }\r
1130 }\r
1131 //------------------------------------------------------------------------\r
1132 \r
1133 \r
1134 AliITSV0Finder::~AliITSV0Finder()\r
1135 {\r
1136   //\r
1137   //destructor\r
1138   //\r
1139   if (fDebugStreamer) {\r
1140     //fDebugStreamer->Close();\r
1141     delete fDebugStreamer;\r
1142   }\r
1143 \r
1144 }\r