100 lines
2.0 KiB
Lua
100 lines
2.0 KiB
Lua
curses = require "curses"
|
|
|
|
window = curses.initscr()
|
|
|
|
tower = {{6, 5, 4, 3, 2}, {}, {}}
|
|
|
|
function len(array)
|
|
local result = 0
|
|
for k in pairs(array) do
|
|
result = result+1
|
|
end
|
|
return result
|
|
end
|
|
|
|
function pop(array)
|
|
return table.remove(array)
|
|
end
|
|
|
|
function lines(window)
|
|
local lines, cols = window:getmaxyx()
|
|
return lines
|
|
end
|
|
|
|
function cols(window)
|
|
local lines, cols = window:getmaxyx()
|
|
return cols
|
|
end
|
|
|
|
|
|
function render_disk(window, line, col, size)
|
|
col = col-size+1
|
|
for i=1,size do
|
|
window:attron(curses.color_pair(size))
|
|
window:mvaddstr(line, col, " ")
|
|
window:attroff(curses.color_pair(size))
|
|
col = col+2
|
|
end
|
|
end
|
|
|
|
function render_tower(window, line, col, tower_index, tower)
|
|
window:attron(curses.A_BOLD)
|
|
window:mvaddch(line+2, col, string.char(96+tower_index))
|
|
window:attroff(curses.A_BOLD)
|
|
window:attron(curses.color_pair(7))
|
|
window:mvaddstr(line+1, col-6, " ")
|
|
window:attroff(curses.color_pair(7))
|
|
for i, n in ipairs(tower) do
|
|
render_disk(window, line, col, n)
|
|
line = line - 1
|
|
end
|
|
for i=1,5-len(tower)+1 do
|
|
window:attron(curses.color_pair(7))
|
|
window:mvaddstr(line, col, " ")
|
|
window:attroff(curses.color_pair(7))
|
|
line = line - 1
|
|
end
|
|
end
|
|
|
|
function render(window)
|
|
window:clear()
|
|
local lines, cols = window:getmaxyx()
|
|
local line = math.floor(lines/2)
|
|
local col = math.floor(cols/4)
|
|
for i,t in ipairs(tower) do
|
|
render_tower(window, line, i*col, i, t)
|
|
end
|
|
curses.refresh()
|
|
end
|
|
|
|
|
|
function make_move(from, to)
|
|
local disk = pop(tower[from])
|
|
table.insert(tower[to], disk)
|
|
end
|
|
|
|
function update(window)
|
|
window:mvaddstr(lines(window)-2, 5, "tower to remove top disk from? ")
|
|
local from = string.byte(curses.getch()) - 96
|
|
window:mvaddstr(lines(window)-1, 5, "tower to stack it on? ")
|
|
local to = string.byte(curses.getch()) - 96
|
|
make_move(from, to)
|
|
end
|
|
|
|
|
|
function main()
|
|
curses.start_color()
|
|
curses.use_default_colors()
|
|
for i=1,7 do
|
|
curses.init_pair(i, 0, i)
|
|
end
|
|
|
|
while true do
|
|
render(window)
|
|
update(window)
|
|
end
|
|
|
|
curses.endwin()
|
|
end
|
|
main()
|