/*
   Copyright (C) 1999-2006 Id Software, Inc. and contributors.
   For a list of contributors, see the accompanying CONTRIBUTORS file.

   This file is part of GtkRadiant.

   GtkRadiant 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.

   GtkRadiant 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 GtkRadiant; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include <assert.h>
#include <math.h>
#include "token.h"
#include "joints.h"
#include "angles.h"
#include "inout.h"

char *SKEL_ROOT_NAMES[] =
{
	"RAVEN_SPINE"
};

char *SKEL_NAMES[] =
{
	"RAVEN_WAIST1",
	"RAVEN_WAIST2",
	"RAVEN_NECK"
};

int NAME_OFFSETS[] =
{
	0
};

int numJointsForSkeleton[] =
{
	NUM_JOINTS_RAVEN,
	NUM_JOINTS_BOX
};

float g_scaling[3];
float g_rotation[3];
float g_translation[3];

//==========================================================================
//
// LoadHRCClustered
//
//==========================================================================

static void LoadHRCClustered( char *fileName, int **clusterList, int *num_verts, int skelType ){
	int i, j;

	TK_OpenSource( fileName );
	TK_FetchRequire( TK_HRCH );
	TK_FetchRequire( TK_COLON );
	TK_FetchRequire( TK_SOFTIMAGE );

	TK_Beyond( TK_CLUSTERS );

	while ( TK_Search( TK_CLUSTER_NAME ) != TK_EOF )
	{
		TK_Require( TK_STRING );

		for ( i = 0; i < numJointsForSkeleton[skelType]; ++i )
		{
			if ( stricmp( tk_String, SKEL_NAMES[NAME_OFFSETS[skelType] + i] ) == 0 ) {
				i = -i + numJointsForSkeleton[skelType] - 1;

				TK_BeyondRequire( TK_NUM_CLUSTER_VERTICES, TK_INTNUMBER );

				num_verts[i + 1] = tk_IntNumber;

				clusterList[i] = (int *) SafeMalloc( num_verts[i + 1] * sizeof( int ), "LoadHRCClustered" );
				assert( clusterList[i] );
				// currently this function is only called by LoadTriangleListClustered, which in turn is only
				// called by Cmd_Base in qdata.  This is where the only call to free for this memory is being made.

				TK_Beyond( TK_LBRACE );

				for ( j = 0; j < num_verts[i + 1]; ++j )
				{
					TK_Require( TK_INTNUMBER );
					clusterList[i][j] = tk_IntNumber;
					TK_Fetch();
				}

				break;
			}
		}
	}

	num_verts[0] = numJointsForSkeleton[skelType];
}

static void LoadHRCGlobals( char *fileName ){
	int i;

	TK_OpenSource( fileName );
	TK_FetchRequire( TK_HRCH );
	TK_FetchRequire( TK_COLON );
	TK_FetchRequire( TK_SOFTIMAGE );
	TK_Beyond( TK_MODEL );

	TK_Beyond( TK_SCALING );
	for ( i = 0; i < 3; i++ )
	{
		TK_Require( TK_FLOATNUMBER );
		g_scaling[i] = tk_FloatNumber;
		TK_Fetch();
	}

	TK_Beyond( TK_ROTATION );
	for ( i = 0; i < 3; i++ )
	{
		TK_Require( TK_FLOATNUMBER );
		g_rotation[i] = tk_FloatNumber;
		TK_Fetch();
	}

	TK_Beyond( TK_TRANSLATION );
	for ( i = 0; i < 3; i++ )
	{
		TK_Require( TK_FLOATNUMBER );
		g_translation[i] = tk_FloatNumber;
		TK_Fetch();
	}
}

static void ParseVec3( vec3_t in ){
	TK_Require( TK_FLOATNUMBER );
	in[1] = tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[2] = tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[0] = tk_FloatNumber;
}

static void ParseRotation3( vec3_t in ){
	TK_Require( TK_FLOATNUMBER );
	in[1] = -tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[2] = tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[0] = tk_FloatNumber;
}

static void ParseTranslation3( vec3_t in ){
	TK_Require( TK_FLOATNUMBER );
	in[1] = tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[2] = tk_FloatNumber;
	TK_FetchRequire( TK_FLOATNUMBER );
	in[0] = tk_FloatNumber;
}

static void LoadHRCJointList( char *fileName, QDataJoint_t *jointList, int skelType ){
#define MAX_STACK 64
	int i, j;
	vec3_t curTranslation[MAX_STACK], curRotation[MAX_STACK], curScale[MAX_STACK];
	int curCorrespondingJoint[MAX_STACK];
	int currentStack = 0, stackSize;
	int baseJoint;
	float cx, sx, cy, sy, cz, sz;
	float rx, ry, rz;
	float x2, y2, z2;


	TK_OpenSource( fileName );
	TK_FetchRequire( TK_HRCH );
	TK_FetchRequire( TK_COLON );
	TK_FetchRequire( TK_SOFTIMAGE );

	TK_Beyond( TK_MODEL );
	TK_Beyond( TK_MODEL );

/*	while(1)
    {
        TK_Beyond(TK_MODEL);
        TK_BeyondRequire(TK_NAME, TK_STRING);

        if(_stricmp(tk_String, SKEL_ROOT_NAMES[skelType]) == 0)
            break;
    }*/

	TK_Beyond( TK_SCALING );

	ParseVec3( curScale[currentStack] );

	TK_Beyond( TK_ROTATION );

	ParseRotation3( curRotation[currentStack] );

	TK_Beyond( TK_TRANSLATION );

	ParseVec3( curTranslation[currentStack] );

	// account for global model translation
	curTranslation[currentStack][1] += g_translation[0];
	curTranslation[currentStack][2] += g_translation[1];
	curTranslation[currentStack][0] += g_translation[2];

	++currentStack;

	for ( i = 0; i < NUM_JOINTS_RAVEN; ++i )
	{
		while ( 1 )
		{
			TK_Beyond( TK_MODEL );

//			TK_BeyondRequire(TK_NAME, TK_STRING);

//			if(_stricmp(tk_String, SKEL_NAMES[NAME_OFFSETS[skelType]+i]) == 0)
			break;

			TK_Beyond( TK_SCALING );

			ParseVec3( curScale[currentStack] );

			TK_Beyond( TK_ROTATION );

			ParseRotation3( curRotation[currentStack] );

			TK_Beyond( TK_TRANSLATION );

			ParseVec3( curTranslation[currentStack] );

			curCorrespondingJoint[currentStack] = -1;

			++currentStack;
		}

		TK_Beyond( TK_SCALING );

		ParseVec3( curScale[currentStack] );

		TK_Beyond( TK_ROTATION );

		ParseRotation3( curRotation[currentStack] );

		jointList[i].rotation[1] = curRotation[currentStack][1];
		jointList[i].rotation[2] = curRotation[currentStack][2];
		jointList[i].rotation[0] = curRotation[currentStack][0];

		TK_Beyond( TK_TRANSLATION );

		ParseVec3( curTranslation[currentStack] );

		jointList[i].placement.origin[1] = curTranslation[currentStack][1];
		jointList[i].placement.origin[2] = curTranslation[currentStack][2];
		jointList[i].placement.origin[0] = curTranslation[currentStack][0];

		jointList[i].placement.direction[1] = 7.5;
		jointList[i].placement.direction[2] = 0.0;
		jointList[i].placement.direction[0] = 0.0;

		jointList[i].placement.up[1] = 0.0;
		jointList[i].placement.up[2] = 7.5;
		jointList[i].placement.up[0] = 0.0;

		curCorrespondingJoint[currentStack] = i;

		++currentStack;
	}

	stackSize = currentStack;

	for ( i = 0; i < NUM_JOINTS_RAVEN; ++i )
	{
		rx = jointList[i].rotation[0] * ANGLE_TO_RAD;
		ry = jointList[i].rotation[1] * ANGLE_TO_RAD;
		rz = jointList[i].rotation[2] * ANGLE_TO_RAD;

		cx = cos( rx );
		sx = sin( rx );

		cy = cos( ry );
		sy = sin( ry );

		cz = cos( rz );
		sz = sin( rz );

		// y-axis rotation for direction
		x2 = jointList[i].placement.direction[0] * cy + jointList[i].placement.direction[2] * sy;
		z2 = -jointList[i].placement.direction[0] * sy + jointList[i].placement.direction[2] * cy;
		jointList[i].placement.direction[0] = x2;
		jointList[i].placement.direction[2] = z2;

		// y-axis rotation for up
		x2 = jointList[i].placement.up[0] * cy + jointList[i].placement.up[2] * sy;
		z2 = -jointList[i].placement.up[0] * sy + jointList[i].placement.up[2] * cy;
		jointList[i].placement.up[0] = x2;
		jointList[i].placement.up[2] = z2;

		// z-axis rotation for direction
		x2 = jointList[i].placement.direction[0] * cz - jointList[i].placement.direction[1] * sz;
		y2 = jointList[i].placement.direction[0] * sz + jointList[i].placement.direction[1] * cz;
		jointList[i].placement.direction[0] = x2;
		jointList[i].placement.direction[1] = y2;

		// z-axis rotation for up
		x2 = jointList[i].placement.up[0] * cz - jointList[i].placement.up[1] * sz;
		y2 = jointList[i].placement.up[0] * sz + jointList[i].placement.up[1] * cz;
		jointList[i].placement.up[0] = x2;
		jointList[i].placement.up[1] = y2;

		// x-axis rotation for direction vector
		y2 = jointList[i].placement.direction[1] * cx - jointList[i].placement.direction[2] * sx;
		z2 = jointList[i].placement.direction[1] * sx + jointList[i].placement.direction[2] * cx;
		jointList[i].placement.direction[1] = y2;
		jointList[i].placement.direction[2] = z2;

		// x-axis rotation for up vector
		y2 = jointList[i].placement.up[1] * cx - jointList[i].placement.up[2] * sx;
		z2 = jointList[i].placement.up[1] * sx + jointList[i].placement.up[2] * cx;
		jointList[i].placement.up[1] = y2;
		jointList[i].placement.up[2] = z2;

		// translate to position in model
		jointList[i].placement.direction[0] += jointList[i].placement.origin[0];
		jointList[i].placement.direction[1] += jointList[i].placement.origin[1];
		jointList[i].placement.direction[2] += jointList[i].placement.origin[2];

		// translate to position in model
		jointList[i].placement.up[0] += jointList[i].placement.origin[0];
		jointList[i].placement.up[1] += jointList[i].placement.origin[1];
		jointList[i].placement.up[2] += jointList[i].placement.origin[2];
	}

	baseJoint = NUM_JOINTS_RAVEN;

	for ( i = stackSize /*NUM_JOINTS_RAVEN*/ - 1; i > 0; --i )
	{

		rx = curRotation[i - 1][0] * ANGLE_TO_RAD;
		ry = curRotation[i - 1][1] * ANGLE_TO_RAD;
		rz = curRotation[i - 1][2] * ANGLE_TO_RAD;

		cx = cos( rx );
		sx = sin( rx );

		cy = cos( ry );
		sy = sin( ry );

		cz = cos( rz );
		sz = sin( rz );

		for ( j = i - 1; j < stackSize - 1; ++j )
		{
			// y-axis rotation for origin
			x2 = jointList[j].placement.origin[0] * cy + jointList[j].placement.origin[2] * sy;
			z2 = -jointList[j].placement.origin[0] * sy + jointList[j].placement.origin[2] * cy;
			jointList[j].placement.origin[0] = x2;
			jointList[j].placement.origin[2] = z2;

			// y-axis rotation for direction
			x2 = jointList[j].placement.direction[0] * cy + jointList[j].placement.direction[2] * sy;
			z2 = -jointList[j].placement.direction[0] * sy + jointList[j].placement.direction[2] * cy;
			jointList[j].placement.direction[0] = x2;
			jointList[j].placement.direction[2] = z2;

			// y-axis rotation for up
			x2 = jointList[j].placement.up[0] * cy + jointList[j].placement.up[2] * sy;
			z2 = -jointList[j].placement.up[0] * sy + jointList[j].placement.up[2] * cy;
			jointList[j].placement.up[0] = x2;
			jointList[j].placement.up[2] = z2;

			// z-axis rotation for origin
			x2 = jointList[j].placement.origin[0] * cz - jointList[j].placement.origin[1] * sz;
			y2 = jointList[j].placement.origin[0] * sz + jointList[j].placement.origin[1] * cz;
			jointList[j].placement.origin[0] = x2;
			jointList[j].placement.origin[1] = y2;

			// z-axis rotation for direction
			x2 = jointList[j].placement.direction[0] * cz - jointList[j].placement.direction[1] * sz;
			y2 = jointList[j].placement.direction[0] * sz + jointList[j].placement.direction[1] * cz;
			jointList[j].placement.direction[0] = x2;
			jointList[j].placement.direction[1] = y2;

			// z-axis rotation for up
			x2 = jointList[j].placement.up[0] * cz - jointList[j].placement.up[1] * sz;
			y2 = jointList[j].placement.up[0] * sz + jointList[j].placement.up[1] * cz;
			jointList[j].placement.up[0] = x2;
			jointList[j].placement.up[1] = y2;

			// x-axis rotation for origin
			y2 = jointList[j].placement.origin[1] * cx - jointList[j].placement.origin[2] * sx;
			z2 = jointList[j].placement.origin[1] * sx + jointList[j].placement.origin[2] * cx;
			jointList[j].placement.origin[1] = y2;
			jointList[j].placement.origin[2] = z2;

			// x-axis rotation for direction vector
			y2 = jointList[j].placement.direction[1] * cx - jointList[j].placement.direction[2] * sx;
			z2 = jointList[j].placement.direction[1] * sx + jointList[j].placement.direction[2] * cx;
			jointList[j].placement.direction[1] = y2;
			jointList[j].placement.direction[2] = z2;

			// x-axis rotation for up vector
			y2 = jointList[j].placement.up[1] * cx - jointList[j].placement.up[2] * sx;
			z2 = jointList[j].placement.up[1] * sx + jointList[j].placement.up[2] * cx;
			jointList[j].placement.up[1] = y2;
			jointList[j].placement.up[2] = z2;

			if ( curCorrespondingJoint[j + 1] != -1 ) {
				// translate origin
				jointList[j].placement.origin[0] += curTranslation[i - 1][0];
				jointList[j].placement.origin[1] += curTranslation[i - 1][1];
				jointList[j].placement.origin[2] += curTranslation[i - 1][2];

				// translate back to local coord
				jointList[j].placement.direction[0] += curTranslation[i - 1][0];
				jointList[j].placement.direction[1] += curTranslation[i - 1][1];
				jointList[j].placement.direction[2] += curTranslation[i - 1][2];

				// translate back to local coord
				jointList[j].placement.up[0] += curTranslation[i - 1][0];
				jointList[j].placement.up[1] += curTranslation[i - 1][1];
				jointList[j].placement.up[2] += curTranslation[i - 1][2];
			}
		}
	}
}

