Disable Season and TimeFlow on game ready

MafiaHub · Feb 15, 2024 · 65ec031 · 65ec031
1 parent d33bc7b
commit 65ec031
Show file tree

Hide file tree

Showing 7 changed files with 90 additions and 15 deletions.
diff --git a/code/client/CMakeLists.txt b/code/client/CMakeLists.txt
@@ -16,6 +16,7 @@ set(MAFIAMP_CLIENT_FILES
     src/core/ui/web/sdk.cpp
     src/core/ui/web/view.cpp
     src/core/ui/world_debug.cpp
+    src/core/states/game_ready.cpp
     src/core/states/initialize.cpp
     src/core/states/main_menu.cpp
     src/core/states/session_connected.cpp
@@ -39,7 +40,6 @@ set(MAFIAMP_CLIENT_FILES
     src/core/hooks/slot_wrapper.cpp
     src/core/hooks/stream_map.cpp
     src/core/hooks/tables.cpp
-    src/core/hooks/traffic.cpp
     src/core/hooks/translocator.cpp
     src/core/hooks/vehicle.cpp
     src/core/hooks/vfs.cpp

diff --git a/code/client/src/core/application.cpp b/code/client/src/core/application.cpp
@@ -2,6 +2,7 @@
 
 #include <logging/logger.h>
 
+#include "states/game_ready.h"
 #include "states/initialize.h"
 #include "states/main_menu.h"
 #include "states/session_connected.h"
@@ -42,6 +43,7 @@ namespace MafiaMP::Core {
         // Create the state machine and initialize
         _stateMachine = std::make_shared<Framework::Utils::States::Machine>();
         _stateMachine->RegisterState<States::InitializeState>();
+        _stateMachine->RegisterState<States::GameReadyState>();
         _stateMachine->RegisterState<States::MainMenuState>();
         _stateMachine->RegisterState<States::SessionConnectionState>();
         _stateMachine->RegisterState<States::SessionConnectedState>();

diff --git a/code/client/src/core/hooks/traffic.cpp b/code/client/src/core/hooks/traffic.cpp
diff --git a/code/client/src/core/states/game_ready.cpp b/code/client/src/core/states/game_ready.cpp
@@ -0,0 +1,66 @@
+#include "game_ready.h"
+
+#include "states.h"
+
+#include "sdk/ue/game/traffic/c_streaming_traffic_module.h"
+#include "sdk/ue/gfx/environmenteffects/c_gfx_environment_effects.h"
+
+#include <utils/states/machine.h>
+
+namespace MafiaMP::Core::States {
+    GameReadyState::GameReadyState() {}
+
+    GameReadyState::~GameReadyState() {}
+
+    int32_t GameReadyState::GetId() const {
+        return StateIds::GameReady;
+    }
+
+    const char *GameReadyState::GetName() const {
+        return "GameReady";
+    }
+
+    bool GameReadyState::OnEnter(Framework::Utils::States::Machine *) {
+        return true;
+    }
+
+    bool GameReadyState::OnExit(Framework::Utils::States::Machine *machine) {
+        machine->RequestNextState(StateIds::MainMenu);
+        return true;
+    }
+
+    bool GameReadyState::OnUpdate(Framework::Utils::States::Machine *) {
+        const auto streamingTrafficModule = SDK::ue::game::traffic::C_StreamingTrafficModule::GetInstance();
+        if (!streamingTrafficModule) {
+            return false;
+        }
+
+        const auto gfxEnvironmentEffects = SDK::ue::gfx::environmenteffects::C_GfxEnvironmentEffects::GetInstance();
+        if (!gfxEnvironmentEffects) {
+            return false;
+        }
+        const auto weatherManager = gfxEnvironmentEffects->GetWeatherManager();
+        if (!weatherManager) {
+            return false;
+        }
+
+        /**
+         * Disable traffic
+         *
+         * Traffic is automaticaly loaded by the game via C_StreamingTrafficModule::OpenSeason
+         * after C_StreamMap::OpenPart("freeride") is called.
+         *
+         * We close the season here which disable the traffic.
+         */
+        streamingTrafficModule->CloseSeason(true);
+
+        /**
+         * Disable TimeFlow
+         *
+         * We give scripters the option to update the time manually.
+         */
+        weatherManager->SetUserTimeFlowSpeedMult(0);
+
+        return true;
+    }
+} // namespace MafiaMP::Core::States
diff --git a/code/client/src/core/states/game_ready.h b/code/client/src/core/states/game_ready.h
@@ -0,0 +1,19 @@
+#pragma once
+
+#include <utils/states/state.h>
+
+namespace MafiaMP::Core::States {
+    class GameReadyState: public Framework::Utils::States::IState {
+      public:
+        GameReadyState();
+        ~GameReadyState();
+
+        virtual const char *GetName() const override;
+        virtual int32_t GetId() const override;
+
+        virtual bool OnEnter(Framework::Utils::States::Machine *) override;
+        virtual bool OnExit(Framework::Utils::States::Machine *) override;
+
+        virtual bool OnUpdate(Framework::Utils::States::Machine *) override;
+    };
+} // namespace MafiaMP::Core::States
diff --git a/code/client/src/core/states/initialize.cpp b/code/client/src/core/states/initialize.cpp
@@ -24,7 +24,7 @@ namespace MafiaMP::Core::States {
     }
 
     bool InitializeState::OnExit(Framework::Utils::States::Machine *machine) {
-        machine->RequestNextState(StateIds::MainMenu);
+        machine->RequestNextState(StateIds::GameReady);
         return true;
     }
 

diff --git a/code/client/src/core/states/states.h b/code/client/src/core/states/states.h
@@ -3,6 +3,7 @@
 namespace MafiaMP::Core::States {
     enum StateIds: int32_t {
         Initialize,
+        GameReady,
         MainMenu,
         SessionConnection,
         SessionConnected,