first-light of anari_tsd device

Author: jeffamstutz

tsd/src/CMakeLists.txt

@@ -1,4 +1,5 @@
 ## Copyright 2024-2025 NVIDIA Corporation
 ## SPDX-License-Identifier: Apache-2.0
 
+add_subdirectory(anari_tsd)
 add_subdirectory(tsd)

tsd/src/anari_tsd/Array.cpp

@@ -0,0 +1,140 @@
+// Copyright 2026 NVIDIA Corporation
+// SPDX-License-Identifier: Apache-2.0
+
+#include "Array.h"
+#include "Object.h"
+
+namespace tsd_device {
+
+Array::Array(DeviceGlobalState *state, Array1DMemoryDescriptor desc)
+    : helium::BaseArray(ANARI_ARRAY1D, state),
+      m_appMemory(desc.appMemory),
+      m_deleter(desc.deleter),
+      m_deleterPtr(desc.deleterPtr)
+{
+  m_array = state->scene.createArray(desc.elementType, desc.numItems);
+  syncSharedData();
+}
+
+Array::Array(DeviceGlobalState *state, Array2DMemoryDescriptor desc)
+    : helium::BaseArray(ANARI_ARRAY2D, state),
+      m_appMemory(desc.appMemory),
+      m_deleter(desc.deleter),
+      m_deleterPtr(desc.deleterPtr)
+{
+  m_array = state->scene.createArray(
+      desc.elementType, desc.numItems1, desc.numItems2);
+  syncSharedData();
+}
+
+Array::Array(DeviceGlobalState *state, Array3DMemoryDescriptor desc)
+    : helium::BaseArray(ANARI_ARRAY3D, state),
+      m_appMemory(desc.appMemory),
+      m_deleter(desc.deleter),
+      m_deleterPtr(desc.deleterPtr)
+{
+  m_array = state->scene.createArray(
+      desc.elementType, desc.numItems1, desc.numItems2, desc.numItems3);
+  syncSharedData();
+}
+
+Array::~Array()
+{
+  freeAppMemory();
+}
+
+bool Array::isShared() const
+{
+  return m_appMemory != nullptr;
+}
+
+void *Array::map()
+{
+  return const_cast<void *>(isShared() ? m_appMemory : m_array->map());
+}
+
+void Array::unmap()
+{
+  syncSharedData();
+  if (!isShared())
+    m_array->unmap();
+}
+
+void Array::privatize()
+{
+  freeAppMemory();
+}
+
+void Array::commitParameters()
+{
+  auto *object = tsdObject();
+  std::for_each(params_begin(), params_end(), [&](auto &p) {
+    if (anari::isObject(p.second.type())) {
+      if (anari::isArray(p.second.type())) {
+        auto *arr = p.second.template getObject<Array>();
+        if (arr) {
+          object->setParameterObject(
+              tsd::core::Token(p.first), *arr->tsdObject());
+        }
+      } else {
+        auto *obj = p.second.template getObject<TSDObject>();
+        if (obj && obj->tsdObject() != nullptr) {
+          object->setParameterObject(
+              tsd::core::Token(p.first), *obj->tsdObject());
+        }
+      }
+    } else if (p.second.type() != ANARI_UNKNOWN) {
+      object->setParameter(
+          tsd::core::Token(p.first), p.second.type(), p.second.data());
+    } else {
+      reportMessage(ANARI_SEVERITY_WARNING,
+          "skip setting parameter '%s' of unknown type",
+          p.first.c_str());
+    }
+  });
+}
+
+void Array::finalize()
+{
+  // no-op
+}
+
+bool Array::getProperty(const std::string_view &name,
+    ANARIDataType type,
+    void *ptr,
+    uint64_t size,
+    uint32_t flags)
+{
+  // no-op
+  return 0;
+}
+
+tsd::core::Object *Array::tsdObject() const
+{
+  return m_array.data();
+}
+
+DeviceGlobalState *Array::deviceState() const
+{
+  return (DeviceGlobalState *)m_state;
+}
+
+void Array::syncSharedData()
+{
+  if (isShared())
+    m_array->setData(m_appMemory);
+}
+
+void Array::freeAppMemory()
+{
+  if (isShared() && m_deleter != nullptr)
+    m_deleter(m_deleterPtr, m_appMemory);
+
+  m_appMemory = nullptr;
+  m_deleter = nullptr;
+  m_deleterPtr = nullptr;
+}
+
+} // namespace tsd_device
+
+TSD_DEVICE_ANARI_TYPEFOR_DEFINITION(tsd_device::Array *);

