// * This file is part of the COLOBOT source code
// * Copyright (C) 2001-2008, Daniel ROUX & EPSITEC SA, www.epsitec.ch
// *
// * 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 3 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, see  http://www.gnu.org/licenses/.

#define STRICT
#define D3D_OVERLOADS

#include <windows.h>
#include <stdio.h>
#include <d3d.h>

#include "struct.h"
#include "D3DEngine.h"
#include "D3DMath.h"
#include "math3d.h"
#include "event.h"
#include "misc.h"
#include "iman.h"
#include "terrain.h"
#include "water.h"
#include "object.h"
#include "pyro.h"
#include "physics.h"
#include "brain.h"
#include "camera.h"
#include "motion.h"
#include "motionhuman.h"
#include "sound.h"
#include "task.h"
#include "taskflag.h"





// Constructeur de l'objet.

CTaskFlag::CTaskFlag(CInstanceManager* iMan, CObject* object)
					   : CTask(iMan, object)
{
	CTask::CTask(iMan, object);
}

// Destructeur de l'objet.

CTaskFlag::~CTaskFlag()
{
}


// Gestion d'un �v�nement.

BOOL CTaskFlag::EventProcess(const Event &event)
{
	if ( m_bError )  return TRUE;
	if ( m_engine->RetPause() )  return TRUE;
	if ( event.event != EVENT_FRAME )  return TRUE;

	m_time += event.rTime;

	return TRUE;
}



// Assigne le but � atteindre.

Error CTaskFlag::Start(TaskFlagOrder order, int rank)
{
	D3DVECTOR	pos, speed;
	Error		err;

	m_order = order;
	m_time = 0.0f;

	m_bError = TRUE;  // op�ration impossible
	if ( !m_physics->RetLand() )
	{
		pos = m_object->RetPosition(0);
		if ( pos.y < m_water->RetLevel() )  return ERR_FLAG_WATER;
		return ERR_FLAG_FLY;
	}

	speed = m_physics->RetMotorSpeed();
	if ( speed.x != 0.0f ||
		 speed.z != 0.0f )  return ERR_FLAG_MOTOR;

	if ( m_object->RetFret() != 0 )  return ERR_FLAG_BUSY;

	if ( order == TFL_CREATE )
	{
		err = CreateFlag(rank);
		if ( err != ERR_OK )  return err;
	}

	if ( order == TFL_DELETE )
	{
		err = DeleteFlag();
		if ( err != ERR_OK )  return err;
	}

	m_bError = FALSE;

	m_motion->SetAction(MHS_FLAG);  // met/enl�ve drapeau
	m_camera->StartCentering(m_object, PI*0.3f, 99.9f, 0.0f, 0.5f);

	return ERR_OK;
}

// Indique si l'action est termin�e.

Error CTaskFlag::IsEnded()
{
	if ( m_engine->RetPause() )  return ERR_CONTINUE;

	if ( m_bError )  return ERR_STOP;
	if ( m_time < 2.0f )  return ERR_CONTINUE;

	Abort();
	return ERR_STOP;
}

// Termine brutalement l'action en cours.

BOOL CTaskFlag::Abort()
{
	m_motion->SetAction(-1);
	m_camera->StopCentering(m_object, 2.0f);
	return TRUE;
}



// Retourne l'objet le plus proche d'une position donn�e.

CObject* CTaskFlag::SearchNearest(D3DVECTOR pos, ObjectType type)
{
	ObjectType	oType;
	CObject		*pObj, *pBest;
	D3DVECTOR	oPos;
	float		min, dist;
	int			i;

	min = 100000.0f;
	pBest = 0;
	for ( i=0 ; i<1000000 ; i++ )
	{
		pObj = (CObject*)m_iMan->SearchInstance(CLASS_OBJECT, i);
		if ( pObj == 0 )  break;

		if ( !pObj->RetEnable() )  continue;

		oType = pObj->RetType();
		if ( type == OBJECT_NULL )
		{
			if ( oType != OBJECT_FLAGb &&
				 oType != OBJECT_FLAGr &&
				 oType != OBJECT_FLAGg &&
				 oType != OBJECT_FLAGy &&
				 oType != OBJECT_FLAGv )  continue;
		}
		else
		{
			if ( oType != type )  continue;
		}

		oPos = pObj->RetPosition(0);
		dist = Length2d(oPos, pos);
		if ( dist < min )
		{
			min = dist;
			pBest = pObj;
		}
	}
	return pBest;
}

