diff --git a/src/tenvex/expressions/add.h b/src/tenvex/expressions/add.h new file mode 100644 index 0000000..0dbb6a5 --- /dev/null +++ b/src/tenvex/expressions/add.h @@ -0,0 +1,36 @@ +#pragma once + +#include "defines.h" +#include "expression.h" +#include "concepts.h" +#include "traits.h" + +#include + +namespace tnvx +{ + +template +struct Add : Expr> +{ + TNVX_INLINE + Add(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; + + [[nodiscard]] TNVX_INLINE + __m128 eval() const noexcept; + +private: + tnvx_ref_or_value_t _l; + tnvx_ref_or_value_t _r; +}; + +template +[[nodiscard]] TNVX_INLINE +Add operator+(const L& l, const R& r) noexcept; + +template +inline constexpr bool is_vec_expr> = true; + +} + +#include "add_impl.hpp" diff --git a/src/tenvex/expressions/add_impl.hpp b/src/tenvex/expressions/add_impl.hpp new file mode 100644 index 0000000..1df6b07 --- /dev/null +++ b/src/tenvex/expressions/add_impl.hpp @@ -0,0 +1,31 @@ +#pragma once + +#include "defines.h" +#include "add.h" + +namespace tnvx +{ + +template +TNVX_INLINE +Add::Add(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept + : _l(l) + , _r(r) +{ +} + +template +[[nodiscard]] TNVX_INLINE +__m128 Add::eval() const noexcept +{ + return _mm_add_ps(_l.eval(), _r.eval()); +} + +template +[[nodiscard]] TNVX_INLINE +Add operator+(const L& l, const R& r) noexcept +{ + return { l, r }; +} + +} diff --git a/src/tenvex/expressions/concepts.h b/src/tenvex/expressions/concepts.h new file mode 100644 index 0000000..59db405 --- /dev/null +++ b/src/tenvex/expressions/concepts.h @@ -0,0 +1,25 @@ +#pragma once + +#include "traits.h" +#include "expression.h" + +#include + +namespace tnvx +{ + +template +concept expression = requires (T t) +{ + { t.eval() } -> std::same_as<__m128>; + { t.self() } -> std::same_as; +} +&& std::is_base_of_v, T>; + +template +concept vec_expr = expression && is_vec_expr; + +template +concept scalar_expr = expression && is_scalar_expr; + +} diff --git a/src/tenvex/internal/defines.h b/src/tenvex/expressions/defines.h similarity index 92% rename from src/tenvex/internal/defines.h rename to src/tenvex/expressions/defines.h index 28f1b98..4eb4064 100644 --- a/src/tenvex/internal/defines.h +++ b/src/tenvex/expressions/defines.h @@ -1,4 +1,4 @@ -#pragma once +#pragma once #if defined(_MSC_VER) #define TNVX_INLINE __forceinline @@ -9,4 +9,4 @@ #else #define TNVX_INLINE inline #define TNVX_RESTRICT -#endif +#endif \ No newline at end of file diff --git a/src/tenvex/expressions/expression.h b/src/tenvex/expressions/expression.h new file mode 100644 index 0000000..f309e5e --- /dev/null +++ b/src/tenvex/expressions/expression.h @@ -0,0 +1,19 @@ +#pragma once + +#include "defines.h" + +#include + +namespace tnvx +{ + +template +struct Expr +{ + [[nodiscard]] TNVX_INLINE + const Derived& self() const noexcept; +}; + +} + +#include "expression_impl.hpp" diff --git a/src/tenvex/expressions/expression_impl.hpp b/src/tenvex/expressions/expression_impl.hpp new file mode 100644 index 0000000..c6b60ca --- /dev/null +++ b/src/tenvex/expressions/expression_impl.hpp @@ -0,0 +1,14 @@ +#include "defines.h" +#include "expression.h" + +namespace tnvx +{ + +template +TNVX_INLINE +const Derived& Expr::self() const noexcept +{ + return static_cast(*this); +} + +} diff --git a/src/tenvex/expressions/mul.h b/src/tenvex/expressions/mul.h new file mode 100644 index 0000000..b8e14b0 --- /dev/null +++ b/src/tenvex/expressions/mul.h @@ -0,0 +1,48 @@ +#pragma once + +#include "defines.h" +#include "traits.h" +#include "expression.h" +#include "concepts.h" + +#include "scalar.h" + +#include + +namespace tnvx +{ + +template +struct Mul : Expr> +{ + TNVX_INLINE + Mul(const L& l, const R& r) noexcept; + + [[nodiscard]] TNVX_INLINE + __m128 eval() const noexcept; + +private: + tnvx_ref_or_value_t _l; + tnvx_ref_or_value_t _r; +}; + +template +[[nodiscard]] TNVX_INLINE +Mul operator*(const E& l, float r) noexcept; + +template +[[nodiscard]] TNVX_INLINE +Mul operator*(float l, const E& r) noexcept; + +template +[[nodiscard]] TNVX_INLINE +Mul operator*(const L& l, const R& r) noexcept; + +template inline constexpr bool is_vec_expr> = true; +template inline constexpr bool is_vec_expr> = true; + +template inline constexpr bool is_scalar_expr> = true; + +} + +#include "mul_impl.hpp" diff --git a/src/tenvex/expressions/mul_impl.hpp b/src/tenvex/expressions/mul_impl.hpp new file mode 100644 index 0000000..80278d7 --- /dev/null +++ b/src/tenvex/expressions/mul_impl.hpp @@ -0,0 +1,44 @@ +#pragma once + +#include "mul.h" + +namespace tnvx +{ + +template +TNVX_INLINE +Mul::Mul(const L& l, const R& r) noexcept + : _l(l) + , _r(r) +{ +} + +template +TNVX_INLINE +__m128 Mul::eval() const noexcept +{ + return _mm_mul_ps(_l.eval(), _r.eval()); +} + +template +TNVX_INLINE +Mul operator*(const E& l, float r) noexcept +{ + return { l, Scalar(r) }; +} + +template +TNVX_INLINE +Mul operator*(float l, const E& r) noexcept +{ + return { Scalar(l), r }; +} + +template +TNVX_INLINE +Mul operator*(const L& l, const R& r) noexcept +{ + return { l, r }; +} + +} diff --git a/src/tenvex/expressions/scalar.h b/src/tenvex/expressions/scalar.h new file mode 100644 index 0000000..a58a36c --- /dev/null +++ b/src/tenvex/expressions/scalar.h @@ -0,0 +1,30 @@ +#pragma once + +#include "defines.h" +#include "expression.h" +#include "traits.h" +#include "concepts.h" + +#include + +namespace tnvx +{ + +struct Scalar : Expr +{ + TNVX_INLINE + explicit Scalar(float value) noexcept; + + [[nodiscard]] TNVX_INLINE + __m128 eval() const noexcept; + +private: + __m128 _value; +}; + +template<> +inline constexpr bool is_scalar_expr = true; + +} + +#include "scalar_impl.hpp" diff --git a/src/tenvex/expressions/scalar_impl.hpp b/src/tenvex/expressions/scalar_impl.hpp new file mode 100644 index 0000000..4db581f --- /dev/null +++ b/src/tenvex/expressions/scalar_impl.hpp @@ -0,0 +1,20 @@ +#pragma once + +#include "scalar.h" + +namespace tnvx +{ + +TNVX_INLINE +Scalar::Scalar(float value) noexcept +{ + _value = _mm_set1_ps(value); +} + +TNVX_INLINE +__m128 Scalar::eval() const noexcept +{ + return _value; +} + +} diff --git a/src/tenvex/expressions/sub.h b/src/tenvex/expressions/sub.h new file mode 100644 index 0000000..65d317f --- /dev/null +++ b/src/tenvex/expressions/sub.h @@ -0,0 +1,36 @@ +#pragma once + +#include "defines.h" +#include "expression.h" +#include "concepts.h" +#include "traits.h" + +#include + +namespace tnvx +{ + +template +struct Sub : Expr> +{ + TNVX_INLINE + Sub(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; + + [[nodiscard]] TNVX_INLINE + __m128 eval() const noexcept; + +private: + tnvx_ref_or_value_t _l; + tnvx_ref_or_value_t _r; +}; + +template +[[nodiscard]] TNVX_INLINE +Sub operator-(const L& l, const R& r) noexcept; + +template +inline constexpr bool is_vec_expr> = true; + +} + +#include "sub_impl.hpp" diff --git a/src/tenvex/expressions/sub_impl.hpp b/src/tenvex/expressions/sub_impl.hpp new file mode 100644 index 0000000..eeebe9c --- /dev/null +++ b/src/tenvex/expressions/sub_impl.hpp @@ -0,0 +1,30 @@ +#pragma once + +#include "sub.h" + +namespace tnvx +{ + +template +TNVX_INLINE +Sub::Sub(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept + : _l(l) + , _r(r) +{ +} + +template +TNVX_INLINE +__m128 Sub::eval() const noexcept +{ + return _mm_sub_ps(_l.eval(), _r.eval()); +} + +template +TNVX_INLINE +Sub operator-(const L& l, const R& r) noexcept +{ + return { l, r }; +} + +} diff --git a/src/tenvex/expressions/traits.h b/src/tenvex/expressions/traits.h new file mode 100644 index 0000000..0a48524 --- /dev/null +++ b/src/tenvex/expressions/traits.h @@ -0,0 +1,22 @@ +#pragma once + +#include + +namespace tnvx +{ + +template +inline constexpr bool is_stored_by_value_v = std::is_trivially_copyable_v + && std::is_trivially_destructible_v + && sizeof(T) <= 16; + +template +using tnvx_ref_or_value_t = std::conditional_t, T, const T&>; + +template +inline constexpr bool is_scalar_expr = false; + +template +inline constexpr bool is_vec_expr = false; + +} diff --git a/src/tenvex/expressions/vec4.h b/src/tenvex/expressions/vec4.h new file mode 100644 index 0000000..b4b52ec --- /dev/null +++ b/src/tenvex/expressions/vec4.h @@ -0,0 +1,41 @@ +#pragma once + +#include "defines.h" +#include "expression.h" +#include "traits.h" +#include "concepts.h" + +#include + +namespace tnvx +{ + +struct alignas(16) vec4 : Expr +{ + TNVX_INLINE + vec4() noexcept = default; + + TNVX_INLINE + vec4(float x, float y, float z, float w = 0) noexcept; + + template + TNVX_INLINE + vec4(const E& e) noexcept; + + [[nodiscaed]] TNVX_INLINE + __m128 eval() const noexcept; + +private: + __m128 _data; +}; + +template<> +inline constexpr bool is_vec_expr = true; + +TNVX_INLINE +bool approx_eq(const vec4& lhs, const vec4& rhs, float eps = 1e-6f) noexcept; + +TNVX_INLINE +bool operator==(const vec4& lhs, const vec4& rhs) noexcept; + +} diff --git a/src/tenvex/expressions/vec4_impl.hpp b/src/tenvex/expressions/vec4_impl.hpp new file mode 100644 index 0000000..5f28270 --- /dev/null +++ b/src/tenvex/expressions/vec4_impl.hpp @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/tenvex/internal/add.h b/src/tenvex/internal/add.h deleted file mode 100644 index f6d52fa..0000000 --- a/src/tenvex/internal/add.h +++ /dev/null @@ -1,28 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Add : Expr> -{ - inline - Add(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - -private: - tnvx_type_storage _l; - tnvx_type_storage _r; -}; - -template -[[nodiscard]] inline -Add operator+(const Expr& l, const Expr& r) noexcept; - -} - -#include "add_impl.hpp" diff --git a/src/tenvex/internal/add_impl.hpp b/src/tenvex/internal/add_impl.hpp deleted file mode 100644 index c124370..0000000 --- a/src/tenvex/internal/add_impl.hpp +++ /dev/null @@ -1,30 +0,0 @@ -#pragma once - -#include "add.h" - -namespace tnvx -{ - -template -Add::Add(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept - : _l(l) - , _r(r) -{ - -} - -template -TNVX_INLINE -__m128 Add::eval() const noexcept -{ - return _mm_add_ps(_l.eval(), _r.eval()); -} - -template -TNVX_INLINE -Add operator+(const Expr& l, const Expr& r) noexcept -{ - return { l.self(), r.self() }; -} - -} // namespace tnvx diff --git a/src/tenvex/internal/cross.h b/src/tenvex/internal/cross.h deleted file mode 100644 index c7b3e94..0000000 --- a/src/tenvex/internal/cross.h +++ /dev/null @@ -1,28 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Cross : Expr> -{ - inline - Cross(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - -private: - tnvx_type_storage _l; - tnvx_type_storage _r; -}; - -template -[[nodiscard]] inline -Cross cross(const Expr& l, const Expr& r) noexcept; - -} - -#include "cross_impl.hpp" diff --git a/src/tenvex/internal/cross_impl.hpp b/src/tenvex/internal/cross_impl.hpp deleted file mode 100644 index 4f723ca..0000000 --- a/src/tenvex/internal/cross_impl.hpp +++ /dev/null @@ -1,47 +0,0 @@ -#pragma once - -#include "cross.h" - -namespace tnvx -{ - -template -TNVX_INLINE -Cross::Cross(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept - : _l(l) - , _r(r) -{ -} - -template -TNVX_INLINE -__m128 Cross::eval() const noexcept -{ - __m128 a = _l.eval(); - __m128 b = _r.eval(); - - // [ay, az, ax, 0] - __m128 a_yzx = _mm_shuffle_ps(a, a, _MM_SHUFFLE(3, 0, 2, 1)); - // [by, bz, bx, 0] - __m128 b_yzx = _mm_shuffle_ps(b, b, _MM_SHUFFLE(3, 0, 2, 1)); - - // [ay*bz, az*bx, ax*by, 0] - __m128 mul0 = _mm_mul_ps(a, b_yzx); - // [az*by, ax*bz, ay*bx, 0] - __m128 mul1 = _mm_mul_ps(a_yzx, b); - - // [ay*bz - az*by, az*bx - ax*bz, ax*by - ay*bx, 0] - __m128 sub = _mm_sub_ps(mul0, mul1); - - // shuffle обратно в [x, y, z, 0] - return _mm_shuffle_ps(sub, sub, _MM_SHUFFLE(3, 0, 2, 1)); -} - -template -Cross cross(const Expr& l, const Expr& r) noexcept -{ - return { l.self(), r.self() }; -} - - -} diff --git a/src/tenvex/internal/dot.h b/src/tenvex/internal/dot.h deleted file mode 100644 index 1727731..0000000 --- a/src/tenvex/internal/dot.h +++ /dev/null @@ -1,39 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Dot : Expr> -{ - inline - Dot(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - - [[nodiscard]] inline - operator float() const noexcept; - -private: - tnvx_type_storage _l; - tnvx_type_storage _r; -}; - -template -[[nodiscard]] inline -Dot dot(const Expr& l, const Expr& r) noexcept; - -template -[[nodiscard]] inline -float operator*(const Dot& l, float r) noexcept; - -template -[[nodiscard]] inline -float operator*(float l, const Dot& r) noexcept; - -} - -#include "dot_impl.hpp" diff --git a/src/tenvex/internal/dot_impl.hpp b/src/tenvex/internal/dot_impl.hpp deleted file mode 100644 index bc3bce5..0000000 --- a/src/tenvex/internal/dot_impl.hpp +++ /dev/null @@ -1,60 +0,0 @@ -#pragma once - -#include "dot.h" - -namespace tnvx -{ - -template -TNVX_INLINE -Dot::Dot(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept - : _l(l) - , _r(r) -{ -} - -template -TNVX_INLINE -__m128 Dot::eval() const noexcept -{ - __m128 a = _l.eval(); - __m128 b = _r.eval(); - - __m128 mul = _mm_mul_ps(a, b); - __m128 shuf = _mm_shuffle_ps(mul, mul, _MM_SHUFFLE(2, 3, 0, 1)); - __m128 sum = _mm_add_ps(mul, shuf); - shuf = _mm_movehl_ps(shuf, sum); - sum = _mm_add_ss(sum, shuf); - return _mm_shuffle_ps(sum, sum, 0); -} - -template -TNVX_INLINE -Dot::operator float() const noexcept -{ - return _mm_cvtss_f32(eval()); -} - -template -TNVX_INLINE -Dot dot(const Expr& l, const Expr& r) noexcept -{ - return { l.self(), r.self() }; -} - -template -TNVX_INLINE -float operator*(const Dot& l, float r) noexcept -{ - return static_cast(l) * r; -} - -template -float operator*(float l, const Dot& r) noexcept -{ - return l * static_cast(r); -} - - - -} diff --git a/src/tenvex/internal/expression.h b/src/tenvex/internal/expression.h deleted file mode 100644 index 4c3a61e..0000000 --- a/src/tenvex/internal/expression.h +++ /dev/null @@ -1,47 +0,0 @@ -#pragma once - -#include "defines.h" - -#include -#include -#include - -namespace tnvx -{ - -template -struct Expr -{ - [[nodiscard]] inline - const Derived& self() const noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - - [[nodiscard]] inline - float operator[](size_t i) const noexcept; -}; - -template -concept expression = requires(T a) -{ - { a.eval() } -> std::same_as<__m128>; -} -&& std::derived_from>; - -struct Scalar; - -template -concept vector_expression = expression && !std::same_as; - -template -constexpr bool tnvx_store_by_value = std::is_trivially_copyable_v - && std::is_trivially_destructible_v - && sizeof(T) <= 16; - -template -using tnvx_type_storage = std::conditional_t, T, const T&>; - -} // namespace tnvx - -#include "expression_impl.hpp" diff --git a/src/tenvex/internal/expression_impl.hpp b/src/tenvex/internal/expression_impl.hpp deleted file mode 100644 index c52d41a..0000000 --- a/src/tenvex/internal/expression_impl.hpp +++ /dev/null @@ -1,31 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -TNVX_INLINE -const Derived& Expr::self() const noexcept -{ - return static_cast(*this); -} - -template -TNVX_INLINE -__m128 Expr::eval() const noexcept -{ - return self().eval(); -} - -template -TNVX_INLINE -float Expr::operator[](size_t i) const noexcept -{ - alignas(16) float tmp[4]; - _mm_store_ps(tmp, eval()); - return tmp[i]; -} - -} // namespace tnvx \ No newline at end of file diff --git a/src/tenvex/internal/magnitude.h b/src/tenvex/internal/magnitude.h deleted file mode 100644 index 678c214..0000000 --- a/src/tenvex/internal/magnitude.h +++ /dev/null @@ -1,30 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Magnitude : Expr> -{ - inline - Magnitude(const E& TNVX_RESTRICT e) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - - [[nodiscard]] inline - operator float() const noexcept; - -private: - tnvx_type_storage _e; -}; - -template -[[nodiscard]] inline -Magnitude magnitude(const Expr& e) noexcept; - -} - -#include "magnitude_impl.hpp" diff --git a/src/tenvex/internal/magnitude_impl.hpp b/src/tenvex/internal/magnitude_impl.hpp deleted file mode 100644 index ce2723a..0000000 --- a/src/tenvex/internal/magnitude_impl.hpp +++ /dev/null @@ -1,37 +0,0 @@ -#pragma once - -#include "magnitude.h" -#include "dot.h" - -namespace tnvx -{ - -template -TNVX_INLINE -Magnitude::Magnitude(const E& TNVX_RESTRICT e) noexcept - : _e(e) -{ -} - -template -TNVX_INLINE -__m128 Magnitude::eval() const noexcept -{ - return _mm_sqrt_ss(Dot(_e, _e).eval()); -} - -template -TNVX_INLINE -Magnitude::operator float() const noexcept -{ - return _mm_cvtss_f32(eval()); -} - -template -TNVX_INLINE -Magnitude magnitude(const Expr& e) noexcept -{ - return { e.self() }; -} - -} diff --git a/src/tenvex/internal/mul.h b/src/tenvex/internal/mul.h deleted file mode 100644 index 3d08947..0000000 --- a/src/tenvex/internal/mul.h +++ /dev/null @@ -1,37 +0,0 @@ -#pragma once - -#include "expression.h" -#include "scalar.h" - -namespace tnvx -{ - -template -struct Mul : Expr> -{ - inline - Mul(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - -private: - tnvx_type_storage _l; - tnvx_type_storage _r; -}; - -//template -//[[nodiscard]] inline -//Mul operator*(const Expr& l, const Expr& r) noexcept; - -template -[[nodiscard]] inline -Mul operator*(const Expr& l, float r) noexcept; - -template -[[nodiscard]] inline -Mul operator*(float l, const Expr& r) noexcept; - -} - -#include "mul_impl.hpp" diff --git a/src/tenvex/internal/mul_impl.hpp b/src/tenvex/internal/mul_impl.hpp deleted file mode 100644 index 8e6c50f..0000000 --- a/src/tenvex/internal/mul_impl.hpp +++ /dev/null @@ -1,44 +0,0 @@ -#pragma once - -#include "mul.h" - -namespace tnvx -{ - -template -TNVX_INLINE -inline Mul::Mul(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept - : _l(l) - , _r(r) -{ -} - -template -TNVX_INLINE -__m128 Mul::eval() const noexcept -{ - return _mm_mul_ps(_l.eval(), _r.eval()); -} - -//template -//TNVX_INLINE -//Mul operator*(const Expr& l, const Expr& r) noexcept -//{ -// return { l.self(), r.self() }; -//} - -template -TNVX_INLINE -Mul operator*(const Expr& l, float r) noexcept -{ - return { l.self(), Scalar(r) }; -} - -template -TNVX_INLINE -Mul operator*(float l, const Expr& r) noexcept -{ - return { Scalar(l), r.self() }; -} - -} diff --git a/src/tenvex/internal/norm.h b/src/tenvex/internal/norm.h deleted file mode 100644 index dd0432d..0000000 --- a/src/tenvex/internal/norm.h +++ /dev/null @@ -1,27 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Norm : Expr> -{ - inline - Norm(const E& TNVX_RESTRICT e) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - -private: - tnvx_type_storage _e; -}; - -template -[[nodiscard]] inline -Norm norm(const Expr& e) noexcept; - -} - -#include "norm_impl.hpp" diff --git a/src/tenvex/internal/norm_impl.hpp b/src/tenvex/internal/norm_impl.hpp deleted file mode 100644 index 6ca617e..0000000 --- a/src/tenvex/internal/norm_impl.hpp +++ /dev/null @@ -1,38 +0,0 @@ -#pragma once - -#include "norm.h" -#include "dot.h" - -namespace tnvx -{ - -template -TNVX_INLINE -Norm::Norm(const E& TNVX_RESTRICT e) noexcept - : _e(e) -{ -} - -template -TNVX_INLINE -__m128 Norm::eval() const noexcept -{ - __m128 data = _e.eval(); - __m128 mul = _mm_mul_ps(data, data); - __m128 shuf = _mm_shuffle_ps(mul, mul, _MM_SHUFFLE(2, 3, 0, 1)); - __m128 sum = _mm_add_ps(mul, shuf); - shuf = _mm_movehl_ps(shuf, sum); - sum = _mm_add_ss(sum, shuf); - __m128 len = _mm_shuffle_ps(sum, sum, 0); - __m128 sqrt_len = _mm_sqrt_ps(len); - return _mm_div_ps(data, sqrt_len); -} - -template -TNVX_INLINE -Norm norm(const Expr& e) noexcept -{ - return { e.self() }; -} - -} diff --git a/src/tenvex/internal/scalar.h b/src/tenvex/internal/scalar.h deleted file mode 100644 index 2fc298c..0000000 --- a/src/tenvex/internal/scalar.h +++ /dev/null @@ -1,25 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -struct Scalar : Expr -{ - inline - Scalar(float value) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - - [[nodiscard]] inline - float operator[](size_t) const noexcept; - -private: - float _value; -}; - -} - -#include "scalar_impl.hpp" diff --git a/src/tenvex/internal/scalar_impl.hpp b/src/tenvex/internal/scalar_impl.hpp deleted file mode 100644 index f4804c7..0000000 --- a/src/tenvex/internal/scalar_impl.hpp +++ /dev/null @@ -1,26 +0,0 @@ -#pragma once - -#include "scalar.h" - -namespace tnvx -{ - -TNVX_INLINE -Scalar::Scalar(float value) noexcept - : _value(value) -{ -} - -[[nodiscard]] TNVX_INLINE -__m128 Scalar::eval() const noexcept -{ - return _mm_set1_ps(_value); -} - -[[nodiscard]] TNVX_INLINE -float Scalar::operator[](size_t) const noexcept -{ - return _value; -} - -} diff --git a/src/tenvex/internal/sub.h b/src/tenvex/internal/sub.h deleted file mode 100644 index a762069..0000000 --- a/src/tenvex/internal/sub.h +++ /dev/null @@ -1,28 +0,0 @@ -#pragma once - -#include "expression.h" - -namespace tnvx -{ - -template -struct Sub : Expr> -{ - inline - Sub(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - -private: - tnvx_type_storage _l; - tnvx_type_storage _r; -}; - -template -[[nodiscard]] inline -Sub operator-(const Expr& l, const Expr& r) noexcept; - -} - -#include "sub_impl.hpp" diff --git a/src/tenvex/internal/sub_impl.hpp b/src/tenvex/internal/sub_impl.hpp deleted file mode 100644 index cbf092e..0000000 --- a/src/tenvex/internal/sub_impl.hpp +++ /dev/null @@ -1,28 +0,0 @@ -#pragma once - -#include "sub.h" - -namespace tnvx -{ - -template -tnvx::Sub::Sub(const L& TNVX_RESTRICT l, const R& TNVX_RESTRICT r) noexcept - : _l(l), _r(r) -{ -} - -template -TNVX_INLINE -__m128 Sub::eval() const noexcept -{ - return _mm_sub_ps(_l.eval(), _r.eval()); -} - -template -TNVX_INLINE -Sub operator-(const Expr& l, const Expr& r) noexcept -{ - return { l.self(), r.self() }; -} - -} diff --git a/src/tenvex/tenvex.h b/src/tenvex/tenvex.h index c10d5f1..b44d4ea 100644 --- a/src/tenvex/tenvex.h +++ b/src/tenvex/tenvex.h @@ -1,12 +1,11 @@ -#pragma once +#pragma once -#include "vec4.h" - -#include "internal/add.h" -#include "internal/sub.h" -#include "internal/mul.h" -#include "internal/dot.h" -#include "internal/cross.h" -#include "internal/norm.h" -#include "internal/magnitude.h" -#include "internal/scalar.h" +#include "expressions/defines.h" +#include "expressions/traits.h" +#include "expressions/expression.h" +#include "expressions/concepts.h" +#include "expressions/add.h" +#include "expressions/sub.h" +#include "expressions/scalar.h" +#include "expressions/mul.h" +#include "expressions/vec4.h" diff --git a/src/tenvex/vec4.h b/src/tenvex/vec4.h deleted file mode 100644 index 03bc9af..0000000 --- a/src/tenvex/vec4.h +++ /dev/null @@ -1,63 +0,0 @@ -#pragma once - -#include "internal/expression.h" - -#include - -#include - -namespace tnvx -{ - -struct alignas(16) vec4 : Expr -{ - vec4() = default; - - inline - vec4(float x, float y, float z, float w) noexcept; - - template - inline - vec4(const Expr& expr) noexcept; - - template - inline - vec4& operator=(const Expr& expr) noexcept; - - [[nodiscard]] inline - __m128 eval() const noexcept; - - [[nodiscard]] inline - auto magnitude() const noexcept; - - [[nodiscard]] inline - auto norm() const noexcept; - - [[nodiscard]] inline - float x() const noexcept; - - [[nodiscard]] inline - float y() const noexcept; - - [[nodiscard]] inline - float z() const noexcept; - - [[nodiscard]] inline - float w() const noexcept; - - [[nodiscard]] inline - std::array data() const noexcept; - -private: - __m128 _data; -}; - -inline -bool operator==(const vec4& a, const vec4& b) noexcept; - -inline -bool approx_eq(const vec4& a, const vec4& b, float eps = 1e-3f) noexcept; - -} - -#include "vec4_impl.hpp" diff --git a/src/tenvex/vec4_impl.hpp b/src/tenvex/vec4_impl.hpp deleted file mode 100644 index 133d419..0000000 --- a/src/tenvex/vec4_impl.hpp +++ /dev/null @@ -1,108 +0,0 @@ -#pragma once - -#include "vec4.h" -#include "internal/magnitude.h" -#include "internal/norm.h" - -namespace tnvx -{ - -template -TNVX_INLINE -vec4::vec4(const Expr& expr) noexcept - : _data(expr.eval()) -{ -} - -template -TNVX_INLINE -vec4& vec4::operator=(const Expr& expr) noexcept -{ - _data = expr.eval(); - return *this; -} - -TNVX_INLINE -vec4::vec4(float x, float y, float z, float w) noexcept -{ - _data = _mm_set_ps(w, z, y, x); -} - -TNVX_INLINE -__m128 vec4::eval() const noexcept -{ - return _data; -} - -TNVX_INLINE -auto vec4::magnitude() const noexcept -{ - return Magnitude(*this); -} - -TNVX_INLINE -auto vec4::norm() const noexcept -{ - return Norm(*this); -} - -//TNVX_INLINE -//Magnitude vec4::magnitude() const noexcept -//{ -// return { *this }; -//} -// -//TNVX_INLINE -//Norm vec4::norm() const noexcept -//{ -// return { *this }; -//} - -TNVX_INLINE -float vec4::x() const noexcept -{ - return _mm_cvtss_f32(_data); -} - -TNVX_INLINE -float vec4::y() const noexcept -{ - return _mm_cvtss_f32(_mm_shuffle_ps(_data, _data, _MM_SHUFFLE(1, 1, 1, 1))); -} - -TNVX_INLINE -float vec4::z() const noexcept -{ - return _mm_cvtss_f32(_mm_shuffle_ps(_data, _data, _MM_SHUFFLE(2, 2, 2, 2))); -} - -TNVX_INLINE -float vec4::w() const noexcept -{ - return _mm_cvtss_f32(_mm_shuffle_ps(_data, _data, _MM_SHUFFLE(3, 3, 3, 3))); -} - -TNVX_INLINE -std::array vec4::data() const noexcept -{ - std::array result; - _mm_storeu_ps(result.data(), _data); - return result; -} - -TNVX_INLINE -bool operator==(const vec4& a, const vec4& b) noexcept -{ - return _mm_movemask_ps(_mm_cmpeq_ps(a.eval(), b.eval())) == 0xF; -} - -TNVX_INLINE -bool approx_eq(const vec4& a, const vec4& b, float eps) noexcept -{ - __m128 epsilon = _mm_set1_ps(eps); - __m128 diff = _mm_sub_ps(a.eval(), b.eval()); - __m128 abs_diff = _mm_andnot_ps(_mm_set1_ps(-0.0f), diff); - return _mm_movemask_ps(_mm_cmplt_ps(abs_diff, epsilon)) == 0xF; -} - -} diff --git a/src/tests/vec4_tests.cpp b/src/tests/vec4_tests.cpp index 5186e30..e817aef 100644 --- a/src/tests/vec4_tests.cpp +++ b/src/tests/vec4_tests.cpp @@ -9,7 +9,7 @@ TEST(vec4, add) { vec4 a = { 1.0f, 2.0f, 3.0f, 4.0f }; vec4 b = { 4.0f, 3.0f, 2.0f, 1.0f }; - vec4 check{ 5.0f, 5.0f, 5.0f, 5.0f }; + vec4 check = { 5.0f, 5.0f, 5.0f, 5.0f }; vec4 result = a + b; EXPECT_TRUE(approx_eq(check, result)); @@ -140,6 +140,19 @@ TEST(vec4, expression_c) EXPECT_TRUE(approx_eq(check, result)); } +TEST(vec4, expression_d) +{ + vec4 a = { 1.0f, 2.0f, 3.0f, 0.0f }; + vec4 b = { 4.0f, 5.0f, 6.0f, 0.0f }; + vec4 c = { 1.0f, 0.0f, 1.0f, 0.0f }; + + vec4 check = { 7.243f, 5.657f, 10.071f, 0.0f }; + + vec4 result = magnitude(a) * dot(b, c) + c * 3.0f; + + EXPECT_TRUE(approx_eq(check, result)); +} + TEST(vec4, cross_anticommutative) { vec4 a = { 1.0f, 0.0f, 0.0f, 0.0f };