Added WeatherMgr::rollNewWeather (#100)

1 files changed, 38 insertions, 0 deletions

diff --git a/src/KingSystem/World/worldWeatherMgr.cpp b/src/KingSystem/World/worldWeatherMgr.cpp
index d43706ba..d3407a83 100644
--- a/src/KingSystem/World/worldWeatherMgr.cpp
+++ b/src/KingSystem/World/worldWeatherMgr.cpp
@@ -1 +1,39 @@
 #include "KingSystem/World/worldWeatherMgr.h"
+#include <random/seadGlobalRandom.h>
+#include "KingSystem/GameData/gdtManager.h"
+#include "KingSystem/World/worldManager.h"
+
+namespace ksys::world {
+
+WeatherType WeatherMgr::rollNewWeather(Climate climate) {
+    auto* wm = Manager::instance();
+    int random = sead::GlobalRandom::instance()->getU32(99);
+    WeatherType weather = [&] {
+        const std::pair<WeatherType, int> rates[] = {
+            {WeatherType::Bluesky, wm->getWeatherBlueskyRate(climate) - 1},
+            {WeatherType::Cloudy, wm->getWeatherCloudyRate(climate)},
+            {WeatherType::Rain, wm->getWeatherRainRate(climate)},
+            {WeatherType::HeavyRain, wm->getWeatherHeavyRainRate(climate)},
+        };
+        for (auto [type, rate] : rates) {
+            if (random <= rate)
+                return type;
+            random -= rate;
+        }
+        return WeatherType::ThunderRain;
+    }();
+
+    if (wm->getEnvMgr()->isWaterRelicRainOn(climate))
+        weather = WeatherType::Bluesky;
+
+    bool plateau_done = false;
+    auto glider_handle = Manager::instance()->getTimeMgr()->isGetPlayerStole2Flag();
+    if (glider_handle != gdt::InvalidHandle) {
+        gdt::Manager::instance()->getBool(glider_handle, &plateau_done, true);
+    }
+
+    if (plateau_done || weather == WeatherType::Bluesky || weather == WeatherType::Cloudy)
+        return weather;
+    return WeatherType::Bluesky;
+}
+}  // namespace ksys::world