nv-sgl 0.6.0__cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl

sgl/include/sgl/device/query.h

@@ -0,0 +1,52 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "sgl/device/fwd.h"
+#include "sgl/device/types.h"
+#include "sgl/device/device_resource.h"
+
+#include "sgl/core/macros.h"
+#include "sgl/core/object.h"
+#include "sgl/core/enum.h"
+
+#include <slang-gfx.h>
+
+#include <span>
+
+namespace sgl {
+
+struct QueryPoolDesc {
+    /// Query type.
+    QueryType type;
+    /// Number of queries in the pool.
+    uint32_t count;
+};
+
+class SGL_API QueryPool : public DeviceResource {
+    SGL_OBJECT(QueryPool)
+public:
+    QueryPool(ref<Device> device, QueryPoolDesc desc);
+
+    const QueryPoolDesc& desc() const { return m_desc; }
+
+    void reset();
+
+    void get_results(uint32_t index, uint32_t count, std::span<uint64_t> result);
+    std::vector<uint64_t> get_results(uint32_t index, uint32_t count);
+    uint64_t get_result(uint32_t index);
+
+    void get_timestamp_results(uint32_t index, uint32_t count, std::span<double> result);
+    std::vector<double> get_timestamp_results(uint32_t index, uint32_t count);
+    double get_timestamp_result(uint32_t index);
+
+    gfx::IQueryPool* gfx_query_pool() const { return m_gfx_query_pool; }
+
+    std::string to_string() const override;
+
+private:
+    QueryPoolDesc m_desc;
+    Slang::ComPtr<gfx::IQueryPool> m_gfx_query_pool;
+};
+
+} // namespace sgl

sgl/include/sgl/device/raytracing.h

@@ -0,0 +1,84 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "sgl/device/fwd.h"
+#include "sgl/device/types.h"
+#include "sgl/device/device_resource.h"
+#include "sgl/device/resource.h"
+
+#include "sgl/math/vector_types.h"
+#include "sgl/math/matrix_types.h"
+
+#include "sgl/core/macros.h"
+#include "sgl/core/object.h"
+#include "sgl/core/enum.h"
+
+#include <slang-gfx.h>
+
+namespace sgl {
+
+struct AccelerationStructureQueryDesc {
+    QueryType query_type;
+    QueryPool* query_pool;
+    uint32_t first_query_index;
+};
+
+struct AccelerationStructureBuildDesc {
+    AccelerationStructureBuildInputs inputs;
+    AccelerationStructure* src;
+    AccelerationStructure* dst;
+    DeviceAddress scratch_data;
+};
+
+struct AccelerationStructureDesc {
+    AccelerationStructureKind kind;
+    ref<Buffer> buffer;
+    DeviceOffset offset{0};
+    DeviceSize size{0};
+};
+
+class SGL_API AccelerationStructure : public DeviceResource {
+    SGL_OBJECT(AccelerationStructure)
+public:
+    AccelerationStructure(ref<Device> device, AccelerationStructureDesc desc);
+    ~AccelerationStructure();
+
+    const AccelerationStructureDesc& desc() const { return m_desc; }
+
+    AccelerationStructureKind kind() const { return m_desc.kind; }
+
+    DeviceAddress device_address() const;
+
+    gfx::IAccelerationStructure* gfx_acceleration_structure() const { return m_gfx_acceleration_structure; }
+
+    std::string to_string() const override;
+
+private:
+    AccelerationStructureDesc m_desc;
+    Slang::ComPtr<gfx::IAccelerationStructure> m_gfx_acceleration_structure;
+};
+
+struct ShaderTableDesc {
+    ref<ShaderProgram> program;
+    std::vector<std::string> ray_gen_entry_points;
+    std::vector<std::string> miss_entry_points;
+    std::vector<std::string> hit_group_names;
+    std::vector<std::string> callable_entry_points;
+};
+
+class SGL_API ShaderTable : public DeviceResource {
+    SGL_OBJECT(ShaderTable)
+public:
+    ShaderTable(ref<Device> device, ShaderTableDesc desc);
+    ~ShaderTable();
+
+    gfx::IShaderTable* gfx_shader_table() const { return m_gfx_shader_table; }
+
+    std::string to_string() const override;
+
+private:
+    Slang::ComPtr<gfx::IShaderTable> m_gfx_shader_table;
+};
+
+} // namespace sgl