// Compte le nombre d'objets existants.

int CTaskFlag::CountObject(ObjectType type)
{
	ObjectType	oType;
	CObject		*pObj;
	D3DVECTOR	oPos;
	int			i, count;

	count = 0;
	for ( i=0 ; i<1000000 ; i++ )
	{
		pObj = (CObject*)m_iMan->SearchInstance(CLASS_OBJECT, i);
		if ( pObj == 0 )  break;

		if ( !pObj->RetEnable() )  continue;

		oType = pObj->RetType();
		if ( type == OBJECT_NULL )
		{
			if ( oType != OBJECT_FLAGb &&
				 oType != OBJECT_FLAGr &&
				 oType != OBJECT_FLAGg &&
				 oType != OBJECT_FLAGy &&
				 oType != OBJECT_FLAGv )  continue;
		}
		else
		{
			if ( oType != type )  continue;
		}

		count ++;
	}
	return count;
}

// Cr�e un indicateur de couleur.

Error CTaskFlag::CreateFlag(int rank)
{
	CObject*	pObj;
	CObject*	pNew;
	CPyro*		pyro;
	D3DMATRIX*	mat;
	D3DVECTOR	pos;
	float		dist;
	int			i;

	ObjectType	table[5] =
	{
		OBJECT_FLAGb,
		OBJECT_FLAGr,
		OBJECT_FLAGg,
		OBJECT_FLAGy,
		OBJECT_FLAGv,
	};

	mat = m_object->RetWorldMatrix(0);
	pos = Transform(*mat, D3DVECTOR(4.0f, 0.0f, 0.0f));

	pObj = SearchNearest(pos, OBJECT_NULL);
	if ( pObj != 0 )
	{
		dist = Length(pos, pObj->RetPosition(0));
		if ( dist < 10.0f )
		{
			return ERR_FLAG_PROXY;
		}
	}

	i = rank;
	if ( CountObject(table[i]) >= 5 )
	{
		return ERR_FLAG_CREATE;
	}

	pNew = new CObject(m_iMan);
	if ( !pNew->CreateFlag(pos, 0.0f, table[i]) )
	{
		delete pNew;
		return ERR_TOOMANY;
	}
	pNew->SetZoom(0, 0.0f);

	m_sound->Play(SOUND_WAYPOINT, pos);
	pyro = new CPyro(m_iMan);
	pyro->Create(PT_FLCREATE, pNew);

	return ERR_OK;
}

// Supprime un indicateur de couleur.

Error CTaskFlag::DeleteFlag()
{
	CObject*	pObj;
	CPyro*		pyro;
	D3DVECTOR	iPos, oPos;
	float		iAngle, angle, aLimit, dist;

	iPos = m_object->RetPosition(0);
	iAngle = m_object->RetAngleY(0);
	iAngle = NormAngle(iAngle);  // 0..2*PI

	pObj = SearchNearest(iPos, OBJECT_NULL);
	if ( pObj == 0 )
	{
		return ERR_FLAG_DELETE;
	}
	dist = Length(iPos, pObj->RetPosition(0));
	if ( dist > 10.0f )
	{
		return ERR_FLAG_DELETE;
	}

	oPos = pObj->RetPosition(0);
	angle = RotateAngle(oPos.x-iPos.x, iPos.z-oPos.z);  // CW !
	aLimit = 45.0f*PI/180.0f;
	if ( !TestAngle(angle, iAngle-aLimit, iAngle+aLimit) )
	{
		return ERR_FLAG_DELETE;
	}

	m_sound->Play(SOUND_WAYPOINT, iPos);
	pyro = new CPyro(m_iMan);
	pyro->Create(PT_FLDELETE, pObj);

	return ERR_OK;
}