//  ---------------------------------------------------------------------------------
//  Copyright (c) Microsoft Corporation.  All rights reserved.
// 
//  The MIT License (MIT)
// 
//  Permission is hereby granted, free of charge, to any person obtaining a copy
//  of this software and associated documentation files (the "Software"), to deal
//  in the Software without restriction, including without limitation the rights
//  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//  copies of the Software, and to permit persons to whom the Software is
//  furnished to do so, subject to the following conditions:
// 
//  The above copyright notice and this permission notice shall be included in
//  all copies or substantial portions of the Software.
// 
//  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
//  THE SOFTWARE.
//  ---------------------------------------------------------------------------------

using System.Runtime.InteropServices;
using Windows.Graphics.DirectX.Direct3D11;
using Windows.Win32;
using Windows.Win32.Graphics.Direct3D;
using Windows.Win32.Graphics.Direct3D11;
using Windows.Win32.Graphics.Dxgi;
using Windows.Win32.System.WinRT.Direct3D11;
using WinRT;
using static Windows.Win32.PInvoke;

namespace Vision.WindowCapture.GraphicsCapture.Helpers
{
    public static class Direct3D11Helper
    {
        static Guid IInspectable = new Guid("AF86E2E0-B12D-4c6a-9C5A-D7AA65101E90");
        static Guid ID3D11Resource = new Guid("dc8e63f3-d12b-4952-b47b-5e45026a862d");
        static Guid IDXGIAdapter3 = new Guid("645967A4-1392-4310-A798-8053CE3E93FD");
        //static Guid ID3D11Device = new Guid("db6f6ddb-ac77-4e88-8253-819df9bbf140");
        static Guid ID3D11Texture2D = new Guid("6f15aaf2-d208-4e89-9ab4-489535d34f9c");

        public static IDirect3DDevice? CreateDevice()
        {
            return CreateDevice(false);
        }

        public static unsafe IDirect3DDevice? CreateDevice(bool useWARP)
        {
            D3D_DRIVER_TYPE driverType = useWARP ? D3D_DRIVER_TYPE.D3D_DRIVER_TYPE_SOFTWARE : D3D_DRIVER_TYPE.D3D_DRIVER_TYPE_HARDWARE;
            D3D11CreateDevice(
                default,
                driverType,
                default,
                D3D11_CREATE_DEVICE_FLAG.D3D11_CREATE_DEVICE_BGRA_SUPPORT,
                default,
                D3D11_SDK_VERSION,
                out ID3D11Device? d3d11Device,
                default,
                out _);

            return CreateDirect3DDeviceFromD3D11Device(d3d11Device);
        }

        public static IDirect3DDevice? CreateDirect3DDeviceFromD3D11Device(ID3D11Device d3d11Device)
        {
            // Acquire the DXGI interface for the Direct3D device.
            // Wrap the native device using a WinRT interop object.
            if (CreateDirect3D11DeviceFromDXGIDevice(d3d11Device.As<IDXGIDevice>(), out Windows.Win32.System.WinRT.IInspectable iInspectable).Succeeded)
            {
                nint pointer = Marshal.GetIUnknownForObject(iInspectable);
                return MarshalInterface<IDirect3DDevice>.FromAbi(pointer);
            }

            return default;
        }

        public static IDirect3DSurface? CreateDirect3DSurfaceFromSharpDXTexture(ID3D11Texture2D texture)
        {
            // Acquire the DXGI interface for the Direct3D surface.
            // Wrap the native device using a WinRT interop object.
            if (CreateDirect3D11SurfaceFromDXGISurface(texture.As<IDXGISurface>(), out Windows.Win32.System.WinRT.IInspectable iInspectable).Succeeded)
            {
                nint pointer = Marshal.GetIUnknownForObject(iInspectable);
                return MarshalInterface<IDirect3DSurface>.FromAbi(pointer);
            }

            return default;
        }

        public static ID3D11Device CreateD3D11Device(IDirect3DDevice device)
        {
            device
                .As<IDirect3DDxgiInterfaceAccess>()
                .GetInterface(typeof(ID3D11Device).GUID, out var d3dDevice);
            return MarshalInterface<ID3D11Device>.FromAbi(d3dDevice);
        }

        public static ID3D11Texture2D CreateD3D11Texture2D(IDirect3DSurface surface)
        {
            surface
                .As<IDirect3DDxgiInterfaceAccess>()
                .GetInterface(typeof(ID3D11Texture2D).GUID, out var d3dSurface);
            return MarshalInterface<ID3D11Texture2D>.FromAbi(d3dSurface);
        }
    }
}