X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PHOS%2FAliPHOSTrackSegmentMakerv1.cxx;h=44ca8d40c0968b459b470bc26e36abeef450834d;hb=4d4750bd504453bf48a82734d39514ce20f484b7;hp=50a1c9c1cbfb314617ce867bef3e379fad96d53a;hpb=2731cd1e25d597f63ec4341cc5b3c4b90ef1d012;p=u%2Fmrichter%2FAliRoot.git diff --git a/PHOS/AliPHOSTrackSegmentMakerv1.cxx b/PHOS/AliPHOSTrackSegmentMakerv1.cxx index 50a1c9c1cbf..44ca8d40c09 100644 --- a/PHOS/AliPHOSTrackSegmentMakerv1.cxx +++ b/PHOS/AliPHOSTrackSegmentMakerv1.cxx @@ -13,267 +13,381 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ /* $Id$ */ + +/* History of cvs commits: + * + * $Log$ + * Revision 1.78 2005/11/18 13:04:51 hristov + * Bug fix + * + * Revision 1.77 2005/11/17 23:34:36 hristov + * Corrected logics + * + * Revision 1.76 2005/11/17 22:29:12 hristov + * Faster version, no attempt to match tracks outside the PHOS acceptance + * + * Revision 1.75 2005/11/17 12:35:27 hristov + * Use references instead of objects. Avoid to create objects when they are not really needed + * + * Revision 1.74 2005/07/08 14:01:36 hristov + * Tracking in non-uniform nmagnetic field (Yu.Belikov) + * + * Revision 1.73 2005/05/28 14:19:05 schutz + * Compilation warnings fixed by T.P. + * + */ + //_________________________________________________________________________ // Implementation version 1 of algorithm class to construct PHOS track segments -// Associates EMC and PPSD clusters -// Unfolds the EMC cluster -// -//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH) +// Track segment for PHOS is list of +// EMC RecPoint + (possibly) CPV RecPoint +// To find TrackSegments we do the following: +// for each EMC RecPoints we look at +// CPV RecPoints in the radious fRcpv. +// If there is such a CPV RecPoint, +// we make "Link" it is just indexes of EMC and CPV RecPoint and distance +// between them in the PHOS plane. +// Then we sort "Links" and starting from the +// least "Link" pointing to the unassined EMC and CPV RecPoints assing them to +// new TrackSegment. +// If there is no CPV RecPoint we make TrackSegment +// consisting from EMC alone. There is no TrackSegments without EMC RecPoint. +//// In principle this class should be called from AliPHOSReconstructor, but +// one can use it as well in standalone mode. +// Use case: +// root [0] AliPHOSTrackSegmentMakerv1 * t = new AliPHOSTrackSegmentMaker("galice.root", "tracksegmentsname", "recpointsname") +// Warning in : object already instantiated +// // reads gAlice from header file "galice.root", uses recpoints stored in the branch names "recpointsname" (default = "Default") +// // and saves recpoints in branch named "tracksegmentsname" (default = "recpointsname") +// root [1] t->ExecuteTask() +// root [3] t->SetTrackSegmentsBranch("max distance 5 cm") +// root [4] t->ExecuteTask("deb all time") +// +//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH) & Yves Schutz (SUBATECH) // // --- ROOT system --- -#include "TROOT.h" -#include "TFile.h" #include "TTree.h" -#include "TSystem.h" #include "TBenchmark.h" -// --- Standard library --- - -#include -#include +// --- Standard library --- +#include "Riostream.h" // --- AliRoot header files --- - +#include "AliPHOSGeometry.h" #include "AliPHOSTrackSegmentMakerv1.h" -#include "AliPHOSClusterizerv1.h" #include "AliPHOSTrackSegment.h" -#include "AliPHOSCpvRecPoint.h" -#include "AliPHOSPpsdRecPoint.h" #include "AliPHOSLink.h" -#include "AliPHOSv0.h" -#include "AliRun.h" +#include "AliPHOSGetter.h" +#include "AliESD.h" +#include "AliESDtrack.h" ClassImp( AliPHOSTrackSegmentMakerv1) //____________________________________________________________________________ - AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() : AliPHOSTrackSegmentMaker() + AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() : AliPHOSTrackSegmentMaker() { - // ctor - SetTitle("version 1") ; - SetName("AliPHOSTrackSegmentMaker") ; - fR0 = 10. ; - fEmcFirst = 0 ; - fEmcLast = 0 ; - fCpvFirst = 0 ; - fCpvLast = 0 ; - fPpsdFirst= 0 ; - fPpsdLast = 0 ; - fLinkLowArray = 0 ; - fLinkUpArray = 0 ; - fIsInitialized = kFALSE ; + // default ctor (to be used mainly by Streamer) + + InitParameters() ; + fDefaultInit = kTRUE ; } + //____________________________________________________________________________ - AliPHOSTrackSegmentMakerv1:: AliPHOSTrackSegmentMakerv1(const char* headerFile, const char* branchTitle): AliPHOSTrackSegmentMaker() + AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString alirunFileName, const TString eventFolderName) + :AliPHOSTrackSegmentMaker(alirunFileName, eventFolderName) { // ctor - SetTitle("version 1") ; - SetName("AliPHOSTrackSegmentMaker") ; - fR0 = 10. ; - fEmcFirst = 0 ; - fEmcLast = 0 ; - fCpvFirst = 0 ; - fCpvLast = 0 ; - fPpsdFirst= 0 ; - fPpsdLast = 0 ; - - fHeaderFileName = headerFile ; - fRecPointsBranchTitle = branchTitle ; - - TFile * file = (TFile*) gROOT->GetFile(fHeaderFileName.Data() ) ; - - if(file == 0){ - file = new TFile(fHeaderFileName.Data(),"update") ; - gAlice = (AliRun *) file->Get("gAlice") ; - } - - AliPHOS * phos = (AliPHOS *) gAlice->GetDetector("PHOS") ; - fGeom = AliPHOSGeometry::GetInstance(phos->GetGeometry()->GetName(),phos->GetGeometry()->GetTitle() ); - - fEmcRecPoints = new TObjArray(200) ; - fCpvRecPoints = new TObjArray(200) ; - fClusterizer = new AliPHOSClusterizerv1() ; - - fTrackSegments = new TClonesArray("AliPHOSTrackSegment",200) ; - - fLinkLowArray = new TClonesArray("AliPHOSLink", 1000); - fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); - - fIsInitialized = kTRUE ; -} -//____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::Init(){ - - if(!fIsInitialized){ - if(fHeaderFileName.IsNull()) - fHeaderFileName = "galice.root" ; - - - TFile * file = (TFile*) gROOT->GetFile(fHeaderFileName.Data() ) ; - - if(file == 0){ - file = new TFile(fHeaderFileName.Data(),"update") ; - gAlice = (AliRun *) file->Get("gAlice") ; - } - - AliPHOS * phos = (AliPHOS *) gAlice->GetDetector("PHOS") ; - fGeom = AliPHOSGeometry::GetInstance(phos->GetGeometry()->GetName(),phos->GetGeometry()->GetTitle() ); - - - fEmcRecPoints = new TObjArray(200) ; - fCpvRecPoints = new TObjArray(200) ; - fClusterizer = new AliPHOSClusterizerv1() ; - - - fTrackSegments = new TClonesArray("AliPHOSTrackSegment",200) ; - - fLinkLowArray = new TClonesArray("AliPHOSLink", 1000); - fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); - - fIsInitialized = kTRUE ; - } + InitParameters() ; + Init() ; + fDefaultInit = kFALSE ; + fESD = 0; } //____________________________________________________________________________ AliPHOSTrackSegmentMakerv1::~AliPHOSTrackSegmentMakerv1() { // dtor - if(fLinkLowArray) delete fLinkLowArray ; - if(fLinkUpArray) delete fLinkUpArray ; + // fDefaultInit = kTRUE if TrackSegmentMaker created by default ctor (to get just the parameters) + if (!fDefaultInit) + delete fLinkUpArray ; +} + + +//____________________________________________________________________________ +const TString AliPHOSTrackSegmentMakerv1::BranchName() const +{ + + return GetName() ; } //____________________________________________________________________________ void AliPHOSTrackSegmentMakerv1::FillOneModule() { - // Finds bounds in which clusters from one PHOS module are - + // Finds first and last indexes between which + // clusters from one PHOS module are + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + + TObjArray * emcRecPoints = gime->EmcRecPoints() ; + TObjArray * cpvRecPoints = gime->CpvRecPoints() ; + //First EMC clusters - Int_t totalEmc = fEmcRecPoints->GetEntriesFast() ; + Int_t totalEmc = emcRecPoints->GetEntriesFast() ; for(fEmcFirst = fEmcLast; (fEmcLast < totalEmc) && - (((AliPHOSRecPoint *) fEmcRecPoints->At(fEmcLast))->GetPHOSMod() == fModule ); + ((dynamic_cast(emcRecPoints->At(fEmcLast)))->GetPHOSMod() == fModule ); fEmcLast ++) ; - //Now CPV clusters - Int_t totalCpv = fCpvRecPoints->GetEntriesFast() ; + Int_t totalCpv = cpvRecPoints->GetEntriesFast() ; - if(fModule <= fGeom->GetNCPVModules()){ // in CPV geometry - for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) && - (((AliPHOSRecPoint *) fCpvRecPoints->At(fCpvLast))->GetPHOSMod() == fModule ); - fCpvLast ++) ; - - fPpsdFirst = fCpvLast ; //To avoid scanning RecPoints between fPpsdFirst and fPpsdLast - fPpsdLast = fCpvLast ; //and to be ready to switch to mixed geometry - } - else{ //in PPSD geometry - fCpvLast = fPpsdLast ; - //Upper layer first - for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) && - (((AliPHOSPpsdRecPoint *) fCpvRecPoints->At(fCpvLast))->GetPHOSMod() == fModule ) && - (((AliPHOSPpsdRecPoint *) fCpvRecPoints->At(fCpvLast))->GetUp()) ; - fCpvLast ++) ; - - fPpsdLast= fCpvLast ; - for(fPpsdFirst = fPpsdLast; (fPpsdLast < totalCpv) && - (((AliPHOSPpsdRecPoint *) fCpvRecPoints->At(fPpsdLast))->GetPHOSMod() == fModule ) && - (!((AliPHOSPpsdRecPoint *) fCpvRecPoints->At(fPpsdLast))->GetUp()) ; - fPpsdLast ++) ; - } - + ((dynamic_cast(cpvRecPoints->At(fCpvLast)))->GetPHOSMod() == fModule ); + fCpvLast ++) ; + } + //____________________________________________________________________________ -Float_t AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,AliPHOSRecPoint * cpvClu, Bool_t &toofar) +Float_t AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,AliPHOSCpvRecPoint * cpvClu, Int_t &trackindex) const { - // Calculates the distance between the EMC RecPoint and the PPSD RecPoint - //clusters are sorted in "rows" and "columns" of width 1 cm - Float_t delta = 1 ; // Width of the rows in sorting of RecPoints (in cm) + // Calculates the distance between the EMC RecPoint and the CPV RecPoint + // Clusters are sorted in "rows" and "columns" of width 1 cm + + //Float_t delta = 1 ; // Width of the rows in sorting of RecPoints (in cm) // if you change this value, change it as well in xxxRecPoint::Compare() - Float_t r = fR0 ; - - TVector3 vecEmc ; - TVector3 vecCpv ; + Float_t distance2Cpv = fRcpv ; + Float_t distance2Track = fRtpc ; + + trackindex = -1 ; // closest track within fRCpv + + TVector3 vecEmc ; // Local position of EMC recpoint + TVector3 vecCpv ; // Local position of CPV recpoint propagated to EMC + TVector3 vecDist ; // Distance between local positions of two points emcClu->GetLocalPosition(vecEmc) ; - cpvClu->GetLocalPosition(vecCpv) ; + cpvClu->GetLocalPosition(vecCpv) ; + //toofar = kTRUE ; if(emcClu->GetPHOSMod() == cpvClu->GetPHOSMod()){ - if(vecCpv.X() <= vecEmc.X() + fR0 + 2*delta ){ - vecCpv = vecCpv - vecEmc ; - r = vecCpv.Mag() ; - toofar = kFALSE ; + // Find EMC-CPV distance + distance2Cpv = (vecCpv - vecEmc).Mag() ; + + if (fESD != 0x0) { + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + const AliPHOSGeometry * geom = gime->PHOSGeometry() ; + + Double_t rPHOS = geom->GetIPtoCrystalSurface(); + + //PH Acceptance boundaries for each PHOS module + Int_t nModules = geom->GetNModules(); + Double_t * thmin = new Double_t[nModules];// theta min + Double_t * thmax = new Double_t[nModules];// theta max + Double_t * phmin = new Double_t[nModules];// phi min + Double_t * phmax = new Double_t[nModules];// phi max + + for (Int_t imod=0; imodEmcModuleCoverage(imod, + thmin[imod],thmax[imod], + phmin[imod],phmax[imod]); + } + + // Extrapolate the global track direction if any to CPV and find the closest track + Int_t nTracks = fESD->GetNumberOfTracks(); + Int_t iClosestTrack = -1; + Double_t minDistance = 1e6; + Double_t pxyz[3], xyz[3]; + TVector3 inPHOS; //PH Used to calculate theta and phi + + //PH Loop on tracks + AliESDtrack *track; + for (Int_t iTrack=0; iTrackGetTrack(iTrack); + if (!track->GetXYZAt(rPHOS, fESD->GetMagneticField(), xyz)) + continue; //track coord on the cylinder of PHOS radius + if ((TMath::Abs(xyz[0])+TMath::Abs(xyz[1])+TMath::Abs(xyz[2]))<=0) + continue; + //PH Here one has to cut out the tracks which are not inside the PHOS + //PH acceptance + inPHOS.SetXYZ(xyz[0],xyz[1],xyz[2]); + Double_t inPhi = inPHOS.Phi(); + Double_t inTheta = inPHOS.Theta(); + + Bool_t skip = kTRUE; + for (Int_t imod=0; imod inTheta && + phmin[imod] < inPhi && phmax[imod] > inPhi) { + skip = kFALSE; + break; + } + } + if (skip) continue; //PH Skip, if not in the PHOS acceptance + + if (!track->GetPxPyPzAt(rPHOS, fESD->GetMagneticField(), pxyz)) + continue; // track momentum ibid. + PropagateToPlane(vecDist,xyz,pxyz,"CPV",cpvClu->GetPHOSMod()); + // Info("GetDistanceInPHOSPlane","Track %d propagation to CPV = (%f,%f,%f)", + // iTrack,vecDist.X(),vecDist.Y(),vecDist.Z()); + vecDist -= vecCpv; + distance2Track = TMath::Sqrt(vecDist.X()*vecDist.X() + vecDist.Z()*vecDist.Z()); + // Find the closest track to the EMC recpoint + if (distance2Track < minDistance) { + minDistance = distance2Track; + iClosestTrack = iTrack; + } + } + + delete [] thmin; + delete [] thmax; + delete [] phmin; + delete [] phmax; + + if (iClosestTrack != -1) { + track = fESD->GetTrack(iClosestTrack); + if (track->GetPxPyPzAt(rPHOS, fESD->GetMagneticField(), pxyz)) { // track momentum ibid. + TVector3 vecCpvGlobal; // Global position of the CPV recpoint + geom->GetGlobal((AliRecPoint*)cpvClu,vecCpvGlobal); + for (Int_t ixyz=0; ixyz<3; ixyz++) + xyz[ixyz] = vecCpvGlobal[ixyz]; + PropagateToPlane(vecDist,xyz,pxyz,"EMC",cpvClu->GetPHOSMod()); +// Info("GetDistanceInPHOSPlane","Track %d propagation to EMC = (%f,%f,%f)", +// iClosestTrack,vecDist.X(),vecDist.Y(),vecDist.Z()); + vecDist -= vecEmc; + distance2Track = TMath::Sqrt(vecDist.X()*vecDist.X() + vecDist.Z()*vecDist.Z()); + } + } +// } else { +// // If no ESD exists, than simply find EMC-CPV distance +// distance = (vecCpv - vecEmc).Mag() ; + + //if(distance2Track < fRcpv + 2*delta ) + if(distance2Track < fRtpc ) + trackindex = iClosestTrack ; + // toofar = kFALSE ; + } + // Info("GetDistanceInPHOSPlane","cpv-emc distance is %f cm", + // distance); + } + + return distance2Cpv ; +} - } // if xPpsd >= xEmc + ... - else - toofar = kTRUE ; - } - else - toofar = kTRUE ; +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::PropagateToPlane(TVector3& globalIntersection, + Double_t *x, + Double_t *p, + const char *det, + Int_t moduleNumber) const +{ + // Propagate a straight-line track from the origin point x + // along the direction p to the CPV or EMC module moduleNumber + // Returns a local position of such a propagation + + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + const AliPHOSGeometry * geom = gime->PHOSGeometry() ; + TVector3 moduleCenter; + geom->GetModuleCenter(moduleCenter,det,moduleNumber); + TVector3 vertex; vertex.SetXYZ(x[0],x[1],x[2]); + TVector3 direction; direction.SetXYZ(p[0],p[1],p[2]); + +// Info("PropagateToCPV","Center of the %s module %d is (%f,%f,%f)", +// det,moduleNumber,moduleCenter[0],moduleCenter[1],moduleCenter[2]); + + Double_t time = (moduleCenter.Mag2() - vertex.Dot(moduleCenter)) / + (direction.Dot(moduleCenter)); + vertex += direction*time; + geom->Global2Local(globalIntersection,vertex,moduleNumber); +} - //toofar = kFALSE ; - +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::Init() +{ + // Make all memory allocations that are not possible in default constructor + + AliPHOSGetter* gime = AliPHOSGetter::Instance(); + if(!gime) + gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data()); - return r ; + fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); + if ( !gime->TrackSegmentMaker() ) { + gime->PostTrackSegmentMaker(this); + } +} + +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::InitParameters() +{ + //Initializes parameters + fRcpv = 10. ; + fRtpc = 4. ; + fEmcFirst = 0 ; + fEmcLast = 0 ; + fCpvFirst = 0 ; + fCpvLast = 0 ; + fLinkUpArray = 0 ; + fWrite = kTRUE ; + fTrackSegmentsInRun = 0 ; + SetEventRange(0,-1) ; } + //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::MakeLinks() +void AliPHOSTrackSegmentMakerv1::MakeLinks()const { - // Finds distances (links) between all EMC and PPSD clusters, which are not further apart from each other than fR0 + // Finds distances (links) between all EMC and CPV clusters, + // which are not further apart from each other than fRcpv + // and sort them in accordance with this distance + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + TObjArray * emcRecPoints = gime->EmcRecPoints() ; + TObjArray * cpvRecPoints = gime->CpvRecPoints() ; + fLinkUpArray->Clear() ; - fLinkLowArray->Clear() ; - AliPHOSRecPoint * ppsd ; - AliPHOSRecPoint * cpv ; + AliPHOSCpvRecPoint * cpv ; AliPHOSEmcRecPoint * emcclu ; - Int_t iLinkLow = 0 ; Int_t iLinkUp = 0 ; Int_t iEmcRP; for(iEmcRP = fEmcFirst; iEmcRP < fEmcLast; iEmcRP++ ) { - emcclu = (AliPHOSEmcRecPoint *) fEmcRecPoints->At(iEmcRP) ; - - Bool_t toofar ; - Int_t iPpsd ; - for(iPpsd = fPpsdFirst; iPpsd < fPpsdLast;iPpsd++ ) { - - ppsd = (AliPHOSRecPoint *) fCpvRecPoints->At(iPpsd) ; - Float_t r = GetDistanceInPHOSPlane(emcclu, ppsd, toofar) ; + emcclu = dynamic_cast(emcRecPoints->At(iEmcRP)) ; - if(toofar) - break ; - if(r < fR0) - new ((*fLinkLowArray)[iLinkLow++]) AliPHOSLink(r, iEmcRP, iPpsd) ; - } - + //Bool_t toofar ; Int_t iCpv = 0 ; for(iCpv = fCpvFirst; iCpv < fCpvLast;iCpv++ ) { - cpv = (AliPHOSRecPoint *) fCpvRecPoints->At(iCpv) ; - Float_t r = GetDistanceInPHOSPlane(emcclu, cpv, toofar) ; - - if(toofar) - break ; - if(r < fR0) { - new ((*fLinkUpArray)[iLinkUp++]) AliPHOSLink(r, iEmcRP, iCpv) ; + cpv = dynamic_cast(cpvRecPoints->At(iCpv)) ; + Int_t track = -1 ; + Float_t r = GetDistanceInPHOSPlane(emcclu, cpv, track) ; + // if(toofar) + // continue ; + if(r < fRcpv) { + new ((*fLinkUpArray)[iLinkUp++]) AliPHOSLink(r, iEmcRP, iCpv, track) ; } } } - fLinkLowArray->Sort() ; //first links with smallest distances - fLinkUpArray->Sort() ; + fLinkUpArray->Sort() ; //first links with smallest distances } //____________________________________________________________________________ void AliPHOSTrackSegmentMakerv1::MakePairs() { - - //Make arrays to mark clusters already chousen + // Using the previously made list of "links", we found the smallest link - i.e. + // link with the least distance between EMC and CPV and pointing to still + // unassigned RecParticles. We assign these RecPoints to TrackSegment and + // remove them from the list of "unassigned". + + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + + TObjArray * emcRecPoints = gime->EmcRecPoints() ; + TObjArray * cpvRecPoints = gime->CpvRecPoints() ; + TClonesArray * trackSegments = gime->TrackSegments(); + + //Make arrays to mark clusters already chosen Int_t * emcExist = 0; if(fEmcLast > fEmcFirst) emcExist = new Int_t[fEmcLast-fEmcFirst] ; @@ -288,340 +402,190 @@ void AliPHOSTrackSegmentMakerv1::MakePairs() for(index = 0; index fPpsdFirst) - ppsdExist = new Bool_t[fPpsdLast-fPpsdFirst] ; - for(index = 0; index GetEmc()-fEmcFirst]> 0) && ppsdExist[linkLow->GetPpsd()-fPpsdFirst] ){ // RecPoints not removed yet - new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fEmcRecPoints->At(linkLow->GetEmc()), - nullpointer, - (AliPHOSPpsdRecPoint *)fCpvRecPoints->At(linkLow->GetPpsd()) ) ; - - ((AliPHOSTrackSegment* )fTrackSegments->At(fNTrackSegments))->SetIndexInList(fNTrackSegments); - //replace index of emc to negative and shifted index of TS - emcExist[linkLow->GetEmc()-fEmcFirst] = -2 - fNTrackSegments ; - //mark ppsd as used - ppsdExist[linkLow->GetPpsd()-fPpsdFirst] = kFALSE ; - fNTrackSegments++ ; - } - } - - - while ( (linkUp = (AliPHOSLink *)nextUp() ) ){ - if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){ //without ppsd Up yet - - if(cpvExist[linkUp->GetPpsd()-fCpvFirst]){ //CPV still exist - - if(emcExist[linkUp->GetEmc()-fEmcFirst] > 0){ //without ppsd Low => create new TS + AliPHOSCpvRecPoint * nullpointer = 0 ; + + while ( (linkUp = static_cast(nextUp()) ) ){ - new ((* fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fEmcRecPoints->At(linkUp->GetEmc()) , - (AliPHOSPpsdRecPoint *)fCpvRecPoints->At(linkUp->GetPpsd()), - nullpointer) ; - ((AliPHOSTrackSegment *) fTrackSegments->At(fNTrackSegments))->SetIndexInList(fNTrackSegments); - fNTrackSegments++ ; - } - else{ // append ppsd Up to existing TS - ((AliPHOSTrackSegment *)fTrackSegments->At(-2-emcExist[linkUp->GetEmc()-fEmcFirst]))->SetCpvRecPoint((AliPHOSCpvRecPoint *)fCpvRecPoints->At(linkUp->GetPpsd())); - } + if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){ - emcExist[linkUp->GetEmc()-fEmcFirst] = -1 ; //Mark emc that Cpv was found - //mark CPV recpoint as already used - cpvExist[linkUp->GetPpsd()-fCpvFirst] = kFALSE ; - } //if ppsdUp still exist + if(cpvExist[linkUp->GetCpv()-fCpvFirst]){ //CPV still exist + new ((* trackSegments)[fNTrackSegments]) + AliPHOSTrackSegment(dynamic_cast(emcRecPoints->At(linkUp->GetEmc())) , + dynamic_cast(cpvRecPoints->At(linkUp->GetCpv())) , + linkUp->GetTrack()) ; + + (dynamic_cast(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); + fNTrackSegments++ ; + emcExist[linkUp->GetEmc()-fEmcFirst] = -1 ; //Mark emc that Cpv was found + //mark CPV recpoint as already used + cpvExist[linkUp->GetCpv()-fCpvFirst] = kFALSE ; + } //if CpvUp still exist } - } + } - //look through emc recPoints left without CPV/PPSD + //look through emc recPoints left without CPV if(emcExist){ //if there is emc rec point Int_t iEmcRP ; for(iEmcRP = 0; iEmcRP < fEmcLast-fEmcFirst ; iEmcRP++ ){ if(emcExist[iEmcRP] > 0 ){ - new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *)fEmcRecPoints->At(iEmcRP+fEmcFirst), - nullpointer, - nullpointer ) ; - ((AliPHOSTrackSegment *) fTrackSegments->At(fNTrackSegments))->SetIndexInList(fNTrackSegments); - fNTrackSegments++; + new ((*trackSegments)[fNTrackSegments]) + AliPHOSTrackSegment(dynamic_cast(emcRecPoints->At(iEmcRP+fEmcFirst)), + nullpointer) ; + (dynamic_cast(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); + fNTrackSegments++; } } } - + delete [] emcExist ; + delete [] cpvExist ; } //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::Exec(Option_t * option) +void AliPHOSTrackSegmentMakerv1::Exec(Option_t *option) { - // Makes the track segments out of the list of EMC and PPSD Recpoints and stores them in a list - - if(! fIsInitialized) Init() ; - + // Steering method to perform track segment construction for events + // in the range from fFirstEvent to fLastEvent. + // This range is optionally set by SetEventRange(). + // if fLastEvent=-1 (by default), then process events until the end. + if(strstr(option,"tim")) - gBenchmark->Start("PHOSTSMaker"); - - Int_t nEvents = (Int_t) gAlice->TreeE()->GetEntries() ; + gBenchmark->Start("PHOSTSMaker"); + + if(strstr(option,"print")) { + Print() ; + return ; + } - for(fEvent = 0;fEvent< nEvents; fEvent++){ - if(!ReadRecPoints()) //reads RecPoints for event fEvent - return; + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + + const AliPHOSGeometry * geom = gime->PHOSGeometry() ; + + if (fLastEvent == -1) + fLastEvent = gime->MaxEvent() - 1 ; + else + fLastEvent = TMath::Min(fFirstEvent,gime->MaxEvent()); + Int_t nEvents = fLastEvent - fFirstEvent + 1; + + Int_t ievent ; + for (ievent = fFirstEvent; ievent <= fLastEvent; ievent++) { + gime->Event(ievent,"R") ; + //Make some initializations + fNTrackSegments = 0 ; + fEmcFirst = 0 ; + fEmcLast = 0 ; + fCpvFirst = 0 ; + fCpvLast = 0 ; - for(fModule = 1; fModule <= fGeom->GetNModules() ; fModule++ ){ - + gime->TrackSegments()->Clear(); + + // if(!ReadRecPoints(ievent)) continue; //reads RecPoints for event ievent + + for(fModule = 1; fModule <= geom->GetNModules() ; fModule++ ) { FillOneModule() ; - MakeLinks() ; - MakePairs() ; - } WriteTrackSegments() ; + if(strstr(option,"deb")) - PrintTrackSegments(option) ; + PrintTrackSegments(option); + + //increment the total number of track segments per run + fTrackSegmentsInRun += gime->TrackSegments()->GetEntriesFast() ; } - + if(strstr(option,"tim")){ gBenchmark->Stop("PHOSTSMaker"); - cout << "AliPHOSTSMaker:" << endl ; - cout << " took " << gBenchmark->GetCpuTime("PHOSTSMaker") << " seconds for making TS " - << gBenchmark->GetCpuTime("PHOSTSMaker")/nEvents << " seconds per event " << endl ; - cout << endl ; - } - + Info("Exec", "took %f seconds for making TS %f seconds per event", + gBenchmark->GetCpuTime("PHOSTSMaker"), + gBenchmark->GetCpuTime("PHOSTSMaker")/nEvents) ; + } + if(fWrite) //do not unload in "on flight" mode + Unload(); +} +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::Unload() +{ + // Unloads the task from the folder + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + gime->PhosLoader()->UnloadRecPoints() ; + gime->PhosLoader()->UnloadTracks() ; } + //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::Print(Option_t * option)const { - if(fIsInitialized){ - cout << "======== AliPHOSTrackSegmentMakerv1 ========" << endl ; - cout << "Making Track segments "<< endl ; - cout << " Headers file: " << fHeaderFileName.Data() << endl ; - cout << " RecPoints branch file name: " <Clear() ; - fCpvRecPoints->Clear() ; - fTrackSegments->Clear() ; - fNTrackSegments = 0 ; - fEmcFirst = 0 ; - fEmcLast = 0 ; - fCpvFirst = 0 ; - fCpvLast = 0 ; - fPpsdFirst= 0 ; - fPpsdLast = 0 ; - - - gAlice->GetEvent(fEvent) ; - - // Get TreeR header from file - char treeName[20]; - sprintf(treeName,"TreeR%d",fEvent); - - if(gAlice->TreeR()==0){ - cout << "Error in AliPHOSTrackSegmentMakerv1 : no "<TreeR()->GetListOfBranches() ; - Int_t ibranch; - Bool_t emcNotFound = kTRUE ; - Bool_t cpvNotFound = kTRUE ; - Bool_t clusterizerNotFound = kTRUE ; - - for(ibranch = 0;ibranch GetEntries();ibranch++){ + AliPHOSGetter *gime = AliPHOSGetter::Instance() ; - if(emcNotFound){ - emcBranch=(TBranch *) branches->At(ibranch) ; - if( fRecPointsBranchTitle.CompareTo(emcBranch->GetTitle())==0 ) - if( strcmp(emcBranch->GetName(),emcBranchName) == 0) { - emcNotFound = kFALSE ; - } - } - - if(cpvNotFound){ - cpvBranch=(TBranch *) branches->At(ibranch) ; - if( fRecPointsBranchTitle.CompareTo(cpvBranch->GetTitle())==0 ) - if( strcmp(cpvBranch->GetName(),cpvBranchName) == 0) - cpvNotFound = kFALSE ; - } + TClonesArray * trackSegments = gime->TrackSegments() ; + trackSegments->Expand(trackSegments->GetEntriesFast()) ; + + if(fWrite){ //We write TreeT + TTree * treeT = gime->TreeT(); - if(clusterizerNotFound){ - clusterizerBranch = (TBranch *) branches->At(ibranch) ; - if( fRecPointsBranchTitle.CompareTo(clusterizerBranch->GetTitle()) == 0) - if( strcmp(clusterizerBranch->GetName(),cluBranchName) == 0) - clusterizerNotFound = kFALSE ; - } + //First TS + Int_t bufferSize = 32000 ; + TBranch * tsBranch = treeT->Branch("PHOSTS",&trackSegments,bufferSize); + tsBranch->Fill() ; + gime->WriteTracks("OVERWRITE"); + gime->WriteTrackSegmentMaker("OVERWRITE"); } - - if(clusterizerNotFound || emcNotFound || cpvNotFound){ - cout << "AliPHOSTrackSegmentMakerv1: " << endl ; - cout << " Can't find Branch with RecPoints or Clusterizer " ; - cout << " Do nothing" <SetAddress(&fEmcRecPoints) ; - cpvBranch->SetAddress(&fCpvRecPoints) ; - clusterizerBranch->SetAddress(&fClusterizer) ; - - gAlice->TreeR()->GetEvent(0) ; - - delete emcBranchName; - delete cpvBranchName; - delete cluBranchName; - - return kTRUE ; - } -//____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::WriteTrackSegments(){ - - char treeName[20]; - sprintf(treeName,"TreeR%d",fEvent); - - - //First, check, if branches already exist - TBranch * tsMakerBranch = 0; - TBranch * tsBranch = 0; - - TObjArray * branches = gAlice->TreeR()->GetListOfBranches() ; - Int_t ibranch; - Bool_t tsMakerNotFound = kTRUE ; - Bool_t tsNotFound = kTRUE ; - - for(ibranch = 0;(ibranch GetEntries())&&(tsMakerNotFound||tsNotFound);ibranch++){ - if(tsMakerNotFound){ - tsMakerBranch=(TBranch *) branches->At(ibranch) ; - if( (strcmp(tsMakerBranch->GetName(),"AliPHOSTrackSegmentMaker") == 0) && - (fTSBranchTitle.CompareTo( tsMakerBranch->GetTitle())==0 )) - tsMakerNotFound = kFALSE ; - } - if(tsNotFound){ - tsBranch=(TBranch *) branches->At(ibranch) ; - if( (strcmp(tsBranch->GetName(),"PHOSTS") == 0) && - (fTSBranchTitle.CompareTo( tsBranch->GetTitle())==0 )) - tsNotFound = kFALSE ; - } - } - if(!(tsMakerNotFound && tsNotFound )){ - cout << "AliPHOSTrackSegmentMakerv1 error:"<< endl ; - cout << " Branches PHOSTS and AliPHOSTrackSegementMaker " << endl ; - cout << " with title '"<Getenv("CONFIG_SPLIT_FILE")!=0){ //generating file name - filename = new char[strlen(gAlice->GetBaseFile())+20] ; - sprintf(filename,"%s/PHOS.Reco.root",gAlice->GetBaseFile()) ; - } - TDirectory *cwd = gDirectory; - - //First TS - Int_t bufferSize = 32000 ; - tsBranch = gAlice->TreeR()->Branch("PHOSTS",&fTrackSegments,bufferSize); - tsBranch->SetTitle(fTSBranchTitle.Data()); - if (filename) { - tsBranch->SetFile(filename); - TIter next( tsBranch->GetListOfBranches()); - while ((tsBranch=(TBranch*)next())) { - tsBranch->SetFile(filename); - } - cwd->cd(); - } - - //Second -TSMaker - Int_t splitlevel = 0 ; - AliPHOSTrackSegmentMakerv1 * ts = this ; - tsMakerBranch = gAlice->TreeR()->Branch("AliPHOSTrackSegmentMaker","AliPHOSTrackSegmentMakerv1", - &ts,bufferSize,splitlevel); - tsMakerBranch->SetTitle(fTSBranchTitle.Data()); - if (filename) { - tsMakerBranch->SetFile(filename); - TIter next( tsMakerBranch->GetListOfBranches()); - while ((tsMakerBranch=(TBranch*)next())) { - tsMakerBranch->SetFile(filename); - } - cwd->cd(); - } - - gAlice->TreeR()->Fill() ; - gAlice->TreeR()->Write(0,kOverwrite) ; - -} +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option) +{ + // option deb - prints # of found TrackSegments + // option deb all - prints as well indexed of found RecParticles assigned to the TS + TClonesArray * trackSegments = AliPHOSGetter::Instance()->TrackSegments() ; -//____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option){ - - cout << "AliPHOSTrackSegmentMakerv1: " << endl ; - cout << " Found " << fTrackSegments->GetEntriesFast() << " trackSegments " << endl ; + Info("PrintTrackSegments", "Results from TrackSegmentMaker:") ; + printf("nevent: %d\n", gAlice->GetEvNumber()) ; + printf(" Found %d TrackSegments\n", trackSegments->GetEntriesFast() ); if(strstr(option,"all")) { // printing found TS - cout << "TrackSegment # " << " EMC RP# " << " CPV RP# " << " PPSD RP#" << endl ; - + printf("TrackSegment # EMC RP# CPV RP#\n") ; Int_t index; - for (index = 0 ; index GetEntriesFast() ; index++) { - AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )fTrackSegments->At(index) ; - cout<<" "<< setw(4) << ts->GetIndexInList() << " " - <GetEmcIndex()<< " " - <GetCpvIndex()<< " " - <GetPpsdIndex()<< endl ; + for (index = 0 ; index GetEntriesFast() ; index++) { + AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ; + printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ; } - - cout << "-------------------------------------------------------"<< endl ; } } -//____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::SetRecPointsBranch(const char * title){ - //set the title of RecPoints - fRecPointsBranchTitle = title ; - -} -//____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::SetTrackSegmentsBranch(const char * title){ - - fTSBranchTitle = title ; -}