#pragma region Hidden Setup Stuff
#define OLC_PGE_APPLICATION
#include "olcPixelGameEngine.h"

using namespace olc;

enum Run{
	FILE1,
	FILE2
};

// Override base class with your custom functionality
class AoC2023 : public olc::PixelGameEngine
{
std::vector<std::string>lines;
bool waitForRender=false;

void wait(int pauseMs=0){
	waitForRender=true;
	while(waitForRender);
	std::this_thread::sleep_for(std::chrono::milliseconds(pauseMs));
}

#pragma endregion

const int DAY = 6;
Run runInput=FILE2;

std::vector<long long>time;
std::vector<long long>distance;

void doStuff2(){
	while(true){ //lines is accessible as a global.
		bool firstLine=true;
		for(std::string&line:lines){
			if(firstLine){//Time line.
				std::string data=line.substr(5);
				std::string numb="";
				while(data.length()>0){
					if(data[0]>='0'&&data[0]<='9'){
						numb+=data[0];
					}
					data=data.substr(1);
				}
				if(numb.length()>0){
					time.push_back(std::stoll(numb));
					numb="";
				}
				firstLine=false;
			}else{//Distance line.
				std::string data=line.substr(9);
				std::string numb="";
				while(data.length()>0){
					if(data[0]>='0'&&data[0]<='9'){
						numb+=data[0];
					}
					data=data.substr(1);
				}
				if(numb.length()>0){
					distance.push_back(std::stoll(numb));
					numb="";
				}
			}
		}
		long sum=0;
		for(int i=0;i<time.size();i++){
			long long targetTime=time[i];
			long long targetDist=distance[i];
			long long wins=0;
			for(long long holdTime=0;holdTime<=targetTime;holdTime++){
				long long timeRemaining=targetTime-holdTime;
				long long speed=holdTime;
				long long distanceTraveled=speed*timeRemaining;
				if(distanceTraveled>targetDist){
					wins++;
				}
			}
			if(sum==0){
				sum=wins;
			}else{
				sum*=wins;
			}
		}
		std::cout<<sum<<std::endl;

		break;
		//wait(0); //Wait for 0ms and render the screen (calls draw())
	}
}

void doStuff(){
	while(true){ //lines is accessible as a global.
		bool firstLine=true;
		for(std::string&line:lines){
			if(firstLine){//Time line.
				std::string data=line.substr(5);
				std::string numb="";
				while(data.length()>0){
					if(data[0]>='0'&&data[0]<='9'){
						numb+=data[0];
					}else
					if(numb.length()>0){
						time.push_back(std::stoi(numb));
						numb="";
					}
					data=data.substr(1);
				}
				if(numb.length()>0){
					time.push_back(std::stoi(numb));
					numb="";
				}
				firstLine=false;
			}else{//Distance line.
				std::string data=line.substr(9);
				std::string numb="";
				while(data.length()>0){
					if(data[0]>='0'&&data[0]<='9'){
						numb+=data[0];
					}else
					if(numb.length()>0){
						distance.push_back(std::stoi(numb));
						numb="";
					}
					data=data.substr(1);
				}
				if(numb.length()>0){
					distance.push_back(std::stoi(numb));
					numb="";
				}
			}
		}
		long sum=0;
		for(int i=0;i<time.size();i++){
			int targetTime=time[i];
			int targetDist=distance[i];
			int wins=0;
			for(int holdTime=0;holdTime<=targetTime;holdTime++){
				int timeRemaining=targetTime-holdTime;
				int speed=holdTime;
				int distanceTraveled=speed*timeRemaining;
				if(distanceTraveled>targetDist){
					wins++;
				}
			}
			if(sum==0){
				sum=wins;
			}else{
				sum*=wins;
			}
		}
		std::cout<<sum<<std::endl;

		break;
		//wait(0); //Wait for 0ms and render the screen (calls draw())
	}
}

void draw(){ //Only use Sprites! If using decals, you must reference global variables!
	Clear(BLACK);
	int count=0;
	for(std::string&line:lines){
		DrawString({0,count*32},line,WHITE,4);
		count++;
	}
}

#pragma region Hidden Engine Stuff
public:
	AoC2023()
	{
		// Name your application
		std::string fileName="day"+std::to_string(DAY)+"_1.txt";
		if(runInput==FILE2){fileName="day"+std::to_string(DAY)+"_2.txt";}
		std::ifstream file(fileName);
		while(file.good()){
			std::string line;
			std::getline(file,line);
			lines.push_back(line);
		}

		sAppName = "Advent of Code 2023 - Day "+std::to_string(DAY);
	}

public:
	bool OnUserCreate() override
	{

		return true;
	}

	bool OnUserUpdate(float fElapsedTime) override
	{
		static std::thread aocSolver(&AoC2023::doStuff2,this);

		if(waitForRender){
			draw();
			waitForRender=false;
		}
		return true;
	}
};

int main()
{
	AoC2023 game;
	if (game.Construct(640, 480, 2,2))
		game.Start();
	return 0;
}
#pragma endregion