Module:Sandbox/Samboy/Chessboard mxn

local p = {}

function chessboard(args, size, rows, cols, rev, trans, lightdark, altprefix, letters, numbers, header, footer, align, clear) function colchar( col ) return (col <= 26) and ( "abcdefghijklmnopqrstuvwxyz" ):sub( col, col ) or ( "abcdefghijklmnopqrstuvwxyz" ):sub( math.floor((col-1)/26), math.floor((col-1)/26) ) .. ( "abcdefghijklmnopqrstuvwxyz" ):sub( col-math.floor((col-1)/26)*26, col-math.floor((col-1)/26)*26) end function image_square( pc, row, col, size, t, flip, altprefix ) local colornames = { l = 'white', d = 'black', u = 'unknown color' } local piecenames = { p = 'pawn', r = 'rook', n = 'knight', b = 'bishop', q = 'queen', k = 'king', a = 'princess', c = 'empress', 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', G = 'giraffe', U = 'unicorn', Z = 'zebra' }       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', db = 'up-right and down-left arrow', dw = 'up-left and down-right 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 = altprefix .. 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 = t and 't' or ((((row + col + flip) % 2) == 0) and 'd' or 'l') return string.format( '', piece, color, ld, size, size, alt, alt ) end

function letters_row( rev, num_lt, num_rt, cols ) local res = '' .. ( num_lt and ' ' or '' ) .. '' for k = 1, cols do           res = res .. colchar(rev and (cols - k + 1) or k) .. ' ' end res = res .. ' ' .. ( num_lt and ' ' or '' ) .. ' '       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 local flip = lightdark and 1 or 0 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 .. '' if ( numbers_lt ) then b = b .. '' .. row .. ' ' 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, trans, flip, altprefix ) local bg = (((trow + tcol + flip) % 2) == 0) and '#ffce9e' or '#d18b47' b = b .. '' .. img .. ' '       end if ( numbers_rt ) then b = b .. '' .. row .. ' ' end b = b .. ' '   end if ( letters_bt ) then b = b .. letters_row(rev, numbers_lt, numbers_rt, cols) .. '\n' end

if footer:match('^%s*$') then else caption = ' ' .. footer .. ' \n' end b = ' '

if noframe then return b   else return ' ' .. ' ' .. header .. ' ' .. '\n \n' .. b .. '\n' .. caption .. ' '    end 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 trans = (args.transparent or pargs.transparent or '' ):lower == "true" local lightdark = (args.lightdark or pargs.lightdark or '' ):lower == "swap" local altprefix = args.altprefix or pargs.altprefix or '' 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 = mw.ustring.gsub( args[2] or pargs[2] or '', '^%s*(.-)%s*$', '%1' ) 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 noframe = (args.noframe or pargs.noframe or ''):lower == "true" 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, trans, lightdark, altprefix, letters, numbers, header, footer, align, clear, noframe) end if args[3] then return chessboard(args, size, rows, cols, reverse, trans, lightdark, altprefix, letters, numbers, header, footer, align, clear, noframe) else return chessboard(pargs, size, rows, cols, reverse, trans, lightdark, altprefix, letters, numbers, header, footer, align, clear, noframe) end end

return p