#include <TtFullHadKinFitProducer.h>

Inheritance diagram for TtFullHadKinFitProducer:

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

	~TtFullHadKinFitProducer ()
	default destructor More...

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

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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 ()

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, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

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

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

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 &	itemsToGetFromEvent () 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

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

virtual	~EDConsumerBase () noexcept(false)

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

Private Member Functions
virtual void	produce (edm::Event &event, const edm::EventSetup &setup)
	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
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)

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 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 ( )

default destructor

Definition at line 58 of file TtFullHadKinFitProducer.cc.

References kinFitter.

 {
   delete kinFitter;
 }

Member Function Documentation

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

privatevirtual

produce fitted object collections and meta data describing fit quality

set match to be used

set the validity of a match

Definition at line 65 of file TtFullHadKinFitProducer.cc.

References DEFINE_FWK_MODULE, TtFullHadKinFitter::KinFit::fit(), training_settings::idx, fwrapper::jets, jetsToken_, kinFitter, match(), patRefSel_triggerMatching_cfi::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" );
 }