NtupleCnvSvc.cxx

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// NtupleCnvSvc.cxx 
// Implementation file for class Athena::NtupleCnvSvc
// Author: S.Binet<binet@cern.ch>
 
// AthenaRootComps includes
#include "NtupleCnvSvc.h"
#include "LeafCnv.h"
#include "RootBranchAddress.h"
 
// STL includes
 
// FrameWork includes
#include "GaudiKernel/GenericAddress.h"
#include "GaudiKernel/IRegistry.h"
#include "Gaudi/Property.h"
 
// ROOT includes
#include "TFile.h"
 
namespace Athena {
 
// Public methods: 
 
// Constructors
NtupleCnvSvc::NtupleCnvSvc( const std::string& name, 
                            ISvcLocator* pSvcLocator ) : 
  ::AthCnvSvc( name, pSvcLocator, ROOT_StorageType )
{
  //
  // Property declaration
  // 
  //declareProperty( "Property", m_nProperty );
 
  // std::cerr << "::NtupleCnvSvc(" << name << ")...\n";
}
 
// Destructor
NtupleCnvSvc::~NtupleCnvSvc()
{}
 
// Athena Service's Hooks
StatusCode
NtupleCnvSvc::initialize()
{
  ATH_MSG_DEBUG ("Initializing " << name() << "...");
 
  return AthCnvSvc::initialize();
}
 
StatusCode
NtupleCnvSvc::finalize()
{
  ATH_MSG_DEBUG ("Finalizing " << name() << "...");
 
  return AthCnvSvc::finalize();
}
 
// Query the interfaces.
//   Input: riid, Requested interface ID
//          ppvInterface, Pointer to requested interface
//   Return: StatusCode indicating SUCCESS or FAILURE.
// N.B. Don't forget to release the interface after use!!!
StatusCode 
NtupleCnvSvc::queryInterface(const InterfaceID& riid, void** ppvInterface) 
{
  if ( IConversionSvc::interfaceID().versionMatch(riid) ) {
    *ppvInterface = dynamic_cast<IConversionSvc*>(this);
  } else {
    // Interface is not directly available : try out a base class
    return ::AthCnvSvc::queryInterface(riid, ppvInterface);
  }
  addRef();
  return StatusCode::SUCCESS;
}
 
StatusCode 
NtupleCnvSvc::updateServiceState(IOpaqueAddress* pAddress)
{
  ATH_MSG_DEBUG("updateServiceState(" << pAddress << ")...");
  if (pAddress) {
    return dynamic_cast<GenericAddress*>(pAddress) 
      ? StatusCode::SUCCESS
      : StatusCode::FAILURE;
  }
  return StatusCode::FAILURE;
}
 
StatusCode 
NtupleCnvSvc::createObj(IOpaqueAddress* pAddress, DataObject*& refpObject)
{
  ATH_MSG_DEBUG("-->createObj(" << pAddress->par()[0] << "/" << pAddress->par()[1]
                << ", " << refpObject << ")...");
  GenericAddress *addr = dynamic_cast<GenericAddress*>(pAddress);
  if (!addr) {
    ATH_MSG_DEBUG(pAddress->par()[0] << "/" << pAddress->par()[1]
                  << " is NOT a GenericAddress!");
    return StatusCode::FAILURE;
  }
  ATH_MSG_DEBUG("  par:  " << pAddress->par()[0] << "/" << pAddress->par()[1]
                << endmsg <<
                "  ipar: " << *pAddress->ipar());
  return AthCnvSvc::createObj(pAddress, refpObject);
}
 
StatusCode 
NtupleCnvSvc::createRep(DataObject* pObject, IOpaqueAddress*& refpAddress)
{
  ATH_MSG_DEBUG("::createRep(" << refpAddress << ", " << pObject << ")...");
  //std::cerr << "::NCS::createRep(" << this->name() << ", " << pObject << ")...\n";
  if (0==pObject) {
    ATH_MSG_ERROR("::createRep: received null ptr to DataObject");
    return StatusCode::FAILURE;
  }
  return ::AthCnvSvc::createRep(pObject, refpAddress);
}
 
StatusCode 
NtupleCnvSvc::fillRepRefs(IOpaqueAddress* pAddress, DataObject* pObject)
{
  ATH_MSG_DEBUG("::fillRepRefs(" << pAddress << ", " << pObject << ")...");
  if (0==pAddress || 0==pObject) {
    ATH_MSG_ERROR("::fillRepRefs: null ptr received! (paddr=" << pAddress
                  << ", pobj=" << pObject << ")");
    return StatusCode::FAILURE;
  }
  return ::AthCnvSvc::fillRepRefs(pAddress, pObject);
}
 
StatusCode 
NtupleCnvSvc::connectOutput(const std::string& outputConnectionSpec,
                const std::string& openMode)
{
  ATH_MSG_DEBUG("::connectOutput(" 
                << outputConnectionSpec << ", " 
                << openMode << ")");
  // std::cerr << "::connectOutput(" << outputConnectionSpec << ", openmode="
  //       << openMode << ")...\n";
  TFile *f = TFile::Open(outputConnectionSpec.c_str(), "recreate");
  if (f == 0) {
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
StatusCode
NtupleCnvSvc::connectOutput(const std::string& outputConnectionSpec)
{
  ATH_MSG_DEBUG("::connectOutput(" 
                << outputConnectionSpec << ")");
  // std::cerr << "::connectOutput(" << outputConnectionSpec << ")...\n";
  TFile *f = TFile::Open(outputConnectionSpec.c_str(), "recreate");
  if (f == 0) {
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
StatusCode 
NtupleCnvSvc::commitOutput(const std::string& outputConnectionSpec,
                        bool doCommit)
{
  ATH_MSG_DEBUG("::commitOutput(" << outputConnectionSpec
               << ", " << doCommit << ")...");
  return StatusCode::SUCCESS;
}
 
 
 
 
StatusCode
NtupleCnvSvc::createAddress(long svcType,
                         const CLID& clid,
                         const std::string* par,
                         const unsigned long* ip,
                         IOpaqueAddress*& refpAddress)
{
  ATH_MSG_DEBUG("::createAddress(" << svcType << ", " << clid 
                << ", " << par << ", " << ip << ", " << refpAddress
                << ")...");
  return ::AthCnvSvc::createAddress(svcType, clid, par, ip, refpAddress);
}
 
StatusCode 
NtupleCnvSvc::convertAddress(const IOpaqueAddress* pAddress,
                          std::string& refAddress)
{
  ATH_MSG_DEBUG("::convertAddress(" << pAddress << ", " 
                << refAddress << ")...");
  if (pAddress) {
    refAddress = *pAddress->par();
    return StatusCode::SUCCESS;
  }
  ATH_MSG_ERROR("::convertAddress: null pointer to IOpaqueAddress !");
  return StatusCode::FAILURE;
}
 
StatusCode 
NtupleCnvSvc::createAddress(long svcType,
                            const CLID& clid,
                            const std::string& refAddress,
                            IOpaqueAddress*& refpAddress)
{
  ATH_MSG_DEBUG("::createAddress(" << svcType << ", " << clid 
               << ", " << refAddress << ", " << refpAddress
               << ")...");
  if (svcType != ROOT_StorageType) {
    ATH_MSG_ERROR("::createAddress: svcType != root-type");
    ATH_MSG_ERROR("    svcType   = [" << svcType << "]");
    ATH_MSG_ERROR("    root-type = [" << ROOT_StorageType << "]");
    return StatusCode::FAILURE;
  }
  refpAddress = new GenericAddress(ROOT_StorageType, clid, refAddress);
  return StatusCode::SUCCESS;
}
 
// Const methods: 
 
// Non-const methods: 
 
// Protected methods: 
 
IConverter* 
NtupleCnvSvc::createConverter(long typ, 
                              const CLID& clid, 
                              const ICnvFactory* fac)
{
  ATH_MSG_DEBUG("::createConverter(" << typ << ", " << clid
                << ", " << fac << ")...");
  IConverter* cnv = new Athena::LeafCnv(clid, serviceLocator());
  if (cnv) {
    return cnv;
  }
  return AthCnvSvc::createConverter(typ, clid, fac);
}
 
// Const methods: 
 
// Non-const methods: 
 
} //> end namespace Athena