#include "mpeg3private.h"
#include <stdlib.h>
#include <string.h>




// This is basically qtcache.c with quicktime_ replaced by mpeg3_


mpeg3_cache_t* mpeg3_new_cache()
{
	mpeg3_cache_t *result = calloc(1, sizeof(mpeg3_cache_t));
	return result;
}

void mpeg3_delete_cache(mpeg3_cache_t *ptr)
{
	if(ptr->frames) 
	{
		int i;
		for(i = 0; i < ptr->allocation; i++)
		{
			mpeg3_cacheframe_t *frame = &ptr->frames[i];
			if(frame->y) free(frame->y);
			if(frame->u) free(frame->u);
			if(frame->v) free(frame->v);
		}
		free(ptr->frames);
		free(ptr);
	}
}

void mpeg3_reset_cache(mpeg3_cache_t *ptr)
{
	ptr->total = 0;
}

void mpeg3_cache_put_frame(mpeg3_cache_t *ptr,
	int64_t frame_number,
	unsigned char *y,
	unsigned char *u,
	unsigned char *v,
	int y_size,
	int u_size,
	int v_size)
{
	mpeg3_cacheframe_t *frame = 0;
	int i;

//printf("mpeg3_put_frame 1\n");
// Get existing frame
	for(i = 0; i < ptr->total; i++)
	{
		if(ptr->frames[i].frame_number == frame_number)
		{
			frame = &ptr->frames[i];
			break;
		}
	}


	if(!frame)
	{
		if(ptr->total >= ptr->allocation)
		{
			int new_allocation = ptr->allocation * 2;
//printf("mpeg3_put_frame 10 %d\n", new_allocation);
			if(!new_allocation) new_allocation = 32;
			ptr->frames = realloc(ptr->frames, 
				sizeof(mpeg3_cacheframe_t) * new_allocation);
			bzero(ptr->frames + ptr->total,
				sizeof(mpeg3_cacheframe_t) * (new_allocation - ptr->allocation));
//printf("mpeg3_put_frame 20 %d %d %d\n", new_allocation, ptr->allocation, ptr->total);
			ptr->allocation = new_allocation;
		}

		frame = &ptr->frames[ptr->total];
//printf("mpeg3_put_frame 30 %d %p %p %p\n", ptr->total, frame->y, frame->u, frame->v);
		ptr->total++;

// Memcpy is a lot slower than just dropping the seeking frames.
		if(y) 
		{
			frame->y = realloc(frame->y, y_size);
			frame->y_size = y_size;
			memcpy(frame->y, y, y_size);
		}

		if(u)
		{
			frame->u = realloc(frame->u, u_size);
			frame->u_size = u_size;
			memcpy(frame->u, u, u_size);
		}

		if(v)
		{
			frame->v = realloc(frame->v, v_size);
			frame->v_size = v_size;
			memcpy(frame->v, v, v_size);
		}
		frame->frame_number = frame_number;
	}
//printf("mpeg3_put_frame 100\n");
}

int mpeg3_cache_get_frame(mpeg3_cache_t *ptr,
	int64_t frame_number,
	unsigned char **y,
	unsigned char **u,
	unsigned char **v)
{
	int i;

	for(i = 0; i < ptr->total; i++)
	{
		mpeg3_cacheframe_t *frame = &ptr->frames[i];
		if(frame->frame_number == frame_number)
		{
			
			*y = frame->y;
			*u = frame->u;
			*v = frame->v;
			return 1;
			break;
		}
	}

	return 0;
}


int mpeg3_cache_has_frame(mpeg3_cache_t *ptr,
	int64_t frame_number)
{
	int i;

	for(i = 0; i < ptr->total; i++)
	{
		mpeg3_cacheframe_t *frame = &ptr->frames[i];
		if(frame->frame_number == frame_number)
			return 1;
	}

	return 0;
}

int64_t mpeg3_cache_usage(mpeg3_cache_t *ptr)
{
	int64_t result = 0;
	int i;
	for(i = 0; i < ptr->allocation; i++)
	{
		mpeg3_cacheframe_t *frame = &ptr->frames[i];
		result += frame->y_size + frame->u_size + frame->v_size;
	}
	return result;
}