]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONTrack.cxx
change class name: AliMUONEventReconstructor -> AliMUONTrackReconstructor
[u/mrichter/AliRoot.git] / MUON / AliMUONTrack.cxx
index bedca622275754e12c49905501468e3acca364a9..92b32b7354555bd310dbbcf1bb2e4ea5b6750d8b 100644 (file)
 //
 ///////////////////////////////////////////////////
 
-#include <Riostream.h> // for cout
 #include <stdlib.h> // for exit()
 
-#include <TClonesArray.h>
+#include <Riostream.h> // for cout
 #include <TMath.h>
 #include <TMatrixD.h>
 #include <TObjArray.h>
 #include <TVirtualFitter.h>
 
-#include "AliMUONEventReconstructor.h" 
+#include "AliLog.h"
+
+#include "AliMUONTrackReconstructor.h" 
 #include "AliMUONHitForRec.h" 
 #include "AliMUONSegment.h" 
 #include "AliMUONTrack.h"
 #include "AliMUONTrackHit.h"
+#include "AliMUONTriggerTrack.h"
+#include "AliMUONConstants.h"
 
 // Functions to be minimized with Minuit
 void TrackChi2(Int_t &NParam, Double_t *Gradient, Double_t &Chi2, Double_t *Param, Int_t Flag);
@@ -53,10 +56,24 @@ ClassImp(AliMUONTrack) // Class implementation in ROOT context
 TVirtualFitter* AliMUONTrack::fgFitter = NULL; 
 
   //__________________________________________________________________________
-AliMUONTrack::AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegment, AliMUONEventReconstructor* EventReconstructor)
+AliMUONTrack::AliMUONTrack()
+  : TObject() 
+{
+  // Default constructor
+  fgFitter = 0;
+  fTrackReconstructor = 0;
+  fTrackHitsPtr = new TObjArray(10);
+  fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);  
+  fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10); 
+  fTrackID = 0;
+}
+
+  //__________________________________________________________________________
+AliMUONTrack::AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegment, AliMUONTrackReconstructor* TrackReconstructor)
+  : TObject()
 {
   // Constructor from two Segment's
-  fEventReconstructor = EventReconstructor; // link back to EventReconstructor
+  fTrackReconstructor = TrackReconstructor; // link back to TrackReconstructor
   // memory allocation for the TObjArray of pointers to reconstructed TrackHit's
   fTrackHitsPtr = new TObjArray(10);
   fNTrackHits = 0;
@@ -64,19 +81,25 @@ AliMUONTrack::AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegmen
   AddSegment(EndSegment); // add hits from EndSegment
   fTrackHitsPtr->Sort(); // sort TrackHits according to increasing Z
   SetTrackParamAtVertex(); // set track parameters at vertex
+  fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
+  fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
   // set fit conditions...
   fFitMCS = 0;
   fFitNParam = 3;
   fFitStart = 1;
   fFitFMin = -1.0;
+  fMatchTrigger = kFALSE;
+  fChi2MatchTrigger = 0;
+  fTrackID = 0;
   return;
 }
 
   //__________________________________________________________________________
