]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PHOS/AliPHOSTrackSegmentMakerv1.cxx
Removing the flat makefiles
[u/mrichter/AliRoot.git] / PHOS / AliPHOSTrackSegmentMakerv1.cxx
index 444f36629b9e452892fe48ee8a024d0e84d847eb..ba31b3cc1f3c683d97bf5fdba2511598bf578f93 100644 (file)
 //
 
 // --- ROOT system ---
+#include "TVector3.h"
 #include "TTree.h"
 #include "TBenchmark.h"
 
 #include "AliESDtrack.h"
 #include "AliPHOSEmcRecPoint.h"
 #include "AliPHOSCpvRecPoint.h"
+#include "AliLog.h"
+#include "AliMagF.h"
+#include "AliMagF.h"
+#include "AliTracker.h"
+#include "AliGeomManager.h"
+#include "AliCluster.h"
+#include "AliKalmanTrack.h"
+#include "AliGlobalQADataMaker.h"
+#include "AliVParticle.h"
+
 
 ClassImp( AliPHOSTrackSegmentMakerv1) 
 
@@ -238,16 +249,160 @@ void  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * em
                                                          Float_t &dx, Float_t &dz) const
 {
   // Calculates the distance between the EMC RecPoint and the CPV RecPoint
+  // If no CPV, calculates the distance between the EMC RecPoint and the track
+  // prolongation to the PHOS module plane.
   // 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 distance2Track = fRtpc ; 
 
-  trackindex = -1 ; // closest track within fRCpv 
+  trackindex = -1;
+  dx         = 999.;
+  dz         = 999.;
+
+  if(!cpvClu) {
+    
+    if(!emcClu) {
+      return;
+    }
 
+    // *** Start the matching
+    Int_t nt=fESD->GetNumberOfTracks();
+    Int_t iPHOSMod = emcClu->GetPHOSMod()  ;
+    //Calculate actual distance to PHOS module
+    TVector3 globaPos ;
+    fGeom->Local2Global(iPHOSMod, 0.,0., globaPos) ;
+    const Double_t rPHOS = globaPos.Pt() ; //Distance to center of  PHOS module
+    const Double_t kYmax = 72.+10. ; //Size of the module (with some reserve) in phi direction
+    const Double_t kZmax = 64.+10. ; //Size of the module (with some reserve) in z direction
+    const Double_t kAlpha0=330./180.*TMath::Pi() ; //First PHOS module angular direction
+    const Double_t kAlpha= 20./180.*TMath::Pi() ; //PHOS module angular size
+    Double_t minDistance = 1.e6;
+
+    TVector3 vecEmc ;   // Local position of EMC recpoint
+    emcClu->GetLocalPosition(vecEmc) ;
+
+    Double_t gposTrack[3] ; 
+    Double_t bz = AliTracker::GetBz() ; //B-Field for approximate matching
+    Double_t b[3]; 
+    for (Int_t i=0; i<nt; i++) {
+      AliESDtrack *esdTrack=fESD->GetTrack(i);
+
+      // Skip the tracks having "wrong" status (has to be checked/tuned)
+      ULong_t status = esdTrack->GetStatus();
+      if ((status & AliESDtrack::kTPCout)   == 0) continue;
+//     if ((status & AliESDtrack::kTRDout)   == 0) continue;
+//     if ((status & AliESDtrack::kTRDrefit) == 1) continue;
+
+      //Continue extrapolation from TPC outer surface
+      const AliExternalTrackParam *outerParam=esdTrack->GetOuterParam();
+      if (!outerParam) continue;
+      AliExternalTrackParam t(*outerParam);
+
+      t.GetBxByBz(b) ;
+      //Direction to the current PHOS module
+      Double_t phiMod=kAlpha0-kAlpha*iPHOSMod ;
+      if(!t.Rotate(phiMod))
+        continue ;
+      Double_t y;                       // Some tracks do not reach the PHOS
+      if (!t.GetYAt(rPHOS,bz,y)) continue; //    because of the bending
+
+      Double_t z; 
+      if(!t.GetZAt(rPHOS,bz,z))
+        continue ;
+      if (TMath::Abs(z) > kZmax) 
+        continue; // Some tracks miss the PHOS in Z
+      if(TMath::Abs(y) < kYmax){
+        t.PropagateToBxByBz(rPHOS,b);        // Propagate to the matching module
+      //t.CorrectForMaterial(...); // Correct for the TOF material, if needed
+        t.GetXYZ(gposTrack) ;
+        TVector3 globalPositionTr(gposTrack) ;
+        TVector3 localPositionTr ;
+        fGeom->Global2Local(localPositionTr,globalPositionTr,iPHOSMod) ;
+        Double_t ddx = vecEmc.X()-localPositionTr.X();
+        Double_t ddz = vecEmc.Z()-localPositionTr.Z();
+        Double_t d2 = ddx*ddx + ddz*ddz;
+        if(d2 < minDistance) {
+         dx = ddx ;
+         dz = ddz ;
+         trackindex=i;
+         minDistance=d2 ;
+        }
+      }
+    } //Scanned all tracks
+    return ;
+  }
+
+
+  TVector3 emcGlobal;
+  fGeom->GetGlobalPHOS((AliPHOSRecPoint*)emcClu,emcGlobal);
+    
+  // Radius from IP to current point
+  Double_t rEMC = TMath::Abs(emcGlobal.Pt());
+    
+  // Extrapolate the global track direction to EMC
+  // and find the closest track
+    
+  Int_t nTracks = fESD->GetNumberOfTracks();
+    
+  AliESDtrack *track;
+  Double_t xyz[] = {-1,-1,-1};
+  Double_t pxyz[3];
+  Double_t zEMC,xEMC;
+  Int_t module;
+  TVector3 vecP;
+  TVector3 locClu;
+
+  Float_t minDistance = 1.e6;
+  Float_t dr;
+
+  for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
+    track = fESD->GetTrack(iTrack);
+    if (!track->GetXYZAt(rEMC, fESD->GetMagneticField(), xyz)) continue;
+    
+    AliDebug(1,Form("Event %d, iTrack: %d, (%.3f,%.3f,%.3f)",
+                   fESD->GetEventNumberInFile(),iTrack,xyz[0],xyz[1],xyz[2]));
+      
+    if (track->GetPxPyPzAt(rEMC,fESD->GetMagneticField(),pxyz)) { 
+
+      vecP.SetXYZ(pxyz[0],pxyz[1],pxyz[2]);
+      fGeom->ImpactOnEmc(xyz,vecP.Theta(),vecP.Phi(),module,zEMC,xEMC) ;
+
+      if(!module) continue;
+      AliDebug(1,Form("\t\tTrack hit PHOS! Module: %d, (x,z)=(%.3f,%.3f)",module,xEMC,zEMC));
+       
+      if(emcClu->GetPHOSMod() != module) continue;
+       
+      // match track to EMC cluster
+      emcClu->GetLocalPosition(locClu);
+       
+      Float_t delta_x = xEMC - locClu.X();
+      Float_t delta_z = zEMC - locClu.Z();
+      dr = TMath::Sqrt(delta_x*delta_x + delta_z*delta_z);
+      AliDebug(1,Form("\tMatch iTrack=%d: (dx,dz)=(%.3f,%.3f)",iTrack,delta_x,delta_z));
+       
+      if(dr<minDistance) {
+       trackindex = iTrack;
+       minDistance = dr;
+       dx = delta_x;
+       dz = delta_z;
+      }
+    }
+      
+  }
+    
+  if(trackindex>=0)
+    AliDebug(1,Form("\t\tBest match for (xClu,zClu,eClu)=(%.3f,%.3f,%.3f): iTrack=%d, dR=%.3f",
+                   locClu.X(),locClu.Z(),emcClu->GetEnergy(),
+                   trackindex,TMath::Sqrt(dx*dx+dz*dz)));        
+  return;
+  
+  Float_t distance2Track = fRtpc ; 
+  
+  trackindex = -1 ; // closest track within fRCpv 
+  
   TVector3 vecEmc ;   // Local position of EMC recpoint
