#include <TtHadEvtSolutionMaker.h>

Inheritance diagram for TtHadEvtSolutionMaker:

Public Member Functions
virtual void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup)

	TtHadEvtSolutionMaker (const edm::ParameterSet &iConfig)
	constructor More...

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

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

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

	EDConsumerBase ()

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &oModuleLabels) const

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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
bool	addLRJetComb_

bool	addLRSignalSel_

std::vector< unsigned int >	constraints_

bool	doKinFit_

edm::EDGetTokenT< TtGenEvent >	genEvtToken_

int	jetCorrScheme_

int	jetParam_

edm::EDGetTokenT< std::vector < pat::Jet > >	jetSrcToken_

std::string	lrJetCombFile_

std::vector< int >	lrJetCombObs_

std::string	lrSignalSelFile_

std::vector< int >	lrSignalSelObs_

int	matchingAlgo_

bool	matchToGenEvt_

double	maxDeltaS_

double	maxDist_

double	maxF_

int	maxNrIter_

TtFullHadKinFitter *	myKinFitter

TtHadLRJetCombCalc *	myLRJetCombCalc

TtHadLRJetCombObservables *	myLRJetCombObservables

TtHadLRSignalSelCalc *	myLRSignalSelCalc

TtHadLRSignalSelObservables *	myLRSignalSelObservables

TtHadSimpleBestJetComb *	mySimpleBestJetComb

bool	useDeltaR_

bool	useMaxDist_

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 25 of file TtHadEvtSolutionMaker.h.

Constructor & Destructor Documentation

TtHadEvtSolutionMaker::TtHadEvtSolutionMaker ( const edm::ParameterSet & iConfig )

explicit

constructor

Definition at line 17 of file TtHadEvtSolutionMaker.cc.

References addLRJetComb_, addLRSignalSel_, constraints_, doKinFit_, genEvtToken_, edm::ParameterSet::getParameter(), HLT_FULL_cff::InputTag, jetCorrScheme_, jetParam_, jetSrcToken_, lrJetCombFile_, lrJetCombObs_, lrSignalSelFile_, lrSignalSelObs_, matchingAlgo_, matchToGenEvt_, maxDeltaS_, maxDist_, maxF_, maxNrIter_, myKinFitter, myLRJetCombCalc, myLRJetCombObservables, myLRSignalSelCalc, myLRSignalSelObservables, mySimpleBestJetComb, AlCaHLTBitMon_QueryRunRegistry::string, useDeltaR_, and useMaxDist_.

                                                                             {
   // configurables
   jetSrcToken_          = consumes<std::vector<pat::Jet> >(iConfig.getParameter<edm::InputTag>    ("jetSource"));
   jetCorrScheme_   = iConfig.getParameter<int>              ("jetCorrectionScheme");
   doKinFit_        = iConfig.getParameter<bool>             ("doKinFit");
   addLRSignalSel_  = iConfig.getParameter<bool>             ("addLRSignalSel");
   lrSignalSelObs_  = iConfig.getParameter<std::vector<int> >("lrSignalSelObs");
   lrSignalSelFile_ = iConfig.getParameter<std::string>      ("lrSignalSelFile");
   addLRJetComb_    = iConfig.getParameter<bool>             ("addLRJetComb");
   lrJetCombObs_    = iConfig.getParameter<std::vector<int> >("lrJetCombObs");
   lrJetCombFile_   = iConfig.getParameter<std::string>      ("lrJetCombFile");
   maxNrIter_       = iConfig.getParameter<int>              ("maxNrIter");
   maxDeltaS_       = iConfig.getParameter<double>           ("maxDeltaS");
   maxF_            = iConfig.getParameter<double>           ("maxF");
   jetParam_        = iConfig.getParameter<int>              ("jetParametrisation");
   constraints_     = iConfig.getParameter<std::vector<unsigned int> >("constraints");
   matchToGenEvt_   = iConfig.getParameter<bool>             ("matchToGenEvt");
   matchingAlgo_    = iConfig.getParameter<bool>             ("matchingAlgorithm");
   useMaxDist_      = iConfig.getParameter<bool>             ("useMaximalDistance");
   useDeltaR_       = iConfig.getParameter<bool>             ("useDeltaR");
   maxDist_         = iConfig.getParameter<double>           ("maximalDistance");
   genEvtToken_ = mayConsume<TtGenEvent>(edm::InputTag("genEvt"));
 
   // define kinfitter
   if(doKinFit_){
     myKinFitter = new TtFullHadKinFitter(jetParam_, maxNrIter_, maxDeltaS_, maxF_, constraints_);
   }
 
 
   // define jet combinations related calculators
   mySimpleBestJetComb                    = new TtHadSimpleBestJetComb();
   myLRSignalSelObservables               = new TtHadLRSignalSelObservables();
   myLRJetCombObservables                 = new TtHadLRJetCombObservables();
   if (addLRJetComb_)   myLRJetCombCalc   = new TtHadLRJetCombCalc(lrJetCombFile_, lrJetCombObs_);
 
   // instantiate signal selection calculator
   if (addLRSignalSel_) myLRSignalSelCalc = new TtHadLRSignalSelCalc(lrSignalSelFile_, lrSignalSelObs_);
 
   // define what will be produced
   produces<std::vector<TtHadEvtSolution> >();
 
 }

