From 6dd72d2448d568e87b40943985a01cba09af1d86 Mon Sep 17 00:00:00 2001
From: Orange <orange-cpp@yandex.ru>
Date: Mon, 17 Mar 2025 03:56:09 +0300
Subject: [PATCH] added convertors

---
 CMakeLists.txt            | 13 +++++++++----
 include/omath/Vector2.hpp | 34 ++++++++++++++++++++++++++++------
 include/omath/Vector3.hpp |  9 ++++-----
 include/omath/Vector4.hpp | 18 ++++++++++++++++++
 4 files changed, 59 insertions(+), 15 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index a8229e3..3c94fc3 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -9,7 +9,7 @@ option(OMATH_BUILD_TESTS "Build unit tests" OFF)
 option(OMATH_THREAT_WARNING_AS_ERROR "Set highest level of warnings and force compiler to treat them as errors" ON)
 option(OMATH_BUILD_AS_SHARED_LIBRARY "Build Omath as .so or .dll" OFF)
 option(OMATH_USE_AVX2 "Omath will use AVX2 to boost performance" ON)
-
+option(OMATH_IMGUI_INTEGRATION "Omath will define method to convert omath types to imgui types" ON)
 
 if (OMATH_BUILD_AS_SHARED_LIBRARY)
     add_library(omath SHARED source/Vector3.cpp)
@@ -17,6 +17,14 @@ else()
     add_library(omath STATIC source/Matrix.cpp)
 endif()
 
+if (OMATH_IMGUI_INTEGRATION)
+    target_compile_definitions(omath PUBLIC OMATH_IMGUI_INTEGRATION)
+endif()
+
+if (OMATH_USE_AVX2)
+    target_compile_definitions(omath PUBLIC OMATH_USE_AVX2)
+endif()
+
 set_target_properties(omath PROPERTIES
         ARCHIVE_OUTPUT_DIRECTORY "${CMAKE_SOURCE_DIR}/out/${CMAKE_BUILD_TYPE}"
         LIBRARY_OUTPUT_DIRECTORY "${CMAKE_SOURCE_DIR}/out/${CMAKE_BUILD_TYPE}"
@@ -31,9 +39,6 @@ endif()
 
 target_compile_features(omath PUBLIC cxx_std_23)
 
-if (OMATH_USE_AVX2)
-    target_compile_definitions(omath PUBLIC OMATH_USE_AVX2)
-endif()
 
 add_subdirectory(source)
 
diff --git a/include/omath/Vector2.hpp b/include/omath/Vector2.hpp
index 60d2ed5..5873eb7 100644
--- a/include/omath/Vector2.hpp
+++ b/include/omath/Vector2.hpp
@@ -3,13 +3,19 @@
 //
 
 #pragma once
-#include <tuple>
 #include <cmath>
+#include <tuple>
+
+#ifdef OMATH_IMGUI_INTEGRATION
+    class ImVec2;
+#endif
+
 
 namespace omath
 {
 
-    template<class Type> requires std::is_arithmetic_v<Type>
+    template<class Type>
+        requires std::is_arithmetic_v<Type>
     class Vector2
     {
     public:
@@ -19,7 +25,9 @@ namespace omath
         // Constructors
         constexpr Vector2() = default;
 
-        constexpr Vector2(const Type& x, const Type& y) : x(x), y(y) {}
+        constexpr Vector2(const Type& x, const Type& y) : x(x), y(y)
+        {
+        }
 
         // Equality operators
         [[nodiscard]]
@@ -145,7 +153,7 @@ namespace omath
 
         constexpr Vector2& Abs()
         {
-            //FIXME: Replace with std::abs, if it will become constexprable
+            // FIXME: Replace with std::abs, if it will become constexprable
             x = x < 0 ? -x : x;
             y = y < 0 ? -y : y;
             return *this;
@@ -188,7 +196,7 @@ namespace omath
             return {x / fl, y / fl};
         }
 
-         // Sum of elements
+        // Sum of elements
         [[nodiscard]] constexpr Type Sum() const
         {
             return x + y;
@@ -199,5 +207,19 @@ namespace omath
         {
             return std::make_tuple(x, y);
         }
+
+#ifdef OMATH_IMGUI_INTEGRATION
+        [[nodiscard]]
+        const ImVec2& ToImVec2() const
+        {
+            return *reinterpret_cast<const ImVec2*>(this);
+        }
+
+        [[nodiscard]]
+        ImVec2& ToImVec2()
+        {
+            return *reinterpret_cast<ImVec2*>(this);
+        }
+#endif
     };
-}
\ No newline at end of file
+} // namespace omath
diff --git a/include/omath/Vector3.hpp b/include/omath/Vector3.hpp
index 588c608..c3ffcd0 100644
--- a/include/omath/Vector3.hpp
+++ b/include/omath/Vector3.hpp
@@ -4,13 +4,12 @@
 
 #pragma once
 
+#include "Vector2.hpp"
 #include <cstdint>
-#include <functional>
-#include "omath/Vector2.hpp"
-#include "omath/Angle.hpp"
 #include <expected>
-#include <immintrin.h>
-
+#include <functional>
+#include "omath/Angle.hpp"
+#include "omath/Vector2.hpp"
 
 namespace omath
 {
diff --git a/include/omath/Vector4.hpp b/include/omath/Vector4.hpp
index f4fcadd..f927445 100644
--- a/include/omath/Vector4.hpp
+++ b/include/omath/Vector4.hpp
@@ -6,6 +6,10 @@
 #include <omath/Vector3.hpp>
 #include <algorithm>
 
+#ifdef OMATH_IMGUI_INTEGRATION
+    class ImVec4;
+#endif
+
 namespace omath
 {
     template <class Type>
@@ -154,5 +158,19 @@ namespace omath
         {
             return Vector3<Type>::Sum() + w;
         }
+
+#ifdef OMATH_IMGUI_INTEGRATION
+        [[nodiscard]]
+        constexpr ImVec4& ToImVec4() const
+        {
+            return *reinterpret_cast<const ImVec4*>(this);
+        }
+
+        [[nodiscard]]
+        constexpr ImVec2& ToImVec2()
+        {
+            return *reinterpret_cast<ImVec4*>(this);
+        }
+#endif
     };
 }