#include <TtFullLepKinSolutionProducer.h>

Inheritance diagram for TtFullLepKinSolutionProducer:

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_

Detailed Description

Definition at line 21 of file TtFullLepKinSolutionProducer.h.

Constructor & Destructor Documentation

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.

{
}

Member Function Documentation

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

{
  return (l1->charge() != l2->charge());
}

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_, analyzePatCleaning_cfg::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"); 
}