模組:AbQ Pie

模块文档[创建]

require('Module:No globals')

local getArgs = require("Module:Arguments").getArgs
local yesno = require("Module:Yesno")
local p = {}

local function list(args)
    local _args = {}
	
    local args = args
    if type(args) == 'nil' then
        args = {}
    elseif type(args) == 'string' then
        args = { args }
    end

    for _, v in ipairs(args) do
        table.insert(_args, v)
    end
	
    return '[ ' .. table.concat(_args, ', ') .. ' ]'
end

local function dict(args)
    local _args = {}
	
    local args = args
    if type(args) == 'nil' then
        args = {}
    elseif type(args) == 'string' then
        args = { args }
    end

    for i, v in pairs(args) do
        table.insert(_args, string.format('"%s" : %s', i, v))
    end
	
    return '{ ' .. table.concat(_args, ', ') .. ' }'
end

local function str(args)
    return '"' .. (args or '') .. '"'
end

local function get_page_count_by_category(category_name)
    return mw.getCurrentFrame():callParserFunction {
        name = 'PAGESINCATEGORY',
        args = { category_name, 'pages', 'R' }
    }
end

local function get_data_by_class(args)
    
    -- Class information
        
    local default_classes = {'fa', 'fl', 'a', 'ga', 'b', 'c', 'start', 'stub', 'list', 'unassessed'}

    local class_list = {
        ['fa'] = { ['name'] = { ['hans'] = '典范级', ['hant'] = '典範級' }, ['color'] = '#9CBDFF' },
        ['fl'] = { ['name'] = { ['hans'] = '特色列表级', ['hant'] = '特色列表級' }, ['color'] = '#9CBDFF' },
        ['a'] = { ['name'] = { ['hans'] = '甲级', ['hant'] = '甲級' }, ['color'] = '#66FFFF' },
        ['al'] = { ['name'] = { ['hans'] = '甲级列表级', ['hant'] = '甲級列表級' }, ['color'] = '#66FFFF' },
        ['ga'] = { ['name'] = { ['hans'] = '优良级', ['hant'] = '優良級' }, ['color'] = '#66FF66' },
        ['b'] = { ['name'] = { ['hans'] = '乙级', ['hant'] = '乙級' }, ['color'] = '#B2FF66' },
        ['bl'] = { ['name'] = { ['hans'] = '乙级列表级', ['hant'] = '乙級列表級' }, ['color'] = '#B2FF66' },
        ['c'] = { ['name'] = { ['hans'] = '丙级', ['hant'] = '丙級' }, ['color'] = '#FFFF66' },
        ['cl'] = { ['name'] = { ['hans'] = '丙级列表级', ['hant'] = '丙級列表級' }, ['color'] = '#FFFF66' },
        ['start'] = { ['name'] = { ['hans'] = '初级', ['hant'] = '初級' }, ['color'] = '#FFAA66' },
        ['list'] = { ['name'] = { ['hans'] = '列表级', ['hant'] = '列表級' }, ['color'] = '#C7B1FF' },
        ['stub'] = { ['name'] = { ['hans'] = '小作品级', ['hant'] = '小作品級' }, ['color'] = '#FFA4A4' },
        ['sl'] = { ['name'] = { ['hans'] = '小列表级', ['hant'] = '小列表級' }, ['color'] = '#FFA4A4' },
        ['unassessed'] = { ['name'] = { ['hans'] = '未评级', ['hant'] = '未評級' }, ['color'] = '#BEBEBE' },
    }
    
    if args.customclass then
        local custom_classes = mw.text.split(args.customclass:gsub('%s', ''), ';')
        for _, v in ipairs(custom_classes) do
            local tab = mw.text.split(v:lower():gsub('%s', ''), ':'), "", "", ""
            class_list[tab[1]] = {
                ['name'] = { ['hans'] = tab[2], ['hant'] = tab[3] },
                ['color'] = tab[4]
            }
        end
    end
    

    local ret = {}
    
    local topic = args.topic or ''
    
    local classes = {}
    if args.class then
        classes = mw.text.split(args.class:lower():gsub('%s', ''), ',')
    else
        classes = default_classes
    end
    
    local counts = {}
    if args.count then
        counts = mw.text.split(args.count:gsub('%s', ''), ',')
    end

    for i, v in ipairs(classes) do
        local info = class_list[v]
        local count, cat
        if #counts == 0 then
            cat = info.name.hans .. topic .. '条目'
            count = get_page_count_by_category(cat)
        else
            count = counts[i] or '0'
        end
        if yesno(args.hidezero) and count == '0' then
            -- pass
        else
            table.insert(ret, {
                ['name'] = info.name,
                ['color'] = info.color,
                ['count'] = count,
            })
        end
    end

    return ret