void LoadGlobals( char *fileName ){
	FILE *file1;
	int dot = '.';
	char *dotstart;
	char InputFileName[256];

	dotstart = strrchr( fileName,dot ); // Does it already have an extension on the file name?

	if ( !dotstart ) {
		strcpy( InputFileName, fileName );
		strcat( InputFileName, ".hrc" );
		if ( ( file1 = fopen( InputFileName, "rb" ) ) != NULL ) {
			fclose( file1 );

			LoadHRCGlobals( InputFileName );

			printf( " - assuming .HRC\n" );
			return;
		}

		Error( "\n Could not open file '%s':\n"
			   "No HRC match.\n", fileName );
	}
	else
	{
		if ( ( file1 = fopen( fileName, "rb" ) ) != NULL ) {
			printf( "\n" );
			fclose( file1 );
			if ( strcmp( dotstart,".hrc" ) == 0 || strcmp( dotstart,".HRC" ) == 0 ) {
				LoadHRCGlobals( fileName );
				return;
			}
		}

		Error( "Could not open file '%s':\n",fileName );
	}
}

void LoadClusters( char *fileName, int **clusterList, int *num_verts, int skelType ){
	FILE *file1;
	int dot = '.';
	char *dotstart;
	char InputFileName[256];

	dotstart = strrchr( fileName,dot ); // Does it already have an extension on the file name?

	if ( !dotstart ) {
		strcpy( InputFileName, fileName );
		strcat( InputFileName, ".hrc" );
		if ( ( file1 = fopen( InputFileName, "rb" ) ) != NULL ) {
			fclose( file1 );

			LoadHRCClustered( InputFileName, clusterList, num_verts, skelType );

			printf( " - assuming .HRC\n" );
			return;
		}

		Error( "\n Could not open file '%s':\n"
			   "No HRC match.\n", fileName );
	}
	else
	{
		if ( ( file1 = fopen( fileName, "rb" ) ) != NULL ) {
			printf( "\n" );
			fclose( file1 );
			if ( strcmp( dotstart,".hrc" ) == 0 || strcmp( dotstart,".HRC" ) == 0 ) {
				LoadHRCClustered( fileName, clusterList, num_verts, skelType );
				return;
			}
		}

		Error( "Could not open file '%s':\n",fileName );
	}
}

void LoadJointList( char *fileName, QDataJoint_t *jointList, int skelType ){
	FILE *file1;
	int dot = '.';
	char *dotstart;
	char InputFileName[256];

	dotstart = strrchr( fileName,dot ); // Does it already have an extension on the file name?

	if ( !dotstart ) {
		strcpy( InputFileName, fileName );
		strcat( InputFileName, ".hrc" );
		if ( ( file1 = fopen( InputFileName, "rb" ) ) != NULL ) {
			fclose( file1 );

			LoadHRCJointList( InputFileName, jointList, skelType );

			printf( " - assuming .HRC\n" );
			return;
		}

		Error( "\n Could not open file '%s':\n"
			   "No HRC.\n", fileName );
	}
	else
	{
		if ( ( file1 = fopen( fileName, "rb" ) ) != NULL ) {
			printf( "\n" );
			fclose( file1 );
			if ( strcmp( dotstart,".hrc" ) == 0 || strcmp( dotstart,".HRC" ) == 0 ) {
				LoadHRCJointList( fileName, jointList, skelType );

				return;
			}
		}

		Error( "Could not open file '%s':\n",fileName );
	}
}