xAODMultiMuonFilter.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#include "GeneratorFilters/xAODMultiMuonFilter.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthParticleAuxContainer.h"
#include "TruthUtils/HepMCHelpers.h"
 
xAODMultiMuonFilter::xAODMultiMuonFilter(const std::string& name, ISvcLocator* pSvcLocator)
  : GenFilter(name,pSvcLocator)
{
  declareProperty("Ptcut",m_Ptmin = 10000.);
  declareProperty("Etacut",m_EtaRange = 10.0);
  declareProperty("NMuons",m_NMuons = 2);
}
 
 
StatusCode xAODMultiMuonFilter::filterEvent() {  
  // Retrieve TruthMuons container
  const xAOD::TruthParticleContainer* xTruthParticleContainer;
  if (evtStore()->retrieve(xTruthParticleContainer, "TruthMuons").isFailure()) {
      ATH_MSG_ERROR("No TruthParticle collection with name " << "TruthMuons" << " found in StoreGate!");
      return StatusCode::FAILURE;
  }
  int numMuons = 0;
  unsigned int nParticles = xTruthParticleContainer->size();
  for (unsigned int iPart=0; iPart<nParticles; ++iPart) {
    const xAOD::TruthParticle* part = (*xTruthParticleContainer)[iPart];
 
    if (MC::isStable(part) && MC::isMuon(part)) //muon
      if(  part->pt()>= m_Ptmin && part->abseta() <= m_EtaRange )
      {
        numMuons++;
        if (numMuons >= m_NMuons)
        {
          setFilterPassed(true);
          return StatusCode::SUCCESS;
        }
      }
  }
 
  setFilterPassed(false);
  return StatusCode::SUCCESS;
 
}