-AliMUONTrack::AliMUONTrack(AliMUONSegment* Segment, AliMUONHitForRec* HitForRec, AliMUONEventReconstructor* EventReconstructor)
+AliMUONTrack::AliMUONTrack(AliMUONSegment* Segment, AliMUONHitForRec* HitForRec, AliMUONTrackReconstructor* TrackReconstructor)
+  : TObject()
 {
   // Constructor from one Segment and one HitForRec
-  fEventReconstructor = EventReconstructor; // link back to EventReconstructor
+  fTrackReconstructor = TrackReconstructor; // link back to TrackReconstructor
   // memory allocation for the TObjArray of pointers to reconstructed TrackHit's
   fTrackHitsPtr = new TObjArray(10);
   fNTrackHits = 0;
@@ -84,11 +107,16 @@ AliMUONTrack::AliMUONTrack(AliMUONSegment* Segment, AliMUONHitForRec* HitForRec,
   AddHitForRec(HitForRec); // add HitForRec
   fTrackHitsPtr->Sort(); // sort TrackHits according to increasing Z
   SetTrackParamAtVertex(); // set track parameters at vertex
+  fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
+  fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
   // set fit conditions...
   fFitMCS = 0;
   fFitNParam = 3;
   fFitStart = 1;
   fFitFMin = -1.0;
+  fMatchTrigger = kFALSE;
+  fChi2MatchTrigger = 0;
+  fTrackID = 0;
   return;
 }
 
@@ -97,38 +125,115 @@ AliMUONTrack::~AliMUONTrack()
 {
   // Destructor
   if (fTrackHitsPtr) {
-    delete fTrackHitsPtr; // delete the TObjArray of pointers to TrackHit's
+    // delete the TObjArray of pointers to TrackHit's   
+    delete fTrackHitsPtr; 
     fTrackHitsPtr = NULL;
   }
+  
+  if (fTrackParamAtHit) {
+    // delete the TClonesArray of pointers to TrackParam
+    delete fTrackParamAtHit;
+    fTrackParamAtHit = NULL;
+  }
+
+  if (fHitForRecAtHit) {
+    // delete the TClonesArray of pointers to HitForRec
+    delete fHitForRecAtHit;
+    fHitForRecAtHit = NULL;
+  }
 }
 
   //__________________________________________________________________________
-AliMUONTrack::AliMUONTrack (const AliMUONTrack& MUONTrack):TObject(MUONTrack)
+AliMUONTrack::AliMUONTrack (const AliMUONTrack& theMUONTrack)
+  :  TObject(theMUONTrack)
 {
-  fEventReconstructor = new AliMUONEventReconstructor(*MUONTrack.fEventReconstructor);
-  fTrackParamAtVertex = MUONTrack.fTrackParamAtVertex;
-  fTrackHitsPtr =  new TObjArray(*MUONTrack.fTrackHitsPtr);
-  fNTrackHits =  MUONTrack.fNTrackHits;
-  fFitMCS     =  MUONTrack.fFitMCS;
-  fFitNParam  =  MUONTrack.fFitNParam;
-  fFitFMin    =  MUONTrack.fFitFMin;
-  fFitStart   =  MUONTrack.fFitStart;
+  //fTrackReconstructor = new AliMUONTrackReconstructor(*MUONTrack.fTrackReconstructor);
+                               // is it right ?
+                              // NO, because it would use dummy copy constructor
+                              // and AliMUONTrack is not the owner of its TrackReconstructor 
+  fTrackReconstructor = theMUONTrack.fTrackReconstructor;
+  fTrackParamAtVertex = theMUONTrack.fTrackParamAtVertex;
+
+ // necessary to make a copy of the objects and not only the pointers in TObjArray.
+  fTrackHitsPtr  =  new TObjArray(10);
+  for (Int_t index = 0; index < (theMUONTrack.fTrackHitsPtr)->GetEntriesFast(); index++) {
+    AliMUONTrackHit *trackHit = new AliMUONTrackHit(*(AliMUONTrackHit*)(theMUONTrack.fTrackHitsPtr)->At(index));
+    fTrackHitsPtr->Add(trackHit);
+  }
+  fTrackHitsPtr->SetOwner(); // nedeed for deleting TClonesArray
+
+  // necessary to make a copy of the objects and not only the pointers in TClonesArray.
+  fTrackParamAtHit  =  new TClonesArray("AliMUONTrackParam",10);
+  for (Int_t index = 0; index < (theMUONTrack.fTrackParamAtHit)->GetEntriesFast(); index++) {
+    {new ((*fTrackParamAtHit)[fTrackParamAtHit->GetEntriesFast()]) 
+       AliMUONTrackParam(*(AliMUONTrackParam*)(theMUONTrack.fTrackParamAtHit)->At(index));}
+  }
+
+  // necessary to make a copy of the objects and not only the pointers in TClonesArray.
+  fHitForRecAtHit  =  new TClonesArray("AliMUONHitForRec",10);
+  for (Int_t index = 0; index < (theMUONTrack.fHitForRecAtHit)->GetEntriesFast(); index++) {
+    {new ((*fHitForRecAtHit)[fHitForRecAtHit->GetEntriesFast()]) 
+       AliMUONHitForRec(*(AliMUONHitForRec*)(theMUONTrack.fHitForRecAtHit)->At(index));}
+  }
+
+  fNTrackHits       =  theMUONTrack.fNTrackHits;
+  fFitMCS           =  theMUONTrack.fFitMCS;
+  fFitNParam        =  theMUONTrack.fFitNParam;
+  fFitFMin          =  theMUONTrack.fFitFMin;
+  fFitStart         =  theMUONTrack.fFitStart;
+  fMatchTrigger     =  theMUONTrack.fMatchTrigger;
+  fChi2MatchTrigger =  theMUONTrack.fChi2MatchTrigger;
+  fTrackID          =  theMUONTrack.fTrackID;
 }
 
   //__________________________________________________________________________
-AliMUONTrack & AliMUONTrack::operator=(const AliMUONTrack& MUONTrack)
+AliMUONTrack & AliMUONTrack::operator=(const AliMUONTrack& theMUONTrack)
 {
-  if (this == &MUONTrack)
+
+  // check assignement to self
+  if (this == &theMUONTrack)
     return *this;
 
-  fEventReconstructor = new AliMUONEventReconstructor(*MUONTrack.fEventReconstructor);
-  fTrackParamAtVertex = MUONTrack.fTrackParamAtVertex;
-  fTrackHitsPtr =  new TObjArray(*MUONTrack.fTrackHitsPtr);
-  fNTrackHits =  MUONTrack.fNTrackHits;
-  fFitMCS     =  MUONTrack.fFitMCS;
-  fFitNParam  =  MUONTrack.fFitNParam;
-  fFitFMin    =  MUONTrack.fFitFMin;
-  fFitStart   =  MUONTrack.fFitStart;
+  // base class assignement
+  TObject::operator=(theMUONTrack);
+
+  // fTrackReconstructor =  new AliMUONTrackReconstructor(*MUONTrack.fTrackReconstructor); 
+  // is it right ?
+  // is it right ? NO because it would use dummy copy constructor
+  fTrackReconstructor =  theMUONTrack.fTrackReconstructor;
+  fTrackParamAtVertex =  theMUONTrack.fTrackParamAtVertex;
+
+ // necessary to make a copy of the objects and not only the pointers in TObjArray.
+  fTrackHitsPtr  =  new TObjArray(10);
+  for (Int_t index = 0; index < (theMUONTrack.fTrackHitsPtr)->GetEntriesFast(); index++) {
+    AliMUONTrackHit *trackHit = new AliMUONTrackHit(*(AliMUONTrackHit*)(theMUONTrack.fTrackHitsPtr)->At(index));
+    fTrackHitsPtr->Add(trackHit);
+  }
+  fTrackHitsPtr->SetOwner();  // nedeed for deleting TClonesArray
+
+  // necessary to make a copy of the objects and not only the pointers in TClonesArray.
+  fTrackParamAtHit  =  new TClonesArray("AliMUONTrackParam",10);
+  for (Int_t index = 0; index < (theMUONTrack.fTrackParamAtHit)->GetEntriesFast(); index++) {
+    {new ((*fTrackParamAtHit)[fTrackParamAtHit->GetEntriesFast()]) 
+       AliMUONTrackParam(*(AliMUONTrackParam*)(theMUONTrack.fTrackParamAtHit)->At(index));}
+  }
+
+  // necessary to make a copy of the objects and not only the pointers in TClonesArray.
+  fHitForRecAtHit  =  new TClonesArray("AliMUONHitForRec",10);
+  for (Int_t index = 0; index < (theMUONTrack.fHitForRecAtHit)->GetEntriesFast(); index++) {
+    {new ((*fHitForRecAtHit)[fHitForRecAtHit->GetEntriesFast()]) 
+       AliMUONHitForRec(*(AliMUONHitForRec*)(theMUONTrack.fHitForRecAtHit)->At(index));}
+  }
+
+  fNTrackHits         =  theMUONTrack.fNTrackHits;
+  fFitMCS             =  theMUONTrack.fFitMCS;
+  fFitNParam          =  theMUONTrack.fFitNParam;
+  fFitFMin            =  theMUONTrack.fFitFMin;
+  fFitStart           =  theMUONTrack.fFitStart;
+  fMatchTrigger       =  theMUONTrack.fMatchTrigger;
+  fChi2MatchTrigger   =  theMUONTrack.fChi2MatchTrigger;
+  fTrackID            =  theMUONTrack.fTrackID;
+
   return *this;
 }
 
@@ -139,7 +244,7 @@ void AliMUONTrack::Remove()
   // and corresponding track hits from array of track hits.
   // Compress the TClonesArray it belongs to.
   AliMUONTrackHit *nextTrackHit;
-  AliMUONEventReconstructor *eventRec = this->fEventReconstructor;
+  AliMUONTrackReconstructor *eventRec = this->fTrackReconstructor;
   TClonesArray *trackHitsPtr = eventRec->GetRecTrackHitsPtr();
   // Loop over all track hits of track
   AliMUONTrackHit *trackHit = (AliMUONTrackHit*) fTrackHitsPtr->First();
@@ -235,9 +340,48 @@ void AliMUONTrack::RecursiveDump(void) const
   }
   return;
 }
