RootCnvSvc.h

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2019 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
//====================================================================
//  RootCnvSvc definition
//--------------------------------------------------------------------
//
//  Author     : M.Frank
//====================================================================
#ifndef GAUDIROOTCNV_GAUDIROOTCNVSVC_H
#define GAUDIROOTCNV_GAUDIROOTCNVSVC_H
 
// Framework include files
#include "GaudiKernel/ClassID.h"
#include "GaudiKernel/ConversionSvc.h"
#include "GaudiKernel/DataObject.h"
#include "GaudiUtils/IIODataManager.h"
 
// C++ include files
#include <set>
 
// Forward declarations
struct IDataManagerSvc;
class IIncidentSvc;
class TClass;
 
// ROOT include files
#include "TBranch.h"
#include "TClass.h"
#include "TROOT.h"
#include "TTree.h"
 
//#include "RootCnv/RootPerfMonSvc.h"
 
/*
 * Gaudi namespace declaration
 */
namespace Gaudi {
 
  // Forward declarations
  class IDataConnection;
  class RootDataConnection;
  class RootConnectionSetup;
 
  class GAUDI_API RootCnvSvc : public ConversionSvc {
  protected:
    Gaudi::Property<std::string> m_ioPerfStats{this, "IOPerfStats", "",
                                               "Enable TTree IOperfStats if not empty; otherwise perf stat file name"};
    Gaudi::Property<std::string> m_shareFiles{this, "ShareFiles", "NO",
                                              "Share files? If set to YES,  files will not be closed on finalize"};
    Gaudi::Property<bool> m_incidentEnabled{this, "EnableIncident", true, "Flag to enable incidents on FILE_OPEN"};
    Gaudi::Property<std::string> m_recordName{this, "RecordsName", "/FileRecords",
                                              "Records name to fire incident for file records"};
 
    // ROOT Write parameters
    Gaudi::Property<int> m_autoFlush{this, "AutoFlush", 100,
                                     "AutoFlush parameter for ROOT TTree (Number of events between auto flushes)"};
    Gaudi::Property<int> m_basketSize{this, "BasketSize", 2 * 1024 * 1024 /*MBYTE*/,
                                      "Basket optimization parameter for ROOT TTree (total basket size)"};
    Gaudi::Property<int> m_bufferSize{this, "BufferSize", 2 * 1024 /*kBYTE*/,
                                      "Buffer size optimization parameter for ROOT TTree"};
    Gaudi::Property<int> m_splitLevel{this, "SplitLevel", 0, "Split level optimization parameter for ROOT TTree"};
    Gaudi::Property<std::string> m_compression{this, "GlobalCompression", "",
                                               "Compression-algorithm:compression-level,  empty: do nothing"};
 
    SmartIF<Gaudi::IIODataManager> m_ioMgr;
    SmartIF<IIncidentSvc> m_incidentSvc;
    Gaudi::RootDataConnection* m_current = nullptr;
    TClass* m_classRefs = nullptr;
    TClass* m_classDO = nullptr;
    std::shared_ptr<RootConnectionSetup> m_setup;
    std::string m_currSection;
 
    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 );
 
    StatusCode updateServiceState( IOpaqueAddress* /* pAddress */ ) override { return StatusCode::SUCCESS; }
 
    StatusCode error( const std::string& msg );
 
    StatusCode connectDatabase( const std::string& dataset, int mode, RootDataConnection** con );
 
    StatusCode initialize() override;
 
    StatusCode finalize() override;
 
    IConverter* createConverter( long typ, const CLID& wanted, const ICnvFactory* fac ) override;
 
    void loadConverter( DataObject* pObj ) override;
 
    StatusCode connectOutput( const std::string& outputFile, const std::string& openMode ) override;
 
    StatusCode connectOutput( const std::string& outputFile ) override;
 
    StatusCode commitOutput( const std::string& outputFile, bool do_commit ) override;
 
    virtual StatusCode disconnect( const std::string& dbName );
 
    StatusCode createAddress( long svc_type, const CLID& clid, const std::string* par, const unsigned long* ip,
                              IOpaqueAddress*& refpAddress ) override;
 
    StatusCode createAddress( long svc_type, const CLID& clid, const std::string& refAddress,
                              IOpaqueAddress*& refpAddress ) override {
      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 );
  };
} // namespace Gaudi
 
#endif // GAUDIROOTCNV_GAUDIROOTCNVSVC_H