TAS.lua

local TAS={}

function deepcopy(orig, copies)
	copies = copies or {}
	local orig_type = type(orig)
	local copy
	if orig_type == 'table' then
		if copies[orig] then
			copy = copies[orig]
		else
			copy = {}
			copies[orig] = copy
			for orig_key, orig_value in next, orig, nil do
				copy[deepcopy(orig_key, copies)] = deepcopy(orig_value, copies)
			end
			setmetatable(copy, getmetatable(orig))
		end
	else -- number, string, boolean, etc
		copy = orig
	end
	return copy
end
local globals={"level_index","level_intro","freeze_time","camera_x","camera_y","collected","show_score","infade","input_x","input_jump_pressed","input_grapple_pressed","axis_x_value","axis_x_turned","input_jump","input_grapple","level_checkpoint"}
local function get_state()
	local copies={}
	for _,o in pairs(pico8.cart.objects) do
		copies[o.base]=o.base
	end
	local state=deepcopy(pico8.cart.objects,copies)
	local state_flag={}
	for _, o in pairs(globals) do
		state_flag[o]=deepcopy(pico8.cart[o])
	end
	state_flag.pico_cam_x=pico8.camera_x
	state_flag.pico_cam_y=pico8.camera_y
	return state, state_flag
end
TAS.get_state=get_state

function TAS.store_state()
	local state, state_flag=get_state()
	TAS.states[TAS.frame]=state
	TAS.states_flags[TAS.frame]=state_flag
end
local function set_state(state, state_flag)
	pico8.cart.objects=state
	for _, o in pairs(globals) do
		pico8.cart[o]=deepcopy(state_flag[o])
	end
	pico8.camera_x=state_flag.pico_cam_x
	pico8.camera_y=state_flag.pico_cam_y
end
TAS.set_state=set_state


local function update()
	if TAS.advance_frame then
		TAS.advance_frame=false

		local player_alive=false
		for _,o in pairs(pico8.cart.objects) do
			if o.base==pico8.cart.player and o.state~=99 then
				player_alive=true
			end
		end
		if TAS.active then
			TAS.frame=TAS.frame+1
		end
		if player_alive and pico8.cart.level_intro==0 and not TAS.active then
			TAS.active=true
			TAS.frame=0
		elseif not TAS.active then
			TAS.active=false
		end
		--TAS.active=true

		if TAS.prev_state.level_index~=-1 and TAS.prev_state.level_index~=pico8.cart.level_index then
			if TAS.save_reproduce then
				if not TAS.final_reproduce then
					TAS.save_reproduce=false
					TAS.reproduce=false
				end
				TAS.save_file(true,TAS.frame-1)
				log("Saved cleaned file to "..love.filesystem.getRealDirectory(""))
			end

			if not TAS.final_reproduce then
				TAS.active=false
				load_level(TAS.prev_state.level_index)
			else
				local numFrames=TAS.frame
				TAS.load_file(love.filesystem.newFile(("TAS/TAS%d.tas"):format(pico8.cart.level_index)))
				TAS.reproduce=true
				log(("%02d:%02d.%02d (%d)"):format(pico8.cart.minutes,pico8.cart.seconds,math.floor(100*pico8.cart.frames/30+0.5),numFrames))
			end
		end

		if not TAS.keypresses[TAS.frame] then
			TAS.keypresses[TAS.frame]=deepcopy(TAS.keypresses[TAS.frame-1])
		end
	end
	if TAS.reproduce then
		TAS.advance_frame=true
		local state, state_flag=get_state()
		TAS.states[TAS.current_frame]=state
		TAS.states_flags[TAS.current_frame]=state_flag
	end
	TAS.prev_state.level_index=pico8.cart.level_index
end
TAS.update=update