+  
+  //__________________________________________________________________________
+Bool_t* AliMUONTrack::CompatibleTrack(AliMUONTrack * Track, Double_t Sigma2Cut) const
+{
+  // Return kTRUE/kFALSE for each chamber if hit is compatible or not 
+  TClonesArray *hitArray, *thisHitArray;
+  AliMUONHitForRec *hit, *thisHit;
+  Int_t chamberNumber;
+  Float_t deltaZ;
+  Float_t deltaZMax = 1.; // 1 cm
+  Float_t chi2 = 0;
+  Bool_t *nCompHit = new Bool_t[AliMUONConstants::NTrackingCh()]; 
+
+  for ( Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
+    nCompHit[ch] = kFALSE;
+  }
+
+  thisHitArray = this->GetHitForRecAtHit();
+
+  hitArray =  Track->GetHitForRecAtHit();
+
+  for (Int_t iHthis = 0; iHthis < thisHitArray->GetEntriesFast(); iHthis++) {
+    thisHit = (AliMUONHitForRec*) thisHitArray->At(iHthis);
+    chamberNumber = thisHit->GetChamberNumber();
+    if (chamberNumber < 0 || chamberNumber > AliMUONConstants::NTrackingCh()) continue; 
+    nCompHit[chamberNumber] = kFALSE;
+    for (Int_t iH = 0; iH < hitArray->GetEntriesFast(); iH++) {
+      hit = (AliMUONHitForRec*) hitArray->At(iH);
+      deltaZ = TMath::Abs(thisHit->GetZ() - hit->GetZ());
+      chi2 = thisHit->NormalizedChi2WithHitForRec(hit,Sigma2Cut); // set cut to 4 sigmas
+      if (chi2 < 3. && deltaZ < deltaZMax) {
+       nCompHit[chamberNumber] = kTRUE;
+       break;
+      }
+    }  
+  }
+  
+  return nCompHit;
+}
 
   //__________________________________________________________________________
