|
|
|
#include <TtFullLepKinSolutionProducer.h>
Classes | |
| struct | Compare |
Public Member Functions | |
| virtual void | beginJob () |
| virtual void | endJob () |
| virtual void | produce (edm::Event &evt, const edm::EventSetup &iSetup) |
| TtFullLepKinSolutionProducer (const edm::ParameterSet &iConfig) | |
| ~TtFullLepKinSolutionProducer () | |
Private Member Functions | |
| bool | HasPositiveCharge (const reco::Candidate *) const |
| bool | LepDiffCharge (const reco::Candidate *, const reco::Candidate *) const |
| bool | PTComp (const reco::Candidate *, const reco::Candidate *) const |
Private Attributes | |
| bool | eeChannel_ |
| edm::InputTag | electrons_ |
| bool | emuChannel_ |
| std::string | jetCorrLevel_ |
| edm::InputTag | jets_ |
| int | maxNComb_ |
| int | maxNJets_ |
| edm::InputTag | mets_ |
| bool | mumuChannel_ |
| edm::InputTag | muons_ |
| std::vector< double > | nupars_ |
| bool | searchWrongCharge_ |
| TtFullLepKinSolver * | solver |
| double | tmassbegin_ |
| double | tmassend_ |
| double | tmassstep_ |
Definition at line 21 of file TtFullLepKinSolutionProducer.h.
| TtFullLepKinSolutionProducer::TtFullLepKinSolutionProducer | ( | const edm::ParameterSet & | iConfig | ) | [explicit] |
Definition at line 74 of file TtFullLepKinSolutionProducer.h.
References eeChannel_, electrons_, emuChannel_, edm::ParameterSet::getParameter(), jetCorrLevel_, jets_, maxNComb_, maxNJets_, mets_, mumuChannel_, muons_, nupars_, searchWrongCharge_, tmassbegin_, tmassend_, and tmassstep_.
{
// configurables
jets_ = iConfig.getParameter<edm::InputTag>("jets");
electrons_ = iConfig.getParameter<edm::InputTag>("electrons");
muons_ = iConfig.getParameter<edm::InputTag>("muons");
mets_ = iConfig.getParameter<edm::InputTag>("mets");
jetCorrLevel_= iConfig.getParameter<std::string> ("jetCorrectionLevel");
maxNJets_ = iConfig.getParameter<int> ("maxNJets");
maxNComb_ = iConfig.getParameter<int> ("maxNComb");
eeChannel_ = iConfig.getParameter<bool>("eeChannel");
emuChannel_ = iConfig.getParameter<bool>("emuChannel");
mumuChannel_ = iConfig.getParameter<bool>("mumuChannel");
searchWrongCharge_ = iConfig.getParameter<bool> ("searchWrongCharge");
tmassbegin_ = iConfig.getParameter<double>("tmassbegin");
tmassend_ = iConfig.getParameter<double>("tmassend");
tmassstep_ = iConfig.getParameter<double>("tmassstep");
nupars_ = iConfig.getParameter<std::vector<double> >("neutrino_parameters");
// define what will be produced
produces<std::vector<std::vector<int> > > (); // vector of the particle inices (b, bbar, e1, e2, mu1, mu2)
produces<std::vector<reco::LeafCandidate> >("fullLepNeutrinos");
produces<std::vector<reco::LeafCandidate> >("fullLepNeutrinoBars");
produces<std::vector<double> >("solWeight"); //weight for a specific kin solution
produces<bool>("isWrongCharge"); //true if leptons have the same charge
}
| TtFullLepKinSolutionProducer::~TtFullLepKinSolutionProducer | ( | ) |
Definition at line 101 of file TtFullLepKinSolutionProducer.h.
{
}
| void TtFullLepKinSolutionProducer::beginJob | ( | void | ) | [virtual] |
Reimplemented from edm::EDProducer.
Definition at line 105 of file TtFullLepKinSolutionProducer.h.
References nupars_, solver, tmassbegin_, tmassend_, and tmassstep_.
{
solver = new TtFullLepKinSolver(tmassbegin_, tmassend_, tmassstep_, nupars_);
}
| void TtFullLepKinSolutionProducer::endJob | ( | void | ) | [virtual] |
Reimplemented from edm::EDProducer.
Definition at line 110 of file TtFullLepKinSolutionProducer.h.
References solver.
{
delete solver;
}
| bool TtFullLepKinSolutionProducer::HasPositiveCharge | ( | const reco::Candidate * | l | ) | const [inline, private] |
Definition at line 69 of file TtFullLepKinSolutionProducer.h.
References reco::Candidate::charge().
Referenced by produce().
{
return (l->charge() > 0);
}
| bool TtFullLepKinSolutionProducer::LepDiffCharge | ( | const reco::Candidate * | l1, |
| const reco::Candidate * | l2 | ||
| ) | const [inline, private] |
Definition at line 64 of file TtFullLepKinSolutionProducer.h.
References reco::Candidate::charge().
Referenced by produce().
| void TtFullLepKinSolutionProducer::produce | ( | edm::Event & | evt, |
| const edm::EventSetup & | iSetup | ||
| ) | [virtual] |
Implements edm::EDProducer.
Definition at line 115 of file TtFullLepKinSolutionProducer.h.
References eeChannel_, HI_PhotonSkim_cff::electrons, electrons_, emuChannel_, relval_parameters_module::energy, first, edm::Event::getByLabel(), TtFullLepKinSolver::getNuSolution(), HasPositiveCharge(), i, jetCorrLevel_, fwrapper::jets, jets_, LepDiffCharge(), maxNComb_, maxNJets_, mets_, mumuChannel_, patZpeak::muons, muons_, TtFullLepKinSolver::NeutrinoSolution::neutrino, TtFullLepKinSolver::NeutrinoSolution::neutrinoBar, PTComp(), edm::Event::put(), searchWrongCharge_, edm::second(), TtFullLepKinSolver::SetConstraints(), reco::Candidate::size, solver, python::multivaluedict::sort(), and TtFullLepKinSolver::NeutrinoSolution::weight.
{
//create vectors fo runsorted output
std::vector<std::vector<int> > idcsV;
std::vector<reco::LeafCandidate> nusV;
std::vector<reco::LeafCandidate> nuBarsV;
std::vector<std::pair<double, int> > weightsV;
//create pointer for products
std::auto_ptr<std::vector<std::vector<int> > > pIdcs(new std::vector<std::vector<int> >);
std::auto_ptr<std::vector<reco::LeafCandidate> > pNus(new std::vector<reco::LeafCandidate>);
std::auto_ptr<std::vector<reco::LeafCandidate> > pNuBars(new std::vector<reco::LeafCandidate>);
std::auto_ptr<std::vector<double> > pWeight(new std::vector<double>);
std::auto_ptr<bool> pWrongCharge(new bool);
edm::Handle<std::vector<pat::Jet> > jets;
evt.getByLabel(jets_, jets);
edm::Handle<std::vector<pat::Electron> > electrons;
evt.getByLabel(electrons_, electrons);
edm::Handle<std::vector<pat::Muon> > muons;
evt.getByLabel(muons_, muons);
edm::Handle<std::vector<pat::MET> > mets;
evt.getByLabel(mets_, mets);
int selMuon1 = -1, selMuon2 = -1;
int selElectron1 = -1, selElectron2 = -1;
bool ee = false;
bool emu = false;
bool mumu = false;
bool isWrongCharge = false;
bool jetsFound = false;
bool METFound = false;
bool electronsFound = false;
bool electronMuonFound = false;
bool muonsFound = false;
//select Jets (TopJet vector is sorted on ET)
if(jets->size()>=2) { jetsFound = true; }
//select MET (TopMET vector is sorted on ET)
if(mets->size()>=1) { METFound = true; }
// If we have electrons and muons available,
// build a solutions with electrons and muons.
if(muons->size() + electrons->size() >=2) {
// select leptons
if(electrons->size() == 0) mumu = true;
else if(muons->size() == 0) ee = true;
else if(electrons->size() == 1) {
if(muons->size() == 1) emu = true;
else if(PTComp(&(*electrons)[0], &(*muons)[1])) emu = true;
else mumu = true;
}
else if(electrons->size() > 1) {
if(PTComp(&(*electrons)[1], &(*muons)[0])) ee = true;
else if(muons->size() == 1) emu = true;
else if(PTComp(&(*electrons)[0], &(*muons)[1])) emu = true;
else mumu = true;
}
if(ee && eeChannel_) {
if(LepDiffCharge(&(*electrons)[0], &(*electrons)[1]) || searchWrongCharge_) {
if(HasPositiveCharge(&(*electrons)[0]) || !LepDiffCharge(&(*electrons)[0], &(*electrons)[1])) {
selElectron1 = 0;
selElectron2 = 1;
}
else{
selElectron1 = 1;
selElectron2 = 0;
}
electronsFound = true;
if(!LepDiffCharge(&(*electrons)[0], &(*electrons)[1])) isWrongCharge = true;
}
}
else if(emu && emuChannel_) {
if(LepDiffCharge(&(*electrons)[0], &(*muons)[0]) || searchWrongCharge_) {
selElectron1 = 0;
selMuon1 = 0;
electronMuonFound = true;
if(!LepDiffCharge(&(*electrons)[0], &(*muons)[0])) isWrongCharge = true;
}
}
else if(mumu && mumuChannel_) {
if(LepDiffCharge(&(*muons)[0], &(*muons)[1]) || searchWrongCharge_) {
if(HasPositiveCharge(&(*muons)[0]) || !LepDiffCharge(&(*muons)[0], &(*muons)[1])) {
selMuon1 = 0;
selMuon2 = 1;
}
else {
selMuon1 = 1;
selMuon2 = 0;
}
muonsFound = true;
if(!LepDiffCharge(&(*muons)[0], &(*muons)[1])) isWrongCharge = true;
}
}
}
*pWrongCharge = isWrongCharge;
// Check that the above work makes sense
if(int(ee)+int(emu)+int(mumu)>1) {
edm::LogWarning("TtFullLepKinSolutionProducer") << "Lepton selection criteria uncorrectly defined";
}
// Check if the leptons for the required Channel are available
bool correctLeptons = ((electronsFound && eeChannel_) || (muonsFound && mumuChannel_) || (electronMuonFound && emuChannel_) );
// Check for equally charged leptons if for wrong charge combinations is searched
if(isWrongCharge) correctLeptons *= searchWrongCharge_;
if(correctLeptons && METFound && jetsFound) {
// run over all jets if input parameter maxNJets is -1 or
// adapt maxNJets if number of present jets is smaller than selected
// number of jets
int stop=maxNJets_;
if(jets->size()<static_cast<unsigned int>(stop) || stop<0) stop=jets->size();
// counter for number of found kinematic solutions
int nSol=0;
// consider all permutations
for (int ib = 0; ib<stop; ib++) {
// second loop of the permutations
for (int ibbar = 0; ibbar<stop; ibbar++) {
// avoid the diagonal: b and bbar must be distinct jets
if(ib==ibbar) continue;
std::vector<int> idcs;
// push back the indices of the jets
idcs.push_back(ib);
idcs.push_back(ibbar);
TLorentzVector LV_l1;
TLorentzVector LV_l2;
TLorentzVector LV_b = TLorentzVector((*jets)[ib].correctedJet(jetCorrLevel_, "bottom").px(),
(*jets)[ib].correctedJet(jetCorrLevel_, "bottom").py(),
(*jets)[ib].correctedJet(jetCorrLevel_, "bottom").pz(),
(*jets)[ib].correctedJet(jetCorrLevel_, "bottom").energy() );
TLorentzVector LV_bbar = TLorentzVector((*jets)[ibbar].correctedJet(jetCorrLevel_, "bottom").px(),
(*jets)[ibbar].correctedJet(jetCorrLevel_, "bottom").py(),
(*jets)[ibbar].correctedJet(jetCorrLevel_, "bottom").pz(),
(*jets)[ibbar].correctedJet(jetCorrLevel_, "bottom").energy());
double xconstraint = 0, yconstraint = 0;
if (ee) {
idcs.push_back(selElectron1);
LV_l1.SetXYZT((*electrons)[selElectron1].px(),
(*electrons)[selElectron1].py(),
(*electrons)[selElectron1].pz(),
(*electrons)[selElectron1].energy());
xconstraint += (*electrons)[selElectron1].px();
yconstraint += (*electrons)[selElectron1].py();
idcs.push_back(selElectron2);
LV_l2.SetXYZT((*electrons)[selElectron2].px(),
(*electrons)[selElectron2].py(),
(*electrons)[selElectron2].pz(),
(*electrons)[selElectron2].energy());
xconstraint += (*electrons)[selElectron2].px();
yconstraint += (*electrons)[selElectron2].py();
idcs.push_back(-1);
idcs.push_back(-1);
}
else if (emu) {
if(!isWrongCharge){
if(HasPositiveCharge(&(*electrons)[selElectron1])){
idcs.push_back(selElectron1);
LV_l1.SetXYZT((*electrons)[selElectron1].px(),
(*electrons)[selElectron1].py(),
(*electrons)[selElectron1].pz(),
(*electrons)[selElectron1].energy());
xconstraint += (*electrons)[selElectron1].px();
yconstraint += (*electrons)[selElectron1].py();
idcs.push_back(-1);
idcs.push_back(-1);
idcs.push_back(selMuon1);
LV_l2.SetXYZT((*muons)[selMuon1].px(),
(*muons)[selMuon1].py(),
(*muons)[selMuon1].pz(),
(*muons)[selMuon1].energy());
xconstraint += (*muons)[selMuon1].px();
yconstraint += (*muons)[selMuon1].py();
}
else{
idcs.push_back(-1);
idcs.push_back(selMuon1);
LV_l1.SetXYZT((*muons)[selMuon1].px(),
(*muons)[selMuon1].py(),
(*muons)[selMuon1].pz(),
(*muons)[selMuon1].energy());
xconstraint += (*muons)[selMuon1].px();
yconstraint += (*muons)[selMuon1].py();
idcs.push_back(selElectron1);
LV_l2.SetXYZT((*electrons)[selElectron1].px(),
(*electrons)[selElectron1].py(),
(*electrons)[selElectron1].pz(),
(*electrons)[selElectron1].energy());
xconstraint += (*electrons)[selElectron1].px();
yconstraint += (*electrons)[selElectron1].py();
idcs.push_back(-1);
}
}
else{ // means "if wrong charge"
if(HasPositiveCharge(&(*electrons)[selElectron1])){ // both leps positive
idcs.push_back(selElectron1);
LV_l1.SetXYZT((*electrons)[selElectron1].px(),
(*electrons)[selElectron1].py(),
(*electrons)[selElectron1].pz(),
(*electrons)[selElectron1].energy());
xconstraint += (*electrons)[selElectron1].px();
yconstraint += (*electrons)[selElectron1].py();
idcs.push_back(-1);
idcs.push_back(selMuon1);
LV_l2.SetXYZT((*muons)[selMuon1].px(),
(*muons)[selMuon1].py(),
(*muons)[selMuon1].pz(),
(*muons)[selMuon1].energy());
xconstraint += (*muons)[selMuon1].px();
yconstraint += (*muons)[selMuon1].py();
idcs.push_back(-1);
}
else{ // both leps negative
idcs.push_back(-1);
idcs.push_back(selElectron1);
LV_l2.SetXYZT((*electrons)[selElectron1].px(),
(*electrons)[selElectron1].py(),
(*electrons)[selElectron1].pz(),
(*electrons)[selElectron1].energy());
xconstraint += (*electrons)[selElectron1].px();
yconstraint += (*electrons)[selElectron1].py();
idcs.push_back(-1);
idcs.push_back(selMuon1);
LV_l1.SetXYZT((*muons)[selMuon1].px(),
(*muons)[selMuon1].py(),
(*muons)[selMuon1].pz(),
(*muons)[selMuon1].energy());
xconstraint += (*muons)[selMuon1].px();
yconstraint += (*muons)[selMuon1].py();
}
}
}
else if (mumu) {
idcs.push_back(-1);
idcs.push_back(-1);
idcs.push_back(selMuon1);
LV_l1.SetXYZT((*muons)[selMuon1].px(),
(*muons)[selMuon1].py(),
(*muons)[selMuon1].pz(),
(*muons)[selMuon1].energy());
xconstraint += (*muons)[selMuon1].px();
yconstraint += (*muons)[selMuon1].py();
idcs.push_back(selMuon2);
LV_l2.SetXYZT((*muons)[selMuon2].px(),
(*muons)[selMuon2].py(),
(*muons)[selMuon2].pz(),
(*muons)[selMuon2].energy());
xconstraint += (*muons)[selMuon2].px();
yconstraint += (*muons)[selMuon2].py();
}
xconstraint += (*jets)[ib].px() + (*jets)[ibbar].px() + (*mets)[0].px();
yconstraint += (*jets)[ib].py() + (*jets)[ibbar].py() + (*mets)[0].py();
// calculate neutrino momenta and eventweight
solver->SetConstraints(xconstraint, yconstraint);
TtFullLepKinSolver::NeutrinoSolution nuSol= solver->getNuSolution( LV_l1, LV_l2 , LV_b, LV_bbar);
// add solution to the vectors of solutions if solution exists
if(nuSol.weight>0){
// add the leptons and jets indices to the vector of combinations
idcsV.push_back(idcs);
// add the neutrinos
nusV.push_back(nuSol.neutrino);
nuBarsV.push_back(nuSol.neutrinoBar);
// add the solution weight
weightsV.push_back(std::make_pair(nuSol.weight, nSol));
nSol++;
}
}
}
}
if(weightsV.size()==0){
//create dmummy vector
std::vector<int> idcs;
for(int i=0; i<6; ++i)
idcs.push_back(-1);
idcsV.push_back(idcs);
weightsV.push_back(std::make_pair(-1,0));
reco::LeafCandidate nu;
nusV.push_back(nu);
reco::LeafCandidate nuBar;
nuBarsV.push_back(nuBar);
}
// check if all vectors have correct length
int weightL = weightsV.size();
int nuL = nusV.size();
int nuBarL = nuBarsV.size();
int idxL = idcsV.size();
if(weightL!=nuL || weightL!=nuBarL || weightL!=idxL){
edm::LogWarning("TtFullLepKinSolutionProducer") << "Output vectors are of different length:"
<< "\n weight: " << weightL
<< "\n nu: " << nuL
<< "\n nubar: " << nuBarL
<< "\n idcs: " << idxL;
}
// sort vectors by weight in decreasing order
if(weightsV.size()>1){
sort(weightsV.begin(), weightsV.end(), Compare());
}
// determine the number of solutions which is written in the event
int stop=weightL;
if(maxNComb_>0 && maxNComb_<stop) stop=maxNComb_;
for(int i=0; i<stop; ++i){
pWeight->push_back(weightsV[i].first);
pNus ->push_back(nusV[weightsV[i].second]);
pNuBars->push_back(nuBarsV[weightsV[i].second]);
pIdcs ->push_back(idcsV[weightsV[i].second]);
}
// put the results in the event
evt.put(pIdcs);
evt.put(pNus, "fullLepNeutrinos");
evt.put(pNuBars, "fullLepNeutrinoBars");
evt.put(pWeight, "solWeight");
evt.put(pWrongCharge, "isWrongCharge");
}
| bool TtFullLepKinSolutionProducer::PTComp | ( | const reco::Candidate * | l1, |
| const reco::Candidate * | l2 | ||
| ) | const [inline, private] |
Definition at line 59 of file TtFullLepKinSolutionProducer.h.
References reco::Candidate::pt().
Referenced by produce().
bool TtFullLepKinSolutionProducer::eeChannel_ [private] |
Definition at line 52 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
Definition at line 46 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
bool TtFullLepKinSolutionProducer::emuChannel_ [private] |
Definition at line 52 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
std::string TtFullLepKinSolutionProducer::jetCorrLevel_ [private] |
Definition at line 50 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
Definition at line 45 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
int TtFullLepKinSolutionProducer::maxNComb_ [private] |
Definition at line 51 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
int TtFullLepKinSolutionProducer::maxNJets_ [private] |
Definition at line 51 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
Definition at line 48 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
bool TtFullLepKinSolutionProducer::mumuChannel_ [private] |
Definition at line 52 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
Definition at line 47 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
std::vector<double> TtFullLepKinSolutionProducer::nupars_ [private] |
Definition at line 54 of file TtFullLepKinSolutionProducer.h.
Referenced by beginJob(), and TtFullLepKinSolutionProducer().
bool TtFullLepKinSolutionProducer::searchWrongCharge_ [private] |
Definition at line 52 of file TtFullLepKinSolutionProducer.h.
Referenced by produce(), and TtFullLepKinSolutionProducer().
Definition at line 56 of file TtFullLepKinSolutionProducer.h.
Referenced by beginJob(), endJob(), and produce().
double TtFullLepKinSolutionProducer::tmassbegin_ [private] |
Definition at line 53 of file TtFullLepKinSolutionProducer.h.
Referenced by beginJob(), and TtFullLepKinSolutionProducer().
double TtFullLepKinSolutionProducer::tmassend_ [private] |
Definition at line 53 of file TtFullLepKinSolutionProducer.h.
Referenced by beginJob(), and TtFullLepKinSolutionProducer().
double TtFullLepKinSolutionProducer::tmassstep_ [private] |
Definition at line 53 of file TtFullLepKinSolutionProducer.h.
Referenced by beginJob(), and TtFullLepKinSolutionProducer().
1.7.3