X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PHOS%2FAliPHOSTrackSegmentMakerv1.cxx;h=44ca8d40c0968b459b470bc26e36abeef450834d;hb=34ab69dd4975d604e58ee869e76585653e80aaf2;hp=e2b2be43b5b0e626a79386db3bb78e074cd8dceb;hpb=b9791748e007ba282c12a057a60db875fe90d08b;p=u%2Fmrichter%2FAliRoot.git diff --git a/PHOS/AliPHOSTrackSegmentMakerv1.cxx b/PHOS/AliPHOSTrackSegmentMakerv1.cxx index e2b2be43b5b..44ca8d40c09 100644 --- a/PHOS/AliPHOSTrackSegmentMakerv1.cxx +++ b/PHOS/AliPHOSTrackSegmentMakerv1.cxx @@ -13,22 +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 fRcpv. +// 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") @@ -36,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") // @@ -44,25 +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 "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) @@ -73,25 +90,19 @@ ClassImp( AliPHOSTrackSegmentMakerv1) // default ctor (to be used mainly by Streamer) InitParameters() ; - - fTrackSegmentsInRun = 0 ; - fDefaultInit = kTRUE ; } //____________________________________________________________________________ - AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const char * headerFile, const char * name, const Bool_t toSplit) : AliPHOSTrackSegmentMaker(headerFile, name, toSplit) + AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString alirunFileName, const TString eventFolderName) + :AliPHOSTrackSegmentMaker(alirunFileName, eventFolderName) { // ctor InitParameters() ; - - fTrackSegmentsInRun = 0 ; - Init() ; - fDefaultInit = kFALSE ; - + fESD = 0; } //____________________________________________________________________________ @@ -99,22 +110,16 @@ ClassImp( AliPHOSTrackSegmentMakerv1) { // dtor // fDefaultInit = kTRUE if TrackSegmentMaker created by default ctor (to get just the parameters) - - if (!fDefaultInit) { - delete fLinkUpArray ; - - fSplitFile = 0 ; - } + if (!fDefaultInit) + delete fLinkUpArray ; } //____________________________________________________________________________ const TString AliPHOSTrackSegmentMakerv1::BranchName() const { - // retrieves the branch name - TString branchName(GetName() ) ; - branchName.Remove(branchName.Index(Version())-1) ; - return branchName ; + + return GetName() ; } //____________________________________________________________________________ @@ -122,8 +127,9 @@ 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() ; TObjArray * cpvRecPoints = gime->CpvRecPoints() ; @@ -137,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 = fRcpv ; - - 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() + fRcpv + 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); } //____________________________________________________________________________ @@ -183,78 +306,47 @@ void AliPHOSTrackSegmentMakerv1::Init() { // Make all memory allocations that are not possible in default constructor - if ( strcmp(GetTitle(), "") == 0 ) - SetTitle("galice.root") ; - - TString branchname = GetName() ; - branchname.Remove(branchname.Index(Version())-1) ; - AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(),branchname.Data(), fToSplit ) ; - if ( gime == 0 ) { - Error("Init", "Could not obtain the Getter object !") ; - return ; - } - - fSplitFile = 0 ; - if(fToSplit){ - //First - extract full path if necessary - TString fileName(GetTitle()) ; - Ssiz_t islash = fileName.Last('/') ; - if(islash(gROOT->GetFile(fileName.Data())); - if(!fSplitFile) - fSplitFile = TFile::Open(fileName.Data(),"update") ; - } + AliPHOSGetter* gime = AliPHOSGetter::Instance(); + if(!gime) + gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data()); fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); - - - gime->PostTrackSegmentMaker(this) ; - gime->PostTrackSegments(BranchName()) ; - + if ( !gime->TrackSegmentMaker() ) { + gime->PostTrackSegmentMaker(this); + } } //____________________________________________________________________________ void AliPHOSTrackSegmentMakerv1::InitParameters() { - // initializes parameters used for the construction of the tacksegments - fRcpv = 10. ; + //Initializes parameters + fRcpv = 10. ; + fRtpc = 4. ; fEmcFirst = 0 ; fEmcLast = 0 ; fCpvFirst = 0 ; fCpvLast = 0 ; fLinkUpArray = 0 ; - TString tsmName( GetName()) ; - if (tsmName.IsNull() ) - tsmName = "Default" ; - tsmName.Append(":") ; - tsmName.Append(Version()) ; - SetName(tsmName) ; + fWrite = kTRUE ; + fTrackSegmentsInRun = 0 ; + SetEventRange(0,-1) ; } //____________________________________________________________________________ void AliPHOSTrackSegmentMakerv1::MakeLinks()const { - // Finds distances (links) between all EMC and PPSD clusters, + // 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() ; + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; TObjArray * emcRecPoints = gime->EmcRecPoints() ; TObjArray * cpvRecPoints = gime->CpvRecPoints() ; fLinkUpArray->Clear() ; - AliPHOSRecPoint * cpv ; + AliPHOSCpvRecPoint * cpv ; AliPHOSEmcRecPoint * emcclu ; Int_t iLinkUp = 0 ; @@ -263,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 ; + 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) ; + new ((*fLinkUpArray)[iLinkUp++]) AliPHOSLink(r, iEmcRP, iCpv, track) ; } } } @@ -288,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() ; + + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + TObjArray * emcRecPoints = gime->EmcRecPoints() ; TObjArray * cpvRecPoints = gime->CpvRecPoints() ; - TClonesArray * trackSegments = gime->TrackSegments(BranchName()) ; + TClonesArray * trackSegments = gime->TrackSegments(); //Make arrays to mark clusters already chosen Int_t * emcExist = 0; @@ -315,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++; } } } @@ -354,71 +447,81 @@ 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 ; } - - AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ; - if(gime->BranchExists("TrackSegments") ) - return ; + + AliPHOSGetter * gime = AliPHOSGetter::Instance() ; + const AliPHOSGeometry * geom = gime->PHOSGeometry() ; - Int_t nevents = gime->MaxEvent() ; //(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"); Info("Exec", "took %f seconds for making TS %f seconds per event", - gBenchmark->GetCpuTime("PHOSTSMaker"), gBenchmark->GetCpuTime("PHOSTSMaker")/nevents) ; - } - + 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 +void AliPHOSTrackSegmentMakerv1::Print(const Option_t *)const { // Print TrackSegmentMaker parameters @@ -427,7 +530,7 @@ void AliPHOSTrackSegmentMakerv1::Print(Option_t * option)const message = "\n======== AliPHOSTrackSegmentMakerv1 ========\n" ; message += "Making Track segments\n" ; message += "with parameters:\n" ; - message += " Maximal EMC - CPV (PPSD) distance (cm) %f\n" ; + message += " Maximal EMC - CPV distance (cm) %f\n" ; message += "============================================\n" ; Info("Print", message.Data(),fRcpv) ; } @@ -436,7 +539,7 @@ void AliPHOSTrackSegmentMakerv1::Print(Option_t * option)const } //____________________________________________________________________________ -void AliPHOSTrackSegmentMakerv1::WriteTrackSegments(Int_t event) +void AliPHOSTrackSegmentMakerv1::WriteTrackSegments() { // Writes found TrackSegments to TreeR. Creates branches // "PHOSTS" and "AliPHOSTrackSegmentMaker" with the same title. @@ -445,48 +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() ; + + AliPHOSGetter *gime = AliPHOSGetter::Instance() ; TClonesArray * trackSegments = gime->TrackSegments() ; trackSegments->Expand(trackSegments->GetEntriesFast()) ; - TTree * treeR ; - - if(fToSplit){ - if(!fSplitFile) - return ; - fSplitFile->cd() ; - char name[10] ; - sprintf(name,"%s%d", "TreeR",event) ; - treeR = dynamic_cast(fSplitFile->Get(name)); - } - else{ - treeR = gAlice->TreeR(); - } - - if(!treeR){ - gAlice->MakeTree("R", fSplitFile); - treeR = gAlice->TreeR() ; + + 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"); } - - //First TS - Int_t bufferSize = 32000 ; - TBranch * tsBranch = treeR->Branch("PHOSTS",&trackSegments,bufferSize); - tsBranch->SetTitle(BranchName()); - - //Second -TSMaker - Int_t splitlevel = 0 ; - AliPHOSTrackSegmentMakerv1 * ts = this ; - TBranch * tsMakerBranch = treeR->Branch("AliPHOSTrackSegmentMaker","AliPHOSTrackSegmentMakerv1", - &ts,bufferSize,splitlevel); - tsMakerBranch->SetTitle(BranchName()); - - tsBranch->Fill() ; - tsMakerBranch->Fill() ; - - treeR->AutoSave() ; //Write(0,kOverwrite) ; - if(gAlice->TreeR()!=treeR) - treeR->Delete(); } @@ -495,30 +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) ; - TString message ; - message = "\nevent " ; - message += gAlice->GetEvNumber() ; - message += "\n Found " ; - message += trackSegments->GetEntriesFast() ; - message += " TrackSegments\n" ; + 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 - message += "TrackSegment # EMC RP# CPV RP#\n" ; + printf("TrackSegment # EMC RP# CPV RP#\n") ; Int_t index; for (index = 0 ; index GetEntriesFast() ; index++) { AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ; - message += "\n" ; - message += ts->GetIndexInList() ; - message += " " ; - message += ts->GetEmcIndex() ; - message += " " ; - message += ts->GetCpvIndex() ; + printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ; } } - Info("Print", message.Data() ) ; }