]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONRecoCheck.cxx
Remove TClonesArray deletions (Philippe Pillot)
[u/mrichter/AliRoot.git] / MUON / AliMUONRecoCheck.cxx
index de76f24d90239436be2c1f9dd3da2eadc4b8c8ae..aa8e2dc7094bae387b208596037d525fdf5eb4c6 100644 (file)
@@ -43,6 +43,8 @@
 #include "AliMUONTrackParam.h"
 #include "AliMUONTriggerTrack.h"
 #include "AliMUONVTriggerTrackStore.h"
+#include "AliMUONTriggerStoreV1.h"
+#include "AliMUONLocalTrigger.h"
 #include "AliMCEventHandler.h"
 #include "AliMCEvent.h"
 #include "AliStack.h"
 #include "AliLog.h"
 #include "AliESDEvent.h"
 #include "AliESDMuonTrack.h"
+#include "AliMUONDigitStoreV2S.h"
+#include "AliMUONDigit.h"
+#include "AliMpVSegmentation.h"
+#include "AliMpSegmentation.h"
+#include "AliMpPad.h"
 
 #include "AliGeomManager.h"
+#include "AliCDBManager.h"
 #include "AliMpCDB.h"
 #include "AliMpDDLStore.h"
 #include "AliMUONCDB.h"
@@ -59,6 +67,9 @@
 #include "AliMUONTriggerCircuit.h"
 #include "AliMUONVTrackReconstructor.h"
 #include "AliMUONVTriggerStore.h"
+#include "AliMUONCalibrationData.h"
+#include "AliMUONTriggerElectronics.h"
+
 
 #include "TGeoManager.h"
 
@@ -89,6 +100,8 @@ fRecoTrackStore(0x0),
 fRecoTriggerTrackStore(0x0),
 fGeometryTransformer(0x0),
 fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
 fESDEventOwner(kTRUE)
 {
   /// Normal ctor
@@ -129,6 +142,8 @@ fRecoTrackStore(0x0),
 fRecoTriggerTrackStore(0x0),
 fGeometryTransformer(0x0),
 fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
 fESDEventOwner(kFALSE)
 {
   /// Normal ctor
@@ -152,6 +167,8 @@ AliMUONRecoCheck::~AliMUONRecoCheck()
   ResetStores();
   delete fGeometryTransformer;
   delete fTriggerCircuit;
+  delete fTriggerElectronics;
+  delete fCalibrationData;
 }
 
 //_____________________________________________________________________________
@@ -170,17 +187,8 @@ Bool_t AliMUONRecoCheck::InitCircuit()
 {
 
   if ( fTriggerCircuit ) return kTRUE;
-
-  if ( !AliMUONCDB::CheckOCDB() ) return kFALSE;
-
-  if ( !AliGeomManager::GetGeometry() )
-    AliGeomManager::LoadGeometry();
-
-  if ( !AliMpDDLStore::Instance(false) )
-    AliMpCDB::LoadDDLStore();
-       
-  fGeometryTransformer = new AliMUONGeometryTransformer();
-  fGeometryTransformer->LoadGeometryData();
+  
+  if ( ! InitGeometryTransformer() ) return kFALSE;
   
   fTriggerCircuit = new AliMUONTriggerCircuit(fGeometryTransformer);
 
@@ -293,7 +301,7 @@ AliMUONVTrackStore* AliMUONRecoCheck::TrackRefs(Int_t event)
   
   if (fTrackRefStore != 0x0) return fTrackRefStore;
   else {
-    if (!fMCEventHandler->GetEvent(event)) {
+    if (!fMCEventHandler->LoadEvent(event)) {
       AliError(Form("fails to read MC objects for event %d", event));
       return 0x0;
     }
@@ -320,7 +328,7 @@ AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggerableTracks(Int_t event)
        
   if (fRecoTriggerRefStore != 0x0) return fRecoTriggerRefStore;
   else {
-    if (!fMCEventHandler->GetEvent(event)) {
+    if (!fMCEventHandler->LoadEvent(event)) {
       AliError(Form("fails to read MC objects for event %d", event));
       return 0x0;
     }
@@ -331,7 +339,8 @@ AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggerableTracks(Int_t event)
 
 
 //_____________________________________________________________________________
-AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t requestedStationMask, Bool_t request2ChInSameSt45)
+AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t requestedStationMask,
+                                                           Bool_t request2ChInSameSt45, Bool_t hitInFrontOfPad)
 {
   /// Return a track store containing the reconstructible tracks for a given event,
   /// according to the mask of requested stations and the minimum number of chambers hit in stations 4 & 5.
@@ -339,7 +348,7 @@ AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t
   if (!fESDEventOwner) {
     if (fRecoTrackRefStore == 0x0) {
       if (TrackRefs(event) == 0x0) return 0x0;
-      MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45);
+      MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
     }
     return fRecoTrackRefStore;
   }
@@ -352,7 +361,7 @@ AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t
   if (fRecoTrackRefStore != 0x0) return fRecoTrackRefStore;
   else {
     if (TrackRefs(event) == 0x0) return 0x0;
-    MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45);
+    MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
     return fRecoTrackRefStore;
   }
 }
@@ -390,10 +399,11 @@ void AliMUONRecoCheck::MakeTriggeredTracks()
     if (esdTrack->ContainTriggerData()) AliMUONESDInterface::Add(*esdTrack, *tmpTriggerStore);
   }
        
-  if ( ! InitCircuit() ) return;
+  if ( InitCircuit() ) {
   
-  AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
-  tracker->EventReconstructTrigger(*fTriggerCircuit, *tmpTriggerStore, *fRecoTriggerTrackStore);
+    AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
+    tracker->EventReconstructTrigger(*fTriggerCircuit, *tmpTriggerStore, *fRecoTriggerTrackStore);
+  }
   
   delete tmpTriggerStore;
 }
