nv-sgl 0.6.0__cp313-cp313-win_amd64.whl

sgl/include/sgl/math/matrix_types.h

@@ -0,0 +1,207 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "sgl/math/scalar_types.h"
+#include "sgl/math/vector_types.h"
+#include "sgl/core/error.h"
+
+#include <array>
+#include <initializer_list>
+#include <limits>
+
+namespace sgl::math {
+
+/**
+ * Matrix type with row-major storage.
+ *
+ * The semantics are aligned with Slang:
+ * - Row major storage
+ * - Math operators are element-wise (e.g. +, -, *, /)
+ * - Free standing functions for matrix operations (e.g. mul(), transpose(), etc.)
+ *
+ * \tparam T Scalar type
+ * \tparam RowCount Number of rows (1-4)
+ * \tparam ColCount Number of columns (1-4)
+ */
+template<typename T, int RowCount, int ColCount>
+class matrix {
+    static_assert(RowCount >= 1 && RowCount <= 4);
+    static_assert(ColCount >= 1 && ColCount <= 4);
+
+    // For now, we only support float, but want to have the type T visible
+    static_assert(std::is_same_v<float, T>);
+
+private:
+    template<typename, int, int>
+    friend class matrix;
+
+public:
+    static constexpr int rows = RowCount;
+    static constexpr int cols = ColCount;
+    using value_type = T;
+    using row_type = vector<T, ColCount>;
+    using col_type = vector<T, RowCount>;
+
+    matrix()
+        : matrix(Form::Identity)
+    {
+    }
+
+    template<arithmetic U>
+    explicit matrix(std::initializer_list<U> a)
+    {
+        SGL_CHECK(
+            a.size() == RowCount * ColCount,
+            "Invalid number of coefficients (expected {}, got {})",
+            RowCount * ColCount,
+            a.size()
+        );
+        T* f = &m_rows[0][0];
+        for (auto it = a.begin(); it != a.end(); ++it, ++f)
+            *f = static_cast<T>(*it);
+    }
+
+    template<typename U>
+    explicit matrix(const std::array<U, RowCount * ColCount>& a)
+    {
+        T* f = &m_rows[0][0];
+        for (const auto v : a)
+            *f++ = static_cast<T>(v);
+    }
+
+    matrix(const matrix&) = default;
+    matrix(matrix&&) noexcept = default;
+
+    /// Construct matrix from another matrix with different dimensions.
+    /// In HLSL/Slang, destination matrix must be equal or smaller than source matrix.
+    /// In sgl, destination matrix can be larger than source matrix (initialized with identity).
+    template<int R, int C>
+    matrix(const matrix<T, R, C>& other)
+        : matrix(Form::Identity)
+    {
+        for (int r = 0; r < std::min(RowCount, R); ++r) {
+            std::memcpy(&m_rows[r], &other.m_rows[r], std::min(ColCount, C) * sizeof(T));
+        }
+    }
+
+    matrix& operator=(const matrix&) = default;
+    matrix& operator=(matrix&&) = default;
+
+    template<int R, int C>
+    matrix& operator=(const matrix<T, R, C>& other)
+    {
+        for (int r = 0; r < std::min(RowCount, R); ++r) {
+            std::memcpy(&m_rows[r], &other.m_rows[r], std::min(ColCount, C) * sizeof(T));
+        }
+
+        return *this;
+    }
+
+    /// Zero matrix.
+    [[nodiscard]] static matrix zeros() { return matrix(Form::Zeros); }
+
+    /// Identity matrix.
+    [[nodiscard]] static matrix identity() { return matrix(Form::Identity); }
+
+    T* data() { return &m_rows[0][0]; }
+    const T* data() const { return &m_rows[0][0]; }
+
+    row_type& operator[](int r)
+    {
+        SGL_ASSERT_LT(r, RowCount);
+        return m_rows[r];
+    }
+    const row_type& operator[](int r) const
+    {
+        SGL_ASSERT_LT(r, RowCount);
+        return m_rows[r];
+    }
+
+    row_type& get_row(int r)
+    {
+        SGL_ASSERT_LT(r, RowCount);
+        return m_rows[r];
+    }
+    const row_type& get_row(int r) const
+    {
+        SGL_ASSERT_LT(r, RowCount);
+        return m_rows[r];
+    }
+
+    void set_row(int r, const row_type& v)
+    {
+        SGL_ASSERT_LT(r, RowCount);
+        m_rows[r] = v;
+    }
+
+    col_type get_col(int col) const
+    {
+        SGL_ASSERT_LT(col, ColCount);
+        col_type result;
+        for (int r = 0; r < RowCount; ++r)
+            result[r] = m_rows[r][col];
+        return result;
+    }
+
+    void set_col(int col, const col_type& v)
+    {
+        SGL_ASSERT_LT(col, ColCount);
+        for (int r = 0; r < RowCount; ++r)
+            m_rows[r][col] = v[r];
+    }
+
+    bool operator==(const matrix& rhs) const { return std::memcmp(this, &rhs, sizeof(*this)) == 0; }
+    bool operator!=(const matrix& rhs) const { return !(*this == rhs); }
+
+private:
+    enum class Form {
+        Undefined,
+        Zeros,
+        Identity,
+    };
+
+    explicit matrix(Form form)
+    {
+        switch (form) {
+        case Form::Undefined:
+#ifdef _DEBUG
+            for (int i = 0; i < RowCount; ++i)
+                m_rows[i] = row_type(std::numeric_limits<T>::quiet_NaN());
+#endif
+            break;
+        case Form::Zeros:
+            std::memset(this, 0, sizeof(*this));
+            break;
+        case Form::Identity:
+            std::memset(this, 0, sizeof(*this));
+            for (int i = 0; i < std::min(RowCount, ColCount); ++i)
+                m_rows[i][i] = T(1);
+            break;
+        }
+    }
+
+    row_type m_rows[RowCount];
+};
+
+using float2x2 = matrix<float, 2, 2>;
+
+using float3x3 = matrix<float, 3, 3>;
+
+using float2x4 = matrix<float, 2, 4>;
+using float3x4 = matrix<float, 3, 4>;
+using float4x4 = matrix<float, 4, 4>;
+
+} // namespace sgl::math
+
+namespace sgl {
+
+using float2x2 = math::float2x2;
+
+using float3x3 = math::float3x3;
+
+using float2x4 = math::float2x4;
+using float3x4 = math::float3x4;
+using float4x4 = math::float4x4;
+
+} // namespace sgl

