#include "fluid_holder.hpp"
#include "../util/constants.hpp"
#include "../conversions/temperature.hpp"
#include "../reactor/fuel/half_life.hpp"

#include <cmath>
#include <iostream>

using namespace sim::coolant;

fluid_holder::fluid_holder(fluid_t fluid, double volume, double extra_mass) : fluid(fluid), volume(volume), extra_mass(extra_mass)
{

}

fluid_holder::fluid_holder(const Json::Value& node) : fluid(node["fluid"]), volume(node["volume"].asDouble()), extra_mass(node["extra_mass"].asDouble())
{
	level = node["level"].asDouble();
	steam = node["steam"].asDouble();
	air = node["air"].asDouble();
	heat = node["heat"].asDouble();
}

double fluid_holder::add_heat(double m1, double t1)
{
	double t2 = get_heat();
	double t = t1 - t2;
	double m2 = get_thermal_mass();
	double m = m1 + m2;

	if(m1 == 0 || m2 == 0)
		return t1;
	
   	heat = t1 - t * m2 / m;

	return heat;
}

double fluid_holder::add_fluid(double v2, double t2)
{
	if(level + v2 <= 0)
	{
		return 0;
	}
	
	if(level + v2 > volume - 1e-3)
	{
		v2 = volume - level - 1e-3;
	}

	double m1 = get_thermal_mass();
	double m2 = fluid.l_to_g(v2);
	
	double t1 = get_heat();
	double t = t1 - t2;

	heat = t1 - t * m2 / (m1 + m2);
	level += v2;

	return v2;
}

double fluid_holder::extract_fluid(double amount)
{
	if(amount < level - 1e-3)
	{
		level -= amount - 1e-3;
	}

	else
	{
		amount = level;
		level = 0;
	}
	
	return amount;
}

void fluid_holder::add_gas(double m_s2, double m_a2, double t_2, double e_2)
{
	double m_2 = m_a2 + m_s2;
	double m_1 = get_thermal_mass();
	double t_1 = heat;
	
	double m = m_1 + m_2;

	if(m > 0)
	{
		heat = t_1 - (t_1 - t_2) * m_2 / m;
	}

	steam += m_s2;
	air += m_a2;
}

double fluid_holder::calc_pressure(double heat, double volume, double mol)
{
	double V = volume * 0.001;

	return V == 0 ? 0 : (mol * heat * constants::R) / V;
}

double fluid_holder::calc_pressure_mol(double heat, double volume, double pressure)
{
	double V = volume * 0.001;

	return (pressure * V) / (constants::R * heat);
}

double fluid_holder::get_pressure() const
{
	return calc_pressure(conversions::temperature::c_to_k(heat), get_gas_volume(), fluid.g_to_mol(steam) + air / constants::M_air);
}

double fluid_holder::get_gas_density() const
{
	double v = get_gas_volume();
	return v > 0 ? get_gas() / v : 0;
}

fluid_holder::operator Json::Value() const
{
	Json::Value node;

	node["level"] = level;
	node["steam"] = steam;
	node["air"] = air;
	node["heat"] = heat;

	node["volume"] = volume;
	node["extra_mass"] = extra_mass;
	node["fluid"] = fluid;

	return node;
}

void fluid_holder::update_base(double secs)
{
	double mass = get_thermal_mass();

	if(mass > 0)
	{
		double K = conversions::temperature::c_to_k(heat);	// K
		double P = fluid.vapor_pressure.calc_p(K);			// Pa
		double R = sim::constants::R;						// J/K/mol

		double J_m = fluid.jPg * fluid.gPmol;							// J/mol
		double n_g = fluid.g_to_mol(steam) + air / constants::M_air;	// mol
		double V_g = (volume - level) * 0.001;							// m^3

		double n = (P * V_g) / (R * K) - n_g;	// mol
		double l = level - fluid.mol_to_l(n);	// L

		if(l < 0)
		{
			n -= fluid.l_to_mol(l);
			l = 0;
		}
		
		steam += fluid.mol_to_g(n);

		if(steam < 0)
		{
			l += fluid.g_to_l(steam);
			n -= fluid.g_to_mol(steam);
			steam = 0;
		}
		
		level = l;
		heat -= n * J_m / mass;
	}
}