zc

Source/WulaFallenEmpire/ThingComp/WULA_MechSpecificWeapon/CompMechSpecificWeapon.cs

@@ -0,0 +1,39 @@
+// File: CompMechOnlyWeapon.cs
+using System.Collections.Generic;
+using Verse;
+
+namespace WulaFallenEmpire
+{
+    /// <summary>
+    /// 简单的机甲专用武器组件
+    /// </summary>
+    public class CompMechOnlyWeapon : ThingComp
+    {
+        public List<ThingDef> allowedMechRaces;
+        
+        public override void Initialize(CompProperties props)
+        {
+            base.Initialize(props);
+            allowedMechRaces = ((CompProperties_MechOnlyWeapon)props).allowedMechRaces;
+        }
+        
+        /// <summary>
+        /// 检查机甲是否可以装备此武器
+        /// </summary>
+        public bool CanBeEquippedByMech(Pawn mech)
+        {
+            if (mech == null || mech.def == null) return false;
+            return allowedMechRaces != null && allowedMechRaces.Contains(mech.def);
+        }
+    }
+    
+    public class CompProperties_MechOnlyWeapon : CompProperties
+    {
+        public List<ThingDef> allowedMechRaces = new List<ThingDef>();
+        
+        public CompProperties_MechOnlyWeapon()
+        {
+            compClass = typeof(CompMechOnlyWeapon);
+        }
+    }
+}