Module:Chessboard/Chu shogi

local p = {}

local function image_square( pc, row, col, size )
	local colornames = {
		l = { m = 'white', f = 'white' },
		d = { m = 'black', f = 'black' }
	}
	local piecenames = {
		am = { name = 'angle mover', gender = 'm' },
		bt = { name = 'blind tiger', gender = 'm' },
		cg = { name = 'copper general', gender = 'm' },
		dh = { name = 'dragon horse', gender = 'm' },
		dk = { name = 'dragon king', gender = 'm' },
		de = { name = 'drunk elephant', gender = 'm' },
		fl = { name = 'ferocious leopard', gender = 'm' },
		fc = { name = 'flying chariot', gender = 'm' },
		fo = { name = 'flying ox', gender = 'm' },
		fs = { name = 'flying stag', gender = 'm' },
		fb = { name = 'free boar', gender = 'm' },
		fk = { name = 'free king', gender = 'm' },
		pn = { name = 'foot soldier', gender = 'm' },
		gb = { name = 'go-between', gender = 'm' },
		gg = { name = 'golden general', gender = 'm' },
		hf = { name = 'horned falcon', gender = 'm' },
		ic = { name = 'incense chariot', gender = 'm' },
		jd = { name = 'jade general', gender = 'm' },
		kg = { name = 'king general', gender = 'm' },
		kn = { name = 'kirin', gender = 'm' },
		ln = { name = 'lion', gender = 'm' },
		px = { name = 'phoenix', gender = 'm' },
		pr = { name = 'prince', gender = 'm' },
		rc = { name = 'reverse chariot', gender = 'm' },
		sm = { name = 'side mover', gender = 'm' },
		sg = { name = 'silver general', gender = 'm' },
		se = { name = 'soaring eagle', gender = 'm' },
		vm = { name = 'vertical mover', gender = 'm' },
		we = { name = 'whale', gender = 'm' },
		wh = { name = 'white horse', gender = 'm' }
	}
	local symnames = {
		xx = 'black cross',
		ox = 'white cross',
		xo = 'black circle',
		oo = 'white circle',
		ul = 'arrow up-left',
		ua = 'arrow up',
		ur = 'arrow up-right',
		la = 'arrow left',
		ra = 'arrow right',
		dl = 'arrow down-left',
		da = 'arrow down',
		dr = 'arrow down-right',
		lr = 'arrow left-right',
		ud = 'arrow up-down',
		x0 = 'zero',
		x1 = 'one',
		x2 = 'two',
		x3 = 'three',
		x4 = 'four',
		x5 = 'five',
		x6 = 'six',
		x7 = 'seven',
		x8 = 'eight',
		x9 = 'nine',
	}
	local colchar = {'A','B','C','D','E','F','G','H','I','J','K','L'}
    local color = string.sub(pc, 3, 1) or ''
    local piece = mw.ustring.gsub( pc, '^.*(%w)(%w)(%w).*$', '%1' ) or ''
    local alt = colchar[col] .. row .. ' '

    if colornames[color] and piecenames[piece] then
		alt = alt .. colornames[color][piecenames[piece]['gender']] .. ' ' .. piecenames[piece]['name']
    else
		alt = alt .. ( symnames[piece .. color] or piece .. ' ' .. color )
	end

	fn = ''

	if pc == 'aml' then
		fn = 'Angle_mover_(Chu_Shogi)'
	elseif pc == 'amd' then
		fn = 'Angle_mover_(Chu_Shogi)'
	elseif pc == 'btl' then
		fn = 'Blind_Tiger_(Chu_Shogi)'
	elseif pc == 'btd' then
		fn = 'Blind_Tiger_(Chu_Shogi)'
	elseif pc == 'cgl' then
		fn = 'Copper_General_(Chu_Shogi)'
	elseif pc == 'cgd' then
		fn = 'Copper_General_(Chu_Shogi)'
	elseif pc == 'dhl' then
		fn = 'Dragon_Horse_(Chu_Shogi)'
	elseif pc == 'dhd' then
		fn = 'Dragon_Horse_(Chu_Shogi)'
	elseif pc == 'dkl' then
		fn = 'Dragon_King_(Chu_Shogi)'
	elseif pc == 'dkd' then
		fn = 'Dragon_King_(Chu_Shogi)'
	elseif pc == 'del' then
		fn = 'Drunk_Elephant_(Chu_Shogi)'
	elseif pc == 'ded' then
		fn = 'Drunk_Elephant_(Chu_Shogi)'
	elseif pc == 'fll' then
		fn = 'Ferocious_Leopard_(Chu_Shogi)'
	elseif pc == 'fld' then
		fn = 'Ferocious_Leopard_(Chu_Shogi)'
	elseif pc == 'fcl' then
		fn = 'Flying_Chariot_(Chu_Shogi)'
	elseif pc == 'fcd' then
		fn = 'Flying_Chariot_(Chu_Shogi)'
	elseif pc == 'fol' then
		fn = 'Flying_Ox_(Chu_Shogi)'
	elseif pc == 'fod' then
		fn = 'Flying_Ox_(Chu_Shogi)'
	elseif pc == 'fsl' then
		fn = 'Flying_Stag_(Chu_Shogi)'
	elseif pc == 'fsd' then
		fn = 'Flying_Stag_(Chu_Shogi)'
	elseif pc == 'pnl' then
		fn = 'Foot_Soldier_(Chu_Shogi)'
	elseif pc == 'pnd' then
		fn = 'Foot_Soldier_(Chu_Shogi)'
	elseif pc == 'fbl' then
		fn = 'Free_Boar_(Chu_Shogi)'
	elseif pc == 'fbd' then
		fn = 'Free_Boar_(Chu_Shogi)'
	elseif pc == 'fkl' then
		fn = 'Free_King_(Chu_Shogi)'
	elseif pc == 'fkd' then
		fn = 'Free_King_(Chu_Shogi)'
	elseif pc == 'gbl' then
		fn = 'Go-Between_(Chu_Shogi)'
	elseif pc == 'gbd' then
		fn = 'Go-Between_(Chu_Shogi)'
	elseif pc == 'ggl' then
		fn = 'Golden_General_(Chu_Shogi)'
	elseif pc == 'ggd' then
		fn = 'Golden_General_(Chu_Shogi)'
	elseif pc == 'hfl' then
		fn = 'Horned_Falcon_(Chu_Shogi)'
	elseif pc == 'hfd' then
		fn = 'Horned_Falcon_(Chu_Shogi)'
	elseif pc == 'icl' then
		fn = 'Incense_Chariot_(Chu_Shogi)'
	elseif pc == 'icd' then
		fn = 'Incense_Chariot_(Chu_Shogi)'
	elseif pc == 'jdl' then
		fn = 'Jade_General_(Chu_Shogi)'
	elseif pc == 'jdd' then
		fn = 'Jade_General_(Chu_Shogi)'
	elseif pc == 'kgl' then
		fn = 'King_General_(Chu_Shogi)'
	elseif pc == 'kgd' then
		fn = 'King_General_(Chu_Shogi)'
	elseif pc == 'knl' then
		fn = 'Kirin_(Chu_Shogi)'
	elseif pc == 'knd' then
		fn = 'Kirin_(Chu_Shogi)'
	elseif pc == 'lnl' then
		fn = 'Lion_(Chu_Shogi)'
	elseif pc == 'lnd' then
		fn = 'Lion_(Chu_Shogi)'
	elseif pc == 'pxl' then
		fn = 'Phoenix_(Chu_Shogi)'
	elseif pc == 'pxd' then
		fn = 'Phoenix_(Chu_Shogi)'
	elseif pc == 'prl' then
		fn = 'Prince_(Chu_Shogi)'
	elseif pc == 'prd' then
		fn = 'Prince_(Chu_Shogi)'
	elseif pc == 'rcl' then
		fn = 'Reverse_Chariot_(Chu_Shogi)'
	elseif pc == 'rcd' then
		fn = 'Reverse_Chariot_(Chu_Shogi)'
	elseif pc == 'sml' then
		fn = 'Side_Mover_(Chu_Shogi)'
	elseif pc == 'smd' then
		fn = 'Side_Mover_(Chu_Shogi)'
	elseif pc == 'sgl' then
		fn = 'Silver_General_(Chu_Shogi)'
	elseif pc == 'sgd' then
		fn = 'Silver_General_(Chu_Shogi)'
	elseif pc == 'sel' then
		fn = 'Soaring_Eagle_(Chu_Shogi)'
	elseif pc == 'sed' then
		fn = 'Soaring_Eagle_(Chu_Shogi)'
	elseif pc == 'vml' then
		fn = 'Vertical_Mover_(Chu_Shogi)'
	elseif pc == 'vmd' then
		fn = 'Vertical_Mover_(Chu_Shogi)'
	elseif pc == 'wel' then
		fn = 'Whale_(Chu_Shogi)'
	elseif pc == 'wed' then
		fn = 'Whale_(Chu_Shogi)'
	elseif pc == 'whl' then
		fn = 'White_Horse_(Chu_Shogi)'
	elseif pc == 'whd' then
		fn = 'White_Horse_(Chu_Shogi)'
	end

	return string.format( '[[File:%s.svg|%dx%dpx|alt=%s|%s]]', fn, size, size, pc, pc )

