]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONTrack.cxx
Change needed on solarisCC5
[u/mrichter/AliRoot.git] / MUON / AliMUONTrack.cxx
index 88a659f6fa86c4a341a2b00a0a421807b8a1fbaf..d490e53512558d0d64f8dd22e989f7f2c62c1b63 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>
@@ -39,6 +38,8 @@
 #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);
@@ -52,8 +53,22 @@ ClassImp(AliMUONTrack) // Class implementation in ROOT context
 
 TVirtualFitter* AliMUONTrack::fgFitter = NULL; 
 
+  //__________________________________________________________________________
+AliMUONTrack::AliMUONTrack()
+  : TObject() 
+{
+  // Default constructor
+  fgFitter = 0;
+  fEventReconstructor = 0;
+  fTrackHitsPtr = new TObjArray(10);
+  fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);  
+  fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10); 
+  fTrackID = 0;
+}
+
   //__________________________________________________________________________
 AliMUONTrack::AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegment, AliMUONEventReconstructor* EventReconstructor)
+  : TObject()
 {
   // Constructor from two Segment's
   fEventReconstructor = EventReconstructor; // link back to EventReconstructor
@@ -64,16 +79,22 @@ 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)
+  : TObject()
 {
   // Constructor from one Segment and one HitForRec
   fEventReconstructor = EventReconstructor; // link back to EventReconstructor
@@ -84,11 +105,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 +123,114 @@ AliMUONTrack::~AliMUONTrack()
 {
   // Destructor
   if (fTrackHitsPtr) {
+    fTrackHitsPtr->Clear();
     delete fTrackHitsPtr; // delete the TObjArray of pointers to TrackHit's
     fTrackHitsPtr = NULL;
   }
+  
+  if (fTrackParamAtHit) {
+    // delete the TClonesArray of pointers to TrackParam
+    fTrackParamAtHit->Clear();
+    delete fTrackParamAtHit;
+    fTrackParamAtHit = NULL;
+  }
+
+  if (fHitForRecAtHit) {
+    // delete the TClonesArray of pointers to HitForRec
+    fHitForRecAtHit->Clear();
+    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;
+  //fEventReconstructor = new AliMUONEventReconstructor(*MUONTrack.fEventReconstructor);
+                               // is it right ?
+                              // NO, because it would use dummy copy constructor
+                              // and AliMUONTrack is not the owner of its EventReconstructor 
+  fEventReconstructor = theMUONTrack.fEventReconstructor;
+  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);
+  }
+
+  // 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);
+
+  // fEventReconstructor =  new AliMUONEventReconstructor(*MUONTrack.fEventReconstructor); // is it right ?
+                               // is it right ? NO because it would use dummy copy constructor
+  fEventReconstructor =  theMUONTrack.fEventReconstructor;
+  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);
+  }
+
+  // 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;
 }
 
@@ -235,9 +337,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 +401,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 =  fEventReconstructor->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()
 {
@@ -748,15 +930,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 */