TableOutputBranches.h

Go to the documentation of this file.

#ifndef PhysicsTools_NanoAOD_TableOutputBranches_h
#define PhysicsTools_NanoAOD_TableOutputBranches_h

#include <string>
#include <vector>
#include <TTree.h>
#include "FWCore/Framework/interface/EventForOutput.h"
#include "DataFormats/NanoAOD/interface/FlatTable.h"
#include "DataFormats/Provenance/interface/BranchDescription.h"
#include "FWCore/Utilities/interface/EDGetToken.h"

class TableOutputBranches {
 public:
    TableOutputBranches(const edm::BranchDescription *desc, const edm::EDGetToken & token ) :
        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");
    }

    void defineBranchesFromFirstEvent(const nanoaod::FlatTable & tab) ;
    void branch(TTree &tree) ;

    void fill(const edm::EventForOutput &iEvent, TTree & tree, bool extensions) ;

 private:
    edm::EDGetToken m_token;
    std::string  m_baseName;
    bool         m_singleton;
    enum { IsMain=0, IsExtension=1, DontKnowYetIfMainOrExtension=2 } m_extension;
    std::string  m_doc;
    UInt_t       m_counter;
    struct NamedBranchPtr {
        std::string name, title, rootTypeCode;
        TBranch * branch;
        NamedBranchPtr(const std::string & aname, const std::string & atitle, const std::string & rootType, TBranch *branchptr = nullptr) : 
            name(aname), title(atitle), rootTypeCode(rootType), branch(branchptr) {}
    };
    TBranch * m_counterBranch;
    std::vector<NamedBranchPtr> m_floatBranches;
    std::vector<NamedBranchPtr>   m_intBranches;
    std::vector<NamedBranchPtr> m_uint8Branches;
    bool m_branchesBooked;

    template<typename T>
    void fillColumn(NamedBranchPtr & pair, const nanoaod::FlatTable & tab) {
        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 * !
    }

};

#endif