//
// Library: Memory Manage Module v1.10, Copyleft, 2009-02-25
// Author : Mandingo, mandingo [ at ]yoire.com
//

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "mmemory.h"

static memInfo *mInfo;

void memStringRealloc(memChunk *chunk) {
	memCheckState ();
	memChunk *temp;
	temp = memString (chunk->address);
	//printf("Reallocating chunk size %d to size %d\n",chunk->size,temp->size);
	memCopy (chunk,temp);
	memFree (temp);
}

void memFree(memChunk *chunk) {
	memCheckState ();
	if (chunk && chunk->address){
		mInfo->allocated-=chunk->size;
		free (chunk->address);
		free (chunk);
	}
}
void memInit() {
	mInfo = (memInfo*) malloc (sizeof (memInfo));
	if (!mInfo) return;
	memset (mInfo, 0, sizeof (memInfo));
	mInfo->state = MEM_STATE_OK;
	mInfo->allocated += sizeof (memInfo);
}
void memCheckState() {
	if (!mInfo) memInit();
	if (mInfo->state != MEM_STATE_OK){
		fprintf (stderr,"\rMemory not initialized :p\n");
		exit (0);
	}
}
memInfo *memInformation() {
	memCheckState ();
	return mInfo;
}
long memAllocated() {
	memCheckState ();
	return mInfo->allocated;
}
memChunk *memReserve(long size) {
	static memChunk *buffer;
	memCheckState ();
	buffer = (memChunk*)malloc (sizeof (memChunk));
	if (!buffer || (buffer->address = (char*)malloc (size)) == NULL) {
		perror ("memReserve");
		exit (0);
  	}
	//printf("- reservando %d bytes\n",size);
	buffer->size = size;
	memset (buffer->address, 0, buffer->size);
	mInfo->allocated += buffer->size;
	return buffer;
}
memChunk *memStringReserve(char *string, long nbytes) {
	static memChunk *buffer;
	buffer=memReserve (nbytes);
	memCopy (buffer, memString (string));
	return buffer;
}
memChunk *memString(char *string) {
	static memChunk *buffer;
	memCheckState ();
	buffer = memReserve (strlen (string) + 1);
	memcpy (buffer->address, string, strlen (string));
	return buffer;
}

void memCopy(memChunk *dest,memChunk *source) {
	long nbytes;
	memCheckState ();
	if ((!source->address) || (!dest->address)) return;
	nbytes=dest->size > source->size ? source->size : dest->size;
	#if DEBUG3
	eprintf ("Copying %d bytes to dest (size %d)\n", nbytes, dest->address, dest->size);
	#endif
	memcpy (dest->address, source->address, nbytes);
}

void memStrCat(memChunk *dest, char *string) {
	memChunk result,*temp;

	temp           = memReserve (dest->size + strlen (string)+1);

	result.address = dest->address + strlen (dest->address);
	result.size    = dest->size - strlen (dest->address) + 1;

	memCopy (temp, memString (string));
	memCopy (&result, temp);

	memFree (temp);
}
