//_________________________________________________________________________
// 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")
+// root [0] AliPHOSTrackSegmentMakerv1 * t = new AliPHOSTrackSegmentMaker("galice.root", "tracksegmentsname", "recpointsname")
// Warning in <TDatabasePDG::TDatabasePDG>: 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 [2] t->SetMaxEmcPpsdDistance(5)
// root [3] t->SetTrackSegmentsBranch("max distance 5 cm")
// root [4] t->ExecuteTask("deb all time")
//
-//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH)
+//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH) & Yves Schutz (SUBATECH)
//
// --- ROOT system ---
-#include "TROOT.h"
-#include "TFile.h"
-#include "TFolder.h"
#include "TTree.h"
-#include "TSystem.h"
#include "TBenchmark.h"
-// --- Standard library ---
-
-#include <iostream.h>
-#include <iomanip.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 "AliPHOSPpsdRecPoint.h"
#include "AliPHOSLink.h"
#include "AliPHOSGetter.h"
-#include "AliPHOS.h"
-#include "AliRun.h"
+#include "AliESD.h"
+#include "AliESDtrack.h"
ClassImp( AliPHOSTrackSegmentMakerv1)
{
// default ctor (to be used mainly by Streamer)
- fR0 = 10. ;
- fEmcFirst = 0 ;
- fEmcLast = 0 ;
- fCpvFirst = 0 ;
- fCpvLast = 0 ;
- fPpsdFirst = 0 ;
- fPpsdLast = 0 ;
- fLinkLowArray = 0 ;
- fLinkUpArray = 0 ;
- fHeaderFileName = "" ;
- fRecPointsBranchTitle = "" ;
- fTrackSegmentsBranchTitle = "" ;
- fTrackSegmentsInRun = 0 ;
+ InitParameters() ;
+ fDefaultInit = kTRUE ;
}
//____________________________________________________________________________
- AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const char * headerFile, const char * name) : 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 ;
- fPpsdFirst = 0 ;
- fPpsdLast = 0 ;
-
- fHeaderFileName = GetTitle() ;
- fRecPointsBranchTitle = GetName() ;
- fTrackSegmentsBranchTitle = GetName() ;
- fTrackSegmentsInRun = 0 ;
-
- TString tempo(GetName()) ;
- tempo.Append(Version()) ;
- SetName(tempo.Data()) ;
-
+ 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() ;
}
//____________________________________________________________________________
{
// Finds first and last indexes between which
// clusters from one PHOS module are
-
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
- const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
+
+ AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
+
TObjArray * emcRecPoints = gime->EmcRecPoints() ;
TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
-
+
//First EMC clusters
Int_t totalEmc = emcRecPoints->GetEntriesFast() ;
for(fEmcFirst = fEmcLast; (fEmcLast < totalEmc) &&
- (((AliPHOSRecPoint *) emcRecPoints->At(fEmcLast))->GetPHOSMod() == fModule );
+ ((dynamic_cast<AliPHOSRecPoint *>(emcRecPoints->At(fEmcLast)))->GetPHOSMod() == fModule );
fEmcLast ++) ;
//Now CPV clusters
Int_t totalCpv = cpvRecPoints->GetEntriesFast() ;
- if(fModule <= geom->GetNCPVModules()){ // in CPV geometry
-
for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) &&
- (((AliPHOSRecPoint *) cpvRecPoints->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 *) cpvRecPoints->At(fCpvLast))->GetPHOSMod() == fModule ) &&
- (((AliPHOSPpsdRecPoint *) cpvRecPoints->At(fCpvLast))->GetUp()) ;
- fCpvLast ++) ;
-
- fPpsdLast= fCpvLast ;
- for(fPpsdFirst = fPpsdLast; (fPpsdLast < totalCpv) &&
- (((AliPHOSPpsdRecPoint *) cpvRecPoints->At(fPpsdLast))->GetPHOSMod() == fModule ) &&
- (!((AliPHOSPpsdRecPoint *) cpvRecPoints->At(fPpsdLast))->GetUp()) ;
- fPpsdLast ++) ;
- }
-
+ ((dynamic_cast<AliPHOSRecPoint *>(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) {
+ // 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];
+ AliESDtrack *track;
+ for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
+ track = fESD->GetTrack(iTrack);
+ if (track->IsPHOS())
+ continue ;
+ track->GetOuterXYZPHOS(xyz); // track coord on the cylinder of PHOS radius
+ if ((TMath::Abs(xyz[0])+TMath::Abs(xyz[1])+TMath::Abs(xyz[2]))<=0)
+ continue;
+ track->GetOuterPxPyPzPHOS(pxyz); // track momentum ibid.
+ vecDist = PropagateToPlane(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;
+ }
+ }
+
+ if (iClosestTrack != -1) {
+ track = fESD->GetTrack(iClosestTrack);
+ track->GetOuterPxPyPzPHOS(pxyz); // track momentum ibid.
+ TVector3 vecCpvGlobal; // Global position of the CPV recpoint
+ AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
+ const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
+ geom->GetGlobal((AliRecPoint*)cpvClu,vecCpvGlobal);
+ for (Int_t ixyz=0; ixyz<3; ixyz++)
+ xyz[ixyz] = vecCpvGlobal[ixyz];
+ vecDist = PropagateToPlane(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 ;
+}
+
+//____________________________________________________________________________
+TVector3 AliPHOSTrackSegmentMakerv1::PropagateToPlane(Double_t *x, Double_t *p,
+ 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
- } // if xPpsd >= xEmc + ...
- else
- toofar = kTRUE ;
- }
- else
- toofar = kTRUE ;
+ AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
+ const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
+ TVector3 moduleCenter = geom->GetModuleCenter(det,moduleNumber);
+ TVector3 vertex(x);
+ TVector3 direction(p);
- //toofar = kFALSE ;
-
-
- return r ;
+// 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));
+ TVector3 globalIntersection = vertex + direction*time;
+ return geom->Global2Local(globalIntersection,moduleNumber);
}
//____________________________________________________________________________
{
// Make all memory allocations that are not possible in default constructor
- if ( strcmp(GetTitle(), "") == 0 )
- SetTitle("galice.root") ;
+ AliPHOSGetter* gime = AliPHOSGetter::Instance();
+ if(!gime)
+ gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data());
- TString taskName(GetName()) ;
- taskName.ReplaceAll(Version(), "") ;
-
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance(GetTitle(), taskName.Data()) ;
- if ( gime == 0 ) {
- cerr << "ERROR: AliPHOSTrackSegmentMakerv1::Init -> Could not obtain the Getter object !" << endl ;
- return ;
- }
-
- fLinkLowArray = new TClonesArray("AliPHOSLink", 1000);
fLinkUpArray = new TClonesArray("AliPHOSLink", 1000);
-
- //add Task to //YSAlice/tasks/Reconstructioner/PHOS
- TTask * aliceRe = (TTask*)gROOT->FindObjectAny("YSAlice/tasks/Reconstructioner") ;
- TTask * phosRe = (TTask*)aliceRe->GetListOfTasks()->FindObject("PHOS") ;
- phosRe->Add(this) ;
- // create a folder on the white board //YSAlice/WhiteBoard/RecPoints/PHOS/trackSegmentsName
-
- gime->Post(GetTitle(), "T", taskName.Data() ) ;
+ 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() ;
+ 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 *) emcRecPoints->At(iEmcRP) ;
+ emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(iEmcRP)) ;
- Bool_t toofar ;
- Int_t iPpsd ;
- for(iPpsd = fPpsdFirst; iPpsd < fPpsdLast;iPpsd++ ) {
-
- ppsd = (AliPHOSRecPoint *) cpvRecPoints->At(iPpsd) ;
- Float_t r = GetDistanceInPHOSPlane(emcclu, ppsd, toofar) ;
-
- 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 *) 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<AliPHOSCpvRecPoint *>(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
}
//____________________________________________________________________________
// 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() ;
-
+ TClonesArray * trackSegments = gime->TrackSegments();
+
//Make arrays to mark clusters already chosen
Int_t * emcExist = 0;
if(fEmcLast > fEmcFirst)
for(index = 0; index <fCpvLast-fCpvFirst; index ++)
cpvExist[index] = kTRUE ;
- Bool_t * ppsdExist = 0;
- if(fPpsdLast > fPpsdFirst)
- ppsdExist = new Bool_t[fPpsdLast-fPpsdFirst] ;
- for(index = 0; index <fPpsdLast-fPpsdFirst; index ++)
- ppsdExist[index] = kTRUE ;
// Finds the smallest links and makes pairs of CPV and EMC clusters with smallest distance
- TIter nextLow(fLinkLowArray) ;
TIter nextUp(fLinkUpArray) ;
- AliPHOSLink * linkLow ;
AliPHOSLink * linkUp ;
-
-
- AliPHOSRecPoint * nullpointer = 0 ;
-
- while ( (linkLow = (AliPHOSLink *)nextLow() ) ){
- if( (emcExist[linkLow->GetEmc()-fEmcFirst]> 0) &&
- ppsdExist[linkLow->GetPpsd()-fPpsdFirst] ){ // RecPoints not removed yet
- new ((*trackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) emcRecPoints->At(linkLow->GetEmc()),
- nullpointer,
- (AliPHOSRecPoint *)cpvRecPoints->At(linkLow->GetPpsd()) ) ;
-
- ((AliPHOSTrackSegment* )trackSegments->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
+ AliPHOSCpvRecPoint * nullpointer = 0 ;
+
+ while ( (linkUp = static_cast<AliPHOSLink *>(nextUp()) ) ){
- if(cpvExist[linkUp->GetPpsd()-fCpvFirst]){ //CPV still exist
-
- if(emcExist[linkUp->GetEmc()-fEmcFirst] > 0){ //without ppsd Low => create new TS
+ if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){
- new ((* trackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) emcRecPoints->At(linkUp->GetEmc()) ,
- (AliPHOSRecPoint *)cpvRecPoints->At(linkUp->GetPpsd()),
- nullpointer) ;
- ((AliPHOSTrackSegment *) trackSegments->At(fNTrackSegments))->SetIndexInList(fNTrackSegments);
- fNTrackSegments++ ;
- }
- else{ // append ppsd Up to existing TS
- ((AliPHOSTrackSegment *)trackSegments->At(-2-emcExist[linkUp->GetEmc()-fEmcFirst]))->SetCpvRecPoint((AliPHOSCpvRecPoint *)cpvRecPoints->At(linkUp->GetPpsd()));
- }
-
- 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<AliPHOSEmcRecPoint *>(emcRecPoints->At(linkUp->GetEmc())) ,
+ dynamic_cast<AliPHOSCpvRecPoint *>(cpvRecPoints->At(linkUp->GetCpv())) ,
+ linkUp->GetTrack()) ;
+
+ (dynamic_cast<AliPHOSTrackSegment *>(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((AliPHOSEmcRecPoint *)emcRecPoints->At(iEmcRP+fEmcFirst),
- nullpointer,
- nullpointer ) ;
- ((AliPHOSTrackSegment *) trackSegments->At(fNTrackSegments))->SetIndexInList(fNTrackSegments);
- fNTrackSegments++;
+ new ((*trackSegments)[fNTrackSegments])
+ AliPHOSTrackSegment(dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(iEmcRP+fEmcFirst)),
+ nullpointer) ;
+ (dynamic_cast<AliPHOSTrackSegment *>(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
+ fNTrackSegments++;
}
}
}
-
+ delete [] emcExist ;
+ delete [] cpvExist ;
}
//____________________________________________________________________________
-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 ;
}
-
- //check, if the branch with name of this" already exits?
- TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
- TIter next(lob) ;
- TBranch * branch = 0 ;
- Bool_t phostsfound = kFALSE, tracksegmentmakerfound = kFALSE ;
- TString taskName(GetName()) ;
- taskName.ReplaceAll(Version(), "") ;
+ AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
+
+ const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
- while ( (branch = (TBranch*)next()) && (!phostsfound || !tracksegmentmakerfound) ) {
- if ( (strcmp(branch->GetName(), "PHOSTS")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
- phostsfound = kTRUE ;
+ 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 ;
- else if ( (strcmp(branch->GetName(), "AliPHOSTrackSegmentMaker")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) )
- tracksegmentmakerfound = kTRUE ;
- }
-
- if ( phostsfound || tracksegmentmakerfound ) {
- cerr << "WARNING: AliPHOSTrackSegmentMakerv1::Exec -> TrackSegments and/or TrackSegmentMaker branch with name "
- << taskName.Data() << " already exits" << endl ;
- return ;
- }
-
- const AliPHOSGeometry * geom = AliPHOSGetter::GetInstance()->PHOSGeometry() ;
- Int_t nevents = (Int_t) gAlice->TreeE()->GetEntries() ;
- Int_t ievent ;
+ gime->TrackSegments()->Clear();
- for(ievent = 0; ievent < nevents; ievent++){
- gAlice->GetEvent(ievent) ;
- gAlice->SetEvent(ievent) ;
-
- if(!ReadRecPoints(ievent)) //reads RecPoints for event ievent
- continue;
+ // 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()->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()
{
- // Print TrackSegmentMaker parameters
-
- 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 ;
- }
- else
- cout << "AliPHOSTrackSegmentMakerv1 not initialized " << endl ;
+ // Unloads the task from the folder
+ AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
+ gime->PhosLoader()->UnloadRecPoints() ;
+ gime->PhosLoader()->UnloadTracks() ;
}
+
//____________________________________________________________________________
-Bool_t AliPHOSTrackSegmentMakerv1::ReadRecPoints(Int_t event)
+void AliPHOSTrackSegmentMakerv1::Print()const
{
- // Reads Emc and CPV recPoints
- // made previously with Clusterizer.
-
-
- //Make some initializations
-
- fNTrackSegments = 0 ;
- fEmcFirst = 0 ;
- fEmcLast = 0 ;
- fCpvFirst = 0 ;
- fCpvLast = 0 ;
- fPpsdFirst= 0 ;
- fPpsdLast = 0 ;
-
- // Get TreeR header from file
- if(gAlice->TreeR()==0){
- cerr << "ERROR: AliPHOSTrackSegmentMakerv1::ReadRecPoints -> There is no Reconstruction Tree" << endl;
- return kFALSE;
- }
-
-
- // Find RecPoints
- TBranch * emcbranch = 0;
- TBranch * cpvbranch = 0;
- TBranch * clusterizerbranch = 0;
- TObjArray * lob = (TObjArray*)gAlice->TreeR()->GetListOfBranches() ;
- TIter next(lob) ;
- TBranch * branch = 0 ;
- Bool_t phosemcfound = kFALSE, phoscpvfound = kFALSE, clusterizerfound = kFALSE ;
-
- TString taskName(GetName()) ;
- taskName.ReplaceAll(Version(), "") ;
+ // Print TrackSegmentMaker parameters
- while ( (branch = (TBranch*)next()) && (!phosemcfound || !phoscpvfound || !clusterizerfound) ) {
- if ( (strcmp(branch->GetName(), "PHOSEmcRP")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
- phosemcfound = kTRUE ;
- emcbranch = branch ;
- }
-
- else if ( (strcmp(branch->GetName(), "PHOSCpvRP")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
- phoscpvfound = kTRUE ;
- cpvbranch = branch ;
-
- } else if ( (strcmp(branch->GetName(), "AliPHOSClusterizer")==0) && (strcmp(branch->GetTitle(), taskName.Data())==0) ) {
- clusterizerfound = kTRUE ;
- clusterizerbranch = branch ;
- }
+ TString message("") ;
+ if( strcmp(GetName(), "") != 0 ) {
+ 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) ;
}
- if ( !phosemcfound || !phoscpvfound || !clusterizerfound ) {
- cerr << "WARNING: AliPHOSTrackSegmentMakerv1::ReadRecPoints -> emc(cpv)RecPoints and/or Clusterizer branch with name " << taskName.Data()
- << " not found" << endl ;
- return kFALSE ;
- }
-
- AliPHOSGetter * gime = AliPHOSGetter::GetInstance() ;
-
- TObjArray * emcRecPoints = gime->EmcRecPoints() ;
- emcRecPoints->Clear() ;
- emcbranch->SetAddress(&emcRecPoints) ;
-
- TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
- cpvRecPoints->Clear() ;
- cpvbranch->SetAddress(&cpvRecPoints) ;
-
- TClonesArray * trackSegments = gime->TrackSegments() ;
- trackSegments->Clear() ;
-
- AliPHOSClusterizer * clusterizer = 0 ;
- clusterizerbranch->SetAddress(&clusterizer) ;
- clusterizerbranch->GetEntry(0) ;
- TString clusterizerName( fTrackSegmentsBranchTitle ) ;
- clusterizerName.Append(clusterizer->Version()) ;
- clusterizer = gime->Clusterizer(clusterizerName) ;
-
- emcbranch->GetEntry(0) ;
- cpvbranch->GetEntry(0) ;
-
- clusterizerbranch->GetEntry(0) ;
-
- printf("ReadRecPoint: EMC=%d, CPV=%d\n",emcRecPoints->GetEntries(),cpvRecPoints->GetEntries());
-
- return kTRUE ;
-
+ else
+ 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.
// 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() ;
- //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()) ;
- }
+ AliPHOSGetter *gime = AliPHOSGetter::Instance() ;
- TDirectory *cwd = gDirectory;
-
- //First TS
- Int_t bufferSize = 32000 ;
- TBranch * tsBranch = gAlice->TreeR()->Branch("PHOSTS",&trackSegments,bufferSize);
- tsBranch->SetTitle(fTrackSegmentsBranchTitle.Data());
- if (filename) {
- tsBranch->SetFile(filename);
- TIter next( tsBranch->GetListOfBranches());
- TBranch * sb ;
- while ((sb=(TBranch*)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(fTrackSegmentsBranchTitle.Data());
- if (filename) {
- tsMakerBranch->SetFile(filename);
- TIter next( tsMakerBranch->GetListOfBranches());
- TBranch * sb;
- while ((sb=(TBranch*)next())) {
- sb->SetFile(filename);
- }
- cwd->cd();
- }
-
- tsBranch->Fill() ;
- tsMakerBranch->Fill() ;
+ TClonesArray * trackSegments = gime->TrackSegments() ;
+ trackSegments->Expand(trackSegments->GetEntriesFast()) ;
- gAlice->TreeR()->Write(0,kOverwrite) ;
-
+ 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");
+ }
}
// option deb - prints # of found TrackSegments
// option deb all - prints as well indexed of found RecParticles assigned to the TS
- TClonesArray * trackSegments = AliPHOSGetter::GetInstance()->TrackSegments() ;
-
- cout << "AliPHOSTrackSegmentMakerv1: event "<<gAlice->GetEvNumber() << endl ;
- cout << " Found " << trackSegments->GetEntriesFast() << " trackSegments " << endl ;
-
- fTrackSegmentsInRun += trackSegments->GetEntriesFast() ;
+ 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# " << " PPSD RP#" << endl ;
-
+ printf("TrackSegment # EMC RP# CPV RP#\n") ;
Int_t index;
for (index = 0 ; index <trackSegments->GetEntriesFast() ; index++) {
AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ;
- cout<<" "<< setw(4) << ts->GetIndexInList() << " "
- <<setw(4) << ts->GetEmcIndex()<< " "
- <<setw(4) << ts->GetCpvIndex()<< " "
- <<setw(4) << ts->GetPpsdIndex()<< endl ;
+ printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ;
}
-
- cout << "-------------------------------------------------------"<< endl ;
}
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff