85 lines
3.0 KiB
C#
85 lines
3.0 KiB
C#
using HarmonyLib;
|
|
using RimWorld;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Verse;
|
|
|
|
namespace WulaFallenEmpire
|
|
{
|
|
[HarmonyPatch(typeof(ThingDefGenerator_Corpses))]
|
|
[HarmonyPatch("GenerateCorpseDef")]
|
|
public static class WulaSpeciesCorpsePatch
|
|
{
|
|
/// <summary>
|
|
/// 在生成尸体定义后，对 WulaSpecies 进行特殊处理
|
|
/// </summary>
|
|
[HarmonyPostfix]
|
|
public static void ModifyWulaSpeciesCorpse(ThingDef pawnDef, ref ThingDef __result)
|
|
{
|
|
// 检查是否是 WulaSpecies 种族
|
|
if (pawnDef?.defName == "WulaSpecies")
|
|
{
|
|
ApplyWulaSpeciesCorpseModifications(__result);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 应用 WulaSpecies 尸体的特殊修改
|
|
/// </summary>
|
|
private static void ApplyWulaSpeciesCorpseModifications(ThingDef corpseDef)
|
|
{
|
|
if (corpseDef == null) return;
|
|
|
|
Log.Message($"[WulaSpecies] Starting corpse modification for WulaSpecies");
|
|
|
|
// 1. 移除腐烂组件（如果存在）
|
|
RemoveCompProperties(corpseDef, typeof(CompProperties_Rottable));
|
|
|
|
// 2. 移除污物生成组件（如果存在）
|
|
RemoveCompProperties(corpseDef, typeof(CompProperties_SpawnerFilth));
|
|
|
|
// 3. 修改可食用属性，设置为 NeverForNutrition
|
|
if (corpseDef.ingestible != null)
|
|
{
|
|
corpseDef.ingestible.preferability = FoodPreferability.NeverForNutrition;
|
|
Log.Message($"[WulaSpecies] Set ingestible preferability to NeverForNutrition");
|
|
}
|
|
|
|
// 4. 移除 HarbingerTreeConsumable 组件（如果存在）
|
|
RemoveCompProperties(corpseDef, typeof(CompProperties), "CompHarbingerTreeConsumable");
|
|
|
|
Log.Message($"[WulaSpecies] Completed corpse modification for WulaSpecies");
|
|
}
|
|
|
|
/// <summary>
|
|
/// 移除指定类型的组件属性
|
|
/// </summary>
|
|
private static void RemoveCompProperties(ThingDef thingDef, System.Type compType, string compClassName = null)
|
|
{
|
|
if (thingDef.comps == null) return;
|
|
|
|
var compsToRemove = new List<CompProperties>();
|
|
|
|
foreach (var comp in thingDef.comps)
|
|
{
|
|
if (comp.GetType() == compType)
|
|
{
|
|
compsToRemove.Add(comp);
|
|
Log.Message($"[WulaSpecies] Found and will remove component: {comp.GetType().Name}");
|
|
}
|
|
else if (!string.IsNullOrEmpty(compClassName) && comp.compClass?.Name == compClassName)
|
|
{
|
|
compsToRemove.Add(comp);
|
|
Log.Message($"[WulaSpecies] Found and will remove component by class name: {compClassName}");
|
|
}
|
|
}
|
|
|
|
foreach (var comp in compsToRemove)
|
|
{
|
|
thingDef.comps.Remove(comp);
|
|
Log.Message($"[WulaSpecies] Removed component: {comp.GetType().Name}");
|
|
}
|
|
}
|
|
}
|
|
}
|