0ad/source/terrain/ParticleEmitter.cpp

//***********************************************************
//
// Name:		ParticleEmitter.cpp
// Last Update: 03/02/04
// Author:		Ben Vinegar
//
// Description: Particle class header
//
//
// To-do: - estimate particles to release / second
//        - randomness, range still needs to be implemented
//        - colour issues
//***********************************************************

#include "ParticleEmitter.h"
#include "time.h"
#include "ogl.h"
#include <stdlib.h>

CParticleEmitter::CParticleEmitter() :
	m_particles(NULL),
	m_origin(0.0f, 0.0f, 0.0f),
	m_originSpread(0.0f, 0.0f, 0.0f),
	m_velocity(0.0f, 0.0f, 0.0f),
	m_velocitySpread(0.0f, 0.0f, 0.0f),
	m_gravity(0.0f, 0.0f, 0.0f),
	m_maxParticles(0),
	m_minParticles(0),
	m_numParticles(0),
	m_maxLifetime(0),
	m_minLifetime(0),
	m_timeOfLastFrame(0.0f),
	m_timeSinceLastEmit(0.0f) {
		m_particles.clear();
}

CParticleEmitter::~CParticleEmitter() {
}

void CParticleEmitter::Frame() {
	Update();
	Render();
}

void CParticleEmitter::Render() {
	
	glEnable(GL_ALPHA_TEST);
	glEnable(GL_BLEND);
	glDisable(GL_DEPTH_TEST);

	glAlphaFunc(GL_GREATER, 0.0f);
	glBlendFunc(GL_SRC_ALPHA,GL_ONE);

	vector<CParticle *>::iterator itor = m_particles.begin();
	while (itor != m_particles.end()) {
		CParticle * curParticle = (*itor);
		
		curParticle->Frame();
		++itor;
	}
	glDisable(GL_ALPHA_TEST);
	glDisable(GL_BLEND);
	glEnable(GL_DEPTH_TEST);
}

void CParticleEmitter::Update() {
	float timeElapsed = get_time() - m_timeOfLastFrame;

	// update existing particles
	vector<CParticle *>::iterator itor = m_particles.begin();
	while (itor != m_particles.end()) {
		CParticle * curParticle = (*itor);

		curParticle->Update();

		// destroy particle if it has lived beyond its duration
		if (curParticle->m_timeElapsedTotal >= curParticle->m_duration) {
			m_particles.erase(itor);
			delete curParticle;
			--m_numParticles;
		}

		++itor;
	}

	float secondsPerEmit = 1 / (m_minParticles / m_minLifetime);

	if (m_timeSinceLastEmit > secondsPerEmit) {
		
		int duration;
		CVector3D position, velocity;
		float colour[4];

		bool moreParticlesToEmit = true;
		while (moreParticlesToEmit) {
			CParticle * newParticle = new CParticle();
			
			// calculate particle duration
			duration = m_minLifetime;
			duration += (rand() % (int)((m_maxLifetime - m_minLifetime) * 1000.0f + 1)) / 1000.0f;
			newParticle->m_duration = duration;

			// calculate particle start position from spread
			position = m_origin;
			position.X += (rand() % (int)(m_originSpread.X * 2000.0f + 1)) / 1000.0f - m_originSpread.X;
			position.Y += (rand() % (int)(m_originSpread.Y * 2000.0f + 1)) / 1000.0f - m_originSpread.Y;
			position.Z += (rand() % (int)(m_originSpread.Z * 2000.0f + 1)) / 1000.0f - m_originSpread.Z;
			newParticle->m_position = position;

			// calculate particle velocity from spread
			velocity = m_velocity;
			velocity.X += (rand() % (int)(m_velocitySpread.X * 2000.0f + 1)) / 1000.0f - m_velocitySpread.X;
			velocity.Y += (rand() % (int)(m_velocitySpread.Y * 2000.0f + 1)) / 1000.0f - m_velocitySpread.Y;
			velocity.Z += (rand() % (int)(m_velocitySpread.Z * 2000.0f + 1)) / 1000.0f - m_velocitySpread.Z;
			newParticle->m_velocity = velocity;

			newParticle->m_gravity = m_gravity;

			// calculate and assign colour
			memcpy(colour, m_startColour, sizeof(float) * 4);
			colour[0] += (rand() % (int)((m_endColour[0] - m_startColour[0]) * 1000.0f + 1)) / 1000.0f;
			colour[1] += (rand() % (int)((m_endColour[1] - m_startColour[1]) * 1000.0f + 1)) / 1000.0f;
			colour[2] += (rand() % (int)((m_endColour[2] - m_startColour[2]) * 1000.0f + 1)) / 1000.0f;
			colour[3] += (rand() % (int)((m_endColour[3] - m_startColour[3]) * 1000.0f + 1)) / 1000.0f;
			memcpy(newParticle->m_colour, colour, sizeof(float) * 4);

			// assign sprite
			newParticle->m_sprite = m_sprite;

			// final pre-processing init call
			newParticle->Init();

			// add to vector of particles
			m_particles.push_back(newParticle);

			timeElapsed -= secondsPerEmit;
			if (timeElapsed < secondsPerEmit) {
				moreParticlesToEmit = false;
			}


			++m_numParticles;
		}
		m_timeSinceLastEmit = 0.0f;
	}
	else
		m_timeSinceLastEmit += timeElapsed;

	m_timeOfLastFrame = get_time();
}

