MuonTrackCleanerBase Class Reference

#include <MuonTrackCleanerBase.h>

Inheritance diagram for MuonTrackCleanerBase:

Classes
struct	muonMomentumType
struct	todoListEntryType

Public Member Functions
	MuonTrackCleanerBase (const edm::ParameterSet &)
	~MuonTrackCleanerBase ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Protected Types
enum	{ kInnerTrack, kOuterTrack, kLink, kTeV }

Protected Member Functions
muonMomentumType	getMuonMomentum (const reco::Candidate &)
virtual void	produce (edm::Event &, const edm::EventSetup &)
virtual void	produceTrackExtras (edm::Event &, const edm::EventSetup &)=0
virtual void	produceTracks (edm::Event &, const edm::EventSetup &)
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Protected Attributes
double	dRmatch_
int	maxWarnings_tooFew_
int	maxWarnings_tooMany_
std::string	moduleLabel_
int	numWarnings_tooFew_
int	numWarnings_tooMany_
bool	removeDuplicates_
edm::InputTag	srcSelectedMuons_
std::vector< todoListEntryType >	todoList_
int	type_
int	verbosity_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Detailed Description

Base class for modules producing collections of reco::Tracks in Z –> mu+ mu- event from which the two muons are removed (later to be replaced by simulated tau decay products)

Authors

Christian Veelken, LLR

Version

Revision:

1.1

Id:

MuonTrackCleanerBase.h,v 1.1 2013/03/30 16:41:11 veelken Exp

Definition at line 30 of file MuonTrackCleanerBase.h.

Member Enumeration Documentation

anonymous enum

protected

Enumerator
kInnerTrack
kOuterTrack
kLink
kTeV

Definition at line 66 of file MuonTrackCleanerBase.h.

{ kInnerTrack, kOuterTrack, kLink, kTeV };

Constructor & Destructor Documentation

MuonTrackCleanerBase::MuonTrackCleanerBase ( const edm::ParameterSet &  cfg )

explicit

Definition at line 14 of file MuonTrackCleanerBase.cc.

References edm::hlt::Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), edm::InputTag::instance(), kInnerTrack, kLink, kOuterTrack, kTeV, MuonTrackCleanerBase::todoListEntryType::srcTracks_, AlCaHLTBitMon_QueryRunRegistry::string, todoList_, type_, and verbosity_.

  : moduleLabel_(cfg.getParameter<std::string>("@module_label")),
    srcSelectedMuons_(cfg.getParameter<edm::InputTag>("selectedMuons")),
    dRmatch_(cfg.getParameter<double>("dRmatch")),
    removeDuplicates_(cfg.getParameter<bool>("removeDuplicates")),
    maxWarnings_tooMany_(100),
    numWarnings_tooMany_(0),
    maxWarnings_tooFew_(3),
    numWarnings_tooFew_(0)
{
  std::string type_string = cfg.getParameter<std::string>("type");
  // CV: types defined in RecoMuon/MuonIdentification/python/muons1stStep_cfi.py
  if      ( type_string == "inner tracks" ) type_ = kInnerTrack;
  else if ( type_string == "outer tracks" ) type_ = kOuterTrack;
  else if ( type_string == "links"        ) type_ = kLink;
  else if ( type_string.find("tev") == 0  ) type_ = kTeV;
  else throw cms::Exception("Configuration") 
    << "Invalid Configuration Parameter 'type' = " << type_string << " !!\n";
  
  typedef std::vector<edm::InputTag> VInputTag;
  VInputTag srcTracks = cfg.getParameter<VInputTag>("tracks");
  for ( VInputTag::const_iterator srcTracks_i = srcTracks.begin();
    srcTracks_i != srcTracks.end(); ++srcTracks_i ) {
    todoListEntryType todoListEntry;
    todoListEntry.srcTracks_ = (*srcTracks_i);
    
    todoList_.push_back(todoListEntry); 

    produces<reco::TrackCollection>(todoListEntry.srcTracks_.instance());
  }

  verbosity_ = ( cfg.exists("verbosity") ) ?
    cfg.getParameter<int>("verbosity") : 0;
}

MuonTrackCleanerBase::~MuonTrackCleanerBase ( )

inline

Definition at line 34 of file MuonTrackCleanerBase.h.

{}

Member Function Documentation

MuonTrackCleanerBase::muonMomentumType MuonTrackCleanerBase::getMuonMomentum ( const reco::Candidate &  muon_candidate )

