colobot/src/object/auto/autopara.cpp

// * 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 "global.h"
#include "event.h"
#include "misc.h"
#include "iman.h"
#include "math3d.h"
#include "particule.h"
#include "light.h"
#include "terrain.h"
#include "camera.h"
#include "object.h"
#include "interface.h"
#include "button.h"
#include "window.h"
#include "sound.h"
#include "displaytext.h"
#include "cmdtoken.h"
#include "auto.h"
#include "autopara.h"




// Object's constructor.

CAutoPara::CAutoPara(CInstanceManager* iMan, CObject* object)
						  : CAuto(iMan, object)
{
	m_channelSound = -1;
	Init();
}

// Object's destructor.

CAutoPara::~CAutoPara()
{
	this->CAuto::~CAuto();
}


// Destroys the object.

void CAutoPara::DeleteObject(BOOL bAll)
{
	if ( m_channelSound != -1 )
	{
		m_sound->FlushEnvelope(m_channelSound);
		m_sound->AddEnvelope(m_channelSound, 0.0f, 1.0f, 1.0f, SOPER_STOP);
		m_channelSound = -1;
	}

	CAuto::DeleteObject(bAll);
}


// Initialize the object.

void CAutoPara::Init()
{
	D3DMATRIX*	mat;

	m_time = 0.0f;
	m_timeVirus = 0.0f;
	m_lastParticule = 0.0f;

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

	m_phase    = APAP_WAIT;  // waiting ...
	m_progress = 0.0f;
	m_speed    = 1.0f/1.0f;

	CAuto::Init();
}


// Reception of lightning.

void CAutoPara::StartBlitz()
{
	m_phase    = APAP_BLITZ;
	m_progress = 0.0f;
	m_speed    = 1.0f/2.0f;
}


// Management of an event.

BOOL CAutoPara::EventProcess(const Event &event)
{
	D3DVECTOR	pos, speed;
	FPOINT		dim;
	int			i;

	CAuto::EventProcess(event);

	if ( m_engine->RetPause() )  return TRUE;
	if ( event.event != EVENT_FRAME )  return TRUE;

	m_progress += event.rTime*m_speed;
	m_timeVirus -= event.rTime;

	if ( m_object->RetVirusMode() )  // contaminated by a virus?
	{
		if ( m_timeVirus <= 0.0f )
		{
			m_timeVirus = 0.1f+Rand()*0.3f;
		}
		return TRUE;
	}

	EventProgress(event.rTime);

	if ( m_phase == APAP_BLITZ )
	{
		if ( m_progress < 1.0f )
		{
			if ( m_lastParticule+m_engine->ParticuleAdapt(0.05f) <= m_time )
			{
				m_lastParticule = m_time;

				for ( i=0 ; i<10 ; i++ )
				{
					pos = m_object->RetPosition(0);
					pos.x += (Rand()-0.5f)*m_progress*40.0f;
					pos.z += (Rand()-0.5f)*m_progress*40.0f;
					pos.y += 50.0f-m_progress*50.0f;
					speed.x = (Rand()-0.5f)*20.0f;
					speed.z = (Rand()-0.5f)*20.0f;
					speed.y = 5.0f+Rand()*5.0f;
					dim.x = 2.0f;
					dim.y = dim.x;
					m_particule->CreateParticule(pos, speed, dim, PARTIBLITZ, 1.0f, 20.0f, 0.5f);
				}
			}
		}
		else
		{
			m_phase    = APAP_CHARGE;
			m_progress = 0.0f;
			m_speed    = 1.0f/2.0f;
		}
	}

	if ( m_phase == APAP_CHARGE )
	{
		if ( m_progress < 1.0f )
		{
			if ( m_lastParticule+m_engine->ParticuleAdapt(0.05f) <= m_time )
			{
				m_lastParticule = m_time;

				for ( i=0 ; i<2 ; i++ )
				{
					pos = m_object->RetPosition(0);
					pos.y += 16.0f;
					speed.x = (Rand()-0.5f)*10.0f;
					speed.z = (Rand()-0.5f)*10.0f;
					speed.y = -Rand()*30.0f;
					dim.x = 1.0f;
					dim.y = dim.x;
					m_particule->CreateParticule(pos, speed, dim, PARTIBLITZ, 1.0f, 0.0f, 0.0f);
				}
			}

			ChargeObject(event.rTime);
		}
		else
		{
			m_phase    = APAP_WAIT;
			m_progress = 0.0f;
			m_speed    = 1.0f/1.0f;
		}
	}

	return TRUE;
}