@@ -540,7 +550,7 @@ void AliMUONRecoCheck::MakeTriggerableTracks()
  if (!(fRecoTriggerRefStore = AliMUONESDInterface::NewTriggerTrackStore()))
    return;
 
-  Double_t x, y, z, slopeX, slopeY, pZ;
+  Double_t x, y, z, slopeX, slopeY, pZ, xLoc, yLoc, zLoc;
   TParticle* particle;
   TClonesArray* trackRefs;
   Int_t nTrackRef = fMCEventHandler->MCEvent()->GetNumberOfTracks();
@@ -553,6 +563,7 @@ void AliMUONRecoCheck::MakeTriggerableTracks()
     if (nHits < 1) continue;
                                
     AliMUONTriggerTrack track;
+    AliMUONDigitStoreV2S digitStore;
     Int_t hitsOnTrigger = 0;
     Int_t currCh = -1;
                
@@ -567,13 +578,39 @@ void AliMUONRecoCheck::MakeTriggerableTracks()
       Int_t detElemId = trackReference->UserId();
       Int_t chamberId = detElemId / 100 - 1;
       if (chamberId < AliMUONConstants::NTrackingCh() || chamberId >= AliMUONConstants::NCh() ) continue;
-                       
-                       
+
+      // Get track parameters of current hit
+      x = trackReference->X();
+      y = trackReference->Y();
+      z = trackReference->Z();
+      
+      if ( InitTriggerResponse() ) {
+        fGeometryTransformer->Global2Local(detElemId, x, y, z, xLoc, yLoc, zLoc);
+      
+        Int_t nboard = 0;
+        for ( Int_t cath = AliMp::kCath0; cath <= AliMp::kCath1; ++cath )
+        {
+          const AliMpVSegmentation* seg 
+          = AliMpSegmentation::Instance()
+          ->GetMpSegmentation(detElemId,AliMp::GetCathodType(cath));
+        
+          AliMpPad pad = seg->PadByPosition(xLoc,yLoc,kFALSE);
+          Int_t ix = pad.GetIx();
+          Int_t iy = pad.GetIy();
+        
+          if ( !pad.IsValid() ) continue;
+        
+          if ( cath == AliMp::kCath0 ) nboard = pad.GetLocalBoardId(0);
+        
+          AliMUONDigit* digit = new AliMUONDigit(detElemId,nboard,
+                                                 pad.GetLocalBoardChannel(0),cath);
+          digit->SetPadXY(ix,iy);
+          digit->SetCharge(1.);
+          digitStore.Add(*digit,AliMUONVDigitStore::kDeny);
+        }
+      }
+    
       if ( hitsOnTrigger == 0 ) {
-        // Get track parameters of current hit
-        x = trackReference->X();
-        y = trackReference->Y();
-        z = trackReference->Z();
         pZ = trackReference->Pz();
         slopeX = ( pZ == 0. ) ? 99999. : trackReference->Px() / pZ;
         slopeY = ( pZ == 0. ) ? 99999. : trackReference->Py() / pZ;
@@ -591,12 +628,32 @@ void AliMUONRecoCheck::MakeTriggerableTracks()
       }
 
     } // loop on hits
