SummaryTableOutputBranches.cc

Go to the documentation of this file.

#include "PhysicsTools/NanoAOD/plugins/SummaryTableOutputBranches.h"

template<typename Col>
void
SummaryTableOutputBranches::makeScalarBranches(const std::vector<Col> & tabcols, TTree & tree, const std::string & rootType, std::vector<NamedBranchPtr> & branches ) {
    for (const auto & col : tabcols) {
        auto * br = tree.Branch(col.name.c_str(), (void*)nullptr, (col.name+"/"+rootType).c_str());
        br->SetTitle(col.doc.c_str());
        branches.emplace_back(col.name, br);
    }
}

template<typename Col>
void
SummaryTableOutputBranches::makeVectorBranches(const std::vector<Col> & tabcols, TTree & tree, const std::string & rootType, std::vector<NamedVectorBranchPtr> & branches ) {
    for (const auto & col : tabcols) {
        auto * cbr = tree.Branch(("n"+col.name).c_str(), (void*)nullptr, ("n"+col.name+"/i").c_str());
        auto * vbr = tree.Branch(col.name.c_str(), (void*)nullptr, (col.name+"[n"+col.name+"]/"+rootType).c_str());
        cbr->SetTitle(("Number of entries in "+col.name).c_str());
        vbr->SetTitle(col.doc.c_str());
        branches.emplace_back(col.name, cbr, vbr);
    }
}

template<typename Col>
void
SummaryTableOutputBranches::fillScalarBranches(const std::vector<Col> & tabcols, std::vector<NamedBranchPtr> & branches ) {
    if (tabcols.size() != branches.size()) throw cms::Exception("LogicError", "Mismatch in table columns");
    for (unsigned int i = 0, n = tabcols.size(); i < n; ++i) {
        if (tabcols[i].name != branches[i].name) throw cms::Exception("LogicError", "Mismatch in table columns");
        branches[i].branch->SetAddress( const_cast<typename Col::value_type *>(& tabcols[i].value) );
    }
}

template<typename Col>
void
SummaryTableOutputBranches::fillVectorBranches(const std::vector<Col> & tabcols, std::vector<NamedVectorBranchPtr> & branches ) {
    if (tabcols.size() != branches.size()) throw cms::Exception("LogicError", "Mismatch in table columns");
    for (unsigned int i = 0, n = tabcols.size(); i < n; ++i) {
        if (tabcols[i].name != branches[i].name) throw cms::Exception("LogicError", "Mismatch in table columns");
        branches[i].count = tabcols[i].values.size();
        branches[i].branch->SetAddress( const_cast<typename Col::element_type *>(& tabcols[i].values.front()) );
    }
}



void 
SummaryTableOutputBranches::defineBranchesFromFirstEvent(const nanoaod::MergeableCounterTable & tab, TTree & tree) 
{
    makeScalarBranches(tab.intCols(), tree, "L", m_intBranches);
    makeScalarBranches(tab.floatCols(), tree, "D", m_floatBranches);
    makeVectorBranches(tab.vintCols(), tree, "L", m_vintBranches);
    makeVectorBranches(tab.vfloatCols(), tree, "D", m_vfloatBranches);

    // now we go set the pointers for the counter branches
    for (auto & vbp : m_vintBranches)   vbp.counterBranch->SetAddress( & vbp.count );
    for (auto & vbp : m_vfloatBranches) vbp.counterBranch->SetAddress( & vbp.count );
}


void SummaryTableOutputBranches::fill(const edm::OccurrenceForOutput &iWhatever, TTree & tree) 
{
    edm::Handle<nanoaod::MergeableCounterTable> handle;
    iWhatever.getByToken(m_token, handle);
    const nanoaod::MergeableCounterTable & tab = *handle;
    
    if(!m_branchesBooked) {
        defineBranchesFromFirstEvent(tab, tree);    
        m_branchesBooked=true;
    } 
    fillScalarBranches(tab.intCols(), m_intBranches);
    fillScalarBranches(tab.floatCols(), m_floatBranches);
    fillVectorBranches(tab.vintCols(), m_vintBranches);
    fillVectorBranches(tab.vfloatCols(), m_vfloatBranches);
}