#include <TtFullHadKinFitProducer.h>

Inheritance diagram for TtFullHadKinFitProducer:

Public Member Functions
	TtFullHadKinFitProducer (const edm::ParameterSet &cfg)
	default constructor More...

	~TtFullHadKinFitProducer () override
	default destructor More...

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

SerialTaskQueue *	globalLuminosityBlocksQueue ()

SerialTaskQueue *	globalRunsQueue ()

ModuleDescription const &	moduleDescription () const

	~EDProducer () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () noexcept(false) override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Public Attributes
TtFullHadKinFitter::KinFit *	kinFitter
	kinematic fit interface More...

Private Member Functions
void	produce (edm::Event &event, const edm::EventSetup &setup) override
	produce fitted object collections and meta data describing fit quality More...

Private Attributes
std::vector< edm::ParameterSet >	bResolutions_

std::string	bTagAlgo_
	input tag for b-tagging algorithm More...

unsigned int	bTags_
	minimal number of b-jets More...

std::vector< unsigned >	constraints_
	numbering of different possible kinematic constraints More...

std::string	jetCorrectionLevel_
	correction level for jets More...

std::vector< double >	jetEnergyResolutionEtaBinning_

std::vector< double >	jetEnergyResolutionScaleFactors_
	scale factors for jet energy resolution More...

unsigned int	jetParam_
	numbering of different possible jet parametrizations More...

edm::EDGetTokenT< std::vector< pat::Jet > >	jetsToken_
	input tag for jets More...

edm::EDGetTokenT< std::vector< std::vector< int > > >	matchToken_
	input tag for matches (in case the fit should be performed on certain matches) More...

double	maxBTagValueNonBJet_
	max value of bTag for a non-b-jet More...

double	maxDeltaS_
	maximal chi2 equivalent More...

double	maxF_
	maximal deviation for contstraints More...

int	maxNComb_
	maximal number of combinations to be written to the event More...

int	maxNJets_
	maximal number of jets (-1 possible to indicate 'all') More...

unsigned int	maxNrIter_
	maximal number of iterations to be performed for the fit More...

double	minBTagValueBJet_
	min value of bTag for a b-jet More...

double	mTop_
	top mass value used for constraints More...

double	mW_
	W mass value used for constraints. More...

std::vector< edm::ParameterSet >	udscResolutions_
	store the resolutions for the jets More...

bool	useBTagging_
	switch to tell whether to use b-tagging or not More...

bool	useOnlyMatch_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

static bool	wantsGlobalLuminosityBlocks ()

static bool	wantsGlobalRuns ()

static bool	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 22 of file TtFullHadKinFitProducer.h.

Constructor & Destructor Documentation

TtFullHadKinFitProducer::TtFullHadKinFitProducer ( const edm::ParameterSet & cfg )

explicit

default constructor

Definition at line 6 of file TtFullHadKinFitProducer.cc.

