Module:Gapnum

Documentation for this module may be created at Module:Gapnum/doc

local p = {}

local getArgs

function p.main(frame)
	if not getArgs then
		getArgs = require('Module:Arguments').getArgs
	end
	local args = getArgs(frame, {wrappers = 'Template:Gapnum'})
	local n = args[1]
	if not n then
		error('Parameter 1 is required')
	end
	local gap =  args.gap
	local precision = tonumber(args.prec)

	return p._gaps(n,{gap=gap,prec=precision})
end

function p._gaps(n,tbl)
	local nstr = tostring(n)
	local gap = tbl.gap or '.25em'
	local precision = tbl.prec or -1

	local decimalloc = nstr:find('.', 1, true)
	local int_part, frac_part
	if decimalloc == nil then
		int_part = nstr
	else
		int_part = nstr:sub(1, decimalloc-1)
		frac_part = nstr:sub(decimalloc + 1)
	end
	local ret = mw.html.create('span')
							:css('white-space','nowrap')
	local int_string = {}
	-- Loop to handle most of the groupings; from right to left, so that if a group has less than 3 members, it will be the first group
	while int_part:len() > 3 do
		-- Insert in first spot, since we're moving backwards
		table.insert(int_string,1,int_part:sub(-3))
		int_part = int_part:sub(1,-4)
	end
	-- handle any left over numbers; add directly
	ret:wikitext(int_part)
	-- Build intstring
	for _, v in ipairs(int_string) do
		ret:tag('span')
				:css('margin-left',gap)
				:wikitext(v)
			:done()
	end

	if precision ~= 0 and frac_part then
		ret:wikitext('.')
		if precision == -1 then
			precision = frac_part:len()
		end
		-- Reduce the length of the string if required precision is less than actual precision
		-- OR
		-- Increase it (by adding 0s) if the required precision is more than actual
		local offset = precision - frac_part:len()
		if offset < 0 then
			frac_part = frac_part:sub(1,precision)
		elseif offset > 0 then
			frac_part = frac_part .. string.rep('0', offset)
		end
		-- The first group after the decimal shouldn't have a gap between the decimal
		ret:wikitext(frac_part:sub(1,3))
		frac_part = frac_part:sub(4)
		local last_span
		-- Allow groups of 3 or 2 (3 first)
		for v in string.gmatch(frac_part,'%d%d%d?') do
			last_span = ret:tag('span')
					:css('margin-left',gap)
					:wikitext(v)
		end
		-- Preference for groups of 4 instead of groups of 1 at the end
		if #frac_part % 3 == 1 then
			if last_span then
				last_span:wikitext(frac_part:sub(-1))
			else
				ret:wikitext(frac_part)
			end
		end
	end

	return ret
end

return p