TableCollection 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)
const std::string &	getCollectionName () const
bool	hasMainTable ()
void	print () const
	TableCollection ()=default

Private Attributes
std::shared_ptr< RNTupleCollectionWriter >	m_collection
std::string	m_collectionName
std::vector< TableOutputFields >	m_extensions
TableOutputFields	m_main

Detailed Description

Definition at line 100 of file TableOutputFields.h.

Constructor & Destructor Documentation

TableCollection()

TableCollection::TableCollection ( )

default

Member Function Documentation

add()

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

Definition at line 136 of file TableOutputFields.cc.

References Exception, hasMainTable(), m_collectionName, m_extensions, m_main, and TableParser::table.

Referenced by counter.Counter::register().

                                                                                         {
  if (m_collectionName.empty()) {
    m_collectionName = table.name();
  }
  if (table.extension()) {
    m_extensions.emplace_back(TableOutputFields(table_token));
    return;
  }
  if (hasMainTable()) {
    throw cms::Exception("LogicError", "Trying to save multiple main tables for " + m_collectionName + "\n");
  }
  m_main = TableOutputFields(table_token);
}

createFields()

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

Definition at line 150 of file TableOutputFields.cc.

References TableOutputFields::createFields(), runGCPTkAlMap::extension, TableOutputFields::getToken(), patZpeak::handle, m_collection, m_collectionName, m_extensions, m_main, eostools::move(), and TableParser::table.

                                                                                           {
  auto collectionModel = RNTupleModel::Create();
  m_main.createFields(event, *collectionModel);
  for (auto& extension : m_extensions) {
    extension.createFields(event, *collectionModel);
  }
  edm::Handle<nanoaod::FlatTable> handle;
  event.getByToken(m_main.getToken(), handle);
  const nanoaod::FlatTable& table = *handle;
  collectionModel->SetDescription(table.doc());
  m_collection = eventModel.MakeCollection(m_collectionName, std::move(collectionModel));
}

fill()

void TableCollection::fill

(

const edm::EventForOutput &

event

)

Definition at line 163 of file TableOutputFields.cc.

References Exception, TableOutputFields::fillEntry(), TableOutputFields::getToken(), patZpeak::handle, mps_fire::i, m_collection, m_collectionName, m_extensions, and m_main.

                                                         {
  edm::Handle<nanoaod::FlatTable> handle;
  event.getByToken(m_main.getToken(), handle);
  const auto& main_table = *handle;
  auto table_size = main_table.size();
  for (std::size_t i = 0; i < table_size; i++) {
    m_main.fillEntry(main_table, i);
    for (auto& ext : m_extensions) {
      edm::Handle<nanoaod::FlatTable> handle;
      event.getByToken(ext.getToken(), handle);
      const auto& ext_table = *handle;
      if (ext_table.size() != table_size) {
        throw cms::Exception("LogicError",
                             "Mismatch in number of entries between extension and main table for " + m_collectionName);
      }
      ext.fillEntry(ext_table, i);
    }
    m_collection->Fill();
  }
}

getCollectionName()

const std::string & TableCollection::getCollectionName

(

)

const

Definition at line 195 of file TableOutputFields.cc.

References m_collectionName.

{ return m_collectionName; }

hasMainTable()

bool TableCollection::hasMainTable

(

)

Definition at line 193 of file TableOutputFields.cc.

References TableOutputFields::getToken(), edm::EDGetToken::isUninitialized(), and m_main.

Referenced by add().

{ return !m_main.getToken().isUninitialized(); }

print()

void TableCollection::print

(

void

)

const

Definition at line 184 of file TableOutputFields.cc.

References gather_cfg::cout, m_collectionName, m_extensions, m_main, and TableOutputFields::print().

                                  {
  std::cout << "Collection: " << m_collectionName << " {\n";
  m_main.print();
  for (const auto& ext : m_extensions) {
    ext.print();
  }
  std::cout << "}\n";
}

Member Data Documentation

m_collection

std::shared_ptr<RNTupleCollectionWriter> TableCollection::m_collection

private

Definition at line 121 of file TableOutputFields.h.

Referenced by createFields(), and fill().

m_collectionName

std::string TableCollection::m_collectionName

private

Definition at line 117 of file TableOutputFields.h.

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

m_extensions

std::vector<TableOutputFields> TableCollection::m_extensions

private

Definition at line 124 of file TableOutputFields.h.

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

m_main

TableOutputFields TableCollection::m_main

private

Definition at line 123 of file TableOutputFields.h.

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