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::MTDTrackQualityMVAProducer ( const ParameterSet & pset )

Definition at line 53 of file MTDTrackQualityMVAProducer.cc.

     : 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);
 }

References mvaName.

Member Function Documentation

◆ fillDescriptions()

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

static

Definition at line 70 of file MTDTrackQualityMVAProducer.cc.

                                                                                             {
   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);
 }

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

◆ fillValueMap()

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.

                                                                            {
   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);
 }

References trigObjTnPSource_cfi::filler, patZpeak::handle, iEvent, eostools::move(), Skims_PA_cff::name, and MillePedeFileConverter_cfg::out.

Referenced by produce().

◆ produce()

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

final

Definition at line 107 of file MTDTrackQualityMVAProducer.cc.

                                                                               {
   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);
 }