GenProtonTableProducer Class Reference

Inheritance diagram for GenProtonTableProducer:

Public Member Functions
	GenProtonTableProducer (const edm::ParameterSet &)
	~GenProtonTableProducer () override=default
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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

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

Private Attributes
const StringCutObjectSelector< reco::Candidate >	protonsCut_
const edm::EDGetTokenT< reco::GenParticleCollection >	prunedCandsToken_
const edm::EDGetTokenT< reco::GenParticleCollection >	puAltCandsToken_
const edm::EDGetTokenT< reco::GenParticleCollection >	puCandsToken_
const std::string	table_name_
const double	tolerance_
bool	use_alt_coll_ {false}
	Are we using premix/mix collection name for PU protons? More...

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

Detailed Description

Definition at line 26 of file GenProtonTableProducer.cc.

Constructor & Destructor Documentation

GenProtonTableProducer()

GenProtonTableProducer::GenProtonTableProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 44 of file GenProtonTableProducer.cc.

    : prunedCandsToken_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("srcPruned"))),
      puCandsToken_(mayConsume<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("srcPUProtons"))),
      puAltCandsToken_(mayConsume<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("srcAltPUProtons"))),
      protonsCut_(iConfig.getParameter<std::string>("cut")),
      table_name_(iConfig.getParameter<std::string>("name")),
      tolerance_(iConfig.getParameter<double>("tolerance")) {
  produces<nanoaod::FlatTable>();
}

~GenProtonTableProducer()

GenProtonTableProducer::~GenProtonTableProducer ( )

overridedefault

Member Function Documentation

fillDescriptions()

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

static

Definition at line 101 of file GenProtonTableProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, HLT_2022v11_cff::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                        {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("srcPruned", edm::InputTag("prunedGenParticles"))
      ->setComment("input source for pruned gen-level particle candidates");
  desc.add<edm::InputTag>("srcPUProtons", edm::InputTag("genPUProtons"))
      ->setComment("input source for pileup protons collection");
  desc.add<edm::InputTag>("srcAltPUProtons", edm::InputTag("genPUProtons", "genPUProtons"))
      ->setComment("alternative input source for pileup protons collection (for premix-mix backward compatibility)");
  desc.add<std::string>("cut", "")->setComment("proton kinematic selection");
  desc.add<std::string>("name", "GenProton")->setComment("flat table name");
  desc.add<std::string>("doc", "generator level information on (signal+PU) protons")
      ->setComment("flat table description");
  desc.add<double>("tolerance", 1.e-3)->setComment("relative difference between the signal and pileup protons momenta");
  descriptions.add("genProtonTable", desc);
}

produce()

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

overrideprivate

Definition at line 54 of file GenProtonTableProducer.cc.

References mps_fire::i, iEvent, eostools::move(), protonsCut_, prunedCandsToken_, puAltCandsToken_, puCandsToken_, table_name_, tolerance_, and use_alt_coll_.

                                                                           {
  // define the variables
  std::vector<float> pxs, pys, pzs, vzs;
  std::vector<bool> isPUs;
  // first loop over signal protons
  for (const auto& pruned_cand : iEvent.get(prunedCandsToken_)) {
    if (!protonsCut_(pruned_cand))
      continue;
    pxs.emplace_back(pruned_cand.px());
    pys.emplace_back(pruned_cand.py());
    pzs.emplace_back(pruned_cand.pz());
    vzs.emplace_back(pruned_cand.vz());
    isPUs.emplace_back(false);
  }
  // then loop over pruned candidates ; if already in signal protons, discard
  edm::Handle<reco::GenParticleCollection> hPUCands;
  if (use_alt_coll_ || !iEvent.getByToken(puCandsToken_, hPUCands))
    use_alt_coll_ = iEvent.getByToken(puAltCandsToken_, hPUCands);
  for (const auto& pu_cand : *hPUCands) {
    if (!protonsCut_(pu_cand))
      continue;
    bool associated{false};
    for (size_t i = 0; i < pzs.size(); ++i) {
      if (fabs(1. - pxs.at(i) / pu_cand.px()) < tolerance_ && fabs(1. - pys.at(i) / pu_cand.py()) < tolerance_ &&
          fabs(1. - pzs.at(i) / pu_cand.pz()) < tolerance_) {
        associated = true;
        break;
      }
    }
    if (associated)
      continue;
    pxs.emplace_back(pu_cand.px());
    pys.emplace_back(pu_cand.py());
    pzs.emplace_back(pu_cand.pz());
    vzs.emplace_back(pu_cand.vz());
    isPUs.emplace_back(true);
  }

  auto protons_table = std::make_unique<nanoaod::FlatTable>(isPUs.size(), table_name_, false);
  protons_table->addColumn<float>("px", pxs, "proton horizontal momentum", 8);
  protons_table->addColumn<float>("py", pys, "proton vertical momentum", 8);
  protons_table->addColumn<float>("pz", pzs, "proton longitudinal momentum", 8);
  protons_table->addColumn<float>("vz", vzs, "proton vertex longitudinal coordinate", 8);
  protons_table->addColumn<bool>("isPU", isPUs, "pileup proton?");
  iEvent.put(std::move(protons_table));
}

Member Data Documentation

protonsCut_

const StringCutObjectSelector<reco::Candidate> GenProtonTableProducer::protonsCut_

private

Definition at line 38 of file GenProtonTableProducer.cc.

Referenced by produce().

prunedCandsToken_

const edm::EDGetTokenT<reco::GenParticleCollection> GenProtonTableProducer::prunedCandsToken_

private

Definition at line 36 of file GenProtonTableProducer.cc.

Referenced by produce().

puAltCandsToken_

const edm::EDGetTokenT<reco::GenParticleCollection> GenProtonTableProducer::puAltCandsToken_

private

Definition at line 37 of file GenProtonTableProducer.cc.

Referenced by produce().

puCandsToken_

const edm::EDGetTokenT<reco::GenParticleCollection> GenProtonTableProducer::puCandsToken_

private

Definition at line 37 of file GenProtonTableProducer.cc.

Referenced by produce().

table_name_

const std::string GenProtonTableProducer::table_name_

private

Definition at line 39 of file GenProtonTableProducer.cc.

Referenced by produce().

tolerance_

const double GenProtonTableProducer::tolerance_

private

Definition at line 40 of file GenProtonTableProducer.cc.

Referenced by produce().

use_alt_coll_

bool GenProtonTableProducer::use_alt_coll_ {false}

private

Are we using premix/mix collection name for PU protons?

Definition at line 41 of file GenProtonTableProducer.cc.

Referenced by produce().