PixelVertexCollectionTrimmer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package: RecoPixelVertexing/PixelVertexFinding
// Class: PixelVertexCollectionTrimmer
//
//
// Original Author: Riccardo Manzoni
// Created: Tue, 01 Apr 2014 10:11:16 GMT
//
//


// system include files
#include <memory>

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

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

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

#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "RecoPixelVertexing/PixelVertexFinding/interface/PVClusterComparer.h"

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

class PixelVertexCollectionTrimmer : public edm::stream::EDProducer<> {
public:
  explicit PixelVertexCollectionTrimmer(const edm::ParameterSet&);
  ~PixelVertexCollectionTrimmer();

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  virtual void produce(edm::Event&, const edm::EventSetup&) override;

  edm::EDGetTokenT<reco::VertexCollection> vtxToken_;
  unsigned int maxVtx_ ;
  double fractionSumPt2_ ;
  double minSumPt2_ ;

  PVClusterComparer* pvComparer_;
};

PixelVertexCollectionTrimmer::PixelVertexCollectionTrimmer(const edm::ParameterSet& iConfig)
{

  edm::InputTag vtxInputTag = iConfig.getParameter<edm::InputTag>("src" );
  vtxToken_ = consumes<reco::VertexCollection>(vtxInputTag);
  maxVtx_         = iConfig.getParameter<unsigned int> ("maxVtx" );
  fractionSumPt2_ = iConfig.getParameter<double> ("fractionSumPt2" );
  minSumPt2_      = iConfig.getParameter<double> ("minSumPt2" );

  edm::ParameterSet PVcomparerPSet = iConfig.getParameter<edm::ParameterSet>("PVcomparer");
  double track_pt_min   = PVcomparerPSet.getParameter<double>("track_pt_min");
  double track_pt_max   = PVcomparerPSet.getParameter<double>("track_pt_max");
  double track_chi2_max = PVcomparerPSet.getParameter<double>("track_chi2_max");
  double track_prob_min = PVcomparerPSet.getParameter<double>("track_prob_min");

  pvComparer_ = new PVClusterComparer(track_pt_min, track_pt_max, track_chi2_max, track_prob_min);

  produces<reco::VertexCollection>();
}

PixelVertexCollectionTrimmer::~PixelVertexCollectionTrimmer()
{
}

// ------------ method called to produce the data ------------
void
PixelVertexCollectionTrimmer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  using namespace edm;
   
  std::auto_ptr<reco::VertexCollection> vtxs_trim(new reco::VertexCollection);

  edm::Handle<reco::VertexCollection> vtxs;
  iEvent.getByToken(vtxToken_,vtxs);

  double sumpt2 ;
  //double sumpt2previous = -99. ;

  // this is not the logic we want, at least for now
  // if requires the sumpt2 for vtx_n to be > threshold * sumpt2 vtx_n-1
  // for (reco::VertexCollection::const_iterator vtx = vtxs->begin(); vtx != vtxs->end(); ++vtx, ++counter){
  // if (counter > maxVtx_) break ;
  // sumpt2 = PVCluster.pTSquaredSum(*vtx) ;
  // if (sumpt2 > sumpt2previous*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
  // else if (counter == 0 ) vtxs_trim->push_back(*vtx) ;
  // sumpt2previous = sumpt2 ;
  // }

  double sumpt2first = pvComparer_->pTSquaredSum(*(vtxs->begin())) ;

  for (reco::VertexCollection::const_iterator vtx = vtxs->begin(), evtx=vtxs->end(); 
       vtx != evtx; ++vtx) {
    if (vtxs_trim->size() >= maxVtx_) break ;
    sumpt2 = pvComparer_->pTSquaredSum(*vtx) ;
    //    std::cout << "sumpt2: " << sumpt2 << "[" << sumpt2first << "]" << std::endl;
    //    if (sumpt2 >= sumpt2first*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
    if (sumpt2 >= sumpt2first*fractionSumPt2_ && sumpt2 > minSumPt2_ ) vtxs_trim->push_back(*vtx) ;
  }
  //  std::cout << " ==> # vertices: " << vtxs_trim->size() << std::endl;
  iEvent.put(vtxs_trim);
    
}

// ------------ method fills 'descriptions' with the allowed parameters for the module ------------
void
PixelVertexCollectionTrimmer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>    ("src",           edm::InputTag(""))
    ->setComment("input (pixel) vertex collection");
  desc.add<unsigned int>     ("maxVtx",        100)
    ->setComment("max output collection size (number of accepted vertices)");
  desc.add<double>           ("fractionSumPt2",  0.3)
    ->setComment("threshold on sumPt2 fraction of the leading vertex");
  desc.add<double>           ("minSumPt2",       0. )
    ->setComment("min sumPt2");
  edm::ParameterSetDescription PVcomparerPSet;
  PVcomparerPSet.add<double>("track_pt_min",1.0)
    ->setComment("min track p_T");
  PVcomparerPSet.add<double>("track_pt_max",10.0)
    ->setComment("max track p_T");
  PVcomparerPSet.add<double>("track_chi2_max",99999.)
    ->setComment("max track chi2");
  PVcomparerPSet.add<double>("track_prob_min",-1.)
    ->setComment("min track prob");
  desc.add<edm::ParameterSetDescription>("PVcomparer", PVcomparerPSet)
    ->setComment("from RecoPixelVertexing/PixelVertexFinding/python/PVClusterComparer_cfi.py");
  descriptions.add("hltPixelVertexCollectionTrimmer",desc);
}

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