-               
-    if ( hitsOnTrigger >= 3 ){
-      // store the track
-      track.SetUniqueID(iTrackRef);
-      fRecoTriggerRefStore->Add(track);
+    
+    if ( hitsOnTrigger < 3 ) continue;
+
+    // Check if the track passes the trigger algorithm
+    if ( InitTriggerResponse() ) {
+      AliMUONTriggerStoreV1 triggerStore;
+      fTriggerElectronics->Digits2Trigger(digitStore,triggerStore);
+    
+      TIter next(triggerStore.CreateIterator());
+      AliMUONLocalTrigger* locTrg(0x0);
+    
+      Int_t ptCutLevel = 0;
+    
+      while ( ( locTrg = static_cast<AliMUONLocalTrigger*>(next()) ) )
+      {
+        if ( locTrg->IsTrigX() && locTrg->IsTrigY() ) 
+        {
+          ptCutLevel = TMath::Max(ptCutLevel, 1);
+          if ( locTrg->LoHpt() ) ptCutLevel = TMath::Max(ptCutLevel, 3);
+          else if ( locTrg->LoLpt() ) ptCutLevel = TMath::Max(ptCutLevel, 2);
+        } // board is fired 
+      } // end of loop on Local Trigger
+      track.SetPtCutLevel(ptCutLevel);
     }
+    track.SetUniqueID(iTrackRef);
+    fRecoTriggerRefStore->Add(track);
   }
 }
 
@@ -718,18 +775,52 @@ void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefSt
 }
 
 //_____________________________________________________________________________
-void AliMUONRecoCheck::MakeReconstructibleTracks(UInt_t requestedStationMask, Bool_t request2ChInSameSt45)
+void AliMUONRecoCheck::MakeReconstructibleTracks(UInt_t requestedStationMask, Bool_t request2ChInSameSt45,
+                                                Bool_t hitInFrontOfPad)
 {
   /// Isolate the reconstructible tracks
   if (!(fRecoTrackRefStore = AliMUONESDInterface::NewTrackStore())) return;
   
+  // need geometry transformer and segmentation to check if trackrefs are located in front of pad(s)
+  if (hitInFrontOfPad) {
+    if (!InitGeometryTransformer()) return;
+    if (!AliMpSegmentation::Instance(kFALSE) && !AliMUONCDB::LoadMapping(kTRUE)) return;
+  }
+  
   // create iterator on trackRef
   TIter next(fTrackRefStore->CreateIterator());
   
   // loop over trackRef and add reconstructible tracks to fRecoTrackRefStore
   AliMUONTrack* track;
   while ( ( track = static_cast<AliMUONTrack*>(next()) ) ) {
-    if (track->IsValid(requestedStationMask, request2ChInSameSt45)) fRecoTrackRefStore->Add(*track);
+    
+    // check if the track is valid as it is
+    if (track->IsValid(requestedStationMask, request2ChInSameSt45)) {
+      
+      if (hitInFrontOfPad) {
+       
+       AliMUONTrack trackTmp(*track);
+       
+       // loop over clusters and remove those which are not in front of pad(s)
+       AliMUONTrackParam *param = static_cast<AliMUONTrackParam*>(trackTmp.GetTrackParamAtCluster()->First());
+       while (param) {
+         AliMUONTrackParam *nextParam = static_cast<AliMUONTrackParam*>(trackTmp.GetTrackParamAtCluster()->After(param));
+         if (!IsHitInFrontOfPad(param)) trackTmp.RemoveTrackParamAtCluster(param);
+         param = nextParam;
+       }
+       
+       // add the track if it is still valid
+       if (trackTmp.IsValid(requestedStationMask, request2ChInSameSt45)) fRecoTrackRefStore->Add(trackTmp);
+       
+      } else {
+       
+       // add the track
+       fRecoTrackRefStore->Add(*track);
+       
+      }
+      
+    }
+      
   }
 
 }
