X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=MUON%2FAliMUONTracker.cxx;h=462d4e3c46e0fd6364ec5021921fe629725e326c;hp=ffd551bcc197cca28c73a64e72979341df253bb7;hb=132872f71b2349c9e3537489a3ab76a4e1d8fc5a;hpb=96ebe67e954961b3c45067cd239082c0d43cd009 diff --git a/MUON/AliMUONTracker.cxx b/MUON/AliMUONTracker.cxx index ffd551bcc19..462d4e3c46e 100644 --- a/MUON/AliMUONTracker.cxx +++ b/MUON/AliMUONTracker.cxx @@ -22,40 +22,43 @@ /// 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 //----------------------------------------------------------------------------- #include "AliMUONTracker.h" -#include "AliMUONReconstructor.h" +#include "AliCodeTimer.h" +#include "AliESDEvent.h" +#include "AliESDMuonTrack.h" +#include "AliESDVertex.h" +#include "AliLog.h" +#include "AliMUONClusterStoreV2.h" +#include "AliMUONESDInterface.h" +#include "AliMUONLegacyClusterServer.h" #include "AliMUONRecoParam.h" +#include "AliMUONReconstructor.h" #include "AliMUONTrack.h" #include "AliMUONTrackExtrap.h" #include "AliMUONTrackHitPattern.h" #include "AliMUONTrackParam.h" -#include "AliMUONVCluster.h" #include "AliMUONTrackReconstructor.h" #include "AliMUONTrackReconstructorK.h" #include "AliMUONTrackStoreV1.h" -#include "AliMUONTriggerChamberEff.h" #include "AliMUONTriggerTrackStoreV1.h" -#include "AliMUONVClusterStore.h" +#include "AliMUONTriggerTrack.h" +#include "AliMUONLocalTrigger.h" +#include "AliMUONVClusterServer.h" +#include "AliMUONVDigitStore.h" #include "AliMUONVTriggerStore.h" - -#include "AliESDEvent.h" -#include "AliESDMuonTrack.h" -#include "AliESDVertex.h" -#include "AliLog.h" -#include "AliCodeTimer.h" - +#include "AliMUONTriggerUtilities.h" #include +#include #include +#include "AliLog.h" /// \cond CLASSIMP ClassImp(AliMUONTracker) @@ -63,62 +66,132 @@ ClassImp(AliMUONTracker) //_____________________________________________________________________________ -AliMUONTracker::AliMUONTracker(const AliMUONDigitMaker* digitMaker, +AliMUONTracker::AliMUONTracker(const AliMUONRecoParam* recoParam, + AliMUONVClusterServer* clusterServer, + AliMUONVDigitStore& digitStore, const AliMUONGeometryTransformer* transformer, const AliMUONTriggerCircuit* triggerCircuit, - AliMUONTriggerChamberEff* chamberEff) + const AliMUONTriggerUtilities* triggerUtilities) : AliTracker(), - fDigitMaker(digitMaker), // not owner - fTransformer(transformer), // not owner - fTriggerCircuit(triggerCircuit), // not owner - fTrigChamberEff(chamberEff), // not owner - fTrackHitPatternMaker(0x0), - fTrackReco(0x0), - fClusterStore(0x0), - fTriggerStore(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) + { + AliDebug(1,"No cluster server given. Will use AliMUONLegacyClusterServer"); + fIsOwnerOfClusterServer = kTRUE; + } + else + { + TIter next(fkDigitStore.CreateIterator()); + fClusterServer->UseDigits(next,&digitStore); + + SetupClusterServer(*fClusterServer); + } } //_____________________________________________________________________________ AliMUONTracker::~AliMUONTracker() { /// dtor + delete fTrackReco; delete fTrackHitPatternMaker; delete fClusterStore; delete fTriggerStore; + if ( fIsOwnerOfClusterServer ) delete fClusterServer; + delete fInputClusterStore; + delete fTriggerTrackStore; + delete fInternalTrackStore; +} + +//_____________________________________________________________________________ +AliMUONVClusterStore* +AliMUONTracker::ClusterStore() const +{ + /// Return (and create if necessary) the cluster container + if (!fClusterStore) + { + fClusterStore = new AliMUONClusterStoreV2; + } + return fClusterStore; +} + +//_____________________________________________________________________________ +AliMUONVTriggerTrackStore* +AliMUONTracker::TriggerTrackStore() const +{ + /// Return (and create if necessary) the trigger track container + if (!fTriggerTrackStore) + { + fTriggerTrackStore = new AliMUONTriggerTrackStoreV1; + } + return fTriggerTrackStore; } //_____________________________________________________________________________ Int_t AliMUONTracker::LoadClusters(TTree* clustersTree) { - /// Load clusterStore and triggerStore from clustersTree - delete fClusterStore; - delete fTriggerStore; + /// Load triggerStore from clustersTree if ( ! clustersTree ) { AliFatal("No clustersTree"); return 1; } - fClusterStore = AliMUONVClusterStore::Create(*clustersTree); - fTriggerStore = AliMUONVTriggerStore::Create(*clustersTree); - - if (!fClusterStore) + if ( !fTriggerStore ) { - AliError("Could not get clusterStore"); - return 1; + fTriggerStore = AliMUONVTriggerStore::Create(*clustersTree); } + if (!fTriggerStore) { AliError("Could not get triggerStore"); return 2; } - fClusterStore->Connect(*clustersTree,kFALSE); + if (!fInputClusterStore) + { + fInputClusterStore = AliMUONVClusterStore::Create(*clustersTree); + if (!fInputClusterStore) + { + AliError("Could not get clusterStore"); + return 3; + } + 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); @@ -130,15 +203,28 @@ 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). - if (!fTrackReco) CreateTrackReconstructor(); + AliCodeTimerAuto("",0) + + if (!fTrackReco) + { + fTrackReco = CreateTrackReconstructor(GetRecoParam(),fClusterServer); + fInternalTrackStore = new AliMUONTrackStoreV1; + } // if the required tracking mode does not exist if (!fTrackReco) return 1; - if (!fClusterStore) { + if ( ! ClusterStore() ) + { AliError("ClusterStore is NULL"); return 2; } @@ -147,124 +233,193 @@ Int_t AliMUONTracker::Clusters2Tracks(AliESDEvent* esd) AliError("TriggerStore is NULL"); return 3; } - - // Make tracker tracks - AliMUONVTrackStore* trackStore = new AliMUONTrackStoreV1; - fTrackReco->EventReconstruct(*fClusterStore,*trackStore); - + // Make trigger tracks - AliMUONVTriggerTrackStore* triggerTrackStore(0x0); - if ( fTriggerCircuit ) { - triggerTrackStore = new AliMUONTriggerTrackStoreV1; - fTrackReco->EventReconstructTrigger(*fTriggerCircuit,*fTriggerStore,*triggerTrackStore); + if ( fkTriggerCircuit ) + { + TriggerTrackStore()->Clear(); + fTrackReco->EventReconstructTrigger(*fkTriggerCircuit,*fTriggerStore,*(TriggerTrackStore())); } + + if ( TriggerTrackStore()->GetSize() > GetRecoParam()->GetMaxTriggerTracks() ) + { + // cut to reject shower events + + AliCodeTimerAuto("MUON Shower events",1); - // Match tracker/trigger tracks - if ( triggerTrackStore && fTrackHitPatternMaker ) { - fTrackReco->ValidateTracksWithTrigger(*trackStore,*triggerTrackStore,*fTriggerStore,*fTrackHitPatternMaker); + AliWarning(Form("Probably got a shower event (%d trigger tracks). Will not reconstruct tracks.", + TriggerTrackStore()->GetSize())); + + return 0; } + + fTrackReco->EventReconstruct(*(ClusterStore()),*fInternalTrackStore); - // Compute trigger chamber efficiency - if( triggerTrackStore && fTrigChamberEff){ - AliCodeTimerStart("EventChamberEff"); - fTrigChamberEff->EventChamberEff(*fTriggerStore,*triggerTrackStore,*trackStore); - AliCodeTimerStop("EventChamberEff"); + // Match tracker/trigger tracks + if ( fTrackHitPatternMaker ) + { + fTrackReco->ValidateTracksWithTrigger(*fInternalTrackStore,*(TriggerTrackStore()),*fTriggerStore,*fTrackHitPatternMaker); } // Fill ESD - FillESD(*trackStore,esd); + FillESD(*fInternalTrackStore,esd); - // cleanup - delete trackStore; - delete triggerTrackStore; + 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("") + AliCodeTimerAuto("",0) - // Get vertex - Double_t vertex[3] = {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])); } - // setting ESD MUON class - AliESDMuonTrack esdTrack; - + // fill ESD event including all info in ESD cluster if required and only for the given fraction of events AliMUONTrack* track; + AliMUONLocalTrigger* locTrg; + AliESDMuonTrack esdTrack; TIter next(trackStore.CreateIterator()); - - while ( ( track = static_cast(next()) ) ) - { - AliMUONTrackParam* trackParam = static_cast((track->GetTrackParamAtCluster())->First()); - AliMUONTrackParam trackParamAtVtx(*trackParam); + if (GetRecoParam()->SaveFullClusterInESD() && + gRandom->Uniform(100.) <= GetRecoParam()->GetPercentOfFullClusterInESD()) { - /// Extrapolate to vertex (which is set to (0,0,0) if not available, see above) - AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, vertex[0],vertex[1],vertex[2]); + while ( ( track = static_cast(next()) ) ) { + + 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); + } - // setting data member of ESD MUON + } else { - // at first station - trackParam->SetParamForUncorrected(esdTrack); - trackParam->SetCovFor(esdTrack); - // at vertex - trackParamAtVtx.SetParamFor(esdTrack); - // global info - esdTrack.SetChi2(track->GetGlobalChi2()); - esdTrack.SetNHit(track->GetNClusters()); - esdTrack.SetLocalTrigger(track->GetLocalTrigger()); - esdTrack.SetChi2MatchTrigger(track->GetChi2MatchTrigger()); - esdTrack.SetHitsPatternInTrigCh(track->GetHitsPatternInTrigCh()); - // muon cluster map - while (trackParam) { - esdTrack.AddInMuonClusterMap(trackParam->GetClusterPtr()->GetChamberId()); - trackParam = static_cast(track->GetTrackParamAtCluster()->After(trackParam)); + while ( ( track = static_cast(next()) ) ) { + + 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); } - // storing ESD MUON Track into ESD Event + } + + // 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); - } // end of loop on tracks + ghostId -= 1; + } + } //_____________________________________________________________________________ -void AliMUONTracker::CreateTrackReconstructor() +AliMUONVTrackReconstructor* AliMUONTracker::CreateTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer) { /// Create track reconstructor, depending on tracking mode set in RecoParam - TString opt(AliMUONReconstructor::GetRecoParam()->GetTrackingMode()); + AliMUONVTrackReconstructor* trackReco(0x0); + + TString opt(recoParam->GetTrackingMode()); opt.ToUpper(); if (strstr(opt,"ORIGINAL")) { - fTrackReco = new AliMUONTrackReconstructor(); + trackReco = new AliMUONTrackReconstructor(recoParam,clusterServer); } else if (strstr(opt,"KALMAN")) { - fTrackReco = new AliMUONTrackReconstructorK(); + trackReco = new AliMUONTrackReconstructorK(recoParam,clusterServer); } else { - AliError(Form("tracking mode \"%s\" does not exist",opt.Data())); - return; + AliErrorClass(Form("tracking mode \"%s\" does not exist",opt.Data())); + return 0x0; } - AliInfo(Form("Will use %s for tracking",fTrackReco->ClassName())); + AliDebugClass(1,Form("Will use %s for tracking",trackReco->ClassName())); + + return trackReco; } //_____________________________________________________________________________ void AliMUONTracker::UnloadClusters() { - /// Delete internal clusterStore - delete fClusterStore; - fClusterStore = 0x0; + /// Clear internal clusterStore + + fInputClusterStore->Clear(); } + +//_____________________________________________________________________________ +void +AliMUONTracker::SetupClusterServer(AliMUONVClusterServer& clusterServer) +{ + /// Setup the cluster server + + 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(Form("WILL USE TRIGGER TRACKS TO GENERATE CLUSTERS IN %s, " + "THUS BYPASSING REAL CLUSTERS IN %s!!!",msg1.Data(),msg2.Data())); + } + else + { + AliWarning("BYPASSING OF ST4 AND/OR 5 REQUESTED, BUT CLUSTERSERVER DOES NOT SEEM TO SUPPORT IT !!!"); + } + } +} + +