Module:Ro-headword

local export = {} local pos_functions = {}

local lang = require("Module:languages").getByCode("ro")

-- The main entry point. -- This is the only function that can be invoked from a template. function export.show(frame) local args = frame:getParent.args NAMESPACE = mw.title.getCurrentTitle.nsText PAGENAME = mw.title.getCurrentTitle.text local poscat = frame.args[1] or error("Part of speech has not been specified. Please pass parameter 1 to the module invocation.") local params = { ["head"] = {}, ["suff"] = {type = "boolean"}, ["cat"] = {list = true}, }	if pos_functions[poscat] then for key, val in pairs(pos_functions[poscat].params) do			params[key] = val end end args = require("Module:parameters").process(args, params) local data = {lang = lang, pos_category = args["suff"] and "suffixes" or poscat, categories = args["suff"] and {"Romanian " .. poscat:sub(1, -2) .. "-forming suffixes"} or {}, heads = {args.head}, genders = {}, inflections = {}} local appendix = {} for key, val in ipairs(args.cat) do table.insert(data.categories, "Romanian " .. val) end if pos_functions[poscat] then pos_functions[poscat].func(class, args, data, appendix) end if #appendix == 0 then appendix[1] = "" end return require("Module:headword").full_headword(data) .. appendix[1] end

local function is_equal(a1, a2) if a1 == a2 then return true end if #a1 == #a2 then for i = 1, #a1 do			if a1[i] ~= a2[i] then return false end end return true else return false end end

pos_functions["adjectives"] = { params = { [1] = {},		["f"] = {list = true, allow_holes = true}, ["mp"] = {list = true, allow_holes = true}, ["fp"] = {list = true, allow_holes = true}, ["pl"] = {list = true, allow_holes = true}, ["only"] = {}, },	func = function(class, args, data, appendix) local mode = args[1] local only = {} if args.only then for i, val in ipairs(mw.text.split(args.only, "-")) do				only[val] = true end end if mode == "inv" or mode == "i" or args["f"].maxindex > 0 or args["mp"].maxindex > 0 or args["fp"].maxindex > 0 or args["pl"].maxindex > 0 then require("Module:debug").track("ro-adj new") end if mode == "inv" or mode == "ind" then table.insert(data.genders, "m") table.insert(data.genders, "f") table.insert(data.genders, "n") table.insert(data.inflections, {label = "indeclinable"}) table.insert(data.categories, "Romanian indeclinable adjectives") else local form_order = {"m_sg", "f_sg", "m_pl", "f_pl"} local only_order = {"m", "f", "sg", "pl"} local allowed = {m_sg = true, f_sg = true, m_pl = true, f_pl = true} local only_names = {m = "masculine", f = "feminine", sg = "singular", pl = "plural"} for i, val in pairs(only_names) do				if only[i] then for key, form in pairs(allowed) do						if not key:find(i) then allowed[key] = false end end end end for i, val in ipairs(form_order) do				if allowed[val] then allowed[val] = "lemma" break end end if allowed.m_sg == "lemma" then table.insert(data.genders, "m") if not only.m then table.insert(data.genders, "n") end elseif allowed.f_sg == "lemma" then table.insert(data.genders, "f") elseif allowed.m_pl == "lemma" then table.insert(data.genders, "m-p") elseif allowed.f_pl == "lemma" then table.insert(data.genders, "f-p") if not only.f then table.insert(data.genders, "n-p") end else error("All forms are excluded by the \"only\" parameter") end for i, val in ipairs(only_order) do				if only[val] then table.insert(data.inflections, {label = only_names[val] .. " only"}) end end local f, mp, fp = args["f"], args["pl"].maxindex > 0 and args["pl"] or args["mp"], args["pl"].maxindex > 0 and args["pl"] or args["fp"] require("Module:ro-adjective").make_basic_forms({PAGENAME}, f, mp, fp, mode == "i", only) local unified_sg, unified_pl = false, false if allowed.m_sg and allowed.f_sg and f.maxindex == 1 and f[1] == PAGENAME then unified_sg = true table.insert(data.genders, 2, "f") end if not unified_sg and allowed.f_sg and allowed.f_sg ~= "lemma" then f.label = "feminine" .. (only.sg and "" or " singular") table.insert(data.inflections, f)			end if allowed.m_pl and allowed.f_pl and is_equal(mp, fp) then unified_pl = true mp.label = "plural" table.insert(data.inflections, mp) end if not unified_pl and allowed.m_pl and allowed.m_pl ~= "lemma" then mp.label = (only.m and "" or "masculine") .. " " .. (only.pl and "" or "plural") table.insert(data.inflections, mp) end if not unified_pl and allowed.f_pl and allowed.f_pl ~= "lemma" then fp.label = (only.f and "" or "feminine and neuter") .. " " .. (only.pl and "" or "plural") table.insert(data.inflections, fp) end end return args["head"] end }

