X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=MUON%2FAliMUONTracker.cxx;h=462d4e3c46e0fd6364ec5021921fe629725e326c;hb=b0ac3c26228328f3f7067aaca8b384d1bdc5ec6a;hp=30053c969b6f288d28c32a1fcb7f21e861532d3f;hpb=9bf6860bdd8c933089cab44a94c95664875e5c5f;p=u%2Fmrichter%2FAliRoot.git diff --git a/MUON/AliMUONTracker.cxx b/MUON/AliMUONTracker.cxx index 30053c969b6..462d4e3c46e 100644 --- a/MUON/AliMUONTracker.cxx +++ b/MUON/AliMUONTracker.cxx @@ -22,11 +22,9 @@ /// reconstruct tracks from recpoints /// /// Actual tracking is performed by some AliMUONVTrackReconstructor children -/// Tracking modes (ORIGINAL, KALMAN) and associated options and parameters -/// can be changed by using: -/// AliMUONRecoParam *muonRecoParam = AliMUONRecoParam::GetLow(High)FluxParam(); -/// muonRecoParam->Set...(); // see methods in AliMUONRecoParam.h for details -/// AliMUONReconstructor::SetRecoParam(muonRecoParam); +/// Tracking modes (ORIGINAL, KALMAN) and associated options and parameters can be changed +/// through the AliMUONRecoParam object set in the reconstruction macro or read from the CDB +/// (see methods in AliMUONRecoParam.h file for details) /// /// \author Christian Finck and Laurent Aphecetche, SUBATECH Nantes //----------------------------------------------------------------------------- @@ -51,13 +49,16 @@ #include "AliMUONTrackReconstructorK.h" #include "AliMUONTrackStoreV1.h" #include "AliMUONTriggerTrackStoreV1.h" -#include "AliMUONVCluster.h" +#include "AliMUONTriggerTrack.h" +#include "AliMUONLocalTrigger.h" #include "AliMUONVClusterServer.h" #include "AliMUONVDigitStore.h" #include "AliMUONVTriggerStore.h" +#include "AliMUONTriggerUtilities.h" #include #include #include +#include "AliLog.h" /// \cond CLASSIMP ClassImp(AliMUONTracker) @@ -65,38 +66,41 @@ ClassImp(AliMUONTracker) //_____________________________________________________________________________ -AliMUONTracker::AliMUONTracker(AliMUONVClusterServer* clusterServer, - const AliMUONVDigitStore& digitStore, - const AliMUONDigitMaker* digitMaker, +AliMUONTracker::AliMUONTracker(const AliMUONRecoParam* recoParam, + AliMUONVClusterServer* clusterServer, + AliMUONVDigitStore& digitStore, const AliMUONGeometryTransformer* transformer, - const AliMUONTriggerCircuit* triggerCircuit) + const AliMUONTriggerCircuit* triggerCircuit, + const AliMUONTriggerUtilities* triggerUtilities) : AliTracker(), - fDigitMaker(digitMaker), // not owner - fTransformer(transformer), // not owner - fTriggerCircuit(triggerCircuit), // not owner - fTrackHitPatternMaker(0x0), - fTrackReco(0x0), - fClusterStore(0x0), - fTriggerStore(0x0), - fClusterServer(clusterServer), - fIsOwnerOfClusterServer(kFALSE), - fDigitStore(digitStore), // not owner - fInputClusterStore(0x0), - fTriggerTrackStore(0x0) +fkTransformer(transformer), // not owner +fkTriggerCircuit(triggerCircuit), // not owner +fTrackHitPatternMaker(0x0), +fTrackReco(0x0), +fClusterStore(0x0), +fTriggerStore(0x0), +fClusterServer(clusterServer), +fIsOwnerOfClusterServer(kFALSE), +fkDigitStore(digitStore), // not owner +fInputClusterStore(0x0), +fTriggerTrackStore(0x0), +fkRecoParam(recoParam), +fInternalTrackStore(0x0) { /// constructor - if (fTransformer && fDigitMaker) - fTrackHitPatternMaker = new AliMUONTrackHitPattern(*fTransformer,*fDigitMaker); + + if (fkTransformer) + fTrackHitPatternMaker = new AliMUONTrackHitPattern(recoParam,*fkTransformer,fkDigitStore,triggerUtilities); if (!fClusterServer) { - AliInfo("No cluster server given. Will use AliMUONLegacyClusterServer"); + AliDebug(1,"No cluster server given. Will use AliMUONLegacyClusterServer"); fIsOwnerOfClusterServer = kTRUE; } else { - TIter next(fDigitStore.CreateIterator()); - fClusterServer->UseDigits(next); + TIter next(fkDigitStore.CreateIterator()); + fClusterServer->UseDigits(next,&digitStore); SetupClusterServer(*fClusterServer); } @@ -106,6 +110,7 @@ AliMUONTracker::AliMUONTracker(AliMUONVClusterServer* clusterServer, AliMUONTracker::~AliMUONTracker() { /// dtor + delete fTrackReco; delete fTrackHitPatternMaker; delete fClusterStore; @@ -113,6 +118,7 @@ AliMUONTracker::~AliMUONTracker() if ( fIsOwnerOfClusterServer ) delete fClusterServer; delete fInputClusterStore; delete fTriggerTrackStore; + delete fInternalTrackStore; } //_____________________________________________________________________________ @@ -144,16 +150,15 @@ Int_t AliMUONTracker::LoadClusters(TTree* clustersTree) { /// Load triggerStore from clustersTree - delete fTriggerStore; - delete fInputClusterStore; - fInputClusterStore=0x0; - if ( ! clustersTree ) { AliFatal("No clustersTree"); return 1; } - fTriggerStore = AliMUONVTriggerStore::Create(*clustersTree); + if ( !fTriggerStore ) + { + fTriggerStore = AliMUONVTriggerStore::Create(*clustersTree); + } if (!fTriggerStore) { @@ -161,20 +166,32 @@ Int_t AliMUONTracker::LoadClusters(TTree* clustersTree) return 2; } - if ( fIsOwnerOfClusterServer ) + if (!fInputClusterStore) { fInputClusterStore = AliMUONVClusterStore::Create(*clustersTree); - if ( fInputClusterStore ) + if (!fInputClusterStore) { - AliInfo(Form("Created %s from cluster tree",fInputClusterStore->ClassName())); - fInputClusterStore->Clear(); - fInputClusterStore->Connect(*clustersTree,kFALSE); + AliError("Could not get clusterStore"); + return 3; } - delete fClusterServer; - fClusterServer = new AliMUONLegacyClusterServer(*fTransformer,fInputClusterStore); - SetupClusterServer(*fClusterServer); + AliDebug(1,Form("Created %s from cluster tree",fInputClusterStore->ClassName())); + } + + if ( !fClusterServer && fIsOwnerOfClusterServer ) + { + if ( !fClusterServer ) + { + fClusterServer = new AliMUONLegacyClusterServer(*fkTransformer,fInputClusterStore, + GetRecoParam()->BypassSt4(), + GetRecoParam()->BypassSt5()); + SetupClusterServer(*fClusterServer); + } + } + fInputClusterStore->Clear(); + fInputClusterStore->Connect(*clustersTree,kFALSE); + fTriggerStore->Clear(); fTriggerStore->Connect(*clustersTree,kFALSE); clustersTree->GetEvent(0); @@ -186,12 +203,21 @@ Int_t AliMUONTracker::LoadClusters(TTree* clustersTree) Int_t AliMUONTracker::Clusters2Tracks(AliESDEvent* esd) { /// Performs the tracking and store the resulting tracks in the ESD - AliDebug(1,""); - AliCodeTimerAuto("") + /// + /// note that we're dealing with two cluster stores here : fInputClusterStore + /// and ClusterStore(). + /// The first one is read from the TreeR and may be used by the cluster server + /// (that's the case for the legacy cluster server) to fill the other one. + /// The second one is more dynamic and might be created on the fly by the cluster + /// server (used by the combined tracking, in which case the first one is not used + /// at all). + + AliCodeTimerAuto("",0) if (!fTrackReco) { - fTrackReco = CreateTrackReconstructor(AliMUONReconstructor::GetRecoParam()->GetTrackingMode(),fClusterServer); + fTrackReco = CreateTrackReconstructor(GetRecoParam(),fClusterServer); + fInternalTrackStore = new AliMUONTrackStoreV1; } // if the required tracking mode does not exist @@ -209,87 +235,130 @@ Int_t AliMUONTracker::Clusters2Tracks(AliESDEvent* esd) } // Make trigger tracks - if ( fTriggerCircuit ) + if ( fkTriggerCircuit ) { TriggerTrackStore()->Clear(); - fTrackReco->EventReconstructTrigger(*fTriggerCircuit,*fTriggerStore,*(TriggerTrackStore())); + fTrackReco->EventReconstructTrigger(*fkTriggerCircuit,*fTriggerStore,*(TriggerTrackStore())); } - // Make tracker tracks - AliMUONVTrackStore* trackStore = new AliMUONTrackStoreV1; - fTrackReco->EventReconstruct(*(ClusterStore()),*trackStore); + if ( TriggerTrackStore()->GetSize() > GetRecoParam()->GetMaxTriggerTracks() ) + { + // cut to reject shower events + + AliCodeTimerAuto("MUON Shower events",1); + + AliWarning(Form("Probably got a shower event (%d trigger tracks). Will not reconstruct tracks.", + TriggerTrackStore()->GetSize())); + + return 0; + } + + fTrackReco->EventReconstruct(*(ClusterStore()),*fInternalTrackStore); // Match tracker/trigger tracks if ( fTrackHitPatternMaker ) { - fTrackReco->ValidateTracksWithTrigger(*trackStore,*(TriggerTrackStore()),*fTriggerStore,*fTrackHitPatternMaker); + fTrackReco->ValidateTracksWithTrigger(*fInternalTrackStore,*(TriggerTrackStore()),*fTriggerStore,*fTrackHitPatternMaker); } // Fill ESD - FillESD(*trackStore,esd); + FillESD(*fInternalTrackStore,esd); - // cleanup - delete trackStore; + fInternalTrackStore->Clear(); + ClusterStore()->Clear(); return 0; } //_____________________________________________________________________________ -void AliMUONTracker::FillESD(AliMUONVTrackStore& trackStore, AliESDEvent* esd) const +void AliMUONTracker::FillESD(const AliMUONVTrackStore& trackStore, AliESDEvent* esd) const { /// Fill the ESD from the trackStore AliDebug(1,""); - AliCodeTimerAuto("") - - AliMUONTrack* track; - AliESDMuonTrack esdTrack; - Double_t vertex[3] = {0., 0., 0.}; - TIter next(trackStore.CreateIterator()); + AliCodeTimerAuto("",0) // get ITS vertex + Double_t vertex[3] = {0., 0., 0.}; const AliESDVertex* esdVert = esd->GetVertex(); - if (esdVert->GetNContributors()) { + if (esdVert->GetNContributors() > 0 || !strcmp(esdVert->GetTitle(),"vertexer: smearMC")) { esdVert->GetXYZ(vertex); AliDebug(1,Form("found vertex (%e,%e,%e)",vertex[0],vertex[1],vertex[2])); } // fill ESD event including all info in ESD cluster if required and only for the given fraction of events - if (AliMUONReconstructor::GetRecoParam()->SaveFullClusterInESD() && - gRandom->Uniform(100.) <= AliMUONReconstructor::GetRecoParam()->GetPercentOfFullClusterInESD()) { + AliMUONTrack* track; + AliMUONLocalTrigger* locTrg; + AliESDMuonTrack esdTrack; + TIter next(trackStore.CreateIterator()); + if (GetRecoParam()->SaveFullClusterInESD() && + gRandom->Uniform(100.) <= GetRecoParam()->GetPercentOfFullClusterInESD()) { while ( ( track = static_cast(next()) ) ) { - AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex, &fDigitStore); + + if (track->GetMatchTrigger() > 0) { + locTrg = static_cast(fTriggerStore->FindLocal(track->LoCircuit())); + AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex, &fkDigitStore, locTrg); + } else AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex, &fkDigitStore); + esd->AddMuonTrack(&esdTrack); } } else { while ( ( track = static_cast(next()) ) ) { - AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex); + + if (track->GetMatchTrigger() > 0) { + locTrg = static_cast(fTriggerStore->FindLocal(track->LoCircuit())); + AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex, 0x0, locTrg); + } else AliMUONESDInterface::MUONToESD(*track, esdTrack, vertex); + esd->AddMuonTrack(&esdTrack); } } + // fill the local trigger decisions not matched with tracks (associate them to "ghost" tracks) + UInt_t ghostId = 0xFFFFFFFF - 1; + Bool_t matched = kFALSE; + AliMUONTriggerTrack *triggerTrack; + TIter itTriggerTrack(fTriggerTrackStore->CreateIterator()); + while ( ( triggerTrack = static_cast(itTriggerTrack()) ) ) { + + locTrg = static_cast(fTriggerStore->FindLocal(triggerTrack->GetLoTrgNum())); + + // check if this local trigger has already been matched + TIter itTrack(trackStore.CreateIterator()); + while ( ( track = static_cast(itTrack()) ) ) { + matched = (track->LoCircuit() == locTrg->LoCircuit()); + if (matched) break; + } + if (matched) continue; + + AliMUONESDInterface::MUONToESD(*locTrg, esdTrack, ghostId, triggerTrack); + + esd->AddMuonTrack(&esdTrack); + ghostId -= 1; + } + } //_____________________________________________________________________________ -AliMUONVTrackReconstructor* AliMUONTracker::CreateTrackReconstructor(const char* trackingMode, AliMUONVClusterServer* clusterServer) +AliMUONVTrackReconstructor* AliMUONTracker::CreateTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer) { /// Create track reconstructor, depending on tracking mode set in RecoParam AliMUONVTrackReconstructor* trackReco(0x0); - TString opt(trackingMode); + TString opt(recoParam->GetTrackingMode()); opt.ToUpper(); if (strstr(opt,"ORIGINAL")) { - trackReco = new AliMUONTrackReconstructor(*clusterServer); + trackReco = new AliMUONTrackReconstructor(recoParam,clusterServer); } else if (strstr(opt,"KALMAN")) { - trackReco = new AliMUONTrackReconstructorK(*clusterServer); + trackReco = new AliMUONTrackReconstructorK(recoParam,clusterServer); } else { @@ -297,7 +366,7 @@ AliMUONVTrackReconstructor* AliMUONTracker::CreateTrackReconstructor(const char* return 0x0; } - AliInfoClass(Form("Will use %s for tracking",trackReco->ClassName())); + AliDebugClass(1,Form("Will use %s for tracking",trackReco->ClassName())); return trackReco; } @@ -307,8 +376,7 @@ void AliMUONTracker::UnloadClusters() { /// Clear internal clusterStore - delete fInputClusterStore; - fInputClusterStore = 0x0; + fInputClusterStore->Clear(); } @@ -318,18 +386,38 @@ AliMUONTracker::SetupClusterServer(AliMUONVClusterServer& clusterServer) { /// Setup the cluster server - if ( AliMUONReconstructor::GetRecoParam()->BypassSt45() ) + if ( GetRecoParam()->BypassSt4() || + GetRecoParam()->BypassSt5() ) { Bool_t ok = clusterServer.UseTriggerTrackStore(TriggerTrackStore()); + TString msg1; + TString msg2; + + if ( GetRecoParam()->BypassSt45() ) + { + msg1 = "STATIONS 4 AND 5"; + msg2 = "THOSE TWO STATIONS"; + } + else if ( GetRecoParam()->BypassSt4() ) + { + msg1 = "STATION 4"; + msg2 = "THAT STATION"; + } + else if ( GetRecoParam()->BypassSt5() ) + { + msg1 = "STATION 5"; + msg2 = "THAT STATION"; + } + if ( ok ) - { - AliWarning("WILL USE TRIGGER TRACKS TO GENERATE CLUSTERS IN STATIONS 4 AND 5, THUS BYPASSING REAL CLUSTERS IN THOSE TWO STATIONS !!!"); + AliWarning(Form("WILL USE TRIGGER TRACKS TO GENERATE CLUSTERS IN %s, " + "THUS BYPASSING REAL CLUSTERS IN %s!!!",msg1.Data(),msg2.Data())); } else { - AliWarning("BYPASSING OF ST45 REQUESTED, BUT CLUSTERSERVER DOES NOT SEEM TO SUPPORT IT !!!"); + AliWarning("BYPASSING OF ST4 AND/OR 5 REQUESTED, BUT CLUSTERSERVER DOES NOT SEEM TO SUPPORT IT !!!"); } } }