Vehicle Upgrades Randomizer: Stability Improvements

* Vehicle upgrades are applied on-at-a-time now to make sure the game
  doesn't get too many stream requests at the same time.

* In addition, the stream request pool is also checked to make sure it
  has reasonable free slots before requesting more vehicle upgrades.

* Added License Plate Randomization using data file with names of
  contributors, patrons and others.

Author: Parik27

build-count.txt

@@ -1 +1 @@
-524
+530

cmake/DebugImgui.cmake

@@ -1,9 +1,10 @@
 include(cmake/imgui.cmake)
+include(cmake/implot.cmake)
 include(cmake/fmt.cmake)
 
 file(GLOB_RECURSE SOURCES CONFIGURE_DEPENDS src/debug/*.cc)
 
 target_sources(${PROJECT_NAME} PUBLIC ${SOURCES})
 target_compile_definitions(${PROJECT_NAME} PRIVATE -DENABLE_DEBUG_MENU)
-target_link_libraries (${PROJECT_NAME} PUBLIC fmt imgui dxguid dwmapi imm32)
+target_link_libraries (${PROJECT_NAME} PUBLIC fmt imgui dxguid dwmapi imm32 implot)
 

cmake/implot.cmake

@@ -0,0 +1,22 @@
+include(FetchContent)
+FetchContent_Declare(LIB_implot
+  GIT_REPOSITORY https://github.com/epezent/implot
+  GIT_TAG v0.12)
+
+message("Fetching implot from GitHub...")
+
+FetchContent_MakeAvailable (LIB_implot)
+FetchContent_GetProperties(LIB_implot
+  SOURCE_DIR LIB_implot_SOURCE_DIR)
+
+file(GLOB SOURCES CONFIGURE_DEPENDS
+  ${LIB_implot_SOURCE_DIR}/implot.cpp
+  ${LIB_implot_SOURCE_DIR}/implot_items.cpp
+  )
+
+add_library (implot STATIC ${SOURCES})
+target_include_directories(implot PUBLIC
+  ${LIB_implot_SOURCE_DIR}
+  )
+
+target_link_libraries(implot PUBLIC imgui)

data/LicensePlates.txt

@@ -0,0 +1,37 @@
+P4R1K
+FRYT3RP
+123R0B0
+5P33DY
+M0X1
+HUG0 0N3
+G1B5T4CK
+
+N4BN00B
+M4T3CZK
+V31G4R
+1GU4N4
+5R3W0
+R13K3LT
+W4R15
+M4DM4N
+
+DV1P3R4U
+H4YW1R3
+JUR4YJ
+53RR4N0
+5H1N3
+T4R45
+
+DRDR3
+L0RDM4U5
+M1CH43L
+FR4NKL1N
+L4M4R
+B1RD
+P1G
+C4T
+D0G
+VR00M
+GT4K1D
+N1K0
+R0M4N

lib/CPools.cc

@@ -11,6 +11,8 @@ CPools::InitialisePatterns ()
 
     g_pVehicleStructPool = GetRelativeReference<CPool<CVehicleStruct> *> (
         "c7 44 ? ? ? ? ? ? 83 64 ? ? 00 e8 ? ? ? ? ? 89 ? ? ? ? ? eb", 21, 25);
-}
 
-CPool<CVehicleStruct> **CPools::g_pVehicleStructPool;
+    g_pVehicleStreamRequestGfxPool
+        = GetRelativeReference<CPool<CVehicleStreamRequestGfx> *> (
+            "8b 15 ? ? ? ? ? 8b ? ? 83 c1 10 e8 ? ? ? ? ? 8b 0d", 21, 25);
+}

lib/CPools.hh

@@ -22,6 +22,13 @@ public:
         return m_nCount & 0x3fffffff;
     }
 
+    /* Returns number of free slots in the pool */
+    uint32_t
+    GetFree ()
+    {
+        return m_nMaxElements - GetCount ();
+    }
+
     // Because why not, right? :P
     class CPoolIterator
     {
@@ -138,17 +145,26 @@ public:
 static_assert (sizeof (CPool<void *>) == 40);
 
 struct CVehicleStruct;
+struct CVehicleStreamRequestGfx;
 
 class CPools
 {
 public:
-    static CPool<CVehicleStruct> **g_pVehicleStructPool;
+    inline static CPool<CVehicleStruct> **g_pVehicleStructPool;
+    inline static CPool<CVehicleStreamRequestGfx> *
+        *g_pVehicleStreamRequestGfxPool;
 
     inline static CPool<CVehicleStruct> *&
     GetVehicleStructPool ()
     {
         return *g_pVehicleStructPool;
     };
 
+    inline static auto *&
+    GetVehicleStreamRequestGxtPool ()
+    {
+        return *g_pVehicleStreamRequestGfxPool;
+    };
+
     static void InitialisePatterns ();
 };

lib/HSL.hh

@@ -1,3 +1,5 @@
+#pragma once
+
 #include "CMath.hh"
 #include "CARGB.hh"
 #include <cmath>

src/common/parser.hh

@@ -143,6 +143,68 @@ public:
     }
 };
 