protected

Definition at line 98 of file MuonTrackCleanerBase.cc.

References reco::Candidate::eta(), MuonTrackCleanerBase::muonMomentumType::eta_, reco::Muon::globalTrack(), reco::Muon::innerTrack(), edm::Ref< C, T, F >::isNonnull(), kInnerTrack, kLink, kOuterTrack, kTeV, trackingTruthProducerFastSim_cfi::muon, reco::Muon::outerTrack(), reco::Candidate::phi(), MuonTrackCleanerBase::muonMomentumType::phi_, reco::Candidate::pt(), MuonTrackCleanerBase::muonMomentumType::pt_, and type_.

Referenced by produceTracks().

{
  muonMomentumType muonMomentum;
  muonMomentum.pt_  = muon_candidate.pt();
  muonMomentum.eta_ = muon_candidate.eta();
  muonMomentum.phi_ = muon_candidate.phi();

  const reco::Muon* muon = dynamic_cast<const reco::Muon*>(&muon_candidate);
  if ( muon ) {
    if ( type_ == kInnerTrack && muon->innerTrack().isNonnull() ) {
      muonMomentum.eta_ = muon->innerTrack()->eta();
      muonMomentum.phi_ = muon->innerTrack()->phi();
    } else if ( type_ == kOuterTrack && muon->outerTrack().isNonnull() ) {
      muonMomentum.eta_ = muon->outerTrack()->eta();
      muonMomentum.phi_ = muon->outerTrack()->phi();
    } else if ( (type_ == kLink || type_ == kTeV) && muon->globalTrack().isNonnull() ) {
      muonMomentum.eta_ = muon->globalTrack()->eta();
      muonMomentum.phi_ = muon->globalTrack()->phi();
    }
  } 

  return muonMomentum;
}

void MuonTrackCleanerBase::produce ( edm::Event &  evt, const edm::EventSetup &  es  )

protectedvirtual

Implements edm::EDProducer.

Definition at line 92 of file MuonTrackCleanerBase.cc.

References produceTrackExtras(), and produceTracks().

{
  produceTracks(evt, es);
  produceTrackExtras(evt, es);
}

virtual void MuonTrackCleanerBase::produceTrackExtras ( edm::Event &  , const edm::EventSetup &    )

protectedpure virtual

Implemented in TeVMuonTrackCleaner, GlobalMuonTrackCleaner, and MuonTrackCleaner.

Referenced by produce().

void MuonTrackCleanerBase::produceTracks ( edm::Event &  evt, const edm::EventSetup &  es  )

protectedvirtual

Definition at line 122 of file MuonTrackCleanerBase.cc.

References gather_cfg::cout, reco::deltaR(), PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, dRmatch_, reco::Candidate::eta(), edm::Event::getByLabel(), getMuonMomentum(), edm::Event::getRefBeforePut(), getSelMuons(), getTheMuMinus(), getTheMuPlus(), customizeTrackingMonitorSeedNumber::idx, reco::isMuon(), edm::RefToBase< T >::isNonnull(), kInnerTrack, kLink, Max(), maxWarnings_tooFew_, maxWarnings_tooMany_, numWarnings_tooFew_, numWarnings_tooMany_, reco::Candidate::phi(), reco::Candidate::pt(), edm::Event::put(), removeDuplicates_, python.multivaluedict::sort(), srcSelectedMuons_, todoList_, testEve_cfg::tracks, type_, and verbosity_.

Referenced by produce().

