TreeManager.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef TREEMANAGER_H_
#define TREEMANAGER_H_
 
#include "TTree.h"
 
#include <string>
#include <list>
#include <sstream>
#include <functional>
#include <vector>
 
class TFile;
 
namespace top {
  class TreeManager {
  public:
    TreeManager();
 
    TreeManager(const std::string& name, TFile* outputFile, const int, const int, const int);
 
    TreeManager(const TreeManager&& rhs);
 
    TreeManager(const TreeManager&) = delete;
 
    TreeManager& operator = (TreeManager const&) = delete;
 
    void initialize(const std::string& name, TFile* outputFile, const int, const int, const int);
 
    void fill();
 
    const std::string& name() const;
 
    typedef std::function<int (top::TreeManager const*, std::string const&)> BranchFilter;
 
    std::vector<BranchFilter>& branchFilters() {return m_branchFilters;}
 
    std::vector<BranchFilter> const& branchFilters() const {return m_branchFilters;}
 
    template<class T>
    void makeOutputVariable(T& obj, const std::string name) {
      for (auto branchFilter : branchFilters()) {
        int status = branchFilter(this, name);
        if (status > 0) break;
        if (status == 0) return;
      }
      const char* type_name = typeid(obj).name();
      if (strlen(type_name) == 1) {
        std::ostringstream leaflist;
        leaflist << name << "/" << RootType(type_name);
        m_tree->Branch(name.c_str(), &obj, leaflist.str().c_str(), m_basketSizePrimitive);
      } else {
        m_outputVarPointers.push_back(&obj);
        T** pointer = reinterpret_cast< T** >(&m_outputVarPointers.back());
        m_tree->Branch(name.c_str(), pointer, m_basketSizeVector);
      }
    }
 
  private:
    std::string m_name;
 
 
    const char* RootType(const char* typeid_type);
 
    const char* TypeidType(const char* root_type);
 
    TTree* m_tree;
 
    std::list< void* > m_outputVarPointers;
 
    std::vector<BranchFilter> m_branchFilters;
 
    // Two member variables to manage the size of the tree branch baskets
    int m_basketSizePrimitive;
    int m_basketSizeVector;
  };
}
 
#endif