TreeHandler.hpp

Go to the documentation of this file.

#ifndef TreeHandler_HPP_
#define TreeHandler_HPP_

#include <map>
#include <string>
#include <stdexcept>

#include "cafe/Event.hpp"
#include "cafe/Collection.hpp"

#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TBranchElement.h"
#include "TDirectory.h"
#include "TClonesArray.h"

namespace caf_util {

  class TreeHandler : public TObject {
    
  private:
    
    TTree *_tree;
    TTree *_tmb_tree;
    std::map<std::string,TClonesArray*> _BranchContent;
    std::map<std::string,TBranch*> _Branches;
    std::string _toTree;
    
  public:
    
    TreeHandler();
    
    inline TTree* GetTree() { return _tree; }
    inline TBranch* GetBranch(std::string BranchName) { return _Branches[BranchName.c_str()]; }
    inline TClonesArray* GetBranchArray(std::string BranchName) { return _BranchContent[BranchName.c_str()]; }
    
    void InitTree(std::string TreeName, TFile *input_file);

    void AddBranch(std::string BranchName, std::string BranchObjType);
    void AddBranch(std::string BranchName, int &intref, std::string BranchObjType);
    void AddBranch(std::string BranchName, float &floatref, std::string BranchObjType);
    void AddBranch(std::string BranchName, double &dbleref, std::string BranchObjType);
    
    // To be called from "processEvent"    
    template <class T> 
    void AddBranch(std::string BranchName, std::string BranchObjType, T** obj) {
      if ( _tree )
        _Branches[BranchName.c_str()] =
          _tree->Branch(BranchName.c_str(), BranchObjType.c_str(), obj);
      else
        std::cerr << " In TreeHandler::AddBranch[" << BranchName.c_str()
                  << "] -- No output tree found. _tree needs to be initialized.\n";
    }

    template<class T>
    void FillBranch(std::string BranchName, cafe::Collection<T> result )
    {
      if ( TClonesArray *array = GetBranchArray(BranchName.c_str()) ) {
        array->Delete();
        
        int next = 0;
        for( typename cafe::Collection<T>::iterator it = result.begin(); it != result.end(); ++it )
          new ((*array)[next++]) T(*it);
      }
    }

    // To be called from "inputFileClosing"
    void ClearIt();

    ClassDef(TreeHandler,1);

  };
  
}

#endif

---