TrigBphysMonitorAlgorithm.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TrigBphysMonitorAlgorithm.h"
 
#include "GaudiKernel/SystemOfUnits.h"
 
#include "FourMomUtils/xAODP4Helpers.h"
#include "xAODBPhys/BPhysHelper.h"
#include "TrigInDetToolInterfaces/TrigParticleTable.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
 
using namespace Gaudi::Units;
 
TrigBphysMonitorAlgorithm::TrigBphysMonitorAlgorithm( const std::string& name, ISvcLocator* pSvcLocator )
  : AthMonitorAlgorithm(name,pSvcLocator)
{}
 
 
TrigBphysMonitorAlgorithm::~TrigBphysMonitorAlgorithm() {}
 
 
StatusCode TrigBphysMonitorAlgorithm::initialize() {
  
  ATH_CHECK( m_TrigBphysContainerKeys.initialize() );
  ATH_CHECK( m_offlineMuonCollectionKey.initialize() );
  ATH_CHECK( m_offlineIDTrackCollectionKey.initialize() );
  ATH_CHECK( m_offlinePvCollectionKey.initialize() );
  
  ATH_CHECK( AthMonitorAlgorithm::initialize() );
  
  return StatusCode::SUCCESS;
  
}
 
 
StatusCode TrigBphysMonitorAlgorithm::fillHistograms( const EventContext& ctx ) const {
    
  // fill the container hists
  ATH_MSG_DEBUG ("fill... Containers");
  if (fillContainers(ctx).isFailure()) {ATH_MSG_ERROR("Problems filling Container histograms");}
  
  // fill the chain hists
  ATH_MSG_DEBUG ("fill... Chains");
  if (fillChains(ctx).isFailure()) {ATH_MSG_ERROR("Problems filling Chain histograms");}
  
  // build offline dimuons and fill offline histograms
  std::vector<std::unique_ptr<xAOD::Vertex>> offlineDimuons;
  if (buildDimuons(ctx, offlineDimuons).isFailure()) {
    ATH_MSG_ERROR("Problems building offline dimuons, won't fill corresponding histograms");
  }
  else {
    if (fillOfflineDimuons(ctx, offlineDimuons).isFailure()) {ATH_MSG_ERROR("Problems filling Offline dimuon histograms");}
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillContainers(const EventContext& ctx) const {
  
  for(const SG::ReadHandleKey<xAOD::TrigBphysContainer>& monitoredContainerKey : m_TrigBphysContainerKeys) {
    ATH_MSG_DEBUG("Process container " << monitoredContainerKey.key().c_str());
    if(fillContainerHists(ctx, monitoredContainerKey).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling %s container histograms", monitoredContainerKey.key().c_str()));
    }
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillContainerHists(const EventContext& ctx, const SG::ReadHandleKey<xAOD::TrigBphysContainer>& trigBphysContainerKey) const {
  SG::ReadHandle<xAOD::TrigBphysContainer> trigBphysContainer(trigBphysContainerKey, ctx);
  if( !trigBphysContainer.isValid() ) {
    ATH_MSG_WARNING("No valid TrigBphysContainer with tag: " << trigBphysContainerKey);
  }
  else {
    // Fill container-wise histograms
    std::string monGroupName = std::string("Container_")+trigBphysContainerKey.key();
    auto monGroup = getGroup(monGroupName);
    
    auto ncandidates = Monitored::Scalar<int>("ncandidates",-999);
    ncandidates = trigBphysContainer->size();
    
    auto bphys_mass = Monitored::Collection("bphys_mass", (*trigBphysContainer),  [](const xAOD::TrigBphys* bphys){ return bphys->fitmass() / GeV;});
    
    fill(monGroup, ncandidates, bphys_mass);
    
  }
  return StatusCode::SUCCESS;
}
 
 
StatusCode TrigBphysMonitorAlgorithm::fillChains(const EventContext& ctx) const {
  
  for(const auto& monitoredChain : m_ChainNames_MuMu) {
    ATH_MSG_DEBUG("Process chain " << monitoredChain);
    if( !isChainPassed(monitoredChain) ) {
      ATH_MSG_DEBUG("Chain " << monitoredChain << " is not passed");
      continue;
    }
    if(fillDimuonChainHists(ctx, monitoredChain).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling %s chain histograms", monitoredChain.c_str()));
    }
  }
  
  for(const auto& monitoredChain : m_ChainNames_MuMuX) {
    ATH_MSG_DEBUG("Process chain " << monitoredChain);
    if( !isChainPassed(monitoredChain) ) {
      ATH_MSG_DEBUG("Chain " << monitoredChain << " is not passed");
      continue;
    }
    if(fillBmumuxChainHists(ctx, monitoredChain).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling %s chain histograms", monitoredChain.c_str()));
    }
  }
  
  for(const auto& monitoredChain : m_ChainNames_ElEl) {
    ATH_MSG_DEBUG("Process chain " << monitoredChain);
    if( !isChainPassed(monitoredChain) ) {
      ATH_MSG_DEBUG("Chain " << monitoredChain << " is not passed");
      continue;
    }
    if(fillDielectronChainHists(ctx, monitoredChain).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling %s chain histograms", monitoredChain.c_str()));
    }
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillDimuonChainHists(const EventContext& ctx, const std::string& chainName) const {
  ATH_MSG_DEBUG("Filling  " << chainName << " chain histograms");
  
  std::string monGroupName = std::string("Chain_")+chainName;
  auto monGroup = getGroup(monGroupName);
    
  if (fillChainGenericHists(ctx, monGroup, chainName).isFailure()) {
    ATH_MSG_ERROR(Form("Problems filling generic histograms for %s chain",chainName.c_str()));
  }
  
  std::vector< TrigCompositeUtils::LinkInfo<xAOD::TrigBphysContainer> > chainFeatureContainer = getTrigDecisionTool()->features<xAOD::TrigBphysContainer>(chainName, TrigDefs::Physics);
  for (const auto& featLinkInfo: chainFeatureContainer){
    ATH_CHECK(featLinkInfo.isValid());
    const auto featLink = featLinkInfo.link;
    if (fillBphysObjectHists(monGroup, featLink, "dimu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling bphys object histograms for %s chain",chainName.c_str()));
    }
    if (fillTrigLeptonHists(monGroup, featLink, "mu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling muon histograms for %s chain",chainName.c_str()));
    }
  }
  return StatusCode::SUCCESS;
}
 
 
StatusCode TrigBphysMonitorAlgorithm::fillBmumuxChainHists(const EventContext& ctx, const std::string& chainName) const {
  ATH_MSG_DEBUG("Filling  " << chainName << " chain histograms");
  
  std::string monGroupName = std::string("Chain_")+chainName;
  auto monGroup = getGroup(monGroupName);
    
  if (fillChainGenericHists(ctx, monGroup, chainName).isFailure()) {
    ATH_MSG_ERROR(Form("Problems filling generic histograms for %s chain",chainName.c_str()));
  }
  
  std::vector< TrigCompositeUtils::LinkInfo<xAOD::TrigBphysContainer> > chainFeatureContainer = getTrigDecisionTool()->features<xAOD::TrigBphysContainer>(chainName, TrigDefs::Physics);
  for (const auto& featLinkInfo: chainFeatureContainer){
    ATH_CHECK(featLinkInfo.isValid());
    const auto featLink = featLinkInfo.link;
    if (fillBphysObjectHists(monGroup, featLink, "B").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling bphys object histograms for %s chain",chainName.c_str()));
    }
    if (fillTrigBmumuxTrkHists(monGroup, featLink).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling muon histograms for %s chain",chainName.c_str()));
    }
    
    auto dimuonLink = (*featLink)->lowerChainLink();
    ATH_CHECK(dimuonLink.isValid());
    if (fillBphysObjectHists(monGroup, dimuonLink, "dimu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling dimuon object histograms for %s chain",chainName.c_str()));
    }
    if (fillTrigLeptonHists(monGroup, dimuonLink, "mu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling muon histograms for %s chain",chainName.c_str()));
    }
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillDielectronChainHists(const EventContext& ctx, const std::string& chainName) const {
  ATH_MSG_DEBUG("Filling  " << chainName << " chain histograms");
  
  std::string monGroupName = std::string("Chain_")+chainName;
  auto monGroup = getGroup(monGroupName);
    
  if (fillChainGenericHists(ctx, monGroup, chainName).isFailure()) {
    ATH_MSG_ERROR(Form("Problems filling generic histograms for %s chain",chainName.c_str()));
  }
  
  std::vector< TrigCompositeUtils::LinkInfo<xAOD::TrigBphysContainer> > chainFeatureContainer = getTrigDecisionTool()->features<xAOD::TrigBphysContainer>(chainName, TrigDefs::Physics);
  for (const auto& featLinkInfo: chainFeatureContainer){
    ATH_CHECK(featLinkInfo.isValid());
    const auto featLink = featLinkInfo.link;
    if (fillBphysObjectHists(monGroup, featLink, "diel").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling bphys object histograms for %s chain",chainName.c_str()));
    }
    if (fillTrigLeptonHists(monGroup, featLink, "el").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling muon histograms for %s chain",chainName.c_str()));
    }
  }
  return StatusCode::SUCCESS;
}
 
 
// Function to fill general per-chain hists, independent on the type of chains
StatusCode TrigBphysMonitorAlgorithm::fillChainGenericHists(const EventContext& /*ctx*/, const ToolHandle<GenericMonitoringTool>& currentMonGroup, const std::string& chainName) const {
  
  std::vector< TrigCompositeUtils::LinkInfo<xAOD::TrigBphysContainer> > chainFeatureContainer = getTrigDecisionTool()->features<xAOD::TrigBphysContainer>(chainName, TrigDefs::Physics);
  auto ncandidates = Monitored::Scalar<int>("ncandidates",-999);
  ncandidates = chainFeatureContainer.size();
  
  fill(currentMonGroup, ncandidates);
  
  return StatusCode::SUCCESS;
}
 
 
// Function to fill per-object hists (e.g. for dimuon, B from Bmumux, or X from Bmumux)
StatusCode TrigBphysMonitorAlgorithm::fillBphysObjectHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, const ElementLink<xAOD::TrigBphysContainer>& bphysLink, const std::string& objStr) const {
  
  auto dimu_mass = Monitored::Scalar<float>(objStr+"_mass",-999.);
  auto dimu_fitmass = Monitored::Scalar<float>(objStr+"_fitmass",-999.);
  auto dimu_pt = Monitored::Scalar<float>(objStr+"_pt",-999.);
  auto dimu_y = Monitored::Scalar<float>(objStr+"_y",-999.);
  auto dimu_chi2 = Monitored::Scalar<float>(objStr+"_chi2",-999.);
  
  dimu_mass = (*bphysLink)->mass() / GeV;
  dimu_fitmass = (*bphysLink)->fitmass() / GeV;
  dimu_pt   = (*bphysLink)->pt() / GeV;
  dimu_chi2 = (*bphysLink)->fitchi2();
  dimu_y    = (*bphysLink)->eta();
  
  fill(currentMonGroup, dimu_mass, dimu_fitmass, dimu_pt, dimu_y, dimu_chi2);
  
  return StatusCode::SUCCESS;
}
 
 
// Function to fill per-muon hists, assuming that the passed object is a dimuon
StatusCode TrigBphysMonitorAlgorithm::fillTrigLeptonHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, const ElementLink<xAOD::TrigBphysContainer>& bphysLink, const std::string& name_prefix) const {
  
  const std::vector<ElementLink<xAOD::TrackParticleContainer> > trackVector = (*bphysLink)->trackParticleLinks();
  ATH_MSG_DEBUG("fillTrigLeptonHists: number of lepton tracks: " << trackVector.size());
  if( fillTracksHists(currentMonGroup, trackVector, name_prefix, true).isFailure() ) {
    ATH_MSG_ERROR(Form("Problems filling muon histograms for a chain"));
    return StatusCode::FAILURE;
  }
  if( fillDiTracksHists(currentMonGroup, trackVector, std::string("di")+name_prefix).isFailure() ) {
    ATH_MSG_ERROR(Form("Problems filling two-muon histograms for a chain"));
    return StatusCode::FAILURE;
  }
  
  return StatusCode::SUCCESS;
}
 
// Function to fill per-track hists for Bmumux candidates
StatusCode TrigBphysMonitorAlgorithm::fillTrigBmumuxTrkHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, const ElementLink<xAOD::TrigBphysContainer>& bphysLink, UInt_t tracksStartFrom) const {
  
  const std::vector<ElementLink<xAOD::TrackParticleContainer> > trackVector = (*bphysLink)->trackParticleLinks();
  if (trackVector.size() < tracksStartFrom) {
      ATH_MSG_WARNING("Unexpected number of tracks: "<< trackVector.size() << " found, while expected at least " << tracksStartFrom);
      return StatusCode::SUCCESS;
  }
  
  if (fillTracksHists(currentMonGroup, trackVector, "trk", false, tracksStartFrom).isFailure()) {
    ATH_MSG_ERROR(Form("Problems filling track histograms for a BMuMuX chain"));
    return StatusCode::FAILURE;
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillTracksHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, 
                                                      const std::vector<ElementLink<xAOD::TrackParticleContainer> >& tpLinkVector, 
                                                      const std::string& prefix, 
                                                      bool separateHists, 
                                                      UInt_t offsetIndex) const {
  for(UInt_t i = offsetIndex; i < tpLinkVector.size(); ++i) {
    ATH_CHECK(tpLinkVector.at(i).isValid());
    std::string curPrefix = prefix;
    if(separateHists)
      curPrefix += std::to_string(i+1);
    if (fillTrkHists(currentMonGroup, *tpLinkVector.at(i), curPrefix).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling track histograms"));
      return StatusCode::FAILURE;
    }
  }
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillDiTracksHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, 
                                                        const std::vector<ElementLink<xAOD::TrackParticleContainer> >& tpLinkVector,
                                                        const std::string& name_prefix) const {
  // Use first two tracks
  if (tpLinkVector.size() <2) {
      ATH_MSG_ERROR("Unexpected number of tracks in a dimuon: " << tpLinkVector.size());
      return StatusCode::FAILURE;
  }
  ATH_CHECK(tpLinkVector.at(0).isValid());
  ATH_CHECK(tpLinkVector.at(1).isValid());
  const xAOD::TrackParticle* trk1 = *tpLinkVector.at(0);
  const xAOD::TrackParticle* trk2 = *tpLinkVector.at(1);
  
  auto ditrk_dR   = Monitored::Scalar<float>(name_prefix+"_dR",-999.);
  auto ditrk_deta = Monitored::Scalar<float>(name_prefix+"_deta",-999.);
  auto ditrk_dphi = Monitored::Scalar<float>(name_prefix+"_dphi",-999.);
  
  ditrk_dR = xAOD::P4Helpers::deltaR(*trk1, *trk2, false); // false for pseudo, not true rapidity
  ditrk_deta = std::abs(trk1->eta()-trk2->eta());
  ditrk_dphi = std::abs(xAOD::P4Helpers::deltaPhi(*trk1, *trk2));
  
  fill(currentMonGroup, ditrk_dR, ditrk_deta, ditrk_dphi);
  
  return StatusCode::SUCCESS;
}
 
// Generic function to fill track hists
StatusCode TrigBphysMonitorAlgorithm::fillTrkHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, const xAOD::TrackParticle* trk, const std::string& name_prefix) const {
  if (!trk) {
    ATH_MSG_ERROR("Null pointer for a track");
    return StatusCode::FAILURE;
  }
  auto trk_pt  = Monitored::Scalar<float>(name_prefix+"_pt",-999.);
  auto trk_eta = Monitored::Scalar<float>(name_prefix+"_eta",-999.);
  auto trk_d0  = Monitored::Scalar<float>(name_prefix+"_d0",-999.);
  
  trk_pt  = trk->pt() / GeV;
  trk_eta = trk->eta();
  trk_d0  = trk->d0();
  
  fill(currentMonGroup, trk_pt, trk_eta, trk_d0);
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillOfflineDimuons(const EventContext& ctx, const std::vector<std::unique_ptr<xAOD::Vertex>>& dimuonContainer) const {
  
  std::vector<std::string> dimuonMonGroupNames = {"Any"};
  dimuonMonGroupNames.insert( dimuonMonGroupNames.end(), m_ChainNames_MuMu.begin(), m_ChainNames_MuMu.end() );
  dimuonMonGroupNames.insert( dimuonMonGroupNames.end(), m_ChainNames_MuMuX.begin(), m_ChainNames_MuMuX.end() );
  
  for(const auto& dimuonMonGroupName : dimuonMonGroupNames) {
    ATH_MSG_DEBUG("Process dimuons for " << dimuonMonGroupName);
    if(dimuonMonGroupName != "Any") {
      auto& monitoredChain = dimuonMonGroupName;
      if( !(dimuonMonGroupName == "Any") && !isChainPassed(monitoredChain) ) {
        ATH_MSG_DEBUG("Chain " << monitoredChain << " is not passed");
        continue;
      }
    }
    if(fillOfflineDimuonHists(ctx, dimuonMonGroupName, dimuonContainer).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling offline dimuon histograms for %s", dimuonMonGroupName.c_str()));
    }
  }
  
  return StatusCode::SUCCESS;
}
 
StatusCode TrigBphysMonitorAlgorithm::fillOfflineDimuonHists(const EventContext& /*ctx*/, const std::string& dimuonMonGroupName, const std::vector<std::unique_ptr<xAOD::Vertex>>& dimuonContainer) const {
  ATH_MSG_DEBUG("Filling  " << dimuonMonGroupName << " offline dimuons histograms");
  
  // Do matching
  std::vector<const xAOD::Vertex*> matchedDimuons;
  for(auto& offlineDimuon : dimuonContainer) {
    if( dimuonMonGroupName == "Any" || matchDimuon(offlineDimuon.get(), dimuonMonGroupName) ) 
      matchedDimuons.push_back(offlineDimuon.get());
  }
  
  std::string monGroupName = std::string("OfflineDimu_")+dimuonMonGroupName;
  auto monGroup = getGroup(monGroupName);
  
  auto ncandidates = Monitored::Scalar<int>("ncandidates",-999);
  ncandidates = matchedDimuons.size();
  
  fill(monGroup, ncandidates);
  
  for(auto matchedDimuon : matchedDimuons) {
    if (fillVertexHists(monGroup, matchedDimuon, "dimu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling histogram for an offline dimuon vertex in %s", dimuonMonGroupName.c_str()));
    }
    if (fillTracksHists(monGroup, matchedDimuon->trackParticleLinks(), "mu", true).isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling histogram for offline dimuon muons in %s", dimuonMonGroupName.c_str()));
    }
    if (fillDiTracksHists(monGroup, matchedDimuon->trackParticleLinks(), "dimu").isFailure()) {
      ATH_MSG_ERROR(Form("Problems filling histogram for offline dimuon muon pairs in %s", dimuonMonGroupName.c_str()));
    }
  }
  
  return StatusCode::SUCCESS;
  
}
 
StatusCode TrigBphysMonitorAlgorithm::fillVertexHists(const ToolHandle<GenericMonitoringTool>& currentMonGroup, const xAOD::Vertex* vertex, const std::string& objStr) const {
  
  xAOD::BPhysHelper dimuonVertexHelper(vertex);
  if(dimuonVertexHelper.nRefTrks() != 2) {
    ATH_MSG_WARNING("Unexpected number of refitted tracks at vertex is " << dimuonVertexHelper.nRefTrks());
    return StatusCode::SUCCESS;
  }
  
  using namespace TrigVtx;
  TLorentzVector dimuonMom = dimuonVertexHelper.totalP(std::array<double,2>{TrigParticleMasses().mass[TrigParticleName::muon], TrigParticleMasses().mass[TrigParticleName::muon]});
  
  auto dimu_mass = Monitored::Scalar<float>(objStr+"_mass",-999.);
  auto dimu_pt = Monitored::Scalar<float>(objStr+"_pt",-999.);
  auto dimu_y = Monitored::Scalar<float>(objStr+"_y",-999.);
  auto dimu_chi2 = Monitored::Scalar<float>(objStr+"_chi2",-999.);
  auto dimu_Lxy = Monitored::Scalar<float>(objStr+"_Lxy",-999.);
  auto dimu_LxySig = Monitored::Scalar<float>(objStr+"_LxySig",-999.);
  
  dimu_mass = dimuonMom.M() / GeV;
  dimu_pt   = dimuonMom.Pt() / GeV;
  dimu_chi2 = vertex->chiSquared();
  dimu_y    = dimuonMom.Rapidity();
  dimu_Lxy  = dimuonVertexHelper.lxy(xAOD::BPhysHelper::PV_MIN_A0);
  dimu_LxySig = dimuonVertexHelper.lxy(xAOD::BPhysHelper::PV_MIN_A0)/dimuonVertexHelper.lxyErr(xAOD::BPhysHelper::PV_MIN_A0);
  
  fill(currentMonGroup, dimu_mass, dimu_pt, dimu_y, dimu_chi2, dimu_Lxy, dimu_LxySig);
  
  return StatusCode::SUCCESS;
}
 
bool TrigBphysMonitorAlgorithm::isChainPassed(const std::string& chain) const {
  // Check if a chain is passed after prescale, accounting for Express Stream prescales if necessary
  if(!m_requireExplicitESDecision) {
    return getTrigDecisionTool()->isPassed(chain);
  }
  else {
    const unsigned int passBits = getTrigDecisionTool()->isPassedBits(chain);
    return passBits & TrigDefs::Express_passed;
  }
}
 
StatusCode TrigBphysMonitorAlgorithm::buildDimuons(const EventContext& ctx, std::vector<std::unique_ptr<xAOD::Vertex>>& vxContainer) const {
  ATH_MSG_DEBUG( "TrigBphysMonitorAlgorithm::buildDimuons" );
  
  // Get the muons
  SG::ReadHandle<xAOD::MuonContainer> muonHandle(m_offlineMuonCollectionKey,ctx);
  ATH_CHECK(muonHandle.isValid());
  const xAOD::MuonContainer* muonContainer = muonHandle.cptr();
  ATH_MSG_DEBUG("Muon container size " << muonContainer->size());
  
  // Muon preselection
  std::vector<const xAOD::Muon*> selectedMuons;
  for (const auto mu : *muonContainer) {
    if ( mu == nullptr ) continue;
    if ( mu->muonType() != xAOD::Muon::Combined ) continue; // require combined muons
    if (!mu->inDetTrackParticleLink()) continue; // No muons without ID tracks
    if (!mu->inDetTrackParticleLink().isValid()) continue; // No muons without ID tracks
    selectedMuons.push_back(mu);
  }
  if(selectedMuons.size() < 2) {
    ATH_MSG_DEBUG("Only " << selectedMuons.size() << "muons pass preselection");
    return StatusCode::SUCCESS;
  }
  std::sort(selectedMuons.begin(), selectedMuons.end(), [](const auto mu1, const auto mu2){ return mu1->pt() > mu2->pt(); });
  
  // Build dimuons
  for(auto outerItr=selectedMuons.begin(); outerItr<selectedMuons.end(); ++outerItr){
    for(auto innerItr=(outerItr+1); innerItr!=selectedMuons.end(); ++innerItr){
      const auto muon1 = *outerItr;
      const auto muon2 = *innerItr;
      const auto trackParticle1 = muon1->trackParticle( xAOD::Muon::InnerDetectorTrackParticle );
      const auto trackParticle2 = muon2->trackParticle( xAOD::Muon::InnerDetectorTrackParticle );
      // Charge selection
      if(trackParticle1->qOverP() * trackParticle2->qOverP() > 0.) 
        continue;
      // Mass selection
      double dimu_momentum_prefit = dimuonMass(trackParticle1, trackParticle2);
      if( !(dimu_momentum_prefit > m_dimuMassLower_prefit && dimu_momentum_prefit < m_dimuMassUpper_prefit) ) 
        continue;
      // Fit
      std::unique_ptr<xAOD::Vertex> dimuon = dimuonFit(trackParticle1, trackParticle2);
      if(!dimuon) continue;
      if(dimuon->chiSquared() > m_dimuChi2Cut) continue;
      vxContainer.push_back(std::move(dimuon));
    }
  }
  ATH_MSG_DEBUG("Found " << vxContainer.size() << " good dimuons");
  
  // Augment the dimuon vertices
  if( !vxContainer.empty() ) {
    SG::ReadHandle<xAOD::VertexContainer> pvHandle(m_offlinePvCollectionKey,ctx);
    ATH_CHECK(pvHandle.isValid());
    const xAOD::VertexContainer* pvContainer = pvHandle.cptr();
    std::vector<const xAOD::Vertex*> goodPVs = GetGoodPVs(pvContainer);
    ATH_MSG_DEBUG("Found " << goodPVs.size() << " good PVs");
    for(auto& dimuon : vxContainer) {
      xAOD::BPhysHelper jpsiHelper(dimuon.get());
      jpsiHelper.setRefTrks();
      const xAOD::Vertex* lowestA0Pv = getPvForDimuon_lowestA0(dimuon.get(), goodPVs);
      if(lowestA0Pv) {
        jpsiHelper.setLxy   ( m_v0Tools->lxy     ( dimuon.get(),lowestA0Pv ), xAOD::BPhysHelper::PV_MIN_A0 );
        jpsiHelper.setLxyErr( m_v0Tools->lxyError( dimuon.get(),lowestA0Pv ), xAOD::BPhysHelper::PV_MIN_A0 );
        ATH_MSG_VERBOSE("Lxy    = " << m_v0Tools->lxy     ( dimuon.get(),lowestA0Pv ));
        ATH_MSG_VERBOSE("LxyErr = " << m_v0Tools->lxyError( dimuon.get(),lowestA0Pv ));
      }
    }
  }
 
  return StatusCode::SUCCESS;
}
 
std::unique_ptr<xAOD::Vertex> TrigBphysMonitorAlgorithm::dimuonFit(const xAOD::TrackParticle* mu1, const xAOD::TrackParticle* mu2) const {
  
  const Trk::Perigee& mu1Perigee = mu1->perigeeParameters();
  const Trk::Perigee& mu2Perigee = mu2->perigeeParameters();
  int sflag = 0; int errorcode = 0;
  Amg::Vector3D startingPoint = m_vertexPointEstimator->getCirclesIntersectionPoint(&mu1Perigee,&mu2Perigee,sflag,errorcode);
  if (errorcode != 0) {startingPoint(0) = 0.0; startingPoint(1) = 0.0; startingPoint(2) = 0.0;}
  const std::vector<const xAOD::TrackParticle*> trackPair = {mu1, mu2};
  std::unique_ptr<xAOD::Vertex> myVxCandidate(m_vertexFitter->fit(trackPair, startingPoint)); 
 
  return myVxCandidate;
}
 
bool TrigBphysMonitorAlgorithm::matchDimuon(const xAOD::Vertex* dimuonVertex, const std::string& chainName) const {
  std::vector< TrigCompositeUtils::LinkInfo<xAOD::TrigBphysContainer> > chainFeatureContainer = getTrigDecisionTool()->features<xAOD::TrigBphysContainer>(chainName, TrigDefs::Physics);
  for (const auto& featLinkInfo: chainFeatureContainer){
    if(!featLinkInfo.isValid())
      return false;
    const auto featLink = featLinkInfo.link;
    if( matchDimuon(dimuonVertex, featLink) )
      return true;
  }
  return false;
}
 
bool TrigBphysMonitorAlgorithm::matchDimuon(const xAOD::Vertex* dimuonVertex, const ElementLink<xAOD::TrigBphysContainer>& bphysLink) const {
  auto offlineTrk1 = dimuonVertex->trackParticle(0);
  auto offlineTrk2 = dimuonVertex->trackParticle(1);
  if(!offlineTrk1 || !offlineTrk2) {
    ATH_MSG_DEBUG("TrackParticle from dimuon is null, won't match");
    return false;
  }
  
  const std::vector<ElementLink<xAOD::TrackParticleContainer> > trackVector = (*bphysLink)->trackParticleLinks();
  if (trackVector.size() < 2) {
      ATH_MSG_ERROR("Unexpected number of tracks in a dimuon: " << trackVector.size() << ", won't match");
      return false;
  }
  
  const xAOD::TrackParticle *hltTrk1(nullptr),*hltTrk2(nullptr);
  // Assume the first two tracks are always muons'
  if( !trackVector.at(0).isValid() || !trackVector.at(1).isValid() )
    return false;
  hltTrk1 = *trackVector.at(0);
  hltTrk2 = *trackVector.at(1);
  if (!hltTrk1 || !hltTrk2) {
      ATH_MSG_ERROR("Null pointer for track in a dimuon!");
      return false;
  }
  if( ( matchTrackParticles(offlineTrk1, hltTrk1) && matchTrackParticles(offlineTrk2, hltTrk2) ) || 
      ( matchTrackParticles(offlineTrk1, hltTrk2) && matchTrackParticles(offlineTrk2, hltTrk1) ) ) 
    return true;
  else
    return false;
}
 
bool TrigBphysMonitorAlgorithm::matchTrackParticles(const xAOD::TrackParticle* trk1, const xAOD::TrackParticle* trk2) const {
  double deltaR = xAOD::P4Helpers::deltaR(*trk1, *trk2);
  if( deltaR < m_deltaRMatchingThreshold)
    return true;
  else
    return false;
}
 
double TrigBphysMonitorAlgorithm::dimuonMass(const xAOD::TrackParticle* mu1, const xAOD::TrackParticle* mu2) const {
  using namespace TrigVtx;
  xAOD::TrackParticle::GenVecFourMom_t mom1 = mu1->genvecP4();
  xAOD::TrackParticle::GenVecFourMom_t mom2 = mu2->genvecP4();
  mom1.SetM(TrigParticleMasses().mass[TrigParticleName::muon]);
  mom2.SetM(TrigParticleMasses().mass[TrigParticleName::muon]);
  xAOD::TrackParticle::GenVecFourMom_t dimu_mom = mom1 + mom2;
  return dimu_mom.M();
}
 
std::vector<const xAOD::Vertex*> TrigBphysMonitorAlgorithm::GetGoodPVs(const xAOD::VertexContainer* pvContainer) const {
  std::vector<const xAOD::Vertex*> goodPrimaryVertices;
  goodPrimaryVertices.reserve(pvContainer->size());
  for (auto pv : *pvContainer) {
      xAOD::VxType::VertexType pvType = pv->vertexType();
      if ( pvType == xAOD::VxType::PriVtx || pvType == xAOD::VxType::PileUp ) {
        goodPrimaryVertices.push_back(pv);
      } 
  }
  return goodPrimaryVertices;
}
 
const xAOD::Vertex* TrigBphysMonitorAlgorithm::getPvForDimuon_lowestA0(const xAOD::Vertex* vtx, const std::vector<const xAOD::Vertex*>& PVs) const {
  std::vector<const xAOD::Vertex*>::const_iterator pv = std::min_element(PVs.begin(), PVs.end(), 
                                                                    [&, vtx](const xAOD::Vertex* pv1, const xAOD::Vertex* pv2)
                                                                    { return (std::abs(m_v0Tools->a0(vtx, pv1)) < std::abs(m_v0Tools->a0(vtx, pv2)));}
                                                                  );
  if(pv == PVs.end()) {
    return nullptr;
  }
  ATH_MSG_VERBOSE("Min-a0 PV has index " << std::distance(PVs.begin(), pv) << ", a0 = " << m_v0Tools->a0(vtx, *pv));
  return *pv;
}