void CParticleEmitter::SetSprite(CSprite * sprite) {
	m_sprite = sprite;
}

void CParticleEmitter::SetOrigin(CVector3D origin) {
	m_origin = origin;
}

void CParticleEmitter::SetOrigin(float x, float y, float z) {
	m_origin.X = x;
	m_origin.Y = y;
	m_origin.Z = z;
}

void CParticleEmitter::SetOriginSpread(CVector3D spread) {
	m_originSpread = spread;
}

void CParticleEmitter::SetOriginSpread(float x, float y, float z) {
	m_originSpread.X = x;
	m_originSpread.Y = y;
	m_originSpread.Z = z;
}

void CParticleEmitter::SetGravity(CVector3D gravity) {
	m_gravity = gravity;
}

void CParticleEmitter::SetGravity(float x, float y, float z) {
	m_gravity.X = x;
	m_gravity.Y = y;
	m_gravity.Z = z;

}

void CParticleEmitter::SetVelocity(CVector3D velocity) {
	m_velocity = velocity;
}

void CParticleEmitter::SetVelocity(float x, float y, float z) {
	m_velocity.X = x;
	m_velocity.Y = y;
	m_velocity.Z = z;
}


void CParticleEmitter::SetVelocitySpread(CVector3D spread) {
	m_velocitySpread = spread;
}

void CParticleEmitter::SetVelocitySpread(float x, float y, float z) {
	m_velocitySpread.X = x;
	m_velocitySpread.Y = y;
	m_velocitySpread.Z = z;
}

void CParticleEmitter::SetStartColour(float r, float g, float b, float a) {
	m_startColour[0] = r;
	m_startColour[1] = g;
	m_startColour[2] = b;
	m_startColour[3] = a;
}

void CParticleEmitter::SetEndColour(float r, float g, float b, float a) {
	m_endColour[0] = r;
	m_endColour[1] = g;
	m_endColour[2] = b;
	m_endColour[3] = a;
}

void CParticleEmitter::SetMaxLifetime(double maxLife) {
	m_maxLifetime = maxLife;
}

void CParticleEmitter::SetMinLifetime(double minLife) {
	m_minLifetime = minLife;
}

void CParticleEmitter::SetMaxParticles(int maxParticles) {
	m_maxParticles = maxParticles;
}

void CParticleEmitter::SetMinParticles(int minParticles) {
	m_minParticles = minParticles;
}