#include "simulation/ElementCommon.h"

static int update(UPDATE_FUNC_ARGS);
static int graphics(GRAPHICS_FUNC_ARGS);
static void create(ELEMENT_CREATE_FUNC_ARGS);

void Element::Element_ELEC()
{
	Identifier = "DEFAULT_PT_ELEC";
	Name = "ELEC";
	Colour = 0xDFEFFF_rgb;
	MenuVisible = 1;
	MenuSection = SC_NUCLEAR;
	Enabled = 1;

	Advection = 0.0f;
	AirDrag = 0.00f * CFDS;
	AirLoss = 1.00f;
	Loss = 1.00f;
	Collision = -0.99f;
	Gravity = 0.0f;
	Diffusion = 0.00f;
	HotAir = 0.000f	* CFDS;
	Falldown = 0;

	Flammable = 0;
	Explosive = 0;
	Meltable = 0;
	Hardness = 0;

	Weight = -1;

	DefaultProperties.temp = R_TEMP + 200.0f + 273.15f;
	HeatConduct = 251;
	Description = "Electrons. Sparks electronics, reacts with NEUT and WATR.";

	Properties = TYPE_ENERGY|PROP_LIFE_DEC|PROP_LIFE_KILL_DEC;

	LowPressure = IPL;
	LowPressureTransition = NT;
	HighPressure = IPH;
	HighPressureTransition = NT;
	LowTemperature = ITL;
	LowTemperatureTransition = NT;
	HighTemperature = ITH;
	HighTemperatureTransition = NT;

	ASSIGN_SIM_CALLBACK(Update, update)
	Graphics = &graphics;
	ASSIGN_SIM_CALLBACK(Create, create)
}

static int update(UPDATE_FUNC_ARGS)
{
	auto &sd = SimulationData::CRef();
	auto &elements = sd.elements;
	for (auto rx = -2; rx <= 2; rx++)
	{
		for (auto ry = -2; ry <= 2; ry++)
		{
			auto r = pmap[y+ry][x+rx];
			if (!r)
				r = sim->photons[y+ry][x+rx];
			if (!r)
				continue;
			auto rt = TYP(r);
			switch (rt)
			{
			case PT_GLAS:
				for (auto rrx=-1; rrx<=1; rrx++)
				{
					for (auto rry=-1; rry<=1; rry++)
					{
						if (x+rx+rrx>=0 && y+ry+rry>=0 && x+rx+rrx<XRES && y+ry+rry<YRES) {
							//@ ELEC + GLAS -> EMBR + GLAS
							auto nb = sim->create_part(-1, x+rx+rrx, y+ry+rry, PT_EMBR);
							if (nb!=-1) {
								parts[nb].tmp = 0;
								parts[nb].life = 50;
								parts[nb].temp = parts[i].temp*0.8f;
								parts[nb].vx = float(rng.between(-10, 10));
								parts[nb].vy = float(rng.between(-10, 10));
							}
						}
					}
				}
				sim->kill_part(i);
				return 1;
			case PT_LCRY:
				parts[ID(r)].tmp2 = rng.between(5, 9);
				break;
			case PT_WATR:
			case PT_DSTW:
			case PT_SLTW:
			case PT_CBNW:
				//@ ELEC + WATR/DSTW/SLTW/CBNW -> O2/H2
				if (rng.chance(1, 3))
					sim->create_part(ID(r), x+rx, y+ry, PT_O2);
				else
					sim->create_part(ID(r), x+rx, y+ry, PT_H2);
				sim->kill_part(i);
				return 1;
			case PT_PROT: // this is the correct reaction, not NEUT, but leaving NEUT in anyway
				if (parts[ID(r)].tmp2 & 0x1)
					break;
			case PT_NEUT:
				//@ ELEC + PROT/NEUT -> H2
				sim->part_change_type(ID(r), x+rx, y+ry, PT_H2);
				parts[ID(r)].life = 0;
				parts[ID(r)].ctype = 0;
				sim->kill_part(i);
				break;
			case PT_DEUT:
				if(parts[ID(r)].life < 6000)
					parts[ID(r)].life += 1;
				parts[ID(r)].temp = 0;
				sim->kill_part(i);
				return 1;
			case PT_EXOT:
				parts[ID(r)].tmp2 += 5;
				parts[ID(r)].life = 1000;
				break;
			case PT_RSST: //Destroy RSST
				if(!rx && !ry)
				{
					sim->kill_part(ID(r));
					sim->kill_part(i);

					return 1;
				}
				break;
			case PT_NONE: //seems to speed up ELEC even if it isn't used
				break;
			default:
				if ((elements[rt].Properties & PROP_CONDUCTS) && (rt!=PT_NBLE||parts[i].temp<2273.15))
				{
					sim->create_part(-1, x+rx, y+ry, PT_SPRK);
					sim->kill_part(i);
					return 1;
				}
				break;
			}
		}
	}
	return 0;
}

static int graphics(GRAPHICS_FUNC_ARGS)
{
	*firea = 70;
	*firer = *colr;
	*fireg = *colg;
	*fireb = *colb;

	*pixel_mode |= FIRE_ADD;
	return 0;
}

static void create(ELEMENT_CREATE_FUNC_ARGS)
{
	float a = rng.between(0, 359) * std::numbers::pi_v<float> / 180.0f;
	sim->parts[i].life = 680;
	sim->parts[i].vx = 2.0f * cosf(a);
	sim->parts[i].vy = 2.0f * sinf(a);
}
