#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)
{
if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0)
fDebugStreamer = new TTreeSRedirector("ITSdebug.root");
}
void AliITSV0Finder::UpdateTPCV0(const AliESDEvent *event,
AliITStrackerMI *tracker)
{
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]);
Int_t clayer = tracker->GetNearestLayer(xrp);
vertex->SetNBefore(clayer);
vertex->SetChi2Before(9*clayer);
vertex->SetNAfter(6-clayer);
vertex->SetChi2After(0);
if (clayer >1 ){
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);
}
if (clayer < 5 ){
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);
}
}
}
void AliITSV0Finder::FindV02(AliESDEvent *event,
AliITStrackerMI *tracker)
{
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);
TObjArray trackarrayc(ntracks+2);
TObjArray trackarrayl(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;
Float_t primvertex[3]={static_cast<Float_t>(tracker->GetX()),static_cast<Float_t>(tracker->GetY()),static_cast<Float_t>(tracker->GetZ())};
for (Int_t itrack=0;itrack<ntracks+2;itrack++) {
itsmap[itrack] = -1;
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;
}
for (Int_t itrack=0;itrack<ntracks;itrack++){
if (itsmap[itrack]>=0) continue;
AliITStrackMI * tpctrack = new AliITStrackMI(*(event->GetTrack(itrack)));
tpctrack->GetDZ(tracker->GetX(),tracker->GetY(),tracker->GetZ(),tpctrack->GetDP());
if (tpctrack->GetD(0)<20 && tpctrack->GetD(1)<20){
AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(tpctrack);
}
itsmap[itrack] = nitstracks;
originalTracks->AddAt(tpctrack,nitstracks);
nitstracks++;
}
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();
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;
break;
}
if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()) continue;
bestConst = trackh;
break;
}
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;
break;
}
if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()) continue;
bestLong = trackh;
}
if (!best) {
best = original;
bestLong = original;
}
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) && kCheckPropagate)continue;
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){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]>kMinNormDistForbTgl0) {
minr[itsindex] = TMath::Max(kMinRTgl0,minr[itsindex]);
}
}
}
}
if (bestConst){
if (bestLong->GetNumberOfClusters()<4 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>kMinClForb0){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]<kMinNormDistForb1 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>5.5){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]<kMinNormDistForb2 && bestConst->GetClIndex(0)>=0 && bestConst->GetClIndex(1)>=0 ){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]<kMinNormDistForb3 && bestConst->GetClIndex(0)>=0){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]<kMinNormDistForb4 && bestConst->GetNormChi2(0)<2){
forbidden[itsindex]=kTRUE;
}
if (normdist[itsindex]<kMinNormDistForb5 && bestConst->GetNormChi2(0)<1){
forbidden[itsindex]=kTRUE;
}
if (bestConst->GetNormChi2(0)<2.5) {
minPointAngle[itsindex]= kMinPABestConst;
maxr[itsindex] = kMaxRBestConst;
}
}
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;
}
if( (pid[1]+pid[2])< kMaxPidProbPionForb ){
forbidden[itsindex]=kFALSE;
}
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);
}
for (Int_t iesd0=0;iesd0<ntracks;iesd0++){
Int_t itrack0 = itsmap[iesd0];
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];
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;
}
AliITStrackMI *trackc1 = (AliITStrackMI*)trackarrayc.At(itrack1);
AliHelix &h1 = helixes[itrack0];
AliHelix &h2 = helixes[itrack1];
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) {
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])){
continue;
}
if (radiusC>TMath::Min(maxr[itrack0],maxr[itrack1])){
continue;
}
Float_t maxDist = TMath::Min(kMaxDist,Float_t(kMaxDist0+radiusC*kMaxDist1));
if (distance>maxDist){
continue;
}
Float_t pointAngle = h1.GetPointAngle(h2,phase[iphase],primvertex);
if (pointAngle<TMath::Max(minPointAngle[itrack0],minPointAngle[itrack1])) {
continue;
}
AliITStrackMI * track0=btrack0;
AliITStrackMI * track1=btrack1;
if (radiusC<3.5){
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());
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);
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();
}
AliKFParticle k0KF(pipKF,pinKF);
k0KF.SetProductionVertex(vertexKF);
Float_t k0KFchi2 = 1000;
if ( k0KF.GetNDF()!=0 ){
k0KFchi2 = k0KF.GetChi2()/k0KF.GetNDF();
}
AliKFParticle lambdaKF(protKF,pinKF);
lambdaKF.SetProductionVertex(vertexKF);
Float_t lambdaKFchi2 = 1000;
if ( lambdaKF.GetNDF()!=0 ){
lambdaKFchi2 = lambdaKF.GetChi2()/lambdaKF.GetNDF();
}
AliKFParticle alambKF(aproKF,pipKF);
alambKF.SetProductionVertex(vertexKF);
Float_t alambKFchi2 = 1000;
if ( alambKF.GetNDF()!=0 ){
alambKFchi2 = alambKF.GetChi2()/alambKF.GetNDF();
}
if (pvertex->GetRr()<kMinR){
continue;
}
if (pvertex->GetRr()>kMaxR){
continue;
}
if (pvertex->GetV0CosineOfPointingAngle()<kMinPointAngle){
continue;
}
if (pvertex->GetDcaV0Daughters()>maxDist){
continue;
}
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]);
Int_t maxLayer = tracker->GetNearestLayer(xrp);
if (array0b) for (Int_t i=0;i<5;i++){
AliITStrackMI * btrack = (AliITStrackMI*)array0b->At(i);
if (!btrack) continue;
if (btrack->GetNumberOfClusters()>track0l->GetNumberOfClusters()) track0l = btrack;
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){
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;
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;
}
track0b = btrack;
minchi2after0 = btrack->GetNormChi2(i);
break;
}
if (array1b) for (Int_t i=0;i<5;i++){
AliITStrackMI * btrack = (AliITStrackMI*)array1b->At(i);
if (!btrack) continue;
if (btrack->GetNumberOfClusters()>track1l->GetNumberOfClusters()) track1l = btrack;
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){
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;
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;
}
track1b = btrack;
minchi2after1 = btrack->GetNormChi2(i);
break;
}
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);
if (vertex2.GetDcaV0Daughters()<=pvertex->GetDcaV0Daughters()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){
pvertex->SetParamN(*track0b);
pvertex->SetParamP(*track1b);
pvertex->Update(primvertex);
pvertex->SetClusters(track0b->ClIndex(),track1b->ClIndex());
pvertex->SetIndex(0,track0->GetESDtrack()->GetID());
pvertex->SetIndex(1,track1->GetESDtrack()->GetID());
}
pvertex->SetDistSigma(sigmad);
pvertex->SetNormDCAPrim(normdist[itrack0],normdist[itrack1]);
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);
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);
Float_t sigmap0 = kSigp0Par0+kSigp0Par1/(kSigp0Par2+pvertex->GetRr());
Float_t sigmap = kSigpPar0*sigmap0*(kSigpPar1+kSigpPar2*p12);
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)));
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){
v0OK = kFALSE;
}
if (TMath::Sqrt(likelihood0*likelihood1)<kLikelihood01Cut){
v0OK = kFALSE;
}
if (likelihood1<kLikelihood1Cut){
v0OK = kFALSE;
}
if (TMath::Power(likelihood0*likelihood1*causalityB,0.33)<kCombinedCut){
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<<
"Tr1.="<<track1<<
"posiKF.="<<&posiKF<<
"elecKF.="<<&elecKF<<
"Tr0B.="<<track0b<<
"Tr1B.="<<track1b<<
"Tr0C.="<<htrackc0<<
"Tr1C.="<<htrackc1<<
"Tr0L.="<<track0l<<
"Tr1L.="<<track1l<<
"Esd0.="<<track0->GetESDtrack()<<
"Esd1.="<<track1->GetESDtrack()<<
"V0.="<<pvertex<<
"V0b.="<<&vertex2<<
"gamKF.="<<&gamKF<<
"k0KF.="<<&k0KF<<
"lambdaKF.="<<&lambdaKF<<
"alambKF.="<<&lambdaKF<<
"gamKFchi2="<<gamKFchi2<<
"gamKFchi2C="<<gamKFchi2C<<
"k0KFchi2="<<k0KFchi2<<
"lambdaKFchi2="<<lambdaKFchi2<<
"alambKFchi2="<<alambKFchi2<<
"ND0="<<normdist[itrack0]<<
"ND1="<<normdist[itrack1]<<
"Gold="<<gold<<
"OK="<<v0OK<<
"rmin="<<rmin<<
"sigmad="<<sigmad<<
"Forbid0="<<forbidden[itrack0]<<
"Forbid1="<<forbidden[itrack1]<<
"\n";
}
pvertex->SetStatus(0);
if (pvertex->GetV0CosineOfPointingAngle()>kMinPointAngle2) {
pvertex->SetIndex(0,track0->GetESDtrack()->GetID());
pvertex->SetIndex(1,track1->GetESDtrack()->GetID());
if (v0OK){
pvertex->SetStatus(100);
}
pvertex->SetOnFlyStatus(kTRUE);
pvertex->ChangeMassHypothesis(kK0Short);
event->AddV0(pvertex);
}
}
}
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)
{
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);
Double_t alpha =TMath::ATan2(y,x);
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";
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";
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 (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";
if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){
v0mi->SetParamN(tpc0);
v0mi->SetParamP(tpc1);
v0mi->Update(primvertex);
}
}
}
}
AliITSV0Finder::~AliITSV0Finder()
{
if (fDebugStreamer) {
delete fDebugStreamer;
}
}