]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONRecoCheck.cxx
Adding bp/nbp info after manu Id.
[u/mrichter/AliRoot.git] / MUON / AliMUONRecoCheck.cxx
index 207a9c9fffd93726c13e97561e492c6b15bdfe48..0661717be3ce287e58e9ca1e8257a5436565150c 100644 (file)
 
 /* $Id$ */
 
-// -----------------------
-// Class AliMUONRecoCheck
-// -----------------------
-// Utility class to check the muon reconstruction. Reconstructed tracks are compared
-// to reference tracks. The reference tracks are built from AliTrackReference for the
-// hit in chamber (0..9) and from kinematics for the vertex parameters.     
+/// \class AliMUONRecoCheck
+/// \brief Utility class to check reconstruction
+/// Reconstructed tracks are compared to reference tracks. 
+/// The reference tracks are built from AliTrackReference for the
+/// hit in chamber (0..9) and from kinematics for the vertex parameters.     
 
-#include <TParticle.h>
+#include "AliMUON.h"
+#include "AliMUONRecoCheck.h"
+#include "AliMUONTrack.h"
+#include "AliMUONData.h"
+#include "AliMUONConstants.h"
 
 #include "AliRun.h" // for gAlice
-#include "AliLog.h" 
 #include "AliLoader.h" 
 #include "AliRunLoader.h" 
-#include "AliTrackReference.h"
 #include "AliHeader.h"
-#include "AliMC.h"
 #include "AliStack.h"
-#include "AliMUON.h"
-#include "AliMUONRecoCheck.h"
-#include "AliMUONTrack.h"
-#include "AliMUONData.h"
-#include "AliMUONConstants.h"
+#include "AliTrackReference.h"
+#include "AliLog.h" 
 
+#include <TParticle.h>
+
+/// \cond CLASSIMP
 ClassImp(AliMUONRecoCheck)
+/// \endcond
 
 //_____________________________________________________________________________
-AliMUONRecoCheck::AliMUONRecoCheck(Char_t *chLoader)
+  AliMUONRecoCheck::AliMUONRecoCheck(Char_t *chLoader)
+  : TObject(),
+  fRunLoader(0x0),
+  fMUONData(0x0),
+  fMuonTrackRef(0x0),
+  fTrackReco(0x0),
+  fReconstructibleTracks(0),
+  fRecoTracks(0),
+  fIsLoadConstructor(kTRUE)
 {
-/// Constructor
+/// Constructor using "galice.root",
+/// takes care of loading/unloading Kinematics, TrackRefs and Tracks
+
   fMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
 
-  // open the run loader
+  // run loader
   fRunLoader = AliRunLoader::Open(chLoader);
   if (!fRunLoader) {
-    AliError(Form("no run loader found in file %s","galice.root" ));
+    AliError(Form("no run loader found " ));
     return;
   }
-  // initialize loader's
+
+ // initialize loader   
   AliLoader *loader = fRunLoader->GetLoader("MUONLoader");
 
-  // initialize container
+  // container
   fMUONData  = new AliMUONData(loader,"MUON","MUON");
+  if (!fMUONData) {
+    AliError(Form("no MUONData found " ));
+    return;
+  }
 
-   // Loading AliRun master
-  if (fRunLoader->GetAliRun() == 0x0) fRunLoader->LoadgAlice();
-
-  fRunLoader->LoadKinematics("READ");
-  fRunLoader->LoadTrackRefs("READ");
+  fRunLoader->LoadKinematics("READ");   
+  fRunLoader->LoadTrackRefs("READ");    
   loader->LoadTracks("READ");
 
-  fReconstructibleTracks = 0; 
-  fRecoTracks = 0;
 }
 
