CMS 3D CMS Logo
List of all members | Public Member Functions | Static Public Member Functions | Protected Attributes
LHETablesProducer Class Reference
Inheritance diagram for LHETablesProducer:
edm::global::EDProducer<> edm::global::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

std::unique_ptr< nanoaod::FlatTable > fillLHEObjectTable (const LHEEventProduct &lheProd, nanoaod::FlatTable &out) const
 
 LHETablesProducer (edm::ParameterSet const &params)
 
void produce (edm::StreamID id, edm::Event &iEvent, const edm::EventSetup &iSetup) const override
 
 ~LHETablesProducer () override
 
- Public Member Functions inherited from edm::global::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::global::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
 ~EDProducerBase () override
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo > consumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBase & operator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 
- Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 

Protected Attributes

const edm::EDGetTokenT< LHEEventProduct > lheTag_
 
const unsigned int precision_
 
const bool storeLHEParticles_
 

Additional Inherited Members

- Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
 
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 14 of file LHETablesProducer.cc.

Constructor & Destructor Documentation

LHETablesProducer::LHETablesProducer ( edm::ParameterSet const &  params)
inline

Definition at line 16 of file LHETablesProducer.cc.

References storeLHEParticles_.

16  :
17  lheTag_(consumes<LHEEventProduct>(params.getParameter<edm::InputTag>("lheInfo"))),
18  precision_(params.existsAs<int>("precision") ? params.getParameter<int>("precision") : -1),
19  storeLHEParticles_(params.existsAs<bool>("storeLHEParticles") ? params.getParameter<bool>("storeLHEParticles") : -1)
20  {
21  produces<nanoaod::FlatTable>("LHE");
22  if (storeLHEParticles_)
23  produces<nanoaod::FlatTable>("LHEPart");
24 
25  }
LHETablesProducer::precision_
const unsigned int precision_
Definition: LHETablesProducer.cc:133
LHETablesProducer::storeLHEParticles_
const bool storeLHEParticles_
Definition: LHETablesProducer.cc:134
LHETablesProducer::lheTag_
const edm::EDGetTokenT< LHEEventProduct > lheTag_
Definition: LHETablesProducer.cc:132
LHETablesProducer::~LHETablesProducer ( )
inlineoverride

Definition at line 27 of file LHETablesProducer.cc.

27 {}

Member Function Documentation

static void LHETablesProducer::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions)
inlinestatic

Definition at line 123 of file LHETablesProducer.cc.

References edm::ConfigurationDescriptions::add(), and edm::ParameterSetDescription::add().

123  {
124  edm::ParameterSetDescription desc;
125  desc.add<edm::InputTag>("lheInfo", edm::InputTag("externalLHEProducer"))->setComment("tag for the LHE information (LHEEventProduct)");
126  desc.add<int>("precision", -1)->setComment("precision on the 4-momenta of the LHE particles");
127  desc.add<bool>("storeLHEParticles", false)->setComment("Whether we want to store the 4-momenta of the status 1 particles at LHE level");
128  descriptions.add("lheInfoTable", desc);
129  }
edm::ParameterSetDescription::add
ParameterDescriptionBase * add(U const &iLabel, T const &value)
Definition: ParameterSetDescription.h:89
edm::ConfigurationDescriptions::add
void add(std::string const &label, ParameterSetDescription const &psetDescription)
Definition: ConfigurationDescriptions.cc:63
std::unique_ptr<nanoaod::FlatTable> LHETablesProducer::fillLHEObjectTable ( const LHEEventProduct &  lheProd,
nanoaod::FlatTable &  out 
) const
inline

Definition at line 46 of file LHETablesProducer.cc.

References funct::abs(), nanoaod::FlatTable::addColumnValue(), nanoaod::FlatTable::FloatColumn, LHEEventProduct::hepeup(), mps_fire::i, nanoaod::FlatTable::IntColumn, SiStripPI::max, gen::n, LHEEventProduct::npLO(), LHEEventProduct::npNLO(), p4, precision_, EnergyCorrector::pt, lhef::HEPEUP::PUP, mps_update::status, nanoaod::FlatTable::UInt8Column, and findQualityFiles::v.

Referenced by produce().

