TableOutputBranches Class Reference

#include <TableOutputBranches.h>

Classes
struct	NamedBranchPtr

Public Member Functions
void	branch (TTree &tree)
void	defineBranchesFromFirstEvent (const nanoaod::FlatTable &tab)
void	fill (const edm::OccurrenceForOutput &iWhatever, TTree &tree, bool extensions)
	TableOutputBranches (const edm::BranchDescription *desc, const edm::EDGetToken &token)

Private Types
enum	{ IsMain = 0, IsExtension = 1, DontKnowYetIfMainOrExtension = 2 }

Private Member Functions
template<typename T >
void	fillColumn (NamedBranchPtr &pair, const nanoaod::FlatTable &tab)

Private Attributes
std::string	m_baseName
bool	m_branchesBooked
UInt_t	m_counter
TBranch *	m_counterBranch
std::string	m_doc
std::vector< NamedBranchPtr >	m_doubleBranches
enum TableOutputBranches:: { ... }	m_extension
std::vector< NamedBranchPtr >	m_floatBranches
std::vector< NamedBranchPtr >	m_intBranches
bool	m_singleton
edm::EDGetToken	m_token
std::vector< NamedBranchPtr >	m_uint32Branches
std::vector< NamedBranchPtr >	m_uint8Branches

Detailed Description

Definition at line 12 of file TableOutputBranches.h.

Member Enumeration Documentation

anonymous enum

private

Enumerator
IsMain
IsExtension
DontKnowYetIfMainOrExtension

Definition at line 31 of file TableOutputBranches.h.

{ IsMain = 0, IsExtension = 1, DontKnowYetIfMainOrExtension = 2 } m_extension;

Constructor & Destructor Documentation

TableOutputBranches::TableOutputBranches ( const edm::BranchDescription *  desc, const edm::EDGetToken &  token  )

inline

Definition at line 14 of file TableOutputBranches.h.

References edm::BranchDescription::className().

      : m_token(token), m_extension(DontKnowYetIfMainOrExtension), m_branchesBooked(false) {
    if (desc->className() != "nanoaod::FlatTable")
      throw cms::Exception("Configuration", "NanoAODOutputModule can only write out nanoaod::FlatTable objects");
  }

Member Function Documentation

void TableOutputBranches::branch

(

TTree &

tree

)

Definition at line 40 of file TableOutputBranches.cc.

References Exception, IsExtension, m_baseName, m_counter, m_counterBranch, m_doc, m_doubleBranches, m_extension, m_floatBranches, m_intBranches, m_singleton, m_uint32Branches, m_uint8Branches, and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by python.cmstools.EventTree::__getattr__(), and fill().

                                            {
  if (!m_singleton) {
    if (m_extension == IsExtension) {
      m_counterBranch = tree.FindBranch(("n" + m_baseName).c_str());
      if (!m_counterBranch) {
        throw cms::Exception("LogicError",
                             "Trying to save an extension table for " + m_baseName +
                                 " before having saved the corresponding main table\n");
      }
    } else {
      if (tree.FindBranch(("n" + m_baseName).c_str()) != nullptr) {
        throw cms::Exception("LogicError", "Trying to save multiple main tables for " + m_baseName + "\n");
      }
      m_counterBranch = tree.Branch(("n" + m_baseName).c_str(), &m_counter, ("n" + m_baseName + "/i").c_str());
      m_counterBranch->SetTitle(m_doc.c_str());
    }
  }
  std::string varsize = m_singleton ? "" : "[n" + m_baseName + "]";
  for (std::vector<NamedBranchPtr> *branches :
       {&m_floatBranches, &m_intBranches, &m_uint8Branches, &m_uint32Branches, &m_doubleBranches}) {
    for (auto &pair : *branches) {
      std::string branchName = makeBranchName(m_baseName, pair.name);
      pair.branch =
          tree.Branch(branchName.c_str(), (void *)nullptr, (branchName + varsize + "/" + pair.rootTypeCode).c_str());
      pair.branch->SetTitle(pair.title.c_str());
    }
  }
}

void TableOutputBranches::defineBranchesFromFirstEvent

(

const nanoaod::FlatTable &

tab

)

Definition at line 11 of file TableOutputBranches.cc.

References nanoaod::FlatTable::Bool, nanoaod::FlatTable::columnDoc(), nanoaod::FlatTable::columnName(), nanoaod::FlatTable::columnType(), nanoaod::FlatTable::Double, Exception, nanoaod::FlatTable::Float, mps_fire::i, nanoaod::FlatTable::Int, m_baseName, m_doubleBranches, m_floatBranches, m_intBranches, m_uint32Branches, m_uint8Branches, nanoaod::FlatTable::name(), nanoaod::FlatTable::nColumns(), AlCaHLTBitMon_QueryRunRegistry::string, nanoaod::FlatTable::UInt32, nanoaod::FlatTable::UInt8, and isotrackApplyRegressor::var.

Referenced by fill().

                                                                                  {
  m_baseName = tab.name();
  for (size_t i = 0; i < tab.nColumns(); i++) {
    const std::string &var = tab.columnName(i);
    switch (tab.columnType(i)) {
      case nanoaod::FlatTable::ColumnType::Float:
        m_floatBranches.emplace_back(var, tab.columnDoc(i), "F");
        break;
      case nanoaod::FlatTable::ColumnType::Int:
        m_intBranches.emplace_back(var, tab.columnDoc(i), "I");
        break;
      case nanoaod::FlatTable::ColumnType::UInt8:
        m_uint8Branches.emplace_back(var, tab.columnDoc(i), "b");
        break;
      case nanoaod::FlatTable::ColumnType::Bool:
        m_uint8Branches.emplace_back(var, tab.columnDoc(i), "O");
        break;
      case nanoaod::FlatTable::ColumnType::UInt32:
        m_uint32Branches.emplace_back(var, tab.columnDoc(i), "i");
        break;
      case nanoaod::FlatTable::ColumnType::Double:
        m_doubleBranches.emplace_back(var, tab.columnDoc(i), "D");
        break;
      default:
        throw cms::Exception("LogicError", "Unsupported type");
    }
  }
}