References bResolutions_, bTagAlgo_, bTags_, constraints_, Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), jetCorrectionLevel_, jetEnergyResolutionEtaBinning_, jetEnergyResolutionScaleFactors_, jetParam_, kinFitter, maxBTagValueNonBJet_, maxDeltaS_, maxF_, maxNComb_, maxNJets_, maxNrIter_, minBTagValueBJet_, mTop_, mW_, udscResolutions_, and useBTagging_.

     : jetsToken_(consumes<std::vector<pat::Jet> >(cfg.getParameter<edm::InputTag>("jets"))),
       matchToken_(mayConsume<std::vector<std::vector<int> > >(cfg.getParameter<edm::InputTag>("match"))),
       useOnlyMatch_(cfg.getParameter<bool>("useOnlyMatch")),
       bTagAlgo_(cfg.getParameter<std::string>("bTagAlgo")),
       minBTagValueBJet_(cfg.getParameter<double>("minBTagValueBJet")),
       maxBTagValueNonBJet_(cfg.getParameter<double>("maxBTagValueNonBJet")),
       useBTagging_(cfg.getParameter<bool>("useBTagging")),
       bTags_(cfg.getParameter<unsigned int>("bTags")),
       jetCorrectionLevel_(cfg.getParameter<std::string>("jetCorrectionLevel")),
       maxNJets_(cfg.getParameter<int>("maxNJets")),
       maxNComb_(cfg.getParameter<int>("maxNComb")),
       maxNrIter_(cfg.getParameter<unsigned int>("maxNrIter")),
       maxDeltaS_(cfg.getParameter<double>("maxDeltaS")),
       maxF_(cfg.getParameter<double>("maxF")),
       jetParam_(cfg.getParameter<unsigned>("jetParametrisation")),
       constraints_(cfg.getParameter<std::vector<unsigned> >("constraints")),
       mW_(cfg.getParameter<double>("mW")),
       mTop_(cfg.getParameter<double>("mTop")),
       jetEnergyResolutionScaleFactors_(cfg.getParameter<std::vector<double> >("jetEnergyResolutionScaleFactors")),
       jetEnergyResolutionEtaBinning_(cfg.getParameter<std::vector<double> >("jetEnergyResolutionEtaBinning")) {
   if (cfg.exists("udscResolutions") && cfg.exists("bResolutions")) {
     udscResolutions_ = cfg.getParameter<std::vector<edm::ParameterSet> >("udscResolutions");
     bResolutions_ = cfg.getParameter<std::vector<edm::ParameterSet> >("bResolutions");
   } else if (cfg.exists("udscResolutions") || cfg.exists("bResolutions")) {
     if (cfg.exists("udscResolutions"))
       throw cms::Exception("Configuration")
           << "Parameter 'bResolutions' is needed if parameter 'udscResolutions' is defined!\n";
     else
       throw cms::Exception("Configuration")
           << "Parameter 'udscResolutions' is needed if parameter 'bResolutions' is defined!\n";
   }
 
   // define kinematic fit interface
   kinFitter = new TtFullHadKinFitter::KinFit(useBTagging_,
                                              bTags_,
                                              bTagAlgo_,
                                              minBTagValueBJet_,
                                              maxBTagValueNonBJet_,
                                              udscResolutions_,
                                              bResolutions_,
                                              jetEnergyResolutionScaleFactors_,
                                              jetEnergyResolutionEtaBinning_,
                                              jetCorrectionLevel_,
                                              maxNJets_,
                                              maxNComb_,
                                              maxNrIter_,
                                              maxDeltaS_,
                                              maxF_,
                                              jetParam_,
                                              constraints_,
                                              mW_,
                                              mTop_);
 
   // produces the following collections
   produces<std::vector<pat::Particle> >("PartonsB");
   produces<std::vector<pat::Particle> >("PartonsBBar");
   produces<std::vector<pat::Particle> >("PartonsLightQ");
   produces<std::vector<pat::Particle> >("PartonsLightQBar");
   produces<std::vector<pat::Particle> >("PartonsLightP");
   produces<std::vector<pat::Particle> >("PartonsLightPBar");
 
   produces<std::vector<std::vector<int> > >();
   produces<std::vector<double> >("Chi2");
   produces<std::vector<double> >("Prob");
   produces<std::vector<int> >("Status");
 }

TtFullHadKinFitProducer::~TtFullHadKinFitProducer ( )

override

default destructor

Definition at line 75 of file TtFullHadKinFitProducer.cc.

References kinFitter.

75 { delete kinFitter; }

TtFullHadKinFitProducer::kinFitter

TtFullHadKinFitter::KinFit * kinFitter

kinematic fit interface

Definition: TtFullHadKinFitProducer.h:79

Member Function Documentation

void TtFullHadKinFitProducer::produce	(	edm::Event &	event,
		const edm::EventSetup &	setup
	)

overrideprivate

produce fitted object collections and meta data describing fit quality

set match to be used

set the validity of a match

Definition at line 78 of file TtFullHadKinFitProducer.cc.

