vtkDataArrayAccessor.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause

#include "vtkDataArray.h"
#include "vtkGenericDataArray.h"

#ifndef vtkDataArrayAccessor_h
#define vtkDataArrayAccessor_h

// Generic form for all (non-bit) vtkDataArray subclasses.
VTK_ABI_NAMESPACE_BEGIN
template <typename ArrayT>
struct vtkDataArrayAccessor
{
  typedef ArrayT ArrayType;
  typedef typename ArrayType::ValueType APIType;

  ArrayType* Array;

  vtkDataArrayAccessor(ArrayType* array)
    : Array(array)
  {
  }

  VTK_ALWAYS_INLINE
  APIType Get(vtkIdType tupleIdx, int compIdx) const
  {
    return this->Array->GetTypedComponent(tupleIdx, compIdx);
  }

  VTK_ALWAYS_INLINE
  APIType Get(vtkIdType valueIdx) const { return this->Array->GetValue(valueIdx); }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType tupleIdx, int compIdx, APIType val) const
  {
    this->Array->SetTypedComponent(tupleIdx, compIdx, val);
  }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType valueIdx, APIType val) const { this->Array->SetValue(valueIdx, val); }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType tupleIdx, int compIdx, APIType val) const
  {
    this->Array->InsertTypedComponent(tupleIdx, compIdx, val);
  }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType valueIdx, APIType val) const { this->Array->InsertValue(valueIdx, val); }

  VTK_ALWAYS_INLINE
  void InsertNext(APIType val) const { this->Array->InsertNextValue(val); }

  VTK_ALWAYS_INLINE
  void InsertNext(APIType* tuple) const { this->Array->InsertNextTypedTuple(tuple); }

  VTK_ALWAYS_INLINE
  void Get(vtkIdType tupleIdx, APIType* tuple) const
  {
    this->Array->GetTypedTuple(tupleIdx, tuple);
  }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType tupleIdx, const APIType* tuple) const
  {
    this->Array->SetTypedTuple(tupleIdx, tuple);
  }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType tupleIdx, const APIType* tuple) const
  {
    this->Array->InsertTypedTuple(tupleIdx, tuple);
  }
};

// Specialization for vtkDataArray.
template <>
struct vtkDataArrayAccessor<vtkDataArray>
{
  typedef vtkDataArray ArrayType;
  typedef double APIType;

  ArrayType* Array;

  vtkDataArrayAccessor(ArrayType* array)
    : Array(array)
  {
  }

  VTK_ALWAYS_INLINE
  APIType Get(vtkIdType tupleIdx, int compIdx) const
  {
    return this->Array->GetComponent(tupleIdx, compIdx);
  }

  VTK_ALWAYS_INLINE
  APIType Get(vtkIdType valueIdx) const
  {
    auto div = std::div(valueIdx, static_cast<vtkIdType>(this->Array->GetNumberOfComponents()));
    return this->Array->GetComponent(div.quot, div.rem);
  }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType tupleIdx, int compIdx, APIType val) const
  {
    this->Array->SetComponent(tupleIdx, compIdx, val);
  }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType valueIdx, APIType val) const
  {
    auto div = std::div(valueIdx, static_cast<vtkIdType>(this->Array->GetNumberOfComponents()));
    this->Array->SetComponent(div.quot, div.rem, val);
  }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType tupleIdx, int compIdx, APIType val) const
  {
    this->Array->InsertComponent(tupleIdx, compIdx, val);
  }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType valueIdx, APIType val) const
  {
    auto div = std::div(valueIdx, static_cast<vtkIdType>(this->Array->GetNumberOfComponents()));
    this->Array->InsertComponent(div.quot, div.rem, val);
  }

  VTK_ALWAYS_INLINE
  void InsertNext(APIType val) const { this->Array->InsertNextTuple1(val); }

  VTK_ALWAYS_INLINE
  void InsertNext(APIType* tuple) const { this->Array->InsertNextTuple(tuple); }

  VTK_ALWAYS_INLINE
  void Get(vtkIdType tupleIdx, APIType* tuple) const { this->Array->GetTuple(tupleIdx, tuple); }

  VTK_ALWAYS_INLINE
  void Set(vtkIdType tupleIdx, const APIType* tuple) const
  {
    this->Array->SetTuple(tupleIdx, tuple);
  }

  VTK_ALWAYS_INLINE
  void Insert(vtkIdType tupleIdx, const APIType* tuple) const
  {
    this->Array->InsertTuple(tupleIdx, tuple);
  }
};

VTK_ABI_NAMESPACE_END
#endif // vtkDataArrayAccessor_h
// VTK-HeaderTest-Exclude: vtkDataArrayAccessor.h