ObjectColumnArray.icc

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
/// @author Nils Krumnack
 
 
#include <ColumnarCore/ColumnarTool.h>
#include <ColumnarCore/ColumnAccessorDataArray.h>
#include <ColumnarCore/ObjectId.h>
#include <ColumnarCore/ObjectRange.h>
#include <ColumnarCore/ContainerId.h>
#include <cassert>
 
namespace columnar
{
  template<ContainerIdConcept CI,ColumnarArrayMode CM>
  class AccessorTemplate<CI,ObjectColumn,ColumnAccessMode::input,CM> final
  {
    /// Common Public Members
    /// =====================
  public:
 
    static_assert (CI::perEventRange || CI::perEventId, "ObjectColumn can only be used with per-event ranges or IDs");
 
    AccessorTemplate () = default;
 
    AccessorTemplate (ColumnarTool<CM>& columnBase,
                      const std::string& name,
                      ColumnAccessorOptions&& options = {})
      : m_columnBase (&columnBase),
        m_name (name)
    {
      if (!name.empty())
        columnBase.setContainerUserName (CI::idName, name);
      m_offsetsData = std::make_unique<ColumnAccessorDataArray> (&m_offsetsIndex, &m_offsetsData, &typeid (ColumnarOffsetType), ColumnAccessMode::input);
      auto info = options.makeColumnInfo();
      info.offsetName = (CI::perEventRange ? numberOfEventsName : "");
      info.isOffset = true;
      columnBase.addColumn (std::string (CI::idName), m_offsetsData.get(), std::move (info));
    }
 
    AccessorTemplate (AccessorTemplate&& that)
    {
      moveAccessor (m_offsetsIndex, m_offsetsData, that.m_offsetsIndex, that.m_offsetsData);
    }
 
    // cppcheck-suppress operatorEqVarError; m_columnBase not copied
    AccessorTemplate& operator = (AccessorTemplate&& that)
    {
      if (this != &that)
        moveAccessor (m_offsetsIndex, m_offsetsData, that.m_offsetsIndex, that.m_offsetsData);
      return *this;
    }
 
    AccessorTemplate (const AccessorTemplate&) = delete;
    AccessorTemplate& operator = (const AccessorTemplate&) = delete;
 
    ObjectRange<CI,CM> operator() (ObjectId<ContainerId::eventContext,CM> eventId) const
      requires (CI::perEventRange)
    {
      auto *offsets = static_cast<const ColumnarOffsetType*>(eventId.getDataArea()[m_offsetsIndex]);
      return ObjectRange<CI,CM> (eventId.getDataArea(), offsets[eventId.getIndex()], offsets[eventId.getIndex() + 1]);
    }
 
    ObjectRange<CI,CM> operator() (ObjectRange<ContainerId::eventContext,CM> eventRange) const
      requires (CI::perEventRange)
    {
      auto *offsets = static_cast<const ColumnarOffsetType*>(eventRange.getDataArea()[m_offsetsIndex]);
      return ObjectRange<CI,CM> (eventRange.getDataArea(), offsets[eventRange.beginIndex()], offsets[eventRange.endIndex()]);
    }
 
    ObjectId<CI,CM> operator() (ObjectId<ContainerId::eventContext,CM> eventId) const
      requires (CI::perEventId)
    {
      return ObjectId<CI,CM> (eventId.getDataArea(), eventId.getIndex());
    }
 
    ObjectRange<CI,CM> operator() (ObjectRange<ContainerId::eventContext,CM> eventRange) const
      requires (CI::perEventId)
    {
      return ObjectRange<CI,CM> (eventRange.getDataArea(), eventRange.beginIndex(), eventRange.endIndex());
    }
 
    [[nodiscard]] bool isAvailable (ObjectId<ContainerId::eventContext,CM> eventId) const noexcept
    {
      auto *offsets = static_cast<const ColumnarOffsetType*>(eventId.getDataArea()[m_offsetsIndex]);
      return offsets != nullptr;
    }      
 
 
 
    /// Private Members
    /// ===============
  private:
 
    ColumnarTool<CM> *m_columnBase = nullptr;
 
    std::string m_name;
    unsigned m_offsetsIndex = 0;
    std::unique_ptr<ColumnAccessorDataArray> m_offsetsData;
  };
}