-- 清空现有数据
SET FOREIGN_KEY_CHECKS=0;
TRUNCATE TABLE dataset_equipment;
TRUNCATE TABLE datasets;
TRUNCATE TABLE cost_data;
TRUNCATE TABLE loitering_munition_params;
TRUNCATE TABLE common_params;
TRUNCATE TABLE equipment;
SET FOREIGN_KEY_CHECKS=1;

-- 按系列插入装备数据，确保ID连续
-- 1. HAROP/Harpy 系列 (ID: 1-3)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(1, 'IAI Harop', '巡飞弹', '以色列'),
(2, 'IAI Harpy', '巡飞弹', '以色列'),
(3, 'IAI Mini Harpy', '巡飞弹', '以色列');

-- 2. Hero 系列 (ID: 4-9)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(4, 'Hero-30', '巡飞弹', '以色列 UVision'),
(5, 'Hero-70', '巡飞弹', '以色列 UVision'),
(6, 'Hero-120', '巡飞弹', '以色列 UVision'),
(7, 'Hero-250', '巡飞弹', '以色列 UVision'),
(8, 'Hero-400EC', '巡飞弹', '以色列 UVision'),
(9, 'Hero-900', '巡飞弹', '以色列 UVision');

-- 3. Switchblade 系列 (ID: 10-13)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(10, 'Switchblade 300', '巡飞弹', '美国 AeroVironment'),
(11, 'Switchblade 600', '巡飞弹', '美国 AeroVironment'),
(12, 'Switchblade 300 Block 10', '巡飞弹', '美国 AeroVironment'),
(13, 'Switchblade 600 Extended Range', '巡飞弹', '美国 AeroVironment');

-- 4. Warmate 系列 (ID: 14-18)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(14, 'Warmate 1.0', '巡飞弹', '波兰 WB Electronics'),
(15, 'Warmate 2.0', '巡飞弹', '波兰 WB Electronics'),
(16, 'Warmate-V', '巡飞弹', '波兰 WB Electronics'),
(17, 'Warmate-L', '巡飞弹', '波兰 WB Electronics'),
(18, 'Warmate 3.0', '巡飞弹', '波兰 WB Electronics');

-- 5. CH-901/902 系列 (ID: 19-23)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(19, 'CH-901', '巡飞弹', '中国航天科工'),
(20, 'CH-901A', '巡飞弹', '中国航天科工'),
(21, 'CH-901H', '巡飞弹', '中国航天科工'),
(22, 'CH-902', '巡飞弹', '中国航天科工'),
(23, 'CH-902A', '巡飞弹', '中国航天科工');

-- 6. WS-43/61 系列 (ID: 24-28)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(24, 'WS-43', '巡飞弹', '中国航天科工'),
(25, 'WS-43A', '巡飞弹', '中国航天科工'),
(26, 'WS-43B', '巡飞弹', '中国航天科工'),
(27, 'WS-61', '巡飞弹', '中国航天科工'),
(28, 'WS-61A', '巡飞弹', '中国航天科工');

-- 7. Kargu/Alpagu 系列 (ID: 29-33)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(29, 'Kargu', '巡飞弹', '土耳其 STM'),
(30, 'Kargu-2', '巡飞弹', '土耳其 STM'),
(31, 'Alpagu', '巡飞弹', '土耳其 STM'),
(32, 'Alpagu Block-II', '巡飞弹', '土耳其 STM'),
(33, 'Kargu Autonomous', '巡飞弹', '土耳其 STM');

-- 8. Shahed 系列 (ID: 34-38)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(34, 'Shahed-131', '巡飞弹', '伊朗'),
(35, 'Shahed-131B', '巡飞弹', '伊朗'),
(36, 'Shahed-136', '巡飞弹', '伊朗'),
(37, 'Shahed-136B', '巡飞弹', '伊朗'),
(38, 'Shahed-136C', '巡飞弹', '伊朗');

