---

PersSvc.cpp

Go to the documentation of this file.

#define RootHistCnv_PERSSVC_CPP


// Include files
#include <iostream>
#include <string>
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/SvcFactory.h"
#include "GaudiKernel/CnvFactory.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/IDataSelector.h"
#include "GaudiKernel/IConverter.h"
#include "GaudiKernel/IHistogramSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "PersSvc.h"

#include "TROOT.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
static SvcFactory<RootHistCnv::PersSvc> s_factory;
const ISvcFactory& RootHistPersSvcFactory = s_factory;
 

namespace RootHistCnv {

  static std::string stat_dir = "/stat";
  static std::string undefFileName = "UndefinedROOTOutputFileName";


  StatusCode PersSvc::initialize()     {
    StatusCode status = ConversionSvc::initialize();

    MsgStream log( messageService(), name() );
//      log << MSG::INFO << "initialize()" << endreq;

    if ( status.isSuccess() ) {
      // Set histogram data store
      IHistogramSvc *pIHS = 0;
      status = serviceLocator()->getService("HistogramDataSvc", IID_IHistogramSvc, (IInterface*&)pIHS);
      if ( status.isSuccess() )   {
        status = setStore ( pIHS );
      }
      else    {
        return status;
      }
      // Get my properties from the JobOptionsSvc
      status = setProperties();

      // Protect against multiple instances of TROOT
      if ( 0 == gROOT )   {
        static TROOT root("root","ROOT I/O");
//      log << MSG::INFO << "Initializing ROOT" << endreq;
        //      gDebug = 99;
      } else {
        log << MSG::INFO << "ROOT already initialized" << endreq;
      }

      // Initialize ROOT if output file name is defined
      if( undefFileName != m_defFileName ) {
        hfile = new TFile(m_defFileName.c_str(),"RECREATE","GAUDI Histograms");
      }
      log << MSG::INFO << "Writing ROOT histograms to: " << m_defFileName 
          << endreq;
    }
    return status;
  }


  StatusCode 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 ) {
      //      gDirectory->cd( hfile->GetPath() );
      hfile->cd();
      hfile->Write();
      hfile->Close();
    }
    return ConversionSvc::finalize();
  }


  StatusCode PersSvc::createReps(IDataSelector* pSelector)            {
    // Convert histograms only if HBOOK output file name is defined
    //   and number of entries in the histogram data store is not equal 1
    //     (1 means, that the HDS is "empty", there is only
    //     the root directory "/stat", and no histograms)
    MsgStream log( messageService(), name() );
//    log << MSG::DEBUG << "PersSvc::createReps()  nObj:" << pSelector->size()
//      << endreq;
    if( 1 != pSelector->size() ) {
      // There are objects in the HDS to be stored

      if( undefFileName != m_defFileName ) {

        TDirectory *gDir;
        gDir = gDirectory;
        hfile->cd("/");

        // Valid output file name
        StatusCode sc = ConversionSvc::createReps(pSelector);
//      log << MSG::DEBUG << "called createReps on conversion svc: " 
//          << sc << endreq;
        if( sc.isSuccess() ) {
          log << MSG::INFO << "ALL Histograms converted successfully" << endreq;

        } else {
          log << MSG::ERROR << "Error while creating Histogram persistent representations" << endreq;
        }
        gDirectory = gDir;
        return sc;
      }
      else {
        log << MSG::ERROR << "ROOT output file name missing, Histograms cannot be converted" << endreq;
      }
    }
    // Nothing to be converted, there are no histograms in the HDS
    return StatusCode::SUCCESS;
  }


  PersSvc::PersSvc(const std::string& name, ISvcLocator* svc)
              : ConversionSvc(name, svc, ROOT_StorageType)           {
    declareProperty("OutputFile", m_defFileName = undefFileName);
  }


  PersSvc::~PersSvc() {
    if ( hfile != 0 ) {
      delete hfile;
      hfile = 0;
    }
  }

}    // namespace RootHistCnv

---