/**
 * 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.
 */

#include <freerdp/config.h>

#include <winpr/assert.h>

#include "shadow_surface.h"

#include "shadow_screen.h"
#include "shadow_lobby.h"

rdpShadowScreen* shadow_screen_new(rdpShadowServer* server)
{
	WINPR_ASSERT(server);
	WINPR_ASSERT(server->subsystem);

	rdpShadowScreen* screen = (rdpShadowScreen*)calloc(1, sizeof(rdpShadowScreen));

	if (!screen)
		goto fail;

	screen->server = server;

	{
		rdpShadowSubsystem* subsystem = server->subsystem;
		if (!InitializeCriticalSectionAndSpinCount(&(screen->lock), 4000))
			goto fail;

		region16_init(&(screen->invalidRegion));

		{
			WINPR_ASSERT(subsystem->selectedMonitor < ARRAYSIZE(subsystem->monitors));
			const MONITOR_DEF* primary = &(subsystem->monitors[subsystem->selectedMonitor]);
			WINPR_ASSERT(primary);

			const INT64 x = primary->left;
			const INT64 y = primary->top;
			const INT64 width = primary->right - primary->left + 1;
			const INT64 height = primary->bottom - primary->top + 1;

			WINPR_ASSERT(x >= 0);
			WINPR_ASSERT(x <= UINT16_MAX);
			WINPR_ASSERT(y >= 0);
			WINPR_ASSERT(y <= UINT16_MAX);
			WINPR_ASSERT(width >= 0);
			WINPR_ASSERT(width <= UINT16_MAX);
			WINPR_ASSERT(height >= 0);
			WINPR_ASSERT(height <= UINT16_MAX);

			screen->width = (UINT16)width;
			screen->height = (UINT16)height;

			screen->primary =
			    shadow_surface_new(server, (UINT16)x, (UINT16)y, (UINT16)width, (UINT16)height);

			if (!screen->primary)
				goto fail;

			server->surface = screen->primary;

			screen->lobby =
			    shadow_surface_new(server, (UINT16)x, (UINT16)y, (UINT16)width, (UINT16)height);
		}
	}

	if (!screen->lobby)
		goto fail;

	server->lobby = screen->lobby;

	if (!shadow_client_init_lobby(server))
		goto fail;

	return screen;

fail:
	WINPR_PRAGMA_DIAG_PUSH
	WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
	shadow_screen_free(screen);
	WINPR_PRAGMA_DIAG_POP

	return nullptr;
}

void shadow_screen_free(rdpShadowScreen* screen)
{
	if (!screen)
		return;

	DeleteCriticalSection(&(screen->lock));

	region16_uninit(&(screen->invalidRegion));

	if (screen->primary)
	{
		shadow_surface_free(screen->primary);
		screen->primary = nullptr;
	}

	if (screen->lobby)
	{
		shadow_surface_free(screen->lobby);
		screen->lobby = nullptr;
	}

	free(screen);
}

BOOL shadow_screen_resize(rdpShadowScreen* screen)
{
	if (!screen)
		return FALSE;

	WINPR_ASSERT(screen->server);

	rdpShadowSubsystem* subsystem = screen->server->subsystem;
	WINPR_ASSERT(subsystem);
	WINPR_ASSERT(subsystem->monitors);

	MONITOR_DEF* primary = &(subsystem->monitors[subsystem->selectedMonitor]);
	WINPR_ASSERT(primary);

	const INT32 x = primary->left;
	const INT32 y = primary->top;
	const INT32 width = primary->right - primary->left + 1;
	const INT32 height = primary->bottom - primary->top + 1;

	WINPR_ASSERT(x >= 0);
	WINPR_ASSERT(x <= UINT16_MAX);
	WINPR_ASSERT(y >= 0);
	WINPR_ASSERT(y <= UINT16_MAX);
	WINPR_ASSERT(width >= 0);
	WINPR_ASSERT(width <= UINT16_MAX);
	WINPR_ASSERT(height >= 0);
	WINPR_ASSERT(height <= UINT16_MAX);

	if (shadow_surface_resize(screen->primary, (UINT16)x, (UINT16)y, (UINT16)width,
	                          (UINT16)height) &&
	    shadow_surface_resize(screen->lobby, (UINT16)x, (UINT16)y, (UINT16)width, (UINT16)height))
	{
		if (((UINT32)width != screen->width) || ((UINT32)height != screen->height))
		{
			/* screen size is changed. Store new size and reinit lobby */
			screen->width = (UINT32)width;
			screen->height = (UINT32)height;
			return shadow_client_init_lobby(screen->server);
		}
		return TRUE;
	}

	return FALSE;
}