-- 9. Green Dragon 系列 (ID: 39-43)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(39, 'Green Dragon', '巡飞弹', '以色列 IAI'),
(40, 'Green Dragon Extended Range', '巡飞弹', '以色列 IAI'),
(41, 'Green Dragon Block 2', '巡飞弹', '以色列 IAI'),
(42, 'Green Dragon Maritime', '巡飞弹', '以色列 IAI'),
(43, 'Green Dragon-S', '巡飞弹', '以色列 IAI');

-- 10. Phoenix Ghost 系列 (ID: 44-48)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(44, 'Phoenix Ghost', '巡飞弹', '美国 AEVEX Aerospace'),
(45, 'Phoenix Ghost Block I', '巡飞弹', '美国 AEVEX Aerospace'),
(46, 'Phoenix Ghost Block II', '巡飞弹', '美国 AEVEX Aerospace'),
(47, 'Phoenix Ghost Maritime', '巡飞弹', '美国 AEVEX Aerospace'),
(48, 'Phoenix Ghost-ER', '巡飞弹', '美国 AEVEX Aerospace');

-- 11. ZALA Lancet 系列 (ID: 49-52)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(49, 'Lancet-1', '巡飞弹', '俄罗斯 ZALA'),
(50, 'Lancet-3', '巡飞弹', '俄罗斯 ZALA'),
(51, 'Lancet-3M', '巡飞弹', '俄罗斯 ZALA'),
(52, 'Lancet-4', '巡飞弹', '俄罗斯 ZALA');

-- 12. Rotem L 系列 (ID: 53-56)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(53, 'Rotem L', '巡飞弹', '以色列 IAI'),
(54, 'Rotem L-X', '巡飞弹', '以色列 IAI'),
(55, 'Rotem L-M', '巡飞弹', '以色列 IAI'),
(56, 'Rotem L-ER', '巡飞弹', '以色列 IAI');

-- 13. KUB-BLA 系列 (ID: 57-60)
INSERT INTO equipment (id, name, type, manufacturer) VALUES
(57, 'KUB-BLA', '巡飞弹', '俄罗斯 ZALA'),
(58, 'KUB-BLA-E', '巡飞弹', '俄罗斯 ZALA'),
(59, 'KUB-BLA-M', '巡飞弹', '俄罗斯 ZALA'),
(60, 'KUB-BLA-ER', '巡飞弹', '俄罗斯 ZALA');