-Int_t AliMUONTrack::HitsInCommon(AliMUONTrack* Track)
+Int_t AliMUONTrack::HitsInCommon(AliMUONTrack* Track) const
 {
   // Returns the number of hits in common
   // between the current track ("this")
@@ -260,6 +404,47 @@ Int_t AliMUONTrack::HitsInCommon(AliMUONTrack* Track)
   return hitsInCommon;
 }
 
+  //__________________________________________________________________________
+void AliMUONTrack::MatchTriggerTrack(TClonesArray *triggerTrackArray)
+{
+  // Match this track with one trigger track if possible
+  AliMUONTrackParam trackParam; 
+  AliMUONTriggerTrack *triggerTrack;
+  Double_t xTrack, yTrack, ySlopeTrack, dTrigTrackMin2, dTrigTrack2;
+  Double_t nSigmaCut2;
+
+  Double_t distSigma[3]={1,1,0.02}; // sigma of distributions (trigger-track) X,Y,slopeY
+  Double_t distTriggerTrack[3] = {0,0,0};
+
+  fMatchTrigger = kFALSE;
+  fChi2MatchTrigger = 0;
+
+  trackParam = *((AliMUONTrackParam*) fTrackParamAtHit->Last()); 
+  trackParam.ExtrapToZ(AliMUONConstants::DefaultChamberZ(10)); // extrap to 1st trigger chamber
+
+  nSigmaCut2 =  fTrackReconstructor->GetMaxSigma2Distance(); // nb of sigma**2 for cut
+  xTrack = trackParam.GetNonBendingCoor();
+  yTrack = trackParam.GetBendingCoor();
+  ySlopeTrack = trackParam.GetBendingSlope();
+  dTrigTrackMin2 = 999;
+  
+  triggerTrack = (AliMUONTriggerTrack*) triggerTrackArray->First();
+  while(triggerTrack){
+    distTriggerTrack[0] = (triggerTrack->GetX11()-xTrack)/distSigma[0];
+    distTriggerTrack[1] = (triggerTrack->GetY11()-yTrack)/distSigma[1];
+    distTriggerTrack[2] = (TMath::Tan(triggerTrack->GetThetay())-ySlopeTrack)/distSigma[2];
+    dTrigTrack2 = 0;
+    for (Int_t iVar = 0; iVar < 3; iVar++)
+      dTrigTrack2 += distTriggerTrack[iVar]*distTriggerTrack[iVar];
+    if (dTrigTrack2 < dTrigTrackMin2 && dTrigTrack2 < nSigmaCut2) {
+      dTrigTrackMin2 = dTrigTrack2;
+      fMatchTrigger = kTRUE;
+      fChi2MatchTrigger =  dTrigTrack2/3.; // Normalized Chi2, 3 variables (X,Y,slopeY)
+    }
+    triggerTrack = (AliMUONTriggerTrack*) triggerTrackArray->After(triggerTrack);
+  }
+
+}
   //__________________________________________________________________________
 void AliMUONTrack::Fit()
 {
@@ -319,7 +504,7 @@ void AliMUONTrack::Fit()
   // minimization
   fgFitter->ExecuteCommand("MINIMIZE", arg, 0);
   // exit from Minuit
-  fgFitter->ExecuteCommand("EXIT", arg, 0); // necessary ????
+  //  fgFitter->ExecuteCommand("EXIT", arg, 0); // necessary ????
   // get results into "invBenP", "benC", "nonBenC" ("x", "y")
   fgFitter->GetParameter(0, parName, invBenP, errorParam, lower, upper);
   fgFitter->GetParameter(1, parName, benC, errorParam, lower, upper);
@@ -356,13 +541,14 @@ void AliMUONTrack::AddHitForRec(AliMUONHitForRec* HitForRec)
   // Add HitForRec to the track:
   // actual TrackHit into TClonesArray of TrackHit's for the event;
   // pointer to actual TrackHit in TObjArray of pointers to TrackHit's for the track
-  TClonesArray *recTrackHitsPtr = this->fEventReconstructor->GetRecTrackHitsPtr();
-  Int_t eventTrackHits = this->fEventReconstructor->GetNRecTrackHits();
+  TClonesArray *recTrackHitsPtr = this->fTrackReconstructor->GetRecTrackHitsPtr();
+  Int_t eventTrackHits = this->fTrackReconstructor->GetNRecTrackHits();
   // event
   AliMUONTrackHit* trackHit =
     new ((*recTrackHitsPtr)[eventTrackHits]) AliMUONTrackHit(HitForRec);
-  this->fEventReconstructor->SetNRecTrackHits(eventTrackHits + 1);
+  this->fTrackReconstructor->SetNRecTrackHits(eventTrackHits + 1);
   // track
+  if (fTrackHitsPtr->IsOwner()) AliFatal("fTrackHitsPtr is owner");
   fTrackHitsPtr->Add(trackHit);
   fNTrackHits++;
 }
