/* $Id$ */
-////////////////////////////////////////////////////
-// Stand alone tracker class //
-// Origin: Elisabetta Crescio //
-// e-mail: crescio@to.infn.it //
-// tracks are saved as AliITStrackV2 objects //
-////////////////////////////////////////////////////
+///////////////////////////////////////////////////////////
+// Stand alone ITS tracker class //
+// Origin: Elisabetta Crescio - crescio@to.infn.it //
+// Updated: Francesco Prino - prino@to.infn.it //
+///////////////////////////////////////////////////////////
#include <stdlib.h>
fVertexer = vertexer;
}
-
+/*
//____________________________________________________________________________
AliITStrackerSA::AliITStrackerSA(const AliITStrackerSA& tracker):AliITStrackerMI(),
fPhiEstimate(tracker.fPhiEstimate),
return *this;
}
-
+*/
//____________________________________________________________________________
AliITStrackerSA::~AliITStrackerSA(){
// destructor
// is done in the ITS only. In the standard reconstruction chain this option
// can be set via AliReconstruction::SetOption("ITS","onlyITS")
Int_t rc=0;
+
if(!fITSStandAlone){
rc=AliITStrackerMI::Clusters2Tracks(event);
}
}
if(!rc){
rc=FindTracks(event,kFALSE);
- if(AliITSReconstructor::GetRecoParam()->GetSAUseAllClusters()==kTRUE) {
+ Int_t nSPDcontr=0;
+ const AliESDVertex *spdv = event->GetPrimaryVertexSPD();
+ if(spdv) nSPDcontr = spdv->GetNContributors();
+ if(AliITSReconstructor::GetRecoParam()->GetSAUseAllClusters()==kTRUE &&
+ nSPDcontr<=AliITSReconstructor::GetRecoParam()->GetMaxSPDcontrForSAToUseAllClusters()) {
rc=FindTracks(event,kTRUE);
}
}
-
return rc;
}
&trs,primaryVertex[2],pflag);
Int_t nextLay=theLay+seedStep;
Bool_t goon=kTRUE;
+ if(nextLay<0 || nextLay == 6) goon = kFALSE;
while(goon){
kk++;
nClusLay[kk] = SearchClusters(nextLay,fPhiWin[nloop],fLambdaWin[nloop],
const Int_t kMaxClu=AliITStrackSA::kMaxNumberOfClusters;
- static Int_t firstmod[AliITSgeomTGeo::kNLayers];
-
+ static Int_t firstmod[AliITSgeomTGeo::kNLayers];
static Int_t clind[AliITSgeomTGeo::kNLayers][kMaxClu];
static Int_t clmark[AliITSgeomTGeo::kNLayers][kMaxClu];
static Int_t end[AliITSgeomTGeo::kNLayers];
+ static Int_t indices[AliITSgeomTGeo::kNLayers];
+
static AliITSRecPoint *listlayer[AliITSgeomTGeo::kNLayers][kMaxClu];
for(Int_t i=0;i<AliITSgeomTGeo::GetNLayers();i++) {
for(Int_t ncl=0;ncl<nclusters;ncl++){
Int_t index = tr->GetClusterIndexSA(ncl);
AliITSRecPoint* cl = (AliITSRecPoint*)GetCluster(index);
- if(cl->TestBit(kSAflag)==kTRUE) cl->ResetBit(kSAflag);
Int_t lay = (index & 0xf0000000) >> 28;
Int_t nInLay=end[lay];
listlayer[lay][nInLay]=cl;
TClonesArray &arrSA= *fListOfSATracks;
Int_t nFoundTracks=0;
+
for(Int_t l0=0;l0<end[0];l0++){ //loop on layer 1
- AliITSRecPoint* cl0 = (AliITSRecPoint*)listlayer[0][l0];
+ indices[0]=l0;
for(Int_t l1=0;l1<end[1];l1++){ //loop on layer 2
- AliITSRecPoint* cl1 = (AliITSRecPoint*)listlayer[1][l1];
+ indices[1]=l1;
for(Int_t l2=0;l2<end[2];l2++){ //loop on layer 3
- AliITSRecPoint* cl2 = (AliITSRecPoint*)listlayer[2][l2];
+ indices[2]=l2;
for(Int_t l3=0;l3<end[3];l3++){ //loop on layer 4
- AliITSRecPoint* cl3 = (AliITSRecPoint*)listlayer[3][l3];
+ indices[3]=l3;
for(Int_t l4=0;l4<end[4];l4++){ //loop on layer 5
- AliITSRecPoint* cl4 = (AliITSRecPoint*)listlayer[4][l4];
+ indices[4]=l4;
for(Int_t l5=0;l5<end[5];l5++){ //loop on layer 6
- AliITSRecPoint* cl5 = (AliITSRecPoint*)listlayer[5][l5];
-
-
- Double_t x1,y1,z1,sx1,sy1,sz1;
- Double_t x2,y2,z2,sx2,sy2,sz2;
- AliITSRecPoint* p1=0;
- AliITSRecPoint* p2=0;
- Int_t index1=0,index2=0;
- Int_t mrk1=0,mrk2=0;
-
- switch(firstLay) {
- case 0:
- p1=cl0;
- index1=clind[0][l0];mrk1=clmark[0][l0];
- break;
- case 1:
- p1=cl1;
- index1=clind[1][l1];mrk1=clmark[1][l1];
- break;
- case 2:
- p1=cl2;
- index1=clind[2][l2];mrk1=clmark[2][l2];
- break;
- case 3:
- p1=cl3;
- index1=clind[3][l3];mrk1=clmark[3][l3];
- break;
- case 4:
- p1=cl4;
- index1=clind[4][l4];mrk1=clmark[4][l4];
- break;
- }
+ indices[5]=l5;
- switch(secondLay) {
- case 1:
- p2=cl1;
- index2=clind[1][l1];mrk2=clmark[1][l1];
- break;
- case 2:
- p2=cl2;
- index2=clind[2][l2];mrk2=clmark[2][l2];
- break;
- case 3:
- p2=cl3;
- index2=clind[3][l3];mrk2=clmark[3][l3];
- break;
- case 4:
- p2=cl4;
- index2=clind[4][l4];mrk2=clmark[4][l4];
- break;
- case 5:
- p2=cl5;
- index2=clind[5][l5];mrk2=clmark[5][l5];
- break;
- default:
- p2=0;
- index2=-1;mrk2=-1;
- break;
- }
+ // estimate curvature from 2 innermost points (or innermost point + vertex)
+ Int_t iFirstLay=indices[firstLay];
+ Int_t mrk1=clmark[firstLay][iFirstLay];
+
+ AliITSRecPoint* p1=(AliITSRecPoint*)listlayer[firstLay][iFirstLay];
Int_t module1 = p1->GetDetectorIndex()+firstmod[firstLay];
Int_t layer,ladder,detector;
AliITSgeomTGeo::GetModuleId(module1,layer,ladder,detector);
Float_t yclu1 = p1->GetY();
Float_t zclu1 = p1->GetZ();
+
+ Double_t x1,y1,z1;
+ Double_t x2,y2,z2;
Double_t cv=0,tgl2=0,phi2=0;
-
- Int_t cln1=mrk1;
- AliITSclusterTable* arr1 = (AliITSclusterTable*)GetClusterCoord(firstLay,cln1);
+ AliITSclusterTable* arr1 = (AliITSclusterTable*)GetClusterCoord(firstLay,mrk1);
x1 = arr1->GetX();
y1 = arr1->GetY();
z1 = arr1->GetZ();
- sx1 = arr1->GetSx();
- sy1 = arr1->GetSy();
- sz1 = arr1->GetSz();
if(secondLay>0) {
- Int_t cln2=mrk2;
- AliITSclusterTable* arr2 = (AliITSclusterTable*)GetClusterCoord(secondLay,cln2);
+ Int_t iSecondLay=indices[secondLay];
+ Int_t mrk2=clmark[secondLay][iSecondLay];
+ AliITSclusterTable* arr2 = (AliITSclusterTable*)GetClusterCoord(secondLay,mrk2);
x2 = arr2->GetX();
y2 = arr2->GetY();
z2 = arr2->GetZ();
- sx2 = arr2->GetSx();
- sy2 = arr2->GetSy();
- sz2 = arr2->GetSz();
cv = Curvature(primaryVertex[0],primaryVertex[1],x1,y1,x2,y2);
tgl2 = (z2-z1)/TMath::Sqrt((x2-x1)*(x2-x1)+(y2-y1)*(y2-y1));
phi2 = TMath::ATan2((y2-y1),(x2-x1));
phi2 = TMath::ATan2((y1-y2),(x1-x2));
}
-
+ // create track and attach it the RecPoints
AliITStrackSA trac(layer,ladder,detector,yclu1,zclu1,phi2,tgl2,cv,1);
-
-
- if(cl5!=0) {
- trac.AddClusterV2(5,(clind[5][l5] & 0x0fffffff)>>0);
- trac.AddClusterMark(5,clmark[5][l5]);
- }
- if(cl4!=0){
- trac.AddClusterV2(4,(clind[4][l4] & 0x0fffffff)>>0);
- trac.AddClusterMark(4,clmark[4][l4]);
- }
- if(cl3!=0){
- trac.AddClusterV2(3,(clind[3][l3] & 0x0fffffff)>>0);
- trac.AddClusterMark(3,clmark[3][l3]);
- }
- if(cl2!=0){
- trac.AddClusterV2(2,(clind[2][l2] & 0x0fffffff)>>0);
- trac.AddClusterMark(2,clmark[2][l2]);
- }
- if(cl1!=0){
- trac.AddClusterV2(1,(clind[1][l1] & 0x0fffffff)>>0);
- trac.AddClusterMark(1,clmark[1][l1]);
- }
- if(cl0!=0){
- trac.AddClusterV2(0,(clind[0][l0] & 0x0fffffff)>>0);
- trac.AddClusterMark(0,clmark[0][l0]);
+ for(Int_t iLay=5; iLay>=0; iLay--){
+ Int_t iInLay=indices[iLay];
+ AliITSRecPoint* cl=(AliITSRecPoint*)listlayer[iLay][iInLay];
+ if(cl!=0){
+ trac.AddClusterV2(iLay,(clind[iLay][iInLay] & 0x0fffffff)>>0);
+ trac.AddClusterMark(iLay,clmark[iLay][iInLay]);
+ }
}
//fit with Kalman filter using AliITStrackerMI::RefitAt()
if(otrack==0) return 0;
- Int_t indexc[AliITSgeomTGeo::kNLayers];
- for(Int_t i=0;i<AliITSgeomTGeo::GetNLayers();i++) indexc[i]=0;
- for(Int_t nind=0;nind<otrack->GetNumberOfClusters();nind++){
- indexc[nind] = otrack->GetClusterIndex(nind);
- }
- Int_t labl[6][3];
- for(Int_t i=0;i<AliITSgeomTGeo::GetNLayers();i++) {
- if(i<otrack->GetNumberOfClusters()) {
- AliITSRecPoint* cl = (AliITSRecPoint*)GetCluster(indexc[i]);
- labl[i][0]=cl->GetLabel(0);
- labl[i][1]=cl->GetLabel(1);
- labl[i][2]=cl->GetLabel(2);
- } else {
- labl[i][0]=-1;
- labl[i][1]=-1;
- labl[i][2]=-1;
- }
- }
-
CookLabel(otrack,0.); //MI change - to see fake ratio
-
- Int_t label=FindLabel(labl[0][0],labl[1][0],labl[2][0],labl[3][0],labl[4][0],labl[5][0]);
- Int_t lflag=0;
- for(Int_t i=0;i<otrack->GetNumberOfClusters();i++)
- if(labl[i][0]==label || labl[i][1]==label || labl[i][2]==label) lflag++;
-
- if(lflag<otrack->GetNumberOfClusters()) label = -label;
+ Int_t label=FindLabel(otrack);
otrack->SetLabel(label);
+ Double_t low=0.;
+ Double_t up=0.51;
+ otrack->CookdEdx(low,up);
//remove clusters of found track
for(Int_t nlay=0;nlay<AliITSgeomTGeo::GetNLayers();nlay++){
Double_t sdedx[4]={0.,0.,0.,0.};
for(Int_t i=0; i<4; i++) sdedx[i]=t->GetSampledEdx(i);
outtrack.SetITSdEdxSamples(sdedx);
+
+
+ if(AliITSReconstructor::GetRecoParam()->GetSAUsedEdxInfo()){
+ Double_t mom=t->P();
+ Double_t ppid[AliPID::kSPECIES];
+ for(Int_t isp=0;isp<AliPID::kSPECIES;isp++) ppid[isp]=0.;
+ ppid[AliPID::kPion]=1.;
+ if(mom<0.7){
+ Double_t truncmean=t->GetdEdx();
+ Int_t ide=fITSPid->GetParticleIdFromdEdxVsP(mom,truncmean,kTRUE);
+ if(ide==AliPID::kProton){
+ ppid[AliPID::kProton]=1.;
+ ppid[AliPID::kPion]=0.;
+ }
+ else if(ide==AliPID::kKaon){
+ ppid[AliPID::kKaon]=1.;
+ ppid[AliPID::kPion]=0.;
+ }
+ }
+ outtrack.SetITSpid(ppid);
+ outtrack.SetESDpid(ppid);
+ }
event->AddTrack(&outtrack);
return;
}
- Int_t ncl = fCluLayer[layer]->GetEntries();
+ Double_t phiExpect=fPhiEstimate;
+ Double_t lamExpect=fLambdac;
+
+ Int_t ncl = fCluLayer[layer]->GetEntriesFast();
for (Int_t index=0; index<ncl; index++) {
- AliITSRecPoint *c = (AliITSRecPoint*)fCluLayer[layer]->At(index);
+ AliITSRecPoint *c = (AliITSRecPoint*)fCluLayer[layer]->UncheckedAt(index);
if (!c) continue;
- if (c->GetQ()<=0) continue;
- if(layer>1 && c->GetQ()<=fMinQ) continue;
- AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(layer,index);
- Double_t phi = arr->GetPhi();
- if (TMath::Abs(phi-fPhiEstimate)>phiwindow) continue;
-
- Double_t lambda = arr->GetLambda();
- if (TMath::Abs(lambda-fLambdac)>lambdawindow) continue;
-
- if(trs->GetNumberOfClustersSA()==trs->GetMaxNumberOfClusters()) return 0;
- if(trs->GetNumberOfMarked(layer)==trs->GetMaxNMarkedPerLayer()) return 0;
- Int_t orind = arr->GetOrInd();
- trs->AddClusterSA(layer,orind);
- trs->AddClusterMark(layer,index);
-
- nc++;
- fLambdac=lambda;
- fPhiEstimate=phi;
-
- fPointc[0]=arr->GetX();
- fPointc[1]=arr->GetY();
+ AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(layer,index);
+
+ Double_t lambda = arr->GetLambda();
+ if (TMath::Abs(lambda-lamExpect)>lambdawindow) continue;
+ Double_t phi = arr->GetPhi();
+ Double_t deltaPhi = phi-phiExpect;
+ if(deltaPhi>TMath::Pi()) deltaPhi-=2*TMath::Pi();
+ else if(deltaPhi<-TMath::Pi()) deltaPhi+=2*TMath::Pi();
+ if (TMath::Abs(deltaPhi)>phiwindow) continue;
+
+ if(trs->GetNumberOfClustersSA()==trs->GetMaxNumberOfClusters()) return 0;
+ if(trs->GetNumberOfMarked(layer)==trs->GetMaxNMarkedPerLayer()) return 0;
+ Int_t orind = arr->GetOrInd();
+ trs->AddClusterSA(layer,orind);
+ trs->AddClusterMark(layer,index);
+ nc++;
+ fLambdac=lambda;
+ fPhiEstimate=phi;
+
+ fPointc[0]=arr->GetX();
+ fPointc[1]=arr->GetY();
+
}
return nc;
}
Bool_t AliITStrackerSA::SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex){
// Sets the first point (seed) for tracking
- AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[lay]->At(clu);
+ AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[lay]->UncheckedAt(clu);
if(!cl) return kFALSE;
if (cl->GetQ()<=0) return kFALSE;
if(lay>1 && cl->GetQ()<=fMinQ) return kFALSE;
}
//__________________________________________________________
-Int_t AliITStrackerSA::FindLabel(Int_t l0, Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5){
-
- //function used to determine the track label
-
- Int_t lb[6] = {l0,l1,l2,l3,l4,l5};
- Int_t aa[6]={1,1,1,1,1,1};
- Int_t ff=0;
- Int_t ll=0;
- Int_t k=0;Int_t w=0;Int_t num=6;
- for(Int_t i=5;i>=0;i--) if(lb[i]==-1) num=i;
-
- while(k<num){
+Int_t AliITStrackerSA::FindLabel(AliITStrackV2* track){
+ //
- for(Int_t i=k+1;i<num;i++){
-
- if(lb[k]==lb[i] && aa[k]!=0){
-
- aa[k]+=1;
- aa[i]=0;
- }
+ Int_t labl[AliITSgeomTGeo::kNLayers][3];
+ Int_t cnts[AliITSgeomTGeo::kNLayers][3];
+ for(Int_t j=0;j<AliITSgeomTGeo::GetNLayers();j++){
+ for(Int_t k=0;k<3;k++){
+ labl[j][k]=-2;
+ cnts[j][k]=1;
}
- k++;
}
-
- while(w<num){
-
- for(Int_t j=0;j<6;j++){
- if(aa[w]<aa[j]){
- ff=aa[w];
- aa[w]=aa[j];
- aa[j]=ff;
- ll=lb[w];
- lb[w]=lb[j];
- lb[j]=ll;
- }
+ Int_t iNotLabel=0;
+ for(Int_t i=0;i<track->GetNumberOfClusters(); i++) {
+ Int_t indexc = track->GetClusterIndex(i);
+ AliITSRecPoint* cl = (AliITSRecPoint*)GetCluster(indexc);
+ Int_t iLayer=cl->GetLayer();
+ for(Int_t k=0;k<3;k++){
+ labl[iLayer][k]=cl->GetLabel(k);
+ if(labl[iLayer][k]<0) iNotLabel++;
+ }
+ }
+ if(iNotLabel==3*track->GetNumberOfClusters()) return -2;
+
+ for(Int_t j1=0;j1<AliITSgeomTGeo::kNLayers; j1++) {
+ for(Int_t j2=0; j2<j1; j2++){
+ for(Int_t k1=0; k1<3; k1++){
+ for(Int_t k2=0; k2<3; k2++){
+ if(labl[j1][k1]>=0 && labl[j1][k1]==labl[j2][k2] && cnts[j2][k2]>0){
+ cnts[j2][k2]++;
+ cnts[j1][k1]=0;
+ }
+ }
+ }
}
- w++;
}
-
- if(num<1) return -1;
- return lb[num-1];
-}
-
-//_____________________________________________________________________________
-Int_t AliITStrackerSA::Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5, Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
-Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16, Int_t gl17, Int_t gl18, Int_t minNPoints){
-
-
- //function used to assign label to the found track. If track is fake, the label is negative
- Int_t lb0[6] = {gl1,gl2,gl3,gl4,gl5,gl6};
- Int_t lb1[6] = {gl7,gl8,gl9,gl10,gl11,gl12};
- Int_t lb2[6] = {gl13,gl14,gl15,gl16,gl17,gl18};
- Int_t ll=FindLabel(lb0[0],lb0[1],lb0[2],lb0[3],lb0[4],lb0[5]);
- Int_t lflag=0;Int_t num=6;
- if(lb0[5]==-1 && lb1[5]==-1 && lb2[5]==-1) num=5;
- for(Int_t i=0;i<num;i++){
- if(lb0[i]==ll || lb1[i]==ll || lb2[i]==ll) lflag+=1;
+ Int_t cntMax=0;
+ Int_t label=-1;
+ for(Int_t j=0;j<AliITSgeomTGeo::kNLayers;j++){
+ for(Int_t k=0;k<3;k++){
+ if(cnts[j][k]>cntMax && labl[j][k]>=0){
+ cntMax=cnts[j][k];
+ label=labl[j][k];
+ }
+ }
}
- if(lflag>=minNPoints) return ll;
- else return -ll;
-
+ Int_t lflag=0;
+ for(Int_t i=0;i<AliITSgeomTGeo::kNLayers;i++)
+ if(labl[i][0]==label || labl[i][1]==label || labl[i][2]==label) lflag++;
+ if(lflag<track->GetNumberOfClusters()) label = -label;
+ return label;
}
//_____________________________________________________________________________
void AliITStrackerSA::SetCalculatedWindowSizes(Int_t n, Float_t phimin, Float_t phimax, Float_t lambdamin, Float_t lambdamax){
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff