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	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () 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(), 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 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);
 }