Module:Chessboard mxn

This is an old revision of this page, as edited by Frietjes (talk | contribs) at 16:02, 28 May 2013 (Created page with 'local p = {} function chessboard(args, size, rows, cols, rev, letters, numbers, header, footer, align, clear) function colchar( col ) return ( "abcdefghijkl...'). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
(diff) ← Previous revision | Latest revision (diff) | Newer revision → (diff)
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">'
        if ( num_lt ) then
            res = res .. '<td></td>'
        end
        res = res .. '<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>'
        if ( num_rt ) then
            res= res .. '<td></td>'
        end
        res = res .. '</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 = 8*size + 6
    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" style="margin-left:4px;margin-right:4px;">' .. footer .. '</div>\n'
    end
    
    return '<div class="thumb ' .. align .. '" style="clear:' .. clear .. '; width:' .. width .. 'px; text-align:center">'
        .. header .. '\n<div class="thumbinner" style="width:', 'px; text-align:center">\n' 
        .. '<table cellpadding=0 cellspacing=0 style="background:white; font-size:88%; border:1px #b0b0b0 solid; padding:0; '
        .. 'margin:2px 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 '26px'
    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[67] or pargs[67]) 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 rows = (args.rows or pargs.rows) or 8
    local cols = (args.cols or pargs.cols) or 8
    local fen = args.fen or pargs.fen
        
    size = mw.ustring.match(size, '%d[%d]*') or '26' -- remove px from size
    if (fen) then
        align = (args.align or pargs.align) or 'tright'
        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)
    else
        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
    
end

return p