598 lines
21 KiB
C#
598 lines
21 KiB
C#
using RimWorld;
|
||
using Verse;
|
||
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class HediffCompProperties_NanoRepair : HediffCompProperties
|
||
{
|
||
public float activeSeverity = 0.5f; // 有能量且损伤时的严重性
|
||
public float inactiveSeverity = 1.5f; // 其他情况的严重性
|
||
public float minEnergyThreshold = 0.1f; // 最低能量阈值
|
||
public float repairCostPerHP = 0.03f; // 每点生命值修复的能量消耗
|
||
public int repairCooldownAfterDamage = 600; // 受到伤害后的修复冷却时间
|
||
// 新增：与 StatDef 的关联
|
||
public StatDef repairCostStatDef;
|
||
public StatDef cooldownStatDef;
|
||
public HediffCompProperties_NanoRepair()
|
||
{
|
||
compClass = typeof(HediffComp_NanoRepair);
|
||
|
||
// 设置默认的 StatDef 引用
|
||
repairCostStatDef = DefDatabase<StatDef>.GetNamedSilentFail("WULA_NanoRepairCostPerHP");
|
||
cooldownStatDef = DefDatabase<StatDef>.GetNamedSilentFail("WULA_NanoRepairCooldownAfterDamage");
|
||
}
|
||
}
|
||
|
||
public class HediffComp_NanoRepair : HediffComp
|
||
{
|
||
public HediffCompProperties_NanoRepair Props => (HediffCompProperties_NanoRepair)props;
|
||
|
||
private int lastDamageTick = -9999;
|
||
private const int CheckInterval = 60;
|
||
private int debugCounter = 0;
|
||
public bool repairSystemEnabled = true; // 默认开启修复系统
|
||
|
||
// 获取可用的能量源
|
||
private Need GetEnergyNeed()
|
||
{
|
||
if (Pawn?.needs == null) return null;
|
||
|
||
// 优先尝试 WULA_Energy
|
||
var wulaEnergy = Pawn.needs.TryGetNeed(DefDatabase<NeedDef>.GetNamedSilentFail("WULA_Energy"));
|
||
if (wulaEnergy != null)
|
||
{
|
||
return wulaEnergy;
|
||
}
|
||
|
||
// 回退到 MechEnergy
|
||
var mechEnergy = Pawn.needs?.TryGetNeed<Need_MechEnergy>();
|
||
if (mechEnergy != null)
|
||
{
|
||
return mechEnergy;
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
// 获取能量源名称（用于显示）
|
||
private string GetEnergyNeedName()
|
||
{
|
||
var energyNeed = GetEnergyNeed();
|
||
if (energyNeed == null) return "Unknown";
|
||
|
||
if (energyNeed.def.defName == "WULA_Energy")
|
||
return "WULA Energy";
|
||
else if (energyNeed is Need_MechEnergy)
|
||
return "Mech Energy";
|
||
else
|
||
return energyNeed.def.defName;
|
||
}
|
||
|
||
public override void CompPostTick(ref float severityAdjustment)
|
||
{
|
||
base.CompPostTick(ref severityAdjustment);
|
||
|
||
// 如果修复系统关闭，跳过所有修复逻辑
|
||
if (!repairSystemEnabled)
|
||
{
|
||
// 如果系统关闭，设置为不活跃状态
|
||
if (parent.Severity != Props.inactiveSeverity)
|
||
{
|
||
parent.Severity = Props.inactiveSeverity;
|
||
}
|
||
return;
|
||
}
|
||
|
||
// 每60 ticks检查一次状态
|
||
if (Find.TickManager.TicksGame % CheckInterval == 0)
|
||
{
|
||
debugCounter++;
|
||
UpdateSeverityAndRepair();
|
||
}
|
||
}
|
||
|
||
private void UpdateSeverityAndRepair()
|
||
{
|
||
if (Pawn == null || Pawn.Dead)
|
||
{
|
||
return;
|
||
}
|
||
|
||
bool shouldBeActive = ShouldBeActive();
|
||
float targetSeverity = shouldBeActive ? Props.activeSeverity : Props.inactiveSeverity;
|
||
|
||
// 更新严重性
|
||
if (parent.Severity != targetSeverity)
|
||
{
|
||
parent.Severity = targetSeverity;
|
||
}
|
||
|
||
// 如果处于活跃状态，执行修复
|
||
if (shouldBeActive)
|
||
{
|
||
TryRepairDamage();
|
||
}
|
||
}
|
||
|
||
private bool ShouldBeActive()
|
||
{
|
||
// 如果修复系统关闭，直接返回不活跃
|
||
if (!repairSystemEnabled)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 检查能量
|
||
var energyNeed = GetEnergyNeed();
|
||
if (energyNeed == null)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
if (energyNeed.CurLevelPercentage < Props.minEnergyThreshold)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 检查是否在冷却期内
|
||
int cooldownRemaining = ActualRepairCooldownAfterDamage - (Find.TickManager.TicksGame - lastDamageTick);
|
||
if (cooldownRemaining > 0)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 检查是否有需要修复的损伤
|
||
if (!HasDamageToRepair())
|
||
{
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
private bool HasDamageToRepair()
|
||
{
|
||
if (Pawn.health == null || Pawn.health.hediffSet == null)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 检查是否有缺失部件
|
||
var missingParts = Pawn.health.hediffSet.GetMissingPartsCommonAncestors();
|
||
if (missingParts.Count > 0)
|
||
{
|
||
return true;
|
||
}
|
||
|
||
// 检查是否有损伤
|
||
if (HasDamagedParts())
|
||
{
|
||
return true;
|
||
}
|
||
|
||
// 不再检查疾病
|
||
return false;
|
||
}
|
||
|
||
// 使用 GetPartHealth 检测损伤
|
||
private bool HasDamagedParts()
|
||
{
|
||
var bodyParts = Pawn.RaceProps.body.AllParts;
|
||
int damagedCount = 0;
|
||
|
||
foreach (var part in bodyParts)
|
||
{
|
||
// 如果部位不是缺失的，但健康值小于最大健康值，说明有损伤
|
||
if (!Pawn.health.hediffSet.PartIsMissing(part))
|
||
{
|
||
float maxHealth = part.def.GetMaxHealth(Pawn);
|
||
float currentHealth = Pawn.health.hediffSet.GetPartHealth(part);
|
||
|
||
// 不再使用修复容忍度，任何损伤都需要修复
|
||
if (currentHealth < maxHealth)
|
||
{
|
||
damagedCount++;
|
||
}
|
||
}
|
||
}
|
||
|
||
return damagedCount > 0;
|
||
}
|
||
|
||
// 获取所有需要修复的部位
|
||
private List<BodyPartRecord> GetDamagedParts()
|
||
{
|
||
var damagedParts = new List<BodyPartRecord>();
|
||
var bodyParts = Pawn.RaceProps.body.AllParts;
|
||
|
||
foreach (var part in bodyParts)
|
||
{
|
||
if (!Pawn.health.hediffSet.PartIsMissing(part))
|
||
{
|
||
float maxHealth = part.def.GetMaxHealth(Pawn);
|
||
float currentHealth = Pawn.health.hediffSet.GetPartHealth(part);
|
||
|
||
// 不再使用修复容忍度，任何损伤都需要修复
|
||
if (currentHealth < maxHealth)
|
||
{
|
||
damagedParts.Add(part);
|
||
}
|
||
}
|
||
}
|
||
|
||
return damagedParts;
|
||
}
|
||
|
||
private void TryRepairDamage()
|
||
{
|
||
var energyNeed = GetEnergyNeed();
|
||
if (energyNeed == null)
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 优先修复缺失部件
|
||
if (TryRepairMissingParts(energyNeed))
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 然后修复损伤
|
||
if (TryRepairDamagedParts(energyNeed))
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 不再修复疾病
|
||
}
|
||
|
||
private bool TryRepairMissingParts(Need energyNeed)
|
||
{
|
||
var missingParts = Pawn.health.hediffSet.GetMissingPartsCommonAncestors();
|
||
if (missingParts == null || missingParts.Count == 0)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 选择最小的缺失部件进行修复（成本较低）
|
||
Hediff_MissingPart partToRepair = null;
|
||
float minHealth = float.MaxValue;
|
||
|
||
foreach (var missingPart in missingParts)
|
||
{
|
||
float partHealth = missingPart.Part.def.GetMaxHealth(Pawn);
|
||
if (partHealth < minHealth)
|
||
{
|
||
minHealth = partHealth;
|
||
partToRepair = missingPart;
|
||
}
|
||
}
|
||
|
||
if (partToRepair != null)
|
||
{
|
||
// 计算修复成本 - 使用正常损伤的修复成本
|
||
float repairCost = minHealth * Props.repairCostPerHP;
|
||
|
||
// 根据机械族的能量消耗属性调整成本
|
||
var mechEnergyLoss = Pawn.GetStatValue(StatDefOf.MechEnergyLossPerHP);
|
||
if (mechEnergyLoss > 0)
|
||
{
|
||
repairCost *= mechEnergyLoss;
|
||
}
|
||
|
||
if (energyNeed.CurLevel >= repairCost)
|
||
{
|
||
if (ConvertMissingPartToInjury(partToRepair, repairCost))
|
||
{
|
||
energyNeed.CurLevel -= repairCost;
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
private bool TryRepairDamagedParts(Need energyNeed)
|
||
{
|
||
var damagedParts = GetDamagedParts();
|
||
if (damagedParts.Count == 0)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 选择健康值最低的部位进行修复
|
||
BodyPartRecord partToRepair = null;
|
||
float minHealthRatio = float.MaxValue;
|
||
|
||
foreach (var part in damagedParts)
|
||
{
|
||
float maxHealth = part.def.GetMaxHealth(Pawn);
|
||
float currentHealth = Pawn.health.hediffSet.GetPartHealth(part);
|
||
float healthRatio = currentHealth / maxHealth;
|
||
|
||
if (healthRatio < minHealthRatio)
|
||
{
|
||
minHealthRatio = healthRatio;
|
||
partToRepair = part;
|
||
}
|
||
}
|
||
|
||
if (partToRepair != null)
|
||
{
|
||
float maxHealth = partToRepair.def.GetMaxHealth(Pawn);
|
||
float currentHealth = Pawn.health.hediffSet.GetPartHealth(partToRepair);
|
||
float healthToRepair = maxHealth - currentHealth;
|
||
|
||
// 计算修复成本
|
||
float repairCost = healthToRepair * ActualRepairCostPerHP;
|
||
|
||
// 根据机械族的能量消耗属性调整成本
|
||
var mechEnergyLoss = Pawn.GetStatValue(StatDefOf.MechEnergyLossPerHP);
|
||
if (mechEnergyLoss > 0)
|
||
{
|
||
repairCost *= mechEnergyLoss;
|
||
}
|
||
|
||
if (energyNeed.CurLevel >= repairCost)
|
||
{
|
||
if (RepairDamagedPart(partToRepair, repairCost))
|
||
{
|
||
energyNeed.CurLevel -= repairCost;
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
// 新的修复逻辑：完美修复所有伤口
|
||
private bool RepairDamagedPart(BodyPartRecord part, float repairCost)
|
||
{
|
||
try
|
||
{
|
||
float maxHealth = part.def.GetMaxHealth(Pawn);
|
||
float currentHealth = Pawn.health.hediffSet.GetPartHealth(part);
|
||
|
||
// 获取该部位的所有hediff
|
||
var hediffsOnPart = new List<Hediff>();
|
||
foreach (var hediff in Pawn.health.hediffSet.hediffs)
|
||
{
|
||
if (hediff.Part == part)
|
||
{
|
||
hediffsOnPart.Add(hediff);
|
||
}
|
||
}
|
||
|
||
if (hediffsOnPart.Count == 0)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
bool anyRepairDone = false;
|
||
|
||
foreach (var hediff in hediffsOnPart)
|
||
{
|
||
// 检查hediff是否可修复
|
||
if (!CanRepairHediff(hediff))
|
||
{
|
||
continue;
|
||
}
|
||
|
||
// 新的修复逻辑：对于小于1的伤口，直接删除
|
||
if (hediff.Severity < 1.0f)
|
||
{
|
||
Pawn.health.RemoveHediff(hediff);
|
||
anyRepairDone = true;
|
||
}
|
||
else
|
||
{
|
||
// 对于大于等于1的伤口，完全修复
|
||
float originalSeverity = hediff.Severity;
|
||
hediff.Severity = 0f;
|
||
Pawn.health.RemoveHediff(hediff);
|
||
anyRepairDone = true;
|
||
}
|
||
}
|
||
|
||
return anyRepairDone;
|
||
}
|
||
catch (System.Exception ex)
|
||
{
|
||
WulaLog.Debug($"[NanoRepair] 修复部位 {part.def.defName} 时出错: {ex}");
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// 检查hediff是否可修复
|
||
private bool CanRepairHediff(Hediff hediff)
|
||
{
|
||
// 跳过疾病
|
||
if (IsDisease(hediff))
|
||
{
|
||
return false;
|
||
}
|
||
|
||
// 如果是机械族特有的hediff，可以修复
|
||
if (IsMechSpecificHediff(hediff))
|
||
return true;
|
||
|
||
// 如果是损伤类型的hediff，可以修复
|
||
if (hediff is Hediff_Injury)
|
||
return true;
|
||
|
||
// 其他情况不可修复
|
||
return false;
|
||
}
|
||
|
||
// 检查是否是疾病
|
||
private bool IsDisease(Hediff hediff)
|
||
{
|
||
// 这里可以定义哪些hediff被认为是疾病
|
||
// 常见的疾病类型
|
||
string[] diseaseKeywords = {
|
||
"Disease", "Flu", "Plague", "Infection", "Malaria",
|
||
"SleepingSickness", "FibrousMechanites", "SensoryMechanites",
|
||
"WoundInfection", "FoodPoisoning", "GutWorms", "MuscleParasites"
|
||
};
|
||
|
||
foreach (string keyword in diseaseKeywords)
|
||
{
|
||
if (hediff.def.defName.Contains(keyword))
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
// 将缺失部件转换为指定的hediff
|
||
private bool ConvertMissingPartToInjury(Hediff_MissingPart missingPart, float repairCost)
|
||
{
|
||
try
|
||
{
|
||
float partMaxHealth = missingPart.Part.def.GetMaxHealth(Pawn);
|
||
|
||
// 关键修复：确保转换后的损伤不会导致部位再次缺失
|
||
// 我们设置损伤严重性为最大健康值-1，这样部位健康值至少为1
|
||
float injurySeverity = partMaxHealth - 1;
|
||
|
||
// 如果最大健康值为1，则设置为0.5，确保部位健康值大于0
|
||
if (partMaxHealth <= 1)
|
||
{
|
||
injurySeverity = 0.5f;
|
||
}
|
||
|
||
// 移除缺失部件hediff
|
||
Pawn.health.RemoveHediff(missingPart);
|
||
|
||
// 添加指定的hediff (Crush)
|
||
HediffDef injuryDef = DefDatabase<HediffDef>.GetNamedSilentFail("Crush");
|
||
if (injuryDef == null)
|
||
{
|
||
WulaLog.Debug($"[NanoRepair] 找不到指定的hediff定义: Crush");
|
||
return false;
|
||
}
|
||
|
||
// 创建损伤
|
||
Hediff injury = HediffMaker.MakeHediff(injuryDef, Pawn, missingPart.Part);
|
||
injury.Severity = injurySeverity;
|
||
|
||
Pawn.health.AddHediff(injury);
|
||
|
||
return true;
|
||
}
|
||
catch (System.Exception ex)
|
||
{
|
||
WulaLog.Debug($"[NanoRepair] 转换缺失部件 {missingPart.Part.def.defName} 时出错: {ex}");
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// 检查是否是机械族特有的hediff（可能不可治疗但需要修复）
|
||
private bool IsMechSpecificHediff(Hediff hediff)
|
||
{
|
||
// 机械族的损伤可能不是标准的Hediff_Injury
|
||
// 这里可以根据需要添加更多机械族特有的hediff判断
|
||
return hediff.def.defName.Contains("Mech") ||
|
||
hediff.def.defName.Contains("Mechanical") ||
|
||
hediff.def.defName.Contains("Gunshot"); // 包括枪伤
|
||
}
|
||
|
||
public override void Notify_PawnPostApplyDamage(DamageInfo dinfo, float totalDamageDealt)
|
||
{
|
||
base.Notify_PawnPostApplyDamage(dinfo, totalDamageDealt);
|
||
|
||
// 记录最后一次受到伤害的时间
|
||
lastDamageTick = Find.TickManager.TicksGame;
|
||
}
|
||
|
||
// 新增：动态获取属性值的方法
|
||
public float ActualRepairCostPerHP
|
||
{
|
||
get
|
||
{
|
||
if (Props.repairCostStatDef != null && Pawn != null)
|
||
{
|
||
return Pawn.GetStatValue(Props.repairCostStatDef);
|
||
}
|
||
return Props.repairCostPerHP;
|
||
}
|
||
}
|
||
public int ActualRepairCooldownAfterDamage
|
||
{
|
||
get
|
||
{
|
||
if (Props.cooldownStatDef != null && Pawn != null)
|
||
{
|
||
return (int)Pawn.GetStatValue(Props.cooldownStatDef);
|
||
}
|
||
return Props.repairCooldownAfterDamage;
|
||
}
|
||
}
|
||
|
||
// 添加Gizmo（小工具）用于切换修复系统
|
||
public override IEnumerable<Gizmo> CompGetGizmos()
|
||
{
|
||
// 只有玩家派系的机械族才显示Gizmo
|
||
if (Pawn.Faction == Faction.OfPlayer)
|
||
{
|
||
Command_Toggle toggleCommand = new Command_Toggle
|
||
{
|
||
defaultLabel = repairSystemEnabled ? "WULA_NanoRepair_Disable".Translate() : "WULA_NanoRepair_Enable".Translate(),
|
||
defaultDesc = repairSystemEnabled ? "WULA_NanoRepair_DisableDesc".Translate() : "WULA_NanoRepair_EnableDesc".Translate(),
|
||
icon = ContentFinder<Texture2D>.Get("Wula/UI/Commands/WULA_NanoRepairHediff_Switch"),
|
||
isActive = () => repairSystemEnabled,
|
||
toggleAction = () => {
|
||
repairSystemEnabled = !repairSystemEnabled;
|
||
Messages.Message(
|
||
repairSystemEnabled ?
|
||
"WULA_NanoRepair_EnabledMsg".Translate(Pawn.LabelShort) :
|
||
"WULA_NanoRepair_DisabledMsg".Translate(Pawn.LabelShort),
|
||
MessageTypeDefOf.SilentInput
|
||
);
|
||
},
|
||
hotKey = KeyBindingDefOf.Misc1
|
||
};
|
||
|
||
yield return toggleCommand;
|
||
}
|
||
}
|
||
|
||
public override string CompTipStringExtra
|
||
{
|
||
get
|
||
{
|
||
string status = repairSystemEnabled ?
|
||
(ShouldBeActive() ?
|
||
"WULA_NanoRepair_Active".Translate() :
|
||
"WULA_NanoRepair_Inactive".Translate()) :
|
||
"WULA_NanoRepair_Disabled".Translate();
|
||
|
||
var energyNeed = GetEnergyNeed();
|
||
|
||
string damageInfo = HasDamageToRepair() ?
|
||
"WULA_DamageDetected".Translate() :
|
||
"WULA_NoDamage".Translate();
|
||
|
||
string cooldownInfo = "";
|
||
int cooldownRemaining = ActualRepairCooldownAfterDamage - (Find.TickManager.TicksGame - lastDamageTick);
|
||
if (cooldownRemaining > 0)
|
||
{
|
||
cooldownInfo = "\n" + "WULA_RepairCooldown".Translate((cooldownRemaining / 60f).ToString("F1"));
|
||
}
|
||
// 添加修复成本信息
|
||
string costInfo = "\n" + "WULA_RepairCostPerHP".Translate(ActualRepairCostPerHP.ToStringPercent());
|
||
return status + "\n" + damageInfo + cooldownInfo + costInfo;
|
||
}
|
||
}
|
||
|
||
public override void CompExposeData()
|
||
{
|
||
base.CompExposeData();
|
||
Scribe_Values.Look(ref lastDamageTick, "lastDamageTick", -9999);
|
||
Scribe_Values.Look(ref repairSystemEnabled, "repairSystemEnabled", true);
|
||
}
|
||
}
|
||
}
|