Module:Mech

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

-- Module for handling information related to a BattleMech

-- p for "package", the interface of this module. Returned at the end
local p = {}

local core = {}
p.core = core

-- imports
local cargo = mw.ext.cargo
local getArgs = require('Module:Arguments').getArgs
local gear = require('Module:Gear').core

p.cargo = {}

-- p.cargo.chassis represents the schema of the information contained in a
-- chassisdef file.
p.cargo.chassis = {
  -- table is the name of the cargo table
  table = 'Chassis',
  -- fields is the schema of all of the fields of the Mech
  fields = {
    -- id is the primary key for this table. We expect it to be unique. It
    -- will be used as the foreign key in other tables to refer to this
    -- particular mech variant.
    id = {
      field = 'Id',
      type = 'String',
    },
    cost = {
      -- field defines the name of the field in the cargo table. for the
      -- field name, we use the name of the property in the ChassisDef
      -- json.
      field = 'Cost',
      -- type defines the type of the field
      type = 'Integer',
    },
    rarity = {
      field = 'Rarity',
      type = 'Integer',
    },
    purchasable = {
      field = 'Purchasable',
      type = 'Boolean',
    },
    manufacturer = {
      field = 'Manufacturer',
      type = 'String',
    },
    model = {
      field = 'Model',
      type = 'String',
    },
    uiName = {
      field = 'UIName',
      type = 'String',
    },
    name = {
      field = 'Name',
      type = 'String',
    },
    details = {
      field = 'Details',
      -- Details use the Text type, which is unindexed and intended for
      -- longer-form data.
      type = 'Text',
    },
    icon = {
      field = 'Icon',
      type = 'String',
    },
    tonnage = {
      field = 'Tonnage',
      type = 'Integer',
    },
    weightClass = {
      -- TODO(rust dev): this is probably an enum type. I bet we can
      -- do something with that.
      field = 'weightClass',
      type = 'String'
    },
    variantName = {
      field = 'VariantName',
      type = 'String',
    },
    stockRole = {
      field = 'StockRole',
      type = 'Text',
    },
    yangsThoughts = {
      field = 'YangsThoughts',
      type = 'Text',
    },
    leftArmActuatorLimit = {
      -- TODO(rust dev): another enum
      field = 'LeftArmActuatorLimit',
      type = 'String',
    },
    rightArmActuatorLimit = {
      field = 'RightArmActuatorLimit',
      type = 'String'
    },
    startingTonnage = {
      field = 'StartingTonnage',
      type = 'Float',
    },
  }
}

-- p.cargo.chassis_locations describes the information about a location on a mech
-- chassis
p.cargo.chassis_locations = {
  table = 'ChassisLocations',
  fields = {
    -- chassisID is the id of the chassis this location belongs to
    chassisID = {
      field = 'ChassisID',
      type = 'String',
    },
    location = {
      -- TODO(rust dev): this is probably an enum, and we can do something with
      -- that.
      field = 'Location',
      type = 'String'
    },
    tonnage = {
      field = 'Tonnage',
      type = 'Integer',
    },
    inventorySlots = {
      field = 'InventorySlots',
      type = 'Integer',
    },
    maxArmor = {
      field = 'MaxArmor',
      type = 'Integer',
    },
    maxRearArmor = {
      field = 'MaxRearArmor',
      type = 'Integer',
    },
    internalStructure = {
      field = 'InternalStructure',
      type = 'Integer',
    },
    hardpoints = {
      -- Hardpoints are done as a list of string. The value of the string is the
      -- type of hardpoint so, it might have the form in the template like
      --
      --   |Hardpoints = Energy,Energy,Missile
      --
      field = 'Hardpoints',
      type = 'List (,) of String',
    },
    omniHardpoints = {
      -- OmniHardpoints are the same, but for hardpoints where Omni == true
      field = 'OmniHardpoints',
      type = 'List (,) of String',
    }
  }
}

