mirror of https://github.com/FreeRDP/FreeRDP
249 lines
4.5 KiB
C
249 lines
4.5 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
*
|
|
* Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <winpr/crt.h>
|
|
#include <winpr/print.h>
|
|
|
|
#include <freerdp/log.h>
|
|
|
|
#include "shadow_surface.h"
|
|
|
|
#include "shadow_capture.h"
|
|
|
|
#define TAG SERVER_TAG("shadow")
|
|
|
|
int shadow_capture_align_clip_rect(RECTANGLE_16* rect, RECTANGLE_16* clip)
|
|
{
|
|
int dx, dy;
|
|
dx = (rect->left % 16);
|
|
|
|
if (dx != 0)
|
|
{
|
|
rect->left -= dx;
|
|
rect->right += dx;
|
|
}
|
|
|
|
dx = (rect->right % 16);
|
|
|
|
if (dx != 0)
|
|
{
|
|
rect->right += (16 - dx);
|
|
}
|
|
|
|
dy = (rect->top % 16);
|
|
|
|
if (dy != 0)
|
|
{
|
|
rect->top -= dy;
|
|
rect->bottom += dy;
|
|
}
|
|
|
|
dy = (rect->bottom % 16);
|
|
|
|
if (dy != 0)
|
|
{
|
|
rect->bottom += (16 - dy);
|
|
}
|
|
|
|
if (rect->left < clip->left)
|
|
rect->left = clip->left;
|
|
|
|
if (rect->top < clip->top)
|
|
rect->top = clip->top;
|
|
|
|
if (rect->right > clip->right)
|
|
rect->right = clip->right;
|
|
|
|
if (rect->bottom > clip->bottom)
|
|
rect->bottom = clip->bottom;
|
|
|
|
return 1;
|
|
}
|
|
|
|
int shadow_capture_compare(BYTE* pData1, UINT32 nStep1, UINT32 nWidth, UINT32 nHeight,
|
|
BYTE* pData2, UINT32 nStep2, RECTANGLE_16* rect)
|
|
{
|
|
BOOL equal;
|
|
BOOL allEqual;
|
|
UINT32 tw, th;
|
|
UINT32 tx, ty, k;
|
|
UINT32 nrow, ncol;
|
|
UINT32 l, t, r, b;
|
|
BYTE* p1, *p2;
|
|
BOOL rows[1024];
|
|
#ifdef WITH_DEBUG_SHADOW_CAPTURE
|
|
BOOL cols[1024];
|
|
#endif
|
|
allEqual = TRUE;
|
|
ZeroMemory(rect, sizeof(RECTANGLE_16));
|
|
FillMemory(rows, sizeof(rows), 0xFF);
|
|
#ifdef WITH_DEBUG_SHADOW_CAPTURE
|
|
FillMemory(cols, sizeof(cols), 0xFF);
|
|
#endif
|
|
nrow = (nHeight + 15) / 16;
|
|
ncol = (nWidth + 15) / 16;
|
|
l = ncol + 1;
|
|
r = 0;
|
|
t = nrow + 1;
|
|
b = 0;
|
|
|
|
for (ty = 0; ty < nrow; ty++)
|
|
{
|
|
th = ((ty + 1) == nrow) ? (nHeight % 16) : 16;
|
|
|
|
if (!th)
|
|
th = 16;
|
|
|
|
for (tx = 0; tx < ncol; tx++)
|
|
{
|
|
equal = TRUE;
|
|
tw = ((tx + 1) == ncol) ? (nWidth % 16) : 16;
|
|
|
|
if (!tw)
|
|
tw = 16;
|
|
|
|
p1 = &pData1[(ty * 16 * nStep1) + (tx * 16 * 4)];
|
|
p2 = &pData2[(ty * 16 * nStep2) + (tx * 16 * 4)];
|
|
|
|
for (k = 0; k < th; k++)
|
|
{
|
|
if (memcmp(p1, p2, tw * 4) != 0)
|
|
{
|
|
equal = FALSE;
|
|
break;
|
|
}
|
|
|
|
p1 += nStep1;
|
|
p2 += nStep2;
|
|
}
|
|
|
|
if (!equal)
|
|
{
|
|
rows[ty] = FALSE;
|
|
#ifdef WITH_DEBUG_SHADOW_CAPTURE
|
|
cols[tx] = FALSE;
|
|
#endif
|
|
|
|
if (l > tx)
|
|
l = tx;
|
|
|
|
if (r < tx)
|
|
r = tx;
|
|
}
|
|
}
|
|
|
|
if (!rows[ty])
|
|
{
|
|
allEqual = FALSE;
|
|
|
|
if (t > ty)
|
|
t = ty;
|
|
|
|
if (b < ty)
|
|
b = ty;
|
|
}
|
|
}
|
|
|
|
if (allEqual)
|
|
return 0;
|
|
|
|
rect->left = l * 16;
|
|
rect->top = t * 16;
|
|
rect->right = (r + 1) * 16;
|
|
rect->bottom = (b + 1) * 16;
|
|
|
|
if (rect->right > nWidth)
|
|
rect->right = nWidth;
|
|
|
|
if (rect->bottom > nHeight)
|
|
rect->bottom = nHeight;
|
|
|
|
#ifdef WITH_DEBUG_SHADOW_CAPTURE
|
|
size_t size = ncol + 1;
|
|
char* col_str = calloc(size, sizeof(char));
|
|
|
|
if (!col_str)
|
|
{
|
|
WLog_ERR(TAG, "calloc failed!");
|
|
return 1;
|
|
}
|
|
|
|
for (tx = 0; tx < ncol; tx++)
|
|
sprintf_s(&col_str[tx], size - tx, "-");
|
|
|
|
WLog_INFO(TAG, "%s", col_str);
|
|
|
|
for (tx = 0; tx < ncol; tx++)
|
|
sprintf_s(&col_str[tx], size - tx, "%c", cols[tx] ? 'O' : 'X');
|
|
|
|
WLog_INFO(TAG, "%s", col_str);
|
|
|
|
for (tx = 0; tx < ncol; tx++)
|
|
sprintf_s(&col_str[tx], size - tx, "-");
|
|
|
|
WLog_INFO(TAG, "%s", col_str);
|
|
|
|
for (ty = 0; ty < nrow; ty++)
|
|
{
|
|
for (tx = 0; tx < ncol; tx++)
|
|
sprintf_s(&col_str[tx], size - tx, "%c", cols[tx] ? 'O' : 'X');
|
|
|
|
WLog_INFO(TAG, "%s", col_str);
|
|
WLog_INFO(TAG, "|%s|", rows[ty] ? "O" : "X");
|
|
}
|
|
|
|
WLog_INFO(TAG, "left: %d top: %d right: %d bottom: %d ncol: %d nrow: %d",
|
|
l, t, r, b, ncol, nrow);
|
|
free(col_str);
|
|
#endif
|
|
return 1;
|
|
}
|
|
|
|
rdpShadowCapture* shadow_capture_new(rdpShadowServer* server)
|
|
{
|
|
rdpShadowCapture* capture;
|
|
capture = (rdpShadowCapture*) calloc(1, sizeof(rdpShadowCapture));
|
|
|
|
if (!capture)
|
|
return NULL;
|
|
|
|
capture->server = server;
|
|
|
|
if (!InitializeCriticalSectionAndSpinCount(&(capture->lock), 4000))
|
|
{
|
|
free(capture);
|
|
return NULL;
|
|
}
|
|
|
|
return capture;
|
|
}
|
|
|
|
void shadow_capture_free(rdpShadowCapture* capture)
|
|
{
|
|
if (!capture)
|
|
return;
|
|
|
|
DeleteCriticalSection(&(capture->lock));
|
|
free(capture);
|
|
}
|
|
|