sgl/include/sgl/math/python/primitivetype.h

@@ -0,0 +1,33 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#include "nanobind.h"
+#include "sgl/core/error.h"
+
+namespace sgl::math {
+
+template<class T>
+struct PrimitiveType {
+    static PyObject* python_type() { SGL_THROW("Unknown primitive type"); }
+};
+template<>
+struct PrimitiveType<float16_t> {
+    static PyObject* python_type() { return (PyObject*)&PyFloat_Type; }
+};
+template<>
+struct PrimitiveType<float> {
+    static PyObject* python_type() { return (PyObject*)&PyFloat_Type; }
+};
+template<>
+struct PrimitiveType<int> {
+    static PyObject* python_type() { return (PyObject*)&PyLong_Type; }
+};
+template<>
+struct PrimitiveType<uint32_t> {
+    static PyObject* python_type() { return (PyObject*)&PyLong_Type; }
+};
+template<>
+struct PrimitiveType<bool> {
+    static PyObject* python_type() { return (PyObject*)&PyBool_Type; }
+};
+
+} // namespace sgl::math

sgl/include/sgl/math/quaternion.h

@@ -0,0 +1,6 @@
+// SPDX-License-Identifier: Apache-2.0
+
+#pragma once
+
+#include "sgl/math/quaternion_types.h"
+#include "sgl/math/quaternion_math.h"