CMSSW_4_4_3_patch1/src/TopQuarkAnalysis/TopEventProducers/src/StEvtSolutionMaker.cc

Go to the documentation of this file.

//
// $Id: StEvtSolutionMaker.cc,v 1.9 2008/02/17 11:09:40 rwolf Exp $
//

#include <memory>

#include "TopQuarkAnalysis/TopEventProducers/interface/StEvtSolutionMaker.h"

StEvtSolutionMaker::StEvtSolutionMaker(const edm::ParameterSet& iConfig) 
{
  // configurables
  electronSrc_    = iConfig.getParameter<edm::InputTag>("electronSource");
  muonSrc_        = iConfig.getParameter<edm::InputTag>("muonSource");
  metSrc_         = iConfig.getParameter<edm::InputTag>("metSource");
  jetSrc_         = iConfig.getParameter<edm::InputTag>("jetSource");
  leptonFlavour_  = iConfig.getParameter< std::string >("leptonFlavour");
  jetCorrScheme_  = iConfig.getParameter<int>          ("jetCorrectionScheme");
  //jetInput_        = iConfig.getParameter< std::string >        ("jetInput");
  doKinFit_       = iConfig.getParameter< bool >       ("doKinFit");
  addLRJetComb_   = iConfig.getParameter< bool >       ("addLRJetComb");
  maxNrIter_      = iConfig.getParameter< int >        ("maxNrIter");
  maxDeltaS_      = iConfig.getParameter< double >     ("maxDeltaS");
  maxF_           = iConfig.getParameter< double >     ("maxF");
  jetParam_       = iConfig.getParameter<int>          ("jetParametrisation");
  lepParam_       = iConfig.getParameter<int>          ("lepParametrisation");
  metParam_       = iConfig.getParameter<int>          ("metParametrisation");
  constraints_    = iConfig.getParameter< std::vector<int> > ("constraints");
  matchToGenEvt_  = iConfig.getParameter< bool >        ("matchToGenEvt");
  
  // define kinfitter
  if(doKinFit_){
    myKinFitter = new StKinFitter(jetParam_, lepParam_, metParam_, maxNrIter_, maxDeltaS_, maxF_, constraints_);
  }
  // define what will be produced
  produces<std::vector<StEvtSolution> >();
}

StEvtSolutionMaker::~StEvtSolutionMaker() 
{
  if (doKinFit_) delete myKinFitter;
}

void StEvtSolutionMaker::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  //
  //  TopObject Selection
  //
  
  // select lepton (the TtLepton vectors are, for the moment, sorted on pT)
  bool leptonFound = false;
  edm::Handle<std::vector<pat::Muon> > muons;
  if(leptonFlavour_ == "muon"){
    iEvent.getByLabel(muonSrc_, muons);
    if( muons->size() > 0 ) leptonFound = true;
  }
  edm::Handle<std::vector<pat::Electron> > electrons;
  if(leptonFlavour_ == "electron"){
    iEvent.getByLabel(electronSrc_, electrons);
    if( electrons->size() > 0 ) leptonFound = true;
  }
  
  // select MET (TopMET vector is sorted on ET)
  bool metFound = false;
  edm::Handle<std::vector<pat::MET> > mets;
  iEvent.getByLabel(metSrc_, mets);
  if( mets->size() > 0 ) metFound = true;
  
  // select Jets
  bool jetsFound = false;
  edm::Handle<std::vector<pat::Jet> > jets;
  iEvent.getByLabel(jetSrc_, jets);
  unsigned int maxJets=2;//this has to become a custom-defined parameter (we may want 2 or 3 jets)
  if (jets->size() >= 2) jetsFound = true;
  
  std::vector<StEvtSolution> *evtsols = new std::vector<StEvtSolution>();
  if(leptonFound && metFound && jetsFound){
    std::cout<<"constructing solutions"<<std::endl;
    for (unsigned int b=0; b<maxJets; b++) {
      for (unsigned int l=0; l<maxJets; l++) {
        if(b!=l){  // to avoid double counting
          StEvtSolution asol;
          asol.setJetCorrectionScheme(jetCorrScheme_);
          if(leptonFlavour_ == "muon")     asol.setMuon(muons, 0);
          if(leptonFlavour_ == "electron") asol.setElectron(electrons, 0);
          asol.setNeutrino(mets, 0);
          asol.setBottom(jets, b);
          asol.setLight(jets, l);
          
          if(doKinFit_) asol = myKinFitter->addKinFitInfo(&asol);
          
          /* to be adapted to ST (Andrea)
             if(addLRJetComb_){
             asol.setPtrueCombExist(jetCombProbs[m].getPTrueCombExist(&afitsol));
             asol.setPtrueBJetSel(jetCombProbs[m].getPTrueBJetSel(&afitsol));
             asol.setPtrueBhadrSel(jetCombProbs[m].getPTrueBhadrSel(&afitsol));
             asol.setPtrueJetComb(afitsol.getPtrueCombExist()*afitsol.getPtrueBJetSel()*afitsol.getPtrueBhadrSel());
             }
          */
          evtsols->push_back(asol);
        }
      }
    }
    
    // if asked for, match the event solutions to the gen Event
    if(matchToGenEvt_){
      /*
        edm::Handle<StGenEvent> genEvt;
        iEvent.getByLabel ("genEvt",genEvt);
        double bestSolDR = 9999.;
        int bestSol = 0;
        for(size_t s=0; s<evtsols->size(); s++) {
        (*evtsols)[s].setGenEvt(genEvt->particles());
        vector<double> bm = BestMatch((*evtsols)[s], false); //false to use DR, true SpaceAngles
        (*evtsols)[s].setSumDeltaRjp(bm[0]); // dRBB + dRLL
        (*evtsols)[s].setChangeBL((int) bm[1]); // has swapped or not
        (*evtsols)[s].setDeltaRB(bm[2]);
        (*evtsols)[s].setDeltaRL(bm[3]);
        if(bm[0]<bestSolDR) { bestSolDR =  bm[0]; bestSol = s; }
        }
        (*evtsols)[bestSol].setBestSol(true);
      */
    }
    
    //store the vector of solutions to the event     
    std::auto_ptr<std::vector<StEvtSolution> > pOut(evtsols);
    iEvent.put(pOut);
  }
  else
    {
      
      std::cout<<"@@@ No calibrated solutions built, because:  " << std::endl;;
      if(jets->size()<maxJets)                                    std::cout<<"@ nr jets = " << jets->size() << " < " << maxJets <<std::endl;
      if(leptonFlavour_ == "muon" && !leptonFound)                std::cout<<"@ no good muon candidate"<<std::endl;
      if(leptonFlavour_ == "electron" && !leptonFound)             std::cout<<"@ no good electron candidate"<<std::endl;
      if(mets->size() == 0)                                       std::cout<<"@ no MET reconstruction"<<std::endl;
      
      StEvtSolution asol;
      evtsols->push_back(asol);
      std::auto_ptr<std::vector<StEvtSolution> > pOut(evtsols);
      iEvent.put(pOut);
    }
}