BSvsPVAnalyzer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    Validation/RecoVertex
// Class:      BSvsPVAnalyzer
//
//
// Original Author:  Andrea Venturi
//         Created:  Mon Oct 27 17:37:53 CET 2008
//
//


// system include files
#include <memory>

// user include files

#include <vector>
#include <map>
#include <limits>
#include <string>

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/Utilities/interface/InputTag.h"

#include "Validation/RecoVertex/interface/BSvsPVHistogramMaker.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

//
// class decleration
//

class BSvsPVAnalyzer : public edm::EDAnalyzer {
   public:
      explicit BSvsPVAnalyzer(const edm::ParameterSet&);
      ~BSvsPVAnalyzer();


private:
  virtual void beginJob() ;
  virtual void analyze(const edm::Event&, const edm::EventSetup&);
  virtual void beginRun(const edm::Run&, const edm::EventSetup&);
  virtual void endRun(const edm::Run&, const edm::EventSetup&);
  virtual void endJob() ;

      // ----------member data ---------------------------

  BSvsPVHistogramMaker _bspvhm;
  edm::EDGetTokenT<reco::VertexCollection> _recoVertexCollectionToken;
  edm::EDGetTokenT<reco::BeamSpot> _recoBeamSpotToken;
  bool _firstOnly;

};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
BSvsPVAnalyzer::BSvsPVAnalyzer(const edm::ParameterSet& iConfig)
  : _bspvhm(iConfig.getParameter<edm::ParameterSet>("bspvHistogramMakerPSet"), consumesCollector())
  , _recoVertexCollectionToken(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("pvCollection")))
  , _recoBeamSpotToken(consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("bsCollection")))
  , _firstOnly(iConfig.getUntrackedParameter<bool>("firstOnly",false))
{
   //now do what ever initialization is needed

  //

  _bspvhm.book();

}


BSvsPVAnalyzer::~BSvsPVAnalyzer()
{

   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to for each event  ------------
void
BSvsPVAnalyzer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{

  // get BS

  edm::Handle<reco::BeamSpot> bs;
  iEvent.getByToken(_recoBeamSpotToken,bs);

  // get PV

  edm::Handle<reco::VertexCollection> pvcoll;
  iEvent.getByToken(_recoVertexCollectionToken,pvcoll);

  if(_firstOnly) {
    reco::VertexCollection firstpv;
    if(pvcoll->size()) firstpv.push_back((*pvcoll)[0]);
    _bspvhm.fill(iEvent,firstpv,*bs);
  }
  else {
    _bspvhm.fill(iEvent,*pvcoll,*bs);
  }
}


// ------------ method called once each job just before starting event loop  ------------
void
BSvsPVAnalyzer::beginJob()
{ }

void
BSvsPVAnalyzer::beginRun(const edm::Run& iRun, const edm::EventSetup& iSetup) {

  _bspvhm.beginRun(iRun.run());

}

void
BSvsPVAnalyzer::endRun(const edm::Run& iRun, const edm::EventSetup& iSetup) {

}
// ------------ method called once each job just after ending the event loop  ------------
void
BSvsPVAnalyzer::endJob() {
}


//define this as a plug-in
DEFINE_FWK_MODULE(BSvsPVAnalyzer);