local p = {}
function chessboard(args, size, rows, cols, rev, letters, numbers, header, footer, align, clear)
function colchar( col )
return ( "abcdefghijklmnopqrstuvwxyz" ):sub( col, col )
end
function image_square( pc, row, col, size )
local colornames = { l = 'white', d = 'black' }
local piecenames = {
p = 'pawn',
r = 'rook',
n = 'knight',
b = 'bishop',
q = 'queen',
k = 'king',
a = 'archbishop',
c = 'chancelor',
z = 'champion',
w = 'wizard',
t = 'fool',
h = 'upside-down pawn',
m = 'upside-down rook',
s = 'upside-down knight',
f = 'upside-down king',
e = 'upside-down bishop',
g = 'upside-down queen',
}
local symnames = {
xx = 'black cross',
ox = 'white cross',
xo = 'black circle',
oo = 'white circle',
ul = 'up-left arrow',
ua = 'up arrow',
ur = 'up-right arrow',
la = 'left arrow',
ra = 'right arrow',
dl = 'down-left arrow',
da = 'down arrow',
dr = 'down-right arrow',
lr = 'left-right arrow',
ud = 'up-down arrow',
x0 = 'zero',
x1 = 'one',
x2 = 'two',
x3 = 'three',
x4 = 'four',
x5 = 'five',
x6 = 'six',
x7 = 'seven',
x8 = 'eight',
x9 = 'nine',
}
local color = mw.ustring.gsub( pc, '^.*(%w)(%w).*$', '%2' ) or ''
local piece = mw.ustring.gsub( pc, '^.*(%w)(%w).*$', '%1' ) or ''
local alt = colchar( col ) .. row .. ' '
if ( colornames[color] and piecenames[piece] ) then
alt = alt .. colornames[color] .. ' ' .. piecenames[piece]
else
alt = alt .. ( symnames[piece .. color] or piece .. ' ' .. color )
end
local ld = (((row + col) % 2) == 0) and 'd' or 'l'
return string.format( '[[File:Chess %s%s%s45.svg|%dx%dpx|alt=%s|%s]]', piece, color, ld, size, size, alt, alt )
end
function letters_row( rev, num_lt, num_rt, cols )
local res = '<tr style="vertical-align:middle">' .. ( num_lt and '<td></td>' or '' ) .. '<td style="height:18px">'
for k = 1, cols do
res = res .. colchar(rev and (cols - k + 1) or k) .. '</td><td>'
end
res = res .. '</td>' .. ( num_lt and '<td></td>' or '' ) .. '</tr>'
return res
end
local letters_tp = letters:match('both') or letters:match('top')
local letters_bt = letters:match('both') or letters:match('bottom')
local numbers_lt = numbers:match('both') or numbers:match('left')
local numbers_rt = numbers:match('both') or numbers:match('right')
local width = cols * size + 2
if ( numbers_lt ) then width = width + 18 end
if ( numbers_rt ) then width = width + 18 end
local b = ''
local caption = ''
if ( letters_tp ) then b = b .. letters_row(rev, numbers_lt, numbers_rt, cols) .. '\n' end
for trow = 1,rows do
local row = rev and trow or (rows - trow + 1)
b = b .. '<tr style="vertical-align:middle">'
if ( numbers_lt ) then b = b .. '<td style="width:18px">' .. row .. '</td>' end
for tcol = 1,cols do
local col = rev and (cols - tcol + 1) or tcol
local idx = cols*(rows - row) + col + 2
if (args[idx] == nil) then args[idx] = ' ' end
local img = image_square(args[idx]:match('%w%w') or '', row, col, size )
b = b .. '<td>' .. img .. '</td>'
end
if ( numbers_rt ) then b = b .. '<td style="width:18px">' .. row .. '</td>' end
end
if ( letters_bt ) then b = b .. letters_row(rev, numbers_lt, numbers_rt, cols) .. '\n' end
if footer:match('^%s*$')
then
else
caption = '<div class="thumbcaption">' .. footer .. '</div>\n'
end
return '<div class="thumb ' .. align .. '" style="clear:' .. clear .. '; text-align:center;">'
.. header .. '\n<div class="thumbinner" style="width:' .. width .. 'px;">\n'
.. '<table cellpadding=0 cellspacing=0 style="background:white; font-size:88%; border:1px #b0b0b0 solid; padding:0;'
.. ' margin:auto">\n' .. b .. '\n</table>\n' .. caption .. '</div></div>'
end
function convertFenToArgs( fen )
-- converts FEN notation to an array of positions, offset by 2
local res = {' ', ' '}
-- Loop over rows, which are delimited by /
for srow in string.gmatch("/" .. fen, "/%w+") do
-- Loop over all letters and numbers in the row
for piece in srow:gmatch( "%w" ) do
if (piece:match("%d")) then
-- if a digit
for k=1,piece do
table.insert(res,' ')
end
else
-- not a digit
local color = piece:match( '%u' ) and 'l' or 'd'
piece = piece:lower()
table.insert(res, piece .. color )
end
end
end
return res
end
function p.board(frame)
local args = frame.args
local pargs = frame:getParent().args
local size = (args.size or pargs.size) or '26'
local reverse = ( (args.reverse or pargs.reverse) or '' ):lower() == "true"
local rows = args.rows or pargs.rows or 8
local cols = args.cols or pargs.cols or 8
local letters = ( args.letters or pargs.letters or 'both' ):lower()
local numbers = ( args.numbers or pargs.numbers or 'both' ):lower()
local header = args[2] or pargs[2] or ''
local footer = args[3 + rows*cols] or pargs[3 + rows*cols] or ''
local align = ( args[1] or pargs[1] or 'tright' ):lower()
local clear = ( args.clear or pargs.clear ) or ( align:match('tright') and 'right' or 'none' )
local fen = args.fen or pargs.fen
size = mw.ustring.match(size, '[%d]+') or '26' -- remove px from size
if (fen) then
align = ( args.align or pargs.align or 'tright' ):lower()
clear = ( args.clear or pargs.clear ) or ( align:match('tright') and 'right' or 'none' )
header = args.header or pargs.header or ''
footer = args.footer or pargs.footer or ''
return chessboard(convertFenToArgs( fen ), size, rows, cols, reverse, letters, numbers, header, footer, align, clear)
end
if args[3] then
return chessboard(args, size, rows, cols, reverse, letters, numbers, header, footer, align, clear)
else
return chessboard(pargs, size, rows, cols, reverse, letters, numbers, header, footer, align, clear)
end
end
return p
