Initial commit for GitHub migration based on spice2x-25-03-03

2026-08-02 23:00:42 -07:00 · 2025-03-10 19:16:49 -07:00
parent 94af8d32c9
commit ed8c72852f
1008 changed files with 296815 additions and 0 deletions
@@ -0,0 +1,141 @@
+#include "d3d9_swapchain.h"
+
+#include <cassert>
+#include <mutex>
+
+#include "avs/game.h"
+#include "hooks/graphics/graphics.h"
+
+#include "d3d9_backend.h"
+#include "d3d9_device.h"
+
+// std::min
+#ifdef min
+#undef min
+#endif
+
+#define CHECK_RESULT(x) \
+    HRESULT ret = (x); \
+    if (GRAPHICS_LOG_HRESULT && FAILED(ret)) [[unlikely]] { \
+        log_warning("graphics::d3d9", "{} failed, hr={}", __FUNCTION__, FMT_HRESULT(ret)); \
+    } \
+    return ret
+
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::QueryInterface(REFIID riid, void **ppvObj) {
+    if (ppvObj == nullptr) {
+        return E_POINTER;
+    }
+
+    if (//riid == __uuidof(IUnknown) || Ignore IUnknown, it's often queried to test object equality between different interfaces
+        riid == IID_IDirect3DSwapChain9 ||
+        riid == IID_IDirect3DSwapChain9Ex)
+    {
+#pragma region Update to IDirect3DSwapChain9Ex interface
+        if (!is_d3d9ex && riid == IID_IDirect3DSwapChain9Ex) {
+            IDirect3DSwapChain9Ex *swapchainex = nullptr;
+
+            if (FAILED(pReal->QueryInterface(IID_PPV_ARGS(&swapchainex)))) {
+                return E_NOINTERFACE;
+            }
+
+            pReal->Release();
+            pReal = swapchainex;
+            is_d3d9ex = true;
+        }
+#pragma endregion
+
+        this->AddRef();
+        *ppvObj = this;
+
+        return S_OK;
+    }
+
+    return pReal->QueryInterface(riid, ppvObj);
+}
+
+ULONG STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::AddRef(void) {
+    return pReal->AddRef();
+}
+ULONG STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::Release(void) {
+    ULONG refs = pReal != nullptr ? pReal->Release() : 0;
+
+    if (refs == 0) {
+        delete this;
+    }
+
+    // Metal Gear Arcade expects the swap chain to only have one reference. The parent
+    // `WrappedIDirect3DDevice9` holds a strong reference to this swap chain which means
+    // the reference count will be above one. Workaround this by returning a maximum of one.
+    if (avs::game::is_model("I36")) {
+        return std::min(refs, 1lu);
+    }
+
+    return refs;
+}
+
+/*
+ * IDirect3DSwapChain9
+ */
+
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::Present(const RECT *pSourceRect, const RECT *pDestRect,
+        HWND hDestWindowOverride, const RGNDATA *pDirtyRegion, DWORD dwFlags)
+{
+    static std::once_flag printed;
+    std::call_once(printed, []() {
+        log_misc("graphics::d3d9", "WrappedIDirect3DSwapChain9::Present");
+    });
+
+    if (should_run_hooks) {
+        graphics_d3d9_on_present(pDev->hFocusWindow, pDev->pReal, pDev);
+    }
+
+    CHECK_RESULT(pReal->Present(pSourceRect, pDestRect, hDestWindowOverride, pDirtyRegion, dwFlags));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetFrontBufferData(IDirect3DSurface9 *pDestSurface) {
+    CHECK_RESULT(pReal->GetFrontBufferData(pDestSurface));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetBackBuffer(UINT iBackBuffer, D3DBACKBUFFER_TYPE Type,
+        IDirect3DSurface9 **ppBackBuffer)
+{
+    CHECK_RESULT(pReal->GetBackBuffer(iBackBuffer, Type, ppBackBuffer));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetRasterStatus(D3DRASTER_STATUS *pRasterStatus) {
+    CHECK_RESULT(pReal->GetRasterStatus(pRasterStatus));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetDisplayMode(D3DDISPLAYMODE *pMode) {
+    CHECK_RESULT(pReal->GetDisplayMode(pMode));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetDevice(IDirect3DDevice9 **ppDevice) {
+    if (ppDevice == nullptr) {
+        return D3DERR_INVALIDCALL;
+    }
+
+    pDev->AddRef();
+    *ppDevice = pDev;
+
+    return D3D_OK;
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetPresentParameters(
+        D3DPRESENT_PARAMETERS *pPresentationParameters)
+{
+    CHECK_RESULT(pReal->GetPresentParameters(pPresentationParameters));
+}
+
+/*
+ * IDirect3DSwapChain9Ex
+ */
+
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetLastPresentCount(UINT *pLastPresentCount) {
+    assert(is_d3d9ex);
+    CHECK_RESULT(static_cast<IDirect3DSwapChain9Ex *>(pReal)->GetLastPresentCount(pLastPresentCount));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetPresentStats(D3DPRESENTSTATS *pPresentationStatistics) {
+    assert(is_d3d9ex);
+    CHECK_RESULT(static_cast<IDirect3DSwapChain9Ex *>(pReal)->GetPresentStats(pPresentationStatistics));
+}
+HRESULT STDMETHODCALLTYPE WrappedIDirect3DSwapChain9::GetDisplayModeEx(D3DDISPLAYMODEEX *pMode,
+        D3DDISPLAYROTATION *pRotation)
+{
+    assert(is_d3d9ex);
+    CHECK_RESULT(static_cast<IDirect3DSwapChain9Ex *>(pReal)->GetDisplayModeEx(pMode, pRotation));
+}