46  {
47  double lheHT = 0, lheHTIncoming = 0;
48  unsigned int lheNj = 0, lheNb = 0, lheNc = 0, lheNuds = 0, lheNglu = 0;
49  double lheVpt = 0;
50 
51  const auto & hepeup = lheProd.hepeup();
52  const auto & pup = hepeup.PUP;
53  int lep = -1, lepBar = -1, nu = -1, nuBar = -1;
54  std::vector<float> vals_pt;
55  std::vector<float> vals_eta;
56  std::vector<float> vals_phi;
57  std::vector<float> vals_mass;
58  std::vector<int> vals_pid;
59  for (unsigned int i = 0, n = pup.size(); i < n; ++i) {
60  int status = hepeup.ISTUP[i];
61  int idabs = std::abs(hepeup.IDUP[i]);
62  if (status == 1){
63  TLorentzVector p4(pup[i][0], pup[i][1], pup[i][2], pup[i][3]); // x,y,z,t
64  vals_pt.push_back(p4.Pt());
65  vals_eta.push_back(p4.Eta());
66  vals_phi.push_back(p4.Phi());
67  vals_mass.push_back(p4.M());
68  vals_pid.push_back(hepeup.IDUP[i]);
69  }
70  if ( (status == 1) && ( ( idabs == 21 ) || (idabs > 0 && idabs < 7) ) ) { //# gluons and quarks
71  // object counters
72  lheNj++;
73  if (idabs==5) lheNb++;
74  else if (idabs==4) lheNc++;
75  else if (idabs <= 3 ) lheNuds++;
76  else if (idabs == 21) lheNglu++;
77  // HT
78  double pt = std::hypot( pup[i][0], pup[i][1] ); // first entry is px, second py
79  lheHT += pt;
80  int mothIdx = std::max(hepeup.MOTHUP[i].first-1,0); //first and last mother as pair; first entry has index 1 in LHE; incoming particles return motherindex 0
81  int mothIdxTwo = std::max(hepeup.MOTHUP[i].second-1,0);
82  int mothStatus = hepeup.ISTUP[mothIdx];
83  int mothStatusTwo = hepeup.ISTUP[mothIdxTwo];
84  bool hasIncomingAsMother = mothStatus<0 || mothStatusTwo<0;
85  if (hasIncomingAsMother) lheHTIncoming += pt;
86  } else if (idabs == 12 || idabs == 14 || idabs == 16) {
87  (hepeup.IDUP[i] > 0 ? nu : nuBar) = i;
88  } else if (idabs == 11 || idabs == 13 || idabs == 15) {
89  (hepeup.IDUP[i] > 0 ? lep : lepBar) = i;
90  }
91  }
92  std::pair<int,int> v(0,0);
93  if (lep != -1 && lepBar != -1) v = std::make_pair(lep,lepBar);
94  else if (lep != -1 && nuBar != -1) v = std::make_pair(lep, nuBar);
95  else if (nu != -1 && lepBar != -1) v = std::make_pair(nu ,lepBar);
96  else if (nu != -1 && nuBar != -1) v = std::make_pair(nu , nuBar);
97  if (v.first != -1 && v.second != -1) {
98  lheVpt = std::hypot( pup[v.first][0] + pup[v.second][0], pup[v.first][1] + pup[v.second][1] );
99  }
100 
101  out.addColumnValue<uint8_t>("Njets", lheNj, "Number of jets (partons) at LHE step", nanoaod::FlatTable::UInt8Column);
102  out.addColumnValue<uint8_t>("Nb", lheNb, "Number of b partons at LHE step", nanoaod::FlatTable::UInt8Column);
103  out.addColumnValue<uint8_t>("Nc", lheNc, "Number of c partons at LHE step", nanoaod::FlatTable::UInt8Column);
104  out.addColumnValue<uint8_t>("Nuds", lheNuds, "Number of u,d,s partons at LHE step", nanoaod::FlatTable::UInt8Column);
105  out.addColumnValue<uint8_t>("Nglu", lheNglu, "Number of gluon partons at LHE step", nanoaod::FlatTable::UInt8Column);
106  out.addColumnValue<float>("HT", lheHT, "HT, scalar sum of parton pTs at LHE step", nanoaod::FlatTable::FloatColumn);
107  out.addColumnValue<float>("HTIncoming", lheHTIncoming, "HT, scalar sum of parton pTs at LHE step, restricted to partons", nanoaod::FlatTable::FloatColumn);
108  out.addColumnValue<float>("Vpt", lheVpt, "pT of the W or Z boson at LHE step", nanoaod::FlatTable::FloatColumn);
109  out.addColumnValue<uint8_t>("NpNLO", lheProd.npNLO(), "number of partons at NLO", nanoaod::FlatTable::UInt8Column);
110  out.addColumnValue<uint8_t>("NpLO", lheProd.npLO(), "number of partons at LO", nanoaod::FlatTable::UInt8Column);
111 
112 
113  auto outPart = std::make_unique<nanoaod::FlatTable>(vals_pt.size(), "LHEPart", false);
114  outPart->addColumn<float>("pt", vals_pt, "Pt of LHE particles", nanoaod::FlatTable::FloatColumn, this->precision_);
115  outPart->addColumn<float>("eta", vals_eta, "Pseodorapidity of LHE particles", nanoaod::FlatTable::FloatColumn, this->precision_);
116  outPart->addColumn<float>("phi", vals_phi, "Phi of LHE particles", nanoaod::FlatTable::FloatColumn, this->precision_);
117  outPart->addColumn<float>("mass", vals_mass, "Mass of LHE particles", nanoaod::FlatTable::FloatColumn, this->precision_);
118  outPart->addColumn<int> ("pdgId", vals_pid, "PDG ID of LHE particles", nanoaod::FlatTable::IntColumn);
119 
120  return outPart;
121  }
mps_fire.i
i
Definition: mps_fire.py:269
LHEEventProduct::hepeup
const lhef::HEPEUP & hepeup() const
Definition: LHEEventProduct.h:47
LHETablesProducer::precision_
const unsigned int precision_
Definition: LHETablesProducer.cc:133
mps_update.status
status
Definition: mps_update.py:67
p4
double p4[4]
Definition: TauolaWrapper.h:92
lhef::HEPEUP::PUP
std::vector< FiveVector > PUP
Definition: LesHouches.h:248
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
nanoaod::FlatTable::addColumnValue
void addColumnValue(const std::string &name, const C &value, const std::string &docString, ColumnType type=defaultColumnType< T >(), int mantissaBits=-1)
Definition: FlatTable.h:101
LHEEventProduct::npLO
int npLO() const
Definition: LHEEventProduct.h:41
LHEEventProduct::npNLO
int npNLO() const
Definition: LHEEventProduct.h:42
void LHETablesProducer::produce ( edm::StreamID  id,
edm::Event &  iEvent,
const edm::EventSetup &  iSetup 
) const
inlineoverride

