* provided "as is" without express or implied warranty. *
**************************************************************************/
/* $Id$ */
+
+/* History of cvs commits:
+ *
+ * $Log$
+ * 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")
+// 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 "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 "AliPHOS.h"
-#include "AliRun.h"
+#include "AliPHOSGetter.h"
+#include "AliESD.h"
+#include "AliESDtrack.h"
ClassImp( AliPHOSTrackSegmentMakerv1)
//____________________________________________________________________________
- 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 ;
-}
-//____________________________________________________________________________
- AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const char* headerFile, const char* branchTitle):
-AliPHOSTrackSegmentMaker()
+ AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() : AliPHOSTrackSegmentMaker()
{
- // 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 ;
+ // default ctor (to be used mainly by Streamer)
+ InitParameters() ;
+ fDefaultInit = kTRUE ;
}
+
//____________________________________________________________________________
-void AliPHOSTrackSegmentMakerv1::Init()
+ AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString alirunFileName, const TString eventFolderName)
+ :AliPHOSTrackSegmentMaker(alirunFileName, eventFolderName)
{
- // Make all memory allocations that are not possible in default constructor
-
- 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) ;
+ // ctor
- 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() ;
}
//____________________________________________________________________________
{
// 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<AliPHOSRecPoint *>(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<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];
+ Double_t rPHOS =
+ AliPHOSGetter::Instance()->PHOSGeometry()->GetIPtoCrystalSurface();
+ AliESDtrack *track;
+ for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
+ track = fESD->GetTrack(iTrack);
+ if (track->IsPHOS())
+ continue ;
+ 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;
+ if (!track->GetPxPyPzAt(rPHOS, fESD->GetMagneticField(), pxyz))
+ continue; // 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);
+ if (track->GetPxPyPzAt(rPHOS, fESD->GetMagneticField(), 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 ;
+}
- } // if xPpsd >= xEmc + ...
- else
- toofar = kTRUE ;
- }
- else
- toofar = kTRUE ;
+//____________________________________________________________________________
+TVector3 AliPHOSTrackSegmentMakerv1::PropagateToPlane(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(det,moduleNumber);
+ TVector3 vertex(x);
+ TVector3 direction(p);
+
+// 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);
+}
- //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()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) ;
+ emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->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) ;
-
- 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<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".
-
- //Make arrays to mark clusters already chousen
+
+ 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] ;
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 ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fEmcRecPoints->At(linkLow->GetEmc()),
- nullpointer,
- (AliPHOSRecPoint *)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<AliPHOSLink *>(nextUp()) ) ){
- new ((* fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *) fEmcRecPoints->At(linkUp->GetEmc()) ,
- (AliPHOSRecPoint *)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<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 ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment((AliPHOSEmcRecPoint *)fEmcRecPoints->At(iEmcRP+fEmcFirst),
- nullpointer,
- nullpointer ) ;
- ((AliPHOSTrackSegment *) fTrackSegments->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(! 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
+void AliPHOSTrackSegmentMakerv1::Print(const Option_t *)const
{
// Print TrackSegmentMaker parameters
- if(fIsInitialized){
- 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: " << fTSBranchTitle.Data() << endl ;
- cout << "with parameters: " << endl ;
- cout << " Maximal EMC - CPV (PPSD) distance (cm)" << fR0 << endl ;
- cout << "============================================" << endl ;
+ 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) ;
}
else
- cout << "AliPHOSTrackSegmentMakerv1 not initialized " << endl ;
+ Info("Print", "AliPHOSTrackSegmentMakerv1 not initialized ") ;
}
-//____________________________________________________________________________
-Bool_t AliPHOSTrackSegmentMakerv1::ReadRecPoints()
-{
- // Reads Emc and CPV recPoints with given title (fRecPointsBranchTitle)
- // made previously with Clusterizer.
-
-
- //Make some initializations
- fEmcRecPoints->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
- if(gAlice->TreeR()==0){
- char treeName[20];
- sprintf(treeName,"TreeR%d",fEvent);
- cout << "Error in AliPHOSTrackSegmentMakerv1 : no "<<treeName << endl ;
- cout << " Do nothing " << endl ;
- return kFALSE ;
- }
-
-
- //Find RecPoints with title fRecPointsBranchTitle
- TBranch * emcBranch = 0;
- TBranch * cpvBranch = 0;
- TBranch * clusterizerBranch = 0;
-
- TObjArray * branches = gAlice->TreeR()->GetListOfBranches() ;
- Int_t ibranch;
- Bool_t emcNotFound = kTRUE ;
- Bool_t cpvNotFound = kTRUE ;
- Bool_t clusterizerNotFound = kTRUE ;
-
- for(ibranch = 0;ibranch <branches->GetEntries();ibranch++){
-
- if(emcNotFound){
- emcBranch=(TBranch *) branches->At(ibranch) ;
- if( fRecPointsBranchTitle.CompareTo(emcBranch->GetTitle())==0 )
- if( strcmp(emcBranch->GetName(),"PHOSEmcRP") == 0) {
- emcNotFound = kFALSE ;
- }
- }
-
- if(cpvNotFound){
- cpvBranch=(TBranch *) branches->At(ibranch) ;
- if( fRecPointsBranchTitle.CompareTo(cpvBranch->GetTitle())==0 )
- if( strcmp(cpvBranch->GetName(),"PHOSCpvRP") == 0)
- cpvNotFound = kFALSE ;
- }
-
- if(clusterizerNotFound){
- clusterizerBranch = (TBranch *) branches->At(ibranch) ;
- if( fRecPointsBranchTitle.CompareTo(clusterizerBranch->GetTitle()) == 0)
- if( strcmp(clusterizerBranch->GetName(),"AliPHOSClusterizer") == 0)
- clusterizerNotFound = kFALSE ;
- }
-
- }
- if(clusterizerNotFound || emcNotFound || cpvNotFound){
- cout << "AliPHOSTrackSegmentMakerv1: " << endl ;
- cout << " Can't find Branch with RecPoints or Clusterizer " ;
- cout << " Do nothing" <<endl ;
- return kFALSE ;
- }
-
- emcBranch->SetAddress(&fEmcRecPoints) ;
- cpvBranch->SetAddress(&fCpvRecPoints) ;
- clusterizerBranch->SetAddress(&fClusterizer) ;
-
- emcBranch->GetEntry(0) ;
- cpvBranch->GetEntry(0) ;
- clusterizerBranch->GetEntry(0) ;
-
- return kTRUE ;
-
-}
//____________________________________________________________________________
void AliPHOSTrackSegmentMakerv1::WriteTrackSegments()
{
// ROOT does not allow overwriting existing branches, therefore
// first we check, if branches with the same title already exist.
// If yes - exits without writing.
-
- //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 <branches->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 '"<<fTSBranchTitle.Data() << "' already exist " << endl ;
- cout << " can not overwrite " << endl ;
- return ;
- }
-
- //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 ;
- tsBranch = gAlice->TreeR()->Branch("PHOSTS",&fTrackSegments,bufferSize);
- tsBranch->SetTitle(fTSBranchTitle.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 ;
- tsMakerBranch = gAlice->TreeR()->Branch("AliPHOSTrackSegmentMaker","AliPHOSTrackSegmentMakerv1",
- &ts,bufferSize,splitlevel);
- tsMakerBranch->SetTitle(fTSBranchTitle.Data());
- if (filename) {
- tsMakerBranch->SetFile(filename);
- TIter next( tsMakerBranch->GetListOfBranches());
- TBranch * sb;
- while ((sb=(TBranch*)next())) {
- sb->SetFile(filename);
- }
- cwd->cd();
- }
-
+ TClonesArray * trackSegments = gime->TrackSegments() ;
+ trackSegments->Expand(trackSegments->GetEntriesFast()) ;
- tsBranch->Fill() ;
- tsMakerBranch->Fill() ;
-// gAlice->TreeR()->Fill() ; // YK 28.05.01
- 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
-
- cout << "AliPHOSTrackSegmentMakerv1: " << endl ;
- cout << " Found " << fTrackSegments->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# " << " PPSD RP#" << endl ;
-
+ printf("TrackSegment # EMC RP# CPV RP#\n") ;
Int_t index;
- for (index = 0 ; index <fTrackSegments->GetEntriesFast() ; index++) {
- AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )fTrackSegments->At(index) ;
- cout<<" "<< setw(4) << ts->GetIndexInList() << " "
- <<setw(4) << ts->GetEmcIndex()<< " "
- <<setw(4) << ts->GetCpvIndex()<< " "
- <<setw(4) << ts->GetPpsdIndex()<< endl ;
+ for (index = 0 ; index <trackSegments->GetEntriesFast() ; index++) {
+ AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ;
+ printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ;
}
-
- cout << "-------------------------------------------------------"<< endl ;
}
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff