修质量

Source/ArachnaeSwarm/ArachnaeSwarm.csproj

@@ -88,6 +88,7 @@
   <ItemGroup>
     <Compile Include="CompWorkForNonMechs.cs" />
     <Compile Include="AnimalWorkSystemPatcher.cs" />
+    <Compile Include="Patch_QualityUtility.cs" />
   </ItemGroup>
   <ItemGroup>
     <Compile Include="CompProperties_DelayedTerrainSpawn.cs" />

Source/ArachnaeSwarm/Patch_QualityUtility.cs

@@ -0,0 +1,62 @@
+using HarmonyLib;
+using RimWorld;
+using Verse;
+using System.Reflection; // For MethodInfo
+
+namespace ArachnaeSwarm
+{
+    [StaticConstructorOnStartup]
+    public static class QualityUtilityPatch
+    {
+        static QualityUtilityPatch()
+        {
+            var harmony = new Harmony("com.yourname.qualityutilitypatch");
+            harmony.Patch(
+                original: AccessTools.Method(typeof(QualityUtility), nameof(QualityUtility.GenerateQualityCreatedByPawn), new[] { typeof(Pawn), typeof(SkillDef), typeof(bool) }),
+                prefix: new HarmonyMethod(typeof(QualityUtilityPatch), nameof(GenerateQualityCreatedByPawn_Prefix))
+            );
+        }
+
+        public static bool GenerateQualityCreatedByPawn_Prefix(Pawn pawn, SkillDef relevantSkill, bool consumeInspiration, ref QualityCategory __result)
+        {
+            // 检查当前 Pawn 是否是我们的自定义动物（通过检查其 ThingDef 是否拥有 CompProperties_WorkForNonMechs）
+            if (pawn != null && pawn.def.comps != null && ShouldEnableWorkSystem(pawn))
+            {
+                // 如果是，强制使用 mechFixedSkillLevel
+                int relevantSkillLevel = pawn.RaceProps.mechFixedSkillLevel;
+                bool inspired = consumeInspiration && pawn.InspirationDef == InspirationDefOf.Inspired_Creativity;
+
+                // 调用 QualityUtility.GenerateQualityCreatedByPawn 的 int 重载
+                __result = QualityUtility.GenerateQualityCreatedByPawn(relevantSkillLevel, inspired);
+
+                // 消耗灵感（如果适用）
+                if (inspired)
+                {
+                    pawn.mindState.inspirationHandler.EndInspiration(InspirationDefOf.Inspired_Creativity);
+                }
+
+                // 返回 false，跳过原版方法执行
+                return false;
+            }
+
+            // 返回 true，执行原版方法
+            return true;
+        }
+
+        private static bool ShouldEnableWorkSystem(Pawn pawn)
+        {
+            // 检查 ThingDef 中是否有 CompProperties_WorkForNonMechs 配置
+            if (pawn.def.comps != null)
+            {
+                foreach (var compProperties in pawn.def.comps)
+                {
+                    if (compProperties is CompProperties_WorkForNonMechs)
+                    {
+                        return true;
+                    }
+                }
+            }
+            return false;
+        }
+    }
+}