#include "stdafx.h"
#include "doctest.h"
#include "vehicle.h"

using test::definition;
using test::vehicle;

// the compressor waiting a while after getting its power before it starts (CompressorStartDelay,
// MaSzyna-EU07/maszyna#67): in a diesel it starts some time after the engine, rather than with it

namespace {

double const step{0.01};

// a compressor fed from the main circuit, which starts on its own, and an empty main reservoir for it to fill
vehicle compressor_vehicle(double const Delay) {
	auto definition{definition::load("wagon")};
	definition.set("Brake", "Vv", 1.0).set("Brake", "MinCP", 0.65).set("Brake", "MaxCP", 0.8);
	definition.set("Brake", "CompressorSpeed", 0.01).set("Brake", "CompressorPower", "Main");
	definition.set("Cntrl", "CompressorStart", "Automatic");
	if (Delay > 0.0) {
		definition.set("Brake", "CompressorStartDelay", Delay);
	}
	vehicle vehicle{definition};
	vehicle->CompressedVolume = 0.0;
	vehicle->LastSwitchingTime = 10.0; // nothing was switched lately
	return vehicle;
}

// runs the compressor for Seconds, the main circuit as it is, the time since the last switching going on as in
// ComputeTotalForce()
void run_compressor(vehicle &Vehicle, double const Seconds) {
	for (auto i{std::lround(Seconds / step)}; i > 0; --i) {
		Vehicle->LastSwitchingTime += step;
		Vehicle->CompressorCheck(step);
	}
}

} // namespace

TEST_CASE("compressor: without a start delay the compressor starts with its power") {
	auto vehicle{compressor_vehicle(0.0)};
	run_compressor(vehicle, 1.0);
	REQUIRE_FALSE(vehicle->CompressorFlag); // no power, no compressor

	vehicle->Mains = true;
	run_compressor(vehicle, step);
	CHECK(vehicle->CompressorFlag);
}

TEST_CASE("compressor: the compressor starts the delay after getting its power") {
	auto vehicle{compressor_vehicle(5.0)};
	vehicle->Mains = true;

	run_compressor(vehicle, 4.9);
	CHECK_FALSE(vehicle->CompressorFlag);
	CHECK(vehicle->Compressor == doctest::Approx(0.0));

	run_compressor(vehicle, 0.2);
	CHECK(vehicle->CompressorFlag);
	run_compressor(vehicle, 10.0);
	CHECK(vehicle->Compressor > 0.0); // and fills the reservoir
}

TEST_CASE("compressor: the delay starts over once the power is gone") {
	auto vehicle{compressor_vehicle(5.0)};
	vehicle->Mains = true;
	run_compressor(vehicle, 10.0);
	REQUIRE(vehicle->CompressorFlag);

	vehicle->Mains = false; // the engine stopped
	run_compressor(vehicle, 1.0);
	CHECK_FALSE(vehicle->CompressorFlag);

	vehicle->Mains = true; // and started again
	run_compressor(vehicle, 4.9);
	CHECK_FALSE(vehicle->CompressorFlag);
	run_compressor(vehicle, 0.2);
	CHECK(vehicle->CompressorFlag);
}