-//____________________________________________________________________
-AliMUONRecoCheck::AliMUONRecoCheck(const AliMUONRecoCheck& rhs)
- : TObject(rhs)
+//_____________________________________________________________________________
+  AliMUONRecoCheck::AliMUONRecoCheck(AliRunLoader *runloader, AliMUONData *muondata)
+  : TObject(),
+  fRunLoader(0x0),
+  fMUONData(0x0),
+  fMuonTrackRef(0x0),
+  fTrackReco(0x0),
+  fReconstructibleTracks(0),
+  fRecoTracks(0),
+  fIsLoadConstructor(kFALSE)
 {
-/// Protected copy constructor
+/// Constructor from AliRunLoader and AliMUONData,
+/// does not load/unload Kinematics, TrackRefs and Tracks internally
+/// it should be done in the execution program (e.g. see MUONRecoCheck.C)
+
+  fMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
+
+  // run loader
+  fRunLoader = runloader;
+  if (!fRunLoader) {
+    AliError(Form("no run loader found " ));
+    return;
+  }
+
+  // container
+  fMUONData  = muondata;
+  if (!fMUONData) {
+    AliError(Form("no MUONData found " ));
+    return;
+  }
 
-  AliFatal("Not implemented.");
 }
 
+
 //_____________________________________________________________________________
 AliMUONRecoCheck::~AliMUONRecoCheck()
 {
 /// Destructor
 
-  fRunLoader->UnloadKinematics();
-  fRunLoader->UnloadTrackRefs();
-  fRunLoader->UnloadTracks();
-  fMuonTrackRef->Delete();
   delete fMuonTrackRef;
-  delete fMUONData;
-}
 
-//________________________________________________________________________
-AliMUONRecoCheck& AliMUONRecoCheck::operator = (const AliMUONRecoCheck& rhs)
-{
-/// Protected assignement operator
-
-  if (this == &rhs) return *this;
+  if(fIsLoadConstructor){
+    fRunLoader->UnloadKinematics();
+    fRunLoader->UnloadTrackRefs();
+    fRunLoader->UnloadTracks();
+    delete fMUONData;
+  }
 
-  AliFatal("Not implemented.");
-    
-  return *this;  
 }
 
 //_____________________________________________________________________________
@@ -133,13 +160,14 @@ void AliMUONRecoCheck::MakeTrackRef()
  
   track = trackSave = -999;
   Bool_t isNewTrack;
-  Int_t iHitMin, iChamber;
+  Int_t iHitMin, iChamber, detElemId;
 
   trackParam = new AliMUONTrackParam();
   hitForRec = new AliMUONHitForRec();
-  muonTrack = new AliMUONTrack();
 
