1-------------------------------------------------------------------------------- 2-------------------------------------------------------------------------------- 3-- 4-- file: setupdefs.lua 5-- brief: setup some custom UnitDefs parameters, 6-- and UnitDefNames, FeatureDefNames, WeaponDefNames 7-- author: Dave Rodgers 8-- 9-- Copyright (C) 2007. 10-- Licensed under the terms of the GNU GPL, v2 or later. 11-- 12-------------------------------------------------------------------------------- 13-------------------------------------------------------------------------------- 14 15for _,ud in pairs(UnitDefs) do 16 17 -- set the cost value (same as shown in the tooltip) 18 ud.cost = ud.metalCost + (ud.energyCost / 60.0) 19 20 -- add the custom weapons based parameters 21 ud.hasShield = false 22 ud.canParalyze = false 23 ud.canStockpile = false 24 ud.canAttackWater = false 25 ud.wDefs = {} 26 for i, wt in ipairs(ud.weapons) do 27 local wd = WeaponDefs[wt.weaponDef] 28 ud.wDefs[i] = wd 29 if (wd) then 30 if (wd.isShield) then ud.hasShield = true end 31 if (wd.paralyzer) then ud.canParalyze = true end 32 if (wd.stockpile) then ud.canStockpile = true end 33 if (wd.waterWeapon) then ud.canAttackWater = true end 34 end 35 end 36end 37 38 39-------------------------------------------------------------------------------- 40-------------------------------------------------------------------------------- 41 42-- setup the UnitDefNames{} table 43do 44 local tbl = {} 45 for _,def in pairs(UnitDefs) do 46 tbl[def.name] = def 47 end 48 UnitDefNames = tbl 49end 50 51-- setup the FeatureDefNames{} table 52do 53 local tbl = {} 54 for _,def in pairs(FeatureDefs) do 55 tbl[def.name] = def 56 end 57 FeatureDefNames = tbl 58end 59 60-- setup the WeaponDefNames{} table 61do 62 local tbl = {} 63 for _,def in pairs(WeaponDefs) do 64 tbl[def.name] = def 65 end 66 WeaponDefNames = tbl 67end 68 69 70-------------------------------------------------------------------------------- 71-------------------------------------------------------------------------------- 72