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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

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< float > >	pathLengthToken_

edm::EDGetTokenT< edm::ValueMap< int > >	trackAssocToken_

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<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 21 of file MTDTrackQualityMVAProducer.cc.

Constructor & Destructor Documentation

◆ MTDTrackQualityMVAProducer()

MTDTrackQualityMVAProducer::MTDTrackQualityMVAProducer ( const ParameterSet & pset )

Definition at line 52 of file MTDTrackQualityMVAProducer.cc.

     : tracksToken_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracksSrc"))),
       tracksMTDToken_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracksMTDSrc"))),
       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"))),
       trackAssocToken_(consumes<edm::ValueMap<int>>(iConfig.getParameter<edm::InputTag>("trackAssocSrc"))),
       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 69 of file MTDTrackQualityMVAProducer.cc.

                                                                                             {
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>("tracksSrc", edm::InputTag("generalTracks"))->setComment("Input tracks collection");
   desc.add<edm::InputTag>("tracksMTDSrc", edm::InputTag("trackExtenderWithMTD"))
       ->setComment("Input tracks collection for MTD extended tracks");
   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", "tmtd"))
       ->setComment("MTD TIme value Map");
   desc.add<edm::InputTag>("pathLengthSrc", edm::InputTag("trackExtenderWithMTD", "pathLength"))
       ->setComment("MTD PathLength value Map");
   desc.add<edm::InputTag>("trackAssocSrc", edm::InputTag("trackExtenderWithMTD", "generalTrackassoc"))
       ->setComment("Association between General and MTD Extended 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(), edm::ParameterSetDescription::add(), and HLT_2018_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 95 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 106 of file MTDTrackQualityMVAProducer.cc.

                                                                               {
   edm::Handle<reco::TrackCollection> tracksH;
   ev.getByToken(tracksToken_, tracksH);
   const auto& tracks = *tracksH;
  
   edm::Handle<reco::TrackCollection> tracksMTDH;
   ev.getByToken(tracksMTDToken_, tracksMTDH);
  
   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& trackAssoc = ev.get(trackAssocToken_);
   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 (trackAssoc[trackref] == -1)
       mvaOutRaw.push_back(-1.);
     else {
       const reco::TrackRef mtdTrackref = reco::TrackRef(tracksMTDH, trackAssoc[trackref]);
       mvaOutRaw.push_back(mva_(
           trackref, mtdTrackref, btlMatchChi2, btlMatchTimeChi2, etlMatchChi2, etlMatchTimeChi2, mtdTime, pathLength));
     }
   }
   fillValueMap(ev, tracksH, mvaOutRaw, mvaName);
 }