vtkAnariPolyDataMapperInheritInterface.h

Go to the documentation of this file.

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

#include <anari/anari_cpp.hpp>
#include <anari/anari_cpp/ext/std.h>

#include <vector>

class vtkDataArray;
class vtkPiecewiseFunction;
class vtkPolyData;
class vtkProperty;

using uvec2 = anari::std_types::uvec2;
using uvec3 = anari::std_types::uvec3;
using vec2 = anari::std_types::vec2;
using vec3 = anari::std_types::vec3;
using vec4 = anari::std_types::vec4;
using mat4 = anari::std_types::mat4;

class vtkAnariPolyDataMapperInheritInterface
{
public:
  virtual ~vtkAnariPolyDataMapperInheritInterface();

  struct ParameterFlags
  {
    bool Positions = true;
    bool Indices = true;
    bool Normals = true;
    bool Scales = true;
    bool Texcoords = true;
    bool Colors = true;
  };

  virtual void SetDevice(
    anari::Device& device, anari::Extensions& extensions, const char* const* anariExtensionStrings);

  virtual int GetSurfaceRepresentation(vtkProperty* property) const;
  virtual ParameterFlags GetBaseUpdateResponsibility() const;

  virtual anari::Geometry InitializeSpheres(vtkPolyData* polyData, vtkProperty* property,
    std::vector<vec3>& vertices, std::vector<uint32_t>& indexArray, double pointSize,
    vtkDataArray* scaleArray, vtkPiecewiseFunction* scaleFunction, std::vector<vec2>& textureCoords,
    std::vector<vec2>& pointValueTextureCoords, std::vector<vec4>& pointColors, int cellFlag);
  virtual anari::Geometry InitializeCurves(vtkPolyData* polyData, vtkProperty* property,
    std::vector<vec3>& vertices, std::vector<uint32_t>& indexArray, double lineWidth,
    vtkDataArray* scaleArray, vtkPiecewiseFunction* scaleFunction, std::vector<vec2>& textureCoords,
    std::vector<vec2>& pointValueTextureCoords, std::vector<vec4>& pointColors, int cellFlag);
  virtual anari::Geometry InitializeCylinders(vtkPolyData* polyData, vtkProperty* property,
    std::vector<vec3>& vertices, std::vector<uint32_t>& indexArray, double lineWidth,
    vtkDataArray* scaleArray, vtkPiecewiseFunction* scaleFunction, std::vector<vec2>& textureCoords,
    std::vector<vec2>& pointValueTextureCoords, std::vector<vec4>& pointColors, int cellFlag);
  virtual anari::Geometry InitializeTriangles(vtkPolyData* polyData, vtkProperty* property,
    std::vector<vec3>& vertices, std::vector<uint32_t>& indexArray, std::vector<vec3>& normals,
    std::vector<vec2>& textureCoords, std::vector<vec2>& pointValueTextureCoords,
    std::vector<vec4>& pointColors, int cellFlag);

  virtual const char* GetSpheresPostfix() const;
  virtual const char* GetCurvesPostfix() const;
  virtual const char* GetCylindersPostfix() const;
  virtual const char* GetTrianglesPostfix() const;

  anari::Device AnariDevice{ nullptr };
  anari::Extensions AnariExtensions{};
  const char* const* AnariExtensionStrings{ nullptr };
};

#endif