#include "fuel_computer.h"
#include "map.h"

EXTERN_ENGINE;

mass_t FuelComputerBase::getCycleFuel(mass_t airmass, int rpm, float load) const {
	load = getTargetLambdaLoadAxis(load);
	
	float stoich = getStoichiometricRatio();
	float lambda = getTargetLambda(rpm, load);
	float afr = stoich * lambda;

	ENGINE(engineState.currentAfrLoad) = load;
	ENGINE(engineState.targetAFR) = afr;

	return airmass / afr;
}

FuelComputer::FuelComputer(const ValueProvider3D& afrTable) : m_afrTable(&afrTable) {}

float FuelComputer::getStoichiometricRatio() const {
	// TODO: vary this with ethanol content/configured setting/whatever
	float rawConfig = (float)CONFIG(stoichRatioPrimary) / PACK_MULT_AFR_CFG;

	// Config compatibility: this field may be zero on ECUs with old defaults
	if (rawConfig < 5) {
		return 14.7f;
	}

	return rawConfig;
}

float FuelComputer::getTargetLambda(int rpm, float load) const {
	efiAssert(OBD_PCM_Processor_Fault, m_afrTable != nullptr, "AFR table null", 0);

	// TODO: set the table value in lambda instead of afr
	return m_afrTable->getValue(rpm, load) / 14.7f;
}

float FuelComputer::getTargetLambdaLoadAxis(float defaultLoad) const {
	return getLoadOverride(defaultLoad, CONFIG(afrOverrideMode) PASS_ENGINE_PARAMETER_SUFFIX);
}

float getLoadOverride(float defaultLoad, afr_override_e overrideMode DECLARE_ENGINE_PARAMETER_SUFFIX) {
	switch(overrideMode) {
		case AFR_None: return defaultLoad;
		case AFR_MAP: return getMap(PASS_ENGINE_PARAMETER_SIGNATURE);
		// TPS/pedal default to 100% - failed TPS goes rich
		case AFR_Tps: return Sensor::get(SensorType::Tps1).value_or(100);
		case AFR_AccPedal: return Sensor::get(SensorType::AcceleratorPedal).value_or(100);
		case AFR_CylFilling: return 100 * ENGINE(engineState.sd.airMassInOneCylinder) / ENGINE(standardAirCharge);
		default: return 0;
	}
}