RootCnvSvc.h

Go to the documentation of this file.

//====================================================================
//  RootCnvSvc definition
//--------------------------------------------------------------------
//
//  Author     : M.Frank
//====================================================================
#ifndef GAUDIROOTCNV_GAUDIROOTCNVSVC_H
#define GAUDIROOTCNV_GAUDIROOTCNVSVC_H

// Framework include files
#include "GaudiKernel/ConversionSvc.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/ClassID.h"

// C++ include files
#include <set>

// Forward declarations
class IDataManagerSvc;
class IIncidentSvc;
class TClass;

// ROOT include files
#include "TROOT.h"
#include "TClass.h"
#include "TTree.h"
#include "TBranch.h"

//#include "RootCnv/RootPerfMonSvc.h"

/*
 * Gaudi namespace declaration
 */
namespace Gaudi {

  // Forward declarations
  class IIODataManager;
  class IDataConnection;
  class RootDataConnection;
  class RootConnectionSetup;

  class GAUDI_API RootCnvSvc : public ConversionSvc  {
  protected:

    SmartIF<Gaudi::IIODataManager>      m_ioMgr;
    SmartIF<IIncidentSvc>               m_incidentSvc;
    Gaudi::RootDataConnection*  m_current = nullptr;
    TClass*                     m_classRefs;
    TClass*                     m_classDO;
    std::shared_ptr<RootConnectionSetup> m_setup;
    std::string                 m_currSection;

    bool                        m_incidentEnabled;
    std::string                 m_shareFiles;
    std::string                 m_recordName;
    std::string                 m_ioPerfStats;
    std::string                 m_compression;

    int                         m_autoFlush;
    int                         m_basketSize;
    int                     m_bufferSize;
    int                     m_splitLevel;

    std::set<std::string>       m_badFiles;

    std::unique_ptr<MsgStream>  m_log;

    TClass* getClass(DataObject* pObject);
    MsgStream& log() const {   return *m_log; }

  public:

    RootCnvSvc(const std::string& name, ISvcLocator* svc);

    virtual ~RootCnvSvc();

    virtual StatusCode updateServiceState(IOpaqueAddress* /* pAddress */)
    { return StatusCode::SUCCESS;                  }

    StatusCode error(const std::string& msg);

    StatusCode connectDatabase(const std::string& dataset, int mode, RootDataConnection** con);

    virtual StatusCode initialize();

    virtual StatusCode finalize();

    virtual IConverter* createConverter(long typ,
                    const CLID& wanted,
                    const ICnvFactory* fac);

    virtual void loadConverter(DataObject* pObj);

    virtual StatusCode connectOutput(const std::string& outputFile,
                     const std::string& openMode);

    virtual StatusCode connectOutput(const std::string& outputFile);

    virtual StatusCode commitOutput(const std::string& outputFile,
                    bool               do_commit);

    virtual StatusCode disconnect(const std::string& dbName);

    virtual StatusCode createAddress( long                 svc_type,
                      const CLID&          clid,
                      const std::string*   par,
                      const unsigned long* ip,
                      IOpaqueAddress*&     refpAddress);

    virtual StatusCode createAddress( long svc_type,
                      const CLID& clid,
                      const std::string& refAddress,
                      IOpaqueAddress*& refpAddress) {
      return this->ConversionSvc::createAddress(svc_type,clid,refAddress,refpAddress);
    }

    virtual StatusCode createNullRep(const std::string& path);

    virtual StatusCode createNullRef(const std::string& path);

    virtual StatusCode i__createRep(DataObject* pObj, IOpaqueAddress*& refpAddr);


    virtual StatusCode i__fillRepRefs(IOpaqueAddress* pAddr,DataObject* pObj);

    virtual StatusCode i__createObj(IOpaqueAddress* pAddr, DataObject*& refpObj);

    virtual StatusCode i__fillObjRefs(IOpaqueAddress* pAddr, DataObject* pObj);
  };
}

#endif  // GAUDIROOTCNV_GAUDIROOTCNVSVC_H