@@ -784,7 +875,7 @@ AliMUONTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTrack &track, AliMUON
 
 
 //_____________________________________________________________________________
-AliMUONTriggerTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTriggerTrack &track, AliMUONVTriggerTrackStore &triggerTrackStore,
+AliMUONTriggerTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTriggerTrack &track, const AliMUONVTriggerTrackStore &triggerTrackStore,
                                                            Double_t sigmaCut)
 {
   /// Return the trigger track from the store matched with the given track (or 0x0).
@@ -796,7 +887,6 @@ AliMUONTriggerTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTriggerTrack &
   TIter next(triggerTrackStore.CreateIterator());
   AliMUONTriggerTrack* track2;
   while ( ( track2 = static_cast<AliMUONTriggerTrack*>(next()) ) ) {
-      
     // check compatibility
     if (track.Match(*track2, sigmaCut)) {
       matchedTrack = track2;
@@ -808,3 +898,78 @@ AliMUONTriggerTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTriggerTrack &
   
 }
 
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitTriggerResponse()
+{
+  /// Initialize trigger electronics
+  /// for building of triggerable tracks from MC
+  
+  if ( fTriggerElectronics ) return kTRUE;
+  
+  if ( ! InitGeometryTransformer() ) return kFALSE;
+  
+  if ( ! AliMpDDLStore::Instance(false) ) AliMpCDB::LoadDDLStore();
+  
+  if ( ! InitCalibrationData() ) return kFALSE;
+  
+  fTriggerElectronics = new AliMUONTriggerElectronics(fCalibrationData);
+  
+  return kTRUE;
+}
+
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitCalibrationData()
+{
+  /// Initialize calibration data
+  if ( ! fCalibrationData ) {
+    if ( !AliMUONCDB::CheckOCDB() ) return kFALSE;
+    fCalibrationData = new AliMUONCalibrationData(AliCDBManager::Instance()->GetRun());
+  }
+  return kTRUE;
+}
+
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitGeometryTransformer()
+{
+  /// Return geometry transformer
+  /// (create it if necessary)
+  if ( ! fGeometryTransformer ) {
+    
+    if ( !AliMUONCDB::CheckOCDB() ) return kFALSE;
+    
+    if (!AliGeomManager::GetGeometry()) {
+      AliGeomManager::LoadGeometry();
+      if (!AliGeomManager::GetGeometry()) return kFALSE;  
+      if (!AliGeomManager::ApplyAlignObjsFromCDB("MUON")) return kFALSE;
+    }
+    
+    fGeometryTransformer = new AliMUONGeometryTransformer();
+    fGeometryTransformer->LoadGeometryData();
+  }
+  
+  return kTRUE;
+}
+
+//________________________________________________________________________
+Bool_t AliMUONRecoCheck::IsHitInFrontOfPad(AliMUONTrackParam *param) const
+{
+  /// Return kTRUE if the hit is located in front of at least 1 pad
+  
+  Int_t deId = param->GetClusterPtr()->GetDetElemId();
+  
+  Double_t xL, yL, zL;
+  fGeometryTransformer->Global2Local(deId, param->GetNonBendingCoor(), param->GetBendingCoor(), param->GetZ(), xL, yL, zL);
+  
+  const AliMpVSegmentation* seg1 = AliMpSegmentation::Instance()->GetMpSegmentation(deId, AliMp::kCath0);
+  const AliMpVSegmentation* seg2 = AliMpSegmentation::Instance()->GetMpSegmentation(deId, AliMp::kCath1);
+  if (!seg1 || !seg2) return kFALSE;
+  
+  AliMpPad pad1 = seg1->PadByPosition(xL, yL, kFALSE);
+  AliMpPad pad2 = seg2->PadByPosition(xL, yL, kFALSE);
+  
+  return (pad1.IsValid() || pad2.IsValid());
+}
+