@@ -404,12 +590,12 @@ void AliMUONTrack::SetTrackParamAtVertex()
     firstHit->GetBendingCoor() - bendingSlope * firstHit->GetZ(); // same if from firstHit and  lastHit ????
   // signed bending momentum
   Double_t signedBendingMomentum =
-    fEventReconstructor->GetBendingMomentumFromImpactParam(impactParam);
+    fTrackReconstructor->GetBendingMomentumFromImpactParam(impactParam);
   trackParam->SetInverseBendingMomentum(1.0 / signedBendingMomentum);
   // bending slope at vertex
   trackParam->
     SetBendingSlope(bendingSlope +
-                   impactParam / fEventReconstructor->GetSimpleBPosition());
+                   impactParam / fTrackReconstructor->GetSimpleBPosition());
   // non bending slope
   Double_t nonBendingSlope =
     (firstHit->GetNonBendingCoor() - lastHit->GetNonBendingCoor()) / deltaZ;
@@ -646,7 +832,7 @@ Double_t MultipleScatteringAngle2(AliMUONTrackHit *TrackHit)
   // thickness in radiation length for the current track,
   // taking local angle into account
   radiationLength =
-    trackBeingFitted->GetEventReconstructor()->GetChamberThicknessInX0() *
+    trackBeingFitted->GetTrackReconstructor()->GetChamberThicknessInX0() *
     TMath::Sqrt(1.0 +
                slopeBending * slopeBending + slopeNonBending * slopeNonBending);
   inverseBendingMomentum2 = 
@@ -748,15 +934,15 @@ L60:
             a[j + k*l] = a[k + j*l];
         }
     }
-    delete localVERTs;
-    delete localVERTq;
-    delete localVERTpp;
+    delete [] localVERTs;
+    delete [] localVERTq;
+    delete [] localVERTpp;
     return;
 //*-*-                  failure return
 L100:
-    delete localVERTs;
-    delete localVERTq;
-    delete localVERTpp;
+    delete [] localVERTs;
+    delete [] localVERTq;
+    delete [] localVERTpp;
     ifail = 1;
 } /* mnvertLocal */