#include "cParticle.h"
#include <iostream>
using namespace std;

// Methods
// Firework Size
int cParticle::fireworkSize(string size)
{
	if (size == "large") {
		// Large
		return 15;
	}
	else if (size == "medium") {
		// Medium
		return 10;
	}
	else if (size == "small") {
		// Small
		return 5;
	}
	else {
		// Default to small
		return 5;
	}
}

// Firework Colour
float* cParticle::fireworkColour(string colour, float *colourArray)
{
	if (colour == "red") {
		// Red Colour
		colourArray[0] = 5.0f;
		colourArray[1] = 0.0f;
		colourArray[2] = 0.0f;
	}
	else if (colour == "blue") {
		// Blue Colour
		colourArray[0] = 0.0f;
		colourArray[1] = 0.0f;
		colourArray[2] = 5.0f;
	}
	else if (colour == "green") {
		// Green Colour
		colourArray[0] = 0.0f;
		colourArray[1] = 5.0f;
		colourArray[2] = 0.0f;
	}
	else {
		// Default to red
		colourArray[0] = 0.0f;
		colourArray[1] = 0.0f;
		colourArray[2] = 0.0f;
	}

	return colourArray;
}