Favorite color WIP

Author: kiwi515

lib/libkiwi/core/kiwiColor.cpp

@@ -1,15 +1,63 @@
 #include <libkiwi.h>
 
+#include <cmath>
+
 namespace kiwi {
 
 /**
  * @brief Converts RGB color to YUV format
  */
-Color Color::Yuv() const {
+Color Color::ToYuv() const {
     u8 y = 0.257f * r + 0.504f * g + 0.098f * b + 16.0f;
     u8 u = -0.148f * r - 0.291f * g + 0.439f * b + 128.0f;
     u8 v = 0.439f * r - 0.368f * g - 0.071f * b + 128.0f;
     return Color(y, u, v, a);
 }
 
+/**
+ * @brief Creates RGB color from HSV format
+ *
+ * @param h Hue [0, 1]
+ * @param s Saturation [0, 1]
+ * @param v Value [0, 1]
+ */
+Color Color::FromHsv(f32 h, f32 s, f32 v) {
+    K_ASSERT(h >= 0.0f && h <= 1.0f);
+    K_ASSERT(s >= 0.0f && s <= 1.0f);
+    K_ASSERT(v >= 0.0f && v <= 1.0f);
+
+    f32 r = 0.0f;
+    f32 g = 0.0f;
+    f32 b = 0.0f;
+
+    s32 i = static_cast<s32>(h * 6.0f);
+    f32 f = h * 6.0f - i;
+
+    f32 p = v * (1.0f - s);
+    f32 q = v * (1.0f - f * s);
+    f32 t = v * (1.0f - (1.0f - f) * s);
+
+    // clang-format off
+    switch (i % 6) {
+        case 0: { r = v, g = t, b = p; break; }
+        case 1: { r = q, g = v, b = p; break; }
+        case 2: { r = p, g = v, b = t; break; }
+        case 3: { r = p, g = q, b = v; break; }
+        case 4: { r = t, g = p, b = v; break; }
+        case 5: { r = v, g = p, b = q; break; }
+    }
+    // clang-format on
+
+    r *= 255;
+    g *= 255;
+    b *= 255;
+
+    // clang-format off
+    return Color(static_cast<u8>(r),
+                 static_cast<u8>(g),
+                 static_cast<u8>(b),
+                 255);
+    // clang-format on
+}
+
 } // namespace kiwi

lib/libkiwi/core/kiwiColor.h

@@ -71,7 +71,16 @@ class Color : public nw4r::ut::Color {
     /**
      * @brief Converts RGB color to YUV format
      */
-    Color Yuv() const;
+    Color ToYuv() const;
+
+    /**
+     * @brief Converts HSV color to RGB format
+     *
+     * @param h Hue [0, 1]
+     * @param s Saturation [0, 1]
+     * @param v Value [0, 1]
+     */
+    static Color FromHsv(f32 h, f32 s, f32 v);
 
 public:
     /**

lib/libkiwi/debug/kiwiNw4rDirectPrint.cpp

@@ -203,7 +203,7 @@ void Nw4rDirectPrint::DrawString(s32 x, s32 y, const char* pMsg, ...) const {
  */
 void Nw4rDirectPrint::SetColor(Color rgb) {
     // Framebuffer uses YUV format, so we convert the color
-    mBufferColor = rgb.Yuv();
+    mBufferColor = rgb.ToYuv();
 }
 
 /**

lib/libkiwi/util/kiwiRandom.h

@@ -112,6 +112,17 @@ class Random {
         return NextF32() * max;
     }
 
+    /**
+     * @brief Get random float (lower+upper bound)
+     *
+     * @param min Lower bound (inclusive)
+     * @param max Upper bound (exclusive)
+     */
+    f32 NextF32(f32 min, f32 max) {
+        K_ASSERT(min < max);
+        return min + NextF32(max - min);
+    }
+
     /**
      * @brief Roll random chance
      *

src/hooks/Cmn/cosmetics_Cmn.cpp

@@ -76,6 +76,17 @@ RPSysScene::ETime InterceptIslandTime(RPSysScene* pScene) {
 }
 KM_CALL(0x8026a56c, InterceptIslandTime);
 
+/**
+ * @brief Overrides the Mii favorite color based on the randomizer settings
+ */
+u32 InterceptFavoriteColor() {
+    return kiwi::Color::FromHsv(kiwi::Random().NextF32(1.0f),
+                                kiwi::Random().NextF32(0.85f, 0.9f),
+                                kiwi::Random().NextF32(0.5f, 0.9f));
+}
+
+KM_BRANCH(0x80102a20, InterceptFavoriteColor);
+
 } // namespace Cosmetic
 } // namespace Cmn
 } // namespace AP