local component = require("component")
local MineOSCore = require("MineOSCore")
local GUI = require("GUI")
local windows = require("windows")
local image = require("image")
local colorlib = require("colorlib")
local buffer = require("doubleBuffering")

--------------------------------------------------------------------------------------------------------------------

component.gpu.setResolution(component.gpu.maxResolution())
buffer.start()

local resourcesDirectory = MineOSCore.getCurrentApplicationResourcesDirectory() 
local earthImage = image.load(resourcesDirectory .. "Earth.pic")

local glasses, geolyzer, projector
if component.isAvailable("glasses") then glasses = component.glasses end
if component.isAvailable("geolyzer") then geolyzer = component.geolyzer else GUI.error("This program requires a geolyzer to work!"); return  end
if component.isAvailable("hologram") then projector = component.hologram else GUI.error("This program requires a hologram projector to work!"); return end

local onScreenDataXOffset, onScreenDataYOffset = math.floor(buffer.screen.width / 2), buffer.screen.height
local onProjectorDataYOffset = 0
local scanResult = {horizontalRange = 0, verticalRange = 0}
local window = windows.fullScreen()

--------------------------------------------------------------------------------------------------------------------

local function getOpenGLValidColorChannels(color)
	local r, g, b = colorlib.HEXtoRGB(color)
	return r / 255, g / 255, b / 255
end

local function createCube(x, y, z, color, isVisThrObj)
	local cube = glasses.addCube3D()
	cube.set3DPos(x, y, z)
	cube.setVisibleThroughObjects(isVisThrObj)
	cube.setColor(getOpenGLValidColorChannels(color))
	cube.setAlpha(0.23)
	return cube
end

local function glassesCreateCube(x, y, z, color, text)
	local cube = createCube(x, y, z, color, true)
	cube.setVisibleThroughObjects(true)

	local floatingText = glasses.addFloatingText()
	floatingText.set3DPos(x + 0.5, y + 0.5, z + 0.5)
	floatingText.setColor(1, 1, 1)
	floatingText.setAlpha(0.6)
	floatingText.setText(text)
	floatingText.setScale(0.015)
end

local function createDick()
	local chance = math.random(1, 100) <= 20
	if chance then
		local range = 48
		local isVisThrObj = false
		local x, y, z = math.random(-range, range), math.random(-range, range), math.random(-range, range)
		createCube(x, y, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 1, y, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 2, y, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 1, y + 1, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 1, y + 2, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 1, y + 3, z, 0xFFFFFF, isVisThrObj)
		createCube(x + 1, y + 5, z, 0xFF8888, isVisThrObj)
	end
end

local function progressReport(value, text)
	local width = 40
	local x, y = math.floor(buffer.screen.width / 2 - width / 2), math.floor(buffer.screen.height / 2)
	GUI.progressBar(x, y, width, 0x00B6FF, 0xFFFFFF, 0xEEEEEE, value, true, true, text, "%"):draw()
	buffer.draw()
end

local function updateData(onScreen, onProjector, onGlasses)
	local glassesAvailable = component.isAvailable("glasses")
	local projectorAvailable = component.isAvailable("hologram")

	if onScreen then buffer.clear(0xEEEEEE) end
	if onProjector and projectorAvailable then
		projector.clear()
		projector.setScale(window.projectorScaleSlider.value)
	end
	if onGlasses and glassesAvailable then glasses.removeAll() end

	local min, max = tonumber(window.minimumHardnessTextBox.text), tonumber(window.maximumHardnessTextBox.text)
	local horizontalRange, verticalRange = math.floor(window.horizontalScanRangeSlider.value), math.floor(window.verticalScanRangeSlider.value)

	for x = -horizontalRange, horizontalRange do
		for z = -horizontalRange, horizontalRange do
			for y = 32 - verticalRange, 32 + verticalRange do
				if scanResult[x] and scanResult[x][z] and scanResult[x][z][y] and scanResult[x][z][y] >= min and scanResult[x][z][y] <= max then
					if onScreen then
						buffer.semiPixelSet(onScreenDataXOffset + x, onScreenDataYOffset + 32 - y, 0x454545)
					end
					if onProjector and window.projectorUpdateSwitch.state and projectorAvailable then
						projector.set(horizontalRange + x, math.floor(window.projectorYOffsetSlider.value) + y - 32, horizontalRange + z, 1)
					end
					if onGlasses and window.glassesUpdateSwitch.state and glassesAvailable then
						glassesCreateCube(x, y - 32, z, window.glassesOreColorButton.colors.default.background, "Hardness: " .. string.format("%.2f", scanResult[x][z][y]))
						os.sleep(0)
					end
				end
			end
		end
	end
end

window.onDrawStarted = function()
	updateData(true, false, false)
end

local panelWidth = 30
local panelX = buffer.screen.width - panelWidth + 1
local buttonX, objectY = panelX + 2, 2
local buttonWidth = panelWidth - 4
window:addPanel(panelX, 1, panelWidth, buffer.screen.height, 0x444444)

window.planetImage = window:addImage(buttonX, objectY, earthImage)
objectY = objectY + window.planetImage.image.height + 1

window:addLabel(buttonX, objectY, buttonWidth, 1, 0xFFFFFF, "GeoScan v2.0"):setAlignment(GUI.alignment.horizontal.center, GUI.alignment.vertical.top)
objectY = objectY + 2

