VertexTableProducer Class Reference

#include <PhysicsTools/VertexTableProducer/plugins/VertexTableProducer.cc>

Inheritance diagram for VertexTableProducer:

Public Member Functions
	VertexTableProducer (const edm::ParameterSet &)
	~VertexTableProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
void	beginStream (edm::StreamID) override
void	endStream () override
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
const double	dlenMin_
const double	dlenSigMin_
const StringCutObjectSelector< reco::Vertex >	goodPvCut_
const std::string	goodPvCutString_
const std::string	pvName_
const edm::EDGetTokenT< std::vector< reco::Vertex > >	pvs_
const edm::EDGetTokenT< edm::ValueMap< float > >	pvsScore_
const bool	storeCharge_
const StringCutObjectSelector< reco::Candidate >	svCut_
const std::string	svDoc_
const std::string	svName_
const edm::EDGetTokenT< edm::View< reco::VertexCompositePtrCandidate > >	svs_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 48 of file VertexTableProducer.cc.

Constructor & Destructor Documentation

VertexTableProducer::VertexTableProducer ( const edm::ParameterSet &  params )

explicit

Definition at line 83 of file VertexTableProducer.cc.

    : pvs_(consumes<std::vector<reco::Vertex>>(params.getParameter<edm::InputTag>("pvSrc"))),
      pvsScore_(consumes<edm::ValueMap<float>>(params.getParameter<edm::InputTag>("pvSrc"))),
      svs_(consumes<edm::View<reco::VertexCompositePtrCandidate>>(params.getParameter<edm::InputTag>("svSrc"))),
      svCut_(params.getParameter<std::string>("svCut"), true),
      goodPvCut_(params.getParameter<std::string>("goodPvCut"), true),
      goodPvCutString_(params.getParameter<std::string>("goodPvCut")),
      pvName_(params.getParameter<std::string>("pvName")),
      svName_(params.getParameter<std::string>("svName")),
      svDoc_(params.getParameter<std::string>("svDoc")),
      dlenMin_(params.getParameter<double>("dlenMin")),
      dlenSigMin_(params.getParameter<double>("dlenSigMin")),
      storeCharge_(params.getParameter<bool>("storeCharge"))

{
  produces<nanoaod::FlatTable>("pv");
  produces<nanoaod::FlatTable>("otherPVs");
  produces<nanoaod::FlatTable>("svs");
  produces<edm::PtrVector<reco::Candidate>>();
}

VertexTableProducer::~VertexTableProducer ( )

override

Definition at line 104 of file VertexTableProducer.cc.

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

Member Function Documentation

void VertexTableProducer::beginStream ( edm::StreamID  )

overrideprivate

Definition at line 215 of file VertexTableProducer.cc.

{}

void VertexTableProducer::endStream ( )

overrideprivate

Definition at line 218 of file VertexTableProducer.cc.

{}

void VertexTableProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 221 of file VertexTableProducer.cc.

References edm::ConfigurationDescriptions::addDefault(), DEFINE_FWK_MODULE, and edm::ParameterSetDescription::setUnknown().

                                                                                     {
  //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.setUnknown();
  descriptions.addDefault(desc);
}

void VertexTableProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 115 of file VertexTableProducer.cc.

