X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITStrackerSA.cxx;h=ce836973c5d5237bc395be001cecf02ebd0f7996;hb=c1caadda8464e94e09f9fd053473f422f6c401db;hp=486769a7de3ae4862e5e3222eefb32f6169bb942;hpb=4e05ab9aa659c1e63cb12400cdad3dcd13a8009c;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITStrackerSA.cxx b/ITS/AliITStrackerSA.cxx index 486769a7de3..ce836973c5d 100644 --- a/ITS/AliITStrackerSA.cxx +++ b/ITS/AliITStrackerSA.cxx @@ -1,4 +1,3 @@ - /************************************************************************** * Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. * * * @@ -14,6 +13,8 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ +/* $Id$ */ + //////////////////////////////////////////////////// // Stand alone tracker class // // Origin: Elisabetta Crescio // @@ -21,102 +22,186 @@ // tracks are saved as AliITStrackV2 objects // //////////////////////////////////////////////////// -#include "TArrayI.h" +#include + +#include #include -#include #include #include -#include "AliRun.h" -#include "AliITSclusterTable.h" -#include "AliITSclusterV2.h" -#include "AliITSgeom.h" -#include "AliITSRiemannFit.h" -#include "AliITStrackerSA.h" -#include "AliITStrackSA.h" -#include "AliITSVertexer.h" + +#include "AliESDEvent.h" #include "AliESDVertex.h" -#include "AliESD.h" #include "AliESDtrack.h" +#include "AliITSVertexer.h" +#include "AliITSclusterTable.h" +#include "AliITSRecPoint.h" +#include "AliITSgeomTGeo.h" +#include "AliITStrackSA.h" +#include "AliITStrackerSA.h" +#include "AliITSReconstructor.h" +#include "AliRun.h" ClassImp(AliITStrackerSA) //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA():AliITStrackerV2(){ +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), +fITSclusters(0), +fSixPoints(0), +fOuterStartLayer(0), +fCluLayer(0), +fCluCoord(0){ // Default constructor Init(); + } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITSgeom *geom):AliITStrackerV2(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), +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):AliITStrackerV2(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), +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(const AliITStrackerSA &trkr) : - AliITStrackerV2(trkr) { - // Copy constructor - // Copies are not allowed. The method is protected to avoid misuse. - Error("AliITStrackerSA","Copy constructor not allowed\n"); -} - -//______________________________________________________________________ -AliITStrackerSA& AliITStrackerSA::operator=(const - AliITStrackerSA& /* trkr */){ - // Assignment operator - // Assignment is not allowed. The method is protected to avoid misuse. - Error("= operator","Assignment operator not allowed\n"); - return *this; + } //____________________________________________________________________________ -AliITStrackerSA::AliITStrackerSA(AliITSgeom *geom, AliITSVertexer *vertexer):AliITStrackerV2(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), +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):AliITStrackerV2(){ +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), +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; + } //____________________________________________________________________________ @@ -129,12 +214,49 @@ AliITStrackerSA::~AliITStrackerSA(){ } fVert = 0; fVertexer = 0; - fGeom = 0; + if(fPhiWin)delete []fPhiWin; if(fLambdaWin)delete []fLambdaWin; - fTable =0; fListOfTracks->Delete(); - } + delete fListOfTracks; + 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(){ @@ -150,13 +272,15 @@ void AliITStrackerSA::Init(){ fPointc[1]=0; fVert = 0; fVertexer = 0; - fGeom = 0; SetWindowSizes(); - fTable = 0; fITSclusters = 0; SetSixPoints(); + SetOuterStartLayer(0); + SetSAFlag(kFALSE); fListOfTracks=new TObjArray(0,0); -} + fCluLayer = 0; + fCluCoord = 0; + } //_______________________________________________________________________ void AliITStrackerSA::ResetForFinding(){ // Reset data members used in all loops during track finding @@ -171,310 +295,131 @@ void AliITStrackerSA::ResetForFinding(){ 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; - 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,fTable); - nn[1] = SearchClusters(1,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); - - 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,fTable); - 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,fTable); - pflag=1; - if(nn[3]!=0) UpdatePoints(); - nn[4] = SearchClusters(4,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); - pflag=1; - if(nn[4]!=0) UpdatePoints(); - nn[5] = SearchClusters(5,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); - - - 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->IsUsed()==1) kl->Use(); - } - continue; - } - outrack=tr2; - out->Fill(); - ntrack++; - Int_t nct = tr2->GetNumberOfClusters(); - - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->IsUsed()==0) kl->Use(); - - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->IsUsed()==1) kl->Use(); - } - } - 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; - 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,fTable); - 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->IsUsed()==1) kl->Use(); - } - 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; - if(kl->IsUsed()==0) kl->Use(); - } - } - 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->IsUsed()==1) kl->Use(); - } - } - delete trs; - }//end loop on clusters of layer2 - } - delete [] nn; - } // if(!fSixPoints.... - - delete [] firstmod; - delete fTable; -} + //______________________________________________________________________ -Int_t AliITStrackerSA::FindTracks(AliESD* event){ +Int_t AliITStrackerSA::FindTracks(AliESDEvent* event){ - // Track finder using the ESD object +// Track finder using the ESD object + //controllare numero cluster sui layer1 e 2 (morti?) + //non trova tracce...controllare.. + 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(); + 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); + //Creates TClonesArray with clusters for each layer. The clusters already used + //by AliITStrackerMI are not considered + Int_t nclusters[6]={0,0,0,0,0,0}; + Int_t dmar[6]={0,0,0,0,0,0}; + if (fCluLayer == 0) { + fCluLayer = new TClonesArray*[AliITSgeomTGeo::GetNLayers()]; + fCluCoord = new TClonesArray*[AliITSgeomTGeo::GetNLayers()]; + for(Int_t i=0;iTestBit(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]); + } - 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; + for(Int_t ilay=0;ilayTestBit(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]++; + } + } + + Int_t minNPoints = (fSixPoints ? AliITSgeomTGeo::GetNLayers() : AliITSgeomTGeo::GetNLayers()-1); + for(Int_t i=0;iGetLayersToSkip(i)) { + minNPoints--; + } } - //Fill array with cluster indices for each module - if(!fTable){ - fTable = new AliITSclusterTable(fGeom,this,primaryVertex); - fTable->FillArray(fITSclusters); - fTable->FillArrayCoorAngles(); - } - 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 + Int_t * nn = new Int_t[AliITSgeomTGeo::GetNLayers()];//counter for clusters on each layer for(Int_t nloop=0;nloopGetEntries();ncl++){ //loop starting from layer 0 + ResetForFinding(); Int_t pflag=0; - AliITSclusterV2* cl = (AliITSclusterV2*)layer.GetCluster(ncl); - if(cl==0) continue; - if(cl->IsUsed()==1) continue; - fPhic = fTable->GetPhiCluster(0,ncl); - fLambdac = fTable->GetLambdaCluster(0,ncl); + + 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(); + 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,fTable); - - nn[1] = SearchClusters(1,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); + + + fPoint2[0]=arr->GetX(); + fPoint2[1]=arr->GetY(); + for(Int_t i=0;i0){ pflag=1; fPoint3[0] = fPointc[0]; fPoint3[1] = fPointc[1]; } - nn[2] = SearchClusters(2,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); + 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){ @@ -483,323 +428,408 @@ Int_t AliITStrackerSA::FindTracks(AliESD* event){ fPoint3[1]=fPointc[1]; } - nn[3] = SearchClusters(3,fPhiWin[nloop],fLambdaWin[nloop],trs,primaryVertex[2],pflag,fTable); + 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,fTable); + 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,fTable); + 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 + //check of the candidate track + for(Int_t nnp=0;nnp=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->IsUsed()==1) kl->Use(); - } - continue; - } - + + if(layOK>=minNPoints){ + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + 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->IsUsed()==0) kl->Use(); - - } - } - else{ - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->IsUsed()==1) kl->Use(); - } } delete trs; - delete[] nn; + }//end loop on clusters of layer1 + //end loop2 + } + + delete[] nn; - }//end loop on clusters of layer1 - }//end loop2 + 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 + if(!fSixPoints) { + //printf("looking from SPD2\n"); // counter for clusters on each layer - Int_t * nn = new Int_t[fGeom->GetNlayers()-1]; + Int_t * nn = new Int_t[AliITSgeomTGeo::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; - 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,fTable); - 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){ - ntrack++; - Int_t nct = trs->GetNumberOfClustersSA(); - while(nct--){ - Int_t index = trs->GetClusterIndexSA(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl->IsUsed()==1) kl->Use(); - } - - 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 + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + AliESDtrack outtrack; outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); event->AddTrack(&outtrack); - Int_t nct = tr2->GetNumberOfClusters(); - while(nct--){ - Int_t index = tr2->GetClusterIndex(nct); - AliITSclusterV2* kl = (AliITSclusterV2*)GetCluster(index); - if(kl==0) continue; - if(kl->IsUsed()==0) kl->Use(); - } - } - 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->IsUsed()==1) kl->Use(); - } - } + ntrack++; + + } + 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) { + 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 + Int_t * nn = new Int_t[AliITSgeomTGeo::GetNLayers()-innLay]; + 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){ + AliITStrackV2* tr2 = 0; + tr2 = FitTrack(trs,primaryVertex); + if(!tr2) continue; + + + AliESDtrack outtrack; + outtrack.UpdateTrackParams(tr2,AliESDtrack::kITSin); + event->AddTrack(&outtrack); + ntrack++; + + } + + delete trs; + }//end loop on clusters of innLay + } //end loop on window sizes + + delete [] nn; + } //end loop on innLay + } //end if(fOuterStartLayer>0) + + delete [] firstmod; - delete fTable; 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) { + //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); - } - AliITStrackV2* otrack2; - Int_t nclusters = tr->GetNumberOfClustersSA(); - TObjArray** listlayer = new TObjArray*[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++){ - listlayer[i] = new TObjArray(0,0); - } + Int_t * firstmod = new Int_t[AliITSgeomTGeo::GetNLayers()]; + TObjArray** listlayer = new TObjArray*[AliITSgeomTGeo::GetNLayers()]; + Int_t ** clind=new Int_t*[AliITSgeomTGeo::GetNLayers()]; + Int_t ** clmark=new Int_t*[AliITSgeomTGeo::GetNLayers()]; + Int_t * nnn = new Int_t[AliITSgeomTGeo::GetNLayers()]; + Int_t * kkk = new Int_t[AliITSgeomTGeo::GetNLayers()]; - TArrayI clind0(20); - TArrayI clind1(20); - TArrayI clind2(20); - TArrayI clind3(20); - TArrayI clind4(20); - TArrayI clind5(20); + const Int_t maxclu=AliITStrackSA::GetMaxNumberOfClusters(); - Int_t * nnn = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++)nnn[i]=0; + for(Int_t i=0;iGetClusterIndexSA(ncl); - AliITSclusterV2* cl = (AliITSclusterV2*)GetCluster(index); - if(cl->IsUsed()==1) cl->Use(); + 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]++;} + listlayer[lay]->AddLast(cl); + Int_t ind=nnn[lay]; + clind[lay][ind]=index; } 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) 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); + delete [] kkk; + + Int_t firstLay=-1,secondLay=-1; + Int_t * end = new Int_t[AliITSgeomTGeo::GetNLayers()]; + for(Int_t i=0;iGetEntries()==0) { + end[i]=1; + } else { + end[i]=listlayer[i]->GetEntries(); + if(firstLay==-1) { + firstLay=i; + } else if(secondLay==-1) { + secondLay=i; + } + } + } + + TClonesArray* listSA = new TClonesArray("AliITStrackSA"); + TClonesArray &tri = *listSA; + Int_t nlist=0; + + for(Int_t l0=0;l0At(l0); + for(Int_t l1=0;l1At(l1); + for(Int_t l2=0;l2At(l2); + for(Int_t l3=0;l3At(l3); + for(Int_t l4=0;l4At(l4); + for(Int_t l5=0;l5At(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; + } + + 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; + } + + Int_t module1 = p1->GetDetectorIndex()+firstmod[firstLay]; + + Int_t cln1=mrk1; + Int_t cln2=mrk2; + AliITSclusterTable* arr1 = (AliITSclusterTable*)GetClusterCoord(firstLay,cln1); + AliITSclusterTable* arr2 = (AliITSclusterTable*)GetClusterCoord(secondLay,cln2); + x1 = arr1->GetX(); + x2 = arr2->GetX(); + y1 = arr1->GetY(); + y2 = arr2->GetY(); + z1 = arr1->GetZ(); + z2 = arr2->GetZ(); + sx1 = arr1->GetSx(); + sx2 = arr2->GetSx(); + sy1 = arr1->GetSy(); + sy2 = arr2->GetSy(); + sz1 = arr1->GetSz(); + sz2 = arr2->GetSz(); + + Int_t layer,ladder,detector; + AliITSgeomTGeo::GetModuleId(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); + Double_t tgl2 = (z2-z1)/TMath::Sqrt((x2-x1)*(x2-x1)+(y2-y1)*(y2-y1)); + Double_t phi2 = TMath::ATan2((y2-y1),(x2-x1)); + AliITStrackSA* trac = new AliITStrackSA(layer,ladder,detector,yclu1,zclu1,phi2,tgl2,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); - - //fit with Kalman filter using AliITStrackerV2::RefitAt() - - AliITStrackV2* ot = new AliITStrackV2(*trac); + 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); - ot->ResetCovariance(); + ot->ResetCovariance(10.); ot->ResetClusters(); - if(RefitAt(49.,ot,trac)){ //fit from layer 1 to layer 6 - - otrack2 = new AliITStrackV2(*ot); - otrack2->ResetCovariance(); + if(RefitAt(AliITSRecoParam::GetrInsideITSscreen(),ot,trac)){ //fit from layer 1 to layer 6 + AliITStrackMI *otrack2 = new AliITStrackMI(*ot); + otrack2->ResetCovariance(10.); otrack2->ResetClusters(); //fit from layer 6 to layer 1 - if(RefitAt(3.7,otrack2,ot)) fListOfTracks->AddLast(otrack2); + if(RefitAt(AliITSRecoParam::GetrInsideSPD1(),otrack2,ot)) { + fListOfTracks->AddLast(otrack2); + new (tri[nlist]) AliITStrackSA(*trac); + nlist++; + } else { + delete otrack2; + } } @@ -807,282 +837,162 @@ AliITStrackV2* AliITStrackerSA::FitTrack(AliITStrackSA* tr,Double_t *primaryVert 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++){ + for(Int_t i=0;iDelete(); + delete listSA; + delete [] firstmod; return 0; } - AliITStrackV2* otrack =(AliITStrackV2*)FindTrackLowChiSquare(fListOfTracks,dim); - - if(otrack==0) return 0; - Int_t * indexc = new Int_t[fGeom->GetNlayers()]; - for(Int_t i=0;iGetNlayers();i++) indexc[i]=0; + Int_t lowchi2 = FindTrackLowChiSquare(fListOfTracks,dim); + AliITStrackV2* otrack =(AliITStrackV2*)fListOfTracks->At(lowchi2); + AliITStrackSA* trsa = (AliITStrackSA*)listSA->At(lowchi2); + + if(otrack==0) { + for(Int_t i=0;iDelete(); + delete listSA; + delete [] firstmod; + delete [] clind; + delete [] clmark; + return 0; + } + Int_t * indexc = new Int_t[AliITSgeomTGeo::GetNLayers()]; + 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); + 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; + } } delete [] indexc; - if(otrack->GetNumberOfClusters()==(fGeom->GetNlayers()-1)){ - labl[0]=-1; - labl[1]=-1; - labl[2]=-1; - } - Int_t numberofpoints; - if(fSixPoints) numberofpoints=6; - else numberofpoints=5; - 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); - - 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) + 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;iGetNumberOfClusters();i++) + if(labl[i][0]==label || labl[i][1]==label || labl[i][2]==label) lflag++; + if(lflagGetNumberOfClusters()) label = -label; + otrack->SetLabel(label); - //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); + listSA->Delete(); + delete listSA; - if(!fTable){ - fTable = new AliITSclusterTable(fGeom,this,primaryVertex); - fTable->FillArray(fITSclusters); - fTable->FillArrayCoorAngles(); + for(Int_t i=0;iGetBranch("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,AliITSclusterTable* table){ +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 + + if(AliITSReconstructor::GetRecoParam()->GetLayersToSkip(layer)) return 0; + 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){ - + 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); - 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); + if (FindIntersection(fCoef1,fCoef2,fCoef3,-r*r,cx1,cy1,cx2,cy2)==0) + return 0; + 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 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*)table->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*)table->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; - 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->Use(); - fPhiEstimate=phi; - fPointc[0]=fTable->GetXCluster(layer,index); - fPointc[1]=fTable->GetYCluster(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 = 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(); } - delete listc; - delete [] firstmod; 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]; @@ -1091,8 +1001,6 @@ void AliITStrackerSA::UpdatePoints(){ } - - //___________________________________________________________________ Int_t AliITStrackerSA::FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2, Float_t x3, Float_t y3,Float_t& a, Float_t& b, Float_t& c){ @@ -1113,14 +1021,15 @@ Int_t AliITStrackerSA::FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t //c2 is -rlayer*rlayer if(a1==0) return 0; - Float_t m = c2-c1; - Float_t aA = (b1*b1)/(a1*a1)+1; - Float_t bB = (-2*m*b1/(a1*a1)); - Float_t cC = c2+(m*m)/(a1*a1); - if((bB*bB-4*aA*cC)<0) return 0; + Double_t m = c2-c1; + Double_t aA = (b1*b1)/(a1*a1)+1; + Double_t bB = (-2*m*b1/(a1*a1)); + Double_t cC = c2+(m*m)/(a1*a1); + Double_t dD = bB*bB-4*aA*cC; + if(dD<0) return 0; - y1 = (-bB+TMath::Sqrt(bB*bB-4*aA*cC))/(2*aA); - y2 = (-bB-TMath::Sqrt(bB*bB-4*aA*cC))/(2*aA); + y1 = (-bB+TMath::Sqrt(dD))/(2*aA); + y2 = (-bB-TMath::Sqrt(dD))/(2*aA); x1 = (c2-c1-b1*y1)/a1; x2 = (c2-c1-b1*y2)/a1; @@ -1150,6 +1059,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){ @@ -1166,11 +1077,12 @@ 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 +Int_t AliITStrackerSA::FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const { + // returns track with lowers chi square if(dim==1){ - AliITStrackV2* trk = (AliITStrackV2*)tracklist->At(0); - return trk; + //AliITStrackV2* trk = (AliITStrackV2*)tracklist->At(0); + //return trk; + return 0; } if(dim==0) return 0; Double_t * chi2 = new Double_t[dim]; @@ -1196,25 +1108,23 @@ AliITStrackV2* AliITStrackerSA::FindTrackLowChiSquare(TObjArray* tracklist, Int_ } w++; } - - AliITStrackV2* trk = (AliITStrackV2*)tracklist->At(index[dim-1]); + Int_t tmp = index[dim-1]; delete [] chi2; delete [] index; - return trk; - + return tmp; } //__________________________________________________________ -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; @@ -1284,34 +1195,89 @@ void AliITStrackerSA::SetWindowSizes(Int_t n, Double_t *phi, Double_t *lam){ } } else { // default values - - Double_t phid[32] = {0.002,0.003,0.004,0.0045,0.0047, + + Double_t phid[33] = {0.002,0.003,0.004,0.0045,0.0047, 0.005,0.0053,0.0055, 0.006,0.0063,0.0065,0.007,0.0073,0.0075,0.0077, 0.008,0.0083,0.0085,0.0087,0.009,0.0095,0.0097, - 0.01,0.0105,0.011,0.0115,0.012,0.0125,0.013,0.0135, - 0.014}; - Double_t lambdad[32] = {0.002,0.003,0.004,0.0045,0.0047, - 0.005,0.0053,0.0055, - 0.006,0.0063,0.0065,0.007,0.0073,0.0075,0.0077, - 0.008,0.0083,0.0085,0.0087,0.009,0.0095,0.0097, - 0.01,0.015,0.011,0.0115,0.012,0.0125,0.013,0.0135, - 0.014}; + 0.01,0.0105,0.011,0.0115,0.012,0.0125,0.013,0.0135,0.0140,0.0145}; + Double_t lambdad[33] = {0.003,0.004,0.005,0.005,0.005, + 0.005,0.005,0.006, + 0.006,0.006,0.006,0.007,0.007,0.007,0.007, + 0.007,0.007,0.007,0.007,0.007,0.007,0.007, + 0.008,0.008,0.008,0.008,0.008,0.008,0.008,0.008,0.008,0.008}; - if(fNloop!=32){ - fNloop = 32; + if(fNloop!=33){ + fNloop = 33; } - + + fPhiWin = new Double_t[fNloop]; fLambdaWin = new Double_t[fNloop]; - + + Double_t factor=AliITSReconstructor::GetRecoParam()->GetFactorSAWindowSizes(); // 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()); +*/ +}