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() override;

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

private:
  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;

  auto vtxs_trim = std::make_unique<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(std::move(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);