Module:Test

From RimWorld Wiki
Revision as of 22:37, 13 April 2021 by Dr. Strangelove (talk | contribs)
Jump to navigation Jump to search

This module is used for development.

Purpose

This module is used to query information from the uploaded and parsed game files.

Its main purpose is to populate the infoboxes.

Usage

A note on the order of named parameters. All of the parameters that look like ...=... are called named parameters and their order is not important (this is true for all templates).

query

{{#invoke:Test|query|<def ID>[|...|][|tag|][|sibling=...]}}

The work-horse. Output varies based on use:

If only the <def ID> parameter is set, it will show the whole Def in the log.
If simple values are queried it will return them.
If lists are queried it will return nothing but call {{#vardefine}} on all the simple values within it. What got defined can be seen in the page's log.

Named parameters:

<def ID>
This parameter identifies the Def so it is mandatory. It can take two forms, if both are defined then defName takes preference.
defName=<defName>
<defName> (case sensitive) should be replaced with the actual defName of a Def.
label=<label>
<label> (case insensitive) should be replaced with the actual label of a Def.
[sibling=...] (optional) (case sensitive)
Allows querying for something if we know its sibling's value (works only for values at the moment).

Anonymous parameters:

[|...|] (optional) (case sensitive)
Anonymous paramaters before the last one ([tag]) are here to help uniquely identify it. If the [tag] is already unique within a Def tree, then these additional parameters are not needed.
[|tag|] (optional) (case sensitive)
The final anonymous parameter defines what is to be queried.

count

{{#invoke:Test|count|<def ID>[|...|][|tag|][|sibling=...]}}

Parameters are the same as for query. It's basically a wrapped up query that behaves a bit differently.

The difference is in how it handles lists. If a list is queried, unlike query, it will return the length of the list.

How-to

Take a look at a Def

{{#invoke:Test|query|label=desert}}

Data is in the log.

Retrieve a simple value

{{#invoke:Test|query|defName=Caribou|description}}

Dealing with lists

{{#invoke:Test|query|defName=Mech_Scyther|tools}}

When a list is retrieved there will be no output but the log will contain a list of defined variables.

For convenience the list is reprinted here:

tools_1_linkedBodyPartsGroup = LeftBlade
tools_1_cooldownTime = 2
tools_1_label = left blade
tools_1_DPS = 10
tools_1_power = 20
tools_1_capacities_1 = Cut
tools_1_capacities_2 = Stab
tools_2_linkedBodyPartsGroup = RightBlade
tools_2_cooldownTime = 2
tools_2_label = right blade
tools_2_DPS = 10
tools_2_power = 20
tools_2_capacities_1 = Cut
tools_2_capacities_2 = Stab
tools_3_linkedBodyPartsGroup = HeadAttackTool
tools_3_capacities_1 = Blunt
tools_3_label = head
tools_3_DPS = 4.5
tools_3_chanceFactor = 0.2
tools_3_power = 9
tools_3_cooldownTime = 2

All of the above can be accessed with the use of {{#var:...}}.

{{#var:tools_1_DPS}}

DPS is not a normal member of this table but has been added with Lua. Let's call it a virtual field.

Retrieve something if a sibling is known

{{#invoke:Test|query|label=guinea pig|minAge|sibling=AnimalAdult}}


ENV = "wiki" -- "dev" or "wiki"
INFO = true

mw = mw or {}
if ENV == "dev" then log = print
elseif ENV == "wiki" then log = mw.log end
---------------
-- load data --
---------------

data = {}

if ENV == "dev" then
  data["Biomes"] = loadfile("./data/BiomeDefs.lua")()
  data["Races"] = loadfile("./data/ThingDefs_Races.lua")()
elseif ENV == "wiki" then
  data["Biomes"] = mw.loadData('Module:Test/data/biomes')
  data["Races"] = mw.loadData('Module:Test/data/races')
end

------------------
-- virtual keys --
------------------

-- this could be implemented with metatable events
-- they get added in get(id_pair)

local virtual_keys = {
  ["Races"] = {
    ["lives_in"] = function (race, biomes)
      local list = {}
      for biome_key, biome in pairs(biomes) do
        for _,animal in ipairs(biome.wildAnimals) do
          if race.defName == animal then
            table.insert(list, biome_key)
          end
        end
      end
      return list
    end
  }
}

--------------------
-- load libraries --
--------------------

if ENV == "dev" then
  inspect = require './lib/inspect'
  util = require("./lib/util")
  search = require("./lib/search")

  function pinspect(tbl, title)
    if INFO then
      util.hl(title)
      print(inspect(tbl))
    end
  end
end

if ENV == "wiki" then
  INFO = false
  util = require("Module:Test/lib/util")
  search = require("Module:Test/lib/search")
  function pinspect() end
end

-------------
-- private --
-------------

local function vardefine(name, value)
  local f_name = "vardefine"
  assert(var_name, string.format("bad argument #1 to '%s' (argument missing, name of variable to define)", f_name))
  assert(var_name == "string", string.format("bad argument #1 to '%s' (string expected, got %s)", f_name, type(var_name)))
  assert(var_value, string.format("bad argument #2 to '%s' (argument missing, value to assign to variable)", f_name))
  assert(var_value == "string" or var_value == "number", string.format("bad argument #2 to '%s' (string or number expected, got %s)", f_name, type(var_value)))

  frame:callParserFunction('#vardefine', var_name, var_value)
end


local function search_parent_def_table(key, def_table)
  local ParentName = getParentName(def_table)
  if not ParentName then return nil end
  local parentdef_table = search_table_recursive(ParentName, data)
  if not parentdef_table then return nil end

  local found = search_table_recursive(key, parentdef_table)
  if found then return found
  else
    found = search_parent_def_table(key, parentdef_table)
    if found then return found end
  end
end


local function merge_def(base_def_table, def_category, ignore_keys)

  local ancestors = {}
  local parent_name = base_def_table["ParentName"]
  local parent_table = data[def_category][parent_name]

  while parent_name do
    table.insert(ancestors, parent_name)
    parent_name = parent_table["ParentName"]
    parent_table = data[def_category][parent_name]
  end

  local inheritance_chain = util.shallowcopy(util.reverse_numeric_table(ancestors))
  table.insert(inheritance_chain, base_def_table.defName)

  local merged = {}
  for i,v in ipairs(inheritance_chain) do
    util.overwrite_first_table_with_second(merged, data[def_category][inheritance_chain[i]], ignore_keys)
  end

  return merged
end


local function get(id_pair)
  local res = search.conductor(id_pair, data)
  if not res then
    log("Def not found")
    return nil
  end

  local base_def_table = res.parent.value
  local def = merge_def(base_def_table, res.ancestors[2], {"ParentName", "Abstract"})

  -- add virtual keys
  if virtual_keys[res.ancestors[2]] then
    def._virtual = {}
    for k,func in pairs(virtual_keys[res.ancestors[2]]) do
      def._virtual[k] = func(def, data.Biomes)
    end
  end

  return def
end

-------------------------------------------------------------------
-- simulate MediaWiki/Scribunto module invocation using 'frame'  --
-------------------------------------------------------------------

local simframe = { ["args"] = {} }

simframe.args[1] = "Fox_Fennec"
--~ simframe.args[2] = "4"
--~ simframe.args[3] = "1"

frame = frame or simframe

----------------------
-- public & related --
----------------------

-- frame.args are always strings on MediaWiki so convert them back to numbers
pinspect(frame.args, "frame.args before")
for k,arg in pairs(frame.args) do
  frame.args[k] = tonumber(arg) or arg
end
pinspect(frame.args, "frame.args after")

local p = {}

function p.query(frame)
  if not frame.args[1] then
    log("missing argument #1 needed to identify a Def (defName or label can be used)")
    return nil
  end

  local def = get({"defName", frame.args[1]}) or get({"label", frame.args[1]})
  return def
end

function p.logObject(frame)
  return get(frame)
end

-------------
-- dev log --
-------------

if ENV == "dev" then
  local clock = string.format("os.clock(): %i ms", os.clock() * 1000)
  print("--" .. string.rep("-", #clock) .. "--")
  print("- " .. clock .. " -")
  print("--" .. string.rep("-", #clock) .. "--")
end

return p