mirror of
https://github.com/gnif/LookingGlass.git
synced 2024-11-13 01:38:20 +00:00
[host] Added cursor support to DXGI capture
This commit is contained in:
parent
f90c258c34
commit
8fadf0a80c
@ -27,7 +27,8 @@ DXGI::DXGI() :
|
||||
m_device(NULL),
|
||||
m_deviceContext(NULL),
|
||||
m_dup(NULL),
|
||||
m_texture(NULL)
|
||||
m_texture(NULL),
|
||||
m_pointer(NULL)
|
||||
{
|
||||
}
|
||||
|
||||
@ -176,6 +177,12 @@ void DXGI::DeInitialize()
|
||||
{
|
||||
m_memcpy.DeInitialize();
|
||||
|
||||
if (m_pointer)
|
||||
{
|
||||
delete[] m_pointer;
|
||||
m_pointer = NULL;
|
||||
}
|
||||
|
||||
if (m_texture)
|
||||
m_texture.Release();
|
||||
|
||||
@ -268,6 +275,26 @@ bool DXGI::GrabFrame(FrameInfo & frame)
|
||||
src->GetDesc(&desc);
|
||||
|
||||
m_deviceContext->CopyResource(m_texture, src);
|
||||
|
||||
// if the pointer shape has changed
|
||||
if (frameInfo.PointerShapeBufferSize > 0)
|
||||
{
|
||||
if (m_pointerBufSize < frameInfo.PointerShapeBufferSize)
|
||||
{
|
||||
if (m_pointer)
|
||||
delete[] m_pointer;
|
||||
m_pointer = new BYTE[frameInfo.PointerShapeBufferSize];
|
||||
m_pointerBufSize = frameInfo.PointerShapeBufferSize;
|
||||
}
|
||||
|
||||
status = m_dup->GetFramePointerShape(m_pointerBufSize, m_pointer, &m_pointerSize, &m_shapeInfo);
|
||||
if (!SUCCEEDED(status))
|
||||
{
|
||||
DEBUG_ERROR("Failed to get the new pointer shape: %08x", status);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
m_dup->ReleaseFrame();
|
||||
res.Release();
|
||||
src.Release();
|
||||
@ -293,10 +320,75 @@ bool DXGI::GrabFrame(FrameInfo & frame)
|
||||
frame.width = desc.Width;
|
||||
frame.height = desc.Height;
|
||||
frame.stride = rect.Pitch / 4;
|
||||
|
||||
frame.outSize = min(frame.bufferSize, m_height * rect.Pitch);
|
||||
m_memcpy.Copy(frame.buffer, rect.pBits, frame.outSize);
|
||||
|
||||
status = surface->Unmap();
|
||||
|
||||
// if we have a mouse update
|
||||
if (frameInfo.LastMouseUpdateTime.QuadPart)
|
||||
{
|
||||
m_pointerVisible = frameInfo.PointerPosition.Visible;
|
||||
m_pointerPos = frameInfo.PointerPosition.Position;
|
||||
}
|
||||
|
||||
// if the pointer is to be drawn
|
||||
if (m_pointerVisible)
|
||||
{
|
||||
const int maxHeight = min(m_shapeInfo.Height, desc.Height - m_pointerPos.y);
|
||||
const int maxWidth = min(m_shapeInfo.Width , desc.Width - m_pointerPos.x);
|
||||
|
||||
switch (m_shapeInfo.Type)
|
||||
{
|
||||
case DXGI_OUTDUPL_POINTER_SHAPE_TYPE_COLOR:
|
||||
{
|
||||
for(int y = abs(min(0, m_pointerPos.y)); y < maxHeight; ++y)
|
||||
for (int x = abs(min(0, m_pointerPos.x)); x < maxWidth; ++x)
|
||||
{
|
||||
BYTE *src = (BYTE *)m_pointer + (m_shapeInfo.Pitch * y) + (x * 4);
|
||||
BYTE *dst = (BYTE *)frame.buffer + (rect.Pitch * (y + m_pointerPos.y)) + ((x + m_pointerPos.x) * 4);
|
||||
|
||||
const unsigned int alpha = src[3] + 1;
|
||||
const unsigned int inv = 256 - alpha;
|
||||
dst[0] = (BYTE)((alpha * src[0] + inv * dst[0]) >> 8);
|
||||
dst[1] = (BYTE)((alpha * src[1] + inv * dst[1]) >> 8);
|
||||
dst[2] = (BYTE)((alpha * src[2] + inv * dst[2]) >> 8);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
case DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MASKED_COLOR:
|
||||
{
|
||||
for (int y = abs(min(0, m_pointerPos.y)); y < maxHeight; ++y)
|
||||
for (int x = abs(min(0, m_pointerPos.x)); x < maxWidth; ++x)
|
||||
{
|
||||
UINT32 *src = (UINT32 *)m_pointer + ((m_shapeInfo.Pitch/4) * y) + x;
|
||||
UINT32 *dst = (UINT32 *)frame.buffer + (frame.stride * (y + m_pointerPos.y)) + (x + m_pointerPos.x);
|
||||
if (*src & 0xff000000)
|
||||
*dst = 0xff000000 | (*dst ^ *src);
|
||||
else *dst = 0xff000000 | *src;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
case DXGI_OUTDUPL_POINTER_SHAPE_TYPE_MONOCHROME:
|
||||
{
|
||||
for (int y = abs(min(0, m_pointerPos.y)); y < maxHeight / 2; ++y)
|
||||
for (int x = abs(min(0, m_pointerPos.x)); x < maxWidth; ++x)
|
||||
{
|
||||
UINT8 *srcAnd = (UINT8 *)m_pointer + (m_shapeInfo.Pitch * y) + (x/8);
|
||||
UINT8 *srcXor = srcAnd + m_shapeInfo.Pitch * (m_shapeInfo.Height / 2);
|
||||
UINT32 *dst = (UINT32 *)frame.buffer + (frame.stride * (y + m_pointerPos.y)) + (x + m_pointerPos.x);
|
||||
const BYTE mask = 0x80 >> (x % 8);
|
||||
const UINT32 andMask = (*srcAnd & mask) ? 0xFFFFFFFF : 0xFF000000;
|
||||
const UINT32 xorMask = (*srcXor & mask) ? 0x00FFFFFF : 0x00000000;
|
||||
*dst = (*dst & andMask) ^ xorMask;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (FAILED(status))
|
||||
{
|
||||
DEBUG_ERROR("Failed to unmap surface: %08x", status);
|
||||
|
@ -54,5 +54,11 @@ namespace Capture
|
||||
CComQIPtr<IDXGIOutput1> m_output;
|
||||
CComPtr<IDXGIOutputDuplication> m_dup;
|
||||
CComPtr<ID3D11Texture2D> m_texture;
|
||||
BYTE * m_pointer;
|
||||
UINT m_pointerBufSize;
|
||||
UINT m_pointerSize;
|
||||
DXGI_OUTDUPL_POINTER_SHAPE_INFO m_shapeInfo;
|
||||
BOOL m_pointerVisible;
|
||||
POINT m_pointerPos;
|
||||
};
|
||||
};
|
Loading…
Reference in New Issue
Block a user