#include "mainwindow.h"
#include <QApplication>
#include "cplaywidget.h"
#include <QTimer>
#include "CameraCapture.h"
#include "mainwindow.h"
#include <qlibrary.h>
#include <qsysinfo.h>
#include <qt_windows.h>
#include "media/screen_capture.h"
#include "media/DXGICapture.h"
#include <QVector>
#include <stdio.h>
#include <tchar.h>
#include <shlobj.h>
#include <D3D9.h>

#if _MSC_VER >= 1600
#pragma execution_character_set("utf-8")
#endif

#ifdef __MINGW32__
#include <Tlhelp32.h>
#include "winuser.h"
#endif

int RegiesterOwnType(){
    return 0;
}

IDirect3DDevice9* g_pd3dDevice = nullptr;

// 全???======================================================

// 建立D3DDevice

LPDIRECT3D9         g_pD3D = nullptr; // Used to create the D3DDevice

// (我?的?)render?置，用??生最後影像的?置


// =============================================================
HRESULT ScreenGrab(LPDIRECT3DDEVICE9 pDev)

{
    HRESULT hr;
    D3DDISPLAYMODE mode;
    if (FAILED(hr=pDev->GetDisplayMode(0, &mode)))  //第一???是Swap Chain，跟?重???有防止破碎
        return hr;

    IDirect3DSurface9* pSurface;

    if (FAILED(hr=pDev->CreateOffscreenPlainSurface(mode.Width,mode.Height,
                            D3DFMT_A8R8G8B8, D3DPOOL_SCRATCH, &pSurface, NULL)))
        return hr;


    if (FAILED(hr=pDev->GetFrontBufferData(0,pSurface))) {
        pSurface->Release();
        return hr;
    }
    D3DLOCKED_RECT	lockedRect;
    if(FAILED(pSurface->LockRect(&lockedRect,NULL,D3DLOCK_NO_DIRTY_UPDATE|D3DLOCK_NOSYSLOCK|D3DLOCK_READONLY)))
    {
        printf("Unable to Lock Front Buffer Surface");
        return 1;
    }
    BYTE* pBits = new BYTE[2960*1440*3];
    for(int i=0;i<mode.Height;i++)
    {
        memcpy((BYTE*)pBits+(mode.Height-i-1)*mode.Width*3/8,(BYTE*)lockedRect.pBits+i*lockedRect.Pitch,
               mode.Width*3/8);//g_d3dpp.BackBufferHeight*g_d3dpp.BackBufferWidth*4);

    }
    pSurface->UnlockRect();

    SetCursor(LoadCursor(NULL,IDC_ARROW));

    // release the image surface

    // ?放影像表面

    pSurface->Release();



    // return status of save operation to caller

    // 回?Handler

    return hr;

}



QVector<tagDublicatorMonitorInfo> GetMonitors()
{
    QVector<tagDublicatorMonitorInfo> ret;
    HRESULT hr = S_OK;
    CDXGICapture dxgiCapture;
    hr = dxgiCapture.Initialize();
    if (FAILED(hr)) {
        printf("Error[0x%08X]: CDXGICapture::Initialize failed.\n", hr);
        return ret;
    }

    int nMonitorCount = dxgiCapture.GetDublicatorMonitorInfoCount();
    printf("{\n");
    if (nMonitorCount > 0) {
        printf("  \"count\" : %u,\n", nMonitorCount);
        printf("  \"monitors\" : [\n");
        for (int i = 0; i < nMonitorCount; ++i) {
            const tagDublicatorMonitorInfo *pInfo = dxgiCapture.GetDublicatorMonitorInfo(i);
            if (nullptr != pInfo) {
                printf("    {\n");
                printf("      \"idx\" : %u,\n", pInfo->Idx);
                printf("      \"name\" : \"%S\",\n", pInfo->DisplayName);
                printf("      \"x\" : %d,\n", pInfo->Bounds.X);
                printf("      \"y\" : %d,\n", pInfo->Bounds.Y);
                printf("      \"width\" : %d,\n", pInfo->Bounds.Width);
                printf("      \"height\" : %d,\n", pInfo->Bounds.Height);
                printf("      \"rotation\" : %d\n", pInfo->RotationDegrees);
                if (i < (nMonitorCount - 1)) {
                    printf("    },\n");
                }
                else {
                    printf("    }\n");
                }
                ret.push_back(*pInfo);
            }
        }
        printf("  ]\n");
    }
    else {
        printf("  \"count\" : %u\n", nMonitorCount);
    }
    printf("}\n");

    dxgiCapture.Terminate();

    return ret;
}