+// Data file containing strings
+template <const char *FileName> class DataFileSourcedRandomizer
+{
+    using Type = std::string;
+
+private:
+    struct ValueGroup
+    {
+        std::vector<Type>   Values;
+        std::vector<double> Weights;
+    };
+
+    bool       m_Initialised = false;
+    ValueGroup m_List;
+
+    /*******************************************************/
+    void
+    Initialise ()
+    {
+        if (std::exchange (m_Initialised, true))
+            return;
+
+        FILE *file = Rainbomizer::Common::GetRainbomizerDataFile (FileName);
+        if (!file)
+            return;
+
+        char line[512] = {0};
+        while (fgets (line, 512, file))
+            {
+                if (strlen (line) < 3)
+                    continue;
+
+                double weight     = 1.0;
+                char   model[256] = {0};
+
+                sscanf (line, "%s %lf ", model, &weight);
+
+                m_List.Values.push_back (model);
+                m_List.Weights.push_back (weight);
+            }
+    }
+
+public:
+    /*******************************************************/
+    bool
+    RandomizeObject (Type &out)
+    {
+        Initialise ();
+
+        if (m_List.Values.size ())
+            out = m_List.Values[RandomWeighed (m_List.Weights)];
+
+        return m_List.Values.size ();
+    }
+
+    /*******************************************************/
+    void
+    AddSample (const Type value)
+    {
+    }
+};
+
 template <typename T, T... Values> class ConstantValues
 {
     constexpr static std::array<T, sizeof...(Values)> m_Values{Values...};
@@ -341,16 +403,16 @@ public:
     void
     AddSample (T *base, uint32_t hash)
     {
-        std::apply ([base,
-                     hash] (auto &...x) { (..., AddSample (x, base, hash)); },
+        std::apply ([base, hash,
+                     this] (auto &...x) { (..., AddSample (x, base, hash)); },
                     randomizers);
     }
 
     template <typename T>
     void
     RandomizeObject (T *base, uint32_t hash)
     {
-        std::apply ([base, hash] (
+        std::apply ([base, hash, this] (
                         auto &...x) { (..., RandomizeObject (x, base, hash)); },
                     randomizers);
     }

src/debug/backend.cc

@@ -6,6 +6,7 @@
 #include <windows.h>
 
 #include <imgui.h>
+#include <implot.h>
 #include <imgui_impl_win32.h>
 #include <imgui_impl_dx11.h>
 
@@ -146,6 +147,7 @@ class DebugServerBackend
                 device->GetImmediateContext (&context);
 
                 ImGui::CreateContext ();
+                ImPlot::CreateContext ();
                 ImGui::StyleColorsDark ();
 
                 HWND window = FindMainWindow (GetCurrentProcessId ());

src/debug/config.hh

@@ -51,6 +51,8 @@ private:
                     i.second);
                 ImGui::PopID ();
                 ImGui::NextColumn ();
+
+                ImGui::Columns ();
             }
     }