ANARIActor.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_ANARIActor_h
#define vtk_m_interop_anari_ANARIActor_h
 
// vtk-m
#include <vtkm/cont/CoordinateSystem.h>
#include <vtkm/cont/DataSet.h>
#include <vtkm/cont/Field.h>
#include <vtkm/cont/UnknownCellSet.h>
#include <vtkm/interop/anari/VtkmANARITypes.h>
// std
#include <array>
#include <memory>
 
#include <vtkm/interop/anari/vtkm_anari_export.h>
 
namespace vtkm
{
namespace interop
{
namespace anari
{
 
using FieldSet = std::array<vtkm::cont::Field, 4>;
 
const char* AnariMaterialInputString(vtkm::IdComponent p);
 
struct VTKM_ANARI_EXPORT ANARIActor
{
  ANARIActor() = default;
 
  ANARIActor(const vtkm::cont::UnknownCellSet& cells,
             const vtkm::cont::CoordinateSystem& coordinates,
             const vtkm::cont::Field& field0 = {},
             const vtkm::cont::Field& field1 = {},
             const vtkm::cont::Field& field2 = {},
             const vtkm::cont::Field& field3 = {});
 
  ANARIActor(const vtkm::cont::UnknownCellSet& cells,
             const vtkm::cont::CoordinateSystem& coordinates,
             const FieldSet& fieldset);
 
  ANARIActor(const vtkm::cont::DataSet& dataset,
             const std::string& field0 = "",
             const std::string& field1 = "",
             const std::string& field2 = "",
             const std::string& field3 = "");
 
  const vtkm::cont::UnknownCellSet& GetCellSet() const;
  const vtkm::cont::CoordinateSystem& GetCoordinateSystem() const;
  const vtkm::cont::Field& GetField(vtkm::IdComponent idx = -1) const;
 
  FieldSet GetFieldSet() const;
 
  void SetPrimaryFieldIndex(vtkm::IdComponent idx);
  vtkm::IdComponent GetPrimaryFieldIndex() const;
 
  vtkm::cont::DataSet MakeDataSet(bool includeFields = false) const;
 
private:
  struct ActorData
  {
    vtkm::cont::UnknownCellSet Cells;
    vtkm::cont::CoordinateSystem Coordinates;
    FieldSet Fields;
    vtkm::IdComponent PrimaryField{ 0 };
  };
 
  std::shared_ptr<ActorData> Data = std::make_shared<ActorData>();
};
 
} // namespace anari
} // namespace interop
} // namespace vtkm
 
#endif