51 lines
1.0 KiB
Lua
51 lines
1.0 KiB
Lua
|
#!/usr/bin/env luajit
|
||
|
|
require("utils")
|
||
|
|
require("set")
|
||
|
|
|
||
|
|
local file = io.open(arg[3])
|
||
|
|
local coords = {}
|
||
|
|
local line = file:read("*l")
|
||
|
|
while line ~= "" do
|
||
|
|
local c, r = unpack(split(line, ","))
|
||
|
|
r, c = tonumber(r), tonumber(c)
|
||
|
|
table.insert(coords, { r, c })
|
||
|
|
line = file:read("*l")
|
||
|
|
end
|
||
|
|
|
||
|
|
local folds = {}
|
||
|
|
line = file:read("*l")
|
||
|
|
while line ~= nil do
|
||
|
|
local dir, v = unpack(split(line, "="))
|
||
|
|
v = tonumber(v)
|
||
|
|
if dir == "fold along y" then
|
||
|
|
for i, p in pairs(coords) do
|
||
|
|
if p[1] > v then p[1] = 2 * v - p[1] end
|
||
|
|
end
|
||
|
|
else
|
||
|
|
for i, p in pairs(coords) do
|
||
|
|
if p[2] > v then p[2] = 2 * v - p[2] end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
line = file:read("*l")
|
||
|
|
end
|
||
|
|
|
||
|
|
local width, height = 0, 0
|
||
|
|
for i, p in pairs(coords) do
|
||
|
|
if p[1] + 1 > height then height = p[1] + 1 end
|
||
|
|
if p[2] + 1 > width then width = p[2] + 1 end
|
||
|
|
end
|
||
|
|
foreach(coords, function(p) return p[1] * width + p[2] end)
|
||
|
|
coords = set(coords)
|
||
|
|
|
||
|
|
for r = 0, height - 1 do
|
||
|
|
for c = 0, width - 1 do
|
||
|
|
if coords:has(r * width + c) then
|
||
|
|
io.write("#")
|
||
|
|
else
|
||
|
|
io.write(".")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
io.write("\n")
|
||
|
|
end
|
||
|
|
io.write("\n")
|