---

RRWNTupleCnv.cpp

Go to the documentation of this file.

#define ROOTHISTCNV_RRWNTUPLECNV_CPP

#define ALLOW_ALL_TYPES

// Compiler include files
#include <cstdio>
#include <map>
#include <string>

// Include files
#include "GaudiKernel/xtoa.h"
#include "GaudiKernel/CnvFactory.h"
#include "GaudiKernel/GenericAddress.h"
#include "GaudiKernel/INTupleSvc.h"

#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/NTuple.h"

#include "RRWNTupleCnv.h"

#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"

//------------------------------------------------------------------------------
//
// Implementation of class :  RootHistCnv::RRWNTupleCnv
//
// Author :                   Charles Leggett
//
//------------------------------------------------------------------------------


// Instantiation of a static factory class used by clients to create
// instances of this service
static CnvFactory<RootHistCnv::RRWNTupleCnv> s_factory;
const ICnvFactory& RootHistRRWNTupleCnvFactory = s_factory;


StatusCode RootHistCnv::RRWNTupleCnv::book(long /* idh */, const std::string& /* loc */, INTuple* nt)  {
  MsgStream log(msgSvc(), "RRWNTupleCnv");
//    log << MSG::INFO << "book(idh, loc, ntup)" << endreq;

  std::string rt_path, rt_fid, rt_id, rt_lid;
  TTree* rtree;

  DataObject* pObj = dynamic_cast<DataObject*>(nt);
  if (pObj != 0) {
    rt_lid = pObj->localPath();
    rt_id = "t" + rt_lid.substr(1,rt_lid.length()-1);
    rt_fid = pObj->fullpath();
    rt_path = pObj->location();
  } else {
    log << MSG::ERROR << "dynamic cast failed" << endreq;
    return StatusCode::FAILURE;
  }
  
  if ( findNTuple(rt_fid, rtree).isFailure() ) {

    const INTuple::ItemContainer& items = nt->items();


    // Book the tree
    //    rt_id = _itoa(idh, text, 10);

    rtree = new TTree(rt_id.c_str(),nt->title().c_str());
//      log << MSG::DEBUG << "created tree id: " << rtree->GetName() 
//      << "  title: "<< nt->title() << endreq;

    // Put it in the treemap
    regNTuple(rt_fid,rtree);
//      log << MSG::DEBUG << "added to treemap: " << rt_fid << endreq;

    // Add the branches
    for (size_t length = items.size(), i = 0; i < length; i++ )    {
      std::string item_name = items[i]->name();
      std::string tag_name = item_name;
//        log << MSG::DEBUG << "item: " << i << "  name: " << items[i]->name()
//        << " typeName: " << items[i]->typeName() << endreq;

      tag_name += rootVarType( items[i]->type() );


      // add the branch
      log << MSG::INFO << "ID " << rt_id << ": added branch: " 
          << item_name << " tag: "
          << tag_name << endreq;
      rtree->Branch(item_name.c_str(), const_cast<void*>(items[i]->buffer()), 
                    tag_name.c_str());

    }

    log << MSG::INFO << "Booked TTree with ID:" << rt_id
        << " \"" << nt->title() << "\" in directory " << rt_path << endreq;

    // Fill it
    //      rtree->Fill();
    return StatusCode::SUCCESS;
  }
  log << MSG::ERROR << "ROOT TTree " << rt_fid  << " already exists." 
      << endreq;
  return StatusCode::FAILURE;
}


StatusCode RootHistCnv::RRWNTupleCnv::writeData(long /* idh */, INTuple* nt)   {

  MsgStream log(msgSvc(), "RRWNTupleCnv");
//    log << MSG::DEBUG << "writeData()" << endreq;

  std::string rt_lid, rt_id, rt_fid, rt_path;
  TTree* rtree;

  DataObject* pObj = dynamic_cast<DataObject*>(nt);
  if (pObj != 0) {
    rt_lid = pObj->localPath();
    rt_id = "t" + rt_lid.substr(1,rt_lid.length()-1);
    rt_fid = pObj->fullpath();
    rt_path = pObj->location();
  } else {
    log << MSG::ERROR << "dynamic cast failed" << endreq;
    return StatusCode::FAILURE;
  }

  if ( findNTuple(rt_fid,rtree).isFailure() ) {
    log << MSG::ERROR << "tree ID: " << rt_fid << " not found in treemap"
        << endreq;
    return StatusCode::FAILURE;
  }
  
  // Fill the tree;
  rtree->Fill();

  // Reset the NTuple
  nt->reset();


  return StatusCode::SUCCESS;
}


StatusCode RootHistCnv::RRWNTupleCnv::readData(long /* id */, INTuple* nt, long /* ievt */)    {
  
  MsgStream log(msgSvc(), "RRWNTupleCnv");
//    log << MSG::INFO << "readData()" << endreq;

  std::string rt_lid, rt_id, rt_fid, rt_path;

  DataObject* pObj = dynamic_cast<DataObject*>(nt);
  if (pObj != 0) {
    rt_lid = pObj->localPath();
    rt_id = "t" + rt_lid.substr(1,rt_lid.length()-1);
    rt_fid = pObj->fullpath();
    rt_path = pObj->location();
  } else {
    log << MSG::ERROR << "dynamic cast failed" << endreq;
    return StatusCode::FAILURE;
  }


  return StatusCode::FAILURE;

}


StatusCode RootHistCnv::RRWNTupleCnv::load(long /* idh */, INTuple*& /* refpObject */)   {
  MsgStream log(msgSvc(), "RRWNTupleCnv");
//    log << MSG::INFO << "load()" << endreq;


  return StatusCode::FAILURE;

}

---