void TableOutputBranches::fill	(	const edm::OccurrenceForOutput &	iWhatever,
		TTree &	tree,
		bool	extensions
	)

Fill the current table, if extensions == table.extension(). This parameter is used so that the fill is called first for non-extensions and then for extensions

Definition at line 69 of file TableOutputBranches.cc.

References branch(), defineBranchesFromFirstEvent(), nanoaod::FlatTable::doc(), DontKnowYetIfMainOrExtension, Exception, nanoaod::FlatTable::extension(), edm::OccurrenceForOutput::getByToken(), patZpeak::handle, IsExtension, IsMain, m_branchesBooked, m_counter, m_counterBranch, m_doc, m_doubleBranches, m_extension, m_floatBranches, m_intBranches, m_singleton, m_token, m_uint32Branches, m_uint8Branches, nanoaod::FlatTable::name(), nanoaod::FlatTable::singleton(), and nanoaod::FlatTable::size().

                                                                                                    {
  if (m_extension != DontKnowYetIfMainOrExtension) {
    if (extensions != m_extension)
      return;  // do nothing, wait to be called with the proper flag
  }

  edm::Handle<nanoaod::FlatTable> handle;
  iWhatever.getByToken(m_token, handle);
  const nanoaod::FlatTable &tab = *handle;
  m_counter = tab.size();
  m_singleton = tab.singleton();
  if (!m_branchesBooked) {
    m_extension = tab.extension() ? IsExtension : IsMain;
    if (extensions != m_extension)
      return;  // do nothing, wait to be called with the proper flag
    defineBranchesFromFirstEvent(tab);
    m_doc = tab.doc();
    m_branchesBooked = true;
    branch(tree);
  }
  if (!m_singleton && m_extension == IsExtension) {
    if (m_counter != *reinterpret_cast<UInt_t *>(m_counterBranch->GetAddress())) {
      throw cms::Exception("LogicError",
                           "Mismatch in number of entries between extension and main table for " + tab.name());
    }
  }
  for (auto &pair : m_floatBranches)
    fillColumn<float>(pair, tab);
  for (auto &pair : m_intBranches)
    fillColumn<int>(pair, tab);
  for (auto &pair : m_uint8Branches)
    fillColumn<uint8_t>(pair, tab);
  for (auto &pair : m_uint32Branches)
    fillColumn<uint32_t>(pair, tab);
  for (auto &pair : m_doubleBranches)
    fillColumn<double>(pair, tab);
}

template<typename T >

void TableOutputBranches::fillColumn ( NamedBranchPtr &  pair, const nanoaod::FlatTable &  tab  )

inlineprivate

Definition at line 52 of file TableOutputBranches.h.

References TableOutputBranches::NamedBranchPtr::branch, nanoaod::FlatTable::columnData(), nanoaod::FlatTable::columnIndex(), Exception, m_baseName, and TableOutputBranches::NamedBranchPtr::name.

                                                                     {
    int idx = tab.columnIndex(pair.name);
    if (idx == -1)
      throw cms::Exception("LogicError", "Missing column in input for " + m_baseName + "_" + pair.name);
    pair.branch->SetAddress(const_cast<T *>(&tab.columnData<T>(idx).front()));  // SetAddress should take a const * !
  }

Member Data Documentation

std::string TableOutputBranches::m_baseName

private

Definition at line 29 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fillColumn().

bool TableOutputBranches::m_branchesBooked

private

Definition at line 49 of file TableOutputBranches.h.

Referenced by fill().

UInt_t TableOutputBranches::m_counter

private

Definition at line 33 of file TableOutputBranches.h.

Referenced by branch(), and fill().

TBranch* TableOutputBranches::m_counterBranch

private

Definition at line 43 of file TableOutputBranches.h.

Referenced by branch(), and fill().

std::string TableOutputBranches::m_doc

private

Definition at line 32 of file TableOutputBranches.h.

Referenced by branch(), and fill().

std::vector<NamedBranchPtr> TableOutputBranches::m_doubleBranches

private

Definition at line 48 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fill().

enum { ... } TableOutputBranches::m_extension

Referenced by branch(), and fill().

std::vector<NamedBranchPtr> TableOutputBranches::m_floatBranches

private

Definition at line 44 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fill().

std::vector<NamedBranchPtr> TableOutputBranches::m_intBranches

private

Definition at line 45 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fill().

bool TableOutputBranches::m_singleton

private

Definition at line 30 of file TableOutputBranches.h.

Referenced by branch(), and fill().

edm::EDGetToken TableOutputBranches::m_token

private

Definition at line 28 of file TableOutputBranches.h.

Referenced by fill().

std::vector<NamedBranchPtr> TableOutputBranches::m_uint32Branches

private

Definition at line 47 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fill().

std::vector<NamedBranchPtr> TableOutputBranches::m_uint8Branches

private

Definition at line 46 of file TableOutputBranches.h.

Referenced by branch(), defineBranchesFromFirstEvent(), and fill().