TtHadEvtSolutionMaker::~TtHadEvtSolutionMaker ( )

destructor

Definition at line 62 of file TtHadEvtSolutionMaker.cc.

References addLRJetComb_, addLRSignalSel_, doKinFit_, myKinFitter, myLRJetCombCalc, myLRJetCombObservables, myLRSignalSelCalc, myLRSignalSelObservables, and mySimpleBestJetComb.

 {
   if (doKinFit_) {
     delete myKinFitter;
   }
   delete mySimpleBestJetComb;
   delete myLRSignalSelObservables;
   delete myLRJetCombObservables;
   if(addLRSignalSel_) delete myLRSignalSelCalc;
   if(addLRJetComb_)   delete myLRJetCombCalc;
 }

Member Function Documentation

void TtHadEvtSolutionMaker::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

virtual

Implements edm::EDProducer.

Definition at line 75 of file TtHadEvtSolutionMaker.cc.

References TtFullHadKinFitter::addKinFitInfo(), addLRJetComb_, addLRSignalSel_, gather_cfg::cout, doKinFit_, TtGenEvtProducer_cfi::genEvt, genEvtToken_, EgammaValidation_cff::genp, edm::Event::getByToken(), JetPartonMatching::getDistanceForParton(), JetPartonMatching::getMatchForParton(), JetPartonMatching::getSumDistances(), i, j, jetCorrScheme_, jetParam_, fwrapper::jets, jetSrcToken_, relval_2017::k, matchingAlgo_, matchToGenEvt_, maxDist_, myKinFitter, AlCaHLTBitMon_ParallelJobs::p, edm::Event::put(), lumiQueryAPI::q, alignCSCRings::s, TtHadEvtSolution::setJetParametrisation(), useDeltaR_, and useMaxDist_.

                                                                                    {
   // TopObject Selection
   // Select Jets
 
   bool jetsFound = false;
   edm::Handle<std::vector<pat::Jet> > jets;
   iEvent.getByToken(jetSrcToken_, jets);
 
   if (jets->size() >= 6) jetsFound = true;
 
   // Build Event solutions according to the ambiguity in the jet combination
   // Note, hardcoded to only run through the 6 most energetic jets - could be changed ....
 
   std::vector<TtHadEvtSolution> * evtsols = new std::vector<TtHadEvtSolution>();
   if(jetsFound){
     for (unsigned int p=0; p<3; p++) {  // loop over light jet p
       for (unsigned int q=p+1; q<4; q++) { // loop over light jet q
     for (unsigned int j=q+1; j<5; j++) {  // loop over light jet j
       for (unsigned int k=j+1; k<6; k++) { // loop over light jet k
         for (unsigned int bh=0; bh!=jets->size(); bh++) { //loop over hadronic b-jet1
           if(!(bh==p || bh==q || bh==j || bh==k)) {
         for (unsigned int bbarh=0; bbarh!=jets->size(); bbarh++) { //loop over hadronic b-jet2
           if (!(bbarh==p || bbarh==q || bbarh==j || bbarh==k) && !(bbarh==bh)) {
             // Make event solutions for all possible combinations of the 4 light
             // jets and 2 possible b-jets, not including the option of the b's being swapped.
             // Hadp,Hadq is one pair, Hadj,Hadk the other
             std::vector<TtHadEvtSolution> asol;
             asol.resize(3);
             //[p][q][b] and [j][k][bbar]
                     asol[0].setJetCorrectionScheme(jetCorrScheme_);
             asol[0].setHadp(jets, p);
             asol[0].setHadq(jets, q);
             asol[0].setHadj(jets, j);
             asol[0].setHadk(jets, k);
             asol[0].setHadb(jets, bh);
             asol[0].setHadbbar(jets, bbarh);
 
             //[p][j][b] and [q][k][bbar]
                     asol[1].setJetCorrectionScheme(jetCorrScheme_);
             asol[1].setHadp(jets, p);
             asol[1].setHadq(jets, j);
             asol[1].setHadj(jets, q);
             asol[1].setHadk(jets, k);
             asol[1].setHadb(jets, bh);
             asol[1].setHadbbar(jets, bbarh);
 
             //[p][k][b] and [j][q][bbar]
                     asol[2].setJetCorrectionScheme(jetCorrScheme_);
             asol[2].setHadp(jets, p);
             asol[2].setHadq(jets, k);
             asol[2].setHadj(jets, j);
             asol[2].setHadk(jets, q);
             asol[2].setHadb(jets, bh);
             asol[2].setHadbbar(jets, bbarh);
 
             if(doKinFit_){
               for(unsigned int i=0;i!=asol.size();i++){
             asol[i] = myKinFitter->addKinFitInfo(&(asol[i]));
             asol[i].setJetParametrisation(jetParam_);
               }
 
             }else{
               std::cout<<"Fitting needed to decide on best solution, enable fitting!"<<std::endl;
             }
             // these lines calculate the observables to be used in the TtHadSignalSelection LR
 
             for(unsigned int i=0;i!=asol.size();i++){
             (*myLRSignalSelObservables)(asol[i]);
             }
             // if asked for, calculate with these observable values the LRvalue and
             // (depending on the configuration) probability this event is signal
             if(addLRSignalSel_){
               for(unsigned int i=0;i!=asol.size();i++){
             (*myLRSignalSelCalc)(asol[i]);
               }
             }
 
             // these lines calculate the observables to be used in the TtHadJetCombination LR
             for(unsigned int i=0;i!=asol.size();i++){
             (*myLRJetCombObservables)(asol[i]);
             }
             // if asked for, calculate with these observable values the LRvalue and
             // (depending on the configuration) probability a jet combination is correct
             if(addLRJetComb_){
               for(unsigned int i=0;i!=asol.size();i++){
             (*myLRJetCombCalc)(asol[i]);
               }
             }
             //std::cout<<"SignalSelLRval = "<<asol.getLRSignalEvtLRval()<<"  JetCombProb = "<<asol.getLRSignalEvtProb()<<std::endl;
             //std::cout<<"JetCombLRval = "<<asol.getLRJetCombLRval()<<"  JetCombProb = "<<asol.getLRJetCombProb()<<std::endl;
 
             // fill solution to vector with all possible solutions
             for(unsigned int i=0;i!=asol.size();i++){
               evtsols->push_back(asol[i]);
             }
           }
         }
           }
         }
       }
     }
       }
     }
 
 
     // add TtHadSimpleBestJetComb to solutions
     int simpleBestJetComb = (*mySimpleBestJetComb)(*evtsols);
 
     for(size_t s=0; s<evtsols->size(); s++){
       (*evtsols)[s].setSimpleBestJetComb(simpleBestJetComb);
       // if asked for, match the event solutions to the gen Event
       if(matchToGenEvt_){
     int bestSolution = -999;
     int bestSolutionChangeW1Q = -999;
     int bestSolutionChangeW2Q = -999;
     edm::Handle<TtGenEvent> genEvt;
     iEvent.getByToken(genEvtToken_,genEvt);
     std::vector<const reco::Candidate*> quarks;
     const reco::Candidate & genp  = *(genEvt->daughterQuarkOfWPlus());
     const reco::Candidate & genq  = *(genEvt->daughterQuarkBarOfWPlus());
     const reco::Candidate & genb  = *(genEvt->b());
     const reco::Candidate & genj  = *(genEvt->daughterQuarkOfWMinus());
     const reco::Candidate & genk  = *(genEvt->daughterQuarkBarOfWMinus());
     const reco::Candidate & genbbar = *(genEvt->bBar());
     quarks.push_back( &genp );
     quarks.push_back( &genq );
     quarks.push_back( &genb );
     quarks.push_back( &genj );
     quarks.push_back( &genk );
     quarks.push_back( &genbbar );
     std::vector<const reco::Candidate*> jets;
     for(size_t s=0; s<evtsols->size(); s++) {
       jets.clear();
       const reco::Candidate & jetp  = (*evtsols)[s].getRecHadp();
       const reco::Candidate & jetq  = (*evtsols)[s].getRecHadq();
       const reco::Candidate & jetbh = (*evtsols)[s].getRecHadb();
       const reco::Candidate & jetj  = (*evtsols)[s].getRecHadj();
       const reco::Candidate & jetk  = (*evtsols)[s].getRecHadk();
       const reco::Candidate & jetbbar = (*evtsols)[s].getRecHadbbar();
       jets.push_back( &jetp );
       jets.push_back( &jetq );
       jets.push_back( &jetbh );
       jets.push_back( &jetj );
       jets.push_back( &jetk );
       jets.push_back( &jetbbar );
       JetPartonMatching aMatch(quarks, jets, matchingAlgo_, useMaxDist_, useDeltaR_, maxDist_);
       (*evtsols)[s].setGenEvt(genEvt);
       (*evtsols)[s].setMCBestSumAngles(aMatch.getSumDistances());
       (*evtsols)[s].setMCBestAngleHadp(aMatch.getDistanceForParton(0));
       (*evtsols)[s].setMCBestAngleHadq(aMatch.getDistanceForParton(1));
       (*evtsols)[s].setMCBestAngleHadb(aMatch.getDistanceForParton(2));
       (*evtsols)[s].setMCBestAngleHadb(aMatch.getDistanceForParton(2));
       (*evtsols)[s].setMCBestAngleHadj(aMatch.getDistanceForParton(3));
       (*evtsols)[s].setMCBestAngleHadk(aMatch.getDistanceForParton(4));
       (*evtsols)[s].setMCBestAngleHadbbar(aMatch.getDistanceForParton(5));
 
       // Check match - checking if two light quarks are swapped wrt matched gen particle
       if((aMatch.getMatchForParton(2) == 2 && aMatch.getMatchForParton(5) == 5)
          || (aMatch.getMatchForParton(2) == 5 && aMatch.getMatchForParton(5) == 2)){ // check b-jets
 
         if(aMatch.getMatchForParton(3) == 3 && aMatch.getMatchForParton(4) == 4){ //check light jets
           bestSolutionChangeW2Q = 0;
           if(aMatch.getMatchForParton(0) == 0 && aMatch.getMatchForParton(1) == 1) {
         bestSolution = s;
         bestSolutionChangeW1Q = 0;
           }else{
         if(aMatch.getMatchForParton(0) == 1 && aMatch.getMatchForParton(1) == 0){
           bestSolution = s;
           bestSolutionChangeW1Q = 1;
         }
           }
         }else{
           if(aMatch.getMatchForParton(2) == 3 && aMatch.getMatchForParton(3) == 2){ // or check if swapped
         bestSolutionChangeW2Q = 1;
         if(aMatch.getMatchForParton(0) == 1 && aMatch.getMatchForParton(1) == 0){
           bestSolution = s;
           bestSolutionChangeW1Q = 1;
         }else{
           if(aMatch.getMatchForParton(0) == 0 && aMatch.getMatchForParton(1) == 1) {
             bestSolution = s;
             bestSolutionChangeW1Q = 0;
           }
         }
           }
           if(aMatch.getMatchForParton(2) == 2 && aMatch.getMatchForParton(3) == 3){
         bestSolutionChangeW2Q = 0;
         if(aMatch.getMatchForParton(0) == 0 && aMatch.getMatchForParton(1) == 1) {
           bestSolution = s;
           bestSolutionChangeW1Q = 0;
         } else if(aMatch.getMatchForParton(0) == 1 && aMatch.getMatchForParton(1) == 0) {
           bestSolution = s;
           bestSolutionChangeW1Q = 1;
         }
           }
         }
       }
       for(size_t s=0; s<evtsols->size(); s++) {
         (*evtsols)[s].setMCBestJetComb(bestSolution);
         (*evtsols)[s].setMCChangeW1Q(bestSolutionChangeW1Q);
         (*evtsols)[s].setMCChangeW2Q(bestSolutionChangeW2Q);
       }
     }
       } // end matchEvt
     }
     //store the vector of solutions to the event
 
     std::auto_ptr<std::vector<TtHadEvtSolution> > pOut(evtsols);
     iEvent.put(pOut);
   }else {     //end loop jet/MET found
     std::cout<<"No calibrated solutions built, because only "<<jets->size()<<" were present";
 
     std::auto_ptr<std::vector<TtHadEvtSolution> > pOut(evtsols);
     iEvent.put(pOut);
   }
 }