Definition at line 29 of file LHETablesProducer.cc.

References fillLHEObjectTable(), edm::Event::getByToken(), nano_cff::lheInfo, lheTag_, eostools::move(), edm::Event::put(), and storeLHEParticles_.

29  {
30  auto lheTab = std::make_unique<nanoaod::FlatTable>(1, "LHE", true);
31 
32  edm::Handle<LHEEventProduct> lheInfo;
33  if (iEvent.getByToken(lheTag_, lheInfo)) {
34  auto lhePartTab = fillLHEObjectTable(*lheInfo, *lheTab);
35  if (storeLHEParticles_) iEvent.put(std::move(lhePartTab), "LHEPart");
36  } else {
37  if (storeLHEParticles_) { // need to store a dummy table anyway to make the framework happy
38  auto lhePartTab = std::make_unique<nanoaod::FlatTable>(1, "LHEPart", true);
39  iEvent.put(std::move(lhePartTab), "LHEPart");
40  }
41  }
42  iEvent.put(std::move(lheTab), "LHE");
43 
44  }
edm::Event::put
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:127
edm::Event::getByToken
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:508
LHETablesProducer::fillLHEObjectTable
std::unique_ptr< nanoaod::FlatTable > fillLHEObjectTable(const LHEEventProduct &lheProd, nanoaod::FlatTable &out) const
Definition: LHETablesProducer.cc:46
LHETablesProducer::storeLHEParticles_
const bool storeLHEParticles_
Definition: LHETablesProducer.cc:134
LHETablesProducer::lheTag_
const edm::EDGetTokenT< LHEEventProduct > lheTag_
Definition: LHETablesProducer.cc:132
nano_cff.lheInfo
lheInfo
Definition: nano_cff.py:100
eostools.move
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

const edm::EDGetTokenT<LHEEventProduct> LHETablesProducer::lheTag_
protected

Definition at line 132 of file LHETablesProducer.cc.

Referenced by produce().

const unsigned int LHETablesProducer::precision_
protected

Definition at line 133 of file LHETablesProducer.cc.

Referenced by fillLHEObjectTable().

const bool LHETablesProducer::storeLHEParticles_
protected

Definition at line 134 of file LHETablesProducer.cc.

Referenced by LHETablesProducer(), and produce().

Generated for CMSSW Reference Manual by doxygen 1.8.11