/*
 * mlt_geometry.h -- provides the geometry API
 * Copyright (C) 2004-2005 Ushodaya Enterprises Limited
 * Author: Charles Yates <charles.yates@pandora.be>
 *
 * This program 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.
 */

#include "mlt_geometry.h"
#include "mlt_tokeniser.h"
#include "mlt_factory.h"

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

typedef struct geometry_item_s
{
	struct mlt_geometry_item_s data;
	struct geometry_item_s *next, *prev;
}
*geometry_item;

typedef struct
{
	char *data;
	int length;
	int nw;
	int nh;
	geometry_item item;
}
geometry_s, *geometry;

// Create a new geometry structure
mlt_geometry mlt_geometry_init( )
{
	mlt_geometry this = calloc( 1, sizeof( struct mlt_geometry_s ) );
	if ( this != NULL )
	{
		this->local = calloc( 1, sizeof( geometry_s ) );
		if ( this->local != NULL )
		{
			geometry self = this->local;
			char *normalisation = mlt_environment( "MLT_NORMALISATION" );
			self->nw = 720;
			if ( normalisation == NULL || strcmp( normalisation, "NTSC" ) )
				self->nh = 576;
			else
				self->nh = 480;
		}
		else
		{
			free( this );
			this = NULL;
		}
	}
	return this;
}

static int mlt_geometry_drop( mlt_geometry this, geometry_item item )
{
	geometry self = this->local;

	if ( item == self->item )
	{
		self->item = item->next;
		if ( self->item != NULL )
			self->item->prev = NULL;
	}
	else if ( item->next != NULL )
	{
		item->next->prev = item->prev;
	}
	else if ( item->prev != NULL )
	{
		item->prev->next = item->next;
		item->next->prev = item->prev;
	}

	free( item );

	return 0;
}

static void mlt_geometry_clean( mlt_geometry this )
{
	geometry self = this->local;
	free( self->data );
	self->data = NULL;
	while( self->item )
		mlt_geometry_drop( this, self->item );
}

// Parse the geometry specification for a given length and normalised width/height (-1 for default)
// data is constructed as: [frame=]X,Y:WxH[:mix][;[frame=]X,Y:WxH[:mix]]*
// and X, Y, W and H can have trailing % chars to indicate percentage of normalised size
int mlt_geometry_parse( mlt_geometry this, char *data, int length, int nw, int nh )
{
	int i = 0;

	// Create a tokeniser
	mlt_tokeniser tokens = mlt_tokeniser_init( );

	// Get the local/private structure
	geometry self = this->local;

	// Clean the existing geometry
	mlt_geometry_clean( this );

	// Update the info on the data
	if ( length != -1 )
		self->length = length;
	if ( nw != -1 )
		self->nw = nw;
	if ( nh != -1 )
		self->nh = nh;
	if ( data != NULL )
		self->data = strdup( data );

	// Tokenise
	if ( data != NULL )
		mlt_tokeniser_parse_new( tokens, data, ";" );

	// Iterate through each token
	for ( i = 0; i < mlt_tokeniser_count( tokens ); i ++ )
	{
		struct mlt_geometry_item_s item;
		char *value = mlt_tokeniser_get_string( tokens, i );

		// Set item to 0
		memset( &item, 0, sizeof( struct mlt_geometry_item_s ) );

		// Now parse the item
		mlt_geometry_parse_item( this, &item, value );

		// Now insert into place
		mlt_geometry_insert( this, &item );
	}

	// Remove the tokeniser
	mlt_tokeniser_close( tokens );

	// ???
	return 0;
}

// Conditionally refresh in case of a change
int mlt_geometry_refresh( mlt_geometry this, char *data, int length, int nw, int nh )
{
	geometry self = this->local;
	int changed = ( length != -1 && length != self->length );
	changed = changed || ( nw != -1 && nw != self->nw );
	changed = changed || ( nh != -1 && nh != self->nh );
	changed = changed || ( data != NULL && ( self->data == NULL || strcmp( data, self->data ) ) );
	if ( changed )
		return mlt_geometry_parse( this, data, length, nw, nh );
	return -1;
}

int mlt_geometry_get_length( mlt_geometry this )
{
	// Get the local/private structure
	geometry self = this->local;

	// return the length
	return self->length;
}

