TableCollectionSet Class Reference

#include <TableOutputFields.h>

Public Member Functions
void	add (const edm::EDGetToken &table_token, const nanoaod::FlatTable &table)
void	createFields (const edm::EventForOutput &event, RNTupleModel &eventModel)
void	fill (const edm::EventForOutput &event)
void	print () const

Static Private Member Functions
static bool	hasAnonymousColumn (const nanoaod::FlatTable &table)

Private Attributes
std::vector< TableCollection >	m_collections
std::vector< TableOutputFields >	m_singletonFields
std::vector< TableOutputVectorFields >	m_vectorFields

Detailed Description

Definition at line 127 of file TableOutputFields.h.

Member Function Documentation

add()

void TableCollectionSet::add	(	const edm::EDGetToken &	table_token,
		const nanoaod::FlatTable &	table
	)

Definition at line 199 of file TableOutputFields.cc.

References HltBtagPostValidation_cff::c, universalConfigTemplate::collection, gather_cfg::cout, Exception, hasAnonymousColumn(), m_collections, m_singletonFields, m_vectorFields, and TableParser::table.

Referenced by NanoAODRNTupleOutputModule::initializeNTuple(), and counter.Counter::register().

                                                                                            {
  // skip empty tables -- requirement of RNTuple to define schema before filling
  if (table.nColumns() == 0) {
    std::cout << "Warning: skipping empty table: \n";
    printTable(table);
    return;
  }
  // Can handle either anonymous table or anonymous column but not both
  // - anonymous table: use column names directly as top-level fields
  // - anonymous column: use the table name as the field name
  if (table.name().empty() && hasAnonymousColumn(table)) {
    throw cms::Exception("LogicError", "Anonymous FlatTable and anonymous field");
  }
  // case 1: create a top-level RNTuple field for each table column
  if (table.name().empty() || hasAnonymousColumn(table)) {
    if (table.singleton()) {
      m_singletonFields.emplace_back(TableOutputFields(table_token));
    } else {
      m_vectorFields.emplace_back(TableOutputVectorFields(table_token));
    }
    return;
  }
  // case 2: Named singleton and vector tables are both written as RNTuple collections.
  auto collection = std::find_if(m_collections.begin(), m_collections.end(), [&](const TableCollection& c) {
    return c.getCollectionName() == table.name();
  });
  if (collection == m_collections.end()) {
    m_collections.emplace_back(TableCollection());
    m_collections.back().add(table_token, table);
    return;
  }
  collection->add(table_token, table);
}

createFields()

void TableCollectionSet::createFields	(	const edm::EventForOutput &	event,
		RNTupleModel &	eventModel
	)

Definition at line 240 of file TableOutputFields.cc.

References universalConfigTemplate::collection, Exception, m_collections, m_singletonFields, m_vectorFields, and TableParser::table.

Referenced by NanoAODRNTupleOutputModule::initializeNTuple().

                                                                                              {
  for (auto& collection : m_collections) {
    if (!collection.hasMainTable()) {
      throw cms::Exception("LogicError",
                           "Trying to save an extension table for " + collection.getCollectionName() +
                               " without the corresponding main table\n");
    }
    collection.createFields(event, eventModel);
  }
  for (auto& table : m_singletonFields) {
    table.createFields(event, eventModel);
  }
  for (auto& table : m_vectorFields) {
    table.createFields(event, eventModel);
  }
}

fill()

void TableCollectionSet::fill

(

const edm::EventForOutput &

event

)

Definition at line 257 of file TableOutputFields.cc.

References universalConfigTemplate::collection, l1GtPatternGenerator_cfi::fields, patZpeak::handle, m_collections, m_singletonFields, m_vectorFields, and TableParser::table.

Referenced by NanoAODRNTupleOutputModule::write().

                                                            {
  for (auto& collection : m_collections) {
    collection.fill(event);
  }
  for (auto& fields : m_singletonFields) {
    edm::Handle<nanoaod::FlatTable> handle;
    event.getByToken(fields.getToken(), handle);
    const auto& table = *handle;
    fields.fillEntry(table, 0);
  }
  for (auto& fields : m_vectorFields) {
    fields.fill(event);
  }
}

hasAnonymousColumn()

bool TableCollectionSet::hasAnonymousColumn ( const nanoaod::FlatTable &  table )

staticprivate

Definition at line 272 of file TableOutputFields.cc.

References Exception, mps_fire::i, TableParser::table, and to_string().

Referenced by add().

                                                                         {
  int num_anon = 0;
  for (std::size_t i = 0; i < table.nColumns(); i++) {
    if (table.columnName(i).empty()) {
      num_anon++;
    }
  }
  if (num_anon > 1) {
    throw cms::Exception("LogicError",
                         "FlatTable `" + table.name() + "` has " + std::to_string(num_anon) + "anonymous fields");
  }
  return num_anon;
}

print()

void TableCollectionSet::print

(

void

)

const

Definition at line 233 of file TableOutputFields.cc.

References universalConfigTemplate::collection, gather_cfg::cout, and m_collections.

                                     {
  for (const auto& collection : m_collections) {
    collection.print();
    std::cout << "\n";
  }
}

Member Data Documentation

m_collections

std::vector<TableCollection> TableCollectionSet::m_collections

private

Definition at line 138 of file TableOutputFields.h.

Referenced by add(), createFields(), fill(), and print().

m_singletonFields

std::vector<TableOutputFields> TableCollectionSet::m_singletonFields

private

Definition at line 139 of file TableOutputFields.h.

Referenced by add(), createFields(), and fill().

m_vectorFields

std::vector<TableOutputVectorFields> TableCollectionSet::m_vectorFields

private

Definition at line 140 of file TableOutputFields.h.

Referenced by add(), createFields(), and fill().