RootCollection.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ROOTCOLLECTION_ROOTCOLLECTION_H
#define ROOTCOLLECTION_ROOTCOLLECTION_H
 
#include "CollectionSvc/ICollection.h"
#include "CollectionSvc/CollectionDescription.h"
#include "CollectionSvc/CollectionRowBuffer.h"
 
#include "POOLCore/DbPrint.h"
 
#include "GaudiKernel/IFileMgr.h"
#include "GaudiKernel/SmartIF.h"
#include "Gaudi/PluginService.h"
 
#include <string>
#include <memory>
 
class TTree;
class TFile;
class IFileMgr;
namespace ROOT {
   class RNTupleModel;
   class RNTupleWriter;
   class RNTupleReader;
}
 
namespace pool {
 
   class ISession;
 
   namespace RootCollection {
 
      class Attribute;
 
      // Create and Overwrite is only option, we'll never update
      constexpr const char* const poolOptToRootOpt[] = {"UPDATE", "READ"};
      // Io flags are not constexpr
      inline const Io::IoFlags poolOptToFileMgrOpt[] = { Io::WRITE|Io::APPEND, Io::READ };

      class RootCollection :  public ICollection, public APRMessaging {
 
     public:
    typedef Gaudi::PluginService::Factory<ICollection*( const CollectionDescription*, ICollection::OpenMode, ISession*)> Factory;

        RootCollection(  const pool::CollectionDescription* description,
                         pool::ICollection::OpenMode mode,
                         pool::ISession* );

        ~RootCollection();
 

        virtual void open() final override;

        virtual void insertRow( const pool::CollectionRowBuffer& inputRowBuffer ) final override;

        virtual void commit( bool restartTransaction = false ) final override;

        virtual void close() final override;

        virtual const CollectionDescription& description() const final override;

        virtual ICollectionCursor& cursor() final override;
 
     private:

        RootCollection(const RootCollection &) = delete;
        RootCollection& operator = (const RootCollection &) = delete;
 
        void setupTree() const;
        virtual void addTreeBranch( const std::string& name, const std::string& type_name );
    void addField(ROOT::RNTupleModel* model, const std::string& field_name, const std::string& field_type);
 
        void cleanup();
 
        CollectionDescription                m_description;
 
        std::string                          m_name;
        std::string                          m_fileName;
        ICollection::OpenMode                m_mode;
        TFile*                               m_file;
        TTree*                               m_tree;
    std::unique_ptr<ROOT::RNTupleReader> m_reader;
    std::unique_ptr<ROOT::RNTupleWriter> m_rntupleWriter;
 
        ISession*                            m_session;
        bool                                 m_open;
 
        SmartIF<IFileMgr>                    m_fileMgr;
      };
   }
}
#endif