+  Int_t max = fRunLoader->GetHeader()->Stack()->GetNtrack();
   for (Int_t iTrackRef  = 0; iTrackRef < nTrackRef; iTrackRef++) {
+
     branch->GetEntry(iTrackRef);
     
     iHitMin = 0;
@@ -148,6 +176,8 @@ void AliMUONRecoCheck::MakeTrackRef()
     if (!trackRefs->GetEntries()) continue; 
 
     while (isNewTrack) {
+
+      muonTrack = new AliMUONTrack();
       
       for (Int_t iHit = iHitMin; iHit < trackRefs->GetEntries(); iHit++) {
       
@@ -161,6 +191,13 @@ void AliMUONRecoCheck::MakeTrackRef()
 
        track = trackReference->GetTrack();
 
+       if(track >= max){
+         AliWarningStream()
+           << "Track ID " << track 
+           << " larger than max number of particles " << max << endl;
+         isNewTrack = kFALSE;
+         break;
+       }
        if (track != trackSave && iHit != 0) {
          iHitMin = iHit;
          trackSave = track;
@@ -171,7 +208,7 @@ void AliMUONRecoCheck::MakeTrackRef()
        trackParam->SetBendingCoor(y);
        trackParam->SetNonBendingCoor(x);
        trackParam->SetZ(z);
-       
+
        if (TMath::Abs(pZ) > 0) {
          bendingSlope = pY/pZ;
          nonBendingSlope = pX/pZ;
@@ -187,21 +224,25 @@ void AliMUONRecoCheck::MakeTrackRef()
        hitForRec->SetNonBendingCoor(x);
        hitForRec->SetZ(z);
        hitForRec->SetBendingReso2(0.0); 
-       hitForRec->SetNonBendingReso2(0.0);  
-       iChamber = AliMUONConstants::ChamberNumber(z);
+       hitForRec->SetNonBendingReso2(0.0);
+       detElemId = hitForRec->GetDetElemId();
+       if (detElemId) iChamber = detElemId / 100 - 1; 
+       else iChamber = AliMUONConstants::ChamberNumber(z);
        hitForRec->SetChamberNumber(iChamber);
 
-       muonTrack->AddTrackParamAtHit(trackParam);
+       muonTrack->AddTrackParamAtHit(trackParam,0);
        muonTrack->AddHitForRecAtHit(hitForRec);
        muonTrack->SetTrackID(track);
-       
+
        trackSave = track;
        if (iHit == trackRefs->GetEntries()-1) isNewTrack = kFALSE;
       }
-      
+
       // track parameters at vertex 
       particle = fRunLoader->GetHeader()->Stack()->Particle(muonTrack->GetTrackID());
+
       if (particle) {
+
        x = particle->Vx();
        y = particle->Vy();
        z = particle->Vz();
@@ -224,20 +265,18 @@ void AliMUONRecoCheck::MakeTrackRef()
       
        muonTrack->SetTrackParamAtVertex(trackParam);
       }
-      
+
       AddMuonTrackReference(muonTrack);
-      muonTrack->ResetTrackParamAtHit();
-      muonTrack->ResetHitForRecAtHit();
+      delete muonTrack;
+      muonTrack = NULL;
       
     } // end while isNewTrack
-    
   }
   
   CleanMuonTrackRef();
   
   ReconstructibleTracks();
 
-  delete muonTrack;
   delete trackParam;
   delete hitForRec;
 
@@ -248,10 +287,10 @@ TClonesArray* AliMUONRecoCheck::GetTrackReco()
 {
 /// Return TClonesArray of reconstructed tracks
 
-  GetMUONData()->ResetRecTracks();
-  GetMUONData()->SetTreeAddress("RT");
-  fTrackReco = GetMUONData()->RecTracks(); 
-  GetMUONData()->GetRecTracks();
+  fMUONData->ResetRecTracks();
+  fMUONData->SetTreeAddress("RT");
+  fTrackReco = fMUONData->RecTracks(); 
+  fMUONData->GetRecTracks();
   fRecoTracks = fTrackReco->GetEntriesFast();
   return fTrackReco;
 }
@@ -313,22 +352,21 @@ void AliMUONRecoCheck::CleanMuonTrackRef()
   Float_t bendingSlope2,nonBendingSlope2,bendingMomentum2;
   TClonesArray *newMuonTrackRef = new TClonesArray("AliMUONTrack", 10);
   Int_t iHit1;
-  Int_t iChamber = 0;
+  Int_t iChamber = 0, detElemId = 0;
   Int_t nRec = 0;
   Int_t nTrackHits = 0;
 
   hitForRec = new AliMUONHitForRec();
   trackParam = new AliMUONTrackParam();
-  trackNew = new AliMUONTrack();
 
   Int_t nTrackRef = fMuonTrackRef->GetEntriesFast();
-  
   for (Int_t index = 0; index < nTrackRef; index++) {
     track = (AliMUONTrack*)fMuonTrackRef->At(index);
     hitForRecAtHit = track->GetHitForRecAtHit();
     trackParamAtHit = track->GetTrackParamAtHit();
     trackParamAtVertex = track->GetTrackParamAtVertex();
     nTrackHits = hitForRecAtHit->GetEntriesFast();
+    trackNew = new AliMUONTrack();
     iHit1 = 0;
     while (iHit1 < nTrackHits) {
       hitForRec1 = (AliMUONHitForRec*) hitForRecAtHit->At(iHit1); 
@@ -361,6 +399,7 @@ void AliMUONRecoCheck::CleanMuonTrackRef()
          bendingMomentum2 = 1./trackParam2->GetInverseBendingMomentum();
        
        if ( TMath::Abs(zRec2-zRec1) < maxGasGap ) {
+
          nRec++;
          xRec += xRec2;
          yRec += yRec2;
@@ -382,7 +421,9 @@ void AliMUONRecoCheck::CleanMuonTrackRef()
       hitForRec->SetNonBendingCoor(xRec);
       hitForRec->SetBendingCoor(yRec);
       hitForRec->SetZ(zRec);
-      iChamber = AliMUONConstants::ChamberNumber(zRec);
+      detElemId = hitForRec->GetDetElemId();
+      if (detElemId) iChamber = detElemId / 100 - 1;
+      else iChamber = AliMUONConstants::ChamberNumber(zRec);
       hitForRec->SetChamberNumber(iChamber);
       hitForRec->SetBendingReso2(0.0); 
       hitForRec->SetNonBendingReso2(0.0); 
@@ -395,16 +436,16 @@ void AliMUONRecoCheck::CleanMuonTrackRef()
        trackParam->SetInverseBendingMomentum(1./bendingMomentum);
 
       trackNew->AddHitForRecAtHit(hitForRec);
-      trackNew->AddTrackParamAtHit(trackParam);
+      trackNew->AddTrackParamAtHit(trackParam,0);
       
       iHit1++;
     } // end iHit1
 
     trackNew->SetTrackID(track->GetTrackID());
     trackNew->SetTrackParamAtVertex(trackParamAtVertex);
-    {new ((*newMuonTrackRef)[newMuonTrackRef->GetEntriesFast()]) AliMUONTrack(*trackNew);}    
-    trackNew->ResetHitForRecAtHit();
-    trackNew->ResetTrackParamAtHit();
+    {new ((*newMuonTrackRef)[newMuonTrackRef->GetEntriesFast()]) AliMUONTrack(*trackNew);}   
+    delete trackNew;
+    trackNew = NULL;
     
   } // end trackRef
 
@@ -415,8 +456,6 @@ void AliMUONRecoCheck::CleanMuonTrackRef()
     AddMuonTrackReference(track);
   }
   
-
-  delete trackNew;
   delete hitForRec;
   delete trackParam;
   newMuonTrackRef->Delete();
@@ -453,16 +492,22 @@ void AliMUONRecoCheck::ReconstructibleTracks()
       iChamber = hitForRec->GetChamberNumber();
       if (iChamber < 0 || iChamber > 10) continue;
       isChamberInTrack[iChamber] = 1;
-      
-    }
-    // track is reconstructible if the particle is crossing every tracking chambers
+    } 
+    // track is reconstructible if the particle is depositing a hit
+    // in the following chamber combinations:
+
     isTrackOK = kTRUE;
-    for (Int_t ch = 0; ch < 10; ch++) {
-      if (!isChamberInTrack[ch]) isTrackOK = kFALSE;
-    }    if (isTrackOK) fReconstructibleTracks++;
+    if (!isChamberInTrack[0] && !isChamberInTrack[1]) isTrackOK = kFALSE;
+    if (!isChamberInTrack[2] && !isChamberInTrack[3]) isTrackOK = kFALSE;
+    if (!isChamberInTrack[4] && !isChamberInTrack[5]) isTrackOK = kFALSE;
+    Int_t nHitsInLastStations=0;
+    for (Int_t ch = 6; ch < AliMUONConstants::NTrackingCh(); ch++)
+      if (isChamberInTrack[ch]) nHitsInLastStations++; 
+    if(nHitsInLastStations < 3) isTrackOK = kFALSE;
+
+    if (isTrackOK) fReconstructibleTracks++;
     if (!isTrackOK) fMuonTrackRef->Remove(track); // remove non reconstructible tracks
   }
   fMuonTrackRef->Compress();
 }
 
-