From 91825578cbdfe88b93191908369e545e53c75ae2 Mon Sep 17 00:00:00 2001
From: iwubcode <iwubcode@users.noreply.github.com>
Date: Mon, 24 Jul 2023 00:40:11 -0500
Subject: [PATCH] VideoCommon: add graphics mod editor types

---
 .../GraphicsModEditor/EditorTypes.h           | 192 ++++++++++++++++++
 1 file changed, 192 insertions(+)
 create mode 100644 Source/Core/VideoCommon/GraphicsModEditor/EditorTypes.h

diff --git a/Source/Core/VideoCommon/GraphicsModEditor/EditorTypes.h b/Source/Core/VideoCommon/GraphicsModEditor/EditorTypes.h
new file mode 100644
index 0000000000..46098b754f
--- /dev/null
+++ b/Source/Core/VideoCommon/GraphicsModEditor/EditorTypes.h
@@ -0,0 +1,192 @@
+// Copyright 2023 Dolphin Emulator Project
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+#pragma once
+
+#include <chrono>
+#include <filesystem>
+#include <memory>
+#include <string>
+#include <variant>
+#include <vector>
+
+#include <imgui.h>
+#include <misc/cpp/imgui_stdlib.h>
+
+#include "Common/CommonTypes.h"
+
+#include "VideoCommon/AbstractTexture.h"
+#include "VideoCommon/Assets/CustomAssetLibrary.h"
+#include "VideoCommon/Assets/CustomTextureData.h"
+#include "VideoCommon/Assets/DirectFilesystemAssetLibrary.h"
+#include "VideoCommon/Assets/MaterialAsset.h"
+#include "VideoCommon/Assets/MeshAsset.h"
+#include "VideoCommon/Assets/ShaderAsset.h"
+#include "VideoCommon/GraphicsModSystem/Runtime/GraphicsModAction.h"
+#include "VideoCommon/GraphicsModSystem/Types.h"
+#include "VideoCommon/XFMemory.h"
+
+namespace GraphicsModEditor
+{
+struct DrawCallData
+{
+  GraphicsModSystem::DrawCallID m_id;
+  std::chrono::steady_clock::time_point m_create_time;
+  std::chrono::steady_clock::time_point m_last_update_time;
+
+  GraphicsModSystem::DrawData draw_data;
+};
+
+struct TextureCacheData
+{
+  GraphicsModSystem::TextureCacheID m_id;
+  std::chrono::steady_clock::time_point m_create_time;
+  std::chrono::steady_clock::time_point m_last_load_time;
+  bool m_active = false;
+
+  GraphicsModSystem::TextureView texture;
+};
+
+struct LightData
+{
+  GraphicsModSystem::LightID m_id;
+  std::chrono::steady_clock::time_point m_create_time;
+  std::chrono::steady_clock::time_point m_last_update_time;
+
+  int4 m_color;
+  float4 m_cosatt;
+  float4 m_distatt;
+  float4 m_pos;
+  float4 m_dir;
+};
+
+struct DrawCallUserData
+{
+  std::string m_friendly_name;
+  std::vector<std::string> m_tag_names;
+};
+
+struct TextureCacheUserData
+{
+  std::string m_friendly_name;
+  std::vector<std::string> m_tag_names;
+};
+
+struct LightUserData
+{
+  std::string m_friendly_name;
+  std::vector<std::string> m_tag_names;
+};
+
+using EditorAssetData = std::variant<
+    std::unique_ptr<VideoCommon::MaterialData>, std::unique_ptr<VideoCommon::PixelShaderData>,
+    std::unique_ptr<VideoCommon::TextureData>, std::unique_ptr<VideoCommon::MeshData>,
+    std::unique_ptr<VideoCommon::RasterMaterialData>,
+    std::unique_ptr<VideoCommon::RasterShaderData>, std::unique_ptr<VideoCommon::RenderTargetData>>;
+enum AssetDataType
+{
+  Camera,
+  Material,
+  RasterMaterial,
+  Mesh,
+  PixelShader,
+  RenderTarget,
+  Shader,
+  Texture
+};
+struct EditorAsset
+{
+  VideoCommon::CustomAssetLibrary::AssetID m_asset_id;
+  std::filesystem::path m_asset_path;
+  EditorAssetData m_data;
+  AssetDataType m_data_type;
+  VideoCommon::CustomAssetLibrary::TimeType m_last_data_write;
+  VideoCommon::Assets::AssetMap m_asset_map;
+  bool m_valid = false;
+};
+
+using SelectableType =
+    std::variant<GraphicsModSystem::DrawCallID, GraphicsModSystem::TextureCacheID,
+                 GraphicsModSystem::LightID, GraphicsModAction*, EditorAsset*>;
+
+class EditorAction final : public GraphicsModAction
+{
+public:
+  explicit EditorAction(std::unique_ptr<GraphicsModAction> action) : m_action(std::move(action)) {}
+  void OnDrawStarted(GraphicsModActionData::DrawStarted* draw) override
+  {
+    if (m_active)
+      m_action->OnDrawStarted(draw);
+  }
+  void OnEFB(GraphicsModActionData::EFB* efb) override
+  {
+    if (m_active)
+      m_action->OnEFB(efb);
+  }
+  void OnXFB() override
+  {
+    if (m_active)
+      m_action->OnXFB();
+  }
+  void OnProjection(GraphicsModActionData::Projection* projection) override
+  {
+    if (m_active)
+      m_action->OnProjection(projection);
+  }
+  void OnProjectionAndTexture(GraphicsModActionData::Projection* projection) override
+  {
+    if (m_active)
+      m_action->OnProjectionAndTexture(projection);
+  }
+  void OnTextureLoad(GraphicsModActionData::TextureLoad* texture_load) override
+  {
+    if (m_active)
+      m_action->OnTextureLoad(texture_load);
+  }
+  void OnTextureCreate(GraphicsModActionData::TextureCreate* texture_create) override
+  {
+    if (m_active)
+      m_action->OnTextureCreate(texture_create);
+  }
+  void OnLight(GraphicsModActionData::Light* light) override
+  {
+    if (m_active)
+      m_action->OnLight(light);
+  }
+  void OnFrameEnd() override
+  {
+    if (m_active)
+      m_action->OnFrameEnd();
+  }
+
+  void SetAllDrawCall(GraphicsModSystem::DrawCallID draw_call)
+  {
+    m_draw_call = draw_call;
+    m_action->SetDrawCall(draw_call);
+  }
+
+  void DrawImGui() override
+  {
+    ImGui::Checkbox("##EmptyCheckbox", &m_active);
+    m_action->DrawImGui();
+  }
+
+  void SerializeToConfig(picojson::object* obj) override
+  {
+    if (!obj) [[unlikely]]
+      return;
+
+    auto& json_obj = *obj;
+    json_obj["active"] = picojson::value{m_active};
+    m_action->SerializeToConfig(&json_obj);
+  }
+
+  std::string GetFactoryName() const override { return m_action->GetFactoryName(); }
+
+  void SetActive(bool active) { m_active = active; }
+
+private:
+  bool m_active = true;
+  std::unique_ptr<GraphicsModAction> m_action;
+};
+}  // namespace GraphicsModEditor