noita-mapcap/files/capture.lua

-- Copyright (c) 2019 David Vogel
--
-- This software is released under the MIT License.
-- https://opensource.org/licenses/MIT

dofile("data/scripts/lib/coroutines.lua")
--dofile("data/scripts/lib/utilities.lua")
dofile("data/scripts/perks/perk_list.lua")

local CAPTURE_GRID_SIZE = 64 -- in ingame pixels
local CAPTURE_DELAY = 30 -- in frames
local CAPTURE_FORCE_HP = 40 -- * 25HP

local function getPlayer()
	local players = EntityGetWithTag("player_unit")
	if players == nil or #players < 1 then
		return nil
	end
	return players[1]
end

local function getPlayerPos()
	return EntityGetTransform(getPlayer())
end

local function teleportPlayer(x, y)
	EntitySetTransform(getPlayer(), x, y)
end

local function setPlayerHP(hp)
	local damagemodels = EntityGetComponent(getPlayer(), "DamageModelComponent")

	if damagemodels ~= nil then
		for i, damagemodel in ipairs(damagemodels) do
			ComponentSetValue(damagemodel, "max_hp", hp)
			ComponentSetValue(damagemodel, "hp", hp)
		end
	end
end

local function addEffectToEntity(entity, gameEffect)
	local gameEffectComp = GetGameEffectLoadTo(entity, gameEffect, true)
	if gameEffectComp ~= nil then
		ComponentSetValue(gameEffectComp, "frames", "-1")
	end
end

local function addPerkToPlayer(perkID)
	local playerEntity = getPlayer()
	local x, y = getPlayerPos()
	local perkData = get_perk_with_id(perk_list, perkID)

	-- Add effect
	addEffectToEntity(playerEntity, perkData.game_effect)

	-- Add ui icon etc
	local perkIcon = EntityCreateNew("")
	EntityAddComponent(
		perkIcon,
		"UIIconComponent",
		{
			name = perkData.ui_name,
			description = perkData.ui_description,
			icon_sprite_file = perkData.ui_icon
		}
	)
	EntityAddChild(playerEntity, perkIcon)

	--local effect = EntityLoad("data/entities/misc/effect_protection_all.xml", x, y)
	--EntityAddChild(playerEntity, effect)
end

local function preparePlayer()
	local playerEntity = getPlayer()
	addEffectToEntity(playerEntity, "PROTECTION_ALL")

	addPerkToPlayer("BREATH_UNDERWATER")
	addPerkToPlayer("INVISIBILITY")
	addPerkToPlayer("REMOVE_FOG_OF_WAR")
	addPerkToPlayer("REPELLING_CAPE")
	addPerkToPlayer("WORM_DETRACTOR")

	setPlayerHP(CAPTURE_FORCE_HP)
end

local function resetPlayer()
	setPlayerHP(CAPTURE_FORCE_HP)
end

local function doCapture()
	local ox, oy = getPlayerPos()
	local x, y = ox, oy

	preparePlayer()

	-- Coroutine to force player to x, y coordinate
	async_loop(
		function()
			teleportPlayer(x, y)
			resetPlayer()
			wait(0)
		end
	)

	-- Coroutine to calculate next coordinate, and trigger screenshots
	local i = 1
	async_loop(
		function()
			-- +x
			for i = 1, i, 1 do
				x, y = x + CAPTURE_GRID_SIZE, y
				wait(CAPTURE_DELAY)
			end
			-- +y
			for i = 1, i, 1 do
				x, y = x, y + CAPTURE_GRID_SIZE
				wait(CAPTURE_DELAY)
			end
			i = i + 1
			-- -x
			for i = 1, i, 1 do
				x, y = x - CAPTURE_GRID_SIZE, y
				wait(CAPTURE_DELAY)
			end
			-- -y
			for i = 1, i, 1 do
				x, y = x, y - CAPTURE_GRID_SIZE
				wait(CAPTURE_DELAY)
			end
			i = i + 1
		end
	)
end

-- #### UI ####

local gui = GuiCreate()

async_loop(
	function()
		if gui ~= nil then
			GuiStartFrame(gui)

			GuiLayoutBeginVertical(gui, 50, 20)
			if GuiButton(gui, 0, 0, "Start capturing map", 1) then
				doCapture()
				GuiDestroy(gui)
				gui = nil
			end
			GuiTextCentered(gui, 0, 0, "Don't do anything while the capturing process is running!")
			GuiTextCentered(gui, 0, 0, "Use ESC and close the game to stop the process.")
			GuiLayoutEnd(gui)
		end

		wait(0)
	end
)