RNTupleContainer.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
//====================================================================
//    Root Database Container RNTuple implementation
//--------------------------------------------------------------------
//    Author     : M.Nowak
//====================================================================
#ifndef POOL_RNTUPLECONTAINER_H
#define POOL_RNTUPLECONTAINER_H 1
 
// Framework include files
 
 
#include "StorageSvc/DbColumn.h"
#include "StorageSvc/DbContainerImp.h"
#include "StorageSvc/DbDatabase.h"
#include <vector>
#include <memory>
#include <string>
 
// Forward declarations
class TClass;
namespace SG { class IAuxStoreIO; }
namespace RootAuxDynIO { class IRootAuxDynReader; class IRNTupleAuxDynWriter; }
namespace RootStorageSvc { class RNTupleWriterHelper; }
namespace ROOT::Experimental { class RNTupleReader; }
 
#include "ROOT/RNTupleView.hxx"
/*
 * POOL namespace declaration
 */
namespace pool {
 
using ROOT::Experimental::RNTupleReader;
using ROOT::Experimental::RNTupleView;
 
// Forward declaration
class DbColumn;
class RootDatabase;
 
class RNTupleContainer : public DbContainerImp
{
  struct FieldDesc : public DbColumn
  {
    std::string fieldname;
#if ROOT_VERSION_CODE >= ROOT_VERSION( 6, 31, 0 )
    std::unique_ptr< RNTupleView<void, true> > view_p;  // pointer because lack of default xtor
#endif
    std::string sgkey;
    TClass*     clazz = nullptr;
    void*       object = nullptr;
 
    // if writing char* keep a copy here as an std::string
    std::string str;
 
    // ----  extra variables used for AuxDyn attributes
    // number of rows written to this branch so far
    size_t rows_written = 0;
 
    int aux_iostore_IFoffset = -1;
 
    // AuxDyn RNTuple reader (managed by the Database)
    std::unique_ptr<RootAuxDynIO::IRootAuxDynReader> auxdyn_reader;
 
    // AuxDyn RNTuple writer (managed by the Database)
    std::unique_ptr<RootAuxDynIO::IRNTupleAuxDynWriter> auxdyn_writer;
 
    FieldDesc(const DbColumn& c);
    FieldDesc(FieldDesc const& other) = delete;
    FieldDesc(FieldDesc&& other) = default;
    ~FieldDesc() = default;
 
    FieldDesc& operator=(FieldDesc const& other) = delete;
    FieldDesc& operator=(FieldDesc&& other) = default;
 
    const std::string typeName();
    bool hasAuxStore() { return aux_iostore_IFoffset >= 0; }
    SG::IAuxStoreIO* getIOStorePtr() {
      return (aux_iostore_IFoffset >= 0
                  ? reinterpret_cast<SG::IAuxStoreIO*>((char*)object +
                                                       aux_iostore_IFoffset)
                  : nullptr);
    }
  };
 
 protected:
   const DbTypeInfo*  m_type{};
   std::vector<FieldDesc>  m_fieldDescs;
   DbDatabase         m_dbH;
   RootDatabase*      m_rootDb;
   int                m_ioBytes;
   bool               m_isDirty;
 
   uint64_t           m_index;
   uint64_t           m_indexSize;
   int64_t            m_indexBump;
   const uint32_t     m_indexMulti;
 
   RootStorageSvc::RNTupleWriterHelper*     m_ntupleWriter = nullptr;
 
   RNTupleReader*       m_ntupleReader{};
 
 public:
  RNTupleContainer();
 
  virtual ~RNTupleContainer();
 
  virtual DbStatus close() override final;
 
  virtual DbStatus open(DbDatabase& dbH, const std::string& nam,
                        const DbTypeInfo* info,
                        DbAccessMode mod) override final;
 
 
  virtual DbStatus getOption(DbOption& opt) override final;
 
 
  virtual DbStatus setOption(const DbOption& opt) override final;
 
  virtual DbStatus isShapeSupported(const DbTypeInfo* typ) const override final;
 
  virtual uint64_t size() override final;
 
  const std::string& getName() const { return m_name; }
 
  bool isDirty() const { return m_isDirty; }
 
  void clearDirty() { m_isDirty = false; }
 
 
  virtual DbStatus loadObject(void** ptr, ShapeH shape,
                              Token::OID_t& oid) override final;
 
  virtual DbStatus writeObject(ActionList::value_type&) override final;
 
  virtual uint64_t nextRecordId() override final;
 
  virtual void useNextRecordId(uint64_t nextID) override final;
 
  virtual DbStatus select(DbSelect& criteria) override final;
 
  using DbContainerImp::fetch;
  virtual DbStatus fetch(DbSelect& sel) override final;
 
  virtual DbStatus transAct(Transaction::Action action) override final;
 
 private:
  DbStatus initObjectFieldDesc(FieldDesc& dsc);
};
 
}  // namespace pool
#endif