{
  if ( verbosity_ ) std::cout << "<MuonTrackCleanerBase::produceTracks>:" << std::endl;

  std::vector<reco::CandidateBaseRef> selMuons = getSelMuons(evt, srcSelectedMuons_);
  const reco::CandidateBaseRef muPlus  = getTheMuPlus(selMuons);
  const reco::CandidateBaseRef muMinus = getTheMuMinus(selMuons);

  std::vector<muonMomentumType> selMuonMomenta;
  if ( muPlus.isNonnull() ) {
    if ( verbosity_ ) std::cout << " muPlus: Pt = " << muPlus->pt() << ", eta = " << muPlus->eta() << ", phi = " << muPlus->phi() << std::endl;
    selMuonMomenta.push_back(getMuonMomentum(*muPlus));
  }
  if ( muMinus.isNonnull() ) {
    if ( verbosity_ ) std::cout << " muMinus: Pt = " << muMinus->pt() << ", eta = " << muMinus->eta() << ", phi = " << muMinus->phi() << std::endl;
    selMuonMomenta.push_back(getMuonMomentum(*muMinus));
  }

//--- produce collection of reco::Tracks excluding muons
  for ( typename std::vector<todoListEntryType>::const_iterator todoItem = todoList_.begin();
    todoItem != todoList_.end(); ++todoItem ) {
    if ( verbosity_ ) std::cout << "processing trackCollection = " << todoItem->srcTracks_ << std::endl;
    
    todoItem->trackRefMap_.clear();
    
    edm::Handle<reco::TrackCollection> tracks;
    evt.getByLabel(todoItem->srcTracks_, tracks);
    
    std::auto_ptr<reco::TrackCollection> tracks_cleaned(new reco::TrackCollection());

    reco::TrackRefProd trackCollectionRefProd_cleaned = evt.getRefBeforePut<reco::TrackCollection>(todoItem->srcTracks_.instance());
    size_t idxTrack_cleaned = 0;
    
    std::vector<muonToTrackMatchInfoType> selMuonToTrackMatches;
    for ( std::vector<muonMomentumType>::const_iterator selMuonMomentum = selMuonMomenta.begin();
      selMuonMomentum != selMuonMomenta.end(); ++selMuonMomentum ) {
      std::vector<muonToTrackMatchInfoType> tmpMatches;
      for ( reco::TrackCollection::const_iterator track = tracks->begin();
        track != tracks->end(); ++track ) {
    double dR = reco::deltaR(track->eta(), track->phi(), selMuonMomentum->eta_, selMuonMomentum->phi_);
    if ( dR < dRmatch_ ) tmpMatches.push_back(muonToTrackMatchInfoType(selMuonMomentum->pt_, &(*track), dR));
      }
      // rank muon-to-track matches by quality
      std::sort(tmpMatches.begin(), tmpMatches.end(), SortMuonToTrackMatchInfosDescendingMatchQuality());
      if ( tmpMatches.size() > 0 ) selMuonToTrackMatches.push_back(tmpMatches.front());
      if ( removeDuplicates_ ) {
    // CV: remove all high Pt tracks very close to muon direction
    //    (duplicate tracks arise in case muon track in SiStrip + Pixel detector is reconstructed as 2 disjoint segments)
    for ( std::vector<muonToTrackMatchInfoType>::const_iterator tmpMatch = tmpMatches.begin();
          tmpMatch != tmpMatches.end(); ++tmpMatch ) {
      if ( (tmpMatch->dR_ < 1.e-3 && tmpMatch->trackPt_ >           (0.33*tmpMatch->muonPt_))      ||
           (tmpMatch->dR_ < 1.e-1 && tmpMatch->trackPt_ > TMath::Max(0.66*tmpMatch->muonPt_, 10.)) ) selMuonToTrackMatches.push_back(*tmpMatch);
    }
      }
    }
    
    std::vector<reco::TrackRef> removedTracks;
    size_t numTracks = tracks->size();
    for ( size_t trackIdx = 0; trackIdx < numTracks; ++trackIdx ) {
      reco::TrackRef track(tracks, trackIdx);
      bool isMuon = false;
      for ( std::vector<muonToTrackMatchInfoType>::const_iterator muonMatchInfo = selMuonToTrackMatches.begin();
        muonMatchInfo != selMuonToTrackMatches.end(); ++muonMatchInfo ) {
    if ( muonMatchInfo->track_ == &(*track) ) isMuon = true;
      }
      if ( verbosity_ && track->pt() > 5. ) {
    std::cout << "track: Pt = " << track->pt() << ", eta = " << track->eta() << ", phi = " << track->phi() << ", isMuon = " << isMuon << std::endl;
      }
      if ( isMuon ) {
    removedTracks.push_back(track); // track belongs to a selected muon, do not copy
      } else { 
    tracks_cleaned->push_back(*track);
    todoItem->trackRefMap_[reco::TrackRef(trackCollectionRefProd_cleaned, idxTrack_cleaned)] = track;
    ++idxTrack_cleaned;
      }
    }
    if ( (type_ >= kInnerTrack && type_ <= kLink       && removedTracks.size() > selMuons.size() && numWarnings_tooMany_ < maxWarnings_tooMany_) &&
     (type_ >= kInnerTrack && type_ <= kInnerTrack && removedTracks.size() < selMuons.size() && numWarnings_tooFew_  < maxWarnings_tooFew_ ) ) {
      edm::LogWarning("MuonTrackCleanerBase") 
    << " (" << runLumiEventNumbers_to_string(evt) << ")" << std::endl
    << " Removed " << removedTracks.size() << " tracks from event containing " << selMuons.size() << " muons !!" << std::endl;
      if ( muPlus.isNonnull() ) std::cout << " muPlus: Pt = " << muPlus->pt() << ", eta = " << muPlus->eta() << ", phi = " << muPlus->phi() << std::endl;
      if ( muMinus.isNonnull() ) std::cout << " muMinus: Pt = " << muMinus->pt() << ", eta = " << muMinus->eta() << ", phi = " << muMinus->phi() << std::endl;
      std::cout << "Removed tracks:" << std::endl;
      int idx = 0;
      for ( std::vector<reco::TrackRef>::const_iterator removedTrack = removedTracks.begin();
        removedTrack != removedTracks.end(); ++removedTrack ) {
    std::cout << "track #" << idx << ":" 
          << " Pt = " << (*removedTrack)->pt() << ", eta = " << (*removedTrack)->eta() << ", phi = " << (*removedTrack)->phi() << std::endl;
    ++idx;
      }
      if ( removedTracks.size() > selMuons.size() ) ++numWarnings_tooMany_;
      if ( removedTracks.size() < selMuons.size() ) ++numWarnings_tooFew_;
    }
    
    evt.put(tracks_cleaned, todoItem->srcTracks_.instance());
  }
}

