Jump to content

Module:Sandbox/SD0001/Chess3

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by SD0001 (talk | contribs) at 18:40, 26 January 2025 (add buttons for jumping to start/end of the game). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
-- Forked from [[Module:Sandbox/Bawolff/Chessboard]]
----------------------------------------------------

local p = {}

local cfg, nrows, ncols
cfg = {}

-- From Module:Chessboard/Chess
function cfg.dims()
	return 8, 8
end

function cfg.letters()
	return { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h' }
end

function cfg.image_board(size)
	return string.format('[[File:Chessboard480.svg|%dx%dpx|link=|class=notpageimage]]', 8 * size, 8 * size)
end

function cfg.image_square(pc, row, col, size)
	local colornames = { l = 'white', d = 'black', u = 'unknown color' }
	local piecenames = {
		p = 'pawn',
		r = 'rook',
		n = 'knight',
		b = 'bishop',
		q = 'queen',
		k = 'king',
		a = 'archbishop',
		c = 'chancellor',
		z = 'champion',
		w = 'wizard',
		t = 'fool',
		M = 'mann',
		h = 'upside-down pawn',
		m = 'upside-down rook',
		B = 'upside-down bishop',
		N = 'upside-down knight',
		f = 'upside-down king',
		g = 'upside-down queen',
		e = 'elephant',
		s = 'boat',
		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 colchar = { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h' }
	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 .. ' '
	local alt = ''

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

	return string.format('[[File:Chess %s%st45.svg|%dx%dpx|alt=%s|%s|link=|class=notpageimage|top]]', piece, color, size, size, alt, alt)

end

-- End Module:Chessboard/Chess
-- start Module:Chessboard


local function innerboard(definedPieces, pieceInfo, size, rev)
	pattern = cfg.pattern or '%w%w'
	local root = mw.html.create('div')
	root:addClass('chess-pieces notheme')
		:css('position', 'relative')
		:wikitext(cfg.image_board(size))

	for piece in pairs(definedPieces) do
		if piece:match(pattern) then
			local img = cfg.image_square(piece:match(pattern), row, col, size)
			local curPos = pieceInfo[1][piece] or { -1, -1 }
			root:tag('div')
				:css('top', 'calc(var(--calculator-top_piece' .. piece .. ',' .. curPos[1] .. ')*1px)')
				:css('left', 'calc(var(--calculator-left_piece' .. piece .. ',' .. curPos[2] .. ')*1px)')
				:css('z-index', 'min(var(--calculator-top_piece' .. piece .. ',' .. curPos[1] .. '),0)')
				:wikitext(img)
		end
	end

	return tostring(root)
end

local function getPositions(args, size, rev)
	pattern = cfg.pattern or '%w%w'
	local pieceInfo = {}
	local definedPieces = {}

	for moveIndex, move in ipairs(args) do
		pieceInfo[moveIndex] = {}
		local layout = convertFenToArgs(move)
		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 = layout[ncols * (nrows - row) + col + 2] or ''
				local pieceName = piece:match(pattern)
				if pieceName then
					while true do
						-- you can have multiple queens at same time.
						if pieceInfo[moveIndex][pieceName] then
							pieceName = pieceName .. '_'
						else
							break
						end
					end
					definedPieces[pieceName] = true
					pieceInfo[moveIndex][pieceName] = { (trow - 1) * size, (tcol - 1) * size }
				end
			end
		end
	end

	return definedPieces, pieceInfo
end

function chessboard(definedPieces, pieceInfo, size, rev, letters, numbers, header, footer, align, clear)
	function letters_row(rev, num_lt, num_rt)
		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('calculator-container')
				   :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(definedPieces, pieceInfo, 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

local function getPieceMoves(definedPieces, pieceInfo)
	local ret = mw.html.create('div')
	for piece in pairs(definedPieces) do
		local curTop = -1
		local curLeft = -1
		local switchTop = 'switch(move,'
		local switchLeft = 'switch(move,'
		for moveNumber, move in ipairs(pieceInfo) do
			if (move[piece] and move[piece][1] or -1) ~= curTop then
				switchTop = switchTop .. (moveNumber - 1) .. ',' .. curTop .. ','
				curTop = (move[piece] and move[piece][1] or -1)
			end
			if (move[piece] and move[piece][2] or -1) ~= curLeft then
				switchLeft = switchLeft .. (moveNumber - 1) .. ',' .. curLeft .. ','
				curLeft = (move[piece] and move[piece][2] or -1)
			end
		end
		switchTop = switchTop .. curTop .. ')'
		switchLeft = switchLeft .. curLeft .. ')'
		ret:tag('span')
		   :addClass('calculator-field')
		   :attr('id', 'calculator-field-top_piece' .. piece)
		   :attr('data-calculator-type', 'hidden')
		   :css('display', 'none')
		   :attr('data-calculator-formula', switchTop)
		ret:tag('span')
		   :addClass('calculator-field')
		   :attr('id', 'calculator-field-left_piece' .. piece)
		   :attr('data-calculator-type', 'hidden')
		   :css('display', 'none')
		   :attr('data-calculator-formula', switchLeft)
	end
	return tostring(ret)
end

function p.board(frame)
	local args = frame.args
	local pargs = frame:getParent().args
	nrows, ncols = cfg.dims()

	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 pgn = args.pgn or pargs.pgn
	local moves, metadata
	local definedPieces, pieceInfo

	size = mw.ustring.match(size, '[%d]+') or '26' -- remove px from size

	assert(pgn, "pgn argument required")
	local pgnModule = require('Module:Pgn')
	metadata, moves = pgnModule.main(pgn)
	definedPieces, pieceInfo = getPositions(moves, size, reverse)

	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 ''
	if args.animate then
		align = align .. ' animate'
		footer = footer .. frame:extensionTag('templatestyles', '', { src = 'Module:Sandbox/SD0001/Chess3/animations.css' })
	end
	header = header .. frame:preprocess('{{calculator button|contents={{!}}←|for=move|type=default|formula=1}}'
		.. '{{calculator button|contents=←|for=move|type=default|formula=max(1, move-1)}}'
		.. '{{calculator button|id=next|contents=→|for=move|type=default|formula=min(' .. (#moves) .. ', move+1)}}'
		.. '{{calculator button|contents=→{{!}}|for=move|type=default|formula=' .. (#moves) .. '}}'
		.. '{{calculator|type=hidden|default=1|id=move}}'
	)
	footer = footer .. getPieceMoves(definedPieces, pieceInfo)
	return chessboard(definedPieces, pieceInfo, size, reverse, letters, numbers, header, footer, align, clear)
end

return p