1 files changed, 279 insertions, 0 deletions
diff --git a/Source/Core/VideoCommon/GraphicsModSystem/Runtime/GraphicsModManager.cpp b/Source/Core/VideoCommon/GraphicsModSystem/Runtime/GraphicsModManager.cpp
new file mode 100644
index 0000000000..9e3d3e9e99
--- /dev/null
+++ b/Source/Core/VideoCommon/GraphicsModSystem/Runtime/GraphicsModManager.cpp
@@ -0,0 +1,279 @@
+// Copyright 2022 Dolphin Emulator Project
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#include "VideoCommon/GraphicsModSystem/Runtime/GraphicsModManager.h"
+
+#include <string>
+#include <string_view>
+#include <variant>
+
+#include "Common/Logging/Log.h"
+#include "Common/VariantUtil.h"
+
+#include "VideoCommon/GraphicsModSystem/Config/GraphicsMod.h"
+#include "VideoCommon/GraphicsModSystem/Config/GraphicsModGroup.h"
+#include "VideoCommon/GraphicsModSystem/Runtime/GraphicsModActionFactory.h"
+#include "VideoCommon/TextureInfo.h"
+
+class GraphicsModManager::DecoratedAction final : public GraphicsModAction
+{
+public:
+  DecoratedAction(std::unique_ptr<GraphicsModAction> action, GraphicsModConfig mod)
+      : m_action_impl(std::move(action)), m_mod(std::move(mod))
+  {
+  }
+  void OnDrawStarted(bool* skip) override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnDrawStarted(skip);
+  }
+  void OnEFB(bool* skip, u32 texture_width, u32 texture_height, u32* scaled_width,
+             u32* scaled_height) override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnEFB(skip, texture_width, texture_height, scaled_width, scaled_height);
+  }
+  void OnProjection(Common::Matrix44* matrix) override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnProjection(matrix);
+  }
+  void OnProjectionAndTexture(Common::Matrix44* matrix) override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnProjectionAndTexture(matrix);
+  }
+  void OnTextureLoad() override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnTextureLoad();
+  }
+  void OnFrameEnd() override
+  {
+    if (!m_mod.m_enabled)
+      return;
+    m_action_impl->OnFrameEnd();
+  }
+
+private:
+  GraphicsModConfig m_mod;
+  std::unique_ptr<GraphicsModAction> m_action_impl;
+};
+
+const std::vector<GraphicsModAction*>&
+GraphicsModManager::GetProjectionActions(ProjectionType projection_type) const
+{
+  if (const auto it = m_projection_target_to_actions.find(projection_type);
+      it != m_projection_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+const std::vector<GraphicsModAction*>&
+GraphicsModManager::GetProjectionTextureActions(ProjectionType projection_type,
+                                                const std::string& texture_name) const
+{
+  const auto lookup = fmt::format("{}_{}", texture_name, static_cast<int>(projection_type));
+  if (const auto it = m_projection_texture_target_to_actions.find(lookup);
+      it != m_projection_texture_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+const std::vector<GraphicsModAction*>&
+GraphicsModManager::GetDrawStartedActions(const std::string& texture_name) const
+{
+  if (const auto it = m_draw_started_target_to_actions.find(texture_name);
+      it != m_draw_started_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+const std::vector<GraphicsModAction*>&
+GraphicsModManager::GetTextureLoadActions(const std::string& texture_name) const
+{
+  if (const auto it = m_load_target_to_actions.find(texture_name);
+      it != m_load_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+const std::vector<GraphicsModAction*>& GraphicsModManager::GetEFBActions(const FBInfo& efb) const
+{
+  if (const auto it = m_efb_target_to_actions.find(efb); it != m_efb_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+const std::vector<GraphicsModAction*>& GraphicsModManager::GetXFBActions(const FBInfo& xfb) const
+{
+  if (const auto it = m_efb_target_to_actions.find(xfb); it != m_efb_target_to_actions.end())
+  {
+    return it->second;
+  }
+
+  return m_default;
+}
+
+void GraphicsModManager::Load(const GraphicsModGroupConfig& config)
+{
+  Reset();
+
+  const auto& mods = config.GetMods();
+
+  std::map<std::string, std::vector<GraphicsTargetConfig>> group_to_targets;
+  for (const auto& mod : mods)
+  {
+    for (const GraphicsTargetGroupConfig& group : mod.m_groups)
+    {
+      if (m_groups.find(group.m_name) != m_groups.end())
+      {
+        WARN_LOG_FMT(
+            VIDEO,
+            "Specified graphics mod group '{}' for mod '{}' is already specified by another mod.",
+            group.m_name, mod.m_title);
+      }
+      m_groups.insert(group.m_name);
+
+      const auto internal_group = fmt::format("{}.{}", mod.m_title, group.m_name);
+      for (const GraphicsTargetConfig& target : group.m_targets)
+      {
+        group_to_targets[group.m_name].push_back(target);
+        group_to_targets[internal_group].push_back(target);
+      }
+    }
+  }
+
+  for (const auto& mod : mods)
+  {
+    for (const GraphicsModFeatureConfig& feature : mod.m_features)
+    {
+      const auto create_action = [](const std::string_view& action_name,
+                                    const picojson::value& json_data,
+                                    GraphicsModConfig mod) -> std::unique_ptr<GraphicsModAction> {
+        auto action = GraphicsModActionFactory::Create(action_name, json_data);
+        if (action == nullptr)
+        {
+          return nullptr;
+        }
+        return std::make_unique<DecoratedAction>(std::move(action), std::move(mod));
+      };
+
+      const auto internal_group = fmt::format("{}.{}", mod.m_title, feature.m_group);
+
+      const auto add_target = [&](const GraphicsTargetConfig& target, GraphicsModConfig mod) {
+        auto action = create_action(feature.m_action, feature.m_action_data, std::move(mod));
+        if (action == nullptr)
+        {
+          WARN_LOG_FMT(VIDEO, "Failed to create action '{}' for group '{}'.", feature.m_action,
+                       feature.m_group);
+          return;
+        }
+        m_actions.push_back(std::move(action));
+        std::visit(
+            overloaded{
+                [&](const DrawStartedTextureTarget& the_target) {
+                  m_draw_started_target_to_actions[the_target.m_texture_info_string].push_back(
+                      m_actions.back().get());
+                },
+                [&](const LoadTextureTarget& the_target) {
+                  m_load_target_to_actions[the_target.m_texture_info_string].push_back(
+                      m_actions.back().get());
+                },
+                [&](const EFBTarget& the_target) {
+                  FBInfo info;
+                  info.m_height = the_target.m_height;
+                  info.m_width = the_target.m_width;
+                  info.m_texture_format = the_target.m_texture_format;
+                  m_efb_target_to_actions[info].push_back(m_actions.back().get());
+                },
+                [&](const XFBTarget& the_target) {
+                  FBInfo info;
+                  info.m_height = the_target.m_height;
+                  info.m_width = the_target.m_width;
+                  info.m_texture_format = the_target.m_texture_format;
+                  m_xfb_target_to_actions[info].push_back(m_actions.back().get());
+                },
+                [&](const ProjectionTarget& the_target) {
+                  if (the_target.m_texture_info_string)
+                  {
+                    const auto lookup = fmt::format("{}_{}", *the_target.m_texture_info_string,
+                                                    static_cast<int>(the_target.m_projection_type));
+                    m_projection_texture_target_to_actions[lookup].push_back(
+                        m_actions.back().get());
+                  }
+                  else
+                  {
+                    m_projection_target_to_actions[the_target.m_projection_type].push_back(
+                        m_actions.back().get());
+                  }
+                },
+            },
+            target);
+      };
+
+      // Prefer groups in the pack over groups from another pack
+      if (const auto local_it = group_to_targets.find(internal_group);
+          local_it != group_to_targets.end())
+      {
+        for (const GraphicsTargetConfig& target : local_it->second)
+        {
+          add_target(target, mod);
+        }
+      }
+      else if (const auto global_it = group_to_targets.find(feature.m_group);
+               global_it != group_to_targets.end())
+      {
+        for (const GraphicsTargetConfig& target : global_it->second)
+        {
+          add_target(target, mod);
+        }
+      }
+      else
+      {
+        WARN_LOG_FMT(VIDEO, "Specified graphics mod group '{}' was not found for mod '{}'",
+                     feature.m_group, mod.m_title);
+      }
+    }
+  }
+}
+
+void GraphicsModManager::EndOfFrame()
+{
+  for (auto&& action : m_actions)
+  {
+    action->OnFrameEnd();
+  }
+}
+
+void GraphicsModManager::Reset()
+{
+  m_actions.clear();
+  m_groups.clear();
+  m_projection_target_to_actions.clear();
+  m_projection_texture_target_to_actions.clear();
+  m_draw_started_target_to_actions.clear();
+  m_load_target_to_actions.clear();
+  m_efb_target_to_actions.clear();
+  m_xfb_target_to_actions.clear();
+}