TreeHandler.cpp

Go to the documentation of this file.

#include "caf_util/TreeHandler.hpp"
#include <iostream>

namespace caf_util {
  
  TreeHandler::TreeHandler()
    : _tree(0),
      _tmb_tree(0)
  {
  }
  
  
  void TreeHandler::InitTree(std::string TreeName, TFile *input_file)
  {
    _toTree = TreeName;
    _tmb_tree = 0; _tree = 0;
    
    if( _tmb_tree = (TTree *)input_file->Get("TMBTree") ) {
      
      if ( TreeName=="TMBTree" ) _tree = _tmb_tree;
      else _tree = dynamic_cast<TTree *>(gROOT->Get(TreeName.c_str()));
      
      if ( _tree == 0 ) {
        gROOT->cd();
        _tree = new TTree(_toTree.c_str(), _toTree.c_str());
        _tmb_tree->AddFriend(_tree);
      }
    }
    else
      std::cerr << " In TreeHandler::InitTree[" << TreeName.c_str() 
                << "] -- No TMBTree in input file!\n";
    
  }
  
  void TreeHandler::AddBranch(std::string BranchName, std::string BranchObjType)
  {
    if ( _tree ) {
      _BranchContent[BranchName.c_str()] = new TClonesArray(BranchObjType.c_str());
      _Branches[BranchName.c_str()] = 
        _tree->Branch(BranchName.c_str(), &_BranchContent[BranchName.c_str()]);
    }
    else
      std::cerr << " In TreeHandler::AddBranch[" << BranchName.c_str() 
                << "] -- No output tree found. _tree needs to be initialized.\n";
  }

  void TreeHandler::AddBranch(std::string BranchName, int &intref, std::string BranchObjType)
  {
    if ( _tree ) {
      _Branches[BranchName.c_str()] =
        _tree->Branch(BranchName.c_str(), &intref, BranchObjType.c_str());
    }
    else
      std::cerr << " In TreeHandler::AddBranch[" << BranchName.c_str() 
                << "] -- No output tree found. _tree needs to be initialized.\n";
  }

  void TreeHandler::AddBranch(std::string BranchName, float &floatref, std::string BranchObjType)
  {
    if ( _tree ) {
      _Branches[BranchName.c_str()] =
        _tree->Branch(BranchName.c_str(), &floatref, BranchObjType.c_str());
    }
    else
      std::cerr << " In TreeHandler::AddBranch[" << BranchName.c_str() 
                << "] -- No output tree found. _tree needs to be initialized.\n";
  }

  void TreeHandler::AddBranch(std::string BranchName, double &dbleref, std::string BranchObjType)
  {
    if ( _tree ) {
      _Branches[BranchName.c_str()] =
        _tree->Branch(BranchName.c_str(), &dbleref, BranchObjType.c_str());
    }
    else
      std::cerr << " In TreeHandler::AddBranch[" << BranchName.c_str() 
                << "] -- No output tree found. _tree needs to be initialized.\n";
  }

  void TreeHandler::ClearIt() 
  {

    // Look for our tree in the memory
    // If it exist -  delete it!

    if ( _tree == (TTree *)gROOT->Get(_toTree.c_str()) ) {

      if ( _tmb_tree )
        _tmb_tree->RemoveFriend(_tree);
      _tree->Delete();
      _tree=0;
    }

    // Clear and remove all TClonesArrays :
    std::map<std::string,TClonesArray*>::iterator it;
    for( it = _BranchContent.begin(); it != _BranchContent.end(); it++ ) {
      if ( TClonesArray *array = it->second )
        array->Delete();
      else
        std::printf("No Array!\n");
    }

    _BranchContent.clear();
    _Branches.clear();
  }

}

---