ToolColumnVectorMap.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

 
 
#ifndef COLUMNAR_TOOL_WRAPPER_TOOL_COLUMN_VECTOR_MAP_H
#define COLUMNAR_TOOL_WRAPPER_TOOL_COLUMN_VECTOR_MAP_H
 
#include <ColumnarToolWrapper/ColumnVectorWrapper.h>
 
#include <unordered_map>
 
namespace columnar
{
 
  class ToolColumnVectorMap final
  {
 
  public:

    explicit ToolColumnVectorMap (ColumnVectorHeader& val_columnHeader, IColumnarTool& val_tool);
 

    [[nodiscard]] const IColumnarTool& getTool () const {
      return *m_tool; }

    [[nodiscard]] const ColumnVectorHeader& getColumnHeader () const {
      return *m_columnHeader; }
 

    [[nodiscard]] std::vector<std::string> getColumnNames () const;

    [[nodiscard]] std::size_t getColumnIndex (const std::string& name) const;
 
 

  

    template<typename CT>
    void setColumn (ColumnVectorData& columnData, const std::string& name, std::size_t size, CT* dataPtr) const {
      columnData.setColumn (getColumnIndex(name), size, dataPtr);
    }

    void setColumnVoid (ColumnVectorData& columnData, const std::string& name, std::size_t size, const void *dataPtr, const std::type_info& type, bool isConst) const {
      columnData.setColumnVoid (getColumnIndex(name), size, dataPtr, type, isConst);
    }
 

    template<typename CT>
    [[nodiscard]] std::pair<std::size_t,CT*>
    getColumn (ColumnVectorData& columnData, const std::string& name) const {
      return columnData.getColumn<CT> (getColumnIndex(name));
    }

    [[nodiscard]] std::pair<std::size_t,const void*>
    getColumnVoid (ColumnVectorData& columnData, const std::string& name, const std::type_info *type, bool isConst) const {
      return columnData.getColumnVoid (getColumnIndex(name), type, isConst);
    }
 
 

 
  private:

    const IColumnarTool *m_tool = nullptr;

    ColumnVectorHeader *m_columnHeader = nullptr;

    struct MyColumnInfo
    {
      unsigned index = 0u;
    };
    std::unordered_map<std::string,MyColumnInfo> m_columns;
  };
}
 
#endif