local function draw()
	if TAS.showdebug and not TAS.final_reproduce then
		--pico8.cart.camera(0,0)
		love.graphics.push()
		love.graphics.origin()
		pico8.cart.rectfill(1,1,17,7,0)
		pico8.cart.print(tostring(TAS.frame),2,2,7)
		local inputs_x=19
		pico8.cart.rectfill(inputs_x,1,inputs_x+24,11,0)
		if TAS.active then
			pico8.cart.rectfill(inputs_x + 12, 7, inputs_x + 14, 9, TAS.keypresses[TAS.frame][0] and 7 or 1) -- l
			pico8.cart.rectfill(inputs_x + 20, 7, inputs_x + 22, 9, TAS.keypresses[TAS.frame][1] and 7 or 1) -- r
			pico8.cart.rectfill(inputs_x + 16, 3, inputs_x + 18, 5, TAS.keypresses[TAS.frame][2] and 7 or 1) -- u
			pico8.cart.rectfill(inputs_x + 16, 7, inputs_x + 18, 9, TAS.keypresses[TAS.frame][3] and 7 or 1) -- d
			pico8.cart.rectfill(inputs_x + 2, 7, inputs_x + 4, 9, TAS.keypresses[TAS.frame][4] and 7 or 1) -- z
			pico8.cart.rectfill(inputs_x + 6, 7, inputs_x + 8, 9, TAS.keypresses[TAS.frame][5] and 7 or 1) -- x
		end
		love.graphics.pop()
	end

end
TAS.draw=draw

local function save_file(compress,idx)
	local file=love.filesystem.newFile("TAS"..tostring(TAS.prev_state.level_index)..".tas")

	file:open("w")
	local finish
	if(compress) then
		finish=idx
	else
		finish=#TAS.keypresses
	end
	for j=0,finish do
		local i=TAS.keypresses[j]
		local line=0
		for x=0,5 do
			if i[x] then
				if x==0 then
					line=line+1
				elseif x==1 then
					line=line+2
				elseif x==2 then
					line=line+4
				elseif x==3 then
					line=line+8
				elseif x==4 then
					line=line+16
				else
					line=line+32
				end
			end
		end
		file:write(tostring(line)..",")
	end
	file:close()
end
TAS.save_file=save_file

local function clear_pico8()
	pico8.cart.frames=0
	pico8.cart.seconds=0
	pico8.cart.minutes=0
	pico8.cart.deaths=0
	pico8.collected={}
end

local function clear_state()
	TAS.states={}
	TAS.state_flags={}
	TAS.keypresses={}
	TAS.keypresses[0]={}
	TAS.frame=0
	TAS.prev_state={level_index=-1}
	TAS.active=false
	TAS.save_reproduce=false
	TAS.reproduce=false
end
function load_level(idx)

	TAS.frame=0
	TAS.states={}
	TAS.states_flags={}
	if not TAS.final_reproduce then
		pico8.cart.collected={}
		pico8.cart.goto_level(idx)
		pico8.cart.level_intro=0
		pico8.cart.infade=15
		--reset input state
		pico8.cart.input_x = 0
		pico8.cart.input_jump = false
		pico8.cart.input_jump_pressed = 0
		pico8.cart.input_grapple = false
		pico8.cart.input_grapple_pressed = 0
		pico8.cart.axis_x_value = 0
		pico8.cart.axis_x_turned = false
	end
	TAS.active=pico8.cart.level_intro<=1
end

local function load_file(file)
	TAS.keypresses={}
	local data=file:read()
	if data~=nil then
		local iterator=0
		for s in data:gmatch("([^,]+)") do
			TAS.keypresses[iterator]={}
			local c=tonumber(s)
			for i=0,5 do
				if math.floor(c/math.pow(2,i))%2==1 then
					TAS.keypresses[iterator][i]=true
				end
			end
			iterator=iterator+1
		end
	end
	TAS.reproduce=false
	TAS.save_reproduce=false
	TAS.advance_frame=false
	load_level(pico8.cart.level_index)
