93 lines
2.9 KiB
C
93 lines
2.9 KiB
C
|
/*****************************************************************************
|
||
|
|
* DXGICapture.h
|
||
|
|
*
|
||
|
|
* Copyright (C) 2020 Gokhan Erdogdu <gokhan_erdogdu - at - yahoo - dot - com>
|
||
|
|
*
|
||
|
|
* DXGICapture is free software; you can redistribute it and/or modify it under
|
||
|
|
* the terms of the GNU Lesser General Public License as published by the Free
|
||
|
|
* Software Foundation; either version 2.1 of the License, or (at your option)
|
||
|
|
* any later version.
|
||
|
|
*
|
||
|
|
* DXGICapture 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 Lesser General Public License for more
|
||
|
|
* details.
|
||
|
|
*
|
||
|
|
******************************************************************************/
|
||
|
|
#pragma once
|
||
|
|
#ifndef __DXGICAPTURE_H__
|
||
|
|
#define __DXGICAPTURE_H__
|
||
|
|
|
||
|
|
#include <atlbase.h>
|
||
|
|
#include <ShellScalingAPI.h>
|
||
|
|
|
||
|
|
#include <dxgi1_2.h>
|
||
|
|
#include <d3d11.h>
|
||
|
|
|
||
|
|
#include <d2d1.h>
|
||
|
|
#include <d2d1_1.h> // for ID2D1Effect
|
||
|
|
#include <wincodec.h>
|
||
|
|
|
||
|
|
#include "DXGICaptureTypes.h"
|
||
|
|
|
||
|
|
#define D3D_FEATURE_LEVEL_INVALID ((D3D_FEATURE_LEVEL)0x0)
|
||
|
|
|
||
|
|
class CDXGICapture
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
ATL::CComAutoCriticalSection m_csLock;
|
||
|
|
|
||
|
|
BOOL m_bInitialized;
|
||
|
|
DublicatorMonitorInfoVec m_monitorInfos;
|
||
|
|
tagRendererInfo m_rendererInfo;
|
||
|
|
|
||
|
|
tagMouseInfo m_mouseInfo;
|
||
|
|
tagFrameBufferInfo m_tempMouseBuffer;
|
||
|
|
DXGI_OUTPUT_DESC m_desktopOutputDesc;
|
||
|
|
|
||
|
|
D3D_FEATURE_LEVEL m_lD3DFeatureLevel;
|
||
|
|
CComPtr<ID3D11Device> m_ipD3D11Device;
|
||
|
|
CComPtr<ID3D11DeviceContext> m_ipD3D11DeviceContext;
|
||
|
|
|
||
|
|
CComPtr<IDXGIOutputDuplication> m_ipDxgiOutputDuplication;
|
||
|
|
CComPtr<ID3D11Texture2D> m_ipCopyTexture2D;
|
||
|
|
|
||
|
|
CComPtr<ID2D1Device> m_ipD2D1Device;
|
||
|
|
CComPtr<ID2D1Factory> m_ipD2D1Factory;
|
||
|
|
CComPtr<IWICImagingFactory> m_ipWICImageFactory;
|
||
|
|
CComPtr<IWICBitmap> m_ipWICOutputBitmap;
|
||
|
|
CComPtr<ID2D1RenderTarget> m_ipD2D1RenderTarget;
|
||
|
|
|
||
|
|
public:
|
||
|
|
CDXGICapture();
|
||
|
|
~CDXGICapture();
|
||
|
|
|
||
|
|
private:
|
||
|
|
HRESULT loadMonitorInfos(ID3D11Device *pDevice);
|
||
|
|
void freeMonitorInfos();
|
||
|
|
|
||
|
|
HRESULT createDeviceResource(
|
||
|
|
const tagScreenCaptureFilterConfig *pConfig,
|
||
|
|
const tagDublicatorMonitorInfo *pSelectedMonitorInfo);
|
||
|
|
void terminateDeviceResource();
|
||
|
|
|
||
|
|
public:
|
||
|
|
HRESULT Initialize();
|
||
|
|
HRESULT Terminate();
|
||
|
|
HRESULT SetConfig(const tagScreenCaptureFilterConfig *pConfig);
|
||
|
|
HRESULT SetConfig(const tagScreenCaptureFilterConfig &config);
|
||
|
|
|
||
|
|
BOOL IsInitialized() const;
|
||
|
|
D3D_FEATURE_LEVEL GetD3DFeatureLevel() const;
|
||
|
|
|
||
|
|
int GetDublicatorMonitorInfoCount() const;
|
||
|
|
const tagDublicatorMonitorInfo* GetDublicatorMonitorInfo(int index) const;
|
||
|
|
const tagDublicatorMonitorInfo* FindDublicatorMonitorInfo(int monitorIdx) const;
|
||
|
|
|
||
|
|
HRESULT CaptureToFile(_In_ LPCWSTR lpcwOutputFileName, _Out_opt_ BOOL *pRetIsTimeout = NULL, _Out_opt_ UINT *pRetRenderDuration = NULL);
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // __DXGICAPTURE_H__
|
||
|
|
|
||
|
|
|