AutomataSpeedrunMod/SwapChainWrapper.hpp at mainline · DisrespectDwardo/AutomataSpeedrunMod · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#pragma once

#include <d2d1_2.h>
#include <dwrite.h>
#include <dxgi1_2.h>
#include <atlbase.h>
#include <chrono>

namespace DxWrappers {

class DXGISwapChainWrapper : public IDXGISwapChain1 {
    static const WCHAR* VC3_NAME;
    static const UINT VC3_LEN;
    static const D2D1::ColorF WATERMARK_COLOR;
    static const D2D1::ColorF SHADOW_COLOR;
    static const D2D1_BITMAP_PROPERTIES1 BITMAP_PROPERTIES;

    ULONG m_refCount;
    CComPtr<IDXGISwapChain1> m_target;
    CComPtr<ID2D1DeviceContext> m_deviceContext;
    CComPtr<ID2D1Device> m_D2DDevice;
    CComPtr<IDWriteTextFormat> m_textFormat;
    CComPtr<ID2D1SolidColorBrush> m_brush;
    CComPtr<ID2D1SolidColorBrush> m_shadowBrush;

    bool m_dvdMode; // true when watermark should bounce around
    D2D1_VECTOR_2F m_location;
    D2D1_VECTOR_2F m_velocity;
    std::chrono::time_point<std::chrono::high_resolution_clock> m_lastFrame;

    void renderWatermark();

    // Roates velocity in a random direction
    void rotateVelocity();

    // Resets location to the default position
    void resetLocation(D2D1_SIZE_F& screenSize);

public:
    DXGISwapChainWrapper(IUnknown* pDevice, CComPtr<IDXGISwapChain1> target, CComPtr<ID2D1Factory2> d2dFactory);
    virtual ~DXGISwapChainWrapper();
    void toggleDvdMode(bool enabled);

    virtual HRESULT __stdcall QueryInterface(REFIID riid, void** ppvObject) override;
    virtual ULONG __stdcall AddRef() override;
    virtual ULONG __stdcall Release() override;
    virtual HRESULT __stdcall SetPrivateData(REFGUID Name, UINT DataSize, const void* pData) override;
    virtual HRESULT __stdcall SetPrivateDataInterface(REFGUID Name, const IUnknown* pUnknown) override;
    virtual HRESULT __stdcall GetPrivateData(REFGUID Name, UINT* pDataSize, void* pData) override;
    virtual HRESULT __stdcall GetParent(REFIID riid, void** ppParent) override;
    virtual HRESULT __stdcall GetDevice(REFIID riid, void** ppDevice) override;
    virtual HRESULT __stdcall Present(UINT SyncInterval, UINT Flags) override;
    virtual HRESULT __stdcall GetBuffer(UINT Buffer, REFIID riid, void** ppSurface) override;
    virtual HRESULT __stdcall SetFullscreenState(BOOL Fullscreen, IDXGIOutput* pTarget) override;
    virtual HRESULT __stdcall GetFullscreenState(BOOL* pFullscreen, IDXGIOutput** ppTarget) override;
    virtual HRESULT __stdcall GetDesc(DXGI_SWAP_CHAIN_DESC* pDesc) override;
    virtual HRESULT __stdcall ResizeBuffers(UINT BufferCount, UINT Width, UINT Height, DXGI_FORMAT NewFormat, UINT SwapChainFlags) override;
    virtual HRESULT __stdcall ResizeTarget(const DXGI_MODE_DESC* pNewTargetParameters) override;
    virtual HRESULT __stdcall GetContainingOutput(IDXGIOutput** ppOutput) override;
    virtual HRESULT __stdcall GetFrameStatistics(DXGI_FRAME_STATISTICS* pStats) override;
    virtual HRESULT __stdcall GetLastPresentCount(UINT* pLastPresentCount) override;
    virtual HRESULT __stdcall GetDesc1(DXGI_SWAP_CHAIN_DESC1* pDesc) override;
    virtual HRESULT __stdcall GetFullscreenDesc(DXGI_SWAP_CHAIN_FULLSCREEN_DESC* pDesc) override;
    virtual HRESULT __stdcall GetHwnd(HWND* pHwnd) override;
    virtual HRESULT __stdcall GetCoreWindow(REFIID refiid, void** ppUnk) override;
    virtual HRESULT __stdcall Present1(UINT SyncInterval, UINT PresentFlags, const DXGI_PRESENT_PARAMETERS* pPresentParameters) override;
    virtual BOOL __stdcall IsTemporaryMonoSupported() override;
    virtual HRESULT __stdcall GetRestrictToOutput(IDXGIOutput** ppRestrictToOutput) override;
    virtual HRESULT __stdcall SetBackgroundColor(const DXGI_RGBA* pColor) override;
    virtual HRESULT __stdcall GetBackgroundColor(DXGI_RGBA* pColor) override;
    virtual HRESULT __stdcall SetRotation(DXGI_MODE_ROTATION Rotation) override;
    virtual HRESULT __stdcall GetRotation(DXGI_MODE_ROTATION* pRotation) override;

};

} // namespace DxWrappers