end

local function build_graph(title, height, width, pie_radius, data, variant)
    local temp_tab = {}

    -- graph
    local graph = {
        ['version'] = 2,
        ['axes'] = list(),
        ['height'] = height or '300',
        ['width'] = width or '300',
        ['padding'] = str('padding'),
    }

    -- graph['legends']
    graph['legends'] = list(dict{
        ['fill'] = str('color'),
        ['offset'] = '20',
        ['properties'] = dict(),
        ['title'] = str(title or ''),
    })

    -- graph['marks']
    graph['marks'] = list(dict{
        ['type'] = str('arc'),
        ['from'] = dict{
            ['data'] = str('table'),
            ['transform'] = list(dict{
                ['type'] = str('pie'),
                ['field'] = str('val'),
            }),
        },
        ['properties'] = dict{
            ['enter'] = dict{
                ['x'] = dict{
                    ['field'] = dict{ ['group'] = str('width'), },
                    ['mult'] = '0.5',
                },
                ['y'] = dict{
                    ['field'] = dict{ ['group'] = str('width'), },
                    ['mult'] = '0.5',
                },
                ['startAngle'] = dict{ ['field'] = str('layout_start'), },
                ['endAngle'] = dict{ ['field'] = str('layout_end'), },
                ['fill'] = dict{
                    ['field'] = str('idx'),
                    ['scale'] = str('color'),
                },
                ['innerRadius'] = dict{ ['value'] = '0', },
                ['outerRadius'] = dict{ ['value'] = pie_radius or '125', },
            },
        },
    })

    -- graph['data']
	local variant = (variant ~= 'hant' and 'hans' or 'hant')
    for _, v in ipairs(data) do
        table.insert(temp_tab, dict{
            ['col'] = str('data'),
            ['idx'] = str(v.name[variant] .. '：' .. mw.language.new('en'):formatNum(tonumber(v.count))),
            ['val'] = v.count,
        })
    end
    graph['data'] = list(dict{
        ['name'] = str('table'),
        ['values'] = list(temp_tab),
    })
    temp_tab = {}

    -- graph['scales']
    for _, v in ipairs(data) do
        table.insert(temp_tab, str(v.color))
    end
    graph['scales'] = list(dict{
        ['domain'] = dict{
            ['data'] = str('table'),
            ['field'] = str('idx'),
        },
        ['name'] = str('color'),
        ['type'] = str('ordinal'),
        ['range'] = list(temp_tab),
    })
    temp_tab = {}

    return mw.getCurrentFrame():extensionTag {
        ['name'] = 'graph',
        ['content'] = dict(graph),
    }
end

function p.main(frame)
    local args = getArgs(frame)
    return p._main(args)
end

function p._main(args)
    local hans_title = (args['topic-hans'] or args['topic'] .. '专题') .. '条目品质统计'
    local hant_title = (args['topic-hant'] or args['topic'] .. '專題') .. '條目品質統計'

    local height, width, radius = args.height, args.width, args.radius
    local data = get_data_by_class(args)

    return require( 'Module:WikitextLC' ).selective { 
        ['zh-hans'] = build_graph(hans_title, height, width, radius, data, 'hans'),
        ['zh-hant'] = build_graph(hant_title, height, width, radius, data, 'hant'),
    }
end

return p