]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PHOS/AliPHOSTrackSegmentMakerv1.cxx
Corrected copy constructor and assignment operator, fixed memory leaks (Christian)
[u/mrichter/AliRoot.git] / PHOS / AliPHOSTrackSegmentMakerv1.cxx
index 118ecf5ade4928608bfd82f3ccf55f42c3f84c83..67fda5813d7617f4adab9cd7c3322758658752f2 100644 (file)
 /* History of cvs commits:
  *
  * $Log$
+ * Revision 1.90  2007/07/11 13:43:30  hristov
+ * New class AliESDEvent, backward compatibility with the old AliESD (Christian)
+ *
+ * Revision 1.89  2007/07/03 08:13:04  kharlov
+ * Bug fix in CPV local coordinates
+ *
+ * Revision 1.88  2007/06/27 09:11:07  kharlov
+ * Bug fix for CPV-EMC distance
+ *
+ * Revision 1.87  2007/05/04 14:49:29  policheh
+ * AliPHOSRecPoint inheritance from AliCluster
+ *
+ * Revision 1.86  2007/04/02 15:00:16  cvetan
+ * No more calls to gAlice in the reconstruction
+ *
+ * Revision 1.85  2007/03/28 19:18:15  kharlov
+ * RecPoints recalculation in TSM removed
+ *
+ * Revision 1.84  2007/03/07 07:01:21  hristov
+ * Fixing copy/paste erro. Additional protections
+ *
+ * Revision 1.83  2007/03/06 21:07:37  kharlov
+ * DP: xz CPV-EMC distance filled to TS
+ *
+ * Revision 1.82  2007/03/06 06:54:48  kharlov
+ * DP:Calculation of cluster properties dep. on vertex added
+ *
+ * Revision 1.81  2007/02/05 10:02:40  kharlov
+ * Module numbering is corrected
+ *
  * Revision 1.80  2006/08/28 10:01:56  kharlov
  * Effective C++ warnings fixed (Timur Pocheptsov)
  *
 #include "AliPHOSTrackSegment.h"
 #include "AliPHOSLink.h"
 #include "AliPHOSGetter.h"
-#include "AliESD.h"
+#include "AliESDEvent.h"
 #include "AliESDtrack.h"
+#include "AliPHOSQualAssDataMaker.h"
 
 ClassImp( AliPHOSTrackSegmentMakerv1) 
 
@@ -98,6 +129,7 @@ AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() :
   fNTrackSegments(0),
   fRcpv(0.f),
   fRtpc(0.f),
+  fVtx(0.f), 
   fLinkUpArray(0),
   fEmcFirst(0),
   fEmcLast(0),
@@ -119,6 +151,7 @@ AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString & alirunFil
   fNTrackSegments(0),
   fRcpv(0.f),
   fRtpc(0.f),
+  fVtx(0.f), 
   fLinkUpArray(0),
   fEmcFirst(0),
   fEmcLast(0),
@@ -141,6 +174,7 @@ AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const AliPHOSTrackSegment
   fNTrackSegments(0),
   fRcpv(0.f),
   fRtpc(0.f),
+  fVtx(0.f), 
   fLinkUpArray(0),
   fEmcFirst(0),
   fEmcLast(0),
@@ -179,10 +213,9 @@ void  AliPHOSTrackSegmentMakerv1::FillOneModule()
   // clusters from one PHOS module are
 
   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) &&  
@@ -199,159 +232,106 @@ void  AliPHOSTrackSegmentMakerv1::FillOneModule()
 }
 
 //____________________________________________________________________________
-Float_t  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,AliPHOSCpvRecPoint * cpvClu, Int_t &trackindex) const
+void  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,
+                                                         AliPHOSCpvRecPoint * cpvClu, 
+                                                         Int_t &trackindex, 
+                                                         Float_t &dx, Float_t &dz) const
 {
   // Calculates the distance between the EMC RecPoint and the CPV RecPoint
   // Clusters are sorted in "rows" and "columns" of width 1 cm
 
-  //Float_t delta = 1 ;  // Width of the rows in sorting of RecPoints (in cm)
-                       // if you change this value, change it as well in xxxRecPoint::Compare()
-  Float_t distance2Cpv   = fRcpv ;
+//  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 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
+  TVector3 vecP ;     // Momentum direction at CPV plain
+  TVector3 vecPloc ;     // Momentum direction at CPV plain
   
-  emcClu->GetLocalPosition(vecEmc) ;
-  cpvClu->GetLocalPosition(vecCpv) ;
-
   //toofar = kTRUE ;
-  if(emcClu->GetPHOSMod() == cpvClu->GetPHOSMod()){ 
+  if(emcClu->GetPHOSMod() != cpvClu->GetPHOSMod()){
+    dx=999. ;
+    dz=999. ;
+    return ;
+  }
+
+  emcClu->GetLocalPosition(vecEmc) ;
 
-    // Find EMC-CPV distance
-    distance2Cpv = (vecCpv - vecEmc).Mag() ;
+  AliPHOSGetter * gime = AliPHOSGetter::Instance() ; 
+  const AliPHOSGeometry * geom = gime->PHOSGeometry() ; 
     
-    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; imod<nModules; imod++) {
-       // Modules are numbered from 1 to 5 in AliPHOSGeometry
-       geom->EmcModuleCoverage(imod+1,
-                               thmin[imod],thmax[imod],
-                               phmin[imod],phmax[imod]);
-      }
+  Double_t xCPV,zCPV ; //EMC-projected coordinates of CPV cluster 
+  TVector3 cpvGlobal; // Global position of the CPV recpoint
+  geom->GetGlobalPHOS((AliPHOSRecPoint*)cpvClu,cpvGlobal);
+  Double_t vtxCPV[3]={cpvGlobal.X(),cpvGlobal.Y(),cpvGlobal.Z()} ;
+  Int_t dummyMod ;
+
+  if (fESD == 0x0) {
+     //if no track information available, assume straight line from IP to emcal
+     geom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,zCPV,xCPV) ;
+     dx=xCPV - vecEmc.X() ;
+     dz=zCPV - vecEmc.Z() ;
+     return ;
+  } 
+
+  //if there is ESD try to correct distance using TPC information on particle direct in CPV 
+  if (fESD != 0x0) {
+
+    Double_t rCPV = cpvGlobal.Pt() ;// Radius from IP to current point 
 
-      // 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; iTrack<nTracks; iTrack++) {
-       track = fESD->GetTrack(iTrack);
-       if (!track->GetXYZAt(rPHOS, fESD->GetMagneticField(), xyz))
+    // 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 = 1.e6;
+    TVector3 inPHOS ; 
+
+    AliESDtrack *track;
+    Double_t xyz[3] ;
+    Double_t pxyz[3]; 
+    for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
+      track = fESD->GetTrack(iTrack);
+      if (!track->GetXYZAt(rCPV, 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)
+      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<nModules; imod++) {
-         //PH Loop on modules to check if the track enters in the acceptance 
-         if (thmin[imod] < inTheta && thmax[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;
-       }
+      //Check if this track hits PHOS
+      inPHOS.SetXYZ(xyz[0],xyz[1],xyz[2]);
+      distance2Track = inPHOS.Angle(cpvGlobal) ;
+      // Find the closest track to the CPV 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());
-       }
+    if (iClosestTrack != -1) {
+      track = fESD->GetTrack(iClosestTrack);
+      if (track->GetPxPyPzAt(rCPV, fESD->GetMagneticField(), pxyz)) { // track momentum ibid.
+        vecP.SetXYZ(pxyz[0],pxyz[1],pxyz[2]);
+        Int_t dummyMod ;
+       geom->ImpactOnEmc(vtxCPV,vecP.Theta(),vecP.Phi(),dummyMod,zCPV,xCPV) ;
       }
-//     } 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 ;
+    if(minDistance < fRtpc ){
+      trackindex = iClosestTrack ; 
     }
-    //     Info("GetDistanceInPHOSPlane","cpv-emc distance is %f cm",
-    //          distance);
   }
-  
-  return distance2Cpv ;
-}
-
-//____________________________________________________________________________
-void AliPHOSTrackSegmentMakerv1::PropagateToPlane(TVector3& globalIntersection,
-                                                 Double_t *x,
-                                                 Double_t *p,
-                                                 const char *det,
-                                                 Int_t moduleNumber) const
-{
-  // Propagate a straight-line track from the origin point x
-  // along the direction p to the CPV or EMC module moduleNumber
-  // Returns a local position of such a propagation
-
-  AliPHOSGetter * gime = AliPHOSGetter::Instance() ; 
-  const AliPHOSGeometry * geom = gime->PHOSGeometry() ; 
-  TVector3 moduleCenter;
-  geom->GetModuleCenter(moduleCenter,det,moduleNumber);
-  TVector3 vertex; vertex.SetXYZ(x[0],x[1],x[2]);
-  TVector3 direction; direction.SetXYZ(p[0],p[1],p[2]);
-
-//   Info("PropagateToCPV","Center of the %s module %d is (%f,%f,%f)",
-//        det,moduleNumber,moduleCenter[0],moduleCenter[1],moduleCenter[2]);
-
-  Double_t time = (moduleCenter.Mag2() - vertex.Dot(moduleCenter)) /
-    (direction.Dot(moduleCenter));
-  vertex += direction*time;
-  geom->Global2Local(globalIntersection,vertex,moduleNumber);
+  if(trackindex!=-1){
+    // If the closest global track is found, calculate EMC-CPV distance from it
+    dx=xCPV - vecEmc.X() ;
+    dz=zCPV - vecEmc.Z() ;
+  }
+  else{
+    // If no global track was found, just take the nearest CPV point
+    geom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,zCPV,xCPV) ;
+    dx=xCPV - vecEmc.X() ;
+    dz=zCPV - vecEmc.Z() ;
+  }
+  return ;
 }
-
 //____________________________________________________________________________
 void  AliPHOSTrackSegmentMakerv1::Init()
 {
@@ -412,11 +392,10 @@ void  AliPHOSTrackSegmentMakerv1::MakeLinks()const
       
       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) ;
+      Float_t dx,dz ;
+      GetDistanceInPHOSPlane(emcclu, cpv, track,dx,dz) ;     
+      if(TMath::Sqrt(dx*dx+dz*dz) < fRcpv ){ 
+        new ((*fLinkUpArray)[iLinkUp++])  AliPHOSLink(dx, dz, iEmcRP, iCpv, track) ;
       }      
     }
   } 
@@ -466,10 +445,12 @@ void  AliPHOSTrackSegmentMakerv1::MakePairs()
     if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){
 
       if(cpvExist[linkUp->GetCpv()-fCpvFirst]){ //CPV still exist
+         Float_t dx,dz ;
+         linkUp->GetXZ(dx,dz) ;
         new ((* trackSegments)[fNTrackSegments]) 
           AliPHOSTrackSegment(dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(linkUp->GetEmc())) , 
                               dynamic_cast<AliPHOSCpvRecPoint *>(cpvRecPoints->At(linkUp->GetCpv())) , 
-                              linkUp->GetTrack()) ;
+                              linkUp->GetTrack(),dx,dz) ;
         
        (dynamic_cast<AliPHOSTrackSegment *>(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
        fNTrackSegments++ ;
@@ -525,7 +506,7 @@ void  AliPHOSTrackSegmentMakerv1::Exec(Option_t *option)
 
   Int_t ievent ; 
   for (ievent = fFirstEvent; ievent <= fLastEvent; ievent++) {
-    gime->Event(ievent,"R") ;
+    gime->Event(ievent,"DR") ;
    //Make some initializations 
     fNTrackSegments = 0 ;
     fEmcFirst = 0 ;    
@@ -535,14 +516,15 @@ void  AliPHOSTrackSegmentMakerv1::Exec(Option_t *option)
     
     gime->TrackSegments()->Clear();
 
-    //    if(!ReadRecPoints(ievent))   continue; //reads RecPoints for event ievent
+   //   if(!ReadRecPoints(ievent))   continue; //reads RecPoints for event ievent
     
     for(fModule = 1; fModule <= geom->GetNModules() ; fModule++ ) {
       FillOneModule() ; 
       MakeLinks() ;
       MakePairs() ;
     }
-
+   
+    
     WriteTrackSegments() ;
 
     if(strstr(option,"deb"))
@@ -561,7 +543,6 @@ void  AliPHOSTrackSegmentMakerv1::Exec(Option_t *option)
   if(fWrite) //do not unload in "on flight" mode
     Unload();
 }
-
 //____________________________________________________________________________
 void AliPHOSTrackSegmentMakerv1::Unload() 
 {
@@ -628,7 +609,7 @@ void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option)
   TClonesArray * trackSegments = AliPHOSGetter::Instance()->TrackSegments() ; 
 
   Info("PrintTrackSegments", "Results from TrackSegmentMaker:") ; 
-  printf("nevent: %d\n", gAlice->GetEvNumber()) ; 
+  printf("nevent: %d\n", AliPHOSGetter::Instance()->EventNumber()) ; 
   printf("        Found %d TrackSegments\n", trackSegments->GetEntriesFast() ); 
   
   if(strstr(option,"all")) {  // printing found TS