X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PHOS%2FAliPHOSTrackSegmentMakerv1.cxx;h=44ca8d40c0968b459b470bc26e36abeef450834d;hb=7e6ee5e39d1f413890d7227279bc5da9fda26849;hp=949dc0590d7d0a3796ae38c638083aa54634dd84;hpb=fc12304fd20b02537e3298b43f8d21e97eb34a41;p=u%2Fmrichter%2FAliRoot.git diff --git a/PHOS/AliPHOSTrackSegmentMakerv1.cxx b/PHOS/AliPHOSTrackSegmentMakerv1.cxx index 949dc0590d7..44ca8d40c09 100644 --- a/PHOS/AliPHOSTrackSegmentMakerv1.cxx +++ b/PHOS/AliPHOSTrackSegmentMakerv1.cxx @@ -13,23 +13,46 @@ * 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 // Track segment for PHOS is list of -// EMC RecPoint + (possibly) CPV RecPoint + (possibly) PPSD RecPoint +// EMC RecPoint + (possibly) CPV RecPoint // To find TrackSegments we do the following: // for each EMC RecPoints we look at -// CPV/PPSD RecPoints in the radious fR0. +// CPV RecPoints in the radious fRcpv. // If there is such a CPV RecPoint, -// we make "Link" it is just indexes of EMC and CPV/PPSD RecPoint and distance +// 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/PPSD RecPoint we make 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 AliPHOSReconstructioner, but +//// 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") @@ -37,7 +60,6 @@ // // 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 [2] t->SetMaxEmcPpsdDistance(5) // root [3] t->SetTrackSegmentsBranch("max distance 5 cm") // root [4] t->ExecuteTask("deb all time") // @@ -45,27 +67,19 @@ // // --- ROOT system --- -#include "TROOT.h" -#include "TFile.h" -#include "TFolder.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 "AliPHOSLink.h" #include "AliPHOSGetter.h" -#include "AliPHOS.h" -#include "AliRun.h" +#include "AliESD.h" +#include "AliESDtrack.h" ClassImp( AliPHOSTrackSegmentMakerv1) @@ -75,62 +89,37 @@ ClassImp( AliPHOSTrackSegmentMakerv1) { // default ctor (to be used mainly by Streamer) - fR0 = 10. ; - fEmcFirst = 0 ; - fEmcLast = 0 ; - fCpvFirst = 0 ; - fCpvLast = 0 ; - fLinkUpArray = 0 ; - fHeaderFileName = "" ; - fRecPointsBranchTitle = "" ; - fTrackSegmentsBranchTitle = "" ; - fFrom = "" ; - - fTrackSegmentsInRun = 0 ; + InitParameters() ; + fDefaultInit = kTRUE ; } //____________________________________________________________________________ - AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const char * headerFile, const char * name, const char * from) : AliPHOSTrackSegmentMaker(headerFile, name) + AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString alirunFileName, const TString eventFolderName) + :AliPHOSTrackSegmentMaker(alirunFileName, eventFolderName) { // ctor - fR0 = 10. ; - fEmcFirst = 0 ; - fEmcLast = 0 ; - fCpvFirst = 0 ; - fCpvLast = 0 ; - fLinkUpArray = 0 ; - - fHeaderFileName = GetTitle() ; - fRecPointsBranchTitle = GetName() ; - fTrackSegmentsBranchTitle = GetName() ; - fTrackSegmentsInRun = 0 ; - - TString tsmName( GetName()) ; - tsmName.Append(":") ; - tsmName.Append(Version()) ; - SetName(tsmName) ; - if ( from == 0 ) - fFrom = name ; - else - fFrom = from ; + InitParameters() ; Init() ; - + fDefaultInit = kFALSE ; + fESD = 0; } //____________________________________________________________________________ AliPHOSTrackSegmentMakerv1::~AliPHOSTrackSegmentMakerv1() { // dtor - delete fLinkUpArray ; + // fDefaultInit = kTRUE if TrackSegmentMaker created by default ctor (to get just the parameters) + if (!fDefaultInit) + delete fLinkUpArray ; } + //____________________________________________________________________________ const TString AliPHOSTrackSegmentMakerv1::BranchName() const { - TString branchName(GetName() ) ; - branchName.Remove(branchName.Index(Version())-1) ; - return branchName ; + + return GetName() ; } //____________________________________________________________________________ @@ -138,10 +127,11 @@ void AliPHOSTrackSegmentMakerv1::FillOneModule() { // Finds first and last indexes between which // clusters from one PHOS module are + + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; - TObjArray * emcRecPoints = gime->EmcRecPoints(fFrom) ; - TObjArray * cpvRecPoints = gime->CpvRecPoints(fFrom) ; + TObjArray * emcRecPoints = gime->EmcRecPoints() ; + TObjArray * cpvRecPoints = gime->CpvRecPoints() ; //First EMC clusters Int_t totalEmc = emcRecPoints->GetEntriesFast() ; @@ -153,45 +143,162 @@ void AliPHOSTrackSegmentMakerv1::FillOneModule() Int_t totalCpv = cpvRecPoints->GetEntriesFast() ; for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) && - ((dynamic_cast(cpvRecPoints->At(fCpvLast)))->GetPHOSMod() == fModule ); - fCpvLast ++) ; + ((dynamic_cast(cpvRecPoints->At(fCpvLast)))->GetPHOSMod() == fModule ); + fCpvLast ++) ; } //____________________________________________________________________________ -Float_t AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,AliPHOSRecPoint * cpvClu, Bool_t &toofar)const +Float_t AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,AliPHOSCpvRecPoint * cpvClu, Int_t &trackindex) const { - // Calculates the distance between the EMC RecPoint and the PPSD RecPoint + // 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) + //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 - //toofar = kFALSE ; - - - return r ; + 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); } //____________________________________________________________________________ @@ -199,39 +306,47 @@ void AliPHOSTrackSegmentMakerv1::Init() { // Make all memory allocations that are not possible in default constructor - if ( strcmp(GetTitle(), "") == 0 ) - SetTitle("galice.root") ; - - AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), fFrom.Data()) ; - if ( gime == 0 ) { - cerr << "ERROR: AliPHOSTrackSegmentMakerv1::Init -> Could not obtain the Getter object !" << endl ; - return ; - } + AliPHOSGetter* gime = AliPHOSGetter::Instance(); + if(!gime) + gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data()); fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); - - //add Task to //YSAlice/tasks/Reconstructioner/PHOS - gime->PostTrackSegmentMaker(this) ; - - // create a folder on the white board //YSAlice/WhiteBoard/RecPoints/PHOS/trackSegmentsName - gime->PostTrackSegments(BranchName()) ; + 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()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::GetInstance() ; - TObjArray * emcRecPoints = gime->EmcRecPoints(fFrom) ; - TObjArray * cpvRecPoints = gime->CpvRecPoints(fFrom) ; + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + TObjArray * emcRecPoints = gime->EmcRecPoints() ; + TObjArray * cpvRecPoints = gime->CpvRecPoints() ; fLinkUpArray->Clear() ; - AliPHOSRecPoint * cpv ; + AliPHOSCpvRecPoint * cpv ; AliPHOSEmcRecPoint * emcclu ; Int_t iLinkUp = 0 ; @@ -240,17 +355,17 @@ void AliPHOSTrackSegmentMakerv1::MakeLinks()const for(iEmcRP = fEmcFirst; iEmcRP < fEmcLast; iEmcRP++ ) { emcclu = dynamic_cast(emcRecPoints->At(iEmcRP)) ; - Bool_t toofar ; + //Bool_t toofar ; Int_t iCpv = 0 ; for(iCpv = fCpvFirst; iCpv < fCpvLast;iCpv++ ) { - cpv = dynamic_cast(cpvRecPoints->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) ; } } } @@ -265,11 +380,12 @@ void AliPHOSTrackSegmentMakerv1::MakePairs() // 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::GetInstance() ; - TObjArray * emcRecPoints = gime->EmcRecPoints(fFrom) ; - TObjArray * cpvRecPoints = gime->CpvRecPoints(fFrom) ; - TClonesArray * trackSegments = gime->TrackSegments(BranchName()) ; + + 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; @@ -292,37 +408,37 @@ void AliPHOSTrackSegmentMakerv1::MakePairs() AliPHOSLink * linkUp ; - AliPHOSRecPoint * nullpointer = 0 ; + AliPHOSCpvRecPoint * nullpointer = 0 ; while ( (linkUp = static_cast(nextUp()) ) ){ - if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){ //without ppsd Up yet - - if(cpvExist[linkUp->GetPpsd()-fCpvFirst]){ //CPV still exist - - new ((* trackSegments)[fNTrackSegments]) - AliPHOSTrackSegment(dynamic_cast(emcRecPoints->At(linkUp->GetEmc())) , - dynamic_cast(cpvRecPoints->At(linkUp->GetPpsd()))) ; - (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->GetPpsd()-fCpvFirst] = kFALSE ; - } //if ppsdUp still exist + if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){ + + 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 ((*trackSegments)[fNTrackSegments]) - AliPHOSTrackSegment(dynamic_cast(emcRecPoints->At(iEmcRP+fEmcFirst)), - nullpointer) ; - (dynamic_cast(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); - fNTrackSegments++; + new ((*trackSegments)[fNTrackSegments]) + AliPHOSTrackSegment(dynamic_cast(emcRecPoints->At(iEmcRP+fEmcFirst)), + nullpointer) ; + (dynamic_cast(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); + fNTrackSegments++; } } } @@ -331,114 +447,99 @@ void AliPHOSTrackSegmentMakerv1::MakePairs() } //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::Exec(Option_t * option) +void AliPHOSTrackSegmentMakerv1::Exec(Option_t *option) { - // STEERing method - - if( strcmp(GetName(), "")== 0 ) - 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"); if(strstr(option,"print")) { - Print("") ; + Print() ; return ; } - - gAlice->GetEvent(0) ; - //check, if the branch with name of this" already exits? - TObjArray * lob = static_cast(gAlice->TreeR()->GetListOfBranches()) ; - TIter next(lob) ; - TBranch * branch = 0 ; - Bool_t phostsfound = kFALSE, tracksegmentmakerfound = kFALSE ; - TString branchname = GetName() ; - branchname.Remove(branchname.Index(Version())-1) ; - - while ( (branch = static_cast(next())) && (!phostsfound || !tracksegmentmakerfound) ) { - if ( (strcmp(branch->GetName(), "PHOSTS")==0) && (strcmp(branch->GetTitle(), branchname.Data())==0) ) - phostsfound = kTRUE ; - - else if ( (strcmp(branch->GetName(), "AliPHOSTrackSegmentMaker")==0) && (strcmp(branch->GetTitle(), GetName())==0) ) - tracksegmentmakerfound = kTRUE ; - } - - if ( phostsfound || tracksegmentmakerfound ) { - cerr << "WARNING: AliPHOSTrackSegmentMakerv1::Exec -> TrackSegments and/or TrackSegmentMaker branch with name " - << branchname.Data() << " already exits" << endl ; - return ; - } - - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + const AliPHOSGeometry * geom = gime->PHOSGeometry() ; - Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ; - Int_t ievent ; - for(ievent = 0; ievent < nevents; ievent++){ + 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 + //Make some initializations fNTrackSegments = 0 ; fEmcFirst = 0 ; fEmcLast = 0 ; fCpvFirst = 0 ; fCpvLast = 0 ; - gime->TrackSegments(BranchName())->Clear() ; + + gime->TrackSegments()->Clear(); // if(!ReadRecPoints(ievent)) continue; //reads RecPoints for event ievent - for(fModule = 1; fModule <= geom->GetNModules() ; fModule++ ){ - + for(fModule = 1; fModule <= geom->GetNModules() ; fModule++ ) { FillOneModule() ; - MakeLinks() ; - MakePairs() ; - } - WriteTrackSegments(ievent) ; + WriteTrackSegments() ; if(strstr(option,"deb")) - PrintTrackSegments(option) ; + PrintTrackSegments(option); //increment the total number of track segments per run - fTrackSegmentsInRun += gime->TrackSegments(BranchName())->GetEntriesFast() ; - + 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::Print(Option_t * option)const +void AliPHOSTrackSegmentMakerv1::Unload() +{ + // Unloads the task from the folder + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + gime->PhosLoader()->UnloadRecPoints() ; + gime->PhosLoader()->UnloadTracks() ; +} + +//____________________________________________________________________________ +void AliPHOSTrackSegmentMakerv1::Print(const Option_t *)const { // Print TrackSegmentMaker parameters + TString message("") ; if( strcmp(GetName(), "") != 0 ) { - cout << "======== AliPHOSTrackSegmentMakerv1 ========" << endl ; - cout << "Making Track segments "<< endl ; - cout << " Headers file: " << fHeaderFileName.Data() << endl ; - cout << " RecPoints branch file name: " << fRecPointsBranchTitle.Data() << endl ; - cout << " TrackSegments Branch file name: " << fTrackSegmentsBranchTitle.Data() << endl ; - cout << "with parameters: " << endl ; - cout << " Maximal EMC - CPV (PPSD) distance (cm)" << fR0 << endl ; - cout << "============================================" << endl ; + message = "\n======== AliPHOSTrackSegmentMakerv1 ========\n" ; + message += "Making Track segments\n" ; + message += "with parameters:\n" ; + message += " Maximal EMC - CPV distance (cm) %f\n" ; + message += "============================================\n" ; + Info("Print", message.Data(),fRcpv) ; } else - cout << "AliPHOSTrackSegmentMakerv1 not initialized " << endl ; + Info("Print", "AliPHOSTrackSegmentMakerv1 not initialized ") ; } //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::WriteTrackSegments(Int_t event) +void AliPHOSTrackSegmentMakerv1::WriteTrackSegments() { // Writes found TrackSegments to TreeR. Creates branches // "PHOSTS" and "AliPHOSTrackSegmentMaker" with the same title. @@ -447,57 +548,23 @@ void AliPHOSTrackSegmentMakerv1::WriteTrackSegments(Int_t event) // ROOT does not allow overwriting existing branches, therefore // first we check, if branches with the same title already exist. // If yes - exits without writing. - - AliPHOSGetter *gime = AliPHOSGetter::GetInstance() ; - TClonesArray * trackSegments = gime->TrackSegments(BranchName()) ; + AliPHOSGetter *gime = AliPHOSGetter::Instance() ; + + TClonesArray * trackSegments = gime->TrackSegments() ; trackSegments->Expand(trackSegments->GetEntriesFast()) ; - //Make branch in TreeR for TrackSegments - char * filename = 0; - if(gSystem->Getenv("CONFIG_SPLIT_FILE")!=0){ //generating file name - filename = new char[strlen(gAlice->GetBaseFile())+20] ; - sprintf(filename,"%s/PHOS.Reco.root",gAlice->GetBaseFile()) ; + if(fWrite){ //We write TreeT + TTree * treeT = gime->TreeT(); + + //First TS + Int_t bufferSize = 32000 ; + TBranch * tsBranch = treeT->Branch("PHOSTS",&trackSegments,bufferSize); + tsBranch->Fill() ; + + gime->WriteTracks("OVERWRITE"); + gime->WriteTrackSegmentMaker("OVERWRITE"); } - - TDirectory *cwd = gDirectory; - - //First TS - Int_t bufferSize = 32000 ; - TBranch * tsBranch = gAlice->TreeR()->Branch("PHOSTS",&trackSegments,bufferSize); - tsBranch->SetTitle(BranchName()); - if (filename) { - tsBranch->SetFile(filename); - TIter next( tsBranch->GetListOfBranches()); - TBranch * sb ; - while ((sb=static_cast(next()))) { - sb->SetFile(filename); - } - cwd->cd(); - } - - //Second -TSMaker - Int_t splitlevel = 0 ; - AliPHOSTrackSegmentMakerv1 * ts = this ; - TBranch * tsMakerBranch = gAlice->TreeR()->Branch("AliPHOSTrackSegmentMaker","AliPHOSTrackSegmentMakerv1", - &ts,bufferSize,splitlevel); - tsMakerBranch->SetTitle(BranchName()); - if (filename) { - tsMakerBranch->SetFile(filename); - TIter next( tsMakerBranch->GetListOfBranches()); - TBranch * sb; - while ((sb=static_cast(next()))) { - sb->SetFile(filename); - } - cwd->cd(); - } - - tsBranch->Fill() ; - tsMakerBranch->Fill() ; - - gAlice->TreeR()->Write(0,kOverwrite) ; - - delete [] filename ; } @@ -506,26 +573,19 @@ 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 - TString taskName(GetName()) ; - taskName.Remove(taskName.Index(Version())-1) ; - - TClonesArray * trackSegments = AliPHOSGetter::GetInstance()->TrackSegments(taskName) ; - - cout << "AliPHOSTrackSegmentMakerv1: event "<GetEvNumber() << endl ; - cout << " Found " << trackSegments->GetEntriesFast() << " trackSegments " << endl ; + TClonesArray * trackSegments = AliPHOSGetter::Instance()->TrackSegments() ; + + 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# " << endl ; - + printf("TrackSegment # EMC RP# CPV RP#\n") ; Int_t index; for (index = 0 ; index GetEntriesFast() ; index++) { AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ; - cout<<" "<< setw(4) << ts->GetIndexInList() << " " - <GetEmcIndex()<< " " - <GetCpvIndex()<< " " << endl ; + printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ; } - - cout << "-------------------------------------------------------"<< endl ; } }