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 138 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 155 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 33 of file MuonAssociatorByHits.h.

References edm::HandleBase::id(), 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 160 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 48 of file MuonAssociatorByHits.h.

References edm::HandleBase::id(), edm::RefToBaseVector< T >::push_back(), edm::RefVector< C, T, F >::push_back(), and GeneralSetup::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 208 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 78 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().

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

private

Definition at line 81 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and MuonAssociatorByHits().

MuonAssociatorByHitsHelper MuonAssociatorByHits::helper_

private

Definition at line 77 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().

TrackerHitAssociator::Config MuonAssociatorByHits::trackerHitAssociatorConfig_

private

Definition at line 79 of file MuonAssociatorByHits.h.

Referenced by associateRecoToSim(), and associateSimToReco().