window.horizontalScanRangeSlider = window:addHorizontalSlider(buttonX, objectY, buttonWidth, 0xFFDB80, 0x000000, 0xFFDB40, 0xBBBBBB, 4, 24, 16, false, "Horizontal scan range: ")
window.horizontalScanRangeSlider.roundValues = true
objectY = objectY + 3
window.verticalScanRangeSlider = window:addHorizontalSlider(buttonX, objectY, buttonWidth, 0xFFDB80, 0x000000, 0xFFDB40, 0xBBBBBB, 4, 32, 16, false, "Vertical show range: ")
window.verticalScanRangeSlider.roundValues = true
objectY = objectY + 4

window:addLabel(buttonX, objectY, buttonWidth, 1, 0xFFFFFF, "Rendering properties"):setAlignment(GUI.alignment.horizontal.center, GUI.alignment.vertical.top)
objectY = objectY + 2

window.minimumHardnessTextBox = window:addInputTextBox(buttonX, objectY, 12, 3, 0x262626, 0xBBBBBB, 0x262626, 0xFFFFFF, 2.7, nil, nil, true)
window.minimumHardnessTextBox.validator = function(text) if tonumber(text) then return true end end
window.maximumHardnessTextBox = window:addInputTextBox(buttonX + 14, objectY, 12, 3, 0x262626, 0xBBBBBB, 0x262626, 0xFFFFFF, 10, nil, nil, true)
window.maximumHardnessTextBox.validator = function(text) if tonumber(text) then return true end end
objectY = objectY + 3
window:addLabel(buttonX, objectY, buttonWidth, 1, 0xBBBBBB, "Hardness min  Hardness max"):setAlignment(GUI.alignment.horizontal.center, GUI.alignment.vertical.top)
objectY = objectY + 2


window.projectorScaleSlider = window:addHorizontalSlider(buttonX, objectY, buttonWidth, 0xFFDB80, 0x000000, 0xFFDB40, 0xBBBBBB, 0.33, 3, projector.getScale(), false, "Projection scale: ")
objectY = objectY + 3
window.projectorYOffsetSlider = window:addHorizontalSlider(buttonX, objectY, buttonWidth, 0xFFDB80, 0x000000, 0xFFDB40, 0xBBBBBB, 0, 64, 4, false, "Projection Y offset: ")
window.projectorYOffsetSlider.roundValues = true
objectY = objectY + 3

local function setButtonColorFromPalette(button)
	local selectedColor = require("palette").show("auto", "auto", button.colors.default.background)
	if selectedColor then button.colors.default.background = selectedColor end
	window:draw(); buffer.draw() 
end

local function updateProjectorColors()
	projector.setPaletteColor(1, window.color1Button.colors.default.background)
end

local color1, color2, color3 = projector.getPaletteColor(1), projector.getPaletteColor(2), projector.getPaletteColor(3)
window.color1Button = window:addButton(buttonX, objectY, buttonWidth, 1, color1, 0xBBBBBB, 0xEEEEEE, 0x262626, "Projector color"); objectY = objectY + 1
window.color1Button.onTouch = function()
	setButtonColorFromPalette(window.color1Button)
	updateProjectorColors()
end
window.glassesOreColorButton = window:addButton(buttonX, objectY, buttonWidth, 1, 0x0044FF, 0xBBBBBB, 0xEEEEEE, 0x262626, "Glasses ore color")
window.glassesOreColorButton.onTouch = function()
	setButtonColorFromPalette(window.glassesOreColorButton)
end
objectY = objectY + 2

window:addLabel(buttonX, objectY, buttonWidth, 1, 0xBBBBBB, "Projector update:")
window.projectorUpdateSwitch = window:addSwitch(buffer.screen.width - 8, objectY, 7, 0xFFDB40, 0xAAAAAA, 0xFFFFFF, true)
objectY = objectY + 2
window:addLabel(buttonX, objectY, buttonWidth, 1, 0xBBBBBB, "Glasses update:")
window.glassesUpdateSwitch = window:addSwitch(buffer.screen.width - 8, objectY, 7, 0xFFDB40, 0xAAAAAA, 0xFFFFFF, true)
objectY = objectY + 2

window:addButton(buffer.screen.width, 1, 1, 1, nil, 0xEEEEEE, nil, 0xFF2222, "X").onTouch = function()
	window:close()
	createDick()
end

window:addButton(panelX, buffer.screen.height - 5, panelWidth, 3, 0x353535, 0xEEEEEE, 0xAAAAAA, 0x262626, "Update").onTouch = function()
	updateData(false, true, true)
end
window.scanButton = window:addButton(panelX, buffer.screen.height - 2, panelWidth, 3, 0x262626, 0xEEEEEE, 0xAAAAAA, 0x262626, "Scan")
window.scanButton.onTouch = function()
	scanResult = {}
	local horizontalRange, verticalRange = math.floor(window.horizontalScanRangeSlider.value), math.floor(window.verticalScanRangeSlider.value)
	local total, current = (horizontalRange * 2 + 1) ^ 2, 0

	buffer.clear(0x0, 0x30)
	for x = -horizontalRange, horizontalRange do
		scanResult[x] = {}
		for z = -horizontalRange, horizontalRange do
			scanResult[x][z] = geolyzer.scan(x, z)
			current = current + 1
			progressReport(math.ceil(current / total * 100), "Scan progress: ")
			buffer.draw()
		end
	end

	window:draw()
	buffer.draw()
	updateData(false, true, true)
end

--------------------------------------------------------------------------------------------------------------------

window:draw()
buffer.draw()
window:handleEvents()