MTDTrackQualityMVAProducer Class Reference

Inheritance diagram for MTDTrackQualityMVAProducer:

Public Member Functions
template<class H , class T >
void	fillValueMap (edm::Event &iEvent, const edm::Handle< H > &handle, const std::vector< T > &vec, const std::string &name) const
	MTDTrackQualityMVAProducer (const ParameterSet &pset)
void	produce (edm::Event &ev, const edm::EventSetup &es) final
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
edm::EDGetTokenT < edm::ValueMap< float > >	btlMatchChi2Token_
edm::EDGetTokenT < edm::ValueMap< float > >	btlMatchTimeChi2Token_
edm::EDGetTokenT < edm::ValueMap< float > >	etlMatchChi2Token_
edm::EDGetTokenT < edm::ValueMap< float > >	etlMatchTimeChi2Token_
edm::EDGetTokenT < edm::ValueMap< float > >	mtdTimeToken_
MTDTrackQualityMVA	mva_
edm::EDGetTokenT < edm::ValueMap< int > >	npixBarrelToken_
edm::EDGetTokenT < edm::ValueMap< int > >	npixEndcapToken_
edm::EDGetTokenT < edm::ValueMap< float > >	pathLengthToken_
edm::EDGetTokenT < reco::TrackCollection >	tracksMTDToken_
edm::EDGetTokenT < reco::TrackCollection >	tracksToken_

Static Private Attributes
static constexpr char	mvaName [] = "mtdQualMVA"

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T...>
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 21 of file MTDTrackQualityMVAProducer.cc.

Constructor & Destructor Documentation

MTDTrackQualityMVAProducer::MTDTrackQualityMVAProducer

(

const ParameterSet &

pset

)

Definition at line 53 of file MTDTrackQualityMVAProducer.cc.

References mvaName.

    : tracksToken_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracksSrc"))),
      btlMatchChi2Token_(consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("btlMatchChi2Src"))),
      btlMatchTimeChi2Token_(
          consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("btlMatchTimeChi2Src"))),
      etlMatchChi2Token_(consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("etlMatchChi2Src"))),
      etlMatchTimeChi2Token_(
          consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("etlMatchTimeChi2Src"))),
      mtdTimeToken_(consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("mtdTimeSrc"))),
      pathLengthToken_(consumes<edm::ValueMap<float>>(iConfig.getParameter<edm::InputTag>("pathLengthSrc"))),
      npixBarrelToken_(consumes<edm::ValueMap<int>>(iConfig.getParameter<edm::InputTag>("npixBarrelSrc"))),
      npixEndcapToken_(consumes<edm::ValueMap<int>>(iConfig.getParameter<edm::InputTag>("npixEndcapSrc"))),
      mva_(iConfig.getParameter<edm::FileInPath>("qualityBDT_weights_file").fullPath()) {
  produces<edm::ValueMap<float>>(mvaName);
}

Member Function Documentation

void MTDTrackQualityMVAProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 70 of file MTDTrackQualityMVAProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), submitPVResolutionJobs::desc, and HLT_FULL_cff::InputTag.

                                                                                            {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("tracksSrc", edm::InputTag("generalTracks"))->setComment("Input tracks collection");
  desc.add<edm::InputTag>("btlMatchChi2Src", edm::InputTag("trackExtenderWithMTD", "btlMatchChi2"))
      ->setComment("BTL Chi2 Matching value Map");
  desc.add<edm::InputTag>("btlMatchTimeChi2Src", edm::InputTag("trackExtenderWithMTD", "btlMatchTimeChi2"))
      ->setComment("BTL Chi2 Matching value Map");
  desc.add<edm::InputTag>("etlMatchChi2Src", edm::InputTag("trackExtenderWithMTD", "etlMatchChi2"))
      ->setComment("ETL Chi2 Matching value Map");
  desc.add<edm::InputTag>("etlMatchTimeChi2Src", edm::InputTag("trackExtenderWithMTD", "etlMatchTimeChi2"))
      ->setComment("ETL Chi2 Matching value Map");
  desc.add<edm::InputTag>("mtdTimeSrc", edm::InputTag("trackExtenderWithMTD", "generalTracktmtd"))
      ->setComment("MTD TIme value Map");
  desc.add<edm::InputTag>("pathLengthSrc", edm::InputTag("trackExtenderWithMTD", "generalTrackPathLength"))
      ->setComment("MTD PathLength value Map");
  desc.add<edm::InputTag>("npixBarrelSrc", edm::InputTag("trackExtenderWithMTD", "npixBarrel"))
      ->setComment("# of Barrel pixel associated to refitted tracks");
  desc.add<edm::InputTag>("npixEndcapSrc", edm::InputTag("trackExtenderWithMTD", "npixEndcap"))
      ->setComment("# of Endcap pixel associated to refitted tracks");
  desc.add<edm::FileInPath>("qualityBDT_weights_file",
                            edm::FileInPath("RecoMTD/TimingIDTools/data/clf4D_MTDquality_bo.xml"))
      ->setComment("Track MTD quality BDT weights");
  descriptions.add("mtdTrackQualityMVAProducer", desc);
}

