Updating the effecftive multiplicity calculation using the official

[u/mrichter/AliRoot.git] / MUON / AliMUONTriggerTrack.cxx

diff --git a/MUON/AliMUONTriggerTrack.cxx b/MUON/AliMUONTriggerTrack.cxx
index aed040f47944c478b8b0a0fead669825e83a0ba2..13bd46c8f1a85aa0c6b6c67836b6d036509bfe09 100644 (file)
--- a/MUON/AliMUONTriggerTrack.cxx
+++ b/MUON/AliMUONTriggerTrack.cxx
@@ -30,6 +30,8 @@
 #include <Riostream.h>
 #include "AliLog.h"
 
+using std::endl;
+using std::cout;
 /// \cond CLASSIMP
 ClassImp(AliMUONTriggerTrack)
 /// \endcond
@@ -45,6 +47,7 @@ AliMUONTriggerTrack::AliMUONTriggerTrack()
     fSlopeY(0),
     floTrgNum(0),
     fGTPattern(0),
+    fPtCutLevel(0),
     fHitsPatternInTrigCh(0),
     fCovariances(0x0)
 {
@@ -52,7 +55,7 @@ AliMUONTriggerTrack::AliMUONTriggerTrack()
       AliDebug(5,Form("this=%p",this));
 }
 //__________________________________________________________________________
-AliMUONTriggerTrack::AliMUONTriggerTrack(Float_t x11, Float_t y11, Float_t z11, Float_t z21, Float_t slopeX, Float_t slopeY, Int_t loTrgNum, Long_t theGTPattern, UShort_t hitsPatternInTrigCh)
+AliMUONTriggerTrack::AliMUONTriggerTrack(Float_t x11, Float_t y11, Float_t z11, Float_t z21, Float_t slopeX, Float_t slopeY, Int_t loTrgNum, Long_t theGTPattern, Int_t ptCutLevel)
     : TObject(),
       fx11(x11),
       fy11(y11),
@@ -62,7 +65,8 @@ AliMUONTriggerTrack::AliMUONTriggerTrack(Float_t x11, Float_t y11, Float_t z11,
       fSlopeY(slopeY),
       floTrgNum(loTrgNum),
       fGTPattern(theGTPattern),
-      fHitsPatternInTrigCh(hitsPatternInTrigCh),
+      fPtCutLevel(ptCutLevel),
+      fHitsPatternInTrigCh(0),
       fCovariances(0x0)
 {
 /// ctor from local trigger output
@@ -93,6 +97,7 @@ AliMUONTriggerTrack::AliMUONTriggerTrack (const AliMUONTriggerTrack& theMUONTrig
       fSlopeY(theMUONTriggerTrack.fSlopeY),
       floTrgNum(theMUONTriggerTrack.floTrgNum),
       fGTPattern(theMUONTriggerTrack.fGTPattern),
+      fPtCutLevel(theMUONTriggerTrack.fPtCutLevel),
       fHitsPatternInTrigCh(theMUONTriggerTrack.fHitsPatternInTrigCh),
       fCovariances(0x0)
 {
@@ -138,6 +143,14 @@ theMUONTriggerTrack)
     return *this;
 }
 
+//__________________________________________________________________________
+void AliMUONTriggerTrack::Clear(Option_t* /*opt*/)
+{
+  /// Clean memory
+  delete fCovariances;
+  fCovariances = 0x0;
+}
+
 //__________________________________________________________________________
 void
 AliMUONTriggerTrack::Print(Option_t* opt) const
@@ -185,21 +198,43 @@ const TMatrixD& AliMUONTriggerTrack::GetCovariances() const
 }
 
 //__________________________________________________________________________
-Bool_t AliMUONTriggerTrack::Match(AliMUONTriggerTrack &track, Double_t sigmaCut) const
+Bool_t AliMUONTriggerTrack::Match(AliMUONTriggerTrack &track,
+                                 Double_t sigmaCut) const
 {
   /// Try to match this track with the given track. Matching conditions:
   /// - x, y position and y slope within sigmaCut
   
+  // Find the track with the covariances correctly set
+  // Extrapolate to the z of the other track
+  Bool_t hasCov1 = ( GetCovariances().NonZeros() != 0 );
+  Bool_t hasCov2 = ( track.GetCovariances().NonZeros() != 0 );
+
+  const AliMUONTriggerTrack* trackToExtrap = ( hasCov2 ) ? &track : this;
+  const AliMUONTriggerTrack* fixedTrack = ( hasCov2 ) ? this : &track;
+
   TMatrixD paramDiff(3,1);
-  Double_t deltaZ = GetZ11() - track.GetZ11();
-  paramDiff(0,0) = GetX11() - track.GetX11();
-  paramDiff(1,0) = GetY11() - ( track.GetY11() + track.GetSlopeY() * deltaZ );
-  paramDiff(2,0) = GetSlopeY() - track.GetSlopeY();
+  Double_t deltaZ = fixedTrack->GetZ11() - trackToExtrap->GetZ11();
+  paramDiff(0,0) = fixedTrack->GetX11() - trackToExtrap->GetX11();
+  paramDiff(1,0) = fixedTrack->GetY11() - ( trackToExtrap->GetY11() + trackToExtrap->GetSlopeY() * deltaZ );
+  paramDiff(2,0) = fixedTrack->GetSlopeY() - trackToExtrap->GetSlopeY();
   Double_t chi2 = 0.;
-  TMatrixD cov1(GetCovariances());
-  TMatrixD cov2(track.GetCovariances());
 
-  AliDebug(3, Form("this Y11 %f  track Y11: %f (Z11 %f)  -> %f (Z11 %f)", GetY11(), track.GetY11(), track.GetZ11(), track.GetY11() + track.GetSlopeY() * deltaZ, GetZ11()));
+  TMatrixD cov1(fixedTrack->GetCovariances());
+  TMatrixD cov2(trackToExtrap->GetCovariances());
+
+  // Extrapolate covariances to z
+  if ( deltaZ != 0 ) {
+    if ( hasCov1 || hasCov2 ){
+      TMatrixD jacob(3,3);
+      jacob.UnitMatrix();
+      jacob(1,2) = deltaZ;
+      TMatrixD tmp(trackToExtrap->GetCovariances(),TMatrixD::kMultTranspose,jacob);
+      TMatrixD tmp2(jacob,TMatrixD::kMult,tmp);
+      cov2 = tmp2;
+    }
+  }
+
+  AliDebug(3, Form("track1 Y11 %f  track2 Y11: %f (Z11 %f)  -> %f (Z11 %f)", fixedTrack->GetY11(), trackToExtrap->GetY11(), trackToExtrap->GetZ11(), trackToExtrap->GetY11() + trackToExtrap->GetSlopeY() * deltaZ, fixedTrack->GetZ11()));
 
   TMatrixD sumCov(cov1,TMatrixD::kPlus,cov2);
   if (sumCov.Determinant() != 0) {