// Creates all the interface when the object is selected.

BOOL CAutoPara::CreateInterface(BOOL bSelect)
{
	CWindow*	pw;
	FPOINT		pos, ddim;
	float		ox, oy, sx, sy;

	CAuto::CreateInterface(bSelect);

	if ( !bSelect )  return TRUE;

	pw = (CWindow*)m_interface->SearchControl(EVENT_WINDOW0);
	if ( pw == 0 )  return FALSE;

	ox = 3.0f/640.0f;
	oy = 3.0f/480.0f;
	sx = 33.0f/640.0f;
	sy = 33.0f/480.0f;

	pos.x = ox+sx*0.0f;
	pos.y = oy+sy*0;
	ddim.x = 66.0f/640.0f;
	ddim.y = 66.0f/480.0f;
	pw->CreateGroup(pos, ddim, 113, EVENT_OBJECT_TYPE);

	pos.x = ox+sx*10.2f;
	pos.y = oy+sy*0.5f;
	ddim.x = 33.0f/640.0f;
	ddim.y = 33.0f/480.0f;
	pw->CreateButton(pos, ddim, 41, EVENT_OBJECT_LIMIT);

	return TRUE;
}


// Returns an error due the state of the automation.

Error CAutoPara::RetError()
{
	if ( m_object->RetVirusMode() )
	{
		return ERR_BAT_VIRUS;
	}
	return ERR_OK;
}


// Load all objects under the lightning rod.

void CAutoPara::ChargeObject(float rTime)
{
	CObject*	pObj;
	CObject*	power;
	D3DVECTOR	sPos, oPos;
	float		dist, energy;
	int			i;

	sPos = m_object->RetPosition(0);

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

		oPos = pObj->RetPosition(0);
		dist = Length(oPos, sPos);
		if ( dist > 20.0f )  continue;

		if ( pObj->RetTruck() == 0 && pObj->RetType() == OBJECT_POWER )
		{
			energy = pObj->RetEnergy();
			energy += rTime/2.0f;
			if ( energy > 1.0f )  energy = 1.0f;
			pObj->SetEnergy(energy);
		}

		power = pObj->RetPower();
		if ( power != 0 && power->RetType() == OBJECT_POWER )
		{
			energy = power->RetEnergy();
			energy += rTime/2.0f;
			if ( energy > 1.0f )  energy = 1.0f;
			power->SetEnergy(energy);
		}

		power = pObj->RetFret();
		if ( power != 0 && power->RetType() == OBJECT_POWER )
		{
			energy = power->RetEnergy();
			energy += rTime/2.0f;
			if ( energy > 1.0f )  energy = 1.0f;
			power->SetEnergy(energy);
		}
	}
}


// Saves all parameters of the controller.

BOOL CAutoPara::Write(char *line)
{
	char	name[100];

	if ( m_phase == APAP_WAIT )  return FALSE;

	sprintf(name, " aExist=%d", 1);
	strcat(line, name);

	CAuto::Write(line);

	sprintf(name, " aPhase=%d", m_phase);
	strcat(line, name);

	sprintf(name, " aProgress=%.2f", m_progress);
	strcat(line, name);

	sprintf(name, " aSpeed=%.2f", m_speed);
	strcat(line, name);

	return TRUE;
}

// Restores all parameters of the controller.

BOOL CAutoPara::Read(char *line)
{
	if ( OpInt(line, "aExist", 0) == 0 )  return FALSE;

	CAuto::Read(line);

	m_phase = (AutoParaPhase)OpInt(line, "aPhase", APAP_WAIT);
	m_progress = OpFloat(line, "aProgress", 0.0f);
	m_speed = OpFloat(line, "aSpeed", 1.0f);

	m_lastParticule = 0.0f;

	return TRUE;
}