HRESULT InitD3D( HWND hWnd )

{

    // Create the D3D object, which is needed to create the D3DDevice.

    // D3DDevice所需要的D3D物件，Direct3DCreate9回?D3DDevice，?入值?使用者安?DirectX的版本

    if( NULL == ( g_pD3D = Direct3DCreate9( D3D_SDK_VERSION ) ) )

        return E_FAIL;



    // Set up the structure used to create the D3DDevice. Most parameters are

    // zeroed out. We set Windowed to TRUE, since we want to do D3D in a

    // window, and then set the SwapEffect to "discard", which is the most

    // efficient method of presenting the back buffer to the display.  And

    // we request a back buffer format that matches the current desktop display

    // format.

    /*

       用?建立D3DDevice的??。大多????0。

       ?定是否?"?窗化"?真，就是?示??窗化。

       如果?D3D於?窗?行，?SwapEffect定"discard"，??在後端??呈???最有效率。

       ?且我?要求後端??的?示格式??在的?面相符。

    */

    D3DPRESENT_PARAMETERS d3dpp;

    ZeroMemory( &d3dpp, sizeof( d3dpp ) ); //初始化d3d的??

    d3dpp.Windowed = TRUE; //?定?窗化

    d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD; //提高效率

    d3dpp.BackBufferFormat = D3DFMT_UNKNOWN; //??有呈??面相符格式(竟然是?定未知= =)



    // Create the Direct3D device. Here we are using the default adapter (most

    // systems only have one, unless they have multiple graphics hardware cards

    // installed) and requesting the HAL (which is saying we want the hardware

    // device rather than a software one). Software vertex processing is

    // specified since we know it will work on all cards. On cards that support

    // hardware vertex processing, though, we would see a big performance gain

    // by specifying hardware vertex processing.

    /*

       建立Direct3D?置。我?使用??的adapter(?似?置的意思= =)。

       ?大多?的使用者??只有一??置(?示卡)，除非他?有其多的?示卡。

       ?且要求HAL，也就是?先用?示卡?示，而不是用???理。

       既然我?知道????(角)?理的?理方式被定?所有?示卡。

       於支援硬????理的?示卡上，我??看到?佳的?行效率。

    */

    if( FAILED( g_pD3D->CreateDevice( D3DADAPTER_DEFAULT,  //??adapter

                                      D3DDEVTYPE_HAL, //硬?加速

                                      hWnd, //Handler名?

                                      //D3DCREATE_SOFTWARE_VERTEXPROCESSING, //最後竟然用??？

                                      D3DCREATE_HARDWARE_VERTEXPROCESSING, //改用硬?加速吧

                                      &d3dpp, //D3D??

                                      &g_pd3dDevice //D3D最後?示?置??

                                      ) ) )

    {

        return E_FAIL;

    }


    // Device state would normally be set here

    return S_OK;

}

int main(int argc, char *argv[])
{
    setbuf(stdout, NULL);//让printf立即输出


    QssEventFilter filter;
    QApplication app(argc, argv);


    MainWindow main;
    InitD3D((HWND)main.winId());
    ScreenGrab( g_pd3dDevice);
    main.setWindowTitle("流媒体测试工具");
    main.setFixedSize(1920,1080);
    main.show();
    return app.exec();
}