Line 1: |
Line 1: |
| + | 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 47: |
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 254: |
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 402: |
Line 393: |
| | | |
| function EquipmentData:is_lbas_plane() | | function EquipmentData:is_lbas_plane() |
− | local eq_type = self:type() | + | self:_load_compatibility() |
− | return eq_type == 6 -- Carrier-based Fighter Aircraft | + | return EquipmentCompatibility.AirBase[self:type()] or false |
− | or eq_type == 7 -- Carrier-based Dive Bomber
| + | -- local eq_type = self:type() |
− | or eq_type == 8 -- Carrier-based Torpedo Bomber
| + | -- return eq_type == 6 -- Carrier-based Fighter Aircraft |
− | or eq_type == 9 -- Carrier-based Reconnaissance Aircraft
| + | -- or eq_type == 7 -- Carrier-based Dive Bomber |
− | or eq_type == 10 -- Reconnaissance Seaplane
| + | -- or eq_type == 8 -- Carrier-based Torpedo Bomber |
− | or eq_type == 11 -- Seaplane Bomber
| + | -- or eq_type == 9 -- Carrier-based Reconnaissance Aircraft |
− | or eq_type == 41 -- Large Flying Boat
| + | -- or eq_type == 10 -- Reconnaissance Seaplane |
− | or eq_type == 45 -- Seaplane Fighter
| + | -- or eq_type == 11 -- Seaplane Bomber |
− | or eq_type == 47 -- Land-based Attack Aircraft
| + | -- or eq_type == 25 -- Rotorcraft |
− | or eq_type == 48 -- Land-based Fighter
| + | -- or eq_type == 26 -- Anti-submarine Patrol Aircraft |
− | or eq_type == 49 -- Land-based Reconnaissance Aircraft
| + | -- or eq_type == 41 -- Large Flying Boat |
− | or eq_type == 56 -- Jet-powered Fighter Aircraft
| + | -- or eq_type == 45 -- Seaplane Fighter |
− | or eq_type == 57 -- Jet-powered Fighter-Bomber
| + | -- or eq_type == 47 -- Land-based Attack Aircraft |
− | or eq_type == 58 -- Jet-powered Attack Aircraft
| + | -- or eq_type == 48 -- Land-based Fighter |
− | or eq_type == 59 -- Jet-powered Recon Aircraft
| + | -- or eq_type == 49 -- Land-based Reconnaissance Aircraft |
− | or eq_type == 94 -- Carrier-based Reconnaissance Aircraft (II)
| + | -- 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 | | end |
| | | |
Line 433: |
Line 429: |
| return eq_type == 3 -- Large Caliber Main Gun | | return eq_type == 3 -- Large Caliber Main Gun |
| or eq_type == 38 -- Large Caliber Main Gun (II) | | 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 | | end |
| | | |
Line 441: |
Line 445: |
| function EquipmentData:is_not_aafd() | | function EquipmentData:is_not_aafd() |
| return not (self:is_aafd()) | | return not (self:is_aafd()) |
| + | end |
| + | |
| + | function EquipmentData:is_fighter_aircraft() |
| + | local eq_type = self:type() |
| + | 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 == 8 -- Carrier-based Torpedo Bomber |
| + | or eq_type == 11 -- Seaplane Bomber |
| + | -- or eq_type == 25 -- Rotorcraft |
| + | or eq_type == 26 -- Anti-submarine Patrol Aircraft |
| + | or eq_type == 47 -- Land-based Attack Aircraft |
| + | or eq_type == 53 -- Large Land-based Aircraft |
| + | or eq_type == 57 -- Jet-powered Fighter-Bomber |
| + | or eq_type == 58 -- Jet-powered Attack Aircraft |
| + | end |
| + | |
| + | function EquipmentData:is_reconnaissance_aircraft() |
| + | local eq_type = self:type() |
| + | return eq_type == 9 -- Carrier-based Reconnaissance Aircraft |
| + | 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 459: |
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 479: |
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 619: |
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()
| |
− | local name = self:name()
| |
| 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[name] or 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[name] or 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 |