Module:Chessboard: Difference between revisions

Content deleted Content added
m Frietjes moved page Module:Chess from fen to Module:Chessboard without leaving a redirect: expanding this to make it work for more than just FEN
m use clearer variable names - took me a while to figure out what these mean
 
(66 intermediate revisions by 17 users not shown)
Line 1:
local p = {}
function chessboard( fen, size, reverse )
local piecenames = { p = 'Pawn', r = 'Rook', n = 'Knight', b = 'Bishop', q = 'Queen', k = 'King' }
local colornames = { l = 'White', d = 'Black' }
 
local cfg, nrows, ncols
function rowchar( row ) return 9 - row end
function filechar( file ) return ( "abcdefgh" ):sub( file, file ) end
function coord( ind ) return ( reverse and ( 8 - ind ) * size ) or ( ind - 1 ) * size end
function piecediv( piece, row, file, res )
local color = piece:match( '%u' ) and 'l' or 'd'
piece = piece:lower()
local alt = string.format("%s%s %s %s", filechar( file ), rowchar( row ), colornames[color], piecenames[piece] or piece)
local img = string.format('[[File:Chess %s%st45.svg|%dx%dpx|alt=%s|%s]]', piece, color, size, size, alt, alt)
table.insert( res, string.format('<div style="position:absolute;z-index:3;top:%dpx;left:%dpx;">%s</div>', coord( row ), coord( file ), img) )
end
 
local function oneRowinnerboard( sargs, rowsize, res rev)
pattern = cfg.pattern or '%w%w'
local file = 1
local root = mw.html.create('div')
for piece in s:gmatch( "%w" ) do -- if a digit, increment "file" by the digit. else, add the piece _and_ increment file by 1
root:addClass('chess-pieces notheme')
file = file + ( piece:match("%d") or piecediv( piece, row, file, res ) or 1 )
:css('position', 'relative')
end
:wikitext(cfg.image_board(size))
end
for trow = 1,nrows do
local row = rev and trow or ( 1 + nrows - trow )
for tcol = 1,ncols do
local col = rev and ( 1 + ncols - tcol ) or tcol
local piece = args[ncols * ( nrows - row ) + col + 2] or ''
if piece:match( pattern ) then
local img = cfg.image_square(piece:match(pattern), row, col, size )
root:tag('div')
:css('top', tostring(( trow - 1 ) * size) .. 'px')
:css('left', tostring(( tcol - 1 ) * size) .. 'px')
:wikitext(img)
end
end
end
 
return tostring(root)
local result = {}
end
table.insert(result, string.format([=[
 
<div class="chess-fen" style="position:relative;">
function chessboard(args, size, rev, letters, numbers, header, footer, align, clear)
[[File:Chessboard480.png|%dx%dpx|link=]]
function letters_row( rev, num_lt, num_rt )
]=], size * 8, size * 8))
local letters = cfg.letters()
local root = mw.html.create('')
if num_lt then
root:tag('td')
end
for k = 1,ncols do
root:tag('td')
:css('height', '18px')
:css('width', size .. 'px')
:wikitext(rev and letters[1+ncols-k] or letters[k])
end
if num_rt then
root:tag('td')
end
return tostring(root)
end
local letters_top = letters:match( 'both' ) or letters:match( 'top' )
local letters_bottom = letters:match( 'both' ) or letters:match( 'bottom' )
local numbers_left = numbers:match( 'both' ) or numbers:match( 'left' )
local numbers_right = numbers:match( 'both' ) or numbers:match( 'right' )
local width = ncols * size + 2
if ( numbers_left ) then width = width + 18 end
if ( numbers_right ) then width = width + 18 end
 
local root = mw.html.create('div')
:addClass('chessboard')
:addClass('thumb')
:addClass('noviewer')
:addClass(align)
if( header and header ~= '' ) then
root:tag('div')
:addClass('center')
:css('line-height', '130%')
:css('margin', '0 auto')
:css('max-width', (width + ncols) .. 'px')
:wikitext(header)
end
local div = root:tag('div')
:addClass('thumbinner')
:css('width', width .. 'px')
local b = div:tag('table')
:attr('cellpadding', '0')
:attr('cellspacing', '0')
 