-  TVector3 vecP ;     // Momentum direction at CPV plain
   TVector3 vecPloc ;     // Momentum direction at CPV plain
   
   //toofar = kTRUE ;
@@ -256,9 +411,9 @@ void  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * em
     dz=999. ;
     return ;
   }
-
+  
   emcClu->GetLocalPosition(vecEmc) ;
-
+  
   Double_t xCPV,zCPV ; //EMC-projected coordinates of CPV cluster 
   TVector3 cpvGlobal; // Global position of the CPV recpoint
   fGeom->GetGlobalPHOS((AliPHOSRecPoint*)cpvClu,cpvGlobal);
@@ -266,40 +421,35 @@ void  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * em
   Int_t dummyMod ;
 
   if (fESD == 0x0) {
-     //if no track information available, assume straight line from IP to emcal
-     fGeom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,zCPV,xCPV) ;
-     dx=xCPV - vecEmc.X() ;
-     dz=zCPV - vecEmc.Z() ;
-     return ;
+    //if no track information available, assume straight line from IP to emcal
+    fGeom->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 = 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
+       continue; //track coord on the cylinder of PHOS radius
       if ((TMath::Abs(xyz[0])+TMath::Abs(xyz[1])+TMath::Abs(xyz[2]))<=0)
-          continue;
+       continue;
       //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;
