ANARIMapperTriangles.h

Go to the documentation of this file.

//============================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//============================================================================
 
#ifndef vtk_m_interop_anari_ANARIMapperTriangles_h
#define vtk_m_interop_anari_ANARIMapperTriangles_h
 
#include <vtkm/interop/anari/ANARIMapper.h>
 
namespace vtkm
{
namespace interop
{
namespace anari
{
 
struct TrianglesParameters
{
  struct VertexData
  {
    anari_cpp::Array1D Position{ nullptr };
    anari_cpp::Array1D Normal{ nullptr };
    std::array<anari_cpp::Array1D, 4> Attribute;
    std::array<std::string, 4> AttributeName;
  } Vertex{};
 
  struct PrimitiveData
  {
    anari_cpp::Array1D Index{ nullptr };
  } Primitive{};
 
  unsigned int NumPrimitives{ 0 };
};
 
struct TriangleArrays
{
  vtkm::cont::ArrayHandle<vtkm::Vec3f_32> Vertices;
  vtkm::cont::ArrayHandle<vtkm::Vec3f_32> Normals;
  std::shared_ptr<vtkm::cont::Token> Token{ new vtkm::cont::Token };
};
 
struct TriangleFieldArrays
{
  vtkm::cont::ArrayHandle<vtkm::Float32> Field1;
  std::string Field1Name;
  vtkm::cont::ArrayHandle<vtkm::Float32> Field2;
  std::string Field2Name;
  vtkm::cont::ArrayHandle<vtkm::Float32> Field3;
  std::string Field3Name;
  vtkm::cont::ArrayHandle<vtkm::Float32> Field4;
  std::string Field4Name;
  std::shared_ptr<vtkm::cont::Token> Token{ new vtkm::cont::Token };
};
 
struct VTKM_ANARI_EXPORT ANARIMapperTriangles : public ANARIMapper
{
  ANARIMapperTriangles(
    anari_cpp::Device device,
    const ANARIActor& actor = {},
    const std::string& name = "<triangles>",
    const vtkm::cont::ColorTable& colorTable = vtkm::cont::ColorTable::Preset::Default);
 
  ~ANARIMapperTriangles() override;
 
  void SetActor(const ANARIActor& actor) override;
 
  void SetMapFieldAsAttribute(bool enabled) override;
 
  void SetANARIColorMap(anari_cpp::Array1D color,
                        anari_cpp::Array1D opacity,
                        bool releaseArrays = true) override;
 
  void SetANARIColorMapValueRange(const vtkm::Vec2f_32& valueRange) override;
 
  void SetCalculateNormals(bool enabled);
 
  anari_cpp::Geometry GetANARIGeometry() override;
  anari_cpp::Surface GetANARISurface() override;
 
private:
  bool NeedToGenerateData() const;
  void ConstructArrays(bool regenerate = false);
  void UpdateGeometry();
  void UpdateMaterial();
 
  struct ANARIHandles
  {
    anari_cpp::Device Device{ nullptr };
    anari_cpp::Geometry Geometry{ nullptr };
    anari_cpp::Sampler Sampler{ nullptr };
    anari_cpp::Material Material{ nullptr };
    anari_cpp::Surface Surface{ nullptr };
    TrianglesParameters Parameters;
    ~ANARIHandles();
    void ReleaseArrays();
  };
 
  std::shared_ptr<ANARIHandles> Handles;
 
  bool CalculateNormals{ false };
  vtkm::IdComponent PrimaryField{ 0 };
  TriangleArrays Arrays;
  TriangleFieldArrays FieldArrays;
};
 
} // namespace anari
} // namespace interop
} // namespace vtkm
 
#endif