end

local function innerboard(args, size, rev)
	local root = mw.html.create('div')
	root:addClass('makruk-board')
		:css('position', 'relative')
		:wikitext(string.format( '[[File:ChuShogiboard.svg|%dx%dpx|link=]]', 12 * size, 12 * size ))

    for trow = 1,12 do
        local row = rev and trow or ( 13 - trow )
        for tcol = 1,12 do
            local col = rev and ( 13 - tcol ) or tcol
            local piece = args[12 * ( 12 - row ) + col + 2] or ''
            if piece:match( '%w%w' ) then
               local img = image_square(piece:match('%w%w'), row, col, size )
               root:tag('div')
               		:css('position', 'absolute')
               		:css('z-index', '3')
               		:css('top', tostring(( trow - 1 ) * size) .. 'px')
               		:css('left', tostring(( tcol - 1 ) * size) .. 'px')
               		:css('width', size .. 'px')
               		:css('height', size .. 'px')
               		:wikitext(img)
            end
        end
    end

    return tostring(root)
end

function ChuShogiboard(args, size, rev, letters, numbers, header, footer, align, clear)
    function letters_row( rev, num_lt, num_rt )
        local letters = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L'}
        local root = mw.html.create('')
        if num_lt then
        	root:tag('td')
        		:css('vertical-align', 'inherit')
        		:css('padding', '0')
        end
        for k = 1,12 do
        	root:tag('td')
        		:css('padding', '0')
        		:css('vartical-align', 'inherit')
        		:css('height', '18px')
        		:css('width', size .. 'px')
        		:wikitext(rev and letters[13-k] or letters[k])
        end
        if num_rt then
        	root:tag('td')
        		:css('vertical-align', 'inherit')
        		:css('padding', '0')
        end
        return tostring(root)
    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 = 12 * size + 2
    if ( numbers_lt ) then width = width + 18 end
    if ( numbers_rt ) then width = width + 18 end

    local root = mw.html.create('div')
    	:addClass('thumb')
    	:addClass(align)
    	:css('clear', clear)
    	:css('text-align', 'center')
    	:wikitext(header)
    local div = root:tag('div')
    	:addClass('thumbinner')
    	:css('width', width .. 'px')
    local b = div:tag('table')
    	:attr('cellpadding', '0')
    	:attr('cellspacing', '0')
    	:css('background', 'white')
    	:css('font-size', '88%')
    	:css('border' , '1px #b0b0b0 solid')
    	:css('padding', '0')
    	:css('margin', 'auto')

    if ( letters_tp ) then
        b:tag('tr')
        	:css('vertical-align', 'middle')
        	:wikitext(letters_row( rev, numbers_lt, numbers_rt ))
    end
    local tablerow = b:tag('tr'):css('vertical-align','middle')
    if ( numbers_lt ) then
    	tablerow:tag('td')
    		:css('padding', '0')
    		:css('vertical-align', 'inherit')
    		:css('width', '18px')
    		:css('height', size .. 'px')
    		:wikitext(rev and 1 or 12)
    end
    local td = tablerow:tag('td')
    	:attr('colspan', 12)
    	:attr('rowspan', 12)
    	:css('padding', '0')
    	:css('vertical-align', 'inherit')
    	:wikitext(innerboard(args, size, rev))

    if ( numbers_rt ) then
    	tablerow:tag('td')
    		:css('padding', '0')
    		:css('vertical-align', 'inherit')
    		:css('width', '18px')
    		:css('height', size .. 'px')
    		:wikitext(rev and 1 or 12)
    end
    if ( numbers_lt or numbers_rt ) then
       for trow = 2, 12 do
          local idx = rev and trow or ( 13 - trow )
          tablerow = b:tag('tr')
          	:css('vertical-align', 'middle')
          if ( numbers_lt ) then
          	tablerow:tag('td')
          		:css('padding', '0')
          		:css('vertical-align', 'inherit')
          		:css('height', size .. 'px')
          		:wikitext(idx)
          end
          if ( numbers_rt ) then
          	tablerow:tag('td')
          		:css('padding', '0')
          		:css('vertical-align', 'inherit')
          		:css('height', size .. 'px')
          		:wikitext(idx)
          end
       end
    end
    if ( letters_bt ) then
        b:tag('tr')
        	:css('vertical-align', 'middle')
        	:wikitext(letters_row( rev, numbers_lt, numbers_rt ))
    end

    if (footer and footer ~= '') then
		div:tag('div')
			:addClass('thumbcaption')
			:wikitext(footer)
	end

    return tostring(root)
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 )
    function nullOrWhitespace( s ) return not s or s:match( '^%s*(.-)%s*$' ) == '' end
    function piece( s )
        return nullOrWhitespace( s ) and 1
        or s:gsub( '%s*(%a)(%a)%s*', function( a, b ) return b == 'l' and a:upper() or a end )
    end

    local res = ''
    offset = offset or 0
    for row = 1, 12 do
        for file = 1, 12 do
            res = res .. piece( args[12*(row - 1) + file + offset] )
        end
        if row < 12 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
    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 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 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'
        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 ChuShogiboard( convertFenToArgs( fen ), size, reverse, letters, numbers, header, footer, align, clear )
    end
    if args[3] then
        return ChuShogiboard(args, size, reverse, letters, numbers, header, footer, align, clear)
    else
        return {pargs=pargs, size=size, reverse=reverse, letters=letters, numbers=numbers, header=header, footer=footer, align=align, clear=clear}
        --return ChuShogiboard(pargs, size, reverse, letters, numbers, header, footer, align, clear)
    end
end

function p.fen2ascii(frame)
    -- {{#invoke:ChuShogiboard|fen2ascii|fen=...}}
    local b = convertFenToArgs( frame.args.fen )
    local res = '|=\n'
    local offset = 2
    for row = 1,12 do
        local n = (13 - row)
        res = res .. n .. ' |' ..
            table.concat(b, '|', 12*(row-1) + 1 + offset, 12*(row-1) + 12 + offset) .. '|=\n'
    end
    res = mw.ustring.gsub( res,'\| \|', '|  |' )
    res = mw.ustring.gsub( res,'\| \|', '|  |' )
    res = res .. '   A  B  C  D  E  F  G  H  I  J  K  L'
    return res
end

function p.ascii2fen( frame )
    -- {{#invoke:ChuShogiboard|ascii2fen|kl| | |....}}
    return convertArgsToFen( frame.args, frame.args.offset or 1 )
end

return p
Usage
