// SPDX-License-Identifier: GPL-2.0
#include "testplan.h"
#include "core/deco.h"
#include "core/dive.h"
#include "core/event.h"
#include "core/errorhelper.h"
#include "core/planner.h"
#include "core/qthelper.h"
#include "core/subsurfacestartup.h"
#include "core/units.h"

#define DEBUG 1

// testing the dive plan algorithm
static struct dive dive;
static struct deco_state test_deco_state;
void setupPrefs()
{
	prefs = default_prefs;
	prefs.ascrate50 = feet_to_mm(30) / 60;
	prefs.ascrate75 = prefs.ascrate50;
	prefs.ascratestops = prefs.ascrate50;
	prefs.ascratelast6m = feet_to_mm(10) / 60;
	prefs.last_stop = true;
}

void setupPrefsVpmb()
{
	prefs = default_prefs;
	prefs.ascrate50 = 10000 / 60;
	prefs.ascrate75 = prefs.ascrate50;
	prefs.ascratestops = prefs.ascrate50;
	prefs.ascratelast6m = prefs.ascrate50;
	prefs.descrate = 99000 / 60;
	prefs.last_stop = false;
	prefs.planner_deco_mode = VPMB;
	prefs.vpmb_conservatism = 0;
}

diveplan setupPlan()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.gfhigh = 100;
	dp.gflow = 100;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{150}, {450}};
	struct gasmix ean36 = {{360}, {0}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean36;
	cyl2->gasmix = oxygen;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(79, 260) * 60 / M_OR_FT(23, 75);
	plan_add_segment(dp, 0, dive.gas_mod(ean36, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(79, 260), 0, 0, 1, OC);
	plan_add_segment(dp, 30 * 60 - droptime, M_OR_FT(79, 260), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb45m30mTx()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.gfhigh = 100;
	dp.gflow = 100;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{210}, {350}};
	struct gasmix ean50 = {{500}, {0}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 24000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean50;
	cyl2->gasmix = oxygen;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(45, 150) * 60 / M_OR_FT(23, 75);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(45, 150), 0, 0, 1, OC);
	plan_add_segment(dp, 30 * 60 - droptime, M_OR_FT(45, 150), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb60m10mTx()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.gfhigh = 100;
	dp.gflow = 100;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{180}, {450}};
	struct gasmix tx50_15 = {{500}, {150}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 24000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = tx50_15;
	cyl2->gasmix = oxygen;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(60, 200) * 60 / M_OR_FT(23, 75);
	plan_add_segment(dp, 0, dive.gas_mod(tx50_15, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	plan_add_segment(dp, 10 * 60 - droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb60m30minAir()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{210}, {0}};
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 100000;
	cyl0->type.workingpressure.mbar = 232000;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(60, 200) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	plan_add_segment(dp, 30 * 60 - droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb60m30minEan50()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{210}, {0}};
	struct gasmix ean50 = {{500}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean50;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(60, 200) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	plan_add_segment(dp, 30 * 60 - droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb60m30minTx()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{180}, {450}};
	struct gasmix ean50 = {{500}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean50;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(60, 200) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	plan_add_segment(dp, 30 * 60 - droptime, M_OR_FT(60, 200), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmbMultiLevelAir()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{210}, {0}};
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 200000;
	cyl0->type.workingpressure.mbar = 232000;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(20, 66) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, droptime, M_OR_FT(20, 66), 0, 0, 1, OC);
	plan_add_segment(dp, 10 * 60 - droptime, M_OR_FT(20, 66), 0, 0, 1, OC);
	plan_add_segment(dp, 1 * 60, M_OR_FT(60, 200), 0, 0, 1, OC);
	plan_add_segment(dp, 29 * 60, M_OR_FT(60, 200), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb100m60min()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{180}, {450}};
	struct gasmix ean50 = {{500}, {0}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 200000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean50;
	cyl2->gasmix = oxygen;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(100, 330) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	plan_add_segment(dp, 60 * 60 - droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb100m10min()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{180}, {450}};
	struct gasmix ean50 = {{500}, {0}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 60000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = ean50;
	cyl2->gasmix = oxygen;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(100, 330) * 60 / M_OR_FT(99, 330);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	plan_add_segment(dp, 10 * 60 - droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb30m20min()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{210}, {0}};
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(30, 100) * 60 / M_OR_FT(18, 60);
	plan_add_segment(dp, droptime, M_OR_FT(30, 100), 0, 0, 1, OC);
	plan_add_segment(dp, 20 * 60 - droptime, M_OR_FT(30, 100), 0, 0, 1, OC);
	return dp;
}

diveplan setupPlanVpmb100mTo70m30min()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix bottomgas = {{120}, {650}};
	struct gasmix tx21_35 = {{210}, {350}};
	struct gasmix ean50 = {{500}, {0}};
	struct gasmix oxygen = {{1000}, {0}};
	pressure_t po2 = {1600};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl3 = dive.get_or_create_cylinder(3);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cyl0->gasmix = bottomgas;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = tx21_35;
	cyl2->gasmix = ean50;
	cyl3->gasmix = oxygen;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	int droptime = M_OR_FT(100, 330) * 60 / M_OR_FT(18, 60);
	plan_add_segment(dp, 0, dive.gas_mod(tx21_35, po2, M_OR_FT(3, 10)).mm, 1, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(ean50, po2, M_OR_FT(3, 10)).mm, 2, 0, 1, OC);
	plan_add_segment(dp, 0, dive.gas_mod(oxygen, po2, M_OR_FT(3, 10)).mm, 3, 0, 1, OC);
	plan_add_segment(dp, droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	plan_add_segment(dp, 20 * 60 - droptime, M_OR_FT(100, 330), 0, 0, 1, OC);
	plan_add_segment(dp, 3 * 60, M_OR_FT(70, 230), 0, 0, 1, OC);
	plan_add_segment(dp, (30 - 20 - 3) * 60, M_OR_FT(70, 230), 0, 0, 1, OC);
	return dp;
}

/* This tests handling different gases in the manually entered part of the dive */

diveplan setupPlanSeveralGases()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	struct gasmix ean36 = {{360}, {0}};
	struct gasmix tx11_50 = {{110}, {500}};

	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cyl0->gasmix = ean36;
	cyl0->type.size.mliter = 36000;
	cyl0->type.workingpressure.mbar = 232000;
	cyl1->gasmix = tx11_50;
	dive.surface_pressure.mbar = 1013;
	reset_cylinders(&dive, true);

	plan_add_segment(dp, 120, 40000, 0, 0, true, OC);
	plan_add_segment(dp, 18 * 60, 40000, 0, 0, true, OC);
	plan_add_segment(dp, 10 * 60, 10000, 1, 0, true, OC);
	plan_add_segment(dp, 5 * 60, 10000, 0, 0, true, OC);
	return dp;
}

diveplan setupPlanCcr()
{
	diveplan dp;
	dp.salinity = 10300;
	dp.surface_pressure = 1013;
	dp.gflow = 50;
	dp.gfhigh = 70;
	dp.bottomsac = prefs.bottomsac;
	dp.decosac = prefs.decosac;

	pressure_t po2 = {1600};
	struct gasmix diluent = {{200}, {210}};
	struct gasmix ean53 = {{530}, {0}};
	struct gasmix tx19_33 = {{190}, {330}};
	// Note: we add the highest-index cylinder first, because
	// pointers to cylinders are not stable when reallocating.
	// For testing OK - don't do this in actual code!
	cylinder_t *cyl2 = dive.get_or_create_cylinder(2);
	cylinder_t *cyl0 = dive.get_or_create_cylinder(0);
	cylinder_t *cyl1 = dive.get_or_create_cylinder(1);
	cyl0->gasmix = diluent;
	cyl0->depth = dive.gas_mod(diluent, po2, M_OR_FT(3, 10));
	cyl0->type.size.mliter = 3000;
	cyl0->type.workingpressure.mbar = 200000;
	cyl0->cylinder_use = DILUENT;
	cyl1->gasmix = ean53;
	cyl1->depth = dive.gas_mod(ean53, po2, M_OR_FT(3, 10));
	cyl2->gasmix = tx19_33;
	cyl2->depth = dive.gas_mod(tx19_33, po2, M_OR_FT(3, 10));
	reset_cylinders(&dive, true);

	plan_add_segment(dp, 0, cyl1->depth.mm, 1, 0, false, OC);
	plan_add_segment(dp, 0, cyl2->depth.mm, 2, 0, false, OC);
	plan_add_segment(dp, 20 * 60, M_OR_FT(60, 197), 0, 1300, true, CCR);

	return dp;
}

/* We compare the calculated runtimes against two values:
 * - Known runtime calculated by Subsurface previously (to detect if anything has changed)
 * - Benchmark runtime (we should be close, but not always exactly the same)
 */
bool compareDecoTime(int actualRunTimeSeconds, int benchmarkRunTimeSeconds, int knownSsrfRunTimeSeconds)
{
	bool result;

	// If the calculated run time equals the expected run time, do a simple comparison
	if (actualRunTimeSeconds == benchmarkRunTimeSeconds) {
		result = true;
	} else {
		/* We want the difference between the expected and calculated total run time to be not more than
		* 1% of total run time + 1 minute */
		int permilDifferenceAllowed = 1 * 10;
		int absoluteDifferenceAllowedSeconds = 60;
		int totalDifferenceAllowed = lrint(0.001 * permilDifferenceAllowed * benchmarkRunTimeSeconds + absoluteDifferenceAllowedSeconds);
		int totalDifference = abs(actualRunTimeSeconds - benchmarkRunTimeSeconds);

		report_info("Calculated run time = %d seconds", actualRunTimeSeconds);
		report_info("Expected run time = %d seconds", benchmarkRunTimeSeconds);
		report_info("Allowed time difference is %g percent plus %d seconds = %d seconds",
		       permilDifferenceAllowed * 0.1, absoluteDifferenceAllowedSeconds, totalDifferenceAllowed);
		report_info("total difference = %d seconds", totalDifference);

		result = (totalDifference <= totalDifferenceAllowed);
	}
	if ((knownSsrfRunTimeSeconds > 0) && (actualRunTimeSeconds != knownSsrfRunTimeSeconds)) {
		report_error("Calculated run time does not match known Subsurface runtime");
		report_error("Calculated runtime: %d", actualRunTimeSeconds);
		report_error("Known Subsurface runtime: %d", knownSsrfRunTimeSeconds);
	}
	return result;
}

void TestPlan::testMetric()
{
	deco_state_cache cache;

	setupPrefs();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;
	prefs.planner_deco_mode = BUEHLMANN;

	auto testPlan = setupPlan();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 148l);
	QVERIFY(dive.dcs[0].events.size() >= 2);
	// check first gas change to EAN36 at 33m
	struct event *ev = &dive.dcs[0].events[0];
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 36);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 33000);
	// check second gas change to Oxygen at 6m
	ev = &dive.dcs[0].events[1];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 2);
	QCOMPARE(ev->value, 100);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 6000);
	// check expected run time of 109 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 109u * 60u, 109u * 60u));
}

void TestPlan::testImperial()
{
	deco_state_cache cache;

	setupPrefs();
	prefs.unit_system = IMPERIAL;
	prefs.units.length = units::FEET;
	prefs.planner_deco_mode = BUEHLMANN;

	auto testPlan = setupPlan();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 155l);
	QVERIFY(dive.dcs[0].events.size() >= 2);
	// check first gas change to EAN36 at 33m
	struct event *ev = &dive.dcs[0].events[0];
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 36);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 33528);
	// check second gas change to Oxygen at 6m
	ev = &dive.dcs[0].events[1];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 2);
	QCOMPARE(ev->value, 100);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 6096);
	// check expected run time of 111 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 111u * 60u - 2u, 111u * 60u - 2u));
}

void TestPlan::testVpmbMetric45m30minTx()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb45m30mTx();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 108l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check benchmark run time of 141 minutes, and known Subsurface runtime of 139 minutes
	//QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 141u * 60u + 20u, 139u * 60u + 20u));
}

void TestPlan::testVpmbMetric60m10minTx()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb60m10mTx();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 162l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check benchmark run time of 141 minutes, and known Subsurface runtime of 139 minutes
	//QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 141u * 60u + 20u, 139u * 60u + 20u));
}

void TestPlan::testVpmbMetric60m30minAir()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb60m30minAir();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 180l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check benchmark run time of 141 minutes, and known Subsurface runtime of 139 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 141u * 60u + 20u, 139u * 60u + 20u));
}

void TestPlan::testVpmbMetric60m30minEan50()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb60m30minEan50();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 155l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	QVERIFY(dive.dcs[0].events.size() >= 1);
	// check first gas change to EAN50 at 21m
	struct event *ev = &dive.dcs[0].events[0];
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 50);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 21000);
	// check benchmark run time of 95 minutes, and known Subsurface runtime of 96 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 95u * 60u + 20u, 96u * 60u + 20u));
}

void TestPlan::testVpmbMetric60m30minTx()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb60m30minTx();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 159l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check first gas change to EAN50 at 21m
	QVERIFY(dive.dcs[0].events.size() >= 1);
	struct event *ev = &dive.dcs[0].events[0];
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 50);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 21000);
	// check benchmark run time of 89 minutes, and known Subsurface runtime of 89 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 89u * 60u + 20u, 89u * 60u + 20u));
}

void TestPlan::testVpmbMetric100m60min()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb100m60min();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 157l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	QVERIFY(dive.dcs[0].events.size() >= 2);
	// check first gas change to EAN50 at 21m
	struct event *ev = &dive.dcs[0].events[0];
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 50);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 21000);
	// check second gas change to Oxygen at 6m
	ev = &dive.dcs[0].events[1];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 2);
	QCOMPARE(ev->value, 100);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 6000);
	// check benchmark run time of 311 minutes, and known Subsurface runtime of 314 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 311u * 60u + 20u, 315u * 60u + 20u));
}

void TestPlan::testMultipleGases()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanSeveralGases();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	gasmix gas = dive.get_gasmix_at_time(dive.dcs[0], {20 * 60 + 1});
	QCOMPARE(get_o2(gas), 110);
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 2480u, 2480u));
}

void TestPlan::testVpmbMetricMultiLevelAir()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmbMultiLevelAir();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 101l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check benchmark run time of 167 minutes, and known Subsurface runtime of 169 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 167u * 60u + 20u, 169u * 60u + 20u));
}

void TestPlan::testVpmbMetric100m10min()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb100m10min();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 175l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	QVERIFY(dive.dcs[0].events.size() >= 2);
	// check first gas change to EAN50 at 21m
	struct event *ev = &dive.dcs[0].events[0];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->value, 50);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 21000);
	// check second gas change to Oxygen at 6m
	ev = &dive.dcs[0].events[1];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 2);
	QCOMPARE(ev->value, 100);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 6000);
	// check benchmark run time of 58 minutes, and known Subsurface runtime of 57 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 58u * 60u + 20u, 57u * 60u + 20u));
}

/* This tests that a previously calculated plan isn't affecting the calculations of the next plan.
 * It is NOT a 'repetitive dive' test (i.e. with a surface interval and considering tissue
 * saturation from the previous dive).
 */
void TestPlan::testVpmbMetricRepeat()
{
	deco_state_cache cache;

	setupPrefsVpmb();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;

	auto testPlan = setupPlanVpmb30m20min();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	auto dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 61l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	// check benchmark run time of 27 minutes, and known Subsurface runtime of 28 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 27u * 60u + 20u, 27u * 60u + 20u));

	int firstDiveRunTimeSeconds = dive.dcs[0].duration.seconds;

	testPlan = setupPlanVpmb100mTo70m30min();
	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 80l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);
	QVERIFY(dive.dcs[0].events.size() >= 3);
	// check first gas change to 21/35 at 66m
	struct event *ev = &dive.dcs[0].events[0];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 1);
	QCOMPARE(ev->gas.mix.o2.permille, 210);
	QCOMPARE(ev->gas.mix.he.permille, 350);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 66000);
	// check second gas change to EAN50 at 21m
	ev = &dive.dcs[0].events[1];
	QCOMPARE(ev->gas.index, 2);
	QCOMPARE(ev->value, 50);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 21000);
	// check third gas change to Oxygen at 6m
	ev = &dive.dcs[0].events[2];
	QVERIFY(ev != NULL);
	QCOMPARE(ev->gas.index, 3);
	QCOMPARE(ev->value, 100);
	QCOMPARE(get_depth_at_time(&dive.dcs[0], ev->time.seconds), 6000);
	// we don't have a benchmark, known Subsurface runtime is 126 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 127u * 60u + 20u, 127u * 60u + 20u));

	testPlan = setupPlanVpmb30m20min();
	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, 1, 0);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check minimum gas result
	dp = std::find_if(testPlan.dp.begin(), testPlan.dp.end(), [](auto &dp) { return dp.minimum_gas.mbar != 0; });
	QCOMPARE(lrint(dp == testPlan.dp.end() ? 0.0 : dp->minimum_gas.mbar / 1000.0), 61l);
	// print first ceiling
	printf("First ceiling %.1f m\n", dive.mbar_to_depth(test_deco_state.first_ceiling_pressure.mbar) * 0.001);

	// check runtime is exactly the same as the first time
	int finalDiveRunTimeSeconds = dive.dcs[0].duration.seconds;
	QCOMPARE(finalDiveRunTimeSeconds, firstDiveRunTimeSeconds);
}


// Test that the correct gases are selected during a CCR dive with bailout ascent
// Includes a regression test for https://groups.google.com/g/subsurface-divelog/c/8N3cTz2Zv5E

void TestPlan::testCcrBailoutGasSelection()
{
	deco_state_cache cache;

	setupPrefs();
	prefs.unit_system = METRIC;
	prefs.units.length = units::METERS;
	prefs.planner_deco_mode = BUEHLMANN;
	dive.dcs[0].divemode = CCR;
	prefs.dobailout = true;

	auto testPlan = setupPlanCcr();

	plan(&test_deco_state, testPlan, &dive, 0, 60, cache, true, false);

#if DEBUG
	dive.notes.clear();
	save_dive(stdout, dive, false);
#endif

	// check diluent used
	cylinder_t *cylinder = dive.get_cylinder(get_cylinderid_at_time(&dive, &dive.dcs[0], { 20 * 60 - 1 }));
	QCOMPARE(cylinder->cylinder_use, DILUENT);
	QCOMPARE(get_o2(cylinder->gasmix), 200);

	// check deep bailout used
	cylinder = dive.get_cylinder(get_cylinderid_at_time(&dive, &dive.dcs[0], { 20 * 60 + 1 }));
	QCOMPARE(cylinder->cylinder_use, OC_GAS);
	QCOMPARE(get_o2(cylinder->gasmix), 190);

	// check shallow bailout used
	cylinder = dive.get_cylinder(get_cylinderid_at_time(&dive, &dive.dcs[0], { 30 * 60 }));
	QCOMPARE(cylinder->cylinder_use, OC_GAS);
	QCOMPARE(get_o2(cylinder->gasmix), 530);

	// check expected run time of 51 minutes
	QVERIFY(compareDecoTime(dive.dcs[0].duration.seconds, 51 * 60, 51 * 60));

}

QTEST_GUILESS_MAIN(TestPlan)