1
This commit is contained in:
@@ -0,0 +1,597 @@
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user