if ( letters_top ) then
b:tag('tr')
:wikitext(letters_row( rev, numbers_left, numbers_right ))
end
local tablerow = b:tag('tr')
if ( numbers_left ) then
tablerow:tag('td')
:css('width', '18px')
:css('height', size .. 'px')
:wikitext(rev and 1 or nrows)
end
local td = tablerow:tag('td')
:attr('colspan', ncols)
:attr('rowspan', nrows)
:wikitext(innerboard(args, size, rev))
if ( numbers_right ) then
tablerow:tag('td')
:css('width', '18px')
:css('height', size .. 'px')
:wikitext(rev and 1 or nrows)
end
if ( numbers_left or numbers_right ) then
for trow = 2, nrows do
local idx = rev and trow or ( 1 + nrows - trow )
tablerow = b:tag('tr')
if ( numbers_left ) then
tablerow:tag('td')
:css('height', size .. 'px')
:wikitext(idx)
end
if ( numbers_right ) then
tablerow:tag('td')
:css('height', size .. 'px')
:wikitext(idx)
end
end
end
if ( letters_bottom ) then
b:tag('tr')
:wikitext(letters_row( rev, numbers_left, numbers_right ))
end
 
if footer and mw.text.trim(footer)~='' then
div:tag('div')
:addClass('thumbcaption')
:wikitext(footer)
end
 
return tostring(root) ..
mw.getCurrentFrame():extensionTag( 'templatestyles', '', { src = 'Module:Chessboard/styles.css' } )
end
 
function convertFenToArgs( fen )
-- converts FEN notation to 64 entry 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 convertArgsToFen( args, offset )
local row = 0
function nullOrWhitespace( s ) return not s or s:match( '^%s*(.-)%s*$' ) == '' end
for srow in string.gmatch("/" .. fen, "/%w+") do
function piece( s )
row = row + 1
return nullOrWhitespace( s ) and 1
oneRow(srow, row, result)
or s:gsub( '%s*(%a)(%a)%s*', function( a, b ) return b == 'l' and a:upper() or a end )
end
end
table.insert(result, '</div>')
return result
local res = ''
offset = offset or 0
for row = 1, 8 do
for file = 1, 8 do
res = res .. piece( args[8*(row - 1) + file + offset] )
end
if row < 8 then res = res .. '/' end
end
return mw.ustring.gsub(res, '1+', function( s ) return #s end )
end
 
function p.board(frame)
--[[
local args = frame.args
this function is to be used only from Template:Chess diagram.
local pargs = frame:getParent().args
it provides part of the FEN string - the part that describes the board itself.
local style = args.style or pargs.style or 'Chess'
unfortunately, the template does not carry enough information to create the 2nd
cfg = require('Module:Chessboard/' .. style)
part of the FEN, which includes information about whose turn is it, en-passant state,
nrows, ncols = cfg.dims()
castling, etc.
]]
local size = args.size or pargs.size or '26'
local reverse = ( args.reverse or pargs.reverse or '' ):lower() == "true"
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[nrows*ncols + 3] or pargs[nrows*ncols + 3] 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
local pgn = args.pgn or pargs.pgn
 
size = mw.ustring.match( size, '[%d]+' ) or '26' -- remove px from size
function diagramToFen( frame )
if (pgn) then
function nullOrWhitespace( s ) return not s or s:match( '^%s*(.-)%s*$' ) == '' end
local pgnModule = require('Module:Pgn')
function piece( s )
metadata, moves = pgnModule.main(pgn)
return nullOrWhitespace( s ) and 1
fen = moves[#moves]
or s:gsub( '%s*(%a)(%a)%s*', function( a, b ) return b == 'l' and a:upper() or a end )
end
if (fen) then
align = args.align or pargs.align or 'tright'
local args = frame:getParent().args
clear = args.clear or pargs.clear or ( align:match('tright') and 'right' ) or 'none'
local res = ''
header = args.header or pargs.header or ''
for row = 0, 7 do
footer = args.footer or pargs.footer or ''
for file = 0, 7 do
return chessboard( convertFenToArgs( fen ), size, reverse, letters, numbers, header, footer, align, clear )
res = res .. piece( args[3 + row * 8 + file] )
end
if args[3] then
if row < 7 then res = res .. '/' end
return chessboard(args, size, reverse, letters, numbers, header, footer, align, clear)
end
else
return ( res.gsub('1+', function( s ) return #s end ) )
return chessboard(pargs, size, reverse, letters, numbers, header, footer, align, clear)
end
end
 
return {p
f = function( frame )
local args = frame.args
local t = chessboard( args.fen, args.size or 30, ( args.reverse or '' ):lower() == "true" )
return table.concat( t, "\n" )
end,
['extract fen'] = diagramToFen,
}