RootCollectionSchemaEditor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ROOTCOLLECTION_COLLECTIONSCHEMAEDITOR_H
#define ROOTCOLLECTION_COLLECTIONSCHEMAEDITOR_H
 
#include <string>
#include <vector>
#include <typeinfo>
 
#include "CollectionBase/CollectionDescription.h"
#include "CollectionBase/ICollectionSchemaEditor.h"
 
#include "CoralBase/MessageStream.h"
 
#include "TTree.h"
 
 
namespace pool {
   class ICollectionColumn;
 
   namespace RootCollection {
 
      class RootCollection;
      
      class RootCollectionSchemaEditor : public ICollectionSchemaEditor
      {
        
     friend class RootCollection;
        
     public:
 
        RootCollectionSchemaEditor( RootCollection& collection,
                                    CollectionDescription& description,
                                    TTree *tree  );
 
        
        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 );
 
        virtual ~RootCollectionSchemaEditor();
 
     protected:
        void         addTreeBranch( const std::string& name, const std::string& type_name );
        void        readSchema();
        void        createTreeBranches();
        void        writeSchema();
        
        RootCollection                        &m_collection;
        
        CollectionDescription                &m_description;
 
        TTree                                *m_tree;
 
        bool                                m_schemaWritten;
 
        coral::MessageStream                m_poolOut; 
     };
   }
}
#endif