tsd/src/anari_tsd/Array.h

@@ -0,0 +1,59 @@
+// Copyright 2026 NVIDIA Corporation
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "DeviceGlobalState.h"
+// helium
+#include "helium/array/Array1D.h"
+#include "helium/array/Array2D.h"
+#include "helium/array/Array3D.h"
+#include "helium/array/ObjectArray.h"
+// tsd_core
+#include "tsd/core/scene/objects/Array.hpp"
+
+namespace tsd_device {
+
+using Array1DMemoryDescriptor = helium::Array1DMemoryDescriptor;
+using Array2DMemoryDescriptor = helium::Array2DMemoryDescriptor;
+using Array3DMemoryDescriptor = helium::Array3DMemoryDescriptor;
+using ObjectArray = helium::ObjectArray;
+
+struct Array : public helium::BaseArray
+{
+  Array(DeviceGlobalState *state, Array1DMemoryDescriptor desc);
+  Array(DeviceGlobalState *state, Array2DMemoryDescriptor desc);
+  Array(DeviceGlobalState *state, Array3DMemoryDescriptor desc);
+  ~Array() override;
+
+  bool isShared() const;
+
+  void *map() override;
+  void unmap() override;
+  void privatize() override;
+
+  void commitParameters() override;
+  void finalize() override;
+
+  bool getProperty(const std::string_view &name,
+      ANARIDataType type,
+      void *ptr,
+      uint64_t size,
+      uint32_t flags) override;
+
+  tsd::core::Object *tsdObject() const;
+  DeviceGlobalState *deviceState() const;
+
+ private:
+  void syncSharedData();
+  void freeAppMemory();
+
+  mutable tsd::core::ArrayRef m_array;
+  const void *m_appMemory{nullptr};
+  ANARIMemoryDeleter m_deleter{};
+  const void *m_deleterPtr{nullptr};
+};
+
+} // namespace tsd_device
+
+TSD_DEVICE_ANARI_TYPEFOR_SPECIALIZATION(tsd_device::Array *, ANARI_ARRAY);

tsd/src/anari_tsd/CMakeLists.txt

@@ -0,0 +1,41 @@
+## Copyright 2026 NVIDIA Corporation
+## SPDX-License-Identifier: Apache-2.0
+
+project(anari_library_tsd LANGUAGES CXX)
+
+project_add_library(SHARED
+  Array.cpp
+  Device.cpp
+  DeviceGlobalState.cpp
+  Frame.cpp
+  Group.cpp
+  Instance.cpp
+  Library.cpp
+  Object.cpp
+  World.cpp
+)
+project_link_libraries(PRIVATE tsd_app tsd_io anari::helium)
+project_compile_definitions(PRIVATE "anari_library_tsd_EXPORTS")
+project_include_directories(
+PRIVATE
+  $<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}>
+)
+
+## Generated code ##
+
+include(GenerateExportHeader)
+generate_export_header(${PROJECT_NAME} EXPORT_MACRO_NAME "ANARI_TSD_EXPORT")
+
+anari_generate_queries(
+  DEVICE_TARGET ${PROJECT_NAME}
+  CPP_NAMESPACE tsd_device
+  JSON_DEFINITIONS_FILE ${ANARI_CODE_GEN_ROOT}/devices/experimental_device.json
+)
+
+## Install ##
+
+install(TARGETS ${PROJECT_NAME}
+  LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
+  RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR}
+  ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
+)