Method GetDisplayedImage

Source/DX9VideoProcessor.cpp:1791–1852

}

HRESULT CDX9VideoProcessor::GetDisplayedImage(BYTE **ppDib, unsigned *pSize)
{
	if (!m_pD3DDevEx) {
		return E_ABORT;
	}

	HRESULT hr = S_OK;
	const UINT width  = m_windowRect.Width();
	const UINT height = m_windowRect.Height();

	// use the back buffer, because the display profile is applied to the front buffer
	CComPtr<IDirect3DSurface9> pBackBuffer;
	CComPtr<IDirect3DSurface9> pDestSurface;

	hr = m_pD3DDevEx->GetBackBuffer(0, 0, D3DBACKBUFFER_TYPE_MONO, &pBackBuffer);
	if (SUCCEEDED(hr)) {
		hr = m_pD3DDevEx->CreateRenderTarget(width, height, D3DFMT_X8R8G8B8, D3DMULTISAMPLE_NONE, 0, TRUE, &pDestSurface, nullptr);
	}
	if (SUCCEEDED(hr)) {
		hr = m_pD3DDevEx->StretchRect(pBackBuffer, m_windowRect, pDestSurface, nullptr, D3DTEXF_NONE);
	}
	if (FAILED(hr)) {
		DLog(L"CDX9VideoProcessor::GetDisplayedImage() failed with error {}", HR2Str(hr));
		return hr;
	}

	const UINT dib_bitdepth = 32;
	const UINT dib_pitch    = CalcDibRowPitch(width, dib_bitdepth);
	const UINT dib_size	    = dib_pitch * height;

	*pSize = sizeof(BITMAPINFOHEADER) + dib_size;
	BYTE* p = (BYTE*)LocalAlloc(LMEM_FIXED, *pSize); // only this allocator can be used
	if (!p) {
		return E_OUTOFMEMORY;
	}

	BITMAPINFOHEADER* pBIH = (BITMAPINFOHEADER*)p;
	ZeroMemory(pBIH, sizeof(BITMAPINFOHEADER));
	pBIH->biSize      = sizeof(BITMAPINFOHEADER);
	pBIH->biWidth     = width;
	pBIH->biHeight    = -(LONG)height; // top-down RGB bitmap
	pBIH->biBitCount  = dib_bitdepth;
	pBIH->biPlanes    = 1;
	pBIH->biSizeImage = dib_size;

	D3DLOCKED_RECT lr;
	hr = pDestSurface->LockRect(&lr, nullptr, D3DLOCK_READONLY);
	if (S_OK == hr) {
		CopyPlaneAsIs(height, (BYTE*)(pBIH + 1), dib_pitch, (BYTE*)lr.pBits, lr.Pitch);
		hr = pDestSurface->UnlockRect();
	}

	if (FAILED(hr)) {
		DLog(L"CDX9VideoProcessor::GetDisplayedImage() failed with error {}", HR2Str(hr));
		LocalFree(p);
		return hr;
	}