#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 = 3;
Run runInput=FILE2;

struct numb{
	vi2d pos;
	vi2d endPos;
	int number;
	bool operator<(numb&rhs){
		return number<rhs.number;
	}
	bool operator==(numb&rhs){
		return number==rhs.number;
	}
};

std::vector<numb>parts;
std::vector<std::vector<char>>board;
std::vector<vi2d>gears;

void doStuff2(){
	while(true){ //lines is accessible as a global.
		long sum=0;
		vi2d pos={};
		for(std::string&line:lines){
			std::string numb="";
			bool isAdjacent=false;
			std::vector<char>row;
			vi2d initialNumberPos={};
			for(int i=0;i<line.size();i++){
				if(line[i]=='*')gears.push_back(pos);
				if(line[i]>='0'&&line[i]<='9'){
					if(numb.length()==0){
						initialNumberPos=pos;
					}
					numb+=line[i];
				}else
				if(numb.length()>0){
					//A number has been parsed.
					parts.push_back({initialNumberPos,pos+vi2d{-1,0},atoi(numb.c_str())});
					numb="";
				}
				row.push_back(line[i]);
				pos.x++;
			}
			if(numb.length()>0){
				//A number has been parsed.
				parts.push_back({initialNumberPos,pos+vi2d{-1,0},atoi(numb.c_str())});
			}
			board.push_back(row);
			pos.y++;
			pos.x=0;
			std::cout<<line<<std::endl;
		}
		
		for(vi2d&pos:gears){
			auto validatePos=[&](vi2d checkPos)->numb*{
				if(pos.y+checkPos.y>=0&&pos.y+checkPos.y<=board.size()-1&&
					pos.x+checkPos.x>=0&&pos.x+checkPos.x<=board[pos.y+checkPos.y].size()-1){
					char c=board[pos.y+checkPos.y][pos.x+checkPos.x];
					if(c>='0'&&c<='9'){
						for(numb&number:parts){
							if(number.pos.y==pos.y+checkPos.y&&number.pos.x<=pos.x+checkPos.x&&number.endPos.x>=pos.x+checkPos.x){
								return &number;
							}
						}
						throw; //Something went terribly wrong.
					}
				}
				return nullptr;
			};
			numb*val;
			std::vector<numb>numbers;
			if((val=validatePos({1,0}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({0,1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({-1,0}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({0,-1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({1,1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({1,-1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({-1,1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if((val=validatePos({-1,-1}))!=nullptr){numbers.push_back(*val);val=nullptr;}
			if(numbers.size()>=2){
				std::sort(numbers.begin(),numbers.end());
				auto last=std::unique(numbers.begin(),numbers.end());
				numbers.erase(last,numbers.end());
				if(numbers.size()==2){
					sum+=numbers[0].number*numbers[1].number;
					std::cout<<numbers[0].number<<"*"<<numbers[1].number<<std::endl;
				}
			}
		}
		std::cout<<sum<<std::endl;
		wait(0); //Wait for 0ms and render the screen (calls draw())
		break;
	}
}
/*
void doStuff(){
	while(true){ //lines is accessible as a global.
		int sum=0;
		vi2d pos={};
		for(std::string&line:lines){
			std::string numb="";
			bool isAdjacent=false;
			std::vector<char>row;
			vi2d initialNumberPos={};
			for(int i=0;i<line.size();i++){
				if(line[i]>='0'&&line[i]<='9'){
					if(numb.length()==0){
						initialNumberPos=pos;
					}
					numb+=line[i];
				}else
				if(numb.length()>0){
					//A number has been parsed.
					parts.push_back({initialNumberPos,atoi(numb.c_str())});
					numb="";
				}
				row.push_back(line[i]);
				pos.x++;
			}
			if(numb.length()>0){
				//A number has been parsed.
				parts.push_back({initialNumberPos,atoi(numb.c_str())});
			}
			board.push_back(row);
			pos.y++;
			pos.x=0;
			std::cout<<line<<std::endl;
		}

		for(numb&number:parts){
			int myNumb=number.number;
			vi2d startingPos=number.pos;
			while(myNumb>0){
				auto validatePos=[&](vi2d checkPos){
					return startingPos.y+checkPos.y>0&&startingPos.y+checkPos.y<board.size()-1&&
						startingPos.x+checkPos.x>0&&startingPos.x+checkPos.x<board[startingPos.y+checkPos.y].size()-1;
				};
				auto testChar=[&](char test){
					if(test!='.'&&!(test>='0'&&test<='9')){
						sum+=number.number;
						std::cout<<number.number<<" was adjacent to a symbol"<<std::endl;
						return true;
					}
					return false;
				};
				vi2d checkPos={1,0};
				if(validatePos({1,0})){if(testChar(board[startingPos.y][startingPos.x+1]))break;}
				if(validatePos({0,1})){if(testChar(board[startingPos.y+1][startingPos.x]))break;}
				if(validatePos({-1,0})){if(testChar(board[startingPos.y][startingPos.x-1]))break;}
				if(validatePos({0,-1})){if(testChar(board[startingPos.y-1][startingPos.x]))break;}
				if(validatePos({1,1})){if(testChar(board[startingPos.y+1][startingPos.x+1]))break;}
				if(validatePos({1,-1})){if(testChar(board[startingPos.y-1][startingPos.x+1]))break;}
				if(validatePos({-1,1})){if(testChar(board[startingPos.y+1][startingPos.x-1]))break;}
				if(validatePos({-1,-1})){if(testChar(board[startingPos.y-1][startingPos.x-1]))break;}
				myNumb/=10;
				startingPos.x++;
			}
		}
		std::cout<<sum<<std::endl;
		wait(0); //Wait for 0ms and render the screen (calls draw())
		break;
	}
}
*/
void draw(){ //Only use Sprites! If using decals, you must reference global variables!
	Clear(BLACK);
	int count=0;
	vi2d pos;
	for(std::vector<char>&row:board){
		for(char c:row){
			DrawString(pos,std::string(1,c));
			pos.x+=8;
		}
		pos.y+=8;
		pos.x=0;
	}
	for(numb&number:parts){
		std::string numbStr="";
		int numb=number.number;
		while(numb>0){
			numbStr=std::to_string(numb%10)+numbStr;
			if(numb<10){
				DrawString(number.endPos*8,std::string(1,numbStr[numbStr.size()-1]),GREEN);
			}
			numb/=10;
		}
		DrawString(number.pos*8,numbStr,RED);
	}
	for(vi2d&pos:gears){
		DrawString(pos*8,std::string(1,'*'),BLUE);
	}
}

#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(1280, 960, 2, 2))
		game.Start();
	return 0;
}
#pragma endregion