/**
 * @file	test_accel_enrichment.cpp
 *
 * See also test_fuel_wall_wetting.cpp
 *
 * @date Apr 29, 2014
 *  	Author: Dmitry Sidin
 * @author Andrey Belomutskiy, (c) 2012-2020
 */

#include "pch.h"

#include "accel_enrichment.h"

TEST(fuel, testTpsAccelEnrichmentMath) {
	printf("====================================================================================== testAccelEnrichment\r\n");

	EngineTestHelper eth(engine_type_e::FORD_ASPIRE_1996);

	engine->rpmCalculator.setRpmValue(600);
	engine->periodicFastCallback();

	engine->tpsAccelEnrichment.setLength(4);

	engine->tpsAccelEnrichment.onNewValue(0);
	ASSERT_EQ( 0,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta";
	engine->tpsAccelEnrichment.onNewValue(10);
	ASSERT_EQ( 10,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta#1";
	engine->tpsAccelEnrichment.onNewValue(30);
	ASSERT_EQ( 20,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta#2";

	engine->tpsAccelEnrichment.onNewValue(0);
	ASSERT_EQ( 20,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta#3";
	engine->tpsAccelEnrichment.onNewValue(0);
	ASSERT_EQ( 20,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta#4";
	engine->tpsAccelEnrichment.onNewValue(0);
	ASSERT_EQ( 0,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta#5";
	engine->tpsAccelEnrichment.onNewValue(0);
	ASSERT_EQ( 0,  engine->tpsAccelEnrichment.getMaxDelta()) << "maxDelta";
}

TEST(fuel, testTpsAccelEnrichmentScheduling) {
	EngineTestHelper eth(engine_type_e::FORD_ASPIRE_1996);

	setCrankOperationMode();

	engineConfiguration->tpsAccelEnrichmentThreshold = 5;
	engineConfiguration->tpsAccelLookback = 2;

	eth.setTriggerType(trigger_type_e::TT_HALF_MOON);


	Sensor::setMockValue(SensorType::Tps1, 0);
	engine->periodicSlowCallback();
	Sensor::setMockValue(SensorType::Tps1, 7);
	engine->periodicSlowCallback();

	eth.fireTriggerEvents2(/* count */ 4, 25 /* ms */);
	ASSERT_EQ( 1200,  Sensor::getOrZero(SensorType::Rpm)) << "RPM";
	int expectedInvocationCounter = 1;
	ASSERT_EQ(expectedInvocationCounter, engine->tpsAccelEnrichment.onUpdateInvocationCounter);

	Sensor::setMockValue(SensorType::Tps1, 70);
	eth.fireTriggerEvents2(/* count */ 1, 25 /* ms */);

	float expectedAEValue = 7;
	// it does not matter how many times we invoke 'getTpsEnrichment' - state does not change
	for (int i = 0; i <20;i++) {
		ASSERT_NEAR(expectedAEValue, engine->tpsAccelEnrichment.getTpsEnrichment(), EPS4D);
	}

	expectedInvocationCounter++;
	ASSERT_EQ(expectedInvocationCounter, engine->tpsAccelEnrichment.onUpdateInvocationCounter);

	eth.engine.periodicFastCallback();
	eth.engine.periodicFastCallback();
	eth.engine.periodicFastCallback();

	ASSERT_EQ(expectedInvocationCounter, engine->tpsAccelEnrichment.onUpdateInvocationCounter);
}

static void doFractionalTpsIteration(int period, int divisor, int numCycles, std::vector<floatms_t> &tpsEnrich) {
	// every cycle
	engineConfiguration->tpsAccelFractionPeriod = period;
	// split into 2 portions
	engineConfiguration->tpsAccelFractionDivisor = divisor;

	engine->tpsAccelEnrichment.resetAE();
	engine->tpsAccelEnrichment.onNewValue(0);
	for (int i = 0; i < numCycles; i++) {
		engine->tpsAccelEnrichment.onNewValue(10);
		engine->tpsAccelEnrichment.onEngineCycleTps();
		tpsEnrich[i] = engine->tpsAccelEnrichment.getTpsEnrichment();
	}
}

TEST(fuel, testAccelEnrichmentFractionalTps) {
	printf("====================================================================================== testAccelEnrichmentFractionalTps\r\n");

	EngineTestHelper eth(engine_type_e::FORD_ASPIRE_1996);

	// setup
	engineConfiguration->tpsAccelEnrichmentThreshold = 5;
	engineConfiguration->tpsAccelLookback = 2;

	// fill tps2tps map (todo: there should be a better way?)
	static const float tpsTpsConst = 1.0f;
	for (int loadIndex = 0; loadIndex < TPS_TPS_ACCEL_TABLE; loadIndex++) {
		for (int rpmIndex = 0; rpmIndex < TPS_TPS_ACCEL_TABLE; rpmIndex++) {
			config->tpsTpsAccelTable[loadIndex][rpmIndex] = tpsTpsConst;
		}
	}

	initAccelEnrichment();

	engine->rpmCalculator.setRpmValue(600);
	engine->periodicFastCallback();

	engine->tpsAccelEnrichment.setLength(2);


	const int numCycles = 4;
	std::vector<floatms_t> tpsEnrich(numCycles);

	// first, test the default behavior without fractional division
	doFractionalTpsIteration(1, 1, numCycles, tpsEnrich);
	// the portion for the first cycle is full-enriched and there's no enrichment for next cycles
	EXPECT_THAT(tpsEnrich, testing::ElementsAre(1.0f, 0.0f, 0.0f, 0.0f)) << "fractionalTps#1";

	// divide into 2 each cycle
	doFractionalTpsIteration(1, 2, numCycles, tpsEnrich);
	// we have half-portion for the first cycle, then 1/4-th and 1/8th and so on...
	EXPECT_THAT(tpsEnrich, testing::ElementsAre(0.5f, 0.25f, 0.125f, 0.0625f)) << "fractionalTps#2";

	// split every portion into 3 sub-portions (so the whole enrichment takes longer)
	doFractionalTpsIteration(3, 1, numCycles, tpsEnrich);
	// we have 1/3rd-portion for the first three cycles
	const float th = (1.0f / 3.0f);
	EXPECT_THAT(tpsEnrich, testing::ElementsAre(testing::FloatEq(th), testing::FloatEq(th), testing::FloatEq(th), 0.0f)) << "fractionalTps#3";

	// split every divided portion into 2 sub-portions (so the whole enrichment takes longer)
	doFractionalTpsIteration(2, 2, numCycles, tpsEnrich);
	// we have half-portion for the first two cycles, and 1/4-th portion for the next 2 cycles, and so on...
	EXPECT_THAT(tpsEnrich, testing::ElementsAre(0.25f, 0.25f, 0.125f, 0.125f)) << "fractionalTps#4";

}