RootBranchAddress.cxx

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// RootBranchAddress.cxx 
// Implementation file for class Athena::RootBranchAddress
// Author: S.Binet<binet@cern.ch>
 
// AthenaRootComps includes
#include "RootBranchAddress.h"
 
// STL includes
 
// fwk includes
#include "GaudiKernel/System.h"
//#include "GaudiKernel/ClassID.h"
 
// ROOT includes
#include "TBranch.h"
#include "TTree.h"
#include "TLeaf.h"
 
namespace Athena {
 
// Public methods: 
 
// Constructors
 
RootBranchAddress::RootBranchAddress() :
  GenericAddress(),
  m_ptr    (),
  m_branch (NULL)
{}
 
RootBranchAddress::RootBranchAddress( const RootBranchAddress& rhs ) :
  GenericAddress(rhs),
  m_ptr    (rhs.m_ptr),
  m_branch (rhs.m_branch)
{}
 
RootBranchAddress& 
RootBranchAddress::operator=( const RootBranchAddress& rhs )
{
  if (this != &rhs) {
    GenericAddress::operator=(rhs);
    m_ptr    = rhs.m_ptr;
    m_branch = rhs.m_branch;
  }
  return *this;
}
 
RootBranchAddress::RootBranchAddress(long svc,
                                     const CLID& clid,
                                     const std::string& p1,
                                     const std::string& p2,
                                     unsigned long ip1,
                                     unsigned long ip2) :
  GenericAddress(svc, clid, p1, p2, ip1, ip2),
  m_ptr    (),
  m_branch (NULL)
{
  // std::cerr << "::RBA::+RBA... (this=" << this << ") br=[" 
  //           << this->par()[1] << "]\n";
}
 
RootBranchAddress::~RootBranchAddress()
{
  // std::cerr << "::RBA::~RBA... (this=" << this << ") br=["
  //           << this->par()[1] << "]\n";
}
 
 
// Const methods: 
 
// Non-const methods: 
 
void
RootBranchAddress::setBranchAddress(const RootType& rflx_type) 
{
  const std::string& br_name = this->par()[1];
  //const unsigned long* ipar = this->ipar();
  TTree *tree = this->ttree();
  Long64_t ientry = tree->GetReadEntry();
 
  // std::cerr << "::RBA::setBA(" << this << ") -- tree_nbr=" << tree_nbr << "|"
  //           << " ientry=" << ientry
  //           << "\tname=[" << br_name << "]"
  //           << " addr=" << m_ptr
  //           << "\n";
  if (m_branch && m_ptr != 0) {
    m_branch->GetEntry(ientry);
    return;
  }
 
  // activate branch...
  tree->SetBranchStatus(br_name.c_str(), 1);
 
  // --- inspired from pyroot/src/Pythonize.cxx ---
  // search for branch first (typical for objects)
  TBranch *branch = tree->GetBranch(br_name.c_str());
  if (!branch) {
    //ATH_MSG_INFO("-- branch [" << br_name << "] =0 -> trying a subbranch");
    // for benefit of naming of sub-branches, 
    // the actual name may have a trailing '.'
    branch = tree->GetBranch((br_name+".").c_str());
  }
 
  m_branch = branch;
  if (branch) {
    Long64_t bentry = branch->GetReadEntry();
    if ( bentry != ientry || ientry < 0) {
      // std::cerr << "::RBA:: -- ientry=" << ientry << " bentry="
      //           << bentry 
      //           << "\tname=[" << br_name << "]"
      //           << "\n";
      /*Long64_t nnn =*/ branch->GetEntry(ientry);
      bentry = branch->GetReadEntry();
      // std::cerr << "::RBA:: -- ientry=" << ientry << " bentry="
      //           << bentry 
      //           << "\tname=[" << br_name << "]"
      //           << " --> " << nnn
      //           << "\n";
 
    }
 
    //ATH_MSG_INFO("-- branch [" << br_name << "] = OK");
    // found a branched object, wrap its address for the object it represents
    TClass *cls = TClass::GetClass(branch->GetClassName());
    if (cls) {
      if (!branch->GetAddress()) {
        branch->GetEntry(ientry);
      }
      if (branch->GetAddress()) {
        //ATH_MSG_INFO("-- branch [" << br_name << "] cls: [" << cls->GetName() << "]");
        void* addr = *(char**)branch->GetAddress();
        RootType t = RootType(*cls->GetTypeInfo());
        if (t != rflx_type && !t.Id()) {
            t = rflx_type;
        }
        m_type = t;
        m_ptr = addr;
        return;
 
      } else {
        //ATH_MSG_INFO("-- branch [" << br_name << "] cls: [" << cls << "]");
      }
    } else {
      //ATH_MSG_INFO("-- branch [" << br_name << "] cls: [" << cls << "]");
    }
 
  }
 
  // if not, try a leaf
  TLeaf *leaf = tree->GetLeaf(br_name.c_str());
  //ATH_MSG_INFO("-- leaf [" << br_name << "] ??");
  if (branch && !leaf) {
    leaf = branch->GetLeaf(br_name.c_str());
    if (!leaf) {
      TObjArray *leaves = branch->GetListOfLeaves();
      if (leaves->GetSize() && (leaves->First() == leaves->Last())) {
        // i.e., if unambiguously only this one
        leaf = (TLeaf*)leaves->At(0);
      }
    }
  }
 
  if (leaf) {
    //ATH_MSG_INFO("-- leaf [" << br_name << "] = OK");
    // found a leaf, extract value and wrap
    if ( 1 < leaf->GetLenStatic() || leaf->GetLeafCount() ) {
      // array types --- FIXME
      //ATH_MSG_INFO("-- leaf [" << br_name << "] => array...");
      std::string tname = leaf->GetTypeName();
      void *addr = (void*)leaf->GetValuePointer();
      RootType t = RootType(tname+"*");
      if (t != rflx_type && !t.Id()) {
        t = rflx_type;
      }
      m_type = t;
      m_ptr = &addr;
      return;
 
    } else {
      // value types
      std::string tname = leaf->GetTypeName();
      // std::cerr << "-- leaf [" << br_name << "] => value-type("
      //           << tname << ")...\n";
      void *addr = (void*)leaf->GetValuePointer();
      RootType t = RootType(tname);
      if (t != rflx_type && !t.Id()) {
        t = rflx_type;
      }
      m_type = t;
      m_ptr = addr;
      // std::cerr << "-->ptr: [" << m_type.Name() << "]\n";
    }
  }
 
}
 
TTree*
RootBranchAddress::ttree()
{
  return reinterpret_cast<TTree*>(reinterpret_cast<unsigned long*>(this->ipar()[0]));  
}
 
// Protected methods: 
 
} //> end namespace Athena