-/**************************************************************************\r
- * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *\r
- * *\r
- * Author: The ALICE Off-line Project. *\r
- * Contributors are mentioned in the code where appropriate. *\r
- * *\r
- * Permission to use, copy, modify and distribute this software and its *\r
- * documentation strictly for non-commercial purposes is hereby granted *\r
- * without fee, provided that the above copyright notice appears in all *\r
- * copies and that both the copyright notice and this permission notice *\r
- * appear in the supporting documentation. The authors make no claims *\r
- * about the suitability of this software for any purpose. It is *\r
- * provided "as is" without express or implied warranty. *\r
- **************************************************************************/\r
-/* $Id$ */\r
-\r
-//-------------------------------------------------------------------------\r
-// Implementation of the on-the-fly v0 finder for ITS tracker MI\r
-// Origin: Marian Ivanov, CERN, Marian.Ivanov@cern.ch \r
-// Extraction from AliITStrackerMI: Andrea Dainese\r
-// Current support and development: \r
-//-------------------------------------------------------------------------\r
-\r
-#include <TMatrixD.h>\r
-#include <TTree.h>\r
-#include <TString.h>\r
-#include <TRandom.h>\r
-#include <TTreeStream.h>\r
-\r
-\r
-#include "AliLog.h"\r
-#include "AliESDVertex.h"\r
-#include "AliESDEvent.h"\r
-#include "AliESDtrack.h"\r
-#include "AliV0.h"\r
-#include "AliHelix.h"\r
-#include "AliITSRecPoint.h"\r
-#include "AliITSReconstructor.h"\r
-#include "AliITStrackerMI.h"\r
-#include "AliITSV0Finder.h"\r
-#include "AliKFParticle.h"\r
-\r
-ClassImp(AliITSV0Finder)\r
-\r
-\r
-AliITSV0Finder::AliITSV0Finder():\r
-fDebugStreamer(0)\r
- {\r
- //Default constructor\r
-\r
- fDebugStreamer = new TTreeSRedirector("ITSdebug.root");\r
-} \r
-\r
-//------------------------------------------------------------------------\r
-void AliITSV0Finder::UpdateTPCV0(const AliESDEvent *event,\r
- AliITStrackerMI *tracker)\r
-{\r
- //\r
- //try to update, or reject TPC V0s\r
- //\r
- const Float_t kMinTgl2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl2();\r
-\r
- TObjArray *trackHypothesys = tracker->GetTrackHypothesys();\r
-\r
- Int_t nv0s = event->GetNumberOfV0s();\r
- Int_t nitstracks = trackHypothesys->GetEntriesFast();\r
-\r
- for (Int_t i=0;i<nv0s;i++){\r
- AliESDv0 * vertex = event->GetV0(i);\r
- Int_t ip = vertex->GetIndex(0);\r
- Int_t im = vertex->GetIndex(1);\r
- //\r
- TObjArray * arrayp = (ip<nitstracks) ? (TObjArray*)trackHypothesys->At(ip):0;\r
- TObjArray * arraym = (im<nitstracks) ? (TObjArray*)trackHypothesys->At(im):0;\r
- AliITStrackMI * trackp = (arrayp!=0) ? (AliITStrackMI*)arrayp->At(0):0;\r
- AliITStrackMI * trackm = (arraym!=0) ? (AliITStrackMI*)arraym->At(0):0;\r
- //\r
- //\r
- if (trackp){\r
- if (trackp->GetNumberOfClusters()+trackp->GetNDeadZone()>5.5){\r
- if (trackp->GetConstrain()&&trackp->GetChi2MIP(0)<3) vertex->SetStatus(-100);\r
- if (!trackp->GetConstrain()&&trackp->GetChi2MIP(0)<2) vertex->SetStatus(-100); \r
- }\r
- }\r
-\r
- if (trackm){\r
- if (trackm->GetNumberOfClusters()+trackm->GetNDeadZone()>5.5){\r
- if (trackm->GetConstrain()&&trackm->GetChi2MIP(0)<3) vertex->SetStatus(-100);\r
- if (!trackm->GetConstrain()&&trackm->GetChi2MIP(0)<2) vertex->SetStatus(-100); \r
- }\r
- }\r
- if (vertex->GetStatus()==-100) continue;\r
- //\r
- Double_t xrp[3]; vertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B.\r
- Int_t clayer = tracker->GetNearestLayer(xrp); //I.B.\r
- vertex->SetNBefore(clayer); //\r
- vertex->SetChi2Before(9*clayer); //\r
- vertex->SetNAfter(6-clayer); //\r
- vertex->SetChi2After(0); //\r
- //\r
- if (clayer >1 ){ // calculate chi2 before vertex\r
- Float_t chi2p = 0, chi2m=0; \r
- //\r
- if (trackp){\r
- for (Int_t ilayer=0;ilayer<clayer;ilayer++){\r
- if (trackp->GetClIndex(ilayer)>=0){\r
- chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+\r
- trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));\r
- }\r
- else{\r
- chi2p+=9;\r
- }\r
- }\r
- }else{\r
- chi2p = 9*clayer;\r
- }\r
- //\r
- if (trackm){\r
- for (Int_t ilayer=0;ilayer<clayer;ilayer++){\r
- if (trackm->GetClIndex(ilayer)>=0){\r
- chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+\r
- trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));\r
- }\r
- else{\r
- chi2m+=9;\r
- }\r
- }\r
- }else{\r
- chi2m = 9*clayer;\r
- }\r
- vertex->SetChi2Before(TMath::Min(chi2p,chi2m));\r
- if (TMath::Min(chi2p,chi2m)/Float_t(clayer)<4) vertex->SetStatus(-10); // track exist before vertex\r
- }\r
- \r
- if (clayer < 5 ){ // calculate chi2 after vertex\r
- Float_t chi2p = 0, chi2m=0; \r
- //\r
- if (trackp&&TMath::Abs(trackp->GetTgl())<kMinTgl2){\r
- for (Int_t ilayer=clayer;ilayer<6;ilayer++){\r
- if (trackp->GetClIndex(ilayer)>=0){\r
- chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+\r
- trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));\r
- }\r
- else{\r
- chi2p+=9;\r
- }\r
- }\r
- }else{\r
- chi2p = 0;\r
- }\r
- //\r
- if (trackm&&TMath::Abs(trackm->GetTgl())<kMinTgl2){\r
- for (Int_t ilayer=clayer;ilayer<6;ilayer++){\r
- if (trackm->GetClIndex(ilayer)>=0){\r
- chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+\r
- trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));\r
- }\r
- else{\r
- chi2m+=9;\r
- }\r
- }\r
- }else{\r
- chi2m = 0;\r
- }\r
- vertex->SetChi2After(TMath::Max(chi2p,chi2m));\r
- if (TMath::Max(chi2m,chi2p)/Float_t(6-clayer)>9) vertex->SetStatus(-20); // track not found in ITS\r
- }\r
- }\r
- //\r
-}\r
-//------------------------------------------------------------------------\r
-void AliITSV0Finder::FindV02(AliESDEvent *event,\r
- AliITStrackerMI *tracker) \r
-{\r
- //\r
- // V0 finder\r
- //\r
- // Cuts on DCA - R dependent\r
- // max distance DCA between 2 tracks cut \r
- // maxDist = TMath::Min(kMaxDist,kMaxDist0+pvertex->GetRr()*kMaxDist);\r
- //\r
- const Float_t kMaxDist0 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist0();\r
- const Float_t kMaxDist1 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist1();\r
- const Float_t kMaxDist = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist();\r
- const Float_t kMinPointAngle = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle();\r
- const Float_t kMinPointAngle2 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle2();\r
- const Float_t kMinR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinR();\r
- const Float_t kMaxR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxR();\r
- const Float_t kCausality0Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCausality0Cut();\r
- const Float_t kLikelihood01Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood01Cut();\r
- const Float_t kLikelihood1Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood1Cut();\r
- const Float_t kCombinedCut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCombinedCut();\r
- // const Float_t kMinClFullTrk= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClFullTrk();\r
- const Float_t kMinTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl0();\r
- const Float_t kMinTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTPCdensity();\r
- const Float_t kMinTgl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl1();\r
-\r
- const Float_t kMinchi2before0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before0();\r
- const Float_t kMinchi2before1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before1();\r
- const Float_t kMinchi2after0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after0();\r
- const Float_t kMinchi2after1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after1();\r
- const Float_t kAddchi2SharedCl= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2SharedCl();\r
- const Float_t kAddchi2NegCl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl0();\r
- const Float_t kAddchi2NegCl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl1();\r
-\r
- const Float_t kSigp0Par0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par0();\r
- const Float_t kSigp0Par1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par1();\r
- const Float_t kSigp0Par2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par2();\r
-\r
- const Float_t kSigpPar0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar0();\r
- const Float_t kSigpPar1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar1();\r
- const Float_t kSigpPar2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar2();\r
- \r
-\r
- TObjArray *trackHypothesys = tracker->GetTrackHypothesys();\r
- TObjArray *bestHypothesys = tracker->GetBestHypothesys();\r
- TObjArray *originalTracks = tracker->GetOriginal();\r
- //\r
- //\r
- TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());\r
- Int_t ntracks = event->GetNumberOfTracks(); \r
- Int_t nitstracks = trackHypothesys->GetEntriesFast();\r
- originalTracks->Expand(ntracks);\r
- trackHypothesys->Expand(ntracks);\r
- bestHypothesys->Expand(ntracks);\r
- //\r
- AliHelix * helixes = new AliHelix[ntracks+2];\r
- TObjArray trackarray(ntracks+2); //array with tracks - with vertex constrain\r
- TObjArray trackarrayc(ntracks+2); //array of "best tracks" - without vertex constrain\r
- TObjArray trackarrayl(ntracks+2); //array of "longest tracks" - without vertex constrain\r
- Bool_t * forbidden = new Bool_t [ntracks+2];\r
- Int_t *itsmap = new Int_t [ntracks+2];\r
- Float_t *dist = new Float_t[ntracks+2];\r
- Float_t *normdist0 = new Float_t[ntracks+2];\r
- Float_t *normdist1 = new Float_t[ntracks+2];\r
- Float_t *normdist = new Float_t[ntracks+2];\r
- Float_t *norm = new Float_t[ntracks+2];\r
- Float_t *maxr = new Float_t[ntracks+2];\r
- Float_t *minr = new Float_t[ntracks+2];\r
- Float_t *minPointAngle= new Float_t[ntracks+2];\r
- //\r
- AliV0 *pvertex = new AliV0;\r
- AliITStrackMI * dummy= new AliITStrackMI;\r
- dummy->SetLabel(0);\r
- AliITStrackMI trackat0; //temporary track for DCA calculation\r
- //\r
- Float_t primvertex[3]={tracker->GetX(),tracker->GetY(),tracker->GetZ()};\r
- //\r
- // make ITS - ESD map\r
- //\r
- for (Int_t itrack=0;itrack<ntracks+2;itrack++) {\r
- itsmap[itrack] = -1;\r
- forbidden[itrack] = kFALSE;\r
- maxr[itrack] = kMaxR;\r
- minr[itrack] = kMinR;\r
- minPointAngle[itrack] = kMinPointAngle;\r
- }\r
- for (Int_t itrack=0;itrack<nitstracks;itrack++){\r
- AliITStrackMI * original = (AliITStrackMI*)(originalTracks->At(itrack));\r
- Int_t esdindex = original->GetESDtrack()->GetID();\r
- itsmap[esdindex] = itrack;\r
- }\r
- //\r
- // create ITS tracks from ESD tracks if not done before\r
- //\r
- for (Int_t itrack=0;itrack<ntracks;itrack++){\r
- if (itsmap[itrack]>=0) continue;\r
- AliITStrackMI * tpctrack = new AliITStrackMI(*(event->GetTrack(itrack)));\r
- //tpctrack->fD[0] = tpctrack->GetD(GetX(),GetY());\r
- //tpctrack->fD[1] = tpctrack->GetZat(GetX())-GetZ(); \r
- tpctrack->GetDZ(tracker->GetX(),tracker->GetY(),tracker->GetZ(),tpctrack->GetDP()); //I.B.\r
- if (tpctrack->GetD(0)<20 && tpctrack->GetD(1)<20){\r
- // tracks which can reach inner part of ITS\r
- // propagate track to outer its volume - with correction for material\r
- AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(tpctrack); \r
- }\r
- itsmap[itrack] = nitstracks;\r
- originalTracks->AddAt(tpctrack,nitstracks);\r
- nitstracks++;\r
- }\r
- //\r
- // fill temporary arrays\r
- //\r
- for (Int_t itrack=0;itrack<ntracks;itrack++){\r
- AliESDtrack * esdtrack = event->GetTrack(itrack);\r
- Int_t itsindex = itsmap[itrack];\r
- AliITStrackMI *original = (AliITStrackMI*)originalTracks->At(itsmap[itrack]);\r
- if (!original) continue;\r
- AliITStrackMI *bestConst = 0;\r
- AliITStrackMI *bestLong = 0;\r
- AliITStrackMI *best = 0; \r
- //\r
- //\r
- TObjArray * array = (TObjArray*) trackHypothesys->At(itsindex);\r
- Int_t hentries = (array==0) ? 0 : array->GetEntriesFast();\r
- // Get best track with vertex constrain\r
- for (Int_t ih=0;ih<hentries;ih++){\r
- AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);\r
- if (!trackh->GetConstrain()) continue;\r
- if (!bestConst) bestConst = trackh;\r
- if (trackh->GetNumberOfClusters()>5.0){\r
- bestConst = trackh; // full track - with minimal chi2\r
- break;\r
- }\r
- if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()) continue; \r
- bestConst = trackh;\r
- break;\r
- }\r
- // Get best long track without vertex constrain and best track without vertex constrain\r
- for (Int_t ih=0;ih<hentries;ih++){\r
- AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);\r
- if (trackh->GetConstrain()) continue;\r
- if (!best) best = trackh;\r
- if (!bestLong) bestLong = trackh;\r
- if (trackh->GetNumberOfClusters()>5.0){\r
- bestLong = trackh; // full track - with minimal chi2\r
- break;\r
- }\r
- if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()) continue; \r
- bestLong = trackh; \r
- }\r
- if (!best) {\r
- best = original;\r
- bestLong = original;\r
- }\r
- //I.B. trackat0 = *bestLong;\r
- new (&trackat0) AliITStrackMI(*bestLong);\r
- Double_t xx,yy,zz,alpha; \r
- if (!bestLong->GetGlobalXYZat(bestLong->GetX(),xx,yy,zz)) continue;\r
- alpha = TMath::ATan2(yy,xx); \r
- // if (!trackat0.Propagate(alpha,0)) continue; \r
- trackat0.Propagate(alpha,0); //PH The check on the return value is temporarily disabled (bug 45751) \r
- // calculate normalized distances to the vertex \r
- //\r
- Float_t ptfac = (1.+100.*TMath::Abs(trackat0.GetC()));\r
- if ( bestLong->GetNumberOfClusters()>3 ){ \r
- dist[itsindex] = trackat0.GetY();\r
- norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());\r
- normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);\r
- normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));\r
- normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
- if (!bestConst){\r
- if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<6) normdist[itsindex]*=2.;\r
- if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<5) normdist[itsindex]*=2.;\r
- if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<4) normdist[itsindex]*=2.;\r
- }else{\r
- if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<6) normdist[itsindex]*=1.5;\r
- if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<5) normdist[itsindex]*=1.5;\r
- }\r
- }\r
- else{ \r
- if (bestConst&&bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>4.5){\r
- dist[itsindex] = bestConst->GetD(0);\r
- norm[itsindex] = bestConst->GetDnorm(0);\r
- normdist0[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);\r
- normdist1[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);\r
- normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
- }else{\r
- dist[itsindex] = trackat0.GetY();\r
- norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());\r
- normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);\r
- normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));\r
- normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);\r
- if (TMath::Abs(trackat0.GetTgl())>kMinTgl0){\r
- if (normdist[itsindex]<3) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]>3) {\r
- minr[itsindex] = TMath::Max(Float_t(40.),minr[itsindex]);\r
- }\r
- }\r
- }\r
- }\r
- //\r
- //-----------------------------------------------------------\r
- //Forbid primary track candidates - \r
- //\r
- //treetr->SetAlias("forbidden0","Tr0.fN<4&&Tr1.fN+Tr1.fNDeadZone>4.5");\r
- //treetr->SetAlias("forbidden1","ND<3&&Tr1.fN+Tr1.fNDeadZone>5.5");\r
- //treetr->SetAlias("forbidden2","ND<2&&Tr1.fClIndex[0]>0&&Tr1.fClIndex[0]>0");\r
- //treetr->SetAlias("forbidden3","ND<1&&Tr1.fClIndex[0]>0");\r
- //treetr->SetAlias("forbidden4","ND<4&&Tr1.fNormChi2[0]<2");\r
- //treetr->SetAlias("forbidden5","ND<5&&Tr1.fNormChi2[0]<1");\r
- //-----------------------------------------------------------\r
- if (bestConst){\r
- if (bestLong->GetNumberOfClusters()<4 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>4.5) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]<3 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>5.5) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]<2 && bestConst->GetClIndex(0)>=0 && bestConst->GetClIndex(1)>=0 ) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]<1 && bestConst->GetClIndex(0)>=0) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]<4 && bestConst->GetNormChi2(0)<2) forbidden[itsindex]=kTRUE;\r
- if (normdist[itsindex]<5 && bestConst->GetNormChi2(0)<1) forbidden[itsindex]=kTRUE; \r
- if (bestConst->GetNormChi2(0)<2.5) {\r
- minPointAngle[itsindex]= 0.9999;\r
- maxr[itsindex] = 10;\r
- }\r
- }\r
- //\r
- //forbid daughter kink candidates\r
- //\r
- if (esdtrack->GetKinkIndex(0)>0) forbidden[itsindex] = kTRUE;\r
- Bool_t isElectron = kTRUE;\r
- Bool_t isProton = kTRUE;\r
- Double_t pid[5];\r
- esdtrack->GetESDpid(pid);\r
- for (Int_t i=1;i<5;i++){\r
- if (pid[0]<pid[i]) isElectron= kFALSE;\r
- if (pid[4]<pid[i]) isProton= kFALSE;\r
- }\r
- if (isElectron){\r
- forbidden[itsindex]=kFALSE; \r
- normdist[itsindex]*=-1;\r
- }\r
- if (isProton){\r
- if (normdist[itsindex]>2) forbidden[itsindex]=kFALSE; \r
- normdist[itsindex]*=-1;\r
- }\r
-\r
- //\r
- // Causality cuts in TPC volume\r
- //\r
- if (esdtrack->GetTPCdensity(0,10) >kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(110),maxr[itsindex]);\r
- if (esdtrack->GetTPCdensity(10,30)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(120),maxr[itsindex]);\r
- if (esdtrack->GetTPCdensity(20,40)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(130),maxr[itsindex]);\r
- if (esdtrack->GetTPCdensity(30,50)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(140),maxr[itsindex]);\r
- //\r
- if (esdtrack->GetTPCdensity(0,60)<0.4&&bestLong->GetNumberOfClusters()<3) minr[itsindex]=100; \r
- //\r
- //\r
- if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){\r
- cstream<<"Track"<<\r
- "Tr0.="<<best<<\r
- "Tr1.="<<((bestConst)? bestConst:dummy)<<\r
- "Tr2.="<<bestLong<<\r
- "Tr3.="<<&trackat0<<\r
- "Esd.="<<esdtrack<<\r
- "Dist="<<dist[itsindex]<<\r
- "ND0="<<normdist0[itsindex]<<\r
- "ND1="<<normdist1[itsindex]<<\r
- "ND="<<normdist[itsindex]<<\r
- "Pz="<<primvertex[2]<<\r
- "Forbid="<<forbidden[itsindex]<<\r
- "\n";\r
- //\r
- }\r
- trackarray.AddAt(best,itsindex);\r
- trackarrayc.AddAt(bestConst,itsindex);\r
- trackarrayl.AddAt(bestLong,itsindex);\r
- new (&helixes[itsindex]) AliHelix(*best);\r
- }\r
- //\r
- //\r
- //\r
- // first iterration of V0 finder\r
- //\r
-\r
-\r
-\r
- for (Int_t iesd0=0;iesd0<ntracks;iesd0++){\r
- Int_t itrack0 = itsmap[iesd0];\r
- if (forbidden[itrack0]) continue;\r
- AliITStrackMI * btrack0 = (AliITStrackMI*)trackarray.At(itrack0);\r
- if (!btrack0) continue; \r
- if (btrack0->GetSign()>0 && !AliITSReconstructor::GetRecoParam()->GetStoreLikeSignV0s()) continue;\r
- AliITStrackMI *trackc0 = (AliITStrackMI*)trackarrayc.At(itrack0);\r
- //\r
- for (Int_t iesd1=iesd0+1;iesd1<ntracks;iesd1++){\r
- Int_t itrack1 = itsmap[iesd1];\r
- if (forbidden[itrack1]) continue;\r
-\r
- AliITStrackMI * btrack1 = (AliITStrackMI*)trackarray.At(itrack1); \r
- if (!btrack1) continue;\r
- if (btrack1->GetSign()<0 && !AliITSReconstructor::GetRecoParam()->GetStoreLikeSignV0s()) continue;\r
- Bool_t isGold = kFALSE;\r
- if (TMath::Abs(TMath::Abs(btrack0->GetLabel())-TMath::Abs(btrack1->GetLabel()))==1){\r
- isGold = kTRUE;\r
- }\r
- AliITStrackMI *trackc1 = (AliITStrackMI*)trackarrayc.At(itrack1);\r
- AliHelix &h1 = helixes[itrack0];\r
- AliHelix &h2 = helixes[itrack1];\r
- //\r
- // find linear distance\r
- Double_t rmin =0;\r
- //\r
- //\r
- //\r
- Double_t phase[2][2],radius[2];\r
- Int_t points = h1.GetRPHIintersections(h2, phase, radius);\r
- if (points==0) continue;\r
- Double_t delta[2]={1000000,1000000}; \r
- rmin = radius[0];\r
- h1.ParabolicDCA(h2,phase[0][0],phase[0][1],radius[0],delta[0]);\r
- if (points==2){ \r
- if (radius[1]<rmin) rmin = radius[1];\r
- h1.ParabolicDCA(h2,phase[1][0],phase[1][1],radius[1],delta[1]);\r
- }\r
- rmin = TMath::Sqrt(rmin);\r
- Double_t distance = 0;\r
- Double_t radiusC = 0;\r
- Int_t iphase = 0;\r
- if (points==1 || delta[0]<delta[1]){\r
- distance = TMath::Sqrt(delta[0]);\r
- radiusC = TMath::Sqrt(radius[0]);\r
- }else{\r
- distance = TMath::Sqrt(delta[1]);\r
- radiusC = TMath::Sqrt(radius[1]);\r
- iphase=1;\r
- }\r
- if (radiusC<TMath::Max(minr[itrack0],minr[itrack1])) continue;\r
- if (radiusC>TMath::Min(maxr[itrack0],maxr[itrack1])) continue; \r
- Float_t maxDist = TMath::Min(kMaxDist,Float_t(kMaxDist0+radiusC*kMaxDist1)); \r
- if (distance>maxDist) continue;\r
- Float_t pointAngle = h1.GetPointAngle(h2,phase[iphase],primvertex);\r
- if (pointAngle<TMath::Max(minPointAngle[itrack0],minPointAngle[itrack1])) continue;\r
- //\r
- //\r
- // Double_t distance = TestV0(h1,h2,pvertex,rmin);\r
- //\r
- // if (distance>maxDist) continue;\r
- // if (pvertex->GetRr()<kMinR) continue;\r
- // if (pvertex->GetRr()>kMaxR) continue;\r
- AliITStrackMI * track0=btrack0;\r
- AliITStrackMI * track1=btrack1;\r
- // if (pvertex->GetRr()<3.5){ \r
- if (radiusC<3.5){ \r
- //use longest tracks inside the pipe\r
- track0 = (AliITStrackMI*)trackarrayl.At(itrack0);\r
- track1 = (AliITStrackMI*)trackarrayl.At(itrack1);\r
- } \r
- //\r
- //\r
- \r
-\r
- \r
- pvertex->SetParamN(*track0);\r
- pvertex->SetParamP(*track1);\r
- pvertex->Update(primvertex);\r
- pvertex->SetClusters(track0->ClIndex(),track1->ClIndex()); // register clusters\r
-\r
- AliKFParticle negKF( *(pvertex->GetParamN()) ,11);\r
- AliKFParticle posKF( *(pvertex->GetParamP()) ,-11);\r
-\r
- AliKFParticle v0KF(negKF,posKF);\r
- Float_t v0KFchi2= v0KF.GetChi2();\r
-\r
-\r
- if (pvertex->GetRr()<kMinR) continue;\r
- if (pvertex->GetRr()>kMaxR) continue;\r
- if (pvertex->GetV0CosineOfPointingAngle()<kMinPointAngle) continue;\r
-//Bo: if (pvertex->GetDist2()>maxDist) continue;\r
- if (pvertex->GetDcaV0Daughters()>maxDist) continue;\r
-//Bo: pvertex->SetLab(0,track0->GetLabel());\r
-//Bo: pvertex->SetLab(1,track1->GetLabel());\r
- pvertex->SetIndex(0,track0->GetESDtrack()->GetID());\r
- pvertex->SetIndex(1,track1->GetESDtrack()->GetID());\r
- // \r
- AliITStrackMI * htrackc0 = trackc0 ? trackc0:dummy; \r
- AliITStrackMI * htrackc1 = trackc1 ? trackc1:dummy;\r
-\r
- //\r
- //\r
- TObjArray * array0b = (TObjArray*)bestHypothesys->At(itrack0);\r
- if (!array0b&&pvertex->GetRr()<40 && TMath::Abs(track0->GetTgl())<kMinTgl1) {\r
- tracker->SetCurrentEsdTrack(itrack0);\r
- tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack0),itrack0, kFALSE);\r
- }\r
- TObjArray * array1b = (TObjArray*)bestHypothesys->At(itrack1);\r
- if (!array1b&&pvertex->GetRr()<40 && TMath::Abs(track1->GetTgl())<kMinTgl1) { \r
- tracker->SetCurrentEsdTrack(itrack1);\r
- tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack1),itrack1, kFALSE);\r
- }\r
- //\r
- AliITStrackMI * track0b = (AliITStrackMI*)originalTracks->At(itrack0);\r
- AliITStrackMI * track1b = (AliITStrackMI*)originalTracks->At(itrack1);\r
- AliITStrackMI * track0l = (AliITStrackMI*)originalTracks->At(itrack0);\r
- AliITStrackMI * track1l = (AliITStrackMI*)originalTracks->At(itrack1);\r
- \r
- Float_t minchi2before0=kMinchi2before0;\r
- Float_t minchi2before1=kMinchi2before1;\r
- Float_t minchi2after0 =kMinchi2after0;\r
- Float_t minchi2after1 =kMinchi2after1;\r
- Double_t xrp[3]; pvertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B.\r
- Int_t maxLayer = tracker->GetNearestLayer(xrp); //I.B.\r
- \r
- if (array0b) for (Int_t i=0;i<5;i++){\r
- // best track after vertex\r
- AliITStrackMI * btrack = (AliITStrackMI*)array0b->At(i);\r
- if (!btrack) continue;\r
- if (btrack->GetNumberOfClusters()>track0l->GetNumberOfClusters()) track0l = btrack; \r
- // if (btrack->fX<pvertex->GetRr()-2.-0.5/(0.1+pvertex->GetAnglep()[2])) {\r
- if (btrack->GetX()<pvertex->GetRr()-2.) {\r
- if ( (maxLayer>i+2|| (i==0)) && btrack->GetNumberOfClusters()==(6-i)&&i<3){\r
- Float_t sumchi2= 0;\r
- Float_t sumn = 0;\r
- if (maxLayer<3){ // take prim vertex as additional measurement\r
- if (normdist[itrack0]>0 && htrackc0){\r
- sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack0]*normdist[itrack0],16.);\r
- }else{\r
- sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack0]*normdist[itrack0]+3.),16.);\r
- }\r
- sumn += 3-maxLayer;\r
- }\r
- for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){\r
- sumn+=1.; \r
- if (btrack->GetClIndex(ilayer)<0){\r
- sumchi2+=kAddchi2NegCl0;\r
- continue;\r
- }else{\r
- Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);\r
- for (Int_t itrack=0;itrack<4;itrack++){\r
- AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);\r
- if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack0){\r
- sumchi2+=kAddchi2SharedCl; //shared cluster\r
- break;\r
- }\r
- }\r
- sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));\r
- sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer)); \r
- }\r
- }\r
- sumchi2/=sumn;\r
- if (sumchi2<minchi2before0) minchi2before0=sumchi2; \r
- }\r
- continue; //safety space - Geo manager will give exact layer\r
- }\r
- track0b = btrack;\r
- minchi2after0 = btrack->GetNormChi2(i);\r
- break;\r
- }\r
- if (array1b) for (Int_t i=0;i<5;i++){\r
- // best track after vertex\r
- AliITStrackMI * btrack = (AliITStrackMI*)array1b->At(i);\r
- if (!btrack) continue;\r
- if (btrack->GetNumberOfClusters()>track1l->GetNumberOfClusters()) track1l = btrack; \r
- // if (btrack->fX<pvertex->GetRr()-2-0.5/(0.1+pvertex->GetAnglep()[2])){\r
- if (btrack->GetX()<pvertex->GetRr()-2){\r
- if ((maxLayer>i+2 || (i==0))&&btrack->GetNumberOfClusters()==(6-i)&&(i<3)){\r
- Float_t sumchi2= 0;\r
- Float_t sumn = 0;\r
- if (maxLayer<3){ // take prim vertex as additional measurement\r
- if (normdist[itrack1]>0 && htrackc1){\r
- sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack1]*normdist[itrack1],16.);\r
- }else{\r
- sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack1]*normdist[itrack1]+3.),16.);\r
- }\r
- sumn += 3-maxLayer;\r
- }\r
- for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){\r
- sumn+=1.;\r
- if (btrack->GetClIndex(ilayer)<0){\r
- sumchi2+=kAddchi2NegCl1;\r
- continue;\r
- }else{\r
- Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);\r
- for (Int_t itrack=0;itrack<4;itrack++){\r
- AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);\r
- if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack1){\r
- sumchi2+=kAddchi2SharedCl; //shared cluster\r
- break;\r
- }\r
- }\r
- sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));\r
- sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer)); \r
- }\r
- }\r
- sumchi2/=sumn;\r
- if (sumchi2<minchi2before1) minchi2before1=sumchi2; \r
- }\r
- continue; //safety space - Geo manager will give exact layer \r
- }\r
- track1b = btrack;\r
- minchi2after1 = btrack->GetNormChi2(i);\r
- break;\r
- }\r
- //\r
- // position resolution - used for DCA cut\r
- Float_t sigmad = track0b->GetSigmaY2()+track0b->GetSigmaZ2()+track1b->GetSigmaY2()+track1b->GetSigmaZ2()+\r
- (track0b->GetX()-pvertex->GetRr())*(track0b->GetX()-pvertex->GetRr())*(track0b->GetSigmaSnp2()+track0b->GetSigmaTgl2())+\r
- (track1b->GetX()-pvertex->GetRr())*(track1b->GetX()-pvertex->GetRr())*(track1b->GetSigmaSnp2()+track1b->GetSigmaTgl2());\r
- sigmad =TMath::Sqrt(sigmad)+0.04;\r
- if (pvertex->GetRr()>50){\r
- Double_t cov0[15],cov1[15];\r
- track0b->GetESDtrack()->GetInnerExternalCovariance(cov0);\r
- track1b->GetESDtrack()->GetInnerExternalCovariance(cov1);\r
- sigmad = cov0[0]+cov0[2]+cov1[0]+cov1[2]+\r
- (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov0[5]+cov0[9])+\r
- (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov1[5]+cov1[9]);\r
- sigmad =TMath::Sqrt(sigmad)+0.05;\r
- }\r
- // \r
- AliV0 vertex2;\r
- vertex2.SetParamN(*track0b);\r
- vertex2.SetParamP(*track1b);\r
- vertex2.Update(primvertex);\r
- //Bo: if (vertex2.GetDist2()<=pvertex->GetDist2()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){\r
- if (vertex2.GetDcaV0Daughters()<=pvertex->GetDcaV0Daughters()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){\r
- pvertex->SetParamN(*track0b);\r
- pvertex->SetParamP(*track1b);\r
- pvertex->Update(primvertex);\r
- pvertex->SetClusters(track0b->ClIndex(),track1b->ClIndex()); // register clusters\r
- pvertex->SetIndex(0,track0->GetESDtrack()->GetID());\r
- pvertex->SetIndex(1,track1->GetESDtrack()->GetID());\r
- }\r
- pvertex->SetDistSigma(sigmad);\r
- //Bo: pvertex->SetDistNorm(pvertex->GetDist2()/sigmad); \r
- pvertex->SetNormDCAPrim(normdist[itrack0],normdist[itrack1]);\r
- //\r
- // define likelihhod and causalities\r
- //\r
- Float_t pa0=1, pa1=1, pb0=0.26, pb1=0.26; \r
- if (maxLayer<1){\r
- Float_t fnorm0 = normdist[itrack0];\r
- if (fnorm0<0) fnorm0*=-3;\r
- Float_t fnorm1 = normdist[itrack1];\r
- if (fnorm1<0) fnorm1*=-3;\r
- if ((pvertex->GetAnglep()[2]>0.1) || ( (pvertex->GetRr()<10.5)&& pvertex->GetAnglep()[2]>0.05 ) || (pvertex->GetRr()<3)){\r
- pb0 = TMath::Exp(-TMath::Min(fnorm0,Float_t(16.))/12.);\r
- pb1 = TMath::Exp(-TMath::Min(fnorm1,Float_t(16.))/12.);\r
- }\r
- pvertex->SetChi2Before(normdist[itrack0]);\r
- pvertex->SetChi2After(normdist[itrack1]); \r
- pvertex->SetNAfter(0);\r
- pvertex->SetNBefore(0);\r
- }else{\r
- pvertex->SetChi2Before(minchi2before0);\r
- pvertex->SetChi2After(minchi2before1);\r
- if (pvertex->GetAnglep()[2]>0.1 || ( pvertex->GetRr()<10.5 && pvertex->GetAnglep()[2]>0.05) || pvertex->GetRr()<3){\r
- pb0 = TMath::Exp(-TMath::Min(minchi2before0,Float_t(16))/12.);\r
- pb1 = TMath::Exp(-TMath::Min(minchi2before1,Float_t(16))/12.);\r
- }\r
- pvertex->SetNAfter(maxLayer);\r
- pvertex->SetNBefore(maxLayer); \r
- }\r
- if (pvertex->GetRr()<90){\r
- pa0 *= TMath::Min(track0->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));\r
- pa1 *= TMath::Min(track1->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));\r
- }\r
- if (pvertex->GetRr()<20){\r
- pa0 *= (0.2+TMath::Exp(-TMath::Min(minchi2after0,Float_t(16))/8.))/1.2;\r
- pa1 *= (0.2+TMath::Exp(-TMath::Min(minchi2after1,Float_t(16))/8.))/1.2;\r
- }\r
- //\r
- pvertex->SetCausality(pb0,pb1,pa0,pa1);\r
- //\r
- // Likelihood calculations - apply cuts\r
- // \r
- Bool_t v0OK = kTRUE;\r
- Float_t p12 = pvertex->GetParamP()->GetParameter()[4]*pvertex->GetParamP()->GetParameter()[4];\r
- p12 += pvertex->GetParamN()->GetParameter()[4]*pvertex->GetParamN()->GetParameter()[4];\r
- p12 = TMath::Sqrt(p12); // "mean" momenta\r
-\r
- Float_t sigmap0 = kSigp0Par0+kSigp0Par1/(kSigp0Par2+pvertex->GetRr()); \r
- Float_t sigmap = kSigpPar0*sigmap0*(kSigpPar1+kSigpPar2*p12); // "resolution: of point angle - as a function of radius and momenta\r
-\r
-\r
- Float_t causalityA = (1.0-pvertex->GetCausalityP()[0])*(1.0-pvertex->GetCausalityP()[1]);\r
- Float_t causalityB = TMath::Sqrt(TMath::Min(pvertex->GetCausalityP()[2],Double_t(0.7))*\r
- TMath::Min(pvertex->GetCausalityP()[3],Double_t(0.7)));\r
- //\r
- //Bo: Float_t likelihood0 = (TMath::Exp(-pvertex->GetDistNorm())+0.1) *(pvertex->GetDist2()<0.5)*(pvertex->GetDistNorm()<5);\r
- Float_t lDistNorm = pvertex->GetDcaV0Daughters()/pvertex->GetDistSigma();\r
- Float_t likelihood0 = (TMath::Exp(-lDistNorm)+0.1) *(pvertex->GetDcaV0Daughters()<0.5)*(lDistNorm<5);\r
-\r
- Float_t likelihood1 = TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/sigmap)+\r
- 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(4.*sigmap))+\r
- 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(8.*sigmap))+\r
- 0.1*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/0.01);\r
- //\r
- if (causalityA<kCausality0Cut) v0OK = kFALSE;\r
- if (TMath::Sqrt(likelihood0*likelihood1)<kLikelihood01Cut) v0OK = kFALSE;\r
- if (likelihood1<kLikelihood1Cut) v0OK = kFALSE;\r
- if (TMath::Power(likelihood0*likelihood1*causalityB,0.33)<kCombinedCut) v0OK = kFALSE;\r
- \r
- //\r
- //\r
- if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){ \r
- Bool_t gold = TMath::Abs(TMath::Abs(track0->GetLabel())-TMath::Abs(track1->GetLabel()))==1;\r
- cstream<<"It0"<<\r
- "Tr0.="<<track0<< //best without constrain\r
- "Tr1.="<<track1<< //best without constrain \r
- "posKF.="<<&posKF<< //KF from track0\r
- "negKF.="<<&negKF<< //KF from track1\r
- "Tr0B.="<<track0b<< //best without constrain after vertex\r
- "Tr1B.="<<track1b<< //best without constrain after vertex \r
- "Tr0C.="<<htrackc0<< //best with constrain if exist\r
- "Tr1C.="<<htrackc1<< //best with constrain if exist\r
- "Tr0L.="<<track0l<< //longest best \r
- "Tr1L.="<<track1l<< //longest best\r
- "Esd0.="<<track0->GetESDtrack()<< // esd track0 params\r
- "Esd1.="<<track1->GetESDtrack()<< // esd track1 params\r
- "V0.="<<pvertex<< //vertex properties\r
- "V0b.="<<&vertex2<< //vertex properties at "best" track\r
- "v0KF.="<<&v0KF<< //KF from pvertex\r
- "chiKF.="<<v0KFchi2<< // chi2 from KF\r
- "ND0="<<normdist[itrack0]<< //normalize distance for track0\r
- "ND1="<<normdist[itrack1]<< //normalize distance for track1\r
- "Gold="<<gold<< //\r
- // "RejectBase="<<rejectBase<< //rejection in First itteration\r
- "OK="<<v0OK<<\r
- "rmin="<<rmin<<\r
- "sigmad="<<sigmad<<\r
- "\n";\r
- } \r
- //if (rejectBase) continue;\r
- //\r
- pvertex->SetStatus(0);\r
- // if (rejectBase) {\r
- // pvertex->SetStatus(-100);\r
- //}\r
- if (pvertex->GetV0CosineOfPointingAngle()>kMinPointAngle2) {\r
- //Bo: pvertex->SetESDindexes(track0->GetESDtrack()->GetID(),track1->GetESDtrack()->GetID());\r
- pvertex->SetIndex(0,track0->GetESDtrack()->GetID());//Bo: consistency 0 for neg\r
- pvertex->SetIndex(1,track1->GetESDtrack()->GetID());//Bo: consistency 1 for pos\r
- if (v0OK){\r
- // AliV0vertex vertexjuri(*track0,*track1);\r
- // vertexjuri.SetESDindexes(track0->fESDtrack->GetID(),track1->fESDtrack->GetID());\r
- // event->AddV0(&vertexjuri);\r
- pvertex->SetStatus(100);\r
- }\r
- pvertex->SetOnFlyStatus(kTRUE);\r
- pvertex->ChangeMassHypothesis(kK0Short);\r
- event->AddV0(pvertex);\r
- }\r
- }\r
- }\r
- //\r
- //\r
- // delete temporary arrays\r
- // \r
- delete[] forbidden;\r
- delete[] minPointAngle;\r
- delete[] maxr;\r
- delete[] minr;\r
- delete[] norm;\r
- delete[] normdist;\r
- delete[] normdist1;\r
- delete[] normdist0;\r
- delete[] dist;\r
- delete[] itsmap;\r
- delete[] helixes;\r
- delete pvertex;\r
-}\r
-//------------------------------------------------------------------------\r
-void AliITSV0Finder::RefitV02(const AliESDEvent *event,\r
- AliITStrackerMI *tracker) \r
-{\r
- //\r
- //try to refit V0s in the third path of the reconstruction\r
- //\r
- TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());\r
- //\r
- Int_t nv0s = event->GetNumberOfV0s();\r
- Float_t primvertex[3]={tracker->GetX(),tracker->GetY(),tracker->GetZ()};\r
- AliV0 v0temp;\r
- for (Int_t iv0 = 0; iv0<nv0s;iv0++){\r
- AliV0 * v0mi = (AliV0*)event->GetV0(iv0);\r
- if (!v0mi) continue;\r
- Int_t itrack0 = v0mi->GetIndex(0);\r
- Int_t itrack1 = v0mi->GetIndex(1);\r
- AliESDtrack *esd0 = event->GetTrack(itrack0);\r
- AliESDtrack *esd1 = event->GetTrack(itrack1);\r
- if (!esd0||!esd1) continue;\r
- AliITStrackMI tpc0(*esd0); \r
- AliITStrackMI tpc1(*esd1);\r
- Double_t x,y,z; v0mi->GetXYZ(x,y,z); //I.B. \r
- Double_t alpha =TMath::ATan2(y,x); //I.B.\r
- if (v0mi->GetRr()>85){\r
- if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
- v0temp.SetParamN(tpc0);\r
- v0temp.SetParamP(tpc1);\r
- v0temp.Update(primvertex);\r
- if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
- "V0.="<<v0mi<<\r
- "V0refit.="<<&v0temp<<\r
- "Tr0.="<<&tpc0<<\r
- "Tr1.="<<&tpc1<<\r
- "\n";\r
- //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- v0mi->SetParamN(tpc0);\r
- v0mi->SetParamP(tpc1);\r
- v0mi->Update(primvertex);\r
- }\r
- }\r
- continue;\r
- }\r
- if (v0mi->GetRr()>35){\r
- AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);\r
- AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1);\r
- if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
- v0temp.SetParamN(tpc0);\r
- v0temp.SetParamP(tpc1);\r
- v0temp.Update(primvertex);\r
- if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
- "V0.="<<v0mi<<\r
- "V0refit.="<<&v0temp<<\r
- "Tr0.="<<&tpc0<<\r
- "Tr1.="<<&tpc1<<\r
- "\n";\r
- //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- v0mi->SetParamN(tpc0);\r
- v0mi->SetParamP(tpc1);\r
- v0mi->Update(primvertex);\r
- } \r
- }\r
- continue;\r
- }\r
- AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);\r
- AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1); \r
- // if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){\r
- if (tracker->RefitAt(v0mi->GetRr(),&tpc0, v0mi->GetClusters(0)) && \r
- tracker->RefitAt(v0mi->GetRr(),&tpc1, v0mi->GetClusters(1))){\r
- v0temp.SetParamN(tpc0);\r
- v0temp.SetParamP(tpc1);\r
- v0temp.Update(primvertex);\r
- if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<\r
- "V0.="<<v0mi<<\r
- "V0refit.="<<&v0temp<<\r
- "Tr0.="<<&tpc0<<\r
- "Tr1.="<<&tpc1<<\r
- "\n";\r
- //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){\r
- v0mi->SetParamN(tpc0);\r
- v0mi->SetParamP(tpc1);\r
- v0mi->Update(primvertex);\r
- } \r
- } \r
- }\r
-}\r
-//------------------------------------------------------------------------\r
-\r
-\r
-AliITSV0Finder::~AliITSV0Finder()\r
-{\r
- //\r
- //destructor\r
- //\r
- if (fDebugStreamer) {\r
- //fDebugStreamer->Close();\r
- delete fDebugStreamer;\r
- }\r
-\r
-}\r
+/**************************************************************************
+ * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
+/* $Id$ */
+
+//-------------------------------------------------------------------------
+// Implementation of the on-the-fly v0 finder for ITS tracker MI
+// Origin: Marian Ivanov, CERN, Marian.Ivanov@cern.ch
+// Extraction from AliITStrackerMI: Andrea Dainese
+// Current support and development:
+//-------------------------------------------------------------------------
+
+#include <TMatrixD.h>
+#include <TTree.h>
+#include <TString.h>
+#include <TRandom.h>
+#include <TTreeStream.h>
+
+
+#include "AliLog.h"
+#include "AliESDVertex.h"
+#include "AliESDEvent.h"
+#include "AliESDtrack.h"
+#include "AliESDV0Params.h"
+#include "AliV0.h"
+#include "AliHelix.h"
+#include "AliITSRecPoint.h"
+#include "AliITSReconstructor.h"
+#include "AliITStrackerMI.h"
+#include "AliITSV0Finder.h"
+#include "AliKFParticle.h"
+#include "AliKFVertex.h"
+
+ClassImp(AliITSV0Finder)
+
+
+AliITSV0Finder::AliITSV0Finder():
+fDebugStreamer(0)
+ {
+ //Default constructor
+
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0)
+ fDebugStreamer = new TTreeSRedirector("ITSdebug.root");
+}
+
+//------------------------------------------------------------------------
+void AliITSV0Finder::UpdateTPCV0(const AliESDEvent *event,
+ AliITStrackerMI *tracker)
+{
+ //
+ //try to update, or reject TPC V0s
+ //
+ const Float_t kMinTgl2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl2();
+
+ TObjArray *trackHypothesys = tracker->GetTrackHypothesys();
+
+ Int_t nv0s = event->GetNumberOfV0s();
+ Int_t nitstracks = trackHypothesys->GetEntriesFast();
+
+ for (Int_t i=0;i<nv0s;i++){
+ AliESDv0 * vertex = event->GetV0(i);
+ Int_t ip = vertex->GetIndex(0);
+ Int_t im = vertex->GetIndex(1);
+ //
+ TObjArray * arrayp = (ip<nitstracks) ? (TObjArray*)trackHypothesys->At(ip):0;
+ TObjArray * arraym = (im<nitstracks) ? (TObjArray*)trackHypothesys->At(im):0;
+ AliITStrackMI * trackp = (arrayp!=0) ? (AliITStrackMI*)arrayp->At(0):0;
+ AliITStrackMI * trackm = (arraym!=0) ? (AliITStrackMI*)arraym->At(0):0;
+ //
+ //
+ if (trackp){
+ if (trackp->GetNumberOfClusters()+trackp->GetNDeadZone()>5.5){
+ if (trackp->GetConstrain()&&trackp->GetChi2MIP(0)<3) vertex->SetStatus(-100);
+ if (!trackp->GetConstrain()&&trackp->GetChi2MIP(0)<2) vertex->SetStatus(-100);
+ }
+ }
+
+ if (trackm){
+ if (trackm->GetNumberOfClusters()+trackm->GetNDeadZone()>5.5){
+ if (trackm->GetConstrain()&&trackm->GetChi2MIP(0)<3) vertex->SetStatus(-100);
+ if (!trackm->GetConstrain()&&trackm->GetChi2MIP(0)<2) vertex->SetStatus(-100);
+ }
+ }
+ if (vertex->GetStatus()==-100) continue;
+ //
+ Double_t xrp[3]; vertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B.
+ Int_t clayer = tracker->GetNearestLayer(xrp); //I.B.
+ vertex->SetNBefore(clayer); //
+ vertex->SetChi2Before(9*clayer); //
+ vertex->SetNAfter(6-clayer); //
+ vertex->SetChi2After(0); //
+ //
+ if (clayer >1 ){ // calculate chi2 before vertex
+ Float_t chi2p = 0, chi2m=0;
+ //
+ if (trackp){
+ for (Int_t ilayer=0;ilayer<clayer;ilayer++){
+ if (trackp->GetClIndex(ilayer)>=0){
+ chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+
+ trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));
+ }
+ else{
+ chi2p+=9;
+ }
+ }
+ }else{
+ chi2p = 9*clayer;
+ }
+ //
+ if (trackm){
+ for (Int_t ilayer=0;ilayer<clayer;ilayer++){
+ if (trackm->GetClIndex(ilayer)>=0){
+ chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+
+ trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));
+ }
+ else{
+ chi2m+=9;
+ }
+ }
+ }else{
+ chi2m = 9*clayer;
+ }
+ vertex->SetChi2Before(TMath::Min(chi2p,chi2m));
+ if (TMath::Min(chi2p,chi2m)/Float_t(clayer)<4) vertex->SetStatus(-10); // track exist before vertex
+ }
+
+ if (clayer < 5 ){ // calculate chi2 after vertex
+ Float_t chi2p = 0, chi2m=0;
+ //
+ if (trackp&&TMath::Abs(trackp->GetTgl())<kMinTgl2){
+ for (Int_t ilayer=clayer;ilayer<6;ilayer++){
+ if (trackp->GetClIndex(ilayer)>=0){
+ chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+
+ trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer));
+ }
+ else{
+ chi2p+=9;
+ }
+ }
+ }else{
+ chi2p = 0;
+ }
+ //
+ if (trackm&&TMath::Abs(trackm->GetTgl())<kMinTgl2){
+ for (Int_t ilayer=clayer;ilayer<6;ilayer++){
+ if (trackm->GetClIndex(ilayer)>=0){
+ chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+
+ trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer));
+ }
+ else{
+ chi2m+=9;
+ }
+ }
+ }else{
+ chi2m = 0;
+ }
+ vertex->SetChi2After(TMath::Max(chi2p,chi2m));
+ if (TMath::Max(chi2m,chi2p)/Float_t(6-clayer)>9) vertex->SetStatus(-20); // track not found in ITS
+ }
+ }
+ //
+}
+//------------------------------------------------------------------------
+void AliITSV0Finder::FindV02(AliESDEvent *event,
+ AliITStrackerMI *tracker)
+{
+ //
+ // V0 finder
+ //
+ // Cuts on DCA - R dependent
+ // max distance DCA between 2 tracks cut
+ // maxDist = TMath::Min(kMaxDist,kMaxDist0+pvertex->GetRr()*kMaxDist);
+ //
+ const Bool_t kCheckPropagate = kFALSE;
+ const Float_t kMaxDist0 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist0();
+ const Float_t kMaxDist1 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist1();
+ const Float_t kMaxDist = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist();
+ const Float_t kMinPointAngle = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle();
+ const Float_t kMinPointAngle2 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle2();
+ const Float_t kMinR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinR();
+ const Float_t kMaxR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxR();
+ const Float_t kMinPABestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPABestConst();
+ const Float_t kMaxRBestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRBestConst();
+ const Float_t kCausality0Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCausality0Cut();
+ const Float_t kLikelihood01Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood01Cut();
+ const Float_t kLikelihood1Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood1Cut();
+ const Float_t kCombinedCut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCombinedCut();
+ const Float_t kMinClFullTrk= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClFullTrk();
+ const Float_t kMinTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl0();
+ const Float_t kMinRTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTgl0();
+
+ const Float_t kMinNormDistForbTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbTgl0();
+ const Float_t kMinClForb0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClForb0();
+ const Float_t kMinNormDistForb1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb1();
+ const Float_t kMinNormDistForb2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb2();
+ const Float_t kMinNormDistForb3= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb3();
+ const Float_t kMinNormDistForb4= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb4();
+ const Float_t kMinNormDistForb5= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb5();
+ const Float_t kMinNormDistForbProt= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbProt();
+ const Float_t kMaxPidProbPionForb= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxPidProbPionForb();
+
+ const Float_t kMinRTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTPCdensity();
+ const Float_t kMaxRTPCdensity0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity0();
+ const Float_t kMaxRTPCdensity10= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity10();
+ const Float_t kMaxRTPCdensity20= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity20();
+ const Float_t kMaxRTPCdensity30= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity30();
+
+
+ const Float_t kMinTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTPCdensity();
+ const Float_t kMinTgl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl1();
+
+ const Float_t kMinchi2before0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before0();
+ const Float_t kMinchi2before1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before1();
+ const Float_t kMinchi2after0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after0();
+ const Float_t kMinchi2after1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after1();
+ const Float_t kAddchi2SharedCl= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2SharedCl();
+ const Float_t kAddchi2NegCl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl0();
+ const Float_t kAddchi2NegCl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl1();
+
+ const Float_t kSigp0Par0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par0();
+ const Float_t kSigp0Par1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par1();
+ const Float_t kSigp0Par2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par2();
+
+ const Float_t kSigpPar0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar0();
+ const Float_t kSigpPar1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar1();
+ const Float_t kSigpPar2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar2();
+ const Float_t kMaxDcaLh0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDcaLh0();
+
+
+ TObjArray *trackHypothesys = tracker->GetTrackHypothesys();
+ TObjArray *bestHypothesys = tracker->GetBestHypothesys();
+ TObjArray *originalTracks = tracker->GetOriginal();
+ //
+ //
+ TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());
+ Int_t ntracks = event->GetNumberOfTracks();
+ Int_t nitstracks = originalTracks->GetEntriesFast();
+ originalTracks->Expand(ntracks);
+ trackHypothesys->Expand(ntracks);
+ bestHypothesys->Expand(ntracks);
+ //
+ AliHelix * helixes = new AliHelix[ntracks+2];
+ TObjArray trackarray(ntracks+2); //array with tracks - with vertex constrain
+ TObjArray trackarrayc(ntracks+2); //array of "best tracks" - without vertex constrain
+ TObjArray trackarrayl(ntracks+2); //array of "longest tracks" - without vertex constrain
+ //Change from Bool_t to Int_t for optimization
+ // Int_t forbN=0;
+ // Int_t * forbidden = new Int_t [ntracks+2];
+ Bool_t * forbidden = new Bool_t [ntracks+2];
+ Int_t *itsmap = new Int_t [ntracks+2];
+ Float_t *dist = new Float_t[ntracks+2];
+ Float_t *normdist0 = new Float_t[ntracks+2];
+ Float_t *normdist1 = new Float_t[ntracks+2];
+ Float_t *normdist = new Float_t[ntracks+2];
+ Float_t *norm = new Float_t[ntracks+2];
+ Float_t *maxr = new Float_t[ntracks+2];
+ Float_t *minr = new Float_t[ntracks+2];
+ Float_t *minPointAngle= new Float_t[ntracks+2];
+ //
+ AliV0 *pvertex = new AliV0;
+ AliITStrackMI * dummy= new AliITStrackMI;
+ dummy->SetLabel(0);
+ AliITStrackMI trackat0; //temporary track for DCA calculation
+ //
+ Float_t primvertex[3]={static_cast<Float_t>(tracker->GetX()),static_cast<Float_t>(tracker->GetY()),static_cast<Float_t>(tracker->GetZ())};
+ //
+ // make ITS - ESD map
+ //
+ for (Int_t itrack=0;itrack<ntracks+2;itrack++) {
+ itsmap[itrack] = -1;
+ // forbidden[itrack] = 0;
+ forbidden[itrack] = kFALSE;
+ maxr[itrack] = kMaxR;
+ minr[itrack] = kMinR;
+ minPointAngle[itrack] = kMinPointAngle;
+ }
+ for (Int_t itrack=0;itrack<nitstracks;itrack++){
+ AliITStrackMI * original = (AliITStrackMI*)(originalTracks->At(itrack));
+ Int_t esdindex = original->GetESDtrack()->GetID();
+ itsmap[esdindex] = itrack;
+ }
+ //
+ // create ITS tracks from ESD tracks if not done before
+ //
+ for (Int_t itrack=0;itrack<ntracks;itrack++){
+ if (itsmap[itrack]>=0) continue;
+ AliITStrackMI * tpctrack = new AliITStrackMI(*(event->GetTrack(itrack)));
+ //tpctrack->fD[0] = tpctrack->GetD(GetX(),GetY());
+ //tpctrack->fD[1] = tpctrack->GetZat(GetX())-GetZ();
+ tpctrack->GetDZ(tracker->GetX(),tracker->GetY(),tracker->GetZ(),tpctrack->GetDP()); //I.B.
+ if (tpctrack->GetD(0)<20 && tpctrack->GetD(1)<20){
+ // tracks which can reach inner part of ITS
+ // propagate track to outer its volume - with correction for material
+ AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(tpctrack);
+ }
+ itsmap[itrack] = nitstracks;
+ originalTracks->AddAt(tpctrack,nitstracks);
+ nitstracks++;
+ }
+ //
+ // fill temporary arrays
+ //
+ for (Int_t itrack=0;itrack<ntracks;itrack++){
+ AliESDtrack * esdtrack = event->GetTrack(itrack);
+ Int_t itsindex = itsmap[itrack];
+ AliITStrackMI *original = (AliITStrackMI*)originalTracks->At(itsmap[itrack]);
+ if (!original) continue;
+ AliITStrackMI *bestConst = 0;
+ AliITStrackMI *bestLong = 0;
+ AliITStrackMI *best = 0;
+ //
+ //
+ TObjArray * array = (TObjArray*) trackHypothesys->At(itsindex);
+ Int_t hentries = (array==0) ? 0 : array->GetEntriesFast();
+ // Get best track with vertex constrain
+ for (Int_t ih=0;ih<hentries;ih++){
+ AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);
+ if (!trackh->GetConstrain()) continue;
+ if (!bestConst) bestConst = trackh;
+ if (trackh->GetNumberOfClusters()>kMinClFullTrk ){
+ bestConst = trackh; // full track - with minimal chi2
+ break;
+ }
+ if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()) continue;
+ bestConst = trackh;
+ break;
+ }
+ // Get best long track without vertex constrain and best track without vertex constrain
+ for (Int_t ih=0;ih<hentries;ih++){
+ AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih);
+ if (trackh->GetConstrain()) continue;
+ if (!best) best = trackh;
+ if (!bestLong) bestLong = trackh;
+ if (trackh->GetNumberOfClusters()>kMinClFullTrk){
+ bestLong = trackh; // full track - with minimal chi2
+ break;
+ }
+ if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()) continue;
+ bestLong = trackh;
+ }
+ if (!best) {
+ best = original;
+ bestLong = original;
+ }
+ //I.B. trackat0 = *bestLong;
+ new (&trackat0) AliITStrackMI(*bestLong);
+ Double_t xx,yy,zz,alpha;
+ if (!bestLong->GetGlobalXYZat(bestLong->GetX(),xx,yy,zz)) continue;
+
+
+ alpha = TMath::ATan2(yy,xx);
+ // if (!trackat0.Propagate(alpha,0)) continue;
+ // trackat0.Propagate(alpha,0); //PH The check on the return value is temporarily disabled (bug 45751)
+ if(!trackat0.Propagate(alpha,0) && kCheckPropagate)continue;
+ // calculate normalized distances to the vertex
+ //
+ Float_t ptfac = (1.+100.*TMath::Abs(trackat0.GetC()));
+ if ( bestLong->GetNumberOfClusters()>3 ){
+ dist[itsindex] = trackat0.GetY();
+ norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());
+ normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);
+ normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));
+ normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);
+ if (!bestConst){
+ if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<6) normdist[itsindex]*=2.;
+ if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<5) normdist[itsindex]*=2.;
+ if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<4) normdist[itsindex]*=2.;
+ }else{
+ if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<6) normdist[itsindex]*=1.5;
+ if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<5) normdist[itsindex]*=1.5;
+ }
+ }
+ else{
+ if (bestConst&&bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>4.5){
+ dist[itsindex] = bestConst->GetD(0);
+ norm[itsindex] = bestConst->GetDnorm(0);
+ normdist0[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);
+ normdist1[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]);
+ normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);
+ }else{
+ dist[itsindex] = trackat0.GetY();
+ norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2());
+ normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]);
+ normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2())));
+ normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]);
+ if (TMath::Abs(trackat0.GetTgl())>kMinTgl0){
+ if (normdist[itsindex]<kMinNormDistForbTgl0){
+ // forbN=1;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]>kMinNormDistForbTgl0) {
+ minr[itsindex] = TMath::Max(kMinRTgl0,minr[itsindex]);
+ }
+ }
+ }
+ }
+
+
+ //
+ //-----------------------------------------------------------
+ //Forbid primary track candidates -
+ //
+ //treetr->SetAlias("forbidden0","Tr0.fN<4&&Tr1.fN+Tr1.fNDeadZone>4.5");
+ //treetr->SetAlias("forbidden1","ND<3&&Tr1.fN+Tr1.fNDeadZone>5.5");
+ //treetr->SetAlias("forbidden2","ND<2&&Tr1.fClIndex[0]>0&&Tr1.fClIndex[0]>0");
+ //treetr->SetAlias("forbidden3","ND<1&&Tr1.fClIndex[0]>0");
+ //treetr->SetAlias("forbidden4","ND<4&&Tr1.fNormChi2[0]<2");
+ //treetr->SetAlias("forbidden5","ND<5&&Tr1.fNormChi2[0]<1");
+ //-----------------------------------------------------------
+ if (bestConst){
+ if (bestLong->GetNumberOfClusters()<4 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>kMinClForb0){
+ // forbN=2;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]<kMinNormDistForb1 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>5.5){
+ // forbN=3;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]<kMinNormDistForb2 && bestConst->GetClIndex(0)>=0 && bestConst->GetClIndex(1)>=0 ){
+ // forbN=4;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]<kMinNormDistForb3 && bestConst->GetClIndex(0)>=0){
+ // forbN=5;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]<kMinNormDistForb4 && bestConst->GetNormChi2(0)<2){
+ // forbN=6;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (normdist[itsindex]<kMinNormDistForb5 && bestConst->GetNormChi2(0)<1){
+ // forbN=7;
+ // forbidden[itsindex]+=1<<forbN;
+ forbidden[itsindex]=kTRUE;
+ }
+ if (bestConst->GetNormChi2(0)<2.5) {
+ minPointAngle[itsindex]= kMinPABestConst;
+ maxr[itsindex] = kMaxRBestConst;
+ }
+ }
+ //
+ //forbid daughter kink candidates
+ //
+ if (esdtrack->GetKinkIndex(0)>0) forbidden[itsindex] = kTRUE;
+ Bool_t isElectron = kTRUE;
+ Bool_t isProton = kTRUE;
+ Double_t pid[5];
+ esdtrack->GetESDpid(pid);
+ for (Int_t i=1;i<5;i++){
+ if (pid[0]<pid[i]) isElectron= kFALSE;
+ if (pid[4]<pid[i]) isProton= kFALSE;
+ }
+ if (isElectron){
+ forbidden[itsindex]=kFALSE;
+ normdist[itsindex]*=-1;
+ }
+ if (isProton){
+ if (normdist[itsindex]>kMinNormDistForbProt) forbidden[itsindex]=kFALSE;
+ normdist[itsindex]*=-1;
+ }
+
+ // We allow all tracks that are not pions
+ if( (pid[1]+pid[2])< kMaxPidProbPionForb ){
+ forbidden[itsindex]=kFALSE;
+ }
+
+ //
+ // Causality cuts in TPC volume
+ //
+ if (esdtrack->GetTPCdensity(0,10) >kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity0),maxr[itsindex]);
+ if (esdtrack->GetTPCdensity(10,30)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity10),maxr[itsindex]);
+ if (esdtrack->GetTPCdensity(20,40)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity20),maxr[itsindex]);
+ if (esdtrack->GetTPCdensity(30,50)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity30),maxr[itsindex]);
+ //
+ if (esdtrack->GetTPCdensity(0,60)<0.4&&bestLong->GetNumberOfClusters()<3) minr[itsindex]=kMinRTPCdensity;
+ //
+ //
+
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){
+ cstream<<"Track"<<
+ "Tr0.="<<best<<
+ "Tr1.="<<((bestConst)? bestConst:dummy)<<
+ "Tr2.="<<bestLong<<
+ "Tr3.="<<&trackat0<<
+ "Esd.="<<esdtrack<<
+ "Dist="<<dist[itsindex]<<
+ "ND0="<<normdist0[itsindex]<<
+ "ND1="<<normdist1[itsindex]<<
+ "ND="<<normdist[itsindex]<<
+ "Pz="<<primvertex[2]<<
+ "Forbid="<<forbidden[itsindex]<<
+ "\n";
+ //
+ }
+ trackarray.AddAt(best,itsindex);
+ trackarrayc.AddAt(bestConst,itsindex);
+ trackarrayl.AddAt(bestLong,itsindex);
+ new (&helixes[itsindex]) AliHelix(*best);
+ }
+ //
+ //
+ //
+ // first iterration of V0 finder
+ //
+ // AM Comment out for optimization
+ // Int_t rejectBase=0;
+ // Int_t cutN=0;
+
+ for (Int_t iesd0=0;iesd0<ntracks;iesd0++){
+ Int_t itrack0 = itsmap[iesd0];
+ //-AM comment for optimization and store the forbidden value in the debug streamer
+ if (forbidden[itrack0]) continue;
+ AliITStrackMI * btrack0 = (AliITStrackMI*)trackarray.At(itrack0);
+ if (!btrack0) continue;
+ AliITStrackMI *trackc0 = (AliITStrackMI*)trackarrayc.At(itrack0);
+ //
+ for (Int_t iesd1=iesd0+1;iesd1<ntracks;iesd1++){
+ Int_t itrack1 = itsmap[iesd1];
+ //-AM comment for optimization and store the forbidden value in the debug streamer
+ if (forbidden[itrack1]) continue;
+
+ AliITStrackMI * btrack1 = (AliITStrackMI*)trackarray.At(itrack1);
+ if (!btrack1) continue;
+
+ if ( (btrack0->GetSign()==btrack1->GetSign()) && !AliITSReconstructor::GetRecoParam()->GetStoreLikeSignV0s()) continue;
+
+ Bool_t isGold = kFALSE;
+ if (TMath::Abs(TMath::Abs(btrack0->GetLabel())-TMath::Abs(btrack1->GetLabel()))==1){
+ isGold = kTRUE;
+ }
+ // rejectBase=0;
+ AliITStrackMI *trackc1 = (AliITStrackMI*)trackarrayc.At(itrack1);
+ AliHelix &h1 = helixes[itrack0];
+ AliHelix &h2 = helixes[itrack1];
+ //
+ // find linear distance
+ Double_t rmin =0;
+ //
+ //
+ //
+ Double_t phase[2][2]={{0.,0.},{0.,0.}};
+ Double_t radius[2]={0.,0.};
+ Int_t points = h1.GetRPHIintersections(h2, phase, radius);
+ if (points==0) {
+ // cutN=1;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ Double_t delta[2]={1000000,1000000};
+ rmin = radius[0];
+ h1.ParabolicDCA(h2,phase[0][0],phase[0][1],radius[0],delta[0]);
+ if (points==2){
+ if (radius[1]<rmin) rmin = radius[1];
+ h1.ParabolicDCA(h2,phase[1][0],phase[1][1],radius[1],delta[1]);
+ }
+ rmin = TMath::Sqrt(rmin);
+ Double_t distance = 0;
+ Double_t radiusC = 0;
+ Int_t iphase = 0;
+ if (points==1 || delta[0]<delta[1]){
+ distance = TMath::Sqrt(delta[0]);
+ radiusC = TMath::Sqrt(radius[0]);
+ }else{
+ distance = TMath::Sqrt(delta[1]);
+ radiusC = TMath::Sqrt(radius[1]);
+ iphase=1;
+ }
+ if (radiusC<TMath::Max(minr[itrack0],minr[itrack1])){
+ // cutN=2;
+ //rejectBase+=1<<cutN;
+ continue;
+ }
+ if (radiusC>TMath::Min(maxr[itrack0],maxr[itrack1])){
+ // cutN=3;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ Float_t maxDist = TMath::Min(kMaxDist,Float_t(kMaxDist0+radiusC*kMaxDist1));
+ if (distance>maxDist){
+ // cutN=4;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ Float_t pointAngle = h1.GetPointAngle(h2,phase[iphase],primvertex);
+ if (pointAngle<TMath::Max(minPointAngle[itrack0],minPointAngle[itrack1])) {
+ // cutN=5;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ //
+ //
+ // Double_t distance = TestV0(h1,h2,pvertex,rmin);
+ //
+ // if (distance>maxDist) continue;
+ // if (pvertex->GetRr()<kMinR) continue;
+ // if (pvertex->GetRr()>kMaxR) continue;
+ AliITStrackMI * track0=btrack0;
+ AliITStrackMI * track1=btrack1;
+ // if (pvertex->GetRr()<3.5){
+ if (radiusC<3.5){
+ //use longest tracks inside the pipe
+ track0 = (AliITStrackMI*)trackarrayl.At(itrack0);
+ track1 = (AliITStrackMI*)trackarrayl.At(itrack1);
+ }
+ //
+ //
+
+
+
+ pvertex->SetParamN(*track0);
+ pvertex->SetParamP(*track1);
+ pvertex->Update(primvertex);
+ pvertex->SetClusters(track0->ClIndex(),track1->ClIndex()); // register clusters
+
+ // Define gamma, K0, lambda and lambda_bar from the decay particles and calculate the chi2
+ AliKFVertex vertexKF;
+
+ vertexKF.X() = tracker->GetX();
+ vertexKF.Y() = tracker->GetY();
+ vertexKF.Z() = tracker->GetZ();
+ vertexKF.Covariance(0,0) = tracker->GetSigmaX()*tracker->GetSigmaX();
+ vertexKF.Covariance(1,1) = tracker->GetSigmaY()*tracker->GetSigmaY();
+ vertexKF.Covariance(2,2) = tracker->GetSigmaZ()*tracker->GetSigmaZ();
+
+ AliKFParticle elecKF( *(pvertex->GetParamN()) ,11);
+ AliKFParticle posiKF( *(pvertex->GetParamP()) ,-11);
+ AliKFParticle pipKF( *(pvertex->GetParamN()) , 211);
+ AliKFParticle pinKF( *(pvertex->GetParamP()) ,-211);
+ AliKFParticle protKF( *(pvertex->GetParamP()) ,2212);
+ AliKFParticle aproKF ( *(pvertex->GetParamN()) ,-2212);
+
+
+ // Gamma
+ AliKFParticle gamKF(elecKF,posiKF);
+ gamKF.SetProductionVertex(vertexKF);
+
+ Float_t gamKFchi2 = 1000;
+ if ( gamKF.GetNDF()!=0 ){
+ gamKFchi2 = gamKF.GetChi2()/gamKF.GetNDF();
+ }
+
+ Float_t massG=0.;
+ Float_t sigmaMG=0.001;
+ gamKF.SetMassConstraint(massG,sigmaMG);
+
+ Float_t gamKFchi2C = 1000;
+ if ( gamKF.GetNDF()!=0 ){
+ gamKFchi2C = gamKF.GetChi2()/gamKF.GetNDF();
+ }
+
+ //K0 short
+ AliKFParticle k0KF(pipKF,pinKF);
+ k0KF.SetProductionVertex(vertexKF);
+
+ Float_t k0KFchi2 = 1000;
+ if ( k0KF.GetNDF()!=0 ){
+ k0KFchi2 = k0KF.GetChi2()/k0KF.GetNDF();
+ }
+
+ //Lambda
+ AliKFParticle lambdaKF(protKF,pinKF);
+ lambdaKF.SetProductionVertex(vertexKF);
+
+ Float_t lambdaKFchi2 = 1000;
+ if ( lambdaKF.GetNDF()!=0 ){
+ lambdaKFchi2 = lambdaKF.GetChi2()/lambdaKF.GetNDF();
+ }
+
+ //Lambda_bar
+ AliKFParticle alambKF(aproKF,pipKF);
+ alambKF.SetProductionVertex(vertexKF);
+
+ Float_t alambKFchi2 = 1000;
+ if ( alambKF.GetNDF()!=0 ){
+ alambKFchi2 = alambKF.GetChi2()/alambKF.GetNDF();
+ }
+
+
+
+
+
+ if (pvertex->GetRr()<kMinR){
+ // cutN=6;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ if (pvertex->GetRr()>kMaxR){
+ // cutN=7;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+ if (pvertex->GetV0CosineOfPointingAngle()<kMinPointAngle){
+ // cutN=8;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+//Bo: if (pvertex->GetDist2()>maxDist) continue;
+
+ if (pvertex->GetDcaV0Daughters()>maxDist){
+ // cutN=9;
+ // rejectBase+=1<<cutN;
+ continue;
+ }
+//Bo: pvertex->SetLab(0,track0->GetLabel());
+//Bo: pvertex->SetLab(1,track1->GetLabel());
+ pvertex->SetIndex(0,track0->GetESDtrack()->GetID());
+ pvertex->SetIndex(1,track1->GetESDtrack()->GetID());
+ //
+ AliITStrackMI * htrackc0 = trackc0 ? trackc0:dummy;
+ AliITStrackMI * htrackc1 = trackc1 ? trackc1:dummy;
+
+ //
+ //
+ TObjArray * array0b = (TObjArray*)bestHypothesys->At(itrack0);
+ if (!array0b&&pvertex->GetRr()<40 && TMath::Abs(track0->GetTgl())<kMinTgl1) {
+ tracker->SetCurrentEsdTrack(itrack0);
+ tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack0),itrack0, kFALSE);
+ }
+ TObjArray * array1b = (TObjArray*)bestHypothesys->At(itrack1);
+ if (!array1b&&pvertex->GetRr()<40 && TMath::Abs(track1->GetTgl())<kMinTgl1) {
+ tracker->SetCurrentEsdTrack(itrack1);
+ tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack1),itrack1, kFALSE);
+ }
+ //
+ AliITStrackMI * track0b = (AliITStrackMI*)originalTracks->At(itrack0);
+ AliITStrackMI * track1b = (AliITStrackMI*)originalTracks->At(itrack1);
+ AliITStrackMI * track0l = (AliITStrackMI*)originalTracks->At(itrack0);
+ AliITStrackMI * track1l = (AliITStrackMI*)originalTracks->At(itrack1);
+
+ Float_t minchi2before0=kMinchi2before0;
+ Float_t minchi2before1=kMinchi2before1;
+ Float_t minchi2after0 =kMinchi2after0;
+ Float_t minchi2after1 =kMinchi2after1;
+ Double_t xrp[3]; pvertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B.
+ Int_t maxLayer = tracker->GetNearestLayer(xrp); //I.B.
+
+ if (array0b) for (Int_t i=0;i<5;i++){
+ // best track after vertex
+ AliITStrackMI * btrack = (AliITStrackMI*)array0b->At(i);
+ if (!btrack) continue;
+ if (btrack->GetNumberOfClusters()>track0l->GetNumberOfClusters()) track0l = btrack;
+ // if (btrack->fX<pvertex->GetRr()-2.-0.5/(0.1+pvertex->GetAnglep()[2])) {
+ if (btrack->GetX()<pvertex->GetRr()-2.) {
+ if ( (maxLayer>i+2|| (i==0)) && btrack->GetNumberOfClusters()==(6-i)&&i<3){
+ Float_t sumchi2= 0;
+ Float_t sumn = 0;
+ if (maxLayer<3){ // take prim vertex as additional measurement
+ if (normdist[itrack0]>0 && htrackc0){
+ sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack0]*normdist[itrack0],16.);
+ }else{
+ sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack0]*normdist[itrack0]+3.),16.);
+ }
+ sumn += 3-maxLayer;
+ }
+ for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){
+ sumn+=1.;
+ if (btrack->GetClIndex(ilayer)<0){
+ sumchi2+=kAddchi2NegCl0;
+ continue;
+ }else{
+ Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);
+ for (Int_t itrack=0;itrack<4;itrack++){
+ AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);
+ if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack0){
+ sumchi2+=kAddchi2SharedCl; //shared cluster
+ break;
+ }
+ }
+ sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));
+ sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer));
+ }
+ }
+ sumchi2/=sumn;
+ if (sumchi2<minchi2before0) minchi2before0=sumchi2;
+ }
+ continue; //safety space - Geo manager will give exact layer
+ }
+ track0b = btrack;
+ minchi2after0 = btrack->GetNormChi2(i);
+ break;
+ }
+ if (array1b) for (Int_t i=0;i<5;i++){
+ // best track after vertex
+ AliITStrackMI * btrack = (AliITStrackMI*)array1b->At(i);
+ if (!btrack) continue;
+ if (btrack->GetNumberOfClusters()>track1l->GetNumberOfClusters()) track1l = btrack;
+ // if (btrack->fX<pvertex->GetRr()-2-0.5/(0.1+pvertex->GetAnglep()[2])){
+ if (btrack->GetX()<pvertex->GetRr()-2){
+ if ((maxLayer>i+2 || (i==0))&&btrack->GetNumberOfClusters()==(6-i)&&(i<3)){
+ Float_t sumchi2= 0;
+ Float_t sumn = 0;
+ if (maxLayer<3){ // take prim vertex as additional measurement
+ if (normdist[itrack1]>0 && htrackc1){
+ sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack1]*normdist[itrack1],16.);
+ }else{
+ sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack1]*normdist[itrack1]+3.),16.);
+ }
+ sumn += 3-maxLayer;
+ }
+ for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){
+ sumn+=1.;
+ if (btrack->GetClIndex(ilayer)<0){
+ sumchi2+=kAddchi2NegCl1;
+ continue;
+ }else{
+ Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff);
+ for (Int_t itrack=0;itrack<4;itrack++){
+ AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer);
+ if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack1){
+ sumchi2+=kAddchi2SharedCl; //shared cluster
+ break;
+ }
+ }
+ sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer));
+ sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer));
+ }
+ }
+ sumchi2/=sumn;
+ if (sumchi2<minchi2before1) minchi2before1=sumchi2;
+ }
+ continue; //safety space - Geo manager will give exact layer
+ }
+ track1b = btrack;
+ minchi2after1 = btrack->GetNormChi2(i);
+ break;
+ }
+ //
+ // position resolution - used for DCA cut
+ Float_t sigmad = track0b->GetSigmaY2()+track0b->GetSigmaZ2()+track1b->GetSigmaY2()+track1b->GetSigmaZ2()+
+ (track0b->GetX()-pvertex->GetRr())*(track0b->GetX()-pvertex->GetRr())*(track0b->GetSigmaSnp2()+track0b->GetSigmaTgl2())+
+ (track1b->GetX()-pvertex->GetRr())*(track1b->GetX()-pvertex->GetRr())*(track1b->GetSigmaSnp2()+track1b->GetSigmaTgl2());
+ sigmad =TMath::Sqrt(sigmad)+0.04;
+ if (pvertex->GetRr()>50){
+ Double_t cov0[15],cov1[15];
+ track0b->GetESDtrack()->GetInnerExternalCovariance(cov0);
+ track1b->GetESDtrack()->GetInnerExternalCovariance(cov1);
+ sigmad = cov0[0]+cov0[2]+cov1[0]+cov1[2]+
+ (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov0[5]+cov0[9])+
+ (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov1[5]+cov1[9]);
+ sigmad =TMath::Sqrt(sigmad)+0.05;
+ }
+ //
+ AliV0 vertex2;
+ vertex2.SetParamN(*track0b);
+ vertex2.SetParamP(*track1b);
+ vertex2.Update(primvertex);
+ //Bo: if (vertex2.GetDist2()<=pvertex->GetDist2()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){
+ if (vertex2.GetDcaV0Daughters()<=pvertex->GetDcaV0Daughters()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){
+ pvertex->SetParamN(*track0b);
+ pvertex->SetParamP(*track1b);
+ pvertex->Update(primvertex);
+ pvertex->SetClusters(track0b->ClIndex(),track1b->ClIndex()); // register clusters
+ pvertex->SetIndex(0,track0->GetESDtrack()->GetID());
+ pvertex->SetIndex(1,track1->GetESDtrack()->GetID());
+ }
+ pvertex->SetDistSigma(sigmad);
+ //Bo: pvertex->SetDistNorm(pvertex->GetDist2()/sigmad);
+ pvertex->SetNormDCAPrim(normdist[itrack0],normdist[itrack1]);
+ //
+ // define likelihhod and causalities
+ //
+ Float_t pa0=1, pa1=1, pb0=0.26, pb1=0.26;
+ if (maxLayer<1){
+ Float_t fnorm0 = normdist[itrack0];
+ if (fnorm0<0) fnorm0*=-3;
+ Float_t fnorm1 = normdist[itrack1];
+ if (fnorm1<0) fnorm1*=-3;
+ if ((pvertex->GetAnglep()[2]>0.1) || ( (pvertex->GetRr()<10.5)&& pvertex->GetAnglep()[2]>0.05 ) || (pvertex->GetRr()<3)){
+ pb0 = TMath::Exp(-TMath::Min(fnorm0,Float_t(16.))/12.);
+ pb1 = TMath::Exp(-TMath::Min(fnorm1,Float_t(16.))/12.);
+ }
+ pvertex->SetChi2Before(normdist[itrack0]);
+ pvertex->SetChi2After(normdist[itrack1]);
+ pvertex->SetNAfter(0);
+ pvertex->SetNBefore(0);
+ }else{
+ pvertex->SetChi2Before(minchi2before0);
+ pvertex->SetChi2After(minchi2before1);
+ if (pvertex->GetAnglep()[2]>0.1 || ( pvertex->GetRr()<10.5 && pvertex->GetAnglep()[2]>0.05) || pvertex->GetRr()<3){
+ pb0 = TMath::Exp(-TMath::Min(minchi2before0,Float_t(16))/12.);
+ pb1 = TMath::Exp(-TMath::Min(minchi2before1,Float_t(16))/12.);
+ }
+ pvertex->SetNAfter(maxLayer);
+ pvertex->SetNBefore(maxLayer);
+ }
+ if (pvertex->GetRr()<90){
+ pa0 *= TMath::Min(track0->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));
+ pa1 *= TMath::Min(track1->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.));
+ }
+ if (pvertex->GetRr()<20){
+ pa0 *= (0.2+TMath::Exp(-TMath::Min(minchi2after0,Float_t(16))/8.))/1.2;
+ pa1 *= (0.2+TMath::Exp(-TMath::Min(minchi2after1,Float_t(16))/8.))/1.2;
+ }
+ //
+ pvertex->SetCausality(pb0,pb1,pa0,pa1);
+ //
+ // Likelihood calculations - apply cuts
+ //
+
+ // AM - Comment out for optimization and store the v0OK value
+ // Int_t v0OK = 0;
+ // Int_t cutOKN=0;
+ Bool_t v0OK = kTRUE;
+
+
+ Float_t p12 = pvertex->GetParamP()->GetParameter()[4]*pvertex->GetParamP()->GetParameter()[4];
+ p12 += pvertex->GetParamN()->GetParameter()[4]*pvertex->GetParamN()->GetParameter()[4];
+ p12 = TMath::Sqrt(p12); // "mean" momenta
+
+ Float_t sigmap0 = kSigp0Par0+kSigp0Par1/(kSigp0Par2+pvertex->GetRr());
+ Float_t sigmap = kSigpPar0*sigmap0*(kSigpPar1+kSigpPar2*p12); // "resolution: of point angle - as a function of radius and momenta
+
+
+ Float_t causalityA = (1.0-pvertex->GetCausalityP()[0])*(1.0-pvertex->GetCausalityP()[1]);
+ Float_t causalityB = TMath::Sqrt(TMath::Min(pvertex->GetCausalityP()[2],Double_t(0.7))*
+ TMath::Min(pvertex->GetCausalityP()[3],Double_t(0.7)));
+ //
+ //Bo: Float_t likelihood0 = (TMath::Exp(-pvertex->GetDistNorm())+0.1) *(pvertex->GetDist2()<0.5)*(pvertex->GetDistNorm()<5);
+ Float_t lDistNorm = pvertex->GetDcaV0Daughters()/pvertex->GetDistSigma();
+ Float_t likelihood0 = (TMath::Exp(-lDistNorm)+0.1) *(pvertex->GetDcaV0Daughters()<kMaxDcaLh0)*(lDistNorm<5);
+
+ Float_t likelihood1 = TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/sigmap)+
+ 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(4.*sigmap))+
+ 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(8.*sigmap))+
+ 0.1*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/0.01);
+ //
+ if (causalityA<kCausality0Cut){
+ // cutOKN=1;
+ // v0OK += 1<<cutOKN;
+ v0OK = kFALSE;
+ }
+ if (TMath::Sqrt(likelihood0*likelihood1)<kLikelihood01Cut){
+ // cutOKN=2;
+ // v0OK += 1<<cutOKN;
+ v0OK = kFALSE;
+ }
+ if (likelihood1<kLikelihood1Cut){
+ // cutOKN=3;
+ // v0OK += 1<<cutOKN;
+ v0OK = kFALSE;
+ }
+ if (TMath::Power(likelihood0*likelihood1*causalityB,0.33)<kCombinedCut){
+ // cutOKN=4;
+ // v0OK += 1<<cutOKN;
+ v0OK = kFALSE;
+ }
+ //
+ //
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){
+ Bool_t gold = TMath::Abs(TMath::Abs(track0->GetLabel())-TMath::Abs(track1->GetLabel()))==1;
+ cstream<<"It0"<<
+ "Tr0.="<<track0<< //best without constrain
+ "Tr1.="<<track1<< //best without constrain
+ "posiKF.="<<&posiKF<< //KF from track0
+ "elecKF.="<<&elecKF<< //KF from track1
+ "Tr0B.="<<track0b<< //best without constrain after vertex
+ "Tr1B.="<<track1b<< //best without constrain after vertex
+ "Tr0C.="<<htrackc0<< //best with constrain if exist
+ "Tr1C.="<<htrackc1<< //best with constrain if exist
+ "Tr0L.="<<track0l<< //longest best
+ "Tr1L.="<<track1l<< //longest best
+ "Esd0.="<<track0->GetESDtrack()<< // esd track0 params
+ "Esd1.="<<track1->GetESDtrack()<< // esd track1 params
+ "V0.="<<pvertex<< //vertex properties
+ "V0b.="<<&vertex2<< //vertex properties at "best" track
+ "gamKF.="<<&gamKF<< //KF from pvertex
+ "k0KF.="<<&k0KF<< //KF from pvertex
+ "lambdaKF.="<<&lambdaKF<< //KF from pvertex
+ "alambKF.="<<&lambdaKF<< //KF from pvertex
+ "gamKFchi2="<<gamKFchi2<< //Normalized chi2 from KF assuming gamma momther
+ "gamKFchi2C="<<gamKFchi2C<< //Normalized chi2 from KF assuming gamma mother+mass constraint
+ "k0KFchi2="<<k0KFchi2<< //Normalized chi2 from KF assuming K0 mother
+ "lambdaKFchi2="<<lambdaKFchi2<< //Normalized chi2 from KF assuming Lambda mother
+ "alambKFchi2="<<alambKFchi2<< //Normalized chi2 from KF assuming lambda_bar mother
+ "ND0="<<normdist[itrack0]<< //normalize distance for track0
+ "ND1="<<normdist[itrack1]<< //normalize distance for track1
+ "Gold="<<gold<< //
+ // "RejectBase="<<rejectBase<< //rejection in First itteration
+ "OK="<<v0OK<<
+ "rmin="<<rmin<<
+ "sigmad="<<sigmad<<
+ "Forbid0="<<forbidden[itrack0]<<
+ "Forbid1="<<forbidden[itrack1]<<
+ "\n";
+ }
+ // if (rejectBase>0) continue;
+ //if (forbidden[itrack0]>0 ||forbidden[itrack1]>0) continue;
+ //
+ pvertex->SetStatus(0);
+ // if (rejectBase) {
+ // pvertex->SetStatus(-100);
+ //}
+ if (pvertex->GetV0CosineOfPointingAngle()>kMinPointAngle2) {
+ //Bo: pvertex->SetESDindexes(track0->GetESDtrack()->GetID(),track1->GetESDtrack()->GetID());
+ pvertex->SetIndex(0,track0->GetESDtrack()->GetID());//Bo: consistency 0 for neg
+ pvertex->SetIndex(1,track1->GetESDtrack()->GetID());//Bo: consistency 1 for pos
+ // if (v0OK==0){
+ if (v0OK){
+ // AliV0vertex vertexjuri(*track0,*track1);
+ // vertexjuri.SetESDindexes(track0->fESDtrack->GetID(),track1->fESDtrack->GetID());
+ // event->AddV0(&vertexjuri);
+ pvertex->SetStatus(100);
+ }
+ pvertex->SetOnFlyStatus(kTRUE);
+ pvertex->ChangeMassHypothesis(kK0Short);
+ event->AddV0(pvertex);
+ }
+ }
+ }
+ //
+ //
+ // delete temporary arrays
+ //
+ delete[] forbidden;
+ delete[] minPointAngle;
+ delete[] maxr;
+ delete[] minr;
+ delete[] norm;
+ delete[] normdist;
+ delete[] normdist1;
+ delete[] normdist0;
+ delete[] dist;
+ delete[] itsmap;
+ delete[] helixes;
+ delete pvertex;
+ delete dummy;
+}
+//------------------------------------------------------------------------
+void AliITSV0Finder::RefitV02(const AliESDEvent *event,
+ AliITStrackerMI *tracker)
+{
+ //
+ //try to refit V0s in the third path of the reconstruction
+ //
+ TTreeSRedirector &cstream = *(tracker->GetDebugStreamer());
+ //
+ Int_t nv0s = event->GetNumberOfV0s();
+ Float_t primvertex[3]={static_cast<Float_t>(tracker->GetX()),static_cast<Float_t>(tracker->GetY()),static_cast<Float_t>(tracker->GetZ())};
+ AliV0 v0temp;
+ for (Int_t iv0 = 0; iv0<nv0s;iv0++){
+ AliV0 * v0mi = (AliV0*)event->GetV0(iv0);
+ if (!v0mi) continue;
+ Int_t itrack0 = v0mi->GetIndex(0);
+ Int_t itrack1 = v0mi->GetIndex(1);
+ AliESDtrack *esd0 = event->GetTrack(itrack0);
+ AliESDtrack *esd1 = event->GetTrack(itrack1);
+ if (!esd0||!esd1) continue;
+ AliITStrackMI tpc0(*esd0);
+ AliITStrackMI tpc1(*esd1);
+ Double_t x,y,z; v0mi->GetXYZ(x,y,z); //I.B.
+ Double_t alpha =TMath::ATan2(y,x); //I.B.
+ if (v0mi->GetRr()>85){
+ if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){
+ v0temp.SetParamN(tpc0);
+ v0temp.SetParamP(tpc1);
+ v0temp.Update(primvertex);
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<
+ "V0.="<<v0mi<<
+ "V0refit.="<<&v0temp<<
+ "Tr0.="<<&tpc0<<
+ "Tr1.="<<&tpc1<<
+ "\n";
+ //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ v0mi->SetParamN(tpc0);
+ v0mi->SetParamP(tpc1);
+ v0mi->Update(primvertex);
+ }
+ }
+ continue;
+ }
+ if (v0mi->GetRr()>35){
+ AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);
+ AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1);
+ if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){
+ v0temp.SetParamN(tpc0);
+ v0temp.SetParamP(tpc1);
+ v0temp.Update(primvertex);
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<
+ "V0.="<<v0mi<<
+ "V0refit.="<<&v0temp<<
+ "Tr0.="<<&tpc0<<
+ "Tr1.="<<&tpc1<<
+ "\n";
+ //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ v0mi->SetParamN(tpc0);
+ v0mi->SetParamP(tpc1);
+ v0mi->Update(primvertex);
+ }
+ }
+ continue;
+ }
+ AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0);
+ AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1);
+ // if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){
+ if (tracker->RefitAt(v0mi->GetRr(),&tpc0, v0mi->GetClusters(0)) &&
+ tracker->RefitAt(v0mi->GetRr(),&tpc1, v0mi->GetClusters(1))){
+ v0temp.SetParamN(tpc0);
+ v0temp.SetParamP(tpc1);
+ v0temp.Update(primvertex);
+ if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<<
+ "V0.="<<v0mi<<
+ "V0refit.="<<&v0temp<<
+ "Tr0.="<<&tpc0<<
+ "Tr1.="<<&tpc1<<
+ "\n";
+ //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
+ v0mi->SetParamN(tpc0);
+ v0mi->SetParamP(tpc1);
+ v0mi->Update(primvertex);
+ }
+ }
+ }
+}
+//------------------------------------------------------------------------
+
+
+AliITSV0Finder::~AliITSV0Finder()
+{
+ //
+ //destructor
+ //
+ if (fDebugStreamer) {
+ //fDebugStreamer->Close();
+ delete fDebugStreamer;
+ }
+
+}