dotfiles/home/.config/mpv/scripts/mines.lua
Marko Korhonen b1335a3628 Removed commit history
Signed-off-by: Marko Korhonen <marko.korhonen@reekynet.com>
2019-10-26 20:05:31 +03:00

368 lines
9.5 KiB
Lua

local assdraw = require 'mp.assdraw'
local W = 0
local H = 0
-- 2D array of size W*H that's addressed via gField[x][y]. Each entry is a
-- table, with the following fields:
-- is_mine: true or false
-- area_mines: mine count in the 3x3 surrounding area
-- is_covered: true or false, for visibility
-- flag: flag put by user, one of FLAG_*
local gField = nil
local FLAG_NONE = "" -- not flagged
local FLAG_MINE = "" -- "⚐" -- flagged as containing mine
local FLAG_MAYBE_MINE = "!" -- flagged as maybe containing mine
local FLAG_MAYBE_SAFE = "?" -- flagged as maybe empty
local STATUS_PLAYING = "playing"
local STATUS_WON = "won"
local STATUS_LOST = "lost"
local gStatus = nil
local gMines = 0
local gX = 0
local gY = 0
local gNeedRefresh = false
local gHidden = true
local gTransparent = false
local PRESETS = {
-- taken from kmines
{ name = "easy", w = 9, h = 9, mines = 10 },
{ name = "medium", w = 16, h = 16, mines = 40 },
{ name = "hard", w = 30, h = 16, mines = 99 },
}
local gCurrentPreset = 2
function init_field()
local preset = PRESETS[gCurrentPreset]
gStatus = STATUS_PLAYING
gField = {}
W = preset.w
H = preset.h
gMines = math.min(preset.mines, W * H - 1)
gX = 1
gY = 1
for x = 1, W do
gField[x] = {}
for y = 1, H do
gField[x][y] = {
is_mine = false,
area_mines = 0,
is_covered = true,
flag = FLAG_NONE,
}
end
end
-- place mines using the dumbfuck algorithm
local place_mines = gMines
while place_mines > 0 do
local x = math.random(1, W)
local y = math.random(1, H)
if not gField[x][y].is_mine then
gField[x][y].is_mine = true
place_mines = place_mines - 1
end
end
-- pick a random start position (also using dumbfuck algorithm)
for i = 1, 1000000 do
local x = math.random(1, W)
local y = math.random(1, H)
if not gField[x][y].is_mine then
gX = x
gY = y
break
end
end
-- compute proximities after mines have been placed
for y = 1, H do
for x = 1, W do
local tile = gField[x][y]
for a_x = -1, 1 do
for a_y = -1, 1 do
local t_x = x + a_x
local t_y = y + a_y
if t_x >= 1 and t_x <= W and t_y >= 1 and t_y <= H and
gField[t_x][t_y].is_mine
then
tile.area_mines = tile.area_mines + 1
end
end
end
end
end
uncover()
check_status()
gNeedRefresh = true
end
function uncover_at(x, y)
if x < 1 or x > W or y < 1 or y > H then
return
end
local tile = gField[x][y]
if not tile.is_covered then
return
end
tile.is_covered = false
if tile.is_mine then
return -- lost anyway
end
-- uncover mines as far as it goes
-- apparently, the standard thing to do is recursively uncovering all
-- tiles which have 0 neightbours - tiles with 1 or more neighbours are
-- uncovered, but not recursively
if tile.area_mines == 0 then
for a_x = -1, 1 do
for a_y = -1, 1 do
uncover_at(x + a_x, y + a_y)
end
end
end
gNeedRefresh = true
end
function check_status()
if gStatus ~= STATUS_PLAYING then
return
end
local won = true
local lost = false
for y = 1, H do
for x = 1, W do
local tile = gField[x][y]
won = won and (tile.is_mine == tile.is_covered)
lost = lost or (tile.is_mine and not tile.is_covered)
end
end
if lost then
gStatus = STATUS_LOST
gNeedRefresh = true
elseif won then
gStatus = STATUS_WON
gNeedRefresh = true
end
end
function uncover()
if gStatus ~= STATUS_PLAYING and not gField[gX][gY].is_covered then
init_field()
render()
return
end
uncover_at(gX, gY)
check_status()
render()
end
function flag()
local tile = gField[gX][gY]
local cycle = {FLAG_NONE, FLAG_MINE, FLAG_MAYBE_MINE, FLAG_MAYBE_SAFE}
for i = 1, #cycle do
if tile.flag == cycle[i] then
tile.flag = cycle[(i - 1 + 1) % #cycle + 1]
break
end
end
if not tile.is_covered then
tile.flag = FLAG_NONE
end
force_render()
end
function move(x, y)
gX = math.min(math.max(gX + x, 1), W)
gY = math.min(math.max(gY + y, 1), H)
force_render()
end
function force_render()
gNeedRefresh = true
render()
end
function render()
if not gNeedRefresh then
return
end
if gHidden then
mp.set_osd_ass(1280, 720, "")
return
end
local canvas_w = 1280
local canvas_h = 720
local dw, dh, da = mp.get_osd_size()
if dw ~= nil and dw > 0 and dh > 0 then
canvas_w = dw / dh * canvas_h
end
local tile_wh = 32
local o_x = canvas_w / 2 - tile_wh * W / 2
local o_y = canvas_h / 2 - tile_wh * (H + 2) / 2 + tile_wh
local ass = assdraw.ass_new()
local transp = nil
if gTransparent then
transp = "{\\1a&HA0&\\3a&HA0&}"
end
-- some shitty background
ass:new_event()
ass:append("{\\1c&Ha3a3a3&\\1a&H30&}")
if transp then
ass:append(transp)
end
ass:pos(o_x - tile_wh, o_y - tile_wh)
ass:draw_start()
ass:rect_cw(0, 0, (W + 2) * tile_wh, (H + 2) * tile_wh)
-- grid
local function grid_line(x0, y0, x1, y1)
ass:new_event()
ass:append("{\\bord0.5}")
if transp then
ass:append(transp)
end
ass:pos(x0, y0)
ass:draw_start()
ass:coord(0, 0)
ass:line_to(x1 - x0, y1 - y0)
end
for x = 0, W do
local p_x = x * tile_wh + o_x
grid_line(p_x, o_y, p_x, o_y + tile_wh * H)
end
for y = 0, H do
local p_y = y * tile_wh + o_y
grid_line(o_x, p_y, o_x + tile_wh * W, p_y)
end
local function draw_sym(x, y, sym, c)
ass:new_event()
ass:pos(x, y)
ass:append("{\\an5\\fs25\\bord0\\1c&H" .. c .. "&\\b1}" .. sym)
end
for x = 1, W do
for y = 1, H do
local tile = gField[x][y]
local p_x = (x - 1) * tile_wh + tile_wh / 2 + o_x
local p_y = (y - 1) * tile_wh + tile_wh / 2 + o_y
local wh = tile_wh - 4
local sym = nil
if tile.is_covered then
ass:new_event()
if transp then
ass:append(transp)
end
ass:pos(p_x, p_y)
ass:draw_start()
ass:round_rect_cw(-wh / 2, -wh / 2, wh / 2, wh / 2, 5)
ass:draw_stop()
elseif tile.is_mine then
draw_sym(p_x, p_y, "💣", "0000FF")
elseif tile.area_mines > 0 then
draw_sym(p_x, p_y, tile.area_mines, "000000")
end
if tile.flag ~= FLAG_NONE then
draw_sym(p_x, p_y, tile.flag, "FF0000")
end
if x == gX and y == gY then
local wh = tile_wh - 12
ass:new_event()
ass:append("{\\1a&HFF&}")
ass:pos(p_x, p_y)
ass:draw_start()
ass:rect_cw(-wh / 2, -wh / 2, wh / 2, wh / 2)
ass:draw_stop()
end
end
end
local banner = nil
if gStatus == STATUS_WON then
banner = "You may have won, but actually you just wasted time."
elseif gStatus == STATUS_LOST then
banner = "You lost (and wasted time)."
end
if banner then
ass:new_event()
ass:pos(o_x + tile_wh * W / 2, o_y - tile_wh - 10)
ass:append("{\\fs40\\b1\\an2}" .. banner)
end
mp.set_osd_ass(canvas_w, canvas_h, ass.text)
end
mp.observe_property("osd-width", "native", force_render)
mp.observe_property("osd-height", "native", force_render)
init_field()
force_render()
function toggle_transp()
gTransparent = not gTransparent
force_render()
end
function cycle_preset()
gCurrentPreset = (gCurrentPreset + 1 - 1) % #PRESETS + 1
init_field()
render()
end
function toggle_show()
if gHidden then
gHidden = false
local REP = {repeatable = true}
mp.add_forced_key_binding("left", "mines-left", function() move(-1, 0) end, REP)
mp.add_forced_key_binding("right", "mines-right", function() move(1, 0) end, REP)
mp.add_forced_key_binding("up", "mines-up", function() move(0, -1) end, REP)
mp.add_forced_key_binding("down", "mines-down", function() move(0, 1) end, REP)
mp.add_forced_key_binding("space", "mines-uncover", uncover)
mp.add_forced_key_binding("b", "mines-flag", flag)
mp.add_forced_key_binding("t", "mines-transp", toggle_transp)
mp.add_forced_key_binding("w", "mines-preset", cycle_preset)
else
gHidden = true
mp.remove_key_binding("mines-left")
mp.remove_key_binding("mines-right")
mp.remove_key_binding("mines-up")
mp.remove_key_binding("mines-down")
mp.remove_key_binding("mines-uncover")
mp.remove_key_binding("mines-flag")
mp.remove_key_binding("mines-transp")
mp.remove_key_binding("mines-preset")
end
force_render()
end
mp.add_forced_key_binding("ctrl+x", "mines-show", toggle_show)