#pragma once
#include "IOCPServer.h"
#include "..\..\client\CursorInfo.h"

extern "C"
{
#include "libavcodec\avcodec.h"
#include "libavutil\avutil.h"
#include "libyuv\libyuv.h"
}

#ifndef _WIN64
// https://github.com/Terodee/FFMpeg-windows-static-build/releases
#pragma comment(lib,"libavcodec.lib")
#pragma comment(lib,"libavutil.lib")
#pragma comment(lib,"libswresample.lib")

#pragma comment(lib,"libyuv/libyuv.lib")
#else
#pragma comment(lib,"x264/libx264_x64.lib")
#pragma comment(lib,"libyuv/libyuv_x64.lib")
// https://github.com/ShiftMediaProject/FFmpeg
#ifdef _DEBUG
#pragma comment(lib,"libavcodec_x64d.lib")
#pragma comment(lib,"libavutil_x64d.lib")
#pragma comment(lib,"libswresample_x64d.lib")
#else
#pragma comment(lib,"libavcodec_x64.lib")
#pragma comment(lib,"libavutil_x64.lib")
#pragma comment(lib,"libswresample_x64.lib")
#endif
#endif

#pragma comment(lib, "Mfplat.lib")
#pragma comment(lib, "Mfuuid.lib")
#pragma comment(lib, "Bcrypt.lib")
#pragma comment(lib, "Strmiids.lib")

// CScreenSpyDlg 对话框

class CScreenSpyDlg : public DialogBase
{
    DECLARE_DYNAMIC(CScreenSpyDlg)

public:
    CScreenSpyDlg(CWnd* Parent, Server* IOCPServer=NULL, CONTEXT_OBJECT *ContextObject=NULL);
    virtual ~CScreenSpyDlg();

    VOID SendNext(void);
    VOID OnReceiveComplete();
    HDC  m_hFullDC;
    HDC  m_hFullMemDC;
    HBITMAP	m_BitmapHandle;
    PVOID   m_BitmapData_Full;
    LPBITMAPINFO m_BitmapInfor_Full;
    VOID DrawFirstScreen(void);
    VOID DrawNextScreenDiff(bool keyFrame);
    BOOL         m_bIsFirst;
    ULONG m_ulHScrollPos;
    ULONG m_ulVScrollPos;
    VOID DrawTipString(CString strString);

    POINT  m_ClientCursorPos;
    BYTE m_bCursorIndex;
    BOOL     m_bIsTraceCursor;
    CCursorInfo	m_CursorInfo; //自定义的一个系统的光标类
    VOID SendCommand(const MYMSG* Msg);
    void SendScaledMouseMessage(MSG* pMsg, bool makeLP);
    VOID UpdateServerClipboard(char *szBuffer,ULONG ulLength);
    VOID SendServerClipboard(void);

    BOOL  m_bIsCtrl;
    LPBYTE m_szData;
    BOOL  m_bSend;
    ULONG m_ulMsgCount;
    int m_FrameID;

    BOOL SaveSnapshot(void);
    // 对话框数据
    enum { IDD = IDD_DIALOG_SCREEN_SPY };

    BOOL m_bFullScreen;

    WINDOWPLACEMENT m_struOldWndpl;

    AVCodec*			m_pCodec;
    AVCodecContext*		m_pCodecContext;
    AVPacket			m_AVPacket;
    AVFrame				m_AVFrame;

    clock_t				m_lastMouseMove; // 鼠标移动时间
    POINT				m_lastMousePoint;// 上次鼠标位置
    BOOL				m_bAdaptiveSize = TRUE;
    HCURSOR				m_hRemoteCursor = NULL;
    CRect				m_CRect;
    double				m_wZoom=1, m_hZoom=1;
    bool				m_bMouseTracking = false;

    bool Decode(LPBYTE Buffer, int size);
    void EnterFullScreen();
    bool LeaveFullScreen();

    afx_msg void OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar);
    afx_msg void OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar);
    afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
    afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
    afx_msg BOOL OnMouseWheel(UINT nFlags, short zDelta, CPoint pt);
    afx_msg void OnMouseMove(UINT nFlags, CPoint point);
    afx_msg void OnMouseLeave();
    afx_msg void OnKillFocus(CWnd* pNewWnd);
    afx_msg void OnSize(UINT nType, int cx, int cy);
	afx_msg void OnActivate(UINT nState, CWnd* pWndOther, BOOL bMinimized);

protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

    DECLARE_MESSAGE_MAP()
public:
    virtual BOOL OnInitDialog();
    afx_msg void OnClose();
    afx_msg void OnPaint();
    afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
    virtual BOOL PreTranslateMessage(MSG* pMsg);
    void OnLButtonDblClk(UINT nFlags, CPoint point);
};