Changes

no edit summary
Line 1: Line 1: −
local _ = require("Module:Core")
+
local U = require("Module:Core")
 
local BaseData = require("Module:BaseData")
 
local BaseData = require("Module:BaseData")
 
local Formatting = require('Module:Formatting')
 
local Formatting = require('Module:Formatting')
Line 48: Line 48:  
--Returns the equipment's id.
 
--Returns the equipment's id.
 
function EquipmentData:id()
 
function EquipmentData:id()
return self._id
+
return self._is_enemy and self._id and self._id < 1501 and self._id + 1000 or self._id
 
end
 
end
   Line 214: Line 214:  
end
 
end
   −
function EquipmentData:card()
+
function EquipmentData:card(as_item)
 +
    if self._item or as_item and self._item_id then
 +
        return Formatting:item_card(self:asset_name())
 +
    end
 
if self._card then
 
if self._card then
 
return self._card
 
return self._card
Line 252: Line 255:  
local api_id = self:api_id()
 
local api_id = self:api_id()
 
if api_id then
 
if api_id then
    if api_id > 500 then
  −
        local equipment_type = self:type()
  −
        if equipment_type < 6
  −
            or (equipment_type > 11 and equipment_type ~= 25 and equipment_type ~= 26 and equipment_type ~= 41
  −
            and equipment_type ~= 45 and equipment_type ~= 47 and equipment_type ~= 48 and equipment_type ~= 94) then
  −
        Equipment = Equipment or require(self._constructor_module)
  −
        EquipmentFrom = EquipmentFrom or require("Module:Collection/EquipmentByApiId")
  −
        return Equipment(EquipmentFrom[api_id - 500]):cg()
  −
    end
  −
    end
   
return Formatting:equipment_cg(api_id, self:asset_name())
 
return Formatting:equipment_cg(api_id, self:asset_name())
 
end
 
end
Line 286: Line 279:  
end
 
end
   −
function EquipmentData:icon()
+
function EquipmentData:icon(as_item)
 +
    if self._item or as_item and self._item_id then
 +
        return Formatting:item_icon(self:asset_name())
 +
    end
 
     if self._icon or self._icon == false then
 
     if self._icon or self._icon == false then
 
         return self._icon
 
         return self._icon
Line 397: Line 393:     
function EquipmentData:is_lbas_plane()
 
function EquipmentData:is_lbas_plane()
 +
self:_load_compatibility()
 +
return EquipmentCompatibility.AirBase[self:type()] or false
 +
-- local eq_type = self:type()
 +
-- return eq_type == 6 -- Carrier-based Fighter Aircraft
 +
-- or eq_type == 7 -- Carrier-based Dive Bomber
 +
-- or eq_type == 8 -- Carrier-based Torpedo Bomber
 +
-- or eq_type == 9 -- Carrier-based Reconnaissance Aircraft
 +
-- or eq_type == 10 -- Reconnaissance Seaplane
 +
-- or eq_type == 11 -- Seaplane Bomber
 +
-- or eq_type == 25 -- Rotorcraft
 +
-- or eq_type == 26 -- Anti-submarine Patrol Aircraft
 +
-- or eq_type == 41 -- Large Flying Boat
 +
-- or eq_type == 45 -- Seaplane Fighter
 +
-- or eq_type == 47 -- Land-based Attack Aircraft
 +
-- or eq_type == 48 -- Land-based Fighter
 +
-- or eq_type == 49 -- Land-based Reconnaissance Aircraft
 +
-- or eq_type == 53 -- Large Land-based Aircraft
 +
-- or eq_type == 56 -- Jet-powered Fighter Aircraft
 +
-- or eq_type == 57 -- Jet-powered Fighter-Bomber
 +
-- or eq_type == 58 -- Jet-powered Attack Aircraft
 +
-- or eq_type == 59 -- Jet-powered Recon Aircraft
 +
