447 lines
15 KiB
C#
447 lines
15 KiB
C#
// CompMechFuel.cs
|
||
using RimWorld;
|
||
using System.Collections.Generic;
|
||
using UnityEngine;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using System.Linq;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
public class CompMechFuel : ThingComp
|
||
{
|
||
public CompProperties_MechFuel Props => (CompProperties_MechFuel)props;
|
||
|
||
private float fuel;
|
||
private bool isShutdown = false;
|
||
private int lastFuelTick = -1;
|
||
|
||
public float Fuel => fuel;
|
||
public float FuelPercent => fuel / Props.fuelCapacity;
|
||
public bool HasFuel => fuel > 0f;
|
||
public bool IsFull => FuelPercent >= 0.999f;
|
||
public bool NeedsRefueling => FuelPercent < Props.autoRefuelThreshold && Props.allowAutoRefuel;
|
||
public bool IsShutdown => isShutdown;
|
||
|
||
public ThingDef FuelType => Props.fuelType;
|
||
|
||
// 停机状态 Hediff
|
||
private HediffDef ShutdownHediffDef => HediffDef.Named("DD_MechShutdown");
|
||
|
||
public override void PostSpawnSetup(bool respawningAfterLoad)
|
||
{
|
||
base.PostSpawnSetup(respawningAfterLoad);
|
||
lastFuelTick = Find.TickManager.TicksGame;
|
||
|
||
// 如果是新生成的机甲，自动加满燃料
|
||
if (!respawningAfterLoad && fuel <= 0f)
|
||
{
|
||
Refuel(Props.fuelCapacity);
|
||
}
|
||
|
||
// 确保停机状态和 Hediff 同步
|
||
SyncShutdownHediff();
|
||
}
|
||
|
||
public override void CompTick()
|
||
{
|
||
base.CompTick();
|
||
|
||
// 每60ticks（1秒）消耗一次燃料
|
||
if (Find.TickManager.TicksGame % 60 == 0)
|
||
{
|
||
ConsumeFuelOverTime();
|
||
}
|
||
|
||
// 检查是否需要关机
|
||
CheckShutdown();
|
||
|
||
// 确保停机状态和 Hediff 同步
|
||
SyncShutdownHediff();
|
||
}
|
||
|
||
private void SyncShutdownHediff()
|
||
{
|
||
var mech = parent as Pawn;
|
||
if (mech == null || mech.health == null || ShutdownHediffDef == null)
|
||
return;
|
||
|
||
bool hasShutdownHediff = mech.health.hediffSet.HasHediff(ShutdownHediffDef);
|
||
|
||
// 如果处于停机状态但没有 Hediff，添加 Hediff
|
||
if (isShutdown && !hasShutdownHediff)
|
||
{
|
||
mech.health.AddHediff(ShutdownHediffDef);
|
||
}
|
||
// 如果不处于停机状态但有 Hediff，移除 Hediff
|
||
else if (!isShutdown && hasShutdownHediff)
|
||
{
|
||
var hediff = mech.health.hediffSet.GetFirstHediffOfDef(ShutdownHediffDef);
|
||
if (hediff != null)
|
||
{
|
||
mech.health.RemoveHediff(hediff);
|
||
}
|
||
}
|
||
}
|
||
|
||
private void ConsumeFuelOverTime()
|
||
{
|
||
if (fuel <= 0f || !parent.Spawned)
|
||
return;
|
||
|
||
// 检查是否有驾驶员 - 没有驾驶员时不消耗燃料
|
||
var pilotComp = parent.TryGetComp<CompMechPilotHolder>();
|
||
bool hasPilot = pilotComp != null && pilotComp.HasPilots;
|
||
|
||
if (!hasPilot)
|
||
return; // 没有驾驶员，不消耗燃料
|
||
|
||
// 获取当前时间
|
||
int currentTick = Find.TickManager.TicksGame;
|
||
|
||
// 计算经过的游戏时间（以天为单位）
|
||
float daysPassed = (currentTick - lastFuelTick) / 60000f; // 60000 ticks = 1天
|
||
|
||
// 计算基础燃料消耗
|
||
float consumption = Props.dailyFuelConsumption * daysPassed;
|
||
|
||
// 如果机甲正在活动（移动、战斗等），消耗更多燃料
|
||
var mech = parent as Pawn;
|
||
if (mech != null)
|
||
{
|
||
// 增加活动消耗
|
||
if (mech.pather.Moving)
|
||
{
|
||
consumption *= 2f; // 移动时消耗加倍
|
||
}
|
||
|
||
// 战斗状态消耗更多
|
||
if (mech.CurJob != null && mech.CurJob.def == JobDefOf.AttackStatic)
|
||
{
|
||
consumption *= 1.5f;
|
||
}
|
||
}
|
||
|
||
// 消耗燃料
|
||
fuel = Mathf.Max(0f, fuel - consumption);
|
||
|
||
// 更新最后消耗时间
|
||
lastFuelTick = currentTick;
|
||
}
|
||
|
||
private void CheckShutdown()
|
||
{
|
||
if (!Props.shutdownWhenEmpty || isShutdown)
|
||
return;
|
||
|
||
// 燃料耗尽时关机
|
||
if (fuel <= 0f && parent.Spawned)
|
||
{
|
||
Shutdown();
|
||
}
|
||
}
|
||
|
||
private void Shutdown()
|
||
{
|
||
if (isShutdown)
|
||
return;
|
||
|
||
isShutdown = true;
|
||
|
||
var mech = parent as Pawn;
|
||
if (mech != null)
|
||
{
|
||
// 取消所有工作
|
||
mech.jobs.StopAll();
|
||
mech.drafter.Drafted = false;
|
||
|
||
// 添加停机 Hediff
|
||
if (ShutdownHediffDef != null)
|
||
{
|
||
mech.health.AddHediff(ShutdownHediffDef);
|
||
}
|
||
|
||
// 播放关机效果
|
||
MoteMaker.ThrowText(mech.DrawPos, mech.Map, "DD_Shutdown".Translate(), Color.gray, 3.5f);
|
||
}
|
||
}
|
||
|
||
public void Startup()
|
||
{
|
||
if (!isShutdown || fuel <= 0f)
|
||
return;
|
||
|
||
isShutdown = false;
|
||
|
||
var mech = parent as Pawn;
|
||
if (mech != null)
|
||
{
|
||
// 移除停机 Hediff
|
||
if (ShutdownHediffDef != null)
|
||
{
|
||
var hediff = mech.health.hediffSet.GetFirstHediffOfDef(ShutdownHediffDef);
|
||
if (hediff != null)
|
||
{
|
||
mech.health.RemoveHediff(hediff);
|
||
}
|
||
}
|
||
|
||
MoteMaker.ThrowText(mech.DrawPos, mech.Map, "DD_Startup".Translate(), Color.green, 3.5f);
|
||
}
|
||
}
|
||
|
||
public bool TryConsumeFuel(float amount)
|
||
{
|
||
if (fuel >= amount)
|
||
{
|
||
fuel -= amount;
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
public bool Refuel(float amount)
|
||
{
|
||
float oldFuel = fuel;
|
||
fuel = Mathf.Min(Props.fuelCapacity, fuel + amount);
|
||
|
||
// 如果之前关机了且现在有燃料了，启动
|
||
if (isShutdown && fuel > 0f)
|
||
{
|
||
Startup();
|
||
}
|
||
|
||
return fuel > oldFuel;
|
||
}
|
||
|
||
// 设置燃料到特定值（测试用）
|
||
public void SetFuel(float amount)
|
||
{
|
||
float oldFuel = fuel;
|
||
fuel = Mathf.Clamp(amount, 0f, Props.fuelCapacity);
|
||
|
||
// 检查是否需要关机或启动
|
||
if (fuel <= 0f)
|
||
{
|
||
if (!isShutdown && Props.shutdownWhenEmpty)
|
||
{
|
||
Shutdown();
|
||
}
|
||
}
|
||
else if (isShutdown && fuel > 0f)
|
||
{
|
||
Startup();
|
||
}
|
||
|
||
// 发送调试消息
|
||
if (DebugSettings.godMode)
|
||
{
|
||
Messages.Message($"DD_Debug_FuelSet".Translate(
|
||
parent.LabelShort,
|
||
fuel.ToString("F1"),
|
||
Props.fuelCapacity.ToString("F1"),
|
||
(fuel / Props.fuelCapacity * 100f).ToString("F0") + "%"
|
||
), parent, MessageTypeDefOf.PositiveEvent);
|
||
}
|
||
}
|
||
|
||
public float FuelSpaceRemaining => Mathf.Max(0f, Props.fuelCapacity - fuel);
|
||
|
||
public int GetFuelCountToFullyRefuel()
|
||
{
|
||
if (FuelType == null)
|
||
return 0;
|
||
|
||
float fuelNeeded = FuelSpaceRemaining;
|
||
return Mathf.CeilToInt(fuelNeeded);
|
||
}
|
||
|
||
// 立刻加注 - 现在触发最近殖民者来加注
|
||
public void RefuelNow()
|
||
{
|
||
if (IsFull || parent.Map == null || FuelType == null)
|
||
return;
|
||
|
||
// 寻找最近的可用殖民者
|
||
Pawn bestColonist = FindBestColonistForRefuel();
|
||
|
||
if (bestColonist == null)
|
||
{
|
||
Messages.Message("DD_NoColonistAvailable".Translate(), parent, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 寻找燃料
|
||
Thing fuel = FindFuelForRefuel(bestColonist);
|
||
if (fuel == null)
|
||
{
|
||
Messages.Message("DD_NoFuelAvailable".Translate(FuelType), parent, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 为殖民者创建强制加注工作
|
||
Job job = JobMaker.MakeJob(Wula_JobDefOf.WULA_RefuelMech, parent, fuel);
|
||
job.count = GetFuelCountToFullyRefuel();
|
||
job.playerForced = true;
|
||
|
||
bestColonist.jobs.StartJob(job, JobCondition.InterruptForced, null, resumeCurJobAfterwards: true);
|
||
|
||
// 显示消息
|
||
Messages.Message("DD_OrderedRefuel".Translate(bestColonist.LabelShort, parent.LabelShort),
|
||
parent, MessageTypeDefOf.PositiveEvent);
|
||
}
|
||
|
||
private Pawn FindBestColonistForRefuel()
|
||
{
|
||
Map map = parent.Map;
|
||
if (map == null)
|
||
return null;
|
||
|
||
// 寻找所有可用的殖民者
|
||
List<Pawn> colonists = map.mapPawns.FreeColonists.ToList();
|
||
|
||
// 过滤掉无法工作或无法到达机甲的殖民者
|
||
colonists = colonists.Where(colonist =>
|
||
colonist.health.capacities.CapableOf(PawnCapacityDefOf.Manipulation) &&
|
||
colonist.health.capacities.CapableOf(PawnCapacityDefOf.Moving) &&
|
||
!colonist.Downed &&
|
||
!colonist.Dead &&
|
||
colonist.CanReserveAndReach(parent, PathEndMode.Touch, Danger.Some)
|
||
).ToList();
|
||
|
||
if (!colonists.Any())
|
||
return null;
|
||
|
||
// 按照距离排序，选择最近的殖民者
|
||
return colonists.OrderBy(colonist => colonist.Position.DistanceTo(parent.Position)).FirstOrDefault();
|
||
}
|
||
|
||
private Thing FindFuelForRefuel(Pawn colonist)
|
||
{
|
||
if (FuelType == null)
|
||
return null;
|
||
|
||
// 先在殖民者库存中寻找
|
||
if (colonist.inventory != null)
|
||
{
|
||
Thing fuelInInventory = colonist.inventory.innerContainer.FirstOrDefault(t => t.def == FuelType);
|
||
if (fuelInInventory != null)
|
||
return fuelInInventory;
|
||
}
|
||
|
||
// 在地图上寻找可用的燃料
|
||
return GenClosest.ClosestThingReachable(
|
||
colonist.Position,
|
||
colonist.Map,
|
||
ThingRequest.ForDef(FuelType),
|
||
PathEndMode.ClosestTouch,
|
||
TraverseParms.For(colonist),
|
||
9999f,
|
||
validator: thing => !thing.IsForbidden(colonist) && colonist.CanReserve(thing)
|
||
);
|
||
}
|
||
|
||
public bool CanRefuelNow()
|
||
{
|
||
if (IsFull || parent.Map == null || FuelType == null)
|
||
return false;
|
||
|
||
// 检查是否有可用殖民者
|
||
if (FindBestColonistForRefuel() == null)
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
// 检查机甲是否有驾驶员
|
||
public bool HasPilot()
|
||
{
|
||
var pilotComp = parent.TryGetComp<CompMechPilotHolder>();
|
||
return pilotComp != null && pilotComp.HasPilots;
|
||
}
|
||
|
||
public override IEnumerable<Gizmo> CompGetGizmosExtra()
|
||
{
|
||
// 添加燃料状态Gizmo
|
||
yield return new Gizmo_MechFuelStatus(this);
|
||
|
||
// 添加立刻加注按钮
|
||
if (!IsFull && parent.Faction == Faction.OfPlayer)
|
||
{
|
||
Command_Action refuelNow = new Command_Action
|
||
{
|
||
defaultLabel = "DD_RefuelNow".Translate(),
|
||
defaultDesc = "DD_RefuelNowDesc".Translate(),
|
||
icon = ContentFinder<Texture2D>.Get("WulaFallenEmpire/UI/Commands/DD_Refuel_Mech"),
|
||
action = () => RefuelNow()
|
||
};
|
||
|
||
// 检查是否可以立刻加注
|
||
if (!CanRefuelNow())
|
||
{
|
||
refuelNow.Disable("DD_CannotRefuelNow".Translate());
|
||
}
|
||
|
||
yield return refuelNow;
|
||
}
|
||
|
||
// 在 God Mode 下显示测试按钮
|
||
if (DebugSettings.godMode && parent.Faction == Faction.OfPlayer)
|
||
{
|
||
// 设置燃料为空
|
||
Command_Action setEmpty = new Command_Action
|
||
{
|
||
defaultLabel = "DD_Debug_SetEmpty".Translate(),
|
||
defaultDesc = "DD_Debug_SetEmptyDesc".Translate(),
|
||
icon = ContentFinder<Texture2D>.Get("UI/Commands/SetEmpty", false) ?? BaseContent.BadTex,
|
||
action = () => SetFuel(0f)
|
||
};
|
||
yield return setEmpty;
|
||
|
||
// 设置燃料为50%
|
||
Command_Action setHalf = new Command_Action
|
||
{
|
||
defaultLabel = "DD_Debug_SetHalf".Translate(),
|
||
defaultDesc = "DD_Debug_SetHalfDesc".Translate(),
|
||
icon = ContentFinder<Texture2D>.Get("UI/Commands/SetHalf", false) ?? BaseContent.BadTex,
|
||
action = () => SetFuel(Props.fuelCapacity * 0.5f)
|
||
};
|
||
yield return setHalf;
|
||
|
||
// 设置燃料为满
|
||
Command_Action setFull = new Command_Action
|
||
{
|
||
defaultLabel = "DD_Debug_SetFull".Translate(),
|
||
defaultDesc = "DD_Debug_SetFullDesc".Translate(),
|
||
icon = ContentFinder<Texture2D>.Get("UI/Commands/SetFull", false) ?? BaseContent.BadTex,
|
||
action = () => SetFuel(Props.fuelCapacity)
|
||
};
|
||
yield return setFull;
|
||
}
|
||
}
|
||
|
||
public override void PostExposeData()
|
||
{
|
||
base.PostExposeData();
|
||
|
||
Scribe_Values.Look(ref fuel, "fuel", Props.fuelCapacity);
|
||
Scribe_Values.Look(ref isShutdown, "isShutdown", false);
|
||
Scribe_Values.Look(ref lastFuelTick, "lastFuelTick", -1);
|
||
}
|
||
|
||
public override string CompInspectStringExtra()
|
||
{
|
||
string baseString = base.CompInspectStringExtra();
|
||
|
||
string fuelString = "DD_Fuel".Translate(FuelType) + ": " +
|
||
fuel.ToString("F1") + " / " + Props.fuelCapacity.ToString("F1") +
|
||
" (" + (FuelPercent * 100f).ToString("F0") + "%)";
|
||
|
||
if (!baseString.NullOrEmpty())
|
||
return baseString + "\n" + fuelString;
|
||
else
|
||
return fuelString;
|
||
}
|
||
}
|
||
}
|