MuonAssociatorByHits Class Reference

#include <MuonAssociatorByHits.h>

Public Member Functions
reco::RecoToSimCollection	associateRecoToSim (edm::Handle< edm::View< reco::Track >> &tCH, edm::Handle< TrackingParticleCollection > &tPCH, const edm::Event *event, const edm::EventSetup *setup) const
reco::RecoToSimCollection	associateRecoToSim (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &, const edm::Event *event=0, const edm::EventSetup *setup=0) const
	Association Reco To Sim with Collections. More...
virtual reco::SimToRecoCollection	associateSimToReco (edm::Handle< edm::View< reco::Track >> &tCH, edm::Handle< TrackingParticleCollection > &tPCH, const edm::Event *event, const edm::EventSetup *setup) const
reco::SimToRecoCollection	associateSimToReco (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &, const edm::Event *event=0, const edm::EventSetup *setup=0) const
	Association Sim To Reco with Collections. More...
	MuonAssociatorByHits (const edm::ParameterSet &conf, edm::ConsumesCollector &&iC)
virtual	~MuonAssociatorByHits ()

Private Attributes
edm::ParameterSet const	conf_
std::unique_ptr< muonAssociatorByHitsDiagnostics::InputDumper >	diagnostics_
MuonAssociatorByHitsHelper	helper_
TrackerHitAssociator::Config	trackerHitAssociatorConfig_

Detailed Description

Definition at line 25 of file MuonAssociatorByHits.h.

Constructor & Destructor Documentation

MuonAssociatorByHits::MuonAssociatorByHits	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector &&	iC
	)

Definition at line 137 of file MuonAssociatorByHits.cc.

References diagnostics_, edm::ParameterSet::getUntrackedParameter(), and eostools::move().

    : helper_(conf), conf_(conf), trackerHitAssociatorConfig_(conf, std::move(iC)) {
  // hack for consumes
  RPCHitAssociator rpctruth(conf, std::move(iC));
  GEMHitAssociator gemtruth(conf, std::move(iC));
  DTHitAssociator dttruth(conf, std::move(iC));
  CSCHitAssociator muonTruth(conf, std::move(iC));
  if (conf.getUntrackedParameter<bool>("dumpInputCollections")) {
    diagnostics_.reset(new InputDumper(conf, std::move(iC)));
  }
}

MuonAssociatorByHits::~MuonAssociatorByHits ( )

virtual

Definition at line 149 of file MuonAssociatorByHits.cc.

{}

Member Function Documentation

reco::RecoToSimCollection MuonAssociatorByHits::associateRecoToSim ( edm::Handle< edm::View< reco::Track >> &  tCH, edm::Handle< TrackingParticleCollection > &  tPCH, const edm::Event *  event, const edm::EventSetup *  setup  ) const

inline

Definition at line 32 of file MuonAssociatorByHits.h.

References edm::HandleBase::id(), dqmiolumiharvest::j, edm::RefToBaseVector< T >::push_back(), and edm::RefVector< C, T, F >::push_back().

Referenced by MuonAssociatorEDProducer::produce().

                                                                                 {
    edm::RefToBaseVector<reco::Track> tc;
    for (unsigned int j = 0; j < tCH->size(); j++)
      tc.push_back(tCH->refAt(j));

    edm::RefVector<TrackingParticleCollection> tpc(tPCH.id());
    for (unsigned int j = 0; j < tPCH->size(); j++)
      tpc.push_back(edm::Ref<TrackingParticleCollection>(tPCH, j));

    return associateRecoToSim(tc, tpc, event, setup);
  }

RecoToSimCollection MuonAssociatorByHits::associateRecoToSim	(	const edm::RefToBaseVector< reco::Track > &	tC,
		const edm::RefVector< TrackingParticleCollection > &	TPCollectionH,
		const edm::Event *	event = 0,
		const edm::EventSetup *	setup = 0
	)		const

Association Reco To Sim with Collections.

Definition at line 151 of file MuonAssociatorByHits.cc.

References MuonAssociatorByHitsHelper::associateRecoToSimIndices(), edm::RefToBaseVector< T >::begin(), conf_, MuonAssociatorByHitsHelper::Resources::diagnostics_, diagnostics_, MillePedeFileConverter_cfg::e, edm::RefToBaseVector< T >::end(), edm::EventSetup::get(), helper_, edm::ESHandle< T >::product(), edm::Event::productGetter(), mps_fire::resources, and trackerHitAssociatorConfig_.

                                      {
  RecoToSimCollection outputCollection(&e->productGetter());

  MuonAssociatorByHitsHelper::TrackHitsCollection tH;
  for (auto it = tC.begin(), ed = tC.end(); it != ed; ++it) {
    tH.push_back(std::make_pair((*it)->recHitsBegin(), (*it)->recHitsEnd()));
  }

  // Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHand;
  setup->get<TrackerTopologyRcd>().get(tTopoHand);
  const TrackerTopology *tTopo = tTopoHand.product();

  // Tracker hit association
  TrackerHitAssociator trackertruth(*e, trackerHitAssociatorConfig_);
  // CSC hit association
  CSCHitAssociator csctruth(*e, *setup, conf_);
  // DT hit association
  bool printRtS(true);
  DTHitAssociator dttruth(*e, *setup, conf_, printRtS);
  // RPC hit association
  RPCHitAssociator rpctruth(*e, *setup, conf_);
  // GEM hit association
  GEMHitAssociator gemtruth(*e, *setup, conf_);

  MuonAssociatorByHitsHelper::Resources resources = {
      tTopo, &trackertruth, &csctruth, &dttruth, &rpctruth, &gemtruth, {}};

  if (diagnostics_) {
    resources.diagnostics_ = [this, e](const TrackHitsCollection &hC, const TrackingParticleCollection &pC) {
      diagnostics_->dump(hC, pC, *e);
    };
  }

  auto bareAssoc = helper_.associateRecoToSimIndices(tH, TPCollectionH, resources);
  for (auto it = bareAssoc.begin(), ed = bareAssoc.end(); it != ed; ++it) {
    for (auto itma = it->second.begin(), edma = it->second.end(); itma != edma; ++itma) {
      outputCollection.insert(tC[it->first], std::make_pair(TPCollectionH[itma->idx], itma->quality));
    }
  }

  outputCollection.post_insert();  // perhaps not even necessary
  return outputCollection;
}

