ANARIMapperVolume.h

Go to the documentation of this file.

//============================================================================
//  The contents of this file are covered by the Viskores license. See
//  LICENSE.txt for details.
//
//  By contributing to this file, all contributors agree to the Developer
//  Certificate of Origin Version 1.1 (DCO 1.1) as stated in DCO.txt.
//============================================================================
 
//============================================================================
//  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 viskores_interop_anari_ANARIMapperVolume_h
#define viskores_interop_anari_ANARIMapperVolume_h
 
#include <viskores/interop/anari/ANARIMapper.h>
 
namespace viskores
{
namespace interop
{
namespace anari
{
 
struct StructuredVolumeParameters
{
  anari_cpp::Array3D Data{ nullptr };
  int Dims[3];
  float Origin[3];
  float Spacing[3];
};
 
struct UnstructuredVolumeParameters
{
  anari_cpp::Array1D VertexPosition{ nullptr };
  anari_cpp::Array1D VertexData{ nullptr };
  anari_cpp::Array1D Index{ nullptr };
  anari_cpp::Array1D CellIndex{ nullptr };
  anari_cpp::Array1D CellData{ nullptr };
  anari_cpp::Array1D CellType{ nullptr };
  bool IndexPrefixed{ false };
};
 
struct StructuredVolumeArrays
{
  viskores::cont::ArrayHandle<viskores::Float32> Data;
  std::shared_ptr<viskores::cont::Token> Token{ new viskores::cont::Token };
};
 
struct UntructuredVolumeArrays
{
  viskores::cont::ArrayHandle<viskores::Vec3f_32> VertexPosition;
  viskores::cont::ArrayHandle<viskores::Float32> VertexData;
  viskores::cont::ArrayHandle<viskores::UInt64> Index;
  viskores::cont::ArrayHandle<viskores::UInt64> CellIndex;
  viskores::cont::ArrayHandle<viskores::Float32> CellData;
  viskores::cont::ArrayHandle<viskores::UInt8> CellType;
  std::shared_ptr<viskores::cont::Token> Token{ new viskores::cont::Token };
};
 
struct VISKORES_ANARI_EXPORT ANARIMapperVolume : public ANARIMapper
{
  ANARIMapperVolume(
    anari_cpp::Device device,
    const ANARIActor& actor = {},
    const std::string& name = "<volume>",
    const viskores::cont::ColorTable& colorTable = viskores::cont::ColorTable::Preset::Default);
 
  ~ANARIMapperVolume() override;
 
  void SetActor(const ANARIActor& actor) override;
 
  void SetANARIColorMap(anari_cpp::Array1D color,
                        anari_cpp::Array1D opacity,
                        bool releaseArrays = true) override;
  void SetANARIColorMapValueRange(const viskores::Vec2f_32& valueRange) override;
 
  void SetANARIColorMapOpacityScale(viskores::Float32 opacityScale) override;
 
  anari_cpp::SpatialField GetANARISpatialField() override;
  anari_cpp::Volume GetANARIVolume() override;
 
private:
  void ConstructArrays(bool regenerate = false);
  void UpdateSpatialField();
 
  struct ANARIHandles
  {
    anari_cpp::Device Device{ nullptr };
    anari_cpp::SpatialField SpatialField{ nullptr };
    anari_cpp::Volume Volume{ nullptr };
    StructuredVolumeParameters StructuredParameters;
    UnstructuredVolumeParameters UnstructuredParameters;
    ~ANARIHandles();
    void ReleaseArrays();
  };
 
  std::shared_ptr<ANARIHandles> Handles;
  StructuredVolumeArrays StructuredArrays;
  UntructuredVolumeArrays UnstructuredArrays;
};
 
} // namespace anari
} // namespace interop
} // namespace viskores
 
#endif