References DEFINE_FWK_MODULE, TtFullHadKinFitter::KinFit::fit(), training_settings::idx, singleTopDQM_cfi::jets, jetsToken_, kinFitter, match(), patCandidatesForDimuonsSequences_cff::matches, matchToken_, maxNComb_, eostools::move(), nPartons, TtFullHadKinFitter::KinFit::setMatch(), TtFullHadKinFitter::KinFit::setMatchInvalidity(), TtFullHadKinFitter::KinFit::setUseOnlyMatch(), and useOnlyMatch_.

                                                                                  {
   // get jet collection
   edm::Handle<std::vector<pat::Jet> > jets;
   event.getByToken(jetsToken_, jets);
 
   // get match in case that useOnlyMatch_ is true
   std::vector<int> match;
   bool invalidMatch = false;
   if (useOnlyMatch_) {
     kinFitter->setUseOnlyMatch(true);
     // in case that only a ceratin match should be used, get match here
     edm::Handle<std::vector<std::vector<int> > > matches;
     event.getByToken(matchToken_, matches);
     match = *(matches->begin());
     // check if match is valid
     if (match.size() != nPartons) {
       invalidMatch = true;
     } else {
       for (unsigned int idx = 0; idx < match.size(); ++idx) {
         if (match[idx] < 0 || match[idx] >= (int)jets->size()) {
           invalidMatch = true;
           break;
         }
       }
     }
     kinFitter->setMatch(match);
   }
 
   kinFitter->setMatchInvalidity(invalidMatch);
 
   std::list<TtFullHadKinFitter::KinFitResult> fitResults = kinFitter->fit(*jets);
 
   // pointer for output collections
   std::unique_ptr<std::vector<pat::Particle> > pPartonsB(new std::vector<pat::Particle>);
   std::unique_ptr<std::vector<pat::Particle> > pPartonsBBar(new std::vector<pat::Particle>);
   std::unique_ptr<std::vector<pat::Particle> > pPartonsLightQ(new std::vector<pat::Particle>);
   std::unique_ptr<std::vector<pat::Particle> > pPartonsLightQBar(new std::vector<pat::Particle>);
   std::unique_ptr<std::vector<pat::Particle> > pPartonsLightP(new std::vector<pat::Particle>);
   std::unique_ptr<std::vector<pat::Particle> > pPartonsLightPBar(new std::vector<pat::Particle>);
   // pointer for meta information
   std::unique_ptr<std::vector<std::vector<int> > > pCombi(new std::vector<std::vector<int> >);
   std::unique_ptr<std::vector<double> > pChi2(new std::vector<double>);
   std::unique_ptr<std::vector<double> > pProb(new std::vector<double>);
   std::unique_ptr<std::vector<int> > pStatus(new std::vector<int>);
 
   unsigned int iComb = 0;
   for (std::list<TtFullHadKinFitter::KinFitResult>::const_iterator res = fitResults.begin(); res != fitResults.end();
        ++res) {
     if (maxNComb_ >= 1 && iComb == (unsigned int)maxNComb_) {
       break;
     }
     ++iComb;
 
     pPartonsB->push_back(res->B);
     pPartonsBBar->push_back(res->BBar);
     pPartonsLightQ->push_back(res->LightQ);
     pPartonsLightQBar->push_back(res->LightQBar);
     pPartonsLightP->push_back(res->LightP);
     pPartonsLightPBar->push_back(res->LightPBar);
 
     pCombi->push_back(res->JetCombi);
     pChi2->push_back(res->Chi2);
     pProb->push_back(res->Prob);
     pStatus->push_back(res->Status);
   }
 
   event.put(std::move(pCombi));
   event.put(std::move(pPartonsB), "PartonsB");
   event.put(std::move(pPartonsBBar), "PartonsBBar");
   event.put(std::move(pPartonsLightQ), "PartonsLightQ");
   event.put(std::move(pPartonsLightQBar), "PartonsLightQBar");
   event.put(std::move(pPartonsLightP), "PartonsLightP");
   event.put(std::move(pPartonsLightPBar), "PartonsLightPBar");
   event.put(std::move(pChi2), "Chi2");
   event.put(std::move(pProb), "Prob");
   event.put(std::move(pStatus), "Status");
 }