-- p.cargo.mech represents the information contained in a mechdef file.
p.cargo.mech = {
  table = 'Mech',
  fields = {
    -- id is the id of this mechdef.
    id = {
      field = 'Id',
      type = 'String',
    },
    -- chassisID is the foreign key pointed to the ChassisDef that corresponds
    -- to this mech.
    chassisID = {
      field = 'ChassisID',
      type = 'String',
    },
    -- mechTags is a list of strings, the tags for the mech. corresponds to
    -- MechTags.items in the json file.
    mechTags = {
      field = 'MechTags',
      type = 'List (,) of String',
    },
    cost = {
      field = 'Cost',
      type = 'Integer',
    },
    rarity = {
      field = 'Rarity',
      type = 'Integer',
    },
    purchasable = {
      field = 'Purchasable',
      type = 'Boolean',
    },
    manufacturer = {
      field = 'Manufacturer',
      type = 'String',
    },
    model = {
      field = 'Model',
      type = 'String',
    },
    uiName = {
      field = 'UIName',
      type = 'String',
    },
    name = {
      field = 'Name',
      type = 'String',
    },
    details = {
      field = 'Details',
      -- Details use the Text type, which is unindexed and intended for
      -- longer-form data.
      type = 'Text',
    },
    icon = {
      field = 'Icon',
      type = 'String',
    },
    simGameMechPartCost = {
      field = 'simGameMechPartCost',
      type = 'Integer',
    },
    version = {
      field = 'Version',
      type = 'Integer',
    },
  },
}

p.cargo.mech_locations = {
  table = 'MechLocations',
  -- a lot of the data in a mechdef seems redundant and not interesting for
  -- our purposes. however, include it here in the interest of correctness.
  fields = {
    -- mechID is the ID of the mech that this location belongs to
    mechID = {
      field = 'MechID',
      type = 'String',
    },
    -- location is the location being described here
    location = {
      field = 'Location',
      type = 'String',
    },
    currentArmor = {
      field = 'CurrentArmor',
      type = 'Integer',
    },
    currentRearArmor = {
      field = 'CurrentRearArmor',
      type = 'Integer',
    },
    currentInternalStructure = {
      field = 'CurrentInternalStructure',
      type = 'Integer',
    },
    assignedArmor = {
      field = 'AssignedArmor',
      type = 'Integer',
    },
    assignedRearArmor = {
      field = 'AssignedRearArmor',
      type = 'Integer',
    },
  },
}

-- p.cargo.mech_inventory describes an item in the inventory of a mech
--
-- excludes some fields which don't appear to be relevant:
--   * SimGameUID
--   * GUID
--   * DamageLevel
--   * prefabName
--   * hasPrefabName
p.cargo.mech_inventory = {
  table = 'MechInventory',
  fields = {
    -- mechID is the ID of the mechdef that this inventory item is from
    mechID = {
      field = 'MechID',
      type = 'String',
    },
    mountedLocation = {
      field = 'MountedLocation',
      type = 'String',
    },
    -- componentDefID is the ID of the component that that is mounted in this
    -- location
    componentDefID = {
      field = 'ComponentDefID',
      type = 'String',
    },
    componentDefType = {
      field = 'ComponentDefType',
      -- probably yet another enum.
      type = 'String'
    },
    hardpointSlot = {
      field = 'HardpointSlot',
      type = 'Integer',
    },
    -- fixedEquipment is a bit different. a mech can have equipment defined in
    -- the chassisdef as well as the mechdef. Equipment defined in the
    -- chassisdef is "FixedEquipment". Instead of having a separate table for
    -- this, we will instead set this flag true, and the mechID will instead
    -- refer to a chassis ID.
    fixedEquipment = {
      field = 'FixedEquipment',
      type = 'Boolean',
    },
  },
}

-- cargo_declare is a function which returns another function. The return value
-- is evaluated to declare the cargo table.
--
-- this is taken in large part from the Path of Exile wiki's Item2 module.
--
-- TODO(rust dev): define this in a separate module for reuse
function cargo_declare (data)
  return function (frame)
    if frame == nil then
      frame = mw.getCurrentFrame()
    end

    -- dcl_args are the arguments we will be passing to #cargo_declare
    local dcl_args = {}

    -- set the table name
    dcl_args._table = data.table

    -- for every field, we create an argument mapping the field name to its
    -- type.
    for k, field_data in pairs(data.fields) do
      if field_data then
        dcl_args[field_data.field] = field_data.type
      end
    end

    -- call #cargo_declare to declare the table before ending the function.
    frame:callParserFunction('#cargo_declare:', dcl_args)
  end
end

-- invoke these functions to declare the tables.
p.table_chassis = cargo_declare(p.cargo.chassis)
p.table_chassis_locations = cargo_declare(p.cargo.chassis_locations)
p.table_mech = cargo_declare(p.cargo.mech)
p.table_mech_locations = cargo_declare(p.cargo.mech_locations)
p.table_inventory = cargo_declare(p.cargo.mech_inventory)

