RNTCollectionSchemaEditor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef RNTCOLLECTION_COLLECTIONSCHEMAEDITOR_H
#define RNTCOLLECTION_COLLECTIONSCHEMAEDITOR_H
 
#include <string>
#include <vector>
#include <typeinfo>
 
#include "CollectionBase/CollectionDescription.h"
#include "CollectionBase/ICollectionSchemaEditor.h"
 
#include "CoralBase/MessageStream.h"
 
 
namespace ROOT::Experimental {
   class RNTupleReader;
}
using RNTupleReader = ROOT::Experimental::RNTupleReader;
 
 
namespace pool {
   class ICollectionColumn;
 
   namespace RootCollection {
 
      class RNTCollection;
 
      class RNTCollectionSchemaEditor : public ICollectionSchemaEditor
      {
 
      friend class RNTCollection;
 
      public:
 
         RNTCollectionSchemaEditor( RNTCollection& collection,
                                    CollectionDescription& description,
                                    RNTupleReader& reader  );
 
 
         virtual void setEventReferenceColumnName( const std::string& columnName );
 
         virtual const ICollectionColumn&    insertColumn(
            const std::string& columnName,
            const std::string& columnType,
            const std::string& annotation = "",
            int maxSize = 0,
            bool sizeIsFixed = true );
 
         virtual const ICollectionColumn&   insertColumn(
            const std::string& columnName,
            const std::type_info& columnType,
            const std::string& annotation = "",
            int maxSize = 0,
            bool sizeIsFixed = true );
 
         virtual const ICollectionColumn&    insertTokenColumn(
            const std::string& columnName,
            const std::string& annotation = "" );
 
 
         virtual  const ICollectionColumn&    annotateColumn(
            const std::string& columnName,
            const std::string& annotation );
 
         virtual void dropColumn( const std::string& columnName );
 
         virtual void renameColumn( const std::string& oldName, const std::string& newName );
 
         void        readSchema();
 
         virtual ~RNTCollectionSchemaEditor();
 
      protected:
         void        addRNTupleField( const std::string& name, const std::string& type_name );
         void        createRNTuple();
 
         RNTCollection                  &m_collection;
 
         CollectionDescription          &m_description;
 
         RNTupleReader                  &m_reader;
 
         coral::MessageStream           m_poolOut;
      };
   }
}
#endif