logan_2023/day5/d5p2.gd

#!/usr/bin/godot -s
#d5p2.gd
#extends Node3D
extends SceneTree

enum Maps {
	seeds,
	seedToSoil,
	soilToFertilizer,
	fertilizerToWater,
	waterToLight,
	lightToTemp,
	temptoHumidity,
	humidityToLocation
}

enum Column {
	destination,
	origin,
	range
}

func file_load(path):
	var file = FileAccess.open(path, FileAccess.READ)
	var content = file.get_as_text()
	return content

func strarr_to_arr(input):
	var new = []
	for i in range(0, input.size()):
		new.append(input[i])

	return new

func input_clean(input):
	for i in range(0, input.size()):
		var regex = RegEx.new()
		regex.compile("^(.*-to-.* map:\n)|^(seeds: )")
		input[i] = regex.sub(input[i], "", true)

# bad variable names :(
func input_format(input):
	var new = []
	for i in range(0, input.size()):
		var temp = strarr_to_arr(input[i].split("\n"))
		var temp4 = []

		for j in range(0, temp.size()):
			var temp2 = strarr_to_arr(temp[j].split(" "))
			var temp3 = []

			for k in range(0, temp2.size()):
				temp3.append(int(temp2[k]))

			temp4.append(temp3)

		new.append(temp4)

	return new

func is_mapped(map, input):
	for v in map:
		#if input in range(v[Column.origin], v[Column.origin]+v[Column.range]): # Fucking Godot runs out of memory when I use this
		if input >= v[Column.origin] && input <= v[Column.origin]+v[Column.range]:
			return v[Column.destination] + (input-v[Column.origin])

	return input

func is_revmapped(map, input):
	for v in map:
		if input >= v[Column.destination] && input <= v[Column.destination]+v[Column.range]:
			return v[Column.origin] + (input-v[Column.destination])

	return input

func map_input(data, seed):
	var temp = seed
	for map in range(Maps.seedToSoil, Maps.humidityToLocation+1):
		temp = is_mapped(data[map], temp)
		#print("Output: %s %s" % [map, temp])

	return temp

func revmap_input(data, location):
	var temp = location
	for map in range(Maps.humidityToLocation, Maps.seedToSoil-1, -1):
		temp = is_revmapped(data[map], temp)
		#print("Output: %s %s" % [map, temp])

	return temp

func _init():
	print("Hello!")

	var temp = file_load(OS.get_cmdline_user_args()[0]).split("\n\n")
	input_clean(temp)

	var input = input_format(temp)
	#print(input[Maps.seeds])
	#print(input[Maps.seedToSoil])

	temp.clear()

	var seeds = input[0][0]
	print(seeds)


	var testlocation = 0
	while true:
		if testlocation % 10000 == 0:
			print(testlocation)
		for i in range(0, seeds.size() / 2):
			var seed = revmap_input(input, testlocation)
			if seed >= seeds[i*2] && seed <= seeds[i*2]+seeds[i*2+1]:
				print("Answer 2: %s" % testlocation)
				print("Value is between seed %s and %s" % [seeds[i*2], seeds[i*2]+seeds[i*2+1]])
				quit()
				return

		testlocation+=1