template<class H , class T >

void MTDTrackQualityMVAProducer::fillValueMap	(	edm::Event &	iEvent,
		const edm::Handle< H > &	handle,
		const std::vector< T > &	vec,
		const std::string &	name
	)		const

Definition at line 96 of file MTDTrackQualityMVAProducer.cc.

References eostools::move(), submitPVResolutionJobs::out, and edm::Event::put().

Referenced by produce().

                                                                           {
  auto out = std::make_unique<edm::ValueMap<T>>();
  typename edm::ValueMap<T>::Filler filler(*out);
  filler.insert(handle, vec.begin(), vec.end());
  filler.fill();
  iEvent.put(std::move(out), name);
}

void MTDTrackQualityMVAProducer::produce ( edm::Event &  ev, const edm::EventSetup &  es  )

final

Definition at line 107 of file MTDTrackQualityMVAProducer.cc.

References btlMatchChi2Token_, btlMatchTimeChi2Token_, etlMatchChi2Token_, etlMatchTimeChi2Token_, fillValueMap(), edm::Event::get(), edm::Event::getByToken(), mtdTimeToken_, mva_, mvaName, npixBarrelToken_, npixEndcapToken_, pathLengthToken_, tracks, and tracksToken_.

                                                                              {
  edm::Handle<reco::TrackCollection> tracksH;
  ev.getByToken(tracksToken_, tracksH);
  const auto& tracks = *tracksH;

  const auto& btlMatchChi2 = ev.get(btlMatchChi2Token_);
  const auto& btlMatchTimeChi2 = ev.get(btlMatchTimeChi2Token_);
  const auto& etlMatchChi2 = ev.get(etlMatchChi2Token_);
  const auto& etlMatchTimeChi2 = ev.get(etlMatchTimeChi2Token_);
  const auto& pathLength = ev.get(pathLengthToken_);
  const auto& npixBarrel = ev.get(npixBarrelToken_);
  const auto& npixEndcap = ev.get(npixEndcapToken_);
  const auto& mtdTime = ev.get(mtdTimeToken_);

  std::vector<float> mvaOutRaw;

  //Loop over tracks collection
  for (unsigned int itrack = 0; itrack < tracks.size(); ++itrack) {
    const reco::TrackRef trackref(tracksH, itrack);
    if (pathLength[trackref] == -1.)
      mvaOutRaw.push_back(-1.);
    else {
      mvaOutRaw.push_back(mva_(trackref,
                               npixBarrel,
                               npixEndcap,
                               btlMatchChi2,
                               btlMatchTimeChi2,
                               etlMatchChi2,
                               etlMatchTimeChi2,
                               mtdTime,
                               pathLength));
    }
  }
  fillValueMap(ev, tracksH, mvaOutRaw, mvaName);
}

Member Data Documentation

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::btlMatchChi2Token_

private

Definition at line 41 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::btlMatchTimeChi2Token_

private

Definition at line 42 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::etlMatchChi2Token_

private

Definition at line 43 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::etlMatchTimeChi2Token_

private

Definition at line 44 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::mtdTimeToken_

private

Definition at line 45 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

MTDTrackQualityMVA MTDTrackQualityMVAProducer::mva_

private

Definition at line 50 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

constexpr char MTDTrackQualityMVAProducer::mvaName[] = "mtdQualMVA"

staticprivate

Definition at line 36 of file MTDTrackQualityMVAProducer.cc.

Referenced by MTDTrackQualityMVAProducer(), and produce().

edm::EDGetTokenT<edm::ValueMap<int> > MTDTrackQualityMVAProducer::npixBarrelToken_

private

Definition at line 47 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<int> > MTDTrackQualityMVAProducer::npixEndcapToken_

private

Definition at line 48 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<edm::ValueMap<float> > MTDTrackQualityMVAProducer::pathLengthToken_

private

Definition at line 46 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().

edm::EDGetTokenT<reco::TrackCollection> MTDTrackQualityMVAProducer::tracksMTDToken_

private

Definition at line 39 of file MTDTrackQualityMVAProducer.cc.

edm::EDGetTokenT<reco::TrackCollection> MTDTrackQualityMVAProducer::tracksToken_

private

Definition at line 38 of file MTDTrackQualityMVAProducer.cc.

Referenced by produce().