/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Window Icon Cache
 *
 * Copyright 2011 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 <winpr/stream.h>

#include <freerdp/rail/icon.h>

ICON_INFO* icon_cache_get(rdpIconCache* cache, BYTE id, UINT16 index, void** extra)
{
	ICON_INFO* entry;

	if (id >= cache->numCaches)
	{
		fprintf(stderr, "invalid window icon cache id:%d\n", id);
		return (ICON_INFO*) NULL;
	}

	if (index >= cache->numCacheEntries)
	{
		fprintf(stderr, "invalid window icon cache index:%d in cache id:%d\n", index, id);
		return (ICON_INFO*) NULL;
	}

	entry = cache->caches[id].entries[index].entry;

	if (extra)
		*extra = cache->caches[id].entries[index].extra;

	return entry;
}

void icon_cache_put(rdpIconCache* cache, BYTE id, UINT16 index, ICON_INFO* entry, void* extra)
{
	if (id >= cache->numCaches)
	{
		fprintf(stderr, "invalid window icon cache id:%d\n", id);
		return;
	}

	if (index >= cache->numCacheEntries)
	{
		fprintf(stderr, "invalid window icon cache index:%d in cache id:%d\n", index, id);
		return;
	}

	cache->caches[id].entries[index].entry = entry;

	if (extra != NULL)
		cache->caches[id].entries[index].extra = extra;
}

rdpIconCache* icon_cache_new(rdpRail* rail)
{
	rdpIconCache* cache;

	cache = (rdpIconCache*) malloc(sizeof(rdpIconCache));

	if (cache != NULL)
	{
		int i;

		ZeroMemory(cache, sizeof(rdpIconCache));

		cache->rail = rail;
		cache->numCaches = (BYTE) rail->settings->RemoteAppNumIconCacheEntries;
		cache->numCacheEntries = rail->settings->RemoteAppNumIconCacheEntries;

		cache->caches = malloc(cache->numCaches * sizeof(WINDOW_ICON_CACHE));
		ZeroMemory(cache->caches, cache->numCaches * sizeof(WINDOW_ICON_CACHE));

		for (i = 0; i < cache->numCaches; i++)
		{
			if (cache->numCacheEntries)
			{
				cache->caches[i].entries = malloc(cache->numCacheEntries * sizeof(rdpIcon));
				ZeroMemory(cache->caches[i].entries, cache->numCacheEntries * sizeof(rdpIcon));
			}
		}
	}

	return cache;
}

void icon_cache_free(rdpIconCache* cache)
{
	if (cache)
	{
		int i;

		for (i = 0; i < cache->numCaches; i++)
		{
			free(cache->caches[i].entries);
		}

		free(cache->caches);

		free(cache);
	}
}