References EnergyCorrector::c, ALCARECOTkAlJpsiMuMu_cff::charge, reco::Candidate::charge(), RecoVertex::convertError(), RecoVertex::convertPos(), VertexDistance3D::distance(), VertexDistanceXY::distance(), dlenMin_, dlenSigMin_, PVValHelper::dx, PVValHelper::dxy, PVValHelper::dy, PVValHelper::dz, nanoaod::FlatTable::FloatColumn, edm::Event::getByToken(), goodPvCut_, goodPvCutString_, AlcaRecoSelection_cff::goodPVs, mps_fire::i, triggerObjects_cff::id, nanoaod::FlatTable::IntColumn, eostools::move(), edm::Event::put(), MetAnalyzer::pv(), pvName_, pvs_, pvsScore_, Measurement1D::significance(), mathSSE::sqrt(), storeCharge_, pfDeepBoostedJetPreprocessParams_cfi::sv, svCut_, svName_, svs_, and Measurement1D::value().

                                                                               {
  using namespace edm;
  edm::Handle<edm::ValueMap<float>> pvsScoreIn;
  edm::Handle<std::vector<reco::Vertex>> pvsIn;
  iEvent.getByToken(pvs_, pvsIn);
  iEvent.getByToken(pvsScore_, pvsScoreIn);
  auto pvTable = std::make_unique<nanoaod::FlatTable>(1, pvName_, true);
  pvTable->addColumnValue<float>(
      "ndof", (*pvsIn)[0].ndof(), "main primary vertex number of degree of freedom", nanoaod::FlatTable::FloatColumn, 8);
  pvTable->addColumnValue<float>(
      "x", (*pvsIn)[0].position().x(), "main primary vertex position x coordinate", nanoaod::FlatTable::FloatColumn, 10);
  pvTable->addColumnValue<float>(
      "y", (*pvsIn)[0].position().y(), "main primary vertex position y coordinate", nanoaod::FlatTable::FloatColumn, 10);
  pvTable->addColumnValue<float>(
      "z", (*pvsIn)[0].position().z(), "main primary vertex position z coordinate", nanoaod::FlatTable::FloatColumn, 16);
  pvTable->addColumnValue<float>(
      "chi2", (*pvsIn)[0].normalizedChi2(), "main primary vertex reduced chi2", nanoaod::FlatTable::FloatColumn, 8);
  int goodPVs = 0;
  for (const auto& pv : *pvsIn)
    if (goodPvCut_(pv))
      goodPVs++;
  pvTable->addColumnValue<int>(
      "npvs", (*pvsIn).size(), "total number of reconstructed primary vertices", nanoaod::FlatTable::IntColumn);
  pvTable->addColumnValue<int>("npvsGood",
                               goodPVs,
                               "number of good reconstructed primary vertices. selection:" + goodPvCutString_,
                               nanoaod::FlatTable::IntColumn);
  pvTable->addColumnValue<float>("score",
                                 (*pvsScoreIn).get(pvsIn.id(), 0),
                                 "main primary vertex score, i.e. sum pt2 of clustered objects",
                                 nanoaod::FlatTable::FloatColumn,
                                 8);

  auto otherPVsTable =
      std::make_unique<nanoaod::FlatTable>((*pvsIn).size() > 4 ? 3 : (*pvsIn).size() - 1, "Other" + pvName_, false);
  std::vector<float> pvsz;
  for (size_t i = 1; i < (*pvsIn).size() && i < 4; i++)
    pvsz.push_back((*pvsIn)[i - 1].position().z());
  otherPVsTable->addColumn<float>(
      "z", pvsz, "Z position of other primary vertices, excluding the main PV", nanoaod::FlatTable::FloatColumn, 8);

  edm::Handle<edm::View<reco::VertexCompositePtrCandidate>> svsIn;
  iEvent.getByToken(svs_, svsIn);
  auto selCandSv = std::make_unique<PtrVector<reco::Candidate>>();
  std::vector<float> dlen, dlenSig, pAngle, dxy, dxySig;
  std::vector<int> charge;
  VertexDistance3D vdist;
  VertexDistanceXY vdistXY;

  size_t i = 0;
  const auto& PV0 = pvsIn->front();
  for (const auto& sv : *svsIn) {
    if (svCut_(sv)) {
      Measurement1D dl =
          vdist.distance(PV0, VertexState(RecoVertex::convertPos(sv.position()), RecoVertex::convertError(sv.error())));
      if (dl.value() > dlenMin_ and dl.significance() > dlenSigMin_) {
        dlen.push_back(dl.value());
        dlenSig.push_back(dl.significance());
        edm::Ptr<reco::Candidate> c = svsIn->ptrAt(i);
        selCandSv->push_back(c);
        double dx = (PV0.x() - sv.vx()), dy = (PV0.y() - sv.vy()), dz = (PV0.z() - sv.vz());
        double pdotv = (dx * sv.px() + dy * sv.py() + dz * sv.pz()) / sv.p() / sqrt(dx * dx + dy * dy + dz * dz);
        pAngle.push_back(std::acos(pdotv));
        Measurement1D d2d = vdistXY.distance(
            PV0, VertexState(RecoVertex::convertPos(sv.position()), RecoVertex::convertError(sv.error())));
        dxy.push_back(d2d.value());
        dxySig.push_back(d2d.significance());

        if (storeCharge_) {
          int sum_charge = 0;
          for (unsigned int id = 0; id < sv.numberOfDaughters(); ++id) {
            const reco::Candidate* daughter = sv.daughter(id);
            sum_charge += daughter->charge();
          }
          charge.push_back(sum_charge);
        }
      }
    }
    i++;
  }

  auto svsTable = std::make_unique<nanoaod::FlatTable>(selCandSv->size(), svName_, false);
  // For SV we fill from here only stuff that cannot be created with the SimpleFlatTableProducer
  svsTable->addColumn<float>("dlen", dlen, "decay length in cm", nanoaod::FlatTable::FloatColumn, 10);
  svsTable->addColumn<float>("dlenSig", dlenSig, "decay length significance", nanoaod::FlatTable::FloatColumn, 10);
  svsTable->addColumn<float>("dxy", dxy, "2D decay length in cm", nanoaod::FlatTable::FloatColumn, 10);
  svsTable->addColumn<float>("dxySig", dxySig, "2D decay length significance", nanoaod::FlatTable::FloatColumn, 10);
  svsTable->addColumn<float>(
      "pAngle", pAngle, "pointing angle, i.e. acos(p_SV * (SV - PV)) ", nanoaod::FlatTable::FloatColumn, 10);
  if (storeCharge_) {
    svsTable->addColumn<int>("charge", charge, "sum of the charge of the SV tracks", nanoaod::FlatTable::IntColumn, 10);
  }

  iEvent.put(std::move(pvTable), "pv");
  iEvent.put(std::move(otherPVsTable), "otherPVs");
  iEvent.put(std::move(svsTable), "svs");
  iEvent.put(std::move(selCandSv));
}

Member Data Documentation

const double VertexTableProducer::dlenMin_

private

Definition at line 76 of file VertexTableProducer.cc.

Referenced by produce().

const double VertexTableProducer::dlenSigMin_

private

Definition at line 76 of file VertexTableProducer.cc.

Referenced by produce().

const StringCutObjectSelector<reco::Vertex> VertexTableProducer::goodPvCut_

private

Definition at line 71 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::goodPvCutString_

private

Definition at line 72 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::pvName_

private

Definition at line 73 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector<reco::Vertex> > VertexTableProducer::pvs_

private

Definition at line 67 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::ValueMap<float> > VertexTableProducer::pvsScore_

private

Definition at line 68 of file VertexTableProducer.cc.

Referenced by produce().

const bool VertexTableProducer::storeCharge_

private

Definition at line 77 of file VertexTableProducer.cc.

Referenced by produce().

const StringCutObjectSelector<reco::Candidate> VertexTableProducer::svCut_

private

Definition at line 70 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::svDoc_

private

Definition at line 75 of file VertexTableProducer.cc.

const std::string VertexTableProducer::svName_

private

Definition at line 74 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::View<reco::VertexCompositePtrCandidate> > VertexTableProducer::svs_

private

Definition at line 69 of file VertexTableProducer.cc.

Referenced by produce().