-- or eq_type == 94 -- Carrier-based Reconnaissance Aircraft (II)
 +
end
 +
 +
function EquipmentData:is_lbas_ha_bomber()
 +
    return self:id() >= 501 and self:type() == 7 and self:range() == 4
 +
end
 +
 +
function EquipmentData:is_lbas_attack_aircraft()
 +
    return self:type() == 47 or self:type() == 53
 +
end
 +
 +
function EquipmentData:is_large_caliber_main_gun()
 +
local eq_type = self:type()
 +
return eq_type == 3 -- Large Caliber Main Gun
 +
or eq_type == 38 -- Large Caliber Main Gun (II)
 +
end
 +
 +
function EquipmentData:is_main_gun()
 +
local eq_type = self:type()
 +
return eq_type == 1 -- Small Caliber Main Guns
 +
or eq_type == 2 -- Medium Caliber Main Guns
 +
or eq_type == 3 -- Large Caliber Main Guns
 +
or eq_type == 38 -- Large Caliber Main Guns (II)
 +
end
 +
 +
function EquipmentData:is_aafd()
 +
return self:aa() and self:aa() >= 8
 +
end
 +
 +
function EquipmentData:is_not_aafd()
 +
return not (self:is_aafd())
 +
end
 +
 +
function EquipmentData:is_fighter_aircraft()
 
local eq_type = self:type()
 
local eq_type = self:type()
 
return eq_type == 6 -- Carrier-based Fighter Aircraft
 
return eq_type == 6 -- Carrier-based Fighter Aircraft
 +
or eq_type == 45 -- Seaplane Fighter
 +
or eq_type == 48 -- Land-based Fighter
 +
or eq_type == 56 -- Jet-powered Fighter Aircraft
 +
end
 +
 +
function EquipmentData:is_bomber_aircraft()
 +
local eq_type = self:type()
 
or eq_type == 7 -- Carrier-based Dive Bomber
 
or eq_type == 7 -- Carrier-based Dive Bomber
 
or eq_type == 8 -- Carrier-based Torpedo Bomber
 
or eq_type == 8 -- Carrier-based Torpedo Bomber
or eq_type == 9 -- Carrier-based Reconnaissance Aircraft
  −
or eq_type == 10 -- Reconnaissance Seaplane
   
or eq_type == 11 -- Seaplane Bomber
 
or eq_type == 11 -- Seaplane Bomber
or eq_type == 41 -- Large Flying Boat
+
-- or eq_type == 25 -- Rotorcraft
or eq_type == 45 -- Seaplane Fighter
+
or eq_type == 26 -- Anti-submarine Patrol Aircraft
 
or eq_type == 47 -- Land-based Attack Aircraft
 
or eq_type == 47 -- Land-based Attack Aircraft
or eq_type == 48 -- Interceptor Fighter
+
or eq_type == 53 -- Large Land-based Aircraft
or eq_type == 56 -- Jet-powered Fighter Aircraft
   
or eq_type == 57 -- Jet-powered Fighter-Bomber
 
or eq_type == 57 -- Jet-powered Fighter-Bomber
 
or eq_type == 58 -- Jet-powered Attack Aircraft
 
or eq_type == 58 -- Jet-powered Attack Aircraft
or eq_type == 59 -- Jet-powered Recon Aircraft
  −
or eq_type == 94 -- Carrier-based Reconnaissance Aircraft (II)
   
end
 
end
   −
function EquipmentData:is_large_caliber_main_gun()
+
function EquipmentData:is_reconnaissance_aircraft()
 
local eq_type = self:type()
 
local eq_type = self:type()
return eq_type == 3 -- Large Caliber Main Gun
+
return eq_type == 9 -- Carrier-based Reconnaissance Aircraft
or eq_type == 38 -- Large Caliber Main Gun (II)
+
or eq_type == 10 -- Reconnaissance Seaplanes
 +
