X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITStrackerSA.cxx;h=cc1db2833924e9642a8c72809202b71505c85444;hb=405be8ef164a6a96b98b5b305db3820ece4ccfcb;hp=6d061c78966580ce574165f186b892dc2ff03e1b;hpb=c930a87490e38ec9dd47b2cc4307f97eb759fb7f;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITStrackerSA.cxx b/ITS/AliITStrackerSA.cxx index 6d061c78966..cc1db283392 100644 --- a/ITS/AliITStrackerSA.cxx +++ b/ITS/AliITStrackerSA.cxx @@ -26,88 +26,187 @@ #include #include -#include #include #include -#include "AliESD.h" +#include "AliESDEvent.h" #include "AliESDVertex.h" #include "AliESDtrack.h" -#include "AliITSRiemannFit.h" #include "AliITSVertexer.h" #include "AliITSclusterTable.h" -#include "AliITSclusterV2.h" -#include "AliITSgeom.h" +#include "AliITSRecPoint.h" +#include "AliITSgeomTGeo.h" #include "AliITStrackSA.h" #include "AliITStrackerSA.h" +#include "AliITSReconstructor.h" #include "AliRun.h" ClassImp(AliITStrackerSA) //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA():AliITStrackerMI(){ +AliITStrackerSA::AliITStrackerSA():AliITStrackerMI(), +fPhiEstimate(0), +fITSStandAlone(0), +fLambdac(0), +fPhic(0), +fCoef1(0), +fCoef2(0), +fCoef3(0), +fNloop(0), +fPhiWin(0), +fLambdaWin(0), +fVert(0), +fVertexer(0), +fListOfTracks(0), +fListOfSATracks(0), +fITSclusters(0), +fSixPoints(0), +fOuterStartLayer(0), +fCluLayer(0), +fCluCoord(0){ // Default constructor Init(); } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITSgeom *geom):AliITStrackerMI(geom) +AliITStrackerSA::AliITStrackerSA(const Char_t *geom):AliITStrackerMI(0), +fPhiEstimate(0), +fITSStandAlone(0), +fLambdac(0), +fPhic(0), +fCoef1(0), +fCoef2(0), +fCoef3(0), +fNloop(0), +fPhiWin(0), +fLambdaWin(0), +fVert(0), +fVertexer(0), +fListOfTracks(0), +fListOfSATracks(0), +fITSclusters(0), +fSixPoints(0), +fOuterStartLayer(0), +fCluLayer(0), +fCluCoord(0) { // Standard constructor (Vertex is known and passed to this obj.) + if (geom) { + AliWarning("\"geom\" is actually a dummy argument !"); + } + Init(); fVert = 0; - fGeom = geom; } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITSgeom *geom, AliESDVertex *vert):AliITStrackerMI(geom) +AliITStrackerSA::AliITStrackerSA(const Char_t *geom, AliESDVertex *vert):AliITStrackerMI(0), +fPhiEstimate(0), +fITSStandAlone(0), +fLambdac(0), +fPhic(0), +fCoef1(0), +fCoef2(0), +fCoef3(0), +fNloop(0), +fPhiWin(0), +fLambdaWin(0), +fVert(vert), +fVertexer(0), +fListOfTracks(0), +fListOfSATracks(0), +fITSclusters(0), +fSixPoints(0), +fOuterStartLayer(0), +fCluLayer(0), +fCluCoord(0) { // Standard constructor (Vertex is known and passed to this obj.) + if (geom) { + AliWarning("\"geom\" is actually a dummy argument !"); + } Init(); - fVert = vert; - fGeom = geom; } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITSgeom *geom, AliITSVertexer *vertexer):AliITStrackerMI(geom) +AliITStrackerSA::AliITStrackerSA(const Char_t *geom, AliITSVertexer *vertexer):AliITStrackerMI(0), +fPhiEstimate(0), +fITSStandAlone(0), +fLambdac(0), +fPhic(0), +fCoef1(0), +fCoef2(0), +fCoef3(0), +fNloop(0), +fPhiWin(0), +fLambdaWin(0), +fVert(), +fVertexer(vertexer), +fListOfTracks(0), +fListOfSATracks(0), +fITSclusters(0), +fSixPoints(0), +fOuterStartLayer(0), +fCluLayer(0), +fCluCoord(0) { // Standard constructor (Vertex is unknown - vertexer is passed to this obj) + if (geom) { + AliWarning("\"geom\" is actually a dummy argument !"); + } Init(); fVertexer = vertexer; - fGeom = geom; } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITStrackerSA& tracker):AliITStrackerMI(){ +AliITStrackerSA::AliITStrackerSA(const AliITStrackerSA& tracker):AliITStrackerMI(), +fPhiEstimate(tracker.fPhiEstimate), +fITSStandAlone(tracker.fITSStandAlone), +fLambdac(tracker.fLambdac), +fPhic(tracker.fPhic), +fCoef1(tracker.fCoef1), +fCoef2(tracker.fCoef2), +fCoef3(tracker.fCoef3), +fNloop(tracker.fNloop), +fPhiWin(tracker.fPhiWin), +fLambdaWin(tracker.fLambdaWin), +fVert(tracker.fVert), +fVertexer(tracker.fVertexer), +fListOfTracks(tracker.fListOfTracks), +fListOfSATracks(tracker.fListOfSATracks), +fITSclusters(tracker.fITSclusters), +fSixPoints(tracker.fSixPoints), +fOuterStartLayer(tracker.fOuterStartLayer), +fCluLayer(tracker.fCluLayer), +fCluCoord(tracker.fCluCoord) { // Copy constructor - fPhiEstimate = tracker.fPhiEstimate; for(Int_t i=0;i<2;i++){ fPoint1[i]=tracker.fPoint1[i]; fPoint2[i]=tracker.fPoint2[i]; fPoint3[i]=tracker.fPoint3[i]; fPointc[i]=tracker.fPointc[i]; } - fLambdac = tracker.fLambdac; - fPhic = tracker.fPhic; - fCoef1 = tracker.fCoef1; - fCoef2 = tracker.fCoef2; - fCoef3 = tracker.fCoef3; - fNloop = tracker.fNloop; - fPhiWin = tracker.fPhiWin; - fLambdaWin = tracker.fLambdaWin; if(tracker.fVertexer && tracker.fVert){ fVert = new AliESDVertex(*tracker.fVert); } else { fVert = tracker.fVert; } - fVertexer = tracker.fVertexer; - fGeom = tracker.fGeom; - fTable = tracker.fTable; - fListOfTracks = tracker.fListOfTracks; + for(Int_t i=0;i~AliITStrackerSA(); + new(this) AliITStrackerSA(source); + return *this; + } //____________________________________________________________________________ @@ -120,12 +219,51 @@ AliITStrackerSA::~AliITStrackerSA(){ } fVert = 0; fVertexer = 0; - fGeom = 0; + if(fPhiWin)delete []fPhiWin; if(fLambdaWin)delete []fLambdaWin; - fTable =0; fListOfTracks->Delete(); + delete fListOfTracks; + fListOfSATracks->Delete(); + delete fListOfSATracks; + if(fCluLayer){ + for(Int_t i=0;iDelete(); + delete fCluLayer[i]; + } + } + delete [] fCluLayer; } + if(fCluCoord){ + for(Int_t i=0;iDelete(); + delete fCluCoord[i]; + } + } + delete [] fCluCoord; + } + +} + +//____________________________________________________________________________ +Int_t AliITStrackerSA::Clusters2Tracks(AliESDEvent *event){ +// This method is used to find and fit the tracks. By default the corresponding +// method in the parent class is invoked. In this way a combined tracking +// TPC+ITS is performed. If the flag fITSStandAlone is true, the tracking +// 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); + } + else { + AliDebug(1,"Stand Alone flag set: doing tracking in ITS alone\n"); + } + if(!rc) rc=FindTracks(event); + return rc; +} //____________________________________________________________________________ void AliITStrackerSA::Init(){ @@ -141,12 +279,24 @@ void AliITStrackerSA::Init(){ fPointc[1]=0; fVert = 0; fVertexer = 0; - fGeom = 0; - SetWindowSizes(); - fTable = 0; + Int_t nLoops=AliITSReconstructor::GetRecoParam()->GetNLoopsSA(); + if(nLoops==33){ + SetFixedWindowSizes(); + }else{ + Double_t phimin=AliITSReconstructor::GetRecoParam()->GetMinPhiSA(); + Double_t phimax=AliITSReconstructor::GetRecoParam()->GetMaxPhiSA(); + Double_t lambmin=AliITSReconstructor::GetRecoParam()->GetMinLambdaSA(); + Double_t lambmax=AliITSReconstructor::GetRecoParam()->GetMaxLambdaSA(); + SetCalculatedWindowSizes(nLoops,phimin,phimax,lambmin,lambmax); + } fITSclusters = 0; SetSixPoints(); + SetOuterStartLayer(0); + SetSAFlag(kFALSE); fListOfTracks=new TObjArray(0,0); + fListOfSATracks=new TObjArray(0,0); + fCluLayer = 0; + fCluCoord = 0; } //_______________________________________________________________________ void AliITStrackerSA::ResetForFinding(){ @@ -161,1012 +311,758 @@ void AliITStrackerSA::ResetForFinding(){ fPointc[0]=0; fPointc[1]=0; fListOfTracks->Delete(); -} -//____________________________________________________________________________ -void AliITStrackerSA::FindTracks(TTree *out,Int_t evnumber){ - - /************************************************************************** - * This function finds primary tracks. - * * - * * - * Example: to execute function with only the ITS (no combined tracking * - * with TPC+ITS) and requiring 5/6 points to define a good track * - * call SetSixPoinbts(kFALSE) in advance and then * - * use: FindTracks(treein,treeout,evnumber) * - * to execute combined tracking, before using FindTracks, use * - * UseFoundTracksV2 * - *************************************************************************/ - - if(!fITSclusters){ - Fatal("FindTracks","ITS cluster tree is not accessed - Abort!!!\n Please use method SetClusterTree to pass the pointer to the tree\n"); - exit(1); - } - //Get primary vertex - if(fVertexer){ - if(fVert)delete fVert; - fVert = fVertexer->FindVertexForCurrentEvent(evnumber); - } - else { - gAlice->GetEvent(evnumber); - if(!fVert){ - Fatal("FindTracks","Vertex is missing\n"); - return; - } - } - Double_t primaryVertex[3]; - Double_t errorsprimvert[3]; - fVert->GetXYZ(primaryVertex); - fVert->GetSigmaXYZ(errorsprimvert); - if(errorsprimvert[0]==0 || errorsprimvert[1]==0){ - Warning("FindTracks","Set errors on vertex positions x and y at 0.0001"); - errorsprimvert[0]=0.0001; - errorsprimvert[1]=0.0001; - } - fVert->PrintStatus(); - - - //Fill array with cluster indices for each module - if(!fTable){ - fTable = new AliITSclusterTable(fGeom,this,primaryVertex); - fTable->FillArray(fITSclusters); - fTable->FillArrayCoorAngles(); - } - - -//Fill tree for found tracks - AliITStrackV2* outrack=0; - TBranch* branch=out->Branch("tracks","AliITStrackV2",&outrack,32000,0); - if (!branch) out->Branch("tracks","AliITStrackV2",&outrack,32000,3); - else branch->SetAddress(&outrack); - - - Int_t * firstmod = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - firstmod[i]=fGeom->GetModuleIndex(i+1,1,1); - } - // firstmod [i] number of the first module in the ITS layer i. - - AliITSlayer &layer=fgLayers[0]; // first layer - Int_t ntrack=0; - Int_t dim=layer.GetNumberOfClusters(); - //loop on the different windows - for(Int_t nloop=0;nloopIsUsed()==1) continue; - if(cl->TestBit(kSAflag)==kTRUE) continue; - - fPhic = fTable->GetPhiCluster(0,ncl); - fLambdac = fTable->GetLambdaCluster(0,ncl); - fPhiEstimate = fPhic; - AliITStrackSA* trs = new AliITStrackSA(); - fPoint1[0]=primaryVertex[0]; - fPoint1[1]=primaryVertex[1]; - fPoint2[0]=fTable->GetXCluster(0,ncl); - fPoint2[1]=fTable->GetYCluster(0,ncl); - - Int_t * nn = new Int_t[fGeom->GetNlayers()];//counter for clusters on each layer - for(Int_t i=0;iGetNlayers();i++){ nn[i]=0;} - nn[0] = SearchClusters(0,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - nn[1] = SearchClusters(1,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - - if(nn[1]>0){ - pflag=1; - fPoint3[0] = fPointc[0]; - fPoint3[1] = fPointc[1]; - } - nn[2] = SearchClusters(2,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - if(nn[1]==0 && nn[2]==0) pflag=0; - if(nn[2]!=0 && nn[1]!=0){ pflag=1; UpdatePoints();} - if(nn[2]!=0 && nn[1]==0){ - pflag=1; - fPoint3[0]=fPointc[0]; - fPoint3[1]=fPointc[1]; - } - - nn[3] = SearchClusters(3,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - pflag=1; - if(nn[3]!=0) UpdatePoints(); - nn[4] = SearchClusters(4,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - pflag=1; - if(nn[4]!=0) UpdatePoints(); - nn[5] = SearchClusters(5,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - - - Int_t layOK=0; - Int_t numberofpoints; - if(fSixPoints) numberofpoints=6; //check of the candidate track - else numberofpoints=5; //if track is good (with the required number - for(Int_t nnp=0;nnpGetNlayers();nnp++){ //of points) it is written on file - if(nn[nnp]!=0) layOK+=1; - } - if(layOK>=numberofpoints){ - AliITStrackV2* tr2 = FitTrack(trs,primaryVertex,errorsprimvert); - if(tr2==0){ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - } - delete [] nn; - delete trs; - continue; - } - outrack=tr2; - out->Fill(); - ntrack++; - Int_t nct = tr2->GetNumberOfClusters(); - - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - kl->SetBit(kSAflag); - - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - } - } - delete [] nn; - delete trs; - - }//end loop on clusters of layer1 - - }//end loop2 - - //if 5/6 points are required, second loop starting - //from second layer, to find tracks with point of - //layer 1 missing - - if(!fSixPoints){ - // counter for clusters on each layer - Int_t * nn = new Int_t[fGeom->GetNlayers()-1]; - for(Int_t nloop=0;nloopIsUsed()==1) continue; - if(cl->TestBit(kSAflag)==kTRUE) continue; - fPhic = fTable->GetPhiCluster(1,ncl2); - fLambdac = fTable->GetLambdaCluster(1,ncl2); - fPhiEstimate = fPhic; - AliITStrackSA* trs = new AliITStrackSA(); - - fPoint1[0]=primaryVertex[0]; - fPoint1[1]=primaryVertex[1]; - fPoint2[0]=fTable->GetXCluster(1,ncl2);; - fPoint2[1]=fTable->GetYCluster(1,ncl2);; - - for(Int_t kk=0;kkGetNlayers()-1;kk++)nn[kk] = 0; - for(Int_t kk=0;kkGetNlayers()-1;kk++){ - nn[kk] = SearchClusters(kk+1,fPhiWin[nloop],fLambdaWin[nloop], - trs,primaryVertex[2],pflag); - if(nn[kk]==0)break; - if(kk>0){ - UpdatePoints(); - pflag = 1; - } - } - Int_t fl=0; - for(Int_t nnp=0;nnpGetNlayers()-1;nnp++){ - if(nn[nnp]!=0) fl+=1; - } - if(fl>=5){ // 5/6 - AliITStrackV2* tr2 = FitTrack(trs,primaryVertex,errorsprimvert); - if(tr2==0){ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - } - delete trs; - continue; - } - outrack=tr2; - out->Fill(); - Int_t nct = tr2->GetNumberOfClusters(); - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl==0) continue; - kl->SetBit(kSAflag); - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl==0) continue; - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - } - } - delete trs; - }//end loop on clusters of layer2 - } - delete [] nn; - } // if(!fSixPoints.... - - delete [] firstmod; - delete fTable; fTable=0; + fListOfSATracks->Delete(); } + //______________________________________________________________________ -Int_t AliITStrackerSA::FindTracks(AliESD* event){ +Int_t AliITStrackerSA::FindTracks(AliESDEvent* event){ - // Track finder using the ESD object +// Track finder using the ESD object + if(!fITSclusters){ Fatal("FindTracks","ITS cluster tree is not accessed - Abort!!!\n Please use method SetClusterTree to pass the pointer to the tree\n"); return -1; } - + //Reads event and mark clusters of traks already found, with flag kITSin + Int_t nentr=event->GetNumberOfTracks(); + if(AliITSReconstructor::GetRecoParam()->GetSAUseAllClusters()==kFALSE) { + while (nentr--) { + AliESDtrack *track=event->GetTrack(nentr); + if (track->GetStatus()&AliESDtrack::kITSin==AliESDtrack::kITSin){ + Int_t idx[12]; + Int_t ncl = track->GetITSclusters(idx); + for(Int_t k=0;kSetBit(kSAflag); + } + } + } + } //Get primary vertex - Double_t errorsprimvert[3]; Double_t primaryVertex[3]; event->GetVertex()->GetXYZ(primaryVertex); - event->GetVertex()->GetSigmaXYZ(errorsprimvert); - - if(errorsprimvert[0]==0 || errorsprimvert[1]==0){ - // Warning("FindTracks","Set errors on vertex positions x and y at 0.005"); - errorsprimvert[0]=0.005; - errorsprimvert[1]=0.005; + //Creates TClonesArray with clusters for each layer. The clusters already used + //by AliITStrackerMI are not considered + Int_t nclusters[AliITSgeomTGeo::kNLayers]={0,0,0,0,0,0}; + Int_t dmar[AliITSgeomTGeo::kNLayers]={0,0,0,0,0,0}; + if (fCluLayer == 0) { + fCluLayer = new TClonesArray*[AliITSgeomTGeo::kNLayers]; + fCluCoord = new TClonesArray*[AliITSgeomTGeo::kNLayers]; + for(Int_t i=0;iFillArray(fITSclusters); - fTable->FillArrayCoorAngles(); + for(Int_t i=0;iGetLayersToSkip(i)) { + for(Int_t cli=0;cliTestBit(kSAflag)==kTRUE) continue; //clusters used by TPC prol. + if(cls->GetQ()==0) continue; //fake clusters dead zones + nclusters[i]++; + } + } + dmar[i]=0; + delete fCluLayer[i]; + fCluLayer[i] = new TClonesArray("AliITSRecPoint",nclusters[i]); + delete fCluCoord[i]; + fCluCoord[i] = new TClonesArray("AliITSclusterTable",nclusters[i]); } - Int_t * firstmod = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - firstmod[i]=fGeom->GetModuleIndex(i+1,1,1); + for(Int_t ilay=0;ilayGetLayersToSkip(ilay)) { + for(Int_t cli=0;cliTestBit(kSAflag)==kTRUE) continue; + if(cls->GetQ()==0) continue; + Double_t phi=0;Double_t lambda=0; + Float_t x=0;Float_t y=0;Float_t z=0; + Float_t sx=0;Float_t sy=0;Float_t sz=0; + GetCoorAngles(cls,phi,lambda,x,y,z,primaryVertex); + GetCoorErrors(cls,sx,sy,sz); + new (clulay[dmar[ilay]]) AliITSRecPoint(*cls); + new (clucoo[dmar[ilay]]) AliITSclusterTable(x,y,z,sx,sy,sz,phi,lambda,cli); + dmar[ilay]++; + } + } } - // firstmod [i] number of the first module in the ITS layer i. - - - AliITSlayer &layer=fgLayers[0]; + + // track counter Int_t ntrack=0; - Int_t dim=layer.GetNumberOfClusters(); - //loop on the different windows - for(Int_t nloop=0;nloopIsUsed()==1) continue; - if(cl->TestBit(kSAflag)==kTRUE) continue; - if (cl->GetQ()<=0) continue; - - fPhic = fTable->GetPhiCluster(0,ncl); - fLambdac = fTable->GetLambdaCluster(0,ncl); - - if (TMath::Abs(fLambdac)>0.26*TMath::Pi()) continue; - - fPhiEstimate = fPhic; - AliITStrackSA* trs = new AliITStrackSA(); - fPoint1[0]=primaryVertex[0]; - fPoint1[1]=primaryVertex[1]; - fPoint2[0]=fTable->GetXCluster(0,ncl); - fPoint2[1]=fTable->GetYCluster(0,ncl); - Int_t * nn = new Int_t[fGeom->GetNlayers()];//counter for clusters on each layer - for(Int_t i=0;iGetNlayers();i++){ nn[i]=0;} - nn[0] = SearchClusters(0,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - - nn[1] = SearchClusters(1,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - if(nn[1]>0){ - pflag=1; - fPoint3[0] = fPointc[0]; - fPoint3[1] = fPointc[1]; - } - nn[2] = SearchClusters(2,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - if(nn[1]==0 && nn[2]==0) pflag=0; - if(nn[2]!=0 && nn[1]!=0){ pflag=1; UpdatePoints();} - if(nn[2]!=0 && nn[1]==0){ - pflag=1; - fPoint3[0]=fPointc[0]; - fPoint3[1]=fPointc[1]; - } - nn[3] = SearchClusters(3,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - pflag=1; - if(nn[3]!=0) UpdatePoints(); - nn[4] = SearchClusters(4,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - pflag=1; - if(nn[4]!=0) UpdatePoints(); - nn[5] = SearchClusters(5,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); - + Int_t minNPoints = (fSixPoints ? AliITSgeomTGeo::GetNLayers() : AliITSgeomTGeo::GetNLayers()-1); + for(Int_t i=0;iGetLayersToSkip(i)) { + minNPoints--; + } + } - Int_t layOK=0; - Int_t numberofpoints; - if(fSixPoints) numberofpoints=6; //check of the candidate track - else numberofpoints=5; //if track is good (with the required number - for(Int_t nnp=0;nnpGetNlayers();nnp++){ //of points) it is written on file - if(nn[nnp]!=0) layOK+=1; - } - if(layOK>=numberofpoints){ - AliITStrackV2* tr2 = FitTrack(trs,primaryVertex,errorsprimvert); - if(tr2==0){ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - - } - continue; - } + static Int_t nClusLay[AliITSgeomTGeo::kNLayers];//counter for clusters on each layer + + //loop on different minNPoints + Int_t minMinNPoints=minNPoints; + if(AliITSReconstructor::GetRecoParam()->GetSAOnePointTracks()) minMinNPoints=2; + for(Int_t iMinNPoints=minNPoints; iMinNPoints>=minMinNPoints; iMinNPoints--) { + //loop on the different windows + for(Int_t nloop=0;nloopGetEntries();ncl++){ //loop starting from layer 0 + + ResetForFinding(); + Int_t pflag=0; + + AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[0]->At(ncl); + + if(!cl) continue; + if (cl->GetQ()<=0) continue; + + AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(0,ncl); + fPhic = arr->GetPhi(); + fLambdac = arr->GetLambda(); + if (TMath::Abs(fLambdac)>0.26*TMath::Pi()) continue; + fPhiEstimate = fPhic; + AliITStrackSA* trs = new AliITStrackSA(); + fPoint1[0]=primaryVertex[0]; + fPoint1[1]=primaryVertex[1]; - AliESDtrack outtrack; - outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); - event->AddTrack(&outtrack); - ntrack++; - Int_t nct = tr2->GetNumberOfClusters(); - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - kl->SetBit(kSAflag); - - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); + + fPoint2[0]=arr->GetX(); + fPoint2[1]=arr->GetY(); + for(Int_t i=0;i0){ + pflag=1; + fPoint3[0] = fPointc[0]; + fPoint3[1] = fPointc[1]; + } + nClusLay[2] = SearchClusters(2,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); + if(nClusLay[1]==0 && nClusLay[2]==0) pflag=0; + if(nClusLay[2]!=0 && nClusLay[1]!=0){ pflag=1; UpdatePoints();} + if(nClusLay[2]!=0 && nClusLay[1]==0){ + pflag=1; + fPoint3[0]=fPointc[0]; + fPoint3[1]=fPointc[1]; + } + + nClusLay[3] = SearchClusters(3,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); + pflag=1; + if(nClusLay[3]!=0) UpdatePoints(); + nClusLay[4] = SearchClusters(4,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); + pflag=1; + if(nClusLay[4]!=0) UpdatePoints(); + nClusLay[5] = SearchClusters(5,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag); + + + Int_t layOK=0; + //check of the candidate track + for(Int_t nnp=0;nnp=iMinNPoints) { + //printf("-NPOINTS: %d; MAP: %d %d %d %d %d %d\n",layOK,nClusLay[0],nClusLay[1],nClusLay[2],nClusLay[3],nClusLay[4],nClusLay[5]); + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + //printf("-NPOINTS fit: %d\n",tr2->GetNumberOfClusters()); + + AliESDtrack outtrack; + outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); + event->AddTrack(&outtrack); + ntrack++; } - } - delete trs; - delete[] nn; - - }//end loop on clusters of layer1 - - }//end loop2 + delete trs; + }//end loop on clusters of layer1 + }//end loop on windows + }//end loop on min points + minNPoints--; //if 5/6 points are required, second loop starting - //from second layer, to find tracks with point of - //layer 1 missing - - if(!fSixPoints){ + //from second layer (SPD2), to find tracks with point of + //layer 1 missing. Not done for cosmics. + if(!fSixPoints && fOuterStartLayer==0) { + //printf("looking from SPD2\n"); // counter for clusters on each layer - Int_t * nn = new Int_t[fGeom->GetNlayers()-1]; for(Int_t nloop=0;nloopGetEntries(); while(ncl2--){ //loop starting from layer 2 - ResetForFinding(); - Int_t pflag=0; - AliITSclusterV2* cl = layer2.GetCluster(ncl2); - if(cl->IsUsed()==1) continue; - if(cl->TestBit(kSAflag)==kTRUE) continue; - fPhic = fTable->GetPhiCluster(1,ncl2); - fLambdac = fTable->GetLambdaCluster(1,ncl2); + ResetForFinding(); + Int_t pflag=0; + AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[1]->At(ncl2); + + if(!cl) continue; + AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(1,ncl2); + fPhic = arr->GetPhi(); + fLambdac = arr->GetLambda(); fPhiEstimate = fPhic; - AliITStrackSA* trs = new AliITStrackSA(); - fPoint1[0]=primaryVertex[0]; - fPoint1[1]=primaryVertex[1]; - fPoint2[0]=fTable->GetXCluster(1,ncl2); - fPoint2[1]=fTable->GetYCluster(1,ncl2); - - for(Int_t kk=0;kkGetNlayers()-1;kk++)nn[kk] = 0; - for(Int_t kk=0;kkGetNlayers()-1;kk++){ - nn[kk] = SearchClusters(kk+1,fPhiWin[nloop],fLambdaWin[nloop], - trs,primaryVertex[2],pflag); - if(nn[kk]==0)break; - if(kk>0){ - UpdatePoints(); - pflag = 1; - } - } - Int_t fl=0; - for(Int_t nnp=0;nnpGetNlayers()-1;nnp++){ - if(nn[nnp]!=0) fl+=1; - } - if(fl>=5){ // 5/6 - AliITStrackV2* tr2 = FitTrack(trs,primaryVertex,errorsprimvert); - if(tr2==0){ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - - } - - continue; - } - + + AliITStrackSA* trs = new AliITStrackSA(); + fPoint1[0]=primaryVertex[0]; + fPoint1[1]=primaryVertex[1]; + + fPoint2[0]=arr->GetX(); + fPoint2[1]=arr->GetY(); + for(Int_t kk=0;kk=minNPoints){ // 5/6 + //printf("--NPOINTS: %d; MAP: %d %d %d %d %d %d\n",layOK,nClusLay[0],nClusLay[1],nClusLay[2],nClusLay[3],nClusLay[4],nClusLay[5]); + + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + //printf("--NPOINTS fit: %d\n",tr2->GetNumberOfClusters()); + AliESDtrack outtrack; outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); event->AddTrack(&outtrack); ntrack++; - Int_t nct = tr2->GetNumberOfClusters(); - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl==0) continue; - kl->SetBit(kSAflag); - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl==0) continue; - if(kl->TestBit(kSAflag)==kTRUE) kl->ResetBit(kSAflag); - - } - } + + } + delete trs; }//end loop on clusters of layer2 } - delete [] nn; - } //end opt="5/6" + + } //end opt="5/6" + + // search for tracks starting from SPD2, SDD1, SDD2, SSD2 + // for cosmics (A. Dainese 31.07.07) + if(fOuterStartLayer>0 && !AliITSReconstructor::GetRecoParam()->GetSAOnePointTracks()) { + for(Int_t innLay=1; innLay<=fOuterStartLayer; innLay++) { + //printf("Searching from layer %d outward\n",innLay); + minNPoints=AliITSgeomTGeo::GetNLayers()-innLay; + for(Int_t i=innLay;iGetLayersToSkip(i)) + minNPoints--; + // counter for clusters on each layer + + for(Int_t nloop=0;nloopGetEntries(); + while(nclInnLay--){ //loop starting from layer innLay + ResetForFinding(); + Int_t pflag=0; + AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[innLay]->At(nclInnLay); + + if(!cl) continue; + AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(innLay,nclInnLay); + fPhic = arr->GetPhi(); + fLambdac = arr->GetLambda(); + fPhiEstimate = fPhic; + + AliITStrackSA* trs = new AliITStrackSA(); + fPoint1[0]=primaryVertex[0]; + fPoint1[1]=primaryVertex[1]; + fPoint2[0]=arr->GetX(); + fPoint2[1]=arr->GetY(); + + Int_t kk; + for(kk=0;kk=minNPoints){ + //printf("---NPOINTS: %d; MAP: %d %d %d %d %d %d\n",layOK,nClusLay[0],nClusLay[1],nClusLay[2],nClusLay[3],nClusLay[4],nClusLay[5]); + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + //printf("---NPOINTS fit: %d\n",tr2->GetNumberOfClusters()); + + AliESDtrack outtrack; + outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); + event->AddTrack(&outtrack); + ntrack++; + + } + + delete trs; + }//end loop on clusters of innLay + } //end loop on window sizes + + } //end loop on innLay + } //end if(fOuterStartLayer>0) + - delete [] firstmod; - delete fTable;fTable=0; + // search for 1-point tracks in SPD, only for cosmics + // (A.Dainese 21.03.08) + if(AliITSReconstructor::GetRecoParam()->GetSAOnePointTracks() && + TMath::Abs(event->GetMagneticField())<0.01) { + Int_t outerLayer=1; // only SPD + for(Int_t innLay=0; innLay<=TMath::Min(1,fOuterStartLayer); innLay++) { + // counter for clusters on each layer + + for(Int_t nloop=0;nloopGetEntries(); + while(nclInnLay--){ //loop starting from layer innLay + ResetForFinding(); + Int_t pflag=0; + AliITSRecPoint* cl = (AliITSRecPoint*)fCluLayer[innLay]->At(nclInnLay); + + if(!cl) continue; + AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(innLay,nclInnLay); + fPhic = arr->GetPhi(); + fLambdac = arr->GetLambda(); + fPhiEstimate = fPhic; + + AliITStrackSA* trs = new AliITStrackSA(); + fPoint1[0]=primaryVertex[0]; + fPoint1[1]=primaryVertex[1]; + fPoint2[0]=arr->GetX(); + fPoint2[1]=arr->GetY(); + + Int_t kk; + for(kk=0;kkGetNumberOfClusters()); + + AliESDtrack outtrack; + outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); + event->AddTrack(&outtrack); + ntrack++; + + } + + delete trs; + }//end loop on clusters of innLay + } //end loop on window sizes + + } //end loop on innLay + } // end search 1-point tracks + Info("FindTracks","Number of found tracks: %d",event->GetNumberOfTracks()); return 0; } - - + //________________________________________________________________________ -AliITStrackV2* AliITStrackerSA::FitTrack(AliITStrackSA* tr,Double_t *primaryVertex,Double_t *errorsprimvert){ - //fit of the found track +AliITStrackV2* AliITStrackerSA::FitTrack(AliITStrackSA* tr,Double_t *primaryVertex,Bool_t onePoint) { + //fit of the found track (most general case, also <6 points, layers missing) + // A.Dainese 16.11.07 - Int_t * firstmod = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - firstmod[i]=fGeom->GetModuleIndex(i+1,1,1); - } + const Int_t kMaxClu=AliITStrackSA::kMaxNumberOfClusters; - Int_t nclusters = tr->GetNumberOfClustersSA(); - TObjArray** listlayer = new TObjArray*[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - listlayer[i] = new TObjArray(0,0); + static Int_t firstmod[AliITSgeomTGeo::kNLayers]; + + static Int_t clind[AliITSgeomTGeo::kNLayers][kMaxClu]; + static Int_t clmark[AliITSgeomTGeo::kNLayers][kMaxClu]; + static Int_t nnn[AliITSgeomTGeo::kNLayers]; + static Int_t kkk[AliITSgeomTGeo::kNLayers]; + static Int_t end[AliITSgeomTGeo::kNLayers]; + static AliITSRecPoint *listlayer[AliITSgeomTGeo::kNLayers][kMaxClu]; + + for(Int_t i=0;iGetNlayers()]; - for(Int_t i=0;iGetNlayers();i++)nnn[i]=0; - for(Int_t ncl=0;nclGetClusterIndexSA(ncl); - AliITSclusterV2* cl = (AliITSclusterV2*)GetCluster(index); + Int_t nclusters = tr->GetNumberOfClustersSA(); + for(Int_t ncl=0;nclGetClusterIndexSA(ncl); + AliITSRecPoint* cl = (AliITSRecPoint*)GetCluster(index); if(cl->TestBit(kSAflag)==kTRUE) cl->ResetBit(kSAflag); Int_t lay = (index & 0xf0000000) >> 28; - if(lay==0) { listlayer[0]->AddLast(cl); clind0[nnn[0]]=index;nnn[0]++;} - if(lay==1) { listlayer[1]->AddLast(cl); clind1[nnn[1]]=index;nnn[1]++;} - if(lay==2) { listlayer[2]->AddLast(cl); clind2[nnn[2]]=index;nnn[2]++;} - if(lay==3) { listlayer[3]->AddLast(cl); clind3[nnn[3]]=index;nnn[3]++;} - if(lay==4) { listlayer[4]->AddLast(cl); clind4[nnn[4]]=index;nnn[4]++;} - if(lay==5) { listlayer[5]->AddLast(cl); clind5[nnn[5]]=index;nnn[5]++;} + Int_t nInLay=end[lay]; + listlayer[lay][nInLay]=cl; + Int_t ind=nnn[lay]; + clind[lay][ind]=index; + end[lay]++; } - delete [] nnn; - - Int_t * end = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - if(listlayer[i]->GetEntries()==0) end[i]=1; - else end[i]=listlayer[i]->GetEntries(); + for(Int_t nlay=0;nlayGetNumberOfMarked(nlay);ncl++){ + Int_t mark = tr->GetClusterMark(nlay,ncl); + Int_t ind=kkk[nlay]; + clmark[nlay][ind]=mark; + } } - for(Int_t l1=0;l1At(l1); - TVector3** recp = new TVector3*[3]; - TVector3** errs = new TVector3*[3]; - recp[0] = new TVector3(primaryVertex[0],primaryVertex[1],primaryVertex[2]); - errs[0] = new TVector3(errorsprimvert[0],errorsprimvert[1],errorsprimvert[2]); - Double_t x1,y1,z1,sx1,sy1,sz1; - Double_t x2,y2,z2,sx2,sy2,sz2; - AliITSclusterV2* p1=0; - AliITSclusterV2* p2=0; - Int_t index1=clind0[l1]; - Int_t index2=0; - for(Int_t l2=0;l2At(l2); - index2=clind1[l2]; - for(Int_t l3=0;l3At(l3); - - if(cl0==0 && cl1!=0) { - p2 = cl2;index1=clind2[l3]; - p1=cl1; - - } - if(cl0!=0 && cl1==0){ - p1=cl0; - p2=cl2;index2=clind2[l3]; - } - if(cl0!=0 && cl1!=0){ - p1=cl0; - p2=cl1; - } - Int_t lay1=(index1 & 0xf0000000) >> 28; - Int_t cln1=(index1 & 0x0fffffff) >> 00; - Int_t lay2=(index2 & 0xf0000000) >> 28; - Int_t cln2=(index2 & 0x0fffffff) >> 00; - x1 = fTable->GetXCluster(lay1,cln1); - x2 = fTable->GetXCluster(lay2,cln2); - y1 = fTable->GetYCluster(lay1,cln1); - y2 = fTable->GetYCluster(lay2,cln2); - z1 = fTable->GetZCluster(lay1,cln1); - z2 = fTable->GetZCluster(lay2,cln2); - sx1 = fTable->GetXClusterError(lay1,cln1); - sx2 = fTable->GetXClusterError(lay2,cln2); - sy1 = fTable->GetYClusterError(lay1,cln1); - sy2 = fTable->GetYClusterError(lay2,cln2); - sz1 = fTable->GetZClusterError(lay1,cln1); - sz2 = fTable->GetZClusterError(lay2,cln2); - Double_t phi1 = fTable->GetPhiCluster(lay1,cln1); - Int_t module1 = p1->GetDetectorIndex()+firstmod[0]; - recp[1] = new TVector3(x1,y1,z1); - errs[1] = new TVector3(sx1,sy1,sz1); - recp[2] = new TVector3(x2,y2,z2); - errs[2] = new TVector3(sx2,sy2,sz2); - - //fit on the Riemann sphere - Float_t seed1,seed2,seed3; - AliITSRiemannFit fit; - Int_t rf = fit.FitHelix(3,recp,errs,seed1,seed2,seed3); //this gives phi,tgl,curvature to start Kalman Filter - if(rf==0) { - for(Int_t i=1;i<3;i++){ - delete recp[i]; - delete errs[i]; - } - continue; - } - Double_t phi=seed1; - Double_t tgl=seed2; - - if(phi1>0){ - if(seed1>-TMath::Pi() && seed1<-0.5*TMath::Pi()){ - phi=seed1+1.5*TMath::Pi(); - tgl=seed2; - } - if(seed1>-0.5*TMath::Pi() && seed1<0.5*TMath::Pi()){ - phi=seed1+0.5*TMath::Pi(); - tgl=(-1)*seed2; - } - if(seed1>0.5*TMath::Pi() && seed1-TMath::Pi() && seed1<-0.5*TMath::Pi()){ - phi=seed1+0.5*TMath::Pi(); - tgl=(-1)*seed2; - } - if(seed1>-0.5*TMath::Pi() && seed1<0.5*TMath::Pi()){ - phi=seed1-0.5*TMath::Pi(); - tgl=seed2; - } - if(seed1>0.5*TMath::Pi() && seed1GetModuleId(module1,layer,ladder,detector); - Float_t yclu1 = p1->GetY(); - Float_t zclu1 = p1->GetZ(); - Double_t cv=Curvature(primaryVertex[0],primaryVertex[1],x1,y1,x2,y2); - - for(Int_t l4=0;l4At(l4); - for(Int_t l5=0;l5At(l5); - for(Int_t l6=0;l6At(l6); - AliITStrackSA* trac = new AliITStrackSA(layer,ladder,detector,yclu1,zclu1,phi,tgl,cv,1); - - if(cl5!=0) trac->AddClusterV2(5,(clind5[l6] & 0x0fffffff)>>0); - if(cl4!=0) trac->AddClusterV2(4,(clind4[l5] & 0x0fffffff)>>0); - if(cl3!=0) trac->AddClusterV2(3,(clind3[l4] & 0x0fffffff)>>0); - if(cl2!=0) trac->AddClusterV2(2,(clind2[l3] & 0x0fffffff)>>0); - if(cl1!=0) trac->AddClusterV2(1,(clind1[l2] & 0x0fffffff)>>0); - if(cl0!=0) trac->AddClusterV2(0,(clind0[l1] & 0x0fffffff)>>0); - + + Int_t firstLay=-1,secondLay=-1; + for(Int_t i=0;iGetDetectorIndex()+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 cv=0,tgl2=0,phi2=0; + + Int_t cln1=mrk1; + AliITSclusterTable* arr1 = (AliITSclusterTable*)GetClusterCoord(firstLay,cln1); + 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); + 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)); + } else { // special case of 1-point tracks, only for cosmics (B=0) + x2 = primaryVertex[0]; + y2 = primaryVertex[1]; + z2 = primaryVertex[2]; + cv = 0; + tgl2 = (z1-z2)/TMath::Sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2)); + phi2 = TMath::ATan2((y1-y2),(x1-x2)); + } + + + AliITStrackSA* trac = new AliITStrackSA(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]); + } + //fit with Kalman filter using AliITStrackerMI::RefitAt() - - AliITStrackMI* ot = new AliITStrackSA(*trac); + AliITStrackMI* ot = new AliITStrackSA(*trac); - ot->ResetCovariance(); + ot->ResetCovariance(10.); ot->ResetClusters(); - if(RefitAt(49.,ot,trac)){ //fit from layer 1 to layer 6 - + if(RefitAt(AliITSRecoParam::GetrInsideITSscreen(),ot,trac)){ //fit from layer 1 to layer 6 AliITStrackMI *otrack2 = new AliITStrackMI(*ot); - otrack2->ResetCovariance(); + otrack2->ResetCovariance(10.); otrack2->ResetClusters(); //fit from layer 6 to layer 1 - if(RefitAt(3.7,otrack2,ot)) { + if(RefitAt(AliITSRecoParam::GetrInsideSPD1(),otrack2,ot)) { fListOfTracks->AddLast(otrack2); + fListOfSATracks->AddLast(trac); } else { delete otrack2; + delete trac; } } delete ot; - delete trac; }//end loop layer 6 }//end loop layer 5 - }//end loop layer 4 - - for(Int_t i=1;i<3;i++){ - delete recp[i]; - delete errs[i]; - } + }//end loop layer 4 }//end loop layer 3 }//end loop layer 2 - delete recp[0]; - delete errs[0]; - delete[] recp; - delete[] errs; }//end loop layer 1 - delete [] end; - Int_t dim=fListOfTracks->GetEntries(); - if(dim==0){ - for(Int_t i=0;iGetNlayers();i++){ - delete listlayer[i]; - } - delete [] listlayer; - delete [] firstmod; - return 0; - } - AliITStrackV2* otrack =(AliITStrackV2*)FindTrackLowChiSquare(fListOfTracks,dim); - if(otrack==0) { - for(Int_t i=0;iGetNlayers();i++){ - delete listlayer[i]; - } - delete [] listlayer; - delete [] firstmod; - return 0; - } - Int_t * indexc = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++) indexc[i]=0; + if(fListOfTracks->GetEntries()==0) return 0; + + Int_t lowchi2 = FindTrackLowChiSquare(); + AliITStrackV2* otrack =(AliITStrackV2*)fListOfTracks->At(lowchi2); + AliITStrackSA* trsa = (AliITStrackSA*)fListOfSATracks->At(lowchi2); + + if(otrack==0) return 0; + + Int_t indexc[AliITSgeomTGeo::kNLayers]; + for(Int_t i=0;iGetNumberOfClusters();nind++){ indexc[nind] = otrack->GetClusterIndex(nind); } - AliITSclusterV2* cl0 = (AliITSclusterV2*)GetCluster(indexc[0]); - AliITSclusterV2* cl1 = (AliITSclusterV2*)GetCluster(indexc[1]); - AliITSclusterV2* cl2 = (AliITSclusterV2*)GetCluster(indexc[2]); - AliITSclusterV2* cl3 = (AliITSclusterV2*)GetCluster(indexc[3]); - AliITSclusterV2* cl4 = (AliITSclusterV2*)GetCluster(indexc[4]); - Int_t labl[3]={-1,-1,-1}; - if(otrack->GetNumberOfClusters()==fGeom->GetNlayers()){ - AliITSclusterV2* cl5 = (AliITSclusterV2*)GetCluster(indexc[5]); - labl[0]=cl5->GetLabel(0); - labl[1]=cl5->GetLabel(1); - labl[2]=cl5->GetLabel(2); - } - delete [] indexc; - if(otrack->GetNumberOfClusters()==(fGeom->GetNlayers()-1)){ - labl[0]=-1; - labl[1]=-1; - labl[2]=-1; + Int_t labl[6][3]; + for(Int_t i=0;iGetNumberOfClusters()) { + 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; + } } - Int_t numberofpoints; - if(fSixPoints) numberofpoints=6; - else numberofpoints=5; + CookLabel(otrack,0.); //MI change - to see fake ratio - Int_t label = Label(cl0->GetLabel(0),cl1->GetLabel(0), - cl2->GetLabel(0),cl3->GetLabel(0), - cl4->GetLabel(0),labl[0], - cl0->GetLabel(1),cl1->GetLabel(1), - cl2->GetLabel(1),cl3->GetLabel(1), - cl4->GetLabel(1),labl[1], - cl0->GetLabel(2),cl1->GetLabel(2), - cl2->GetLabel(2),cl3->GetLabel(2), - cl4->GetLabel(2),labl[2],numberofpoints); + + 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;iGetNumberOfClusters();i++) + if(labl[i][0]==label || labl[i][1]==label || labl[i][2]==label) lflag++; + if(lflagGetNumberOfClusters()) label = -label; otrack->SetLabel(label); - for(Int_t i=0;iGetNlayers();i++){ - delete listlayer[i]; - } - delete [] listlayer; - delete [] firstmod; - return otrack; - -} - -//_______________________________________________________________________ -void AliITStrackerSA::UseFoundTracksV2(Int_t evnum,TTree* treev2){ - // Marks as used clusters belonging to tracks found with V2 TPC+ITS tracking - //(or AliITStrackV2 tracks found with function FindTracks of this class) - - //Get primary vertex - if(fVertexer){ - if(fVert)delete fVert; - fVert = fVertexer->FindVertexForCurrentEvent(evnum); - } - else { - gAlice->GetEvent(evnum); - if(!fVert){ - Fatal("FindTracks","Vertex is missing\n"); - return; - } + //remove clusters of found track + for(Int_t nlay=0;nlayGetNumberOfMarked(nlay);cln++){ + Int_t index = trsa->GetClusterMark(nlay,cln); + fCluLayer[nlay]->RemoveAt(index); + RemoveClusterCoord(nlay,index); + fCluLayer[nlay]->Compress(); + } } - Double_t primaryVertex[3]; - fVert->GetXYZ(primaryVertex); - if(!fTable){ - fTable = new AliITSclusterTable(fGeom,this,primaryVertex); - fTable->FillArray(fITSclusters); - fTable->FillArrayCoorAngles(); - } + return otrack; - TBranch* bra = (TBranch*)treev2->GetBranch("tracks"); - if(!bra) Warning("UseFoundTracksV2","No branch for track tree"); - AliITStrackV2* ttrrt = new AliITStrackV2; - bra->SetAddress(&ttrrt); - - for(Int_t nj=0;njGetEntries();nj++){ - treev2->GetEvent(nj); - Int_t ncl = ttrrt->GetNumberOfClusters(); - for(Int_t k=0;kGetClusterIndex(k); - AliITSclusterV2* clui = (AliITSclusterV2*)GetCluster(index); - if(clui->IsUsed()==0) clui->Use(); - - } - } - delete ttrrt; - } -//_______________________________________________________________________ -void AliITStrackerSA::UseFoundTracksV2(AliESD *event){ - // Marks as used clusters belonging to tracks found with V2 TPC+ITS tracking - - //Get primary vertex - - Double_t primaryVertex[3]; - event->GetVertex()->GetXYZ(primaryVertex); - - if(!fTable){ - fTable = new AliITSclusterTable(fGeom,this,primaryVertex); - fTable->FillArray(fITSclusters); - fTable->FillArrayCoorAngles(); - } - - Int_t ntracks = event->GetNumberOfTracks(); - while (ntracks--) { - AliESDtrack *esd=event->GetTrack(ntracks); - if ((esd->GetStatus()& - AliESDtrack::kITSin|AliESDtrack::kTPCin)==0) continue; - UInt_t idx[6]; - Int_t ncl = esd->GetITSclusters(idx); - for(Int_t clu=0; cluIsUsed()==0) cl->Use(); - } - } - - Info("UseFoundTracksV2","Clusters of tracks prolonged from TPC deleted"); - -} -/* //_______________________________________________________ -Int_t AliITStrackerSA::SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, AliITStrackSA* trs,Double_t zvertex,Int_t pflag){ +Int_t AliITStrackerSA::SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, AliITStrackSA* trs,Double_t /*zvertex*/,Int_t pflag){ //function used to to find the clusters associated to the track - Int_t nc=0; - AliITSlayer &lay = fgLayers[layer]; - Int_t * firstmod = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - firstmod[i]=fGeom->GetModuleIndex(i+1,1,1); - } - if(pflag==1){ - - Float_t cx1,cx2,cy1,cy2; - FindEquation(fPoint1[0],fPoint1[1],fPoint2[0],fPoint2[1],fPoint3[0],fPoint3[1],fCoef1,fCoef2,fCoef3); - Int_t fun = FindIntersection(fCoef1,fCoef2,fCoef3,-(lay.GetR()*lay.GetR()),cx1,cy1,cx2,cy2); - if(fun==0) { - delete[] firstmod; - return 0; - } - - Double_t fi1 =TMath::ATan2(cy1,cx1); - Double_t fi2 =TMath::ATan2(cy2,cx2); - fPhiEstimate = ChoosePoint(fi1,fi2,fPhic); - } - - Double_t zed = TMath::Tan(fLambdac)*lay.GetR()+zvertex; - Double_t zed1 = TMath::Tan(fLambdac+lambdawindow)*lay.GetR()+zvertex; - Double_t zed2 = TMath::Tan(fLambdac-lambdawindow)*lay.GetR()+zvertex; - - Double_t fi = fPhiEstimate; - Int_t nmod = lay.FindDetectorIndex(fi,zed); - if (nmod < 0) { - delete[] firstmod; - return 0; - } - nmod += firstmod[layer]; - - Int_t nm[8]={0,0,0,0,0,0,0,0}; - nm[0] = lay.FindDetectorIndex(fi+phiwindow,zed); - nm[1] = lay.FindDetectorIndex(fi-phiwindow,zed); - nm[2] = lay.FindDetectorIndex(fi,zed1); - nm[3] = lay.FindDetectorIndex(fi,zed2); - nm[4] = lay.FindDetectorIndex(fi+phiwindow,zed1); - nm[5] = lay.FindDetectorIndex(fi-phiwindow,zed1); - nm[6] = lay.FindDetectorIndex(fi+phiwindow,zed2); - nm[7] = lay.FindDetectorIndex(fi-phiwindow,zed2); - - - Int_t nn=0; - TArrayI* array =(TArrayI*)fTable->GetListOfClusters(nmod); - TArrayI* listc = new TArrayI(array->GetSize()); - for(Int_t i=0;iGetSize();i++){ - Int_t in=(Int_t)array->At(i); - listc->AddAt(in,nn); - nn++; - } - - Int_t k=0; - Int_t val; - while(k<8){ - for(Int_t h=k+1;h<8;h++){ - if(nm[k]>nm[h]){ - val=nm[k]; - nm[k]=nm[h]; - nm[h]=val; - } - - } - k++; - } - - Int_t value=-5; - - for(Int_t ii=0;ii<8;ii++){ - if(nm[ii]!=value && nm[ii]!=nmod && nm[ii]>=0){ - TArrayI* ar =(TArrayI*)fTable->GetListOfClusters(nm[ii]+firstmod[layer]); - listc->Set(listc->GetSize()+ar->GetSize()); - for(Int_t j=0;jGetSize();j++){ - Int_t in=(Int_t)ar->At(j); - listc->AddAt(in,nn); - nn++; - value=nm[ii]; - } - } - } - - - for(Int_t i=0;iGetSize();i++){ - Int_t index = (Int_t)listc->At(i); - AliITSclusterV2* cllay = lay.GetCluster(index); - if(cllay==0) continue; - if(cllay->IsUsed()==1) continue; - if(cllay->TestBit(kSAflag)==kTRUE) continue; - Double_t phi = fTable->GetPhiCluster(layer,index); - Double_t lambda= fTable->GetLambdaCluster(layer,index); - - if(TMath::Abs(fLambdac-lambda)GetNumberOfClustersSA()==15){ - delete[] firstmod; - delete listc; - return 0; - } - trs->AddClusterSA(layer,index); - cllay->SetBit(kSAflag); - fPhiEstimate=phi; - fPointc[0]=fTable->GetXCluster(layer,index); - fPointc[1]=fTable->GetYCluster(layer,index); - } - - } - delete listc; - delete [] firstmod; - return nc; - -} -*/ + if(AliITSReconstructor::GetRecoParam()->GetLayersToSkip(layer)) return 0; -//_______________________________________________________ -Int_t AliITStrackerSA::SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, AliITStrackSA* trs,Double_t zvertex,Int_t pflag){ - //function used to to find the clusters associated to the track Int_t nc=0; AliITSlayer &lay = fgLayers[layer]; - Double_t r=lay.GetR(),tgl=TMath::Tan(fLambdac); - + Double_t r=lay.GetR(); if(pflag==1){ Float_t cx1,cx2,cy1,cy2; FindEquation(fPoint1[0],fPoint1[1],fPoint2[0],fPoint2[1],fPoint3[0],fPoint3[1],fCoef1,fCoef2,fCoef3); if (FindIntersection(fCoef1,fCoef2,fCoef3,-r*r,cx1,cy1,cx2,cy2)==0) return 0; - Double_t fi1=TMath::ATan2(cy1,cx1); - Double_t fi2=TMath::ATan2(cy2,cx2); + Double_t fi1=TMath::ATan2(cy1-fPoint1[1],cx1-fPoint1[0]); + Double_t fi2=TMath::ATan2(cy2-fPoint1[1],cx2-fPoint1[0]); fPhiEstimate=ChoosePoint(fi1,fi2,fPhic); } - Double_t dz=r*lambdawindow*TMath::Sqrt(1+tgl*tgl) + 0.3*TMath::Abs(tgl); - Double_t zmax=r*tgl + zvertex + dz; - Double_t zmin=r*tgl + zvertex - dz; - - Int_t indmin = lay.FindClusterIndex(zmin); - Int_t indmax = lay.FindClusterIndex(zmax); - for (Int_t index=indmin; indexIsUsed()) continue; - if (c->GetQ()<=0) continue; - if (c->TestBit(kSAflag)==kTRUE) continue; - - Double_t phi =fTable->GetPhiCluster(layer,index); + + Int_t ncl = fCluLayer[layer]->GetEntries(); + for (Int_t index=0; indexAt(index); + if (!c) continue; + if (c->GetQ()<=0) continue; + + AliITSclusterTable* arr = (AliITSclusterTable*)GetClusterCoord(layer,index); + Double_t phi = arr->GetPhi(); if (TMath::Abs(phi-fPhiEstimate)>phiwindow) continue; - Double_t lambda=fTable->GetLambdaCluster(layer,index); + Double_t lambda = arr->GetLambda(); if (TMath::Abs(lambda-fLambdac)>lambdawindow) continue; - if(trs->GetNumberOfClustersSA()==15) return 0; - - trs->AddClusterSA(layer,index); + 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]=fTable->GetXCluster(layer,index); - fPointc[1]=fTable->GetYCluster(layer,index); - c->SetBit(kSAflag); + fPointc[0]=arr->GetX(); + fPointc[1]=arr->GetY(); + } return nc; } - //________________________________________________________________ void AliITStrackerSA::UpdatePoints(){ //update of points for the estimation of the curvature - //fPoint1[0]=fPoint2[0]; - //fPoint1[1]=fPoint2[1]; fPoint2[0]=fPoint3[0]; fPoint2[1]=fPoint3[1]; fPoint3[0]=fPointc[0]; @@ -1233,6 +1129,8 @@ x2,Double_t y2,Double_t x3,Double_t y3){ return 1/rad; } + + //____________________________________________________________________ Double_t AliITStrackerSA::ChoosePoint(Double_t p1, Double_t p2, Double_t pp){ @@ -1249,55 +1147,35 @@ Double_t AliITStrackerSA::ChoosePoint(Double_t p1, Double_t p2, Double_t pp){ //_________________________________________________________________ -AliITStrackV2* AliITStrackerSA::FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const { - // returns track with lowes chi square - if(dim==1){ - AliITStrackV2* trk = (AliITStrackV2*)tracklist->At(0); - return trk; - } - if(dim==0) return 0; - Double_t * chi2 = new Double_t[dim]; - Int_t * index = new Int_t[dim]; - for(Int_t i=0;iAt(i); - chi2[i]=trk->GetChi2(); - index[i]=i; - } - - Int_t w=0;Double_t value; - Int_t lp; - while(wGetEntries(); + if(dim<=1) return 0; + AliITStrackV2* trk = (AliITStrackV2*)fListOfTracks->At(0); + Double_t minChi2=trk->GetChi2(); + Int_t index=0; + for(Int_t i=1;iAt(i); + Double_t chi2=trk->GetChi2(); + if(chi2At(index[dim-1]); - delete [] chi2; - delete [] index; - return trk; - + return index; } //__________________________________________________________ -Int_t AliITStrackerSA::FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6){ +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] = {l1,l2,l3,l4,l5,l6}; + 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; - if(lb[5]==-1) num=5; + for(Int_t i=5;i>=0;i--) if(lb[i]==-1) num=i; while(k=numberofpoints) return ll; + if(lflag>=minNPoints) return ll; else return -ll; } - //_____________________________________________________________________________ -void AliITStrackerSA::SetWindowSizes(Int_t n, Double_t *phi, Double_t *lam){ +void AliITStrackerSA::SetCalculatedWindowSizes(Int_t n, Float_t phimin, Float_t phimax, Float_t lambdamin, Float_t lambdamax){ + // Set sizes of the phi and lambda windows used for track finding + fNloop = n; + if(fPhiWin) delete [] fPhiWin; + if(fLambdaWin) delete [] fLambdaWin; + fPhiWin = new Double_t[fNloop]; + fLambdaWin = new Double_t[fNloop]; + Float_t stepPhi=(phimax-phimin)/(Float_t)(fNloop-1); + Float_t stepLambda=(lambdamax-lambdamin)/(Float_t)(fNloop-1); + for(Int_t k=0;kGetFactorSAWindowSizes(); // possibility to enlarge windows for cosmics reco with large misalignments (A.Dainese) + for(Int_t k=0;kGetRotMatrix(module,rot); + Int_t lay,lad,det; fGeom->GetModuleId(module,lay,lad,det); + Float_t tx,ty,tz; fGeom->GetTrans(lay,lad,det,tx,ty,tz); + + Double_t alpha=TMath::ATan2(rot[1],rot[0])+TMath::Pi(); + Double_t phi1=TMath::Pi()/2+alpha; + if (lay==1) phi1+=TMath::Pi(); + + Float_t cp=TMath::Cos(phi1), sp=TMath::Sin(phi1); + Float_t r=tx*cp+ty*sp; + + xyz= r*cp - cl->GetY()*sp; + y= r*sp + cl->GetY()*cp; + z=cl->GetZ(); +*/ + Float_t xyz[3]; + cl->GetGlobalXYZ(xyz); + x=xyz[0]; + y=xyz[1]; + z=xyz[2]; + + phi=TMath::ATan2(y-vertex[1],x-vertex[0]); + lambda=TMath::ATan2(z-vertex[2],TMath::Sqrt((x-vertex[0])*(x-vertex[0])+(y-vertex[1])*(y-vertex[1]))); +} + +//________________________________________________________________________ +void AliITStrackerSA::GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz){ + + //returns sigmax, y, z of cluster in global coordinates +/* + Double_t rot[9]; fGeom->GetRotMatrix(module,rot); + Int_t lay,lad,det; + AliITSgeomTGeo::GetModuleId(module,lay,lad,det); + + Double_t alpha=TMath::ATan2(rot[1],rot[0])+TMath::Pi(); + Double_t phi=TMath::Pi()/2+alpha; + if (lay==1) phi+=TMath::Pi(); + Float_t cp=TMath::Cos(phi), sp=TMath::Sin(phi); +*/ + Float_t covm[6]; + cl->GetGlobalCov(covm); + sx=TMath::Sqrt(covm[0]); + sy=TMath::Sqrt(covm[3]); + sz=TMath::Sqrt(covm[5]); +/* + sx = TMath::Sqrt(sp*sp*cl->GetSigmaY2()); + sy = TMath::Sqrt(cp*cp*cl->GetSigmaY2()); + sz = TMath::Sqrt(cl->GetSigmaZ2()); +*/ +}