+       minDistance = distance2Track;
+       iClosestTrack = iTrack;
       }
     }
 
@@ -307,7 +457,6 @@ void  AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * em
       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 ;
        fGeom->ImpactOnEmc(vtxCPV,vecP.Theta(),vecP.Phi(),dummyMod,zCPV,xCPV) ;
       }
     }
@@ -446,11 +595,16 @@ void  AliPHOSTrackSegmentMakerv1::MakePairs()
     Int_t iEmcRP ;
     for(iEmcRP = 0; iEmcRP < fEmcLast-fEmcFirst  ; iEmcRP++ ){
       if(emcExist[iEmcRP] > 0 ){
-       new ((*fTrackSegments)[fNTrackSegments])  
-         AliPHOSTrackSegment(dynamic_cast<AliPHOSEmcRecPoint *>(fEMCRecPoints->At(iEmcRP+fEmcFirst)), 
-                           nullpointer) ;
-       (dynamic_cast<AliPHOSTrackSegment *>(fTrackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
-       fNTrackSegments++;    
+        Int_t track = -1 ;
+        Float_t dx,dz ;
+        AliPHOSEmcRecPoint *emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(fEMCRecPoints->At(iEmcRP+fEmcFirst));
+        GetDistanceInPHOSPlane(emcclu, 0, track,dx,dz);
+        if(track<0)
+         new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment(emcclu,nullpointer) ;
+        else
+          new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment(emcclu,0,track,dx,dz);
+       (dynamic_cast<AliPHOSTrackSegment *>(fTrackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
+       fNTrackSegments++;    
       } 
     }
   }
@@ -482,7 +636,7 @@ void AliPHOSTrackSegmentMakerv1::Clusters2TrackSegments(Option_t *option)
   fTrackSegments->Clear();
 
   //   if(!ReadRecPoints(ievent))   continue; //reads RecPoints for event ievent
-    
+
   for(fModule = 1; fModule <= fGeom->GetNModules() ; fModule++ ) {
     FillOneModule() ; 
     MakeLinks() ;
@@ -535,3 +689,4 @@ void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option)
     }  
   }
 }
+