virtual reco::SimToRecoCollection MuonAssociatorByHits::associateSimToReco ( edm::Handle< edm::View< reco::Track >> &  tCH, edm::Handle< TrackingParticleCollection > &  tPCH, const edm::Event *  event, const edm::EventSetup *  setup  ) const

inlinevirtual

Definition at line 47 of file MuonAssociatorByHits.h.

References edm::HandleBase::id(), dqmiolumiharvest::j, edm::RefToBaseVector< T >::push_back(), edm::RefVector< C, T, F >::push_back(), and singleTopDQM_cfi::setup.

Referenced by MuonAssociatorEDProducer::produce().

                                                                                         {
    edm::RefToBaseVector<reco::Track> tc;
    for (unsigned int j = 0; j < tCH->size(); j++)
      tc.push_back(tCH->refAt(j));

    edm::RefVector<TrackingParticleCollection> tpc(tPCH.id());
    for (unsigned int j = 0; j < tPCH->size(); j++)
      tpc.push_back(edm::Ref<TrackingParticleCollection>(tPCH, j));

    return associateSimToReco(tc, tpc, event, setup);
  }

SimToRecoCollection MuonAssociatorByHits::associateSimToReco	(	const edm::RefToBaseVector< reco::Track > &	tC,
		const edm::RefVector< TrackingParticleCollection > &	TPCollectionH,
		const edm::Event *	event = 0,
		const edm::EventSetup *	setup = 0
	)		const

Association Sim To Reco with Collections.

Definition at line 200 of file MuonAssociatorByHits.cc.

References MuonAssociatorByHitsHelper::associateSimToRecoIndices(), edm::RefToBaseVector< T >::begin(), conf_, edm::RefToBaseVector< T >::end(), edm::EventSetup::get(), helper_, edm::ESHandle< T >::product(), edm::Event::productGetter(), mps_fire::resources, and trackerHitAssociatorConfig_.

                                      {
  SimToRecoCollection outputCollection(&e->productGetter());
  MuonAssociatorByHitsHelper::TrackHitsCollection tH;
  for (auto it = tC.begin(), ed = tC.end(); it != ed; ++it) {
    tH.push_back(std::make_pair((*it)->recHitsBegin(), (*it)->recHitsEnd()));
  }

  // Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHand;
  setup->get<TrackerTopologyRcd>().get(tTopoHand);
  const TrackerTopology *tTopo = tTopoHand.product();

  // Tracker hit association
  TrackerHitAssociator trackertruth(*e, trackerHitAssociatorConfig_);
  // CSC hit association
  CSCHitAssociator csctruth(*e, *setup, conf_);
  // DT hit association
  bool printRtS = false;
  DTHitAssociator dttruth(*e, *setup, conf_, printRtS);
  // RPC hit association
  RPCHitAssociator rpctruth(*e, *setup, conf_);
  // GEM hit association
  GEMHitAssociator gemtruth(*e, *setup, conf_);

  MuonAssociatorByHitsHelper::Resources resources = {
      tTopo, &trackertruth, &csctruth, &dttruth, &rpctruth, &gemtruth, {}};

  auto bareAssoc = helper_.associateSimToRecoIndices(tH, TPCollectionH, resources);
  for (auto it = bareAssoc.begin(), ed = bareAssoc.end(); it != ed; ++it) {
    for (auto itma = it->second.begin(), edma = it->second.end(); itma != edma; ++itma) {
      outputCollection.insert(TPCollectionH[it->first], std::make_pair(tC[itma->idx], itma->quality));
    }
  }

  outputCollection.post_insert();  // perhaps not even necessary
  return outputCollection;
}

Member Data Documentation

edm::ParameterSet const MuonAssociatorByHits::conf_

private

Definition at line 77 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().

std::unique_ptr<muonAssociatorByHitsDiagnostics::InputDumper> MuonAssociatorByHits::diagnostics_

private

Definition at line 80 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and MuonAssociatorByHits().

MuonAssociatorByHitsHelper MuonAssociatorByHits::helper_

private

Definition at line 76 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().

TrackerHitAssociator::Config MuonAssociatorByHits::trackerHitAssociatorConfig_

private

Definition at line 78 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().