void mlt_geometry_set_length( mlt_geometry this, int length )
{
	// Get the local/private structure
	geometry self = this->local;

	// return the length
	self->length = length;
}

int mlt_geometry_parse_item( mlt_geometry this, mlt_geometry_item item, char *value )
{
	int ret = 0;

	// Get the local/private structure
	geometry self = this->local;

	if ( value != NULL && strcmp( value, "" ) )
	{
		char *p = strchr( value, '=' );
		int count = 0;
		float temp;

		// Determine if a position has been specified
		if ( p != NULL )
		{
			item->frame = atoi( value );
			value = p + 1;
		}

		// Special case - frame < 0
		if ( item->frame < 0 )
			item->frame += self->length;

		// Obtain the current value at this position - this allows new
		// frames to be created which don't specify all values
		mlt_geometry_fetch( this, item, item->frame );

		// Iterate through the remainder of value
		while( *value )
		{
			// Get the value
			temp = strtod( value, &p );

			// Check if a value was specified
			if ( p != value )
			{
				// Handle the % case
				if ( *p == '%' )
				{
					if ( count == 0 || count == 2 )
						temp *= self->nw / 100.0;
					else if ( count == 1 || count == 3 )
						temp *= self->nh / 100.0;
					p ++;
				}

				// Special case - distort token
				if ( *p == '!' )
				{
					p ++;
					item->distort = 1;
				}

				// Actually, we don't care about the delimiter at all..
				if ( *p ) p ++;

				// Assign to the item
				switch( count )
				{
					case 0: item->x = temp; break;
					case 1: item->y = temp; break;
					case 2: item->w = temp; break;
					case 3: item->h = temp; break;
					case 4: item->mix = temp; break;
				}
			}
			else
			{
				p ++;
			}

			// Update the value pointer
			value = p;
			count ++;
		}
	}
	else
	{
		ret = 1;
	}

	return ret;
}

/** A linear step
*/

static inline float linearstep( float start, float end, float position, int length )
{
	float o = ( end - start ) / length;
	return start + position * o + 0.5;
}

// Fetch a geometry item for an absolute position
int mlt_geometry_fetch( mlt_geometry this, mlt_geometry_item item, float position )
{
	// Get the local geometry
	geometry self = this->local;

	// Need to find the nearest key to the position specifed
	geometry_item key = self->item;

	// Iterate through the keys until we reach last or have 
	while( key != NULL && key->next != NULL && position >= key->next->data.frame )
		key = key->next;

	if ( key != NULL )
	{
		// Position is situated before the first key - all zeroes
		if ( position < key->data.frame )
		{
			memset( item, 0, sizeof( struct mlt_geometry_item_s ) );
			item->mix = 100;
		}
		// Position is a key itself - no iterpolation need
		else if ( position == key->data.frame )
		{
			memcpy( item, &key->data, sizeof( struct mlt_geometry_item_s ) );
		}
		// Position is after the last key - no interpolation, but not a key frame
		else if ( key->next == NULL )
		{
			memcpy( item, &key->data, sizeof( struct mlt_geometry_item_s ) );
			item->key = 0;
		}
		// Interpolation is needed - position > key and there is a following key
		else
		{
			item->key = 0;
			item->frame = position;
			position -= key->data.frame;
			item->x = linearstep( key->data.x, key->next->data.x, position, key->next->data.frame - key->data.frame );
			item->y = linearstep( key->data.y, key->next->data.y, position, key->next->data.frame - key->data.frame );
			item->w = linearstep( key->data.w, key->next->data.w, position, key->next->data.frame - key->data.frame );
			item->h = linearstep( key->data.h, key->next->data.h, position, key->next->data.frame - key->data.frame );
			item->mix = linearstep( key->data.mix, key->next->data.mix, position, key->next->data.frame - key->data.frame );
			item->distort = key->data.distort;
			position += key->data.frame;
		}

		item->frame = position;
	}
	else
	{
		memset( item, 0, sizeof( struct mlt_geometry_item_s ) );
		item->mix = 100;
	}

	return key == NULL;
}

