PileupVertexAccumulator.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PileupVertexAccumulator
// Class:      PileupVertexAccumulator
// 
//
// Original Author: Mike Hildreth - Notre Dame
//         Created:  Wed Jan 21 05:14:48 CET 2015
//
//


// system include files
#include <memory>
#include <set>

// user include files
#include "PileupVertexAccumulator.h"

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/one/EDProducer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"

#include "SimDataFormats/GeneratorProducts/interface/GenRunInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupVertexContent.h"

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

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

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "SimGeneral/MixingModule/interface/PileUpEventPrincipal.h"

#include "FWCore/Utilities/interface/StreamID.h"
#include "FWCore/Utilities/interface/Exception.h"


//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
//using namespace std;


namespace cms
{
  PileupVertexAccumulator::PileupVertexAccumulator(const edm::ParameterSet& iConfig, edm::stream::EDProducerBase& mixMod, edm::ConsumesCollector& iC) :
    Mtag_(iConfig.getParameter<edm::InputTag>("vtxTag")),
    fallbackMtag_(iConfig.getParameter<edm::InputTag>("vtxFallbackTag")),
    saveVtxTimes_(iConfig.getParameter<bool>("saveVtxTimes"))
  {
    edm::LogInfo ("PixelDigitizer ") <<"Enter the Pixel Digitizer";
    
    const std::string alias ("PileupVertexAccum"); 
    
    mixMod.produces<PileupVertexContent>().setBranchAlias(alias);

    iC.consumes<edm::HepMCProduct>(Mtag_);
    iC.mayConsume<edm::HepMCProduct>(fallbackMtag_);
  }
  
  PileupVertexAccumulator::~PileupVertexAccumulator(){  
  }


  //
  // member functions
  //
  
  void
  PileupVertexAccumulator::initializeEvent(edm::Event const& e, edm::EventSetup const& iSetup) {
    // Make sure that the first crossing processed starts indexing the minbias events from zero.

    pT_Hats_.clear();
    z_posns_.clear();
    t_posns_.clear();
  }

  void
  PileupVertexAccumulator::accumulate(edm::Event const& iEvent, edm::EventSetup const& iSetup) {
    // don't do anything for hard-scatter signal events
  }

  void
  PileupVertexAccumulator::accumulate(PileUpEventPrincipal const& iEvent, edm::EventSetup const& iSetup, edm::StreamID const& streamID) {

    edm::Handle<edm::HepMCProduct> MCevt;
    iEvent.getByLabel(Mtag_, MCevt);
    if(MCevt.whyFailed()) {
      iEvent.getByLabel(fallbackMtag_, MCevt);
    }

    const HepMC::GenEvent *myGenEvent = MCevt->GetEvent();

    double pthat = myGenEvent->event_scale();
    float pt_hat = float(pthat);

    pT_Hats_.push_back(pt_hat);

    HepMC::GenEvent::vertex_const_iterator viter;
    HepMC::GenEvent::vertex_const_iterator vbegin = myGenEvent->vertices_begin();
    HepMC::GenEvent::vertex_const_iterator vend = myGenEvent->vertices_end();

    // for production point, pick first vertex
    viter=vbegin; 

    if(viter!=vend){
      // The origin vertex (turn it to cm's from GenEvent mm's)
      HepMC::GenVertex* v = *viter;   
      float zpos = v->position().z()*0.1;
 
      z_posns_.push_back(zpos);

      if (saveVtxTimes_) {
          float tpos = v->position().t()/299792458e-6; // turn from mm to ns
          t_posns_.push_back(tpos);
      }
    }

    //    delete myGenEvent;

  }

  // ------------ method called to produce write the data  ------------
  void
  PileupVertexAccumulator::finalizeEvent(edm::Event& iEvent, const edm::EventSetup& iSetup) {

    std::unique_ptr<PileupVertexContent> PUVtxC(new PileupVertexContent(pT_Hats_, z_posns_, t_posns_));

    // write output to event
    iEvent.put(std::move(PUVtxC));
  }


}// end namespace cms::