mirror of
https://github.com/gnif/LookingGlass.git
synced 2024-11-10 08:38:20 +00:00
94 lines
3.2 KiB
C++
94 lines
3.2 KiB
C++
/*
|
|
Looking Glass - KVM FrameRelay (KVMFR) Client
|
|
Copyright (C) 2017 Geoffrey McRae <geoff@hostfission.com>
|
|
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 <windows.h>
|
|
#include <dxgi1_2.h>
|
|
#include <d3d11.h>
|
|
#include <stdio.h>
|
|
#include <comdef.h>
|
|
|
|
_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;
|
|
};
|
|
}; |