-- stores the data in cargo. We don't just pass in all the template arguments,
-- as the template may have parameters that are not part of the chassis
-- definition. here instead, we iterate through the known fields of the table
-- and look for matching argument names. If we find one, we add it to the data
-- we use in the cargo store call.
function cargo_store(frame, schema, tpl_args)
  local cargo_data = {_table = schema.table}

  for _, field in pairs(schema.fields) do
    local arg = tpl_args[field.field]
    if arg ~= nil then
      -- consistently cast boolean values to the same type as 1 or 0.
      if field.type == 'Boolean' then
        if arg == '1' or arg ==  'true' then
          arg = '1'
        else
          arg ='0'
        end
      end

      cargo_data[field.field] = arg
    end
  end
  return frame:callParserFunction('#cargo_store:', cargo_data)
end

--
-- Classes

-- core.mech defines the lua-centric mech object
core.mech = {}

function core.mech:new(variantName)
  local mech = {
    variantName = variantName
  }

  setmetatable(mech, self)
  self.__index = self

  local chassisfields = {
    'Id',
    'Name',
    'UIName',
    'Details',
    'StockRole',
    'weightClass',
    'Tonnage',
    'StartingTonnage',
    'LeftArmActuatorLimit',
    'RightArmActuatorLimit',
  }

  where = string.format(
    '%s.%s="%s"', p.cargo.chassis.table, 'VariantName', variantName
  )

  local chassisRow = cargo.query(
    p.cargo.chassis.table, table.concat(chassisfields, ','), { where = where }
  )
  chassisRow = chassisRow[1]

  mech.chassisID = chassisRow['Id']
  mech.uiName = chassisRow['UIName']
  mech.name = chassisRow['Name']
  mech.details = chassisRow['Details']
  mech.weightClass = chassisRow['weightClass']
  mech.stockRole = chassisRow['StockRole']
  mech.leftArmActuatorLimit = chassisRow['LeftArmActuatorLimit']
  mech.rightArmActuatorLimit = chassisRow['RightArmActuatorLimit']
  mech.tonnage = tonumber(chassisRow['Tonnage'], 10)
  mech.startingTonnage = tonumber(chassisRow['StartingTonnage'], 10)

  local mechfields = {
    'MechTags',
    'Id',
  }
  local mechwhere = string.format(
    '%s.%s="%s"', p.cargo.mech.table, 'ChassisID', mech.chassisID
  )

  local mechRow = cargo.query(
    p.cargo.mech.table, table.concat(mechfields, ','), { where = mechwhere }
  )

  mechRow = mechRow[1]
  mech.mechID = mechRow['Id']
  mech.tags = mw.text.split(mechRow['MechTags'], ',')

  return mech
end

local standard_actuators = {
}
function core.mech:getLocations()
  if self.locations ~= nil then
    return self.locations
  end

  where = string.format(
    '%s.%s="%s"', 
    p.cargo.chassis_locations.table, 'ChassisID', self.chassisID
  )
  fields = {
    'Location',
    'InventorySlots',
    'MaxArmor',
    'MaxRearArmor',
    'InternalStructure',
    'Hardpoints',
    'OmniHardpoints',
  }

  local chassisLocations = cargo.query(
    p.cargo.chassis_locations.table, 
    table.concat(fields, ','),
    { where = where }
  )

  self.locations = {}
  for _, row in ipairs(chassisLocations) do
    local location = {
      tonnage = tonumber(row['Tonnage'], 10),
      location = row['Location'],
      maxArmor = tonumber(row['MaxArmor'], 10),
      maxRearArmor = tonumber(row['MaxRearArmor'], 10),
      internalStructure = tonumber(row['InternalStructure'], 10),
      hardpoints = mw.text.split(row['Hardpoints'], ','),
      omniHardpoints = mw.text.split(row['OmniHardpoints'], ','),
      gear = {},
    }

    self.locations[row['Location']] = location
  end

  local mechwhere = string.format(
    '%s.%s="%s"', 
    p.cargo.mech_locations.table, 'MechID', self.mechID
  )
  local mechfields = {
    'Location',
    'AssignedArmor',
    'AssignedRearArmor',
  }


  local mechlocations = cargo.query(
    p.cargo.mech_locations.table,
    table.concat(mechfields, ','),
    { where = mechwhere }
  )

  for _, row in ipairs(mechlocations) do
    self.locations[row['Location']].assignedArmor = tonumber(row['AssignedArmor'], 10)
    self.locations[row['Location']].assignedRearArmor = tonumber(row['AssignedRearArmor'], 10)
  end

  local mechGear = gear.get_gear(self.chassisID, self.mechID)
  for _, gear in ipairs(mechGear) do
    -- do not include in the gear list fixed-location gear like structure,
    -- gyro, or armor
    local exempt = false
    for _, category in ipairs(gear.categories) do
      if string.match(gear.id, 'emod_arm_part') or string.match(gear.id, 'emod_leg') then
        -- don't display standard actuators, which have IDs starting with emod
        -- all other actuators, display
        if string.match(gear.id, 'emod_') then
          gear.nodisplay = true
        end
      elseif category == 'EnginePart' then
        gear.nodisplay = true
      elseif category == 'Gyro' then
        mech.gyro = gear
        exempt = true
      elseif category == 'Armor' then
        mech.armor = gear
        exempt = true
      elseif category == 'Structure' then
        mech.structure = gear
        exempt = true
      end

      if category == 'Quirk' then
        mech.quirk = gear
      end
    end

    if not exempt then
      table.insert(self.locations[gear.location].gear, gear)
    end
  end

  return self.locations
