In This Package:

PersSvc.cpp

Go to the documentation of this file.

// $Id: PersSvc.cpp,v 1.15 2007/03/15 15:53:15 hmd Exp $

// Include files
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/SvcFactory.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/IRegistry.h"
#include "GaudiKernel/SmartDataPtr.h"
#include "RootObjAddress.h"
#include "RConverter.h"
#include "PersSvc.h"
#include "TFile.h"

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

// Instantiation of a static factory class used by clients to create
// instances of this service
DECLARE_NAMESPACE_SERVICE_FACTORY(RootHistCnv,PersSvc)

namespace RootHistCnv {
  static std::string stat_dir = "/stat";
  static std::string undefFileName = "UndefinedROOTOutputFileName";
};

//-----------------------------------------------------------------------------
StatusCode RootHistCnv::PersSvc::initialize()
//-----------------------------------------------------------------------------
{
  MsgStream log( messageService(), name() );

  StatusCode status = ConversionSvc::initialize();
  if( status.isFailure() ) return status;

  // Get my properties from the JobOptionsSvc
  if (setProperties().isFailure()) {
    log << MSG::ERROR << "Could not set my properties" << endreq;
    return StatusCode::FAILURE;
  }
  // Initialize ROOT if output file name is defined
  if( undefFileName != m_defFileName ) {
    m_hfile = TFile::Open(m_defFileName.c_str(),"RECREATE","GAUDI Histograms");
  } else {
    m_hfile = 0;
  }
  log << MSG::INFO << "Writing ROOT histograms to: " << m_defFileName 
      << endreq;

  return StatusCode(StatusCode::SUCCESS,true);
}


//-----------------------------------------------------------------------------
StatusCode RootHistCnv::PersSvc::finalize()
//-----------------------------------------------------------------------------
{
  // Close ROOT only if the output file name is defined
  MsgStream log( messageService(), name() );
  log << MSG::DEBUG << "RootHistCnv::PersSvc::finalize()" << endreq;
  if( undefFileName != m_defFileName ) {
    m_hfile->Write("",TObject::kOverwrite);
    m_hfile->Close();
  }
  return ConversionSvc::finalize();
}

//-----------------------------------------------------------------------------
StatusCode RootHistCnv::PersSvc::createRep(DataObject* pObject, 
                                           IOpaqueAddress*& refpAddress)
//-----------------------------------------------------------------------------
{
  // There are objects in the HDS to be stored
  if( undefFileName != m_defFileName )  {
    SmartDataPtr<DataObject> top(dataProvider(), stat_dir);
    if ( 0 != top )    {
      IRegistry* pReg = top->registry();
      if ( pReg )   {
        if ( top.ptr() == pObject )   {
          TDirectory* pDir = m_hfile;
          refpAddress = new RootObjAddress( repSvcType(), 
                                            CLID_DataObject,
                                            stat_dir, 
                                            m_defFileName,
                                            long(pDir),
                                            long(0));
          return StatusCode::SUCCESS;
        }
        else    {
          StatusCode sc = ConversionSvc::createRep(pObject, refpAddress);
          if( sc.isFailure() )   {
            MsgStream log( messageService(), name() );
            log << MSG::ERROR
                << "Error while creating persistent Histogram:"
                << pReg->identifier()
                << endreq;
          }
          return sc;
        }
      }
    }
    MsgStream err( messageService(), name() );
    err << MSG::ERROR
        << "Internal error while creating Histogram persistent representations" 
        << endreq;
    return StatusCode::FAILURE;
  } else {
    if (!m_prtWar) {
      m_prtWar = true;
      MsgStream log( messageService(), name() );
      log << MSG::WARNING
          << "no ROOT output file name, "
          << "Histograms cannot be persistified" << endreq;
    }
  }
  return StatusCode::SUCCESS;
}

//-----------------------------------------------------------------------------
RootHistCnv::PersSvc::PersSvc(const std::string& name, ISvcLocator* svc)
//-----------------------------------------------------------------------------
: ConversionSvc(name, svc, ROOT_StorageType), m_prtWar(false) {
  declareProperty("OutputFile", m_defFileName = undefFileName);
  declareProperty("ForceAlphaIds", m_alphaIds = false);
}

//-----------------------------------------------------------------------------
RootHistCnv::PersSvc::~PersSvc()
//-----------------------------------------------------------------------------
{
  if ( m_hfile != 0 ) {
    delete m_hfile;
    m_hfile = 0;
  }
}

---