/* Looking Glass - KVM FrameRelay (KVMFR) Client Copyright (C) 2017 Geoffrey McRae https://looking-glass.hostfission.com This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #pragma once #include "ICapture.h" #define W32_LEAN_AND_MEAN #include #include #include #include #include _COM_SMARTPTR_TYPEDEF(IDXGIFactory1 , __uuidof(IDXGIFactory1 )); _COM_SMARTPTR_TYPEDEF(ID3D11Device , __uuidof(ID3D11Device )); _COM_SMARTPTR_TYPEDEF(ID3D11DeviceContext , __uuidof(ID3D11DeviceContext )); _COM_SMARTPTR_TYPEDEF(IDXGIDevice , __uuidof(IDXGIDevice )); _COM_SMARTPTR_TYPEDEF(IDXGIOutput1 , __uuidof(IDXGIOutput1 )); _COM_SMARTPTR_TYPEDEF(IDXGIOutput , __uuidof(IDXGIOutput )); _COM_SMARTPTR_TYPEDEF(IDXGIAdapter1 , __uuidof(IDXGIAdapter1 )); _COM_SMARTPTR_TYPEDEF(IDXGIOutputDuplication, __uuidof(IDXGIOutputDuplication)); _COM_SMARTPTR_TYPEDEF(ID3D11Texture2D , __uuidof(ID3D11Texture2D )); _COM_SMARTPTR_TYPEDEF(IDXGIResource , __uuidof(IDXGIResource )); namespace Capture { class DXGI : public ICapture { public: DXGI(); ~DXGI(); const char * GetName() { return "DXGI"; } bool Initialize(CaptureOptions * options); void DeInitialize(); bool ReInitialize() { DeInitialize(); /* DXGI needs some time when mode switches occur, failing to do so causes failure to start and exceptions internal to DXGI */ Sleep(400); return Initialize(m_options); } enum FrameType GetFrameType(); size_t GetMaxFrameSize(); enum GrabStatus GrabFrame(struct FrameInfo & frame); private: void WaitForDesktop(); CaptureOptions * m_options; bool m_initialized; unsigned int m_width; unsigned int m_height; IDXGIFactory1Ptr m_dxgiFactory; ID3D11DevicePtr m_device; D3D_FEATURE_LEVEL m_featureLevel; ID3D11DeviceContextPtr m_deviceContext; IDXGIOutput1Ptr m_output; IDXGIOutputDuplicationPtr m_dup; bool m_releaseFrame; ID3D11Texture2DPtr m_texture; D3D11_MAPPED_SUBRESOURCE m_mapping; bool m_surfaceMapped; BYTE * m_pointer; UINT m_pointerBufSize; UINT m_pointerSize; BOOL m_lastMouseVis; POINT m_lastMousePos; }; };