pos_functions["determiners"] = pos_functions["adjectives"]

pos_functions["pronouns"] = pos_functions["adjectives"]

pos_functions["articles"] = pos_functions["adjectives"]

pos_functions["nouns"] = { params = { [1] = {list = "g", default = "?"}, [2] = {list = "pl"}, ["f"] = {list = true}, ["m"] = {list = true}, ["sg"] = {}, ["pl=_qual"] = {list = true, allow_holes = true}, ["f=_qual"] = {list = true, allow_holes = true}, ["m=_qual"] = {list = true, allow_holes = true}, ["sort"] = {} },	func = function(class, args, data, appendix) local type for _, val in ipairs(args[1]) do			table.insert(data.genders, val) if val:match("p") then type = "plural" break end end if args["suff"] then data.pos_category = "suffixes" table.insert(data.categories, "Romanian noun-forming suffixes") type = "suffix" end if type == "plural" then if args["sg"] then args[2] = {label = "normally plural"} else args[2] = {label = "plural only"} end table.insert(data.categories, "Romanian pluralia tantum") else if args[2][1] == "-" then args[2] = {label = "uncountable"} if type ~= "suffix" then table.insert(data.categories, "Romanian uncountable nouns") end elseif args[2][1] == "!" then args[2] = {} if type ~= "suffix" then table.insert(data.categories, "Romanian nouns with unattested forms") end else args[2].label = "plural" args[2].accel = {form = "p"} if #args[2] > 0 then for key, val in ipairs(args[2]) do						if ((val == "e" or val == "uri") and args[1][1] == "n") or (val == "i" and args[1][1] == "m") then args[2][key] = PAGENAME .. val end end for key, val in ipairs(args[2]) do						if not mw.title.new(val).exists then table.insert(data.categories, "Romanian nouns with red links in their headword lines") end if args.pl_qual[key] then args[2][key] = {term = args[2][key], qualifiers = {args.pl_qual[key]}} end end if type ~= "suffix" then table.insert(data.categories, "Romanian countable nouns") end else if type ~= "suffix" then args[2].request = true end end end end if args[2].label then table.insert(data.inflections, args[2]) end if args["sg"] then if type == "plural" then table.insert(data.inflections, {label = "singular", args["sg"]}) else error("Parameter \"sg\" can only be used for pluralia tantum") end end if #args["f"] > 0 then if args.f_qual.maxindex > 0 then for key, val in ipairs(args["f"]) do					if args.f_qual[key] then args["f"][key] = {term = args["f"][key], qualifiers = {args.f_qual[key]}} end end end args["f"].label = "feminine equivalent" table.insert(data.inflections, args["f"]) end if #args["m"] > 0 then if args.m_qual.maxindex > 0 then for key, val in ipairs(args["m"]) do					if args.m_qual[key] then args["m"][key] = {term = args["m"][key], qualifiers = {args.m_qual[key]}} end end end args["m"].label = "masculine equivalent" table.insert(data.inflections, args["m"]) end return args["head"] end }

function format_conj(conj) if not conj then return '' else if conj == 1 then return ' 1st conj. ' elseif conj == 2 then return ' 2nd conj. ' elseif conj == 3 then return ' 3rd conj. ' elseif conj == 4 then return ' 4th conj. ' else return ' [conj?] ' end end end

function get_conj(head, ind) local inf, conj = mw.ustring.match(require("Module:links").remove_links(head), '^a (.+)') or error('Head must begin with "a"') inf = mw.ustring.gsub(inf, 'á', 'a') inf = mw.ustring.gsub(inf, 'í', 'i') inf = mw.ustring.gsub(inf, '́', '')		-- remove combining acute if mw.ustring.match(inf, '[iî]$') then conj = 4 elseif mw.ustring.match(inf, 'e$') then conj = 3 elseif mw.ustring.match(inf, 'ea$') then if mw.ustring.match(inf, '[cg]hea$') then conj = 1 elseif ind then if mw.ustring.match(ind, 'ează$') then conj = 1 else conj = 2 end end elseif mw.ustring.match(inf, 'a$') then conj = 1 else error('Unrecognized verb ending') end return conj end

pos_functions["verbs"] = { params = { [1] = {list = "pres"}, [2] = {list = "past"}, ["inf"] = {}, ["conj"] = {type = "number"}, ["pres_qual"] = {list = "pres=_qual", allow_holes = true}, ["past_qual"] = {list = "past=_qual", allow_holes = true} },	func = function(class, args, data, appendix) local multi = false if mw.ustring.match(PAGENAME, ' ') then multi = true end if #args[1] > 0 and args.pres_qual.maxindex > 0 then for key, val in ipairs(args[1]) do				if args.pres_qual[key] then args[1][key] = {term = args[1][key], qualifiers = {args.pres_qual[key]}} end end end if #args[2] > 0 and args.past_qual.maxindex > 0 then for key, val in ipairs(args[2]) do				if args.past_qual[key] then args[2][key] = {term = args[2][key], qualifiers = {args.past_qual[key]}} end end end if args[1][1] == "!" then table.insert(data.inflections, {label = "third-person singular present not attested"}) table.insert(data.categories, 'Romanian verbs with unattested forms') elseif args[1][1] == "-" then table.insert(data.inflections, {label = "third-person singular present not used"}) table.insert(data.categories, "Romanian defective verbs") else args[1].label = 'third-person singular present' args[1].request = true table.insert(data.inflections, args[1]) end if args[2][1] == "!" then table.insert(data.inflections, {label = "past participle not attested"}) table.insert(data.categories, 'Romanian verbs with unattested forms') elseif args[2][1] == "-" then table.insert(data.inflections, {label = "past participle not used"}) table.insert(data.categories, "Romanian defective verbs") else args[2].label = 'past participle' args[2].request = true table.insert(data.inflections, args[2]) end local sc = require("Module:scripts").findBestScript(data.heads[1] or args.inf or PAGENAME, lang):getCode if not data.heads[1] then if not args.inf then if multi then args.inf = mw.text.split(PAGENAME, ' ') args.inf =  .. table.concat(args.inf, ' ') ..  else args.inf = PAGENAME end end if sc == 'Cyrl' then data.heads = {'а ' .. args.inf} --Cyrillic 'а' else data.heads = {'a ' .. args.inf} end end local conj if multi or NAMESPACE ~= '' or sc == 'Cyrl' then conj = args.conj or 0 else conj = get_conj(data.heads[1], args[1][1]) if args.conj and args.conj ~= conj then require("Module:debug").track('ro-verb with wrong conjugation') end end if conj then if conj == 1 then table.insert(data.categories, 'Romanian verbs in 1st conjugation') elseif conj == 2 then table.insert(data.categories, 'Romanian verbs in 2nd conjugation') elseif conj == 3 then table.insert(data.categories, 'Romanian verbs in 3rd conjugation') elseif conj == 4 then table.insert(data.categories, 'Romanian verbs in 4th conjugation') else table.insert(data.categories, 'Requests for inflections in Romanian entries') end end appendix[1] = format_conj(conj) end }

return export