or eq_type == 49 -- Land-based Reconnaissance Aircraft
 +
or eq_type == 94 -- Carrier-based Reconnaissance Aircraft (II)
 
end
 
end
   Line 437: Line 492:  
return eq_type == 12 -- Small Radar
 
return eq_type == 12 -- Small Radar
 
or eq_type == 13 -- Large Radar
 
or eq_type == 13 -- Large Radar
 +
or eq_type == 93 -- Large Radar (II)
 +
end
 +
 +
function EquipmentData:is_large_radar()
 +
local eq_type = self:type()
 +
return eq_type == 13 -- Large Radar
 
or eq_type == 93 -- Large Radar (II)
 
or eq_type == 93 -- Large Radar (II)
 
end
 
end
Line 457: Line 518:  
or eq_type == 27 -- Extra Armor (Medium)
 
or eq_type == 27 -- Extra Armor (Medium)
 
or eq_type == 28 -- Extra Armor (Large)
 
or eq_type == 28 -- Extra Armor (Large)
 +
end
 +
 +
function EquipmentData:is_secondary_gun()
 +
local eq_type = self:type()
 +
return eq_type == 4 -- Secondary Guns
 +
or eq_type == 95 -- Secondary Guns (II)
 
end
 
end
   Line 468: Line 535:     
function EquipmentData:wikipedia()
 
function EquipmentData:wikipedia()
    local parts = self._wikipedia and _.split(self._wikipedia, ':')
+
  return self._wikipedia
return parts and (parts[2] or parts[1])
   
end
 
end
   Line 598: Line 664:  
function EquipmentData:gun_fit_group()
 
function EquipmentData:gun_fit_group()
 
return self._gun_fit_group
 
return self._gun_fit_group
 +
end
 +
 +
function EquipmentData:_load_compatibility()
 +
if not EquipmentCompatibility then
 +
EquipmentCompatibility = mw.loadData('Module:Data/EquipmentRefit')
 +
end
 +
end
 +
 +
local function convert_compatibility(compatibility_table, equipment_type, name)
 +
local value = compatibility_table[name] or compatibility_table[equipment_type]
 +
return value == true and 1 or type(value) == 'string' and {value = U.startsWith(value, 'Only') and 2 or 3, note = value, code = value} or 0
 
end
 
end
    
function EquipmentData:equippable(ship_type)
 
function EquipmentData:equippable(ship_type)
 
self:_load_compatibility()
 
self:_load_compatibility()
local equipment_type = self:type()
   
if not ship_type then
 
if not ship_type then
 
local result = {}
 
local result = {}
 
for ship_type, compatibility_table in pairs(EquipmentCompatibility) do
 
for ship_type, compatibility_table in pairs(EquipmentCompatibility) do
result[ship_type] = compatibility_table[equipment_type] or 0
+
result[ship_type] = convert_compatibility(compatibility_table, self:type(), self:name())
 
end
 
end
 
return result
 
return result
 +
end
 +
local compatibility_table = EquipmentCompatibility[ship_type]
 +
if not compatibility_table then
 +
return nil
 +
end
 +
local compatibility = convert_compatibility(compatibility_table, self:type(), self:name())
 +
if type(compatibility) ~= "table" then
 +
return compatibility
 
else
 
else
local compatibility_table = EquipmentCompatibility[ship_type]
+
return compatibility.value, compatibility.note
if compatibility_table then
  −
local compatibility = compatibility_table[equipment_type] or 0
  −
if type(compatibility) ~= "table" then
  −
return compatibility
  −
else
  −
return compatibility.value, compatibility.note
  −
end
  −
end
  −
end
  −
end
  −
 
  −
function EquipmentData:_load_compatibility()
  −
if not EquipmentCompatibility then
  −
EquipmentCompatibility = mw.loadData('Module:EquipmentCompatibility')
   
end
 
end
 
end
 
end
cssedit, gkautomate
6,926

edits