end

function core.mech:getEngine()
  if self.engine ~= nil then
    return self.engine
  end

  self.engine = gear.get_engine(self.chassisID, self.mechID)

  return self.engine
end

function core.mech:getGear()
end

function core.mech:getWeapons()
  if self.weapons ~= nil then
    return self.weapons
  end
  
  self.weapons = gear.get_weapons(self.chassisID, self.mechID)

  return self.weapons
end

function core.mech:getHeatsinks()
  if self.heatsinks ~= nil then
    return self.heatsinks
  end

  self.heatsinks = gear.get_heatsinks(self.chassisID, self.mechID)

  return self.heatsinks
end

--
-- Templates
--

-- raw_table generates a raw table consisting of the data
function raw_table(schema, tpl_args)
  local rawTable = mw.html.create('table')
  rawTable:addClass('wikitable')

  for _, field in pairs(schema.fields) do
    rawTable:tag('tr')
      :tag('th'):wikitext(field.field):done()
      :tag('td'):wikitext(tpl_args[field.field])
  end
  return rawTable
end

-- p.chassis is a template which defines a Mech chassis.
function p.chassis (frame)
  tpl_args = getArgs(frame, {
    -- parentFirst tells us to prefer args from the parent page (the template
    -- that calls #invoke).
    parentFirst = true
  })

  -- p.mechChassis is a template for raw data, and so we display the raw data
  -- instead of anything fancy. To really display a mech, we need way more than
  -- the chassis, anyway.
  local chassisTable = raw_table(p.cargo.chassis, tpl_args)

  cargo_store(frame, p.cargo.chassis, tpl_args)

  return tostring(chassisTable)
end

-- p.chassisLocation is a template which defines a Mech chassis location
function p.chassisLocation (frame)
  tpl_args = getArgs(frame, {
    parentFirst = true,
  })

  local locationTable = raw_table(p.cargo.chassis_locations, tpl_args)

  cargo_store(frame, p.cargo.chassis_locations, tpl_args)

  return tostring(locationTable)
end

function p.mech (frame)
  tpl_args = getArgs(frame, {
    parentFirst = true
  })

  cargo_store(frame, p.cargo.mech, tpl_args)

  return tostring(raw_table(p.cargo.mech, tpl_args))
end

function p.mechLocation (frame)
  tpl_args = getArgs(frame, {
    parentFirst = true
  })

  cargo_store(frame, p.cargo.mech_locations, tpl_args)

  return tostring(raw_table(p.cargo.mech_locations, tpl_args))
end

function p.mechInventory (frame)
  tpl_args = getArgs(frame, {
    parentFirst = true
  })

  cargo_store(frame, p.cargo.mech_inventory, tpl_args)

  return tostring(raw_table(p.cargo.mech_inventory, tpl_args))
end

-- mechInfoBox creates a mech infobox.
function p.mechInfoBox (frame)
  tpl_args = getArgs(frame, {
    parentFirst = true
  })

  -- this function takes one argument, the VariantName.
  mech = core.mech:new(tpl_args['VariantName'])

  local structure = 0
  local armor = 0
  -- table keys in lua are also strings. these table keys are the same strings
  -- we expect the hardpoints to be. this is a shortcut to make counting
  -- hardpoints easier.
  local hardpoint_counts = {
    Energy = 0,
    Ballistic = 0,
    Missile = 0,
    AntiPersonnel = 0,
  }
  local omniHardpoints = 0

  -- mechtype is 'Standard BattleMech', unless we find a non-zero number of
  -- Omni hardpoints. then, we'll change it to be 'OmniMech'.
  local mechtype = 'Standard BattleMech'

  for _, location in pairs(mech:getLocations()) do
    -- add the value of each location's InternalStructure together. all of the
    -- fields returned from cargo are strings, so we have to first convert them
    -- to numbers. I think. I might be wrong about this, actually.
    structure = structure + location.internalStructure
    armor = armor + location.assignedArmor
    if location.assignedRearArmor ~= -1 then
      armor = armor + location.assignedRearArmor
    end

    -- hardpoints is a list of strings. we split that list along the commas to
    -- get an actual list we can iterate.
    for _, hardpoint in ipairs(location.hardpoints) do
      -- then, we check to see if the hardpoint is one of the keys of the
      -- hardpoint_counts table
      if hardpoint_counts[hardpoint] ~= nil then
        -- if it is, then we increment that entry's counter.
        hardpoint_counts[hardpoint] = hardpoint_counts[hardpoint] + 1
      end
    end
  end
  
  local damage = 0
  local stab = 0
  local heat_damage = 0
  local heat_gen = 0
  for _, weapon in ipairs(mech:getWeapons()) do
    if weapon.damage ~= nil then
      damage = damage + weapon.damage
      stab = stab + weapon.instability
      heat_damage = heat_damage + weapon.heatDamage
      heat_gen = heat_gen + weapon.heat
    end
  end

  local engine = mech:getEngine()
  local heatsinks = mech:getHeatsinks()

  local dissipation = engine.heat_sinking
  for _, heatsink in ipairs(heatsinks) do
    dissipation = dissipation + heatsink.dissipation
  end

  -- finally, to avoid re-inventing the wheel for this demo, we call the
  -- existing InfoboxVariant template with the data we've computed from the
  -- mech's database entries.
  local template_args = {
    variantname = mech.variantName,
    image = mech.name .. '.png',
    mechtype = 'Standard BattleMech',
    tonnage = mech.tonnage,
    class = mech.weightClass,
    armor = armor,
    structure = structure,
    enginetype = engine.shield.name,
    coresize = engine.rating,
    ecooling = string.format('+%d', engine.ecooling),
    maxdamage = damage,
    maxstability = stab,
    maxheat = heat_damage,
    alphaheat = heat_gen,
    heatsinking = dissipation,
  }

  if mech.gyro ~= nil then
    template_args.gyro = mech.gyro.name
  end
  if mech.armor ~= nil then
    template_args.armor = mech.armor.name
  end
  if mech.structure ~= nil then
    template_args.structure = mech.structure.name
  end


  for locationName, location in pairs(mech.locations) do
    local locationstring = ""
    if locationName == 'Head' then
      locationstring = 'headequip'
    elseif locationName == 'CenterTorso' then
      locationstring = 'ctequip'
    elseif locationName == 'LeftTorso' then
      locationstring = 'ltequip'
    elseif locationName == 'RightTorso' then
      locationstring = 'rtequip'
    elseif locationName == 'LeftArm' then
      locationstring = 'laequip'
    elseif locationName == 'RightArm' then
      locationstring = 'raequip'
    elseif locationname == 'LeftLeg' then 
      locationstring = 'llequip'
    else 
      locationstring = 'rlequip'
    end

    for i, gear in ipairs(location.gear) do
      -- skip gear marked nodisplay, which includes stuff like actuators and
      -- gyros.
      if not gear.nodisplay then
        if gear.fixed == '1' then
          template_args[locationstring .. tonumber(i, 10)] = gear.name .. ' (Fixed)'
        else
          template_args[locationstring .. tonumber(i, 10)] = gear.name
        end
      end
    end
  end

  if hardpoint_counts.Ballistic > 0 then 
    template_args.ballistic = hardpoint_counts.Ballistic
  end
  if hardpoint_counts.Energy > 0 then
    template_args.energy = hardpoint_counts.Energy
  end
  
  if hardpoint_counts.Missile > 0 then
    template_args.missile = hardpoint_counts.Missile
  end

  if hardpoint_counts.AntiPersonnel > 0 then
    template_args.support = hardpoint_counts.AntiPersonnel
  end
  
  if omniHardpoints > 0 then
    template_args.omni = omniHardpoints
    template_args.mechtype = 'OmniMech'
  end

  template_args['stock role'] = mech.stockRole

  return frame:expandTemplate{
    title = 'InfoboxVariant',
    args = template_args
  }
end

return p