Write all menu function definitions on top of every menu list item

src/ObjWriting/Game/IW4/AssetDumpers/AssetDumperMenuList.cpp

@@ -1,8 +1,11 @@
 #include "AssetDumperMenuList.h"
 
+#include <cassert>
 #include <filesystem>
 #include <sstream>
+#include <set>
 
+#include "ObjWriting.h"
 #include "Game/IW4/Menu/MenuDumperIW4.h"
 #include "Menu/AbstractMenuDumper.h"
 
@@ -10,28 +13,70 @@ namespace fs = std::filesystem;
 
 using namespace IW4;
 
-bool AssetDumperMenuList::ShouldDump(XAssetInfo<MenuList>* asset)
+std::vector<const ExpressionSupportingData*> AssetDumperMenuList::GetAllUniqueExpressionSupportingData(const MenuList* menuList)
 {
-    return true;
+    std::vector<const ExpressionSupportingData*> result;
+    std::set<const ExpressionSupportingData*> alreadyAddedSupportingData;
+
+    if (menuList->menus == nullptr)
+        return result;
+
+    for(auto i = 0; i < menuList->menuCount; i++)
+    {
+        if(menuList->menus[i] == nullptr)
+            continue;
+
+        const auto* menu = menuList->menus[i];
+
+        if(menu->expressionData == nullptr)
+            continue;
+
+        if(alreadyAddedSupportingData.find(menu->expressionData) == alreadyAddedSupportingData.end())
+        {
+            result.push_back(menu->expressionData);
+            alreadyAddedSupportingData.emplace(menu->expressionData);
+        }
     }
 
-void AssetDumperMenuList::DumpAsset(AssetDumpingContext& context, XAssetInfo<MenuList>* asset)
+    return result;
-{
+}
-    const auto* menuList = asset->Asset();
-    const auto assetFile = context.OpenAssetFile(asset->m_name);
-    const fs::path p(asset->Asset()->name);
 
-    if (!assetFile)
+void AssetDumperMenuList::DumpFunctions(MenuDumper& menuDumper, const MenuList* menuList)
-        return;
+{
+    const auto allSupportingData = GetAllUniqueExpressionSupportingData(menuList);
+    auto functionIndex = 0u;
+
+    assert(allSupportingData.size() <= 1);
+
+    for (const auto* supportingData : allSupportingData)
+    {
+        if (supportingData->uifunctions.functions == nullptr)
+            continue;
+
+        for(auto i = 0; i < supportingData->uifunctions.totalFunctions; i++)
+        {
+            const auto* function = supportingData->uifunctions.functions[i];
+            if(function == nullptr)
+                continue;
+
+            std::stringstream ss;
+            ss << "FUNC_" << functionIndex;
+
+            menuDumper.WriteFunctionDef(ss.str(), function);
+
+            functionIndex++;
+        }
+    }
+}
+
+void AssetDumperMenuList::DumpMenus(MenuDumper& menuDumper, const MenuList* menuList)
+{
+    const fs::path p(menuList->name);
 
     std::string parentPath;
     if (p.has_parent_path())
         parentPath = p.parent_path().string() + "/";
 
-    MenuDumper menuDumper(*assetFile);
-
-    menuDumper.Start();
-
     for (auto menuNum = 0; menuNum < menuList->menuCount; menuNum++)
     {
         const auto* menu = menuList->menus[menuNum];
@@ -47,6 +92,29 @@ void AssetDumperMenuList::DumpAsset(AssetDumpingContext& context, XAssetInfo<Men
         else
             menuDumper.IncludeMenu(ss.str());
     }
+}
+
+bool AssetDumperMenuList::ShouldDump(XAssetInfo<MenuList>* asset)
+{
+    return true;
+}
+
+void AssetDumperMenuList::DumpAsset(AssetDumpingContext& context, XAssetInfo<MenuList>* asset)
+{
+    const auto* menuList = asset->Asset();
+    const auto assetFile = context.OpenAssetFile(asset->m_name);
+
+    if (!assetFile)
+        return;
+
+    MenuDumper menuDumper(*assetFile);
+
+    menuDumper.Start();
+
+    if(!ObjWriting::Configuration.MenuLegacyMode)
+        DumpFunctions(menuDumper, menuList);
+
+    DumpMenus(menuDumper, menuList);
 
     menuDumper.End();
 }

src/ObjWriting/Game/IW4/AssetDumpers/AssetDumperMenuList.h

@@ -2,11 +2,17 @@
 
 #include "Dumping/AbstractAssetDumper.h"
 #include "Game/IW4/IW4.h"
+#include "Game/IW4/Menu/MenuDumperIW4.h"
 
 namespace IW4
 {
     class AssetDumperMenuList final : public AbstractAssetDumper<MenuList>
     {
+        static std::vector<const ExpressionSupportingData*> GetAllUniqueExpressionSupportingData(const MenuList* menuList);
+
+        static void DumpFunctions(MenuDumper& menuDumper, const MenuList* menuList);
+        static void DumpMenus(MenuDumper& menuDumper, const MenuList* menuList);
+
     protected:
         bool ShouldDump(XAssetInfo<MenuList>* asset) override;
         void DumpAsset(AssetDumpingContext& context, XAssetInfo<MenuList>* asset) override;

src/ObjWriting/Game/IW4/Menu/MenuDumperIW4.cpp

@@ -776,6 +776,16 @@ MenuDumper::MenuDumper(std::ostream& stream)
 {
 }
 
+void MenuDumper::WriteFunctionDef(const std::string& functionName, const Statement_s* statement)
+{
+    StartFunctionDefScope();
+
+    WriteStringProperty("name", functionName);
+    WriteStatementProperty("value", statement, false);
+
+    EndScope();
+}
+
 void MenuDumper::WriteMenu(const menuDef_t* menu)
 {
     StartMenuDefScope();

src/ObjWriting/Game/IW4/Menu/MenuDumperIW4.h

@@ -48,6 +48,7 @@ namespace IW4
     public:
         explicit MenuDumper(std::ostream& stream);
 
+        void WriteFunctionDef(const std::string& functionName, const Statement_s* statement);
         void WriteMenu(const menuDef_t* menu);
     };
 }