TESSerializer.h

Go to the documentation of this file.

#ifndef GAUDIMP_TESSERIALIZER_H
#define GAUDIMP_TESSERIALIZER_H

#include "GaudiKernel/DataObject.h"
#include "GaudiKernel/IAddressCreator.h"
#include "GaudiKernel/IDataStoreAgent.h"

// ROOT includes
#include "TClass.h"

// vector and string
#include <map>
#include <string>
#include <typeindex>
#include <vector>

// forward declarations
class IDataProviderSvc;
struct IDataManagerSvc;
class TBufferFile;
class DataStoreItem;
class IAddressCreator;

namespace GaudiMP {
  class GAUDI_API TESSerializer : virtual public IDataStoreAgent {
    typedef std::vector<DataStoreItem*> Items;
    typedef std::vector<std::string>    ItemNames;
    typedef std::vector<DataObject*>    Objects;

  public:
    TESSerializer( IDataProviderSvc* svc, IAddressCreator* ac );

    void dumpBuffer( TBufferFile& );

    void loadBuffer( TBufferFile& );

    void addItem( const std::string& path );

    void addOptItem( const std::string& path );

    bool analyse( IRegistry* dir, int level ) override;

    void checkItems();

    virtual ~TESSerializer() {}

  protected:
    void addItem( Items& itms, const std::string& descriptor );

    DataStoreItem* findItem( const std::string& path );

  private:
    TClass* getClass( DataObject* obj ) {
      auto& id  = typeid( *obj );
      auto  pos = m_classMap.find( id );
      if ( pos == end( m_classMap ) ) { return m_classMap[id] = TClass::GetClass( id ); }
      return pos->second;
    }

    IDataProviderSvc* m_TES;
    IDataManagerSvc* m_TESMgr;
    ItemNames m_itemNames;
    Items m_itemList;
    ItemNames m_optItemNames;
    Items m_optItemList;
    DataStoreItem* m_currentItem;
    Objects m_objects;

    std::map<std::type_index, TClass*> m_classMap;
    bool m_verifyItems;
    bool m_strict;
    IAddressCreator* m_addressCreator;
  };
} // namespace GaudiMP
#endif