NVAPI抓屏传递给D3D11(D3D9到D3D11的数据传递)

#include "NVAPIGrab.h" NVAPIGrab::NVAPIGrab() { m_pD3DEx = NULL; m_pD3D9Device = NULL; m_pD3D9Surf = NULL; m_bNvFBCLibLoaded = false; m_pNvFBCDX9 = NULL; m_pNVFBCLib = NULL; m_hGrabShare = NULL; m_pGrabResult = NULL; m_width = GetSystemMetrics(SM_CXSCREEN); m_height = GetSystemMetrics(SM_CYSCREEN); cont = 0; } NVAPIGrab::~NVAPIGrab() { ReleaseGrab(); } BOOL NVAPIGrab::InitGrab(HANDLE & sharehand) { DWORD maxDisplayWidth = -1, maxDisplayHeight = -1; NVFBC_TODX9VID_OUT_BUF NvFBC_OutBuf; //加载NvFBCLibary m_pNVFBCLib = new NvFBCLibrary(); if (!m_pNVFBCLib->load()) { fprintf(stderr, "Unable to load the NvFBC library.\n"); return FALSE; } m_bNvFBCLibLoaded = true; if (!InitD3D(0)) { fprintf(stderr, "Unable to create a D3D9Ex Device.\n"); ReleaseGrab(); return FALSE; } if (!InitD3DSurfaces()) { fprintf(stderr, "Unable to create a D3D9Ex Device.\n"); ReleaseGrab(); return FALSE; } m_pNvFBCDX9 = (NvFBCToDx9Vid *)m_pNVFBCLib->create(NVFBC_TO_DX9_VID, &maxDisplayWidth, &maxDisplayHeight, 0, (void *)m_pD3D9Device); if (!m_pNvFBCDX9) { fprintf(stderr, "Failed to create an instance of NvFBCToDx9Vid.\n"); ReleaseGrab(); return FALSE; } NvFBC_OutBuf.pPrimary = m_pD3D9Surf; NVFBC_TODX9VID_SETUP_PARAMS DX9SetupParams = { 0 }; DX9SetupParams.dwVersion = NVFBC_TODX9VID_SETUP_PARAMS_V2_VER; DX9SetupParams.bWithHWCursor = 0; DX9SetupParams.bStereoGrab = 0; DX9SetupParams.bDiffMap = 0; DX9SetupParams.ppBuffer = &NvFBC_OutBuf; DX9SetupParams.eMode = NVFBC_TODX9VID_ARGB; DX9SetupParams.dwNumBuffers = 1; //设置抓屏信息 if (NVFBC_SUCCESS != m_pNvFBCDX9->NvFBCToDx9VidSetUp(&DX9SetupParams)) { fprintf(stderr, "Failed when calling NvFBCDX9->NvFBCToDX9VidSetup()\n"); ReleaseGrab(); return FALSE; } //创建一个新的Texture用于D3D9传递数据给D3D11 D3DSURFACE_DESC _desc; m_pD3D9Surf->GetDesc(&_desc); HRESULT hr = m_pD3D9Device->CreateTexture(m_width, m_height, 1, D3DUSAGE_RENDERTARGET, _desc.Format, D3DPOOL_DEFAULT, &m_pGrabResult, &m_hGrabShare); if (hr != D3D_OK) return FALSE; sharehand = m_hGrabShare; return TRUE; } BOOL NVAPIGrab::GrabOneFarme() { NVFBC_TODX9VID_GRAB_FRAME_PARAMS fbcDX9GrabParams = { 0 }; NvFBCFrameGrabInfo frameGrabInfo = { 0 }; NVFBCRESULT fbcRes = NVFBC_SUCCESS; fbcDX9GrabParams.dwVersion = NVFBC_TODX9VID_GRAB_FRAME_PARAMS_VER; fbcDX9GrabParams.dwFlags = NVFBC_TODX9VID_NOWAIT; fbcDX9GrabParams.eGMode = NVFBC_TODX9VID_SOURCEMODE_SCALE; fbcDX9GrabParams.dwTargetWidth = m_width; fbcDX9GrabParams.dwTargetHeight = m_height; fbcDX9GrabParams.pNvFBCFrameGrabInfo = &frameGrabInfo; fbcRes = m_pNvFBCDX9->NvFBCToDx9VidGrabFrame(&fbcDX9GrabParams); if (fbcRes == NVFBC_SUCCESS) { //抓屏成功 //将OffscreenSurface的数据给m_pGrabResult IDirect3DSurface9* pGrabSurface = NULL; m_pGrabResult->GetSurfaceLevel(0, &pGrabSurface); m_pD3D9Device->StretchRect(m_pD3D9Surf, NULL, pGrabSurface, NULL, D3DTEXF_NONE); CString strIniFileB; strIniFileB.Format(_T(".\\%d.bmp"), cont); D3DXSaveTextureToFile(strIniFileB, D3DXIFF_BMP, m_pGrabResult, NULL); cont++; m_pD3D9Device->Present(0, 0, 0, 0);//提交一次避免数据没有刷新 return TRUE; } else { return FALSE; } return TRUE; } void NVAPIGrab::ReleaseGrab() { if (m_pNvFBCDX9) { m_pNvFBCDX9->NvFBCToDx9VidRelease(); m_pNvFBCDX9 = NULL; } if (m_pD3D9Surf) { m_pD3D9Surf->Release(); m_pD3D9Surf = NULL; } if (m_pD3D9Device) { m_pD3D9Device->Release(); m_pD3D9Device = NULL; } if (m_pD3DEx) { m_pD3DEx->Release(); m_pD3DEx = NULL; } if (m_bNvFBCLibLoaded) { m_pNVFBCLib->close(); } if (m_pNVFBCLib) { delete m_pNVFBCLib; m_pNVFBCLib = NULL; } if (m_pGrabResult != NULL) { m_pGrabResult->Release(); m_pGrabResult = NULL; } } //初始化D3d设备 deviceID为显示器ID BOOL NVAPIGrab::InitD3D(unsigned int deviceID) { HRESULT hr = S_OK; D3DPRESENT_PARAMETERS d3dpp; D3DADAPTER_IDENTIFIER9 adapterId; unsigned int iAdapter = NULL; //判断能否抓对应显示器的数据 Direct3DCreate9Ex(D3D_SDK_VERSION, &m_pD3DEx); if (deviceID >= m_pD3DEx->GetAdapterCount()) { printf("Error: (deviceID=%d) is not a valid GPU device. Headless video devices will not be detected. <<\n\n", deviceID); return FALSE; } hr = m_pD3DEx->GetAdapterIdentifier(deviceID, 0, &adapterId); if (hr != S_OK) { printf("Error: (deviceID=%d) is not a valid GPU device. <<\n\n", deviceID); return FALSE; } ZeroMemory(&d3dpp, sizeof(d3dpp)); d3dpp.Windowed = true; d3dpp.BackBufferFormat = D3DFMT_X8R8G8B8; d3dpp.BackBufferWidth = m_width; d3dpp.BackBufferHeight = m_height; d3dpp.BackBufferCount = 1; d3dpp.SwapEffect = D3DSWAPEFFECT_COPY; d3dpp.PresentationInterval = D3DPRESENT_INTERVAL_IMMEDIATE; d3dpp.Flags = D3DPRESENTFLAG_VIDEO; DWORD dwBehaviorFlags = D3DCREATE_FPU_PRESERVE | D3DCREATE_MULTITHREADED | D3DCREATE_HARDWARE_VERTEXPROCESSING; hr = m_pD3DEx->CreateDeviceEx(deviceID, D3DDEVTYPE_HAL, NULL, dwBehaviorFlags, &d3dpp, NULL, &m_pD3D9Device); if (FAILED(hr)) { OutputDebugString(L"CreateDevice Filed!"); return FALSE; } return TRUE; } //创建抓屏表面 BOOL NVAPIGrab::InitD3DSurfaces() { HRESULT hr = E_FAIL; if (m_pD3D9Device) { hr = m_pD3D9Device->CreateOffscreenPlainSurface(m_width, m_height, D3DFMT_A8R8G8B8, D3DPOOL_DEFAULT, &m_pD3D9Surf, NULL); if (FAILED(hr)) { fprintf(stderr, "Failed to create D3D9 surfaces for output. Error 0x%08x\n", hr); return FALSE; } } return TRUE; }