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

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

// the brake pipe of a lone wagon is held at a pressure the way a coupled locomotive would hold it.
// pressures are in bar, the brake pipe runs at 5.0, a 1.5 bar reduction is a full service application.

TEST_CASE("brakes: KE wagon applies up to its tare limit on a service application and releases again") {
	vehicle wagon{definition::load("wagon")}; // empty wagon, TareMaxBP=2.3

	wagon.hold_brake_pipe(5.0);
	wagon.run(60.0);
	CHECK(wagon->BrakePress < 0.1);
	CHECK(wagon->Hamulec->GetBRP() > 4.5); // auxiliary reservoir charged

	wagon.hold_brake_pipe(3.5);
	wagon.run(10.0);
	CHECK(wagon->BrakePress == doctest::Approx(2.3).epsilon(0.05));

	wagon.hold_brake_pipe(5.0);
	wagon.run(60.0);
	CHECK(wagon->BrakePress < 0.1);
}

TEST_CASE("brakes: a deeper brake pipe reduction brakes harder") {
	auto applied = [](double Pipe) {
		vehicle wagon{definition::load("wagon").set("Brake", "TareMaxBP", 3.8)}; // no tare limit in the way
		wagon.hold_brake_pipe(5.0);
		wagon.run(60.0);
		wagon.hold_brake_pipe(Pipe);
		wagon.run(20.0);
		return wagon->BrakePress;
	};
	auto const light{applied(4.6)};
	auto const medium{applied(4.2)};
	auto const full{applied(3.5)};
	CHECK(light > 0.1);
	CHECK(light < medium);
	CHECK(medium < full);
	CHECK(full <= 3.8 + 0.1); // MaxBP
}

// every distributor valve the engine implements goes through the same cycle; the ESt family needs the size
// of its reservoirs, as in the real .fiz files using it
TEST_CASE("brakes: every implemented distributor valve charges, applies and releases") {
	for (auto const *valve : {"W", "K", "KE", "ESt", "ESt3", "NESt3", "ESt4", "ESt3AL2", "LSt", "EStED", "EP1", "EP2", "CV1"}) {
		CAPTURE(valve);
		vehicle wagon{definition::load("wagon").set("Brake", "BrakeValve", valve).set("Brake", "Size", 16)};
		wagon.hold_brake_pipe(5.0);
		wagon.run(90.0);
		CHECK(wagon->BrakePress < 0.1);
		wagon.hold_brake_pipe(3.5);
		wagon.run(20.0);
		CHECK(wagon->BrakePress > 1.0);
		CHECK(wagon->BrakePress <= 3.8 + 0.1); // MaxBP
		wagon.hold_brake_pipe(5.0);
		wagon.run(90.0);
		CHECK(wagon->BrakePress < 0.1);
	}
}

TEST_CASE("brakes: service braking stops a wagon running at 60 km/h and keeps it stopped") {
	vehicle wagon{definition::load("wagon"), 60.0};
	wagon.hold_brake_pipe(5.0);
	wagon.run(10.0);
	REQUIRE(wagon.speed() > 50.0); // still running before the braking

	wagon.hold_brake_pipe(3.5);
	wagon.run(30.0);
	CHECK(std::abs(wagon.speed()) < 0.1);
	wagon.run(30.0);
	CHECK(std::abs(wagon.speed()) < 0.1);
}

TEST_CASE("brakes: a released wagon rolls on") {
	vehicle wagon{definition::load("wagon"), 60.0};
	wagon.hold_brake_pipe(5.0);
	wagon.run(30.0);
	CHECK(wagon.speed() > 50.0);
	CHECK(wagon.speed() < 60.0); // running resistance slows it down a little
}

// the example in README.md
TEST_CASE("brakes: service braking stops a heavier wagon too") {
	// the default wagon, with the keys the test is about changed
	test::vehicle wagon{test::definition::load("wagon").set("Param", "M", 60000), 40.0 /* km/h */};
	wagon.hold_brake_pipe(3.5); // bar, as a coupled locomotive would hold it
	wagon.run(40.0 /* s */, test::track{0.15 /* friction */, 0.0 /* gradient */});
	CHECK(std::abs(wagon.speed()) < 0.1);
}