Member Data Documentation

double MuonTrackCleanerBase::dRmatch_

protected

Definition at line 63 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

int MuonTrackCleanerBase::maxWarnings_tooFew_

protected

Definition at line 71 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

int MuonTrackCleanerBase::maxWarnings_tooMany_

protected

Definition at line 69 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

std::string MuonTrackCleanerBase::moduleLabel_

protected

Definition at line 39 of file MuonTrackCleanerBase.h.

Referenced by Mixins._TypedParameterizable::insertInto(), Modules.ESSource::nameInProcessDesc_(), Modules.ESProducer::nameInProcessDesc_(), GlobalMuonTrackCleaner::produceTrackExtras(), and TeVMuonTrackCleaner::produceTrackExtras().

int MuonTrackCleanerBase::numWarnings_tooFew_

protected

Definition at line 72 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

int MuonTrackCleanerBase::numWarnings_tooMany_

protected

Definition at line 70 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

bool MuonTrackCleanerBase::removeDuplicates_

protected

Definition at line 64 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

edm::InputTag MuonTrackCleanerBase::srcSelectedMuons_

protected

Definition at line 52 of file MuonTrackCleanerBase.h.

Referenced by produceTracks().

std::vector<todoListEntryType> MuonTrackCleanerBase::todoList_

protected

Definition at line 61 of file MuonTrackCleanerBase.h.

Referenced by GlobalMuonTrackCleaner::GlobalMuonTrackCleaner(), MuonTrackCleanerBase(), GlobalMuonTrackCleaner::produceTrackExtras(), TeVMuonTrackCleaner::produceTrackExtras(), produceTracks(), and TeVMuonTrackCleaner::TeVMuonTrackCleaner().

int MuonTrackCleanerBase::type_

protected

Definition at line 67 of file MuonTrackCleanerBase.h.

Referenced by Modules.Service::_placeImpl(), Modules.ESSource::_placeImpl(), Modules.ESProducer::_placeImpl(), Modules.ESPrefer::copy(), Mixins._TypedParameterizable::dumpPython(), Modules.ESPrefer::dumpPythonAs(), getMuonMomentum(), Modules.Service::insertInto(), RandomRunSource.RandomRunSource::insertInto(), Mixins._TypedParameterizable::insertInto(), Modules.ESSource::moduleLabel_(), Modules.ESProducer::moduleLabel_(), MuonTrackCleanerBase(), Modules.ESSource::nameInProcessDesc_(), Modules.ESProducer::nameInProcessDesc_(), Modules.ESPrefer::nameInProcessDesc_(), and produceTracks().

int MuonTrackCleanerBase::verbosity_

protected

Definition at line 74 of file MuonTrackCleanerBase.h.

Referenced by MuonTrackCleanerBase(), GlobalMuonTrackCleaner::produceTrackExtras(), TeVMuonTrackCleaner::produceTrackExtras(), and produceTracks().