/data/doxygen/doxygen-1.7.3/gen/CMSSW_4_2_8/src/MuonAnalysis/MomentumScaleCalibration/plugins/MuScleFitGenFilter.cc

Go to the documentation of this file.

//#define DEBUG

#include "MuScleFitGenFilter.h"

// Collaborating Class Header
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "RecoMuon/TrackingTools/interface/MuonPatternRecoDumper.h"
#include "RecoMuon/TrackingTools/interface/MuonServiceProxy.h"

#include "MuonAnalysis/MomentumScaleCalibration/interface/MuScleFitUtils.h"

#include <CLHEP/Vector/LorentzVector.h>

// Constructor
// -----------
MuScleFitGenFilter::MuScleFitGenFilter(const edm::ParameterSet& iConfig) :
  genParticlesName_( iConfig.getUntrackedParameter<std::string>("GenParticlesName", "genParticles") ),
  totalEvents_(0),
  eventsPassingTheFilter_(0)
{
  MuScleFitUtils::resfind = iConfig.getParameter<std::vector<int> >("resfind");
}

// Destructor
// ----------
MuScleFitGenFilter::~MuScleFitGenFilter()
{
  std::cout << "Total number of events = " << totalEvents_ << std::endl;
  std::cout << "Events passing the filter = " << eventsPassingTheFilter_ << std::endl;
}

// Method called for each event 
// ----------------------------
bool MuScleFitGenFilter::filter(edm::Event& event, const edm::EventSetup& iSetup)
{
  ++totalEvents_;

  bool ifHepMC = false;
  bool ifGenPart = false;

  edm::Handle<edm::HepMCProduct> evtMC;

  std::pair<lorentzVector,lorentzVector> genPair;

  event.getByLabel( genParticlesName_, evtMC );
  if( evtMC.isValid() ) {

    genPair = MuScleFitUtils::findGenMuFromRes(evtMC.product());

    ifHepMC = true;
  }
  else {
    edm::Handle<reco::GenParticleCollection> genParticles;
    event.getByLabel( genParticlesName_, genParticles );
    if( genParticles.isValid() ) {
      
      genPair = MuScleFitUtils::findGenMuFromRes(genParticles.product());

      ifGenPart=true;
    }
    else {
      std::cout << "ERROR: no generator info found" << std::endl;
      return false;
    }
  }
  lorentzVector emptyVec(0.,0.,0.,0.);
  if( (genPair.first == emptyVec) || (genPair.second == emptyVec) ) {
    return false;
  }

  ++eventsPassingTheFilter_;

  return true;
}