-- 插入通用参数
INSERT INTO common_params (equipment_id, length_m, width_m, height_m, weight_kg, max_range_km) VALUES
(1, 2.5, 0.43, 0.43, 135, 1000),  -- IAI Harop
(2, 2.7, 0.35, 0.35, 125, 500),   -- IAI Harpy
(3, 2.1, 0.30, 0.30, 45, 100),    -- IAI Mini Harpy
(4, 0.76, 0.17, 0.17, 3.0, 15),   -- Hero-30
(5, 0.87, 0.18, 0.18, 6.5, 25),   -- Hero-70
(6, 1.3, 0.23, 0.23, 12.5, 40),   -- Hero-120
(7, 2.1, 0.30, 0.30, 35, 150),    -- Hero-250
(8, 2.4, 0.35, 0.35, 40, 150),    -- Hero-400EC
(9, 2.9, 0.40, 0.40, 90, 250),    -- Hero-900
(10, 0.58, 0.12, 0.12, 2.5, 10),
(11, 1.30, 0.22, 0.22, 15.0, 40),
(12, 0.60, 0.12, 0.12, 2.7, 15),   -- Switchblade 300 Block 10
(13, 1.35, 0.22, 0.22, 16.0, 50),  -- Switchblade 600 Extended Range
(14, 0.68, 0.12, 0.12, 2.5, 10),
(15, 1.30, 0.22, 0.22, 15.0, 40),
(16, 0.68, 0.12, 0.12, 2.5, 10),
(17, 1.30, 0.22, 0.22, 15.0, 40),
(18, 0.68, 0.12, 0.12, 2.5, 10),
(19, 1.2, 0.18, 0.18, 9.0, 20),
(20, 1.2, 0.18, 0.18, 9.3, 25),
(21, 1.2, 0.18, 0.18, 9.5, 20),
(22, 1.4, 0.22, 0.22, 15.0, 30),
(23, 1.4, 0.22, 0.22, 15.5, 35),
(24, 1.8, 0.35, 0.35, 20, 60),
(25, 1.8, 0.35, 0.35, 21, 70),
(26, 1.9, 0.35, 0.35, 22, 80),
(27, 2.2, 0.40, 0.40, 35, 100),
(28, 2.2, 0.40, 0.40, 37, 120),
(29, 0.6, 0.35, 0.35, 7.0, 10),
(30, 0.6, 0.35, 0.35, 7.2, 15),
(31, 1.0, 0.23, 0.23, 3.7, 5),
(32, 1.0, 0.23, 0.23, 3.9, 8),
(33, 0.6, 0.35, 0.35, 7.5, 15),
(34, 2.6, 0.34, 0.34, 135, 900),
(35, 2.6, 0.34, 0.34, 140, 1000),
(36, 3.5, 0.42, 0.42, 200, 2000),
(37, 3.5, 0.42, 0.42, 210, 2200),
(38, 3.5, 0.42, 0.42, 215, 2500),
(39, 1.5, 0.20, 0.20, 15, 40),
(40, 1.6, 0.20, 0.20, 16, 50),
(41, 1.5, 0.20, 0.20, 15.5, 45),
(42, 1.5, 0.20, 0.20, 15.8, 40),
(43, 1.2, 0.18, 0.18, 12, 30),
(44, 1.5, 0.25, 0.25, 14.0, 30),
(45, 1.5, 0.25, 0.25, 14.5, 35),
(46, 1.6, 0.26, 0.26, 15.0, 40),
(47, 1.5, 0.25, 0.25, 14.8, 30),
(48, 1.7, 0.27, 0.27, 16.0, 50),
(49, 1.0, 0.20, 0.20, 5.0, 40),
(50, 1.65, 0.35, 0.35, 12.0, 70),
(51, 1.65, 0.35, 0.35, 12.5, 80),
(52, 1.80, 0.40, 0.40, 15.0, 100),
(53, 0.8, 0.25, 0.25, 4.5, 10),   -- Rotem L
(54, 0.8, 0.25, 0.25, 4.8, 15),   -- Rotem L-X
(55, 0.8, 0.25, 0.25, 4.7, 10),   -- Rotem L-M
(56, 0.9, 0.27, 0.27, 5.2, 20),   -- Rotem L-ER
(57, 1.21, 0.95, 0.165, 3.0, 40),  -- KUB-BLA
(58, 1.21, 0.95, 0.165, 3.2, 50),  -- KUB-BLA-E
(59, 1.21, 0.95, 0.165, 3.3, 45),  -- KUB-BLA-M
(60, 1.25, 1.0, 0.17, 3.5, 60);    -- KUB-BLA-ER

-- 插入特有参数
INSERT INTO loitering_munition_params (equipment_id, wingspan_m, warhead_weight_kg, max_speed_ms, cruise_speed_kmh,
    endurance_min,
    warhead_type,
    launch_mode,
    power_system,
    guidance_system
) VALUES
-- HAROP/Harpy系列
(1, 3.0, 23, 51.4, 185, 360, '高爆战斗部', '箱式发射/空中发射', '活塞发动机', 'GPS/INS/光电/数据链'),
(2, 2.1, 32, 51.4, 148, 120, '高爆战斗部', '箱式发射', '活塞发动机', 'GPS/INS/被动雷达'),
(3, 1.8, 8, 47.2, 130, 120, '高爆战斗部', '箱式发射', '电动机', 'GPS/INS/光电/被动雷达'),

