#include "AliMUONConstants.h"
#include "AliMUONESDInterface.h"
#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 "AliMUONVDigit.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"
+#include "AliMUONGeometryTransformer.h"
+#include "AliMUONTriggerCircuit.h"
+#include "AliMUONVTrackReconstructor.h"
+#include "AliMUONVTriggerStore.h"
+#include "AliMUONCalibrationData.h"
+#include "AliMUONTriggerElectronics.h"
+
+
+#include "TGeoManager.h"
#include <TFile.h>
#include <TTree.h>
#include <TParticlePDG.h>
#include <Riostream.h>
+#include "AliMUONRecoCheck.h"
+
/// \cond CLASSIMP
ClassImp(AliMUONRecoCheck)
/// \endcond
fCurrentEvent(0),
fTrackRefStore(0x0),
fRecoTrackRefStore(0x0),
+fRecoTriggerRefStore(0x0),
fRecoTrackStore(0x0),
+fRecoTriggerTrackStore(0x0),
+fGeometryTransformer(0x0),
+fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
fESDEventOwner(kTRUE)
{
/// Normal ctor
fCurrentEvent(0),
fTrackRefStore(0x0),
fRecoTrackRefStore(0x0),
+fRecoTriggerRefStore(0x0),
fRecoTrackStore(0x0),
+fRecoTriggerTrackStore(0x0),
+fGeometryTransformer(0x0),
+fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
fESDEventOwner(kFALSE)
{
/// Normal ctor
// ESD MUON Tracks
fESDEvent = esdEvent;
-
}
//_____________________________________________________________________________
if (fESDFile) fESDFile->Close();
}
ResetStores();
+ delete fGeometryTransformer;
+ delete fTriggerCircuit;
+ delete fTriggerElectronics;
+ delete fCalibrationData;
}
//_____________________________________________________________________________
/// Deletes all the store objects that have been created and resets the pointers to 0x0
delete fTrackRefStore; fTrackRefStore = 0x0;
delete fRecoTrackRefStore; fRecoTrackRefStore = 0x0;
+ delete fRecoTriggerRefStore; fRecoTriggerRefStore = 0x0;
delete fRecoTrackStore; fRecoTrackStore = 0x0;
+ delete fRecoTriggerTrackStore; fRecoTriggerTrackStore = 0x0;
+}
+
+//_____________________________________________________________________________
+Bool_t AliMUONRecoCheck::InitCircuit()
+{
+
+ if ( fTriggerCircuit ) return kTRUE;
+
+ if ( ! InitGeometryTransformer() ) return kFALSE;
+
+ // reset tracker for local trigger to trigger track conversion
+ if ( ! AliMUONESDInterface::GetTracker() ) AliMUONESDInterface::ResetTracker();
+
+ fTriggerCircuit = new AliMUONTriggerCircuit(fGeometryTransformer);
+
+ return kTRUE;
}
+
//_____________________________________________________________________________
Int_t AliMUONRecoCheck::GetRunNumber()
{
/// Return the run number of the current ESD event
- if (fESDEventOwner && fRecoTrackStore == 0x0) {
+ if (fESDEventOwner && fRecoTrackStore == 0x0 && fRecoTriggerTrackStore == 0x0) {
if (!fESDTree || fESDTree->GetEvent(fCurrentEvent) <= 0) {
AliError(Form("fails to read ESD object for event %d: cannot get the run number",fCurrentEvent));
return -1;
}
}
+
+//_____________________________________________________________________________
+AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggeredTracks(Int_t event)
+{
+ /// Return a track store containing the reconstructed trigger tracks (converted into
+ /// MUONTriggerTrack objects) for a given event.
+
+ if (!fESDEventOwner) {
+ if (fRecoTriggerTrackStore == 0x0) MakeTriggeredTracks();
+ return fRecoTriggerTrackStore;
+ }
+
+ if (event != fCurrentEvent) {
+ ResetStores();
+ fCurrentEvent = event;
+ }
+
+ if (fRecoTriggerTrackStore != 0x0) return fRecoTriggerTrackStore;
+ else {
+ if (!fESDTree) return 0x0;
+ if (fESDTree->GetEvent(event) <= 0) {
+ AliError(Form("fails to read ESD object for event %d", event));
+ return 0x0;
+ }
+ MakeTriggeredTracks();
+ return fRecoTriggerTrackStore;
+ }
+}
+
+
//_____________________________________________________________________________
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;
}
}
//_____________________________________________________________________________
-AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t requestedStationMask, Bool_t request2ChInSameSt45)
+AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggerableTracks(Int_t event)
+{
+ /// Return a trigger track store containing the triggerable track references (converted into
+ /// AliMUONTriggerTrack objects) for a given event
+
+ if (!fESDEventOwner) {
+ if (fRecoTriggerRefStore == 0x0) MakeTriggerableTracks();
+ return fRecoTriggerRefStore;
+ }
+
+ if (event != fCurrentEvent) {
+ ResetStores();
+ fCurrentEvent = event;
+ }
+
+ if (fRecoTriggerRefStore != 0x0) return fRecoTriggerRefStore;
+ else {
+ if (!fMCEventHandler->LoadEvent(event)) {
+ AliError(Form("fails to read MC objects for event %d", event));
+ return 0x0;
+ }
+ MakeTriggerableTracks();
+ return fRecoTriggerRefStore;
+ }
+}
+
+
+//_____________________________________________________________________________
+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.
if (!fESDEventOwner) {
if (fRecoTrackRefStore == 0x0) {
if (TrackRefs(event) == 0x0) return 0x0;
- MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45);
+ MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
}
return fRecoTrackRefStore;
}
if (fRecoTrackRefStore != 0x0) return fRecoTrackRefStore;
else {
if (TrackRefs(event) == 0x0) return 0x0;
- MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45);
+ MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
return fRecoTrackRefStore;
}
}
+
//_____________________________________________________________________________
void AliMUONRecoCheck::MakeReconstructedTracks(Bool_t refit)
{
}
+
+//_____________________________________________________________________________
+void AliMUONRecoCheck::MakeTriggeredTracks()
+{
+ /// Make reconstructed trigger tracks
+ if (!(fRecoTriggerTrackStore = AliMUONESDInterface::NewTriggerTrackStore())) return;
+
+ AliMUONVTriggerStore* tmpTriggerStore = AliMUONESDInterface::NewTriggerStore();
+ if ( ! tmpTriggerStore ) return;
+
+ // loop over all reconstructed tracks and add them to the store (include ghosts)
+ Int_t nTracks = (Int_t) fESDEvent->GetNumberOfMuonTracks();
+ for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
+ AliESDMuonTrack* esdTrack = fESDEvent->GetMuonTrack(iTrack);
+ if (esdTrack->ContainTriggerData()) AliMUONESDInterface::Add(*esdTrack, *tmpTriggerStore);
+ }
+
+ if ( InitCircuit() ) {
+
+ AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
+ tracker->EventReconstructTrigger(*fTriggerCircuit, *tmpTriggerStore, *fRecoTriggerTrackStore);
+ }
+
+ delete tmpTriggerStore;
+}
+
+//_____________________________________________________________________________
+Bool_t AliMUONRecoCheck::TriggerToTrack(const AliMUONLocalTrigger& locTrg, AliMUONTriggerTrack& triggerTrack)
+{
+ /// Make trigger track from local trigger info
+ if ( ! InitCircuit() ) return kFALSE;
+ AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
+ return tracker->TriggerToTrack(*fTriggerCircuit, locTrg, triggerTrack);
+}
+
+
//_____________________________________________________________________________
void AliMUONRecoCheck::MakeTrackRefs()
{
TParticle* particle;
TClonesArray* trackRefs;
Int_t nTrackRef = fMCEventHandler->MCEvent()->GetNumberOfTracks();
- AliMUONVClusterStore* cStore = AliMUONESDInterface::NewClusterStore();
- if (!cStore) return;
- AliMUONVCluster* hit = cStore->CreateCluster(0,0,0);
+ AliMUONVCluster* hit = AliMUONESDInterface::NewCluster();
// loop over simulated tracks
for (Int_t iTrackRef = 0; iTrackRef < nTrackRef; ++iTrackRef) {
CleanMuonTrackRef(tmpTrackRefStore);
delete hit;
- delete cStore;
delete tmpTrackRefStore;
}
+//_____________________________________________________________________________
+void AliMUONRecoCheck::MakeTriggerableTracks()
+{
+ /// Make triggerable tracks
+ if (!(fRecoTriggerRefStore = AliMUONESDInterface::NewTriggerTrackStore()))
+ return;
+
+ Double_t x, y, z, slopeX, slopeY, pZ, xLoc, yLoc, zLoc;
+ TParticle* particle;
+ TClonesArray* trackRefs;
+ Int_t nTrackRef = fMCEventHandler->MCEvent()->GetNumberOfTracks();
+
+ // loop over simulated tracks
+ for (Int_t iTrackRef = 0; iTrackRef < nTrackRef; ++iTrackRef) {
+ Int_t nHits = fMCEventHandler->GetParticleAndTR(iTrackRef, particle, trackRefs);
+
+ // skip empty trackRefs
+ if (nHits < 1) continue;
+
+ AliMUONTriggerTrack track;
+ AliMUONDigitStoreV2S digitStore;
+ Int_t hitsOnTrigger = 0;
+ Int_t currCh = -1;
+
+ // loop over simulated track hits
+ for (Int_t iHit = 0; iHit < nHits; ++iHit) {
+ AliTrackReference* trackReference = static_cast<AliTrackReference*>(trackRefs->UncheckedAt(iHit));
+
+ // skip trackRefs not in MUON
+ if (trackReference->DetectorId() != AliTrackReference::kMUON) continue;
+
+ // check chamberId of current trackReference
+ 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);
+
+ AliMUONVDigit* digit = digitStore.Add(detElemId, nboard, pad.GetLocalBoardChannel(0),
+ cath, AliMUONVDigitStore::kDeny);
+ if (digit) {
+ digit->SetPadXY(ix,iy);
+ digit->SetCharge(1.);
+ }
+ }
+ }
+
+ if ( hitsOnTrigger == 0 ) {
+ pZ = trackReference->Pz();
+ slopeX = ( pZ == 0. ) ? 99999. : trackReference->Px() / pZ;
+ slopeY = ( pZ == 0. ) ? 99999. : trackReference->Py() / pZ;
+
+ track.SetX11(x);
+ track.SetY11(y);
+ track.SetZ11(z);
+ track.SetSlopeX(slopeX);
+ track.SetSlopeY(slopeY);
+ }
+
+ if ( currCh != chamberId ) {
+ hitsOnTrigger++;
+ currCh = chamberId;
+ }
+
+ } // loop on hits
+
+ 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);
+ }
+}
+
+
//_____________________________________________________________________________
void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefStore)
{
Double_t maxGasGap = 1.; // cm
Double_t x, y, z, pX, pY, pZ, x1, y1, z1, pX1, pY1, pZ1, z2;
Double_t bendingSlope,nonBendingSlope,inverseBendingMomentum;
- AliMUONVClusterStore* cStore = AliMUONESDInterface::NewClusterStore();
- if (!cStore) return;
- AliMUONVCluster* hit = cStore->CreateCluster(0,0,0);
+ AliMUONVCluster* hit = AliMUONESDInterface::NewCluster();
// create iterator
TIter next(tmpTrackRefStore->CreateIterator());
}
delete hit;
- delete cStore;
}
//_____________________________________________________________________________
-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);
+
+ }
+
+ }
+
}
}
}
+
+//_____________________________________________________________________________
+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).
+ /// Matching is done by comparing cluster/TrackRef positions.
+
+ AliMUONTriggerTrack *matchedTrack = 0x0;
+
+ // look for the corresponding simulated track if any
+ TIter next(triggerTrackStore.CreateIterator());
+ AliMUONTriggerTrack* track2;
+ while ( ( track2 = static_cast<AliMUONTriggerTrack*>(next()) ) ) {
+ // check compatibility
+ if (track.Match(*track2, sigmaCut)) {
+ matchedTrack = track2;
+ break;
+ }
+ }
+
+ return matchedTrack;
+
+}
+
+
+//____________________________________________________________________________
+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()) return kFALSE;
+
+ 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());
+}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff