main.cpp

#include "include/configuration.hpp"

int main() {
	fmt::print(fmt::fg(fmt::color::orange_red) | fmt::emphasis::bold,
			   "Welcome to Flappy Bird AI!\n");

	librapid::setNumThreads(1);

#if defined(LIBRAPID_HAS_OPENCL)
	fmt::print(fmt::fg(fmt::color::lime_green) | fmt::emphasis::bold, "OpenCL is enabled.\n");
	librapid::configureOpenCL(true);
#else
	fmt::print(fmt::fg(fmt::color::red) | fmt::emphasis::bold, "OpenCL is disabled.\n");
#endif

#if defined(LIBRAPID_HAS_CUDA)
	fmt::print(fmt::fg(fmt::color::lime_green) | fmt::emphasis::bold, "CUDA is enabled.\n");
#else
	fmt::print(fmt::fg(fmt::color::red) | fmt::emphasis::bold, "CUDA is disabled.\n");
#endif

	// Configure the window
	surge::Window mainWindow(librapid::Vec2i(1000, 600), "Flappy Bird AI");

	// Configure the walls
	std::vector<Wall> walls(NUM_WALLS);
	resetWalls(walls);

	// The bird population
	generationStartTime = librapid::now();
	std::vector<Bird> birds(NUM_BIRDS);

	// Configure each bird
	for (auto &bird : birds) {
		bird		 = createBird();
		bird.brain() = createBirdBrain();
	}

	// Information about the generations and birds
	std::vector<double> wallDistances;
	std::vector<double> generationBirdsAlive;
	std::vector<double> generationBirdsAliveDistance;

	// Style settings
	surge::Font textFont("Arial", 20);
	surge::Font mathFont("Cambria", 20);

	ImGui::SetFont(textFont);
	ImGui::GetStyle().WindowPadding = {16, 12};

	// The main loop
	while (!mainWindow.shouldClose()) {
		// Begin a drawing and clear the screen
		mainWindow.beginDrawing();
		mainWindow.clear(surge::Color::veryDarkGray);

		// Update the birds and walls
		updateWalls(walls);
		wallDistance += 0.1; // Arbitrary. So long as it's increasing, it's fine.
		int64_t alive = updateBirds(birds, walls);

		// Occasionally log some information about the current generation
		if (mainWindow.frameCount() % 10 == 0) {
			fmt::print(fmt::fg(fmt::color::purple) | fmt::emphasis::bold,
					   "Alive: {:>7} / {:>7}\r",
					   alive,
					   NUM_BIRDS);
			generationBirdsAlive.emplace_back(((double)alive / (double)NUM_BIRDS) * 100.0);
			generationBirdsAliveDistance.emplace_back(wallDistance);
		}

		if (alive == 0) {
			// All birds are dead, so start a new generation
			++generationNumber;
			double generationTime = librapid::now() - generationStartTime;

			fmt::print(fmt::fg(fmt::color::orange_red) | fmt::emphasis::bold,
					   "\n\nGeneration {} lasted {}.\n",
					   generationNumber,
					   librapid::formatTime(generationTime));

			wallDistances.emplace_back(wallDistance);

			// Reset the walls before the birds, since they may collide with "ghost" walls
			// and cause some strange bugs
			resetWalls(walls);

			// Create the next generation of mutated bird brains
			std::vector<std::pair<Bird::BirdBrain, double>> birdBrains;

			birdBrains.reserve(birds.size());
			for (auto &bird : birds) { birdBrains.emplace_back(bird.brain(), bird.fitness()); }

			std::vector<Bird::BirdBrain> nextGeneration = newGeneration(birdBrains);

			for (int64_t i = 0; i < NUM_BIRDS; ++i) {
				birds[i]		 = createBird();
				birds[i].brain() = nextGeneration[i];
			}

			generationBirdsAlive.clear();
			generationBirdsAliveDistance.clear();

			wallDistance		= 0;
			generationStartTime = librapid::now();
		}

		mainWindow.drawFPS(librapid::Vec2i(20, 20));
		mainWindow.drawFrameTime(librapid::Vec2i(20, 40));
		mainWindow.drawTime(librapid::Vec2i(20, 60));

		if (ImGui::Begin("Statistics")) {
			ImGui::Text("%s", fmt::format("Generation: {}", generationNumber).c_str());
			ImGui::Text("%s", fmt::format("Alive: {}", alive).c_str());
			ImGui::Text(
			  "%s",
			  fmt::format("Time: {}", librapid::formatTime(librapid::now() - generationStartTime))
				.c_str());

			ImGui::Separator();

			ImGui::SliderFloat("Learning Rate", &mutationRate, 0.0f, 0.2f);

			ImGui::Separator();

			ImGui::PushFont(mathFont);
			if (ImPlot::BeginSubplots("", 2, 1, ImVec2(-1, -1))) {
				ImPlot::SetNextAxesLimits(0, wallDistance, 0, 100, ImPlotCond_Always);
				if (ImPlot::BeginPlot("Birds Alive", ImVec2(0, 0))) {
					ImPlot::SetupAxis(ImAxis_X1, "Time/s");
					ImPlot::SetupAxis(ImAxis_Y1, "Alive %");

					ImPlot::PlotLine(
					  "Birds Alive", generationBirdsAliveDistance, generationBirdsAlive);
					ImPlot::EndPlot();
				}

				if (ImPlot::BeginPlot("Survival Distance", ImVec2(0, 0))) {
					ImPlot::SetupAxis(ImAxis_X1, "Generation", ImPlotAxisFlags_AutoFit);
					ImPlot::SetupAxis(ImAxis_Y1, "Distance", ImPlotAxisFlags_AutoFit);

					ImPlot::PlotLine("Survival Distance", wallDistances);
					ImPlot::PlotInfLines(
					  "Current Distance", &wallDistance, 1, ImPlotInfLinesFlags_Horizontal);
					ImPlot::EndPlot();
				}

				ImPlot::EndSubplots();
			}
			ImGui::PopFont();
			ImGui::End();
		}
		mainWindow.endDrawing();
	}

	return 0;
}