// Specify a geometry item at an absolute position
int mlt_geometry_insert( mlt_geometry this, mlt_geometry_item item )
{
	// Get the local/private geometry structure
	geometry self = this->local;

	// Create a new local item (this may be removed if a key already exists at this position)
	geometry_item new = calloc( 1, sizeof( struct geometry_item_s ) );
	memcpy( &new->data, item, sizeof( struct mlt_geometry_item_s ) );
	new->data.key = 1;

	// Determine if we need to insert or append to the list, or if it's a new list
	if ( self->item != NULL )
	{
		// Get the first item
		geometry_item place = self->item;

		// Locate an existing nearby item
		while ( place->next != NULL && item->frame > place->data.frame )
			place = place->next;

		if ( item->frame < place->data.frame )
		{
			if ( place == self->item )
				self->item = new;
			if ( place->prev )
				place->prev->next = new;
			new->next = place;
			new->prev = place->prev;
			place->prev = new;
		}
		else if ( item->frame > place->data.frame )
		{
			new->next = place->next;
			new->prev = place;
			place->next = new;
		}
		else
		{
			memcpy( &place->data, &new->data, sizeof( struct mlt_geometry_item_s ) );
			free( new );
		}
	}
	else
	{
		self->item = new;
	}

	// TODO: Error checking
	return 0;
}

// Remove the key at the specified position
int mlt_geometry_remove( mlt_geometry this, int position )
{
	int ret = 1;

	// Get the local/private geometry structure
	geometry self = this->local;

	// Get the first item
	geometry_item place = self->item;

	while( place != NULL && position < place->data.frame )
		place = place->next;

	if ( place != NULL && position == place->data.frame )
		ret = mlt_geometry_drop( this, place );

	return ret;
}

// Get the key at the position or the next following
int mlt_geometry_key( mlt_geometry this, mlt_geometry_item item, int position )
{
	// Get the local/private geometry structure
	geometry self = this->local;

	// Get the first item
	geometry_item place = self->item;

	while( place != NULL && position > place->data.frame )
		place = place->next;

	if ( place != NULL )
		memcpy( item, &place->data, sizeof( struct mlt_geometry_item_s ) );

	return place == NULL;
}

char *mlt_geometry_serialise_cut( mlt_geometry this, int in, int out )
{
	struct mlt_geometry_item_s item;
	char *ret = malloc( 1000 );
	int used = 0;
	int size = 1000;

	if ( in == -1 )
		in = 0;
	if ( out == -1 )
		out = mlt_geometry_get_length( this );

	if ( ret != NULL )
	{
		char temp[ 100 ];

		strcpy( ret, "" );

		item.frame = in;

		while( 1 )
		{
			strcpy( temp, "" );

			// If it's the first frame, then it's not necessarily a key
			if ( item.frame == in )
			{
				if ( mlt_geometry_fetch( this, &item, item.frame ) )
					break;
			}
			// Typically, we move from key to key
			else if ( item.frame < out )
			{
				if ( mlt_geometry_key( this, &item, item.frame ) )
					break;

				// Special case - crop at the out point
				if ( item.frame > out )
					mlt_geometry_fetch( this, &item, out );
			}
			// We've handled the last key
			else
			{
				break;
			}

			if ( item.frame - in != 0 )
				sprintf( temp, "%d=", item.frame - in );

			sprintf( temp + strlen( temp ), "%.0f,%.0f:%.0fx%.0f%s", item.x, item.y, item.w, item.h, item.distort ? "!" : "" );

			if ( item.mix )
				sprintf( temp + strlen( temp ), ":%.0f", item.mix );

			if ( used + strlen( temp ) > size )
			{
				size += 1000;
				ret = realloc( ret, size );
			}

			if ( ret != NULL && used != 0 )
			{
				used ++;
				strcat( ret, ";" );
			}
			if ( ret != NULL )
			{
				used += strlen( temp );
				strcat( ret, temp );
			}

			item.frame ++;
		}
	}

	return ret;
}

// Serialise the current geometry
char *mlt_geometry_serialise( mlt_geometry this )
{
	geometry self = this->local;
	char *ret = mlt_geometry_serialise_cut( this, 0, self->length );
	if ( ret )
	{
		free( self->data );
		self->data = ret;
	}
	return ret;
}

// Close the geometry
void mlt_geometry_close( mlt_geometry this )
{
	if ( this != NULL )
	{
		mlt_geometry_clean( this );
		free( this->local );
		free( this );
	}
}