-- Hero系列
(4, 1.0, 0.5, 36.1, 100, 30, '破片杀伤战斗部', '箱式发射/单兵发射', '电动机', 'GPS/INS/光电'),
(5, 1.5, 1.2, 38.9, 105, 45, '破片杀伤战斗部', '箱式发射', '电动机', 'GPS/INS/光电'),
(6, 2.1, 3.5, 41.7, 100, 60, '破片杀伤战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(7, 2.5, 10.0, 47.2, 130, 120, '破片杀伤战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(8, 2.8, 8.0, 47.2, 130, 240, '破片杀伤战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(9, 3.0, 20.0, 51.4, 150, 360, '破片杀伤战斗部', '箱式发射', '活塞发动机', 'GPS/INS/光电/数据链'),

-- Switchblade系列
(10, 0.68, 0.2, 38.9, 98, 15, '破片杀伤战斗部', '单兵发射管', '电动机', 'GPS/INS/光电'),
(11, 2.2, 4.0, 51.4, 115, 40, '破甲战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(12, 0.70, 0.25, 41.7, 100, 20, '破片杀伤战斗部', '单兵发射管', '电动机', 'GPS/INS/光电/数据链'),
(13, 2.3, 4.1, 51.4, 115, 50, '破甲战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链/AI辅助'),

-- Warmate系列
(14, 0.68, 0.2, 38.9, 98, 15, '破片杀伤战斗部', '单兵发射管', '电动机', 'GPS/INS/光电'),
(15, 1.30, 0.22, 0.22, 15.0, 40, '破甲战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(16, 0.68, 0.2, 38.9, 98, 15, '破片杀伤战斗部', '单兵发射管', '电动机', 'GPS/INS/光电/数据链'),
(17, 1.30, 0.22, 0.22, 15.0, 40, '破甲战斗部', '箱式发射', '电动机', 'GPS/INS/光电/数据链'),
(18, 0.68, 0.2, 38.9, 98, 15, '破片杀伤战斗部', '单兵发射管', '电动机', 'GPS/INS/光电/数据链'),

-- CH-901/902系列
(19, 1.8, 2.0, 44.4, 95, 120, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(20, 1.8, 2.2, 47.2, 100, 140, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(21, 1.8, 3.0, 44.4, 95, 120, '破甲战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(22, 2.2, 3.5, 50.0, 110, 180, '模块化战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(23, 2.2, 3.5, 50.0, 110, 200, '模块化战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助/卫通'),
(24, 2.4, 3.8, 47.2, 100, 45, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(25, 2.4, 4.0, 50.0, 110, 60, '破片杀伤/破甲双用战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(26, 2.5, 4.0, 50.0, 110, 80, '破片杀伤/破甲双用战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(27, 3.0, 8.0, 55.6, 120, 120, '模块化战斗部', '箱式发射', '活塞发动机', 'GPS/INS/光电/数据链/AI辅助'),
(28, 3.0, 8.5, 55.6, 120, 150, '模块化战斗部', '箱式发射', '活塞发动机', 'GPS/INS/光电/数据链/AI辅助/卫通'),
(29, 0.7, 1.0, 36.1, 72, 30, '破片杀伤战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别'),
(30, 0.7, 1.1, 38.9, 75, 40, '破片杀伤战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别/数据链'),
(31, 1.3, 0.8, 41.7, 80, 20, '破片杀伤战斗部', '弹射式', '电动机', 'GPS/INS/光电'),
(32, 1.3, 0.9, 44.4, 85, 25, '破片杀伤战斗部', '弹射式', '电动机', 'GPS/INS/光电/AI识别'),
(33, 0.7, 1.2, 38.9, 75, 45, '破片杀伤战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别/自主决策'),
(34, 2.2, 15, 55.6, 150, 180, '高爆战斗部', '箱式发射/弹射式', '活塞发动机', 'GPS/INS/光电'),
(35, 2.2, 15, 58.3, 160, 200, '高爆战斗部', '箱式发射/弹射式', '活塞发动机', 'GPS/INS/光电/数据链'),
(36, 2.5, 30, 61.1, 180, 240, '高爆战斗部', '箱式发射/弹射式', '活塞发动机', 'GPS/INS/光电/数据链'),
(37, 2.5, 35, 63.9, 185, 260, '高爆战斗部', '箱式发射/弹射式', '活塞发动机', 'GPS/INS/光电/数据链/AI辅助'),
(38, 2.5, 40, 66.7, 190, 300, '高爆战斗部', '箱式发射/弹射式', '活塞发动机', 'GPS/INS/光电/数据链/AI辅助/卫通'),
(39, 2.0, 3.0, 47.2, 110, 90, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(40, 2.2, 3.0, 50.0, 115, 120, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(41, 2.0, 3.5, 47.2, 110, 90, '破片杀伤/破甲双用战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(42, 2.0, 3.0, 47.2, 110, 90, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/抗盐雾'),
(43, 1.8, 2.5, 44.4, 100, 60, '破片杀伤战斗部', '箱式发射/单兵发射', '电动机', 'GPS/INS/光电/数据链'),
(44, 2.2, 3.5, 47.2, 110, 120, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链'),
(45, 2.2, 3.8, 50.0, 115, 140, '破片杀伤/破甲双用战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助'),
(46, 2.3, 4.0, 52.8, 120, 160, '模块化战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助/红外'),
(47, 2.2, 3.5, 47.2, 110, 120, '破片杀伤战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/抗盐雾'),
(48, 2.4, 4.2, 55.6, 125, 180, '模块化战斗部', '箱式发射/弹射式', '电动机', 'GPS/INS/光电/数据链/AI辅助/卫通'),
(49, 1.2, 1.0, 44.4, 80, 30, '破片杀伤战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别'),
(50, 2.0, 3.0, 50.0, 110, 40, '破片杀伤/破甲双用战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/数据链'),
(51, 2.0, 3.5, 52.8, 120, 50, '破片杀伤/破甲双用战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/数据链/红外'),
(52, 2.3, 5.0, 55.6, 130, 60, '模块化战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/数据链/红外/卫通'),
(53, 0.9, 1.0, 36.1, 80, 30, '破片杀伤战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别'),
(54, 0.9, 1.2, 38.9, 85, 45, '破片杀伤/破甲双用战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别/数据链'),
(55, 0.9, 1.0, 36.1, 80, 30, '破片杀伤战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别/抗盐雾'),
(56, 1.0, 1.3, 41.7, 90, 60, '破片杀伤/破甲双用战斗部', '垂直起降', '电动机', 'GPS/INS/光电/AI识别/数据链'),
(57, 1.2, 1.0, 41.7, 80, 30, '破片杀伤战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别'),
(58, 1.2, 1.2, 44.4, 85, 40, '破片杀伤/破甲双用战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/数据链'),
(59, 1.2, 1.3, 44.4, 85, 35, '破片杀伤战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/红外'),
(60, 1.3, 1.5, 47.2, 90, 50, '破片杀伤/破甲双用战斗部', '弹射式发射', '电动机', 'GPS/INS/光电/AI识别/数据链/红外');

-- 插入成本数据
INSERT INTO cost_data (equipment_id, actual_cost) VALUES
(1, 800000),  -- IAI Harop
(2, 700000),  -- IAI Harpy
(3, 350000),  -- IAI Mini Harpy
(4, 70000),    -- Hero-30
(5, 120000),   -- Hero-70
(6, 150000),   -- Hero-120
(7, 300000),   -- Hero-250
(8, 400000),   -- Hero-400EC
(9, 650000),   -- Hero-900
(10, 60000),    -- Switchblade 300
(11, 180000),   -- Switchblade 600
(12, 75000),   -- Switchblade 300 Block 10
(13, 200000),  -- Switchblade 600 Extended Range
(14, 60000),    -- Warmate 1.0
(15, 180000),   -- Warmate 2.0
(16, 60000),    -- Warmate-V
(17, 180000),   -- Warmate-L
(18, 60000),    -- Warmate 3.0
(19, 100000),   -- CH-901
(20, 120000),   -- CH-901A
(21, 130000),   -- CH-901H
(22, 180000),   -- CH-902
(23, 200000),   -- CH-902A
(24, 120000),   -- WS-43
(25, 150000),   -- WS-43A
(26, 180000),   -- WS-43B
(27, 300000),   -- WS-61
(28, 350000),   -- WS-61A
(29, 70000),    -- Kargu
(30, 85000),    -- Kargu-2
(31, 45000),    -- Alpagu
(32, 55000),    -- Alpagu Block-II
(33, 95000),    -- Kargu Autonomous
(34, 20000),    -- Shahed-131
(35, 25000),    -- Shahed-131B
(36, 40000),    -- Shahed-136
(37, 45000),    -- Shahed-136B
(38, 50000),    -- Shahed-136C
(39, 160000),   -- Green Dragon
(40, 200000),   -- Green Dragon Extended Range
(41, 180000),   -- Green Dragon Block 2
(42, 190000),   -- Green Dragon Maritime
(43, 140000),   -- Green Dragon-S
(44, 150000),   -- Phoenix Ghost
(45, 180000),   -- Phoenix Ghost Block I
(46, 220000),   -- Phoenix Ghost Block II
(47, 190000),   -- Phoenix Ghost Maritime
(48, 250000),   -- Phoenix Ghost-ER
(49, 80000),    -- Lancet-1
(50, 150000),   -- Lancet-3
(51, 180000),   -- Lancet-3M
(52, 250000),   -- Lancet-4
(53, 65000),    -- Rotem L
(54, 85000),    -- Rotem L-X
(55, 75000),   -- Rotem L-M
(56, 95000),   -- Rotem L-ER
(57, 95000),   -- KUB-BLA
(58, 120000),  -- KUB-BLA-E
(59, 110000),  -- KUB-BLA-M
(60, 150000);  -- KUB-BLA-ER

-- 创建数据集
INSERT INTO datasets (id, name, description, equipment_type, purpose) VALUES
(1, '巡飞弹训练集', '用于训练巡飞弹成本预测模型的数据集', '巡飞弹', '训练'),
(2, '巡飞弹验证集', '用于验证模型效果的数据集', '巡飞弹', '验证');

-- 关联装备到数据集（按照制造商和型号分配）
INSERT INTO dataset_equipment (dataset_id, equipment_id) VALUES
-- 训练集（约80%的数据，48个型号）
-- 以色列系列
(1, 1), (1, 2), (1, 3),  -- HAROP/Harpy系列
(1, 4), (1, 5), (1, 6),  -- Hero系列基础型号
(1, 39), (1, 40), (1, 41), (1, 42), (1, 43),  -- Green Dragon系列
(1, 53), (1, 54), (1, 55), (1, 56),  -- Rotem L系列

-- 美国系列
(1, 10), (1, 11), (1, 12), (1, 13),  -- Switchblade系列
(1, 44), (1, 45), (1, 46), (1, 47), (1, 48),  -- Phoenix Ghost系列

-- 中国系列
(1, 19), (1, 20), (1, 21), (1, 22), (1, 23),  -- CH-901/902系列
(1, 24), (1, 25), (1, 26), (1, 27), (1, 28),  -- WS-43/61系列

-- 波兰和土耳其系列
(1, 14), (1, 15), (1, 16), (1, 17), (1, 18),  -- Warmate系列
(1, 29), (1, 30), (1, 31), (1, 32), (1, 33),  -- Kargu/Alpagu系列

-- 俄罗斯系列
(1, 57), (1, 58), (1, 59), (1, 60),  -- KUB-BLA系列

-- 验证集（约20%的数据，12个型号）
-- 混合系列
(2, 7), (2, 8), (2, 9),  -- Hero系列高级型号
(2, 34), (2, 35), (2, 36), (2, 37), (2, 38),  -- Shahed系列
(2, 49), (2, 50), (2, 51), (2, 52);  -- ZALA Lancet系列

-- 添加分类特征编码
INSERT INTO feature_encoding (feature_type, feature_value, code) VALUES
-- 战斗部类型编码
('warhead_type', '破片杀伤战斗部', 1),
('warhead_type', '破甲战斗部', 2),
('warhead_type', '高爆战斗部', 3),
('warhead_type', '破片杀伤/破甲双用战斗部', 4),
('warhead_type', '模块化战斗部', 5),

-- 发射方式编码
('launch_mode', '箱式发射', 1),
('launch_mode', '弹射式发射', 2),
('launch_mode', '垂直起降', 3),
('launch_mode', '单兵发射管', 4),
('launch_mode', '箱式发射/弹射式', 5),
('launch_mode', '箱式发射/空中发射', 6),

-- 动力装置编码（按复杂度递增）
('power_system', '电动机', 1),
('power_system', '活塞发动机', 2),

-- 制导系统编码（按复杂度递增）
('guidance_system', 'GPS/INS', 1),
('guidance_system', 'GPS/INS/光电', 2),
('guidance_system', 'GPS/INS/光电/数据链', 3),
('guidance_system', 'GPS/INS/光电/AI识别', 4),
('guidance_system', 'GPS/INS/光电/数据链/AI辅助', 5),
('guidance_system', 'GPS/INS/光电/数据链/AI辅助/红外', 6),
('guidance_system', 'GPS/INS/光电/数据链/AI辅助/卫通', 7);

-- 更新巡飞弹特有参数表，添加新的关键参数和特征工程字段
UPDATE loitering_munition_params l
JOIN common_params c ON l.equipment_id = c.equipment_id
SET
    -- 新增关键参数
    l.payload_weight_kg = l.warhead_weight_kg * 1.2,  -- 有效载荷通常比战斗部重量大20%
    l.min_combat_radius_km = c.max_range_km * 0.1,    -- 最小作战半径约为最大航程的10%
    l.engine_power_kw = 
        CASE 
            WHEN l.power_system = '电动机' THEN c.weight_kg * 0.15
            WHEN l.power_system = '活塞发动机' THEN c.weight_kg * 0.25
        END,
    l.engine_thrust_n = c.weight_kg * 9.8 * 0.3,      -- 推力约为重量的30%
    l.datalink_range_km = c.max_range_km * 0.8,       -- 通信链路距离约为最大航程的80%
    l.guidance_accuracy_m = 
        CASE 
            WHEN INSTR(l.guidance_system, 'AI') > 0 THEN 1.0
            WHEN INSTR(l.guidance_system, '光电') > 0 THEN 2.0
            ELSE 3.0
        END,
    l.min_altitude_m =                                -- 最小作战高度
        CASE 
            -- 大型巡飞弹（体型大、重量大）
            WHEN equipment_id IN (1, 2, 34, 35, 36, 37, 38) THEN 150  -- HAROP/Harpy系列和 Shahed系列
            
            -- 中型巡飞弹
            WHEN equipment_id IN (3, 7, 8, 9, 27, 28) THEN 100  -- Mini Harpy和高端Hero系列, WS-61系列
            
            -- 中小型巡飞弹
            WHEN equipment_id IN (6, 11, 13, 15, 17, 22, 23, 24, 25, 26) THEN 80  -- Hero-120, Switchblade 600系列等
            
            -- 小型巡飞弹
            WHEN equipment_id IN (4, 5, 10, 12, 14, 16, 18, 19, 20, 21) THEN 50  -- Hero-30/70, Switchblade 300系列等
            
            -- 超小型巡飞弹
            WHEN equipment_id IN (29, 30, 31, 32, 33, 53, 54, 55, 56, 57, 58, 59, 60) THEN 30  -- Kargu/Alpagu系列, Rotem系列, KUB-BLA系列
            
            -- 其他型号使用默认值
            ELSE 50
        END,
    l.max_altitude_m = 
        CASE 
            WHEN c.max_range_km > 500 THEN 5000
            WHEN c.max_range_km > 100 THEN 3000
            ELSE 1500
        END,

    -- 特征工程字段
    l.length_width_ratio = c.length_m / c.width_m,
    l.weight_range_ratio = c.weight_kg / c.max_range_km,
    l.speed_weight_ratio = l.max_speed_ms / c.weight_kg,
    l.guidance_system_score = 
        CASE 
            WHEN INSTR(l.guidance_system, 'AI') > 0 AND INSTR(l.guidance_system, '卫通') > 0 THEN 10
            WHEN INSTR(l.guidance_system, 'AI') > 0 THEN 8
            WHEN INSTR(l.guidance_system, '数据链') > 0 THEN 6
            WHEN INSTR(l.guidance_system, '光电') > 0 THEN 4
            ELSE 2
        END,
    l.warhead_power_score = 
        CASE 
            WHEN l.warhead_type = '模块化战斗部' THEN 10
            WHEN l.warhead_type = '破片杀伤/破甲双用战斗部' THEN 8
            WHEN l.warhead_type = '高爆战斗部' THEN 7
            WHEN l.warhead_type = '破甲战斗部' THEN 6
            WHEN l.warhead_type = '破片杀伤战斗部' THEN 5
            ELSE 4
        END,

    -- 分类特征编码
    l.warhead_type_code = 
        CASE 
            WHEN l.warhead_type = '破片杀伤战斗部' THEN 1
            WHEN l.warhead_type = '破甲战斗部' THEN 2
            WHEN l.warhead_type = '高爆战斗部' THEN 3
            WHEN l.warhead_type = '破片杀伤/破甲双用战斗部' THEN 4
            WHEN l.warhead_type = '模块化战斗部' THEN 5
            ELSE 0
        END,
    l.launch_mode_code = 
        CASE 
            WHEN l.launch_mode = '箱式发射' THEN 1
            WHEN l.launch_mode = '弹射式发射' THEN 2
            WHEN l.launch_mode = '垂直起降' THEN 3
            WHEN l.launch_mode = '单兵发射管' THEN 4
            WHEN l.launch_mode = '箱式发射/弹射式' THEN 5
            WHEN l.launch_mode = '箱式发射/空中发射' THEN 6
            ELSE 0
        END,
    l.power_system_code = 
        CASE 
            WHEN l.power_system = '电动机' THEN 1
            WHEN l.power_system = '活塞发动机' THEN 2
            ELSE 0
        END,
    l.guidance_system_code = 
        CASE 
            WHEN l.guidance_system = 'GPS/INS' THEN 1
            WHEN l.guidance_system = 'GPS/INS/光电' THEN 2
            WHEN l.guidance_system = 'GPS/INS/光电/数据链' THEN 3
            WHEN l.guidance_system = 'GPS/INS/光电/AI识别' THEN 4
            WHEN l.guidance_system = 'GPS/INS/光电/数据链/AI辅助' THEN 5
            WHEN l.guidance_system = 'GPS/INS/光电/数据链/AI辅助/红外' THEN 6
            WHEN l.guidance_system = 'GPS/INS/光电/数据链/AI辅助/卫通' THEN 7
            ELSE 0
        END;