end
TAS.load_file=load_file
local function keypress(key)
	if key=='p' then
		TAS.reproduce=not TAS.reproduce
		TAS.save_reproduce=false
	elseif key=='e' then
		TAS.showdebug=not TAS.showdebug
	elseif key=='n' or key=='i' then
		TAS.final_reproduce=not TAS.final_reproduce
		if TAS.final_reproduce then
			clear_pico8()
			clear_state()
			pico8.cart.goto_level(1)
			TAS.load_file(love.filesystem.newFile("TAS/TAS1.tas"))
		end
		TAS.reproduce=TAS.final_reproduce
		TAS.save_reproduce=key=='i' and TAS.final_reproduce
	elseif key=='y' then
		for _,o in pairs(pico8.cart.objects) do
			if o.base==pico8.cart.player then
				log("----------------------------------")
				log("position: "..tostring(o.x)..", "..tostring(o.y))
				log("rem values: "..tostring(o.remainder_x)..", "..tostring(o.remainder_y))
				log("speed: "..tostring(o.speed_x)..", "..tostring(o.speed_y))
				log("camera: "..tostring(pico8.cart.camera_x)..", "..tostring(pico8.cart.camera_y))
			end
		end
	end
	if not TAS.final_reproduce then
		if key=='f' or key=='s' or key=='r' then
			clear_state()
			local off=key=='f' and 1 or key=='s' and -1 or 0
			local new_idx=math.min(math.max(pico8.cart.level_index+off,1),8)
			load_level(new_idx)
		elseif key=='d' then
			TAS.reproduce=false
			TAS.save_reproduce=false
			load_level(pico8.cart.level_index)
		elseif key=='t' then
			for i=TAS.frame+1,#TAS.keypresses do
				TAS.keypresses[i]=nil
			end
		elseif key=='l' then
			TAS.advance_frame=true
		elseif key=='k' then
			if TAS.frame>0 then
				TAS.frame=TAS.frame-1
				set_state(TAS.states[TAS.frame], TAS.states_flags[TAS.frame])
			end
		elseif key=='backspace' then
			for i=TAS.frame,#TAS.keypresses do
				TAS.keypresses[i]=TAS.keypresses[i+1]
			end
			if not TAS.keypresses[TAS.frame] then
				TAS.keypresses[TAS.frame]={}
			end
		elseif key=="=" then
			for i=#TAS.keypresses,TAS.frame,-1 do
				TAS.keypresses[i+1]=TAS.keypresses[i]
			end
			TAS.keypresses[TAS.frame]={}
		elseif key=='up' then
			TAS.keypresses[TAS.frame][2]=not TAS.keypresses[TAS.frame][2]
		elseif key=='down' then
			TAS.keypresses[TAS.frame][3]=not TAS.keypresses[TAS.frame][3]
		elseif key=='left' then
			TAS.keypresses[TAS.frame][0]=not TAS.keypresses[TAS.frame][0]
		elseif key=='right' then
			TAS.keypresses[TAS.frame][1]=not TAS.keypresses[TAS.frame][1]
		elseif key=='c' or key=='z' then
			TAS.keypresses[TAS.frame][4]=not TAS.keypresses[TAS.frame][4]
		elseif key=='x' then
			TAS.keypresses[TAS.frame][5]=not TAS.keypresses[TAS.frame][5]
		elseif key=='m' then
			TAS.save_file(false)
			log("Saved file to "..love.filesystem.getRealDirectory(""))
		elseif key=='u' then
			TAS.save_file(false)
			log("Saved uncleaned file to "..love.filesystem.getRealDirectory(""))
			TAS.reproduce=true
			TAS.save_reproduce=true
		elseif key=='w' then
			TAS.load_file(love.filesystem.newFile("TAS/TAS"..(pico8.cart.level_index)..".tas"))
		end
	end
end
TAS.keypress=keypress

local function init()
	--setfenv(draw_time,pico8.cart)
	--pico8.cart.draw_time=draw_time
	--[[local draw_orb=clone_function(pico8.cart.orb.draw)
	local draw_chest=clone_function(pico8.cart.big_chest.draw)
	local draw_time=clone_function(pico8.cart.draw_time)]]--

	if pico8.cart.draw_time~=nil then
		local draw_time=pico8.cart.draw_time
		pico8.cart.draw_time=function(...)
			local arg={...}
			if TAS.final_reproduce then
				draw_time(...)
			end
		end
	end
	local _draw=pico8.cart._draw
	pico8.cart._draw=function()
		_draw()
		if pico8.cart.shake > 0 then
			pico8.cart.shake=pico8.cart.shake-1
		end
		--[[if pico8.cart.level_index<8 then
			pico8.cart.draw_time(1,1)
		end]]
	end
	load_level(1)
end
TAS.init=init
local function restart()
	TAS.frame=0
	TAS.advance_frame=false
	TAS.keypresses={}
	TAS.keypresses[0]={}
	TAS.active=false
	TAS.states={}
	TAS.states_flags={}
	TAS.current_frame=0
	TAS.showdebug=true
	TAS.reproduce=false
	TAS.final_reproduce=false
	TAS.save_reproduce=false
	TAS.start=false
	TAS.prev_state={level_index=-1}
end
TAS.restart=restart
restart()
return TAS