297 lines
11 KiB
C#
297 lines
11 KiB
C#
// HediffCompProperties_SwitchableHediff.cs
|
||
using RimWorld;
|
||
using Verse;
|
||
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
using System.Text;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class HediffCompProperties_SwitchableHediff : HediffCompProperties
|
||
{
|
||
// 可切换的hediff列表
|
||
public List<HediffDef> availableHediffs = new List<HediffDef>();
|
||
|
||
// 默认选择的hediff索引
|
||
public int defaultHediffIndex = 0;
|
||
|
||
// Gizmo图标路径
|
||
public string gizmoIconPath = "UI/Commands/Default";
|
||
|
||
// 是否显示当前状态的提示
|
||
public bool showStatusInGizmo = true;
|
||
|
||
// 可自定义的标签和描述
|
||
public string switchLabel = "WULA_SwitchableHediff_SwitchLabel"; // 默认翻译键
|
||
public string switchDesc = "WULA_SwitchableHediff_SwitchDesc"; // 默认翻译键
|
||
public string statusLabel = "WULA_SwitchableHediff_StatusLabel"; // 默认翻译键
|
||
public string statusDesc = "WULA_SwitchableHediff_StatusDesc"; // 默认翻译键
|
||
|
||
public HediffCompProperties_SwitchableHediff()
|
||
{
|
||
compClass = typeof(HediffComp_SwitchableHediff);
|
||
}
|
||
}
|
||
|
||
public class HediffComp_SwitchableHediff : HediffComp
|
||
{
|
||
public HediffCompProperties_SwitchableHediff Props => (HediffCompProperties_SwitchableHediff)props;
|
||
|
||
// 当前选择的hediff索引
|
||
private int currentHediffIndex = -1;
|
||
|
||
// 当前激活的hediff引用
|
||
private Hediff activeHediff;
|
||
|
||
public override void CompPostMake()
|
||
{
|
||
base.CompPostMake();
|
||
|
||
// 初始化当前选择
|
||
if (currentHediffIndex == -1 && Props.availableHediffs.Count > 0)
|
||
{
|
||
currentHediffIndex = Props.defaultHediffIndex;
|
||
if (currentHediffIndex >= Props.availableHediffs.Count)
|
||
currentHediffIndex = 0;
|
||
|
||
// 应用初始hediff
|
||
ApplySelectedHediff();
|
||
}
|
||
}
|
||
|
||
// 应用当前选择的hediff
|
||
private void ApplySelectedHediff()
|
||
{
|
||
// 移除之前激活的hediff
|
||
if (activeHediff != null && Pawn.health.hediffSet.hediffs.Contains(activeHediff))
|
||
{
|
||
Pawn.health.RemoveHediff(activeHediff);
|
||
}
|
||
activeHediff = null;
|
||
|
||
// 应用新的hediff
|
||
if (currentHediffIndex >= 0 && currentHediffIndex < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[currentHediffIndex];
|
||
if (hediffDef != null)
|
||
{
|
||
activeHediff = HediffMaker.MakeHediff(hediffDef, Pawn);
|
||
activeHediff.Severity = 1f; // 默认严重性为1
|
||
Pawn.health.AddHediff(activeHediff);
|
||
|
||
Log.Message($"[SwitchableHediff] Applied {hediffDef.defName} to {Pawn.LabelShort}");
|
||
}
|
||
}
|
||
}
|
||
|
||
// 切换到特定索引的hediff
|
||
private void SwitchToHediff(int index)
|
||
{
|
||
if (index >= 0 && index < Props.availableHediffs.Count)
|
||
{
|
||
currentHediffIndex = index;
|
||
ApplySelectedHediff();
|
||
|
||
// 发送切换消息
|
||
var hediffDef = Props.availableHediffs[index];
|
||
if (hediffDef != null)
|
||
{
|
||
Messages.Message("WULA_SwitchableHediff_SwitchedTo".Translate(hediffDef.label),
|
||
Pawn, MessageTypeDefOf.SilentInput);
|
||
}
|
||
}
|
||
}
|
||
|
||
// 获取当前hediff名称
|
||
private string GetCurrentHediffName()
|
||
{
|
||
if (currentHediffIndex >= 0 && currentHediffIndex < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[currentHediffIndex];
|
||
return hediffDef?.label ?? "WULA_Unknown".Translate();
|
||
}
|
||
return "WULA_None".Translate();
|
||
}
|
||
|
||
// 获取当前hediff描述
|
||
private string GetCurrentHediffDesc()
|
||
{
|
||
if (currentHediffIndex >= 0 && currentHediffIndex < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[currentHediffIndex];
|
||
return hediffDef?.description ?? "WULA_NoDescription".Translate();
|
||
}
|
||
return string.Empty;
|
||
}
|
||
|
||
// 获取特定hediff的描述
|
||
private string GetHediffDescription(int index)
|
||
{
|
||
if (index >= 0 && index < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[index];
|
||
return hediffDef?.description ?? "WULA_NoDescription".Translate();
|
||
}
|
||
return string.Empty;
|
||
}
|
||
|
||
// 获取特定hediff的详细工具提示（包含效果信息）
|
||
private string GetHediffDetailedTooltip(int index)
|
||
{
|
||
if (index >= 0 && index < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[index];
|
||
if (hediffDef == null) return string.Empty;
|
||
|
||
StringBuilder sb = new StringBuilder();
|
||
|
||
// 添加描述
|
||
if (!hediffDef.description.NullOrEmpty())
|
||
{
|
||
sb.AppendLine(hediffDef.description);
|
||
}
|
||
return sb.ToString().TrimEndNewlines();
|
||
}
|
||
return string.Empty;
|
||
}
|
||
public override IEnumerable<Gizmo> CompGetGizmos()
|
||
{
|
||
// 只有玩家派系的pawn才显示Gizmo
|
||
if (Pawn.Faction == Faction.OfPlayer && Props.availableHediffs.Count > 1)
|
||
{
|
||
// 主切换按钮 - 显示当前状态
|
||
Command_Action mainButton = new Command_Action
|
||
{
|
||
// 使用可自定义的标签，如果没有自定义则使用翻译键
|
||
defaultLabel = Props.switchLabel.Translate(),
|
||
defaultDesc = "WULA_SwitchableHediff_CurrentMode".Translate(GetCurrentHediffName()) + "\n" +
|
||
Props.switchDesc.Translate(),
|
||
icon = ContentFinder<Texture2D>.Get(Props.gizmoIconPath, false) ?? BaseContent.BadTex,
|
||
action = () => {
|
||
// 显示选择菜单
|
||
ShowHediffSelectionMenu();
|
||
},
|
||
hotKey = KeyBindingDefOf.Misc2
|
||
};
|
||
|
||
yield return mainButton;
|
||
|
||
// 如果启用了状态显示，添加一个信息Gizmo
|
||
if (Props.showStatusInGizmo)
|
||
{
|
||
Command_Action statusButton = new Command_Action
|
||
{
|
||
defaultLabel = Props.statusLabel.Translate(GetCurrentHediffName()),
|
||
defaultDesc = Props.statusDesc.Translate() + "\n\n" +
|
||
"WULA_SwitchableHediff_CurrentDesc".Translate(GetCurrentHediffDesc()),
|
||
icon = ContentFinder<Texture2D>.Get("UI/Commands/Info", false) ?? BaseContent.BadTex,
|
||
action = () => {
|
||
// 显示当前hediff的详细信息
|
||
ShowCurrentHediffInfo();
|
||
}
|
||
};
|
||
|
||
yield return statusButton;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 显示hediff选择菜单
|
||
private void ShowHediffSelectionMenu()
|
||
{
|
||
List<FloatMenuOption> options = new List<FloatMenuOption>();
|
||
|
||
for (int i = 0; i < Props.availableHediffs.Count; i++)
|
||
{
|
||
int index = i; // 捕获当前索引
|
||
var hediffDef = Props.availableHediffs[i];
|
||
string label = hediffDef?.label ?? "WULA_Unknown".Translate();
|
||
string description = GetHediffDetailedTooltip(i); // 获取详细工具提示
|
||
|
||
// 标记当前选择的项目
|
||
string prefix = (i == currentHediffIndex) ? "✓ " : " ";
|
||
|
||
// 创建选项
|
||
var option = new FloatMenuOption(
|
||
prefix + label,
|
||
() => {
|
||
SwitchToHediff(index);
|
||
}
|
||
);
|
||
|
||
// 设置工具提示 - 使用详细的描述信息
|
||
option.tooltip = description;
|
||
|
||
options.Add(option);
|
||
}
|
||
|
||
// 显示浮动菜单
|
||
if (options.Count > 0)
|
||
{
|
||
Find.WindowStack.Add(new FloatMenu(options, "WULA_SwitchableHediff_SelectMode".Translate()));
|
||
}
|
||
}
|
||
|
||
// 显示当前hediff的详细信息
|
||
private void ShowCurrentHediffInfo()
|
||
{
|
||
if (currentHediffIndex >= 0 && currentHediffIndex < Props.availableHediffs.Count)
|
||
{
|
||
var hediffDef = Props.availableHediffs[currentHediffIndex];
|
||
string description = GetHediffDetailedTooltip(currentHediffIndex); // 使用详细工具提示
|
||
|
||
Messages.Message(
|
||
"WULA_SwitchableHediff_CurrentModeInfo".Translate(hediffDef?.label, description),
|
||
MessageTypeDefOf.SilentInput
|
||
);
|
||
}
|
||
}
|
||
|
||
public override string CompTipStringExtra
|
||
{
|
||
get
|
||
{
|
||
string baseTip = base.CompTipStringExtra ?? "";
|
||
string currentEffect = "WULA_SwitchableHediff_CurrentEffect".Translate(GetCurrentHediffName());
|
||
|
||
if (!string.IsNullOrEmpty(baseTip))
|
||
return baseTip + "\n" + currentEffect;
|
||
else
|
||
return currentEffect;
|
||
}
|
||
}
|
||
|
||
public override string CompLabelInBracketsExtra
|
||
{
|
||
get
|
||
{
|
||
return GetCurrentHediffName();
|
||
}
|
||
}
|
||
|
||
public override void CompExposeData()
|
||
{
|
||
base.CompExposeData();
|
||
Scribe_Values.Look(ref currentHediffIndex, "currentHediffIndex", -1);
|
||
|
||
// 加载后重新应用hediff
|
||
if (Scribe.mode == LoadSaveMode.PostLoadInit && currentHediffIndex != -1)
|
||
{
|
||
ApplySelectedHediff();
|
||
}
|
||
}
|
||
|
||
// 当父hediff被移除时，也要移除激活的hediff
|
||
public override void CompPostPostRemoved()
|
||
{
|
||
base.CompPostPostRemoved();
|
||
|
||
if (activeHediff != null && Pawn.health.hediffSet.hediffs.Contains(activeHediff))
|
||
{
|
||
Pawn.health.RemoveHediff(activeHediff);
|
||
}
|
||
activeHediff = null;
|
||
}
|
||
}
|
||
}
|