/***
 **     libface - Library of face recognition and supporting algorithms
        Copyright (c) 2003 Stefan Farthofer

	This file is part of libface, which is

        free software; you can redistribute it and/or modify
        it under the terms of the GNU General Public License as published by
        the Free Software Foundation; either version 2 of the License, or
        (at your option) any later version.

        This program is distributed in the hope that it will be useful,
        but WITHOUT ANY WARRANTY; without even the implied warranty of
        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        GNU General Public License for more details.

        You should have received a copy of the GNU General Public License
        along with this program; if not, write to the Free Software
        Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

	For further information seek us at http://sourceforge.net/projects/openbio/
**	or write an email to dimitri.pissarenko@gmx.net or farthofer@chello.at.
***/

#include <stdio.h>
#include <memory.h>
#include <malloc.h>
#include <IL/IL.h>
#include "frbase.h"
#include "fr.h"
#include "recoalgo.h"

/* prototypes for utility functions */
void frutilFreeTraitArray(FRrecognitionParameters *gParms, FRimageTrait* traits, unsigned int nrTraits);


/* calculate traits */
int frTraitsCalc(FRrecognitionParameters* gParms, FRimage* rawImage, FRimageTrait** traits) {
	unsigned int i;
	int err = FR_OK;
	FRrecoAlgo* algo;
	FRimage image;

	/* TODO: call face finder and resize found face properly */
	/* for now, just resize */
	if (frImgResize(&image, rawImage, gParms->width, gParms->height)) return FR_ERR_IL;


	*traits = (FRimageTrait*) malloc(sizeof(FRimageTrait)*gParms->nrAlgorithms);
	if (traits == NULL) return FR_ERR_NOMEM;

	for (i=0; err == FR_OK && i < gParms->nrAlgorithms; i++) {
		algo = frGetRecoAlgoById(gParms->algorithms[i].algorithmType);
		err = algo->computeTrait(gParms, gParms->algorithms[i].data, &image, &(*traits)->data);
	}

	if (err != FR_OK) {
		frutilFreeTraitArray(gParms, *traits, i-1);
		*traits = NULL;
	}
	return err;
}

/* traits serialize */
size_t frTraitsGetSize(FRrecognitionParameters* gParms, FRimageTrait* traits) {
	unsigned int i;
	size_t sz=0;
	FRrecoAlgo* algo;

	for (i=0; i < gParms->nrAlgorithms; i++) {
		algo = frGetRecoAlgoById(gParms->algorithms[i].algorithmType);
		sz = sz + algo->getTraitSize(gParms, gParms->algorithms[i].data, traits[i].data);
	}
	return sz;
}

void frTraitsFree(FRrecognitionParameters* gParms, FRimageTrait** traits) {
	frutilFreeTraitArray(gParms, *traits, gParms->nrAlgorithms);
	*traits = NULL;
}

int frTraitsSerialize(FRrecognitionParameters* gParms, BYTE** mem, size_t maxsz, FRimageTrait** traits, BYTE direction) {
	BYTE* begin = *mem;
	unsigned int i;
	int err = FR_OK;
	FRrecoAlgo* algo;

	/* sanity check size */
	if (direction == FR_OUT && maxsz < frTraitsGetSize(gParms, *traits))
		return FR_ERR_ALLOCMORE;

	/* if we are reading from serialized form, allocate memory for our structure array */
	if (direction == FR_IN) {
		*traits = (FRimageTrait*) malloc(sizeof(FRimageTrait)*gParms->nrAlgorithms);
		if (*traits == NULL) return FR_ERR_NOMEM;
	}

	/* write/read data */
	for (i=0; err == FR_OK && i < gParms->nrAlgorithms; i++) {
		algo = frGetRecoAlgoById(gParms->algorithms[i].algorithmType);
		err = algo->serializeTrait(gParms, gParms->algorithms[i].data, &(*traits)[i].data, mem, maxsz-(*mem-begin),direction);
	}

	if (err != FR_OK && direction == FR_IN)
		frutilFreeTraitArray(gParms, *traits, i-1);

	return err;
}

int frTraitsSaveFile(FRrecognitionParameters* gParms, FILE* fileHandle, FRimageTrait* traits) {
	BYTE* buffer, * temp;
	size_t sz;
	int err;

	/* copy data to buffer */
	sz = frTraitsGetSize(gParms, traits);
	temp = buffer = (BYTE*) malloc(sz);
	if (buffer == NULL) return FR_ERR_NOMEM;
	err = frTraitsSerialize(gParms, &temp, sz, &traits, FR_OUT);

	/* write size and data */
	if (err == FR_OK && !fwrite(&sz, sizeof(sz),1, fileHandle)) err = FR_ERR_FILE;
	if (err == FR_OK && !fwrite(buffer,sz, 1, fileHandle)) err = FR_ERR_FILE;

	free(buffer);
	return err;
}

int frTraitsLoadFile(FRrecognitionParameters* gParms, FILE* fileHandle, FRimageTrait** traits) {
	BYTE* buffer, * temp;
	size_t sz;
	int err = FR_OK;

	/* read size */
	if (!fread(&sz, sizeof(sz), 1, fileHandle)) return FR_ERR_FILE;

	/* allocate buffer */
	temp = buffer = (BYTE*) malloc(sz);
	if (buffer == NULL) return FR_ERR_NOMEM;

	/* read serialized data */
	if (!fread(buffer, sz, 1, fileHandle)) err = FR_ERR_FILE;

	/* deserialize */
	if (err == FR_OK) err = frTraitsSerialize(gParms, &temp, sz, traits, FR_IN);
	free(buffer);
	return err;
}

void frutilFreeTraitArray(FRrecognitionParameters *gParms, FRimageTrait* traits, unsigned int nrTraits) {
	FRrecoAlgo* algo;
	unsigned int i;

	for (i=0; i < nrTraits; i++) {
		algo = frGetRecoAlgoById(gParms->algorithms[i].algorithmType);
		algo->freeTrait(gParms, gParms->algorithms[i].data, &traits[i].data);
	}
	free(traits);
}