1
This commit is contained in:
@@ -15,16 +15,25 @@ namespace WulaFallenEmpire
|
||||
public SoundDef enterSound;
|
||||
public SoundDef exitSound;
|
||||
public EffecterDef operatingEffecter;
|
||||
public int baseDurationTicks = 60000;
|
||||
public float ticksPerSeverity = 0f;
|
||||
|
||||
// 时间相关
|
||||
public int baseDurationTicks = 60000; // 基础维护时间(1天)
|
||||
public float ticksPerNeedLevel = 120000f; // 每点需求降低需要的时间
|
||||
|
||||
// 电力消耗
|
||||
public float powerConsumptionRunning = 250f;
|
||||
public float powerConsumptionIdle = 50f;
|
||||
public HediffDef hediffToRemove;
|
||||
public float componentCostPerSeverity = 1f;
|
||||
public int baseComponentCost = 0;
|
||||
public float minSeverityToMaintain = 0.75f;
|
||||
public float hediffSeverityAfterCycle = 0.01f;
|
||||
|
||||
// 组件消耗
|
||||
public float componentCostPerNeedLevel = 2f;
|
||||
public int baseComponentCost = 1;
|
||||
|
||||
// 维护效果
|
||||
public float minNeedLevelToMaintain = 0.3f; // 低于此值才需要维护
|
||||
public float needLevelAfterCycle = 1.0f; // 维护后的需求水平
|
||||
public bool healInjuries = true; // 是否治疗损伤
|
||||
public bool healMissingParts = true; // 是否修复缺失部位
|
||||
public int maxInjuriesHealedPerCycle = 5; // 每次维护最多治疗的损伤数量
|
||||
public CompProperties_MaintenancePod()
|
||||
{
|
||||
compClass = typeof(CompMaintenancePod);
|
||||
@@ -40,45 +49,55 @@ namespace WulaFallenEmpire
|
||||
private CompRefuelable refuelableComp;
|
||||
private int ticksRemaining;
|
||||
private MaintenancePodState state = MaintenancePodState.Idle;
|
||||
|
||||
private Effecter operatingEffecter;
|
||||
private static readonly Texture2D CancelIcon = ContentFinder<Texture2D>.Get("UI/Designators/Cancel");
|
||||
private static readonly Texture2D EnterIcon = ContentFinder<Texture2D>.Get("UI/Commands/PodEject");
|
||||
|
||||
// ===================== Properties =====================
|
||||
public CompProperties_MaintenancePod Props => (CompProperties_MaintenancePod)props;
|
||||
public MaintenancePodState State => state;
|
||||
public Pawn Occupant => innerContainer.FirstOrDefault() as Pawn;
|
||||
public bool PowerOn => powerComp != null && powerComp.PowerOn;
|
||||
|
||||
public float RequiredComponents(Pawn pawn)
|
||||
public float RequiredComponents
|
||||
{
|
||||
if (pawn == null || Props.hediffToRemove == null) return Props.baseComponentCost;
|
||||
Hediff hediff = pawn.health.hediffSet.GetFirstHediffOfDef(Props.hediffToRemove);
|
||||
if (hediff == null) return Props.baseComponentCost;
|
||||
return Props.baseComponentCost + (int)(hediff.Severity * Props.componentCostPerSeverity);
|
||||
}
|
||||
get
|
||||
{
|
||||
var occupant = Occupant;
|
||||
if (occupant == null) return Props.baseComponentCost;
|
||||
|
||||
public int RequiredDuration(Pawn pawn)
|
||||
var maintenanceNeed = occupant.needs?.TryGetNeed<Need_Maintenance>();
|
||||
if (maintenanceNeed == null) return Props.baseComponentCost;
|
||||
|
||||
// 计算基于当前需求水平的组件需求
|
||||
float needDeficit = 1.0f - maintenanceNeed.CurLevel;
|
||||
return Props.baseComponentCost + (needDeficit * Props.componentCostPerNeedLevel);
|
||||
}
|
||||
}
|
||||
public int RequiredDuration
|
||||
{
|
||||
if (pawn == null || Props.hediffToRemove == null) return Props.baseDurationTicks;
|
||||
Hediff hediff = pawn.health.hediffSet.GetFirstHediffOfDef(Props.hediffToRemove);
|
||||
if (hediff == null) return Props.baseDurationTicks;
|
||||
return Props.baseDurationTicks + (int)(hediff.Severity * Props.ticksPerSeverity);
|
||||
}
|
||||
get
|
||||
{
|
||||
var occupant = Occupant;
|
||||
if (occupant == null) return Props.baseDurationTicks;
|
||||
|
||||
var maintenanceNeed = occupant.needs?.TryGetNeed<Need_Maintenance>();
|
||||
if (maintenanceNeed == null) return Props.baseDurationTicks;
|
||||
|
||||
// 计算基于当前需求水平的维护时间
|
||||
float needDeficit = 1.0f - maintenanceNeed.CurLevel;
|
||||
return Props.baseDurationTicks + (int)(needDeficit * Props.ticksPerNeedLevel);
|
||||
}
|
||||
}
|
||||
// ===================== Setup =====================
|
||||
public CompMaintenancePod()
|
||||
{
|
||||
innerContainer = new ThingOwner<Thing>(this, false, LookMode.Deep);
|
||||
}
|
||||
|
||||
public override void PostSpawnSetup(bool respawningAfterLoad)
|
||||
{
|
||||
base.PostSpawnSetup(respawningAfterLoad);
|
||||
powerComp = parent.TryGetComp<CompPowerTrader>();
|
||||
refuelableComp = parent.TryGetComp<CompRefuelable>();
|
||||
}
|
||||
|
||||
public override void PostExposeData()
|
||||
{
|
||||
base.PostExposeData();
|
||||
@@ -86,361 +105,335 @@ namespace WulaFallenEmpire
|
||||
Scribe_Values.Look(ref ticksRemaining, "ticksRemaining", 0);
|
||||
Scribe_Deep.Look(ref innerContainer, "innerContainer", this);
|
||||
}
|
||||
|
||||
public override void PostDestroy(DestroyMode mode, Map previousMap)
|
||||
{
|
||||
base.PostDestroy(mode, previousMap);
|
||||
// If the pod is deconstructed or destroyed, eject the occupant to prevent deletion.
|
||||
if (mode == DestroyMode.Deconstruct || mode == DestroyMode.KillFinalize)
|
||||
{
|
||||
Log.Warning($"[WulaPodDebug] Pod destroyed (mode: {mode}). Ejecting pawn.");
|
||||
EjectPawn();
|
||||
}
|
||||
}
|
||||
|
||||
// ===================== IThingHolder Implementation =====================
|
||||
public void GetChildHolders(List<IThingHolder> outChildren)
|
||||
{
|
||||
ThingOwnerUtility.AppendThingHoldersFromThings(outChildren, GetDirectlyHeldThings());
|
||||
}
|
||||
|
||||
public ThingOwner GetDirectlyHeldThings()
|
||||
{
|
||||
return innerContainer;
|
||||
}
|
||||
|
||||
// ===================== Core Logic =====================
|
||||
public override void CompTick()
|
||||
{
|
||||
base.CompTick();
|
||||
if (!parent.Spawned) return;
|
||||
|
||||
if (state == MaintenancePodState.Running)
|
||||
{
|
||||
if (PowerOn)
|
||||
{
|
||||
ticksRemaining--;
|
||||
if (ticksRemaining <= 0)
|
||||
{
|
||||
CycleFinished();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 更新电力消耗
|
||||
if (powerComp != null)
|
||||
{
|
||||
powerComp.PowerOutput = -(state == MaintenancePodState.Running ? Props.powerConsumptionRunning : Props.powerConsumptionIdle);
|
||||
}
|
||||
}
|
||||
// 运行维护周期
|
||||
if (state == MaintenancePodState.Running && PowerOn)
|
||||
{
|
||||
ticksRemaining--;
|
||||
|
||||
// 更新效果器
|
||||
if (Props.operatingEffecter != null)
|
||||
{
|
||||
if (operatingEffecter == null)
|
||||
{
|
||||
operatingEffecter = Props.operatingEffecter.Spawn();
|
||||
}
|
||||
operatingEffecter.EffectTick(parent, Occupant);
|
||||
}
|
||||
if (ticksRemaining <= 0)
|
||||
{
|
||||
CycleFinished();
|
||||
}
|
||||
}
|
||||
else if (operatingEffecter != null)
|
||||
{
|
||||
operatingEffecter.Cleanup();
|
||||
operatingEffecter = null;
|
||||
}
|
||||
}
|
||||
public void StartCycle(Pawn pawn)
|
||||
{
|
||||
Log.Warning($"[WulaPodDebug] StartCycle called for pawn: {pawn.LabelShortCap}");
|
||||
float required = RequiredComponents(pawn);
|
||||
if (refuelableComp.Fuel < required)
|
||||
if (pawn == null) return;
|
||||
// 检查组件是否足够
|
||||
float requiredComponents = RequiredComponents;
|
||||
if (refuelableComp.Fuel < requiredComponents)
|
||||
{
|
||||
Log.Error($"[WulaPodDebug] ERROR: Tried to start cycle for {pawn.LabelShort} without enough components.");
|
||||
Messages.Message("WULA_MaintenancePod_NotEnoughComponents".Translate(requiredComponents.ToString("F0")), MessageTypeDefOf.RejectInput);
|
||||
return;
|
||||
}
|
||||
|
||||
if (required > 0)
|
||||
// 消耗组件
|
||||
if (requiredComponents > 0)
|
||||
{
|
||||
refuelableComp.ConsumeFuel(required);
|
||||
refuelableComp.ConsumeFuel(requiredComponents);
|
||||
}
|
||||
|
||||
Log.Warning($"[WulaPodDebug] Pawn state before action: holdingOwner is {(pawn.holdingOwner == null ? "NULL" : "NOT NULL")}, Spawned is {pawn.Spawned}");
|
||||
|
||||
// THE ACTUAL FIX: A pawn, whether held or not, must be despawned before being put in a container.
|
||||
// 将 pawn 放入容器
|
||||
if (pawn.Spawned)
|
||||
{
|
||||
Log.Warning($"[WulaPodDebug] Pawn is spawned. Despawning...");
|
||||
pawn.DeSpawn(DestroyMode.Vanish);
|
||||
}
|
||||
Log.Warning($"[WulaPodDebug] Attempting to add/transfer pawn to container.");
|
||||
innerContainer.TryAddOrTransfer(pawn);
|
||||
|
||||
|
||||
// 开始维护周期
|
||||
state = MaintenancePodState.Running;
|
||||
ticksRemaining = RequiredDuration(pawn);
|
||||
Log.Warning($"[WulaPodDebug] Cycle started. Ticks remaining: {ticksRemaining}");
|
||||
}
|
||||
ticksRemaining = RequiredDuration;
|
||||
|
||||
// 播放进入音效
|
||||
if (Props.enterSound != null)
|
||||
{
|
||||
Props.enterSound.PlayOneShot(new TargetInfo(parent.Position, parent.Map));
|
||||
}
|
||||
Messages.Message("WULA_MaintenanceCycleStarted".Translate(pawn.LabelShortCap), MessageTypeDefOf.PositiveEvent);
|
||||
}
|
||||
private void CycleFinished()
|
||||
{
|
||||
Pawn occupant = Occupant;
|
||||
Log.Warning($"[WulaPodDebug] CycleFinished. Occupant: {(occupant == null ? "NULL" : occupant.LabelShortCap)}");
|
||||
var occupant = Occupant;
|
||||
if (occupant == null)
|
||||
{
|
||||
Log.Error("[WulaPodDebug] ERROR: Maintenance cycle finished, but no one was inside.");
|
||||
state = MaintenancePodState.Idle;
|
||||
return;
|
||||
}
|
||||
// 执行维护效果
|
||||
PerformMaintenanceEffects(occupant);
|
||||
|
||||
// 1. Fix the maintenance hediff
|
||||
bool maintenanceDone = false;
|
||||
if (Props.hediffToRemove != null)
|
||||
// 弹出 pawn
|
||||
EjectPawn();
|
||||
|
||||
Messages.Message("WULA_MaintenanceCycleComplete".Translate(occupant.LabelShortCap), MessageTypeDefOf.PositiveEvent);
|
||||
}
|
||||
private void PerformMaintenanceEffects(Pawn pawn)
|
||||
{
|
||||
var maintenanceNeed = pawn.needs?.TryGetNeed<Need_Maintenance>();
|
||||
|
||||
// 1. 恢复维护需求
|
||||
if (maintenanceNeed != null)
|
||||
{
|
||||
Hediff hediff = occupant.health.hediffSet.GetFirstHediffOfDef(Props.hediffToRemove);
|
||||
if (hediff != null)
|
||||
{
|
||||
hediff.Severity = Props.hediffSeverityAfterCycle;
|
||||
Messages.Message("WULA_MaintenanceComplete".Translate(occupant.Named("PAWN")), occupant, MessageTypeDefOf.PositiveEvent);
|
||||
maintenanceDone = true;
|
||||
}
|
||||
maintenanceNeed.PerformMaintenance(Props.needLevelAfterCycle);
|
||||
}
|
||||
|
||||
// 2. Heal all other injuries and missing parts
|
||||
// 2. 治疗损伤(如果启用)
|
||||
if (Props.healInjuries)
|
||||
{
|
||||
HealInjuries(pawn);
|
||||
}
|
||||
// 3. 修复缺失部位(如果启用)
|
||||
if (Props.healMissingParts)
|
||||
{
|
||||
HealMissingParts(pawn);
|
||||
}
|
||||
}
|
||||
private void HealInjuries(Pawn pawn)
|
||||
{
|
||||
int injuriesHealed = 0;
|
||||
// Loop until no more health conditions can be fixed
|
||||
while (HealthUtility.TryGetWorstHealthCondition(occupant, out var hediffToFix, out var _))
|
||||
var injuries = pawn.health.hediffSet.hediffs
|
||||
.Where(h => h.def.isBad && h.Visible && h.def != HediffDefOf.BloodLoss)
|
||||
.ToList();
|
||||
foreach (var injury in injuries)
|
||||
{
|
||||
// Ensure we don't try to "heal" the maintenance hediff itself, as it's handled separately.
|
||||
if (hediffToFix != null && hediffToFix.def == Props.hediffToRemove)
|
||||
{
|
||||
if (injuriesHealed >= Props.maxInjuriesHealedPerCycle)
|
||||
break;
|
||||
}
|
||||
|
||||
// Store the state before attempting to fix
|
||||
int initialHediffCount = occupant.health.hediffSet.hediffs.Count;
|
||||
var hediffsBefore = new HashSet<Hediff>(occupant.health.hediffSet.hediffs);
|
||||
|
||||
// Attempt to fix the worst condition
|
||||
HealthUtility.FixWorstHealthCondition(occupant);
|
||||
|
||||
// Check if a change actually occurred
|
||||
bool conditionFixed = initialHediffCount > occupant.health.hediffSet.hediffs.Count ||
|
||||
!hediffsBefore.SetEquals(occupant.health.hediffSet.hediffs);
|
||||
|
||||
if (conditionFixed)
|
||||
{
|
||||
injuriesHealed++;
|
||||
}
|
||||
else
|
||||
{
|
||||
// If FixWorstHealthCondition did nothing, it means it can't handle
|
||||
// the current worst condition. We must break to avoid an infinite loop.
|
||||
Log.Warning($"[WulaPodDebug] Halting healing loop. FixWorstHealthCondition did not resolve: {hediffToFix?.LabelCap ?? "a missing part"}.");
|
||||
break;
|
||||
}
|
||||
pawn.health.RemoveHediff(injury);
|
||||
injuriesHealed++;
|
||||
}
|
||||
|
||||
if (injuriesHealed > 0)
|
||||
{
|
||||
Messages.Message("WULA_MaintenanceHealedAllWounds".Translate(occupant.Named("PAWN")), occupant, MessageTypeDefOf.PositiveEvent);
|
||||
Messages.Message("WULA_MaintenanceHealedInjuries".Translate(pawn.LabelShortCap, injuriesHealed), MessageTypeDefOf.PositiveEvent);
|
||||
}
|
||||
}
|
||||
private void HealMissingParts(Pawn pawn)
|
||||
{
|
||||
var missingParts = pawn.health.hediffSet.GetMissingPartsCommonAncestors();
|
||||
int partsHealed = 0;
|
||||
foreach (var missingPart in missingParts)
|
||||
{
|
||||
if (partsHealed >= 1) // 每次最多修复一个缺失部位
|
||||
break;
|
||||
pawn.health.RemoveHediff(missingPart);
|
||||
partsHealed++;
|
||||
}
|
||||
if (partsHealed > 0)
|
||||
{
|
||||
Messages.Message("WULA_MaintenanceHealedParts".Translate(pawn.LabelShortCap), MessageTypeDefOf.PositiveEvent);
|
||||
}
|
||||
else if (!maintenanceDone)
|
||||
{
|
||||
// If nothing was done at all, give a neutral message
|
||||
Messages.Message("WULA_MaintenanceNoEffect".Translate(occupant.Named("PAWN")), occupant, MessageTypeDefOf.NeutralEvent);
|
||||
}
|
||||
|
||||
EjectPawn();
|
||||
}
|
||||
|
||||
public void EjectPawn(bool interrupted = false)
|
||||
{
|
||||
Pawn occupant = Occupant;
|
||||
Log.Warning($"[WulaPodDebug] EjectPawn. Occupant: {(occupant == null ? "NULL" : occupant.LabelShortCap)}");
|
||||
var occupant = Occupant;
|
||||
if (occupant != null)
|
||||
{
|
||||
Map mapToUse = parent.Map ?? Find.CurrentMap;
|
||||
if (mapToUse == null)
|
||||
// 弹出到交互单元格
|
||||
innerContainer.TryDropAll(parent.InteractionCell, parent.Map, ThingPlaceMode.Near);
|
||||
|
||||
// 播放退出音效
|
||||
if (Props.exitSound != null)
|
||||
{
|
||||
// Try to find the map from nearby things
|
||||
mapToUse = GenClosest.ClosestThing_Global(occupant.Position, Gen.YieldSingle(parent), 99999f, (thing) => thing.Map != null)?.Map;
|
||||
Props.exitSound.PlayOneShot(new TargetInfo(parent.Position, parent.Map));
|
||||
}
|
||||
|
||||
if (mapToUse != null)
|
||||
{
|
||||
innerContainer.TryDropAll(parent.InteractionCell, mapToUse, ThingPlaceMode.Near);
|
||||
if (Props.exitSound != null)
|
||||
{
|
||||
SoundStarter.PlayOneShot(Props.exitSound, new TargetInfo(parent.Position, mapToUse));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Log.Error($"[WulaPodDebug] EjectPawn FAILED: No valid map found to eject {occupant.LabelShortCap}. The pawn will be lost.");
|
||||
}
|
||||
|
||||
// Additional logic to handle occupant if needed
|
||||
// 如果被中断,应用负面效果
|
||||
if (interrupted)
|
||||
{
|
||||
occupant.needs?.mood.thoughts.memories.TryGainMemory(ThoughtDefOf.SoakingWet);
|
||||
occupant.health?.AddHediff(HediffDefOf.BiosculptingSickness);
|
||||
occupant.needs?.mood?.thoughts?.memories?.TryGainMemory(ThoughtDefOf.SoakingWet);
|
||||
}
|
||||
}
|
||||
innerContainer.Clear();
|
||||
state = MaintenancePodState.Idle;
|
||||
Log.Warning($"[WulaPodDebug] EjectPawn finished. State set to Idle.");
|
||||
}
|
||||
|
||||
// 清理效果器
|
||||
if (operatingEffecter != null)
|
||||
{
|
||||
operatingEffecter.Cleanup();
|
||||
operatingEffecter = null;
|
||||
}
|
||||
}
|
||||
// ===================== UI & Gizmos =====================
|
||||
public override string CompInspectStringExtra()
|
||||
{
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.AppendLine("WULA_MaintenancePod_Status".Translate() + ": " + $"WULA_MaintenancePod_State_{state}".Translate());
|
||||
|
||||
if (state == MaintenancePodState.Running)
|
||||
if (state == MaintenancePodState.Running && Occupant != null)
|
||||
{
|
||||
if (Occupant != null)
|
||||
{
|
||||
sb.AppendLine("Contains".Translate() + ": " + Occupant.NameShortColored.Resolve());
|
||||
}
|
||||
sb.AppendLine("Contains".Translate() + ": " + Occupant.NameShortColored.Resolve());
|
||||
sb.AppendLine("TimeLeft".Translate() + ": " + ticksRemaining.ToStringTicksToPeriod());
|
||||
var maintenanceNeed = Occupant.needs?.TryGetNeed<Need_Maintenance>();
|
||||
if (maintenanceNeed != null)
|
||||
{
|
||||
// 直接显示 CurLevel,确保与 Need 显示一致
|
||||
sb.AppendLine("WULA_MaintenanceLevel".Translate() + ": " + maintenanceNeed.CurLevel.ToStringPercent());
|
||||
}
|
||||
}
|
||||
|
||||
if (!PowerOn)
|
||||
{
|
||||
sb.AppendLine("NoPower".Translate().Colorize(Color.red));
|
||||
}
|
||||
|
||||
return sb.ToString().TrimEnd();
|
||||
}
|
||||
|
||||
public override IEnumerable<Gizmo> CompGetGizmosExtra()
|
||||
{
|
||||
foreach (var gizmo in base.CompGetGizmosExtra())
|
||||
{
|
||||
yield return gizmo;
|
||||
}
|
||||
|
||||
// 进入维护舱的按钮
|
||||
if (state == MaintenancePodState.Idle && PowerOn)
|
||||
{
|
||||
var enterCommand = new Command_Action
|
||||
yield return new Command_Action
|
||||
{
|
||||
defaultLabel = "WULA_MaintenancePod_Enter".Translate(),
|
||||
defaultDesc = "WULA_MaintenancePod_EnterDesc".Translate(),
|
||||
icon = EnterIcon,
|
||||
action = () =>
|
||||
{
|
||||
List<FloatMenuOption> options = GetPawnOptions();
|
||||
if (options.Any())
|
||||
{
|
||||
Find.WindowStack.Add(new FloatMenu(options));
|
||||
}
|
||||
else
|
||||
{
|
||||
Messages.Message("WULA_MaintenancePod_NoOneNeeds".Translate(), MessageTypeDefOf.RejectInput);
|
||||
}
|
||||
}
|
||||
action = () => ShowPawnSelectionMenu()
|
||||
};
|
||||
yield return enterCommand;
|
||||
}
|
||||
|
||||
// 取消维护的按钮
|
||||
if (state == MaintenancePodState.Running)
|
||||
{
|
||||
var cancelCommand = new Command_Action
|
||||
yield return new Command_Action
|
||||
{
|
||||
defaultLabel = "CommandCancelConstructionLabel".Translate(),
|
||||
defaultDesc = "WULA_MaintenancePod_CancelDesc".Translate(),
|
||||
icon = CancelIcon,
|
||||
action = () =>
|
||||
{
|
||||
EjectPawn();
|
||||
EjectPawn(true);
|
||||
Messages.Message("WULA_MaintenanceCanceled".Translate(), MessageTypeDefOf.NegativeEvent);
|
||||
}
|
||||
};
|
||||
yield return cancelCommand;
|
||||
}
|
||||
|
||||
// DEV GIZMO
|
||||
if (DebugSettings.godMode && state == MaintenancePodState.Running)
|
||||
{
|
||||
var finishCommand = new Command_Action
|
||||
{
|
||||
defaultLabel = "DEV: Finish Cycle",
|
||||
action = () =>
|
||||
{
|
||||
Log.Warning("[WulaPodDebug] DEV: Force finishing cycle.");
|
||||
CycleFinished();
|
||||
}
|
||||
};
|
||||
yield return finishCommand;
|
||||
}
|
||||
}
|
||||
|
||||
private void ShowPawnSelectionMenu()
|
||||
{
|
||||
var options = GetPawnOptions();
|
||||
if (options.Any())
|
||||
{
|
||||
Find.WindowStack.Add(new FloatMenu(options));
|
||||
}
|
||||
else
|
||||
{
|
||||
Messages.Message("WULA_MaintenancePod_NoOneNeeds".Translate(), MessageTypeDefOf.RejectInput);
|
||||
}
|
||||
}
|
||||
private List<FloatMenuOption> GetPawnOptions()
|
||||
{
|
||||
List<FloatMenuOption> options = new List<FloatMenuOption>();
|
||||
// Now iterates over all pawns on the map, not just colonists.
|
||||
foreach (Pawn p in parent.Map.mapPawns.AllPawns.Where(pawn => pawn.def.defName == "WulaSpecies" || pawn.def.defName == "WulaSpeciesReal"))
|
||||
{
|
||||
if (p.health.hediffSet.HasHediff(Props.hediffToRemove))
|
||||
{
|
||||
// If the pawn is downed or not a free colonist, they need to be brought to the pod.
|
||||
if (p.Downed || !p.IsFreeColonist)
|
||||
{
|
||||
float required = RequiredComponents(p);
|
||||
if (refuelableComp.Fuel < required)
|
||||
{
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap + " (" + p.KindLabel + ", " + "WULA_MaintenancePod_NotEnoughComponents".Translate(required.ToString("F0")) + ")", null));
|
||||
}
|
||||
else
|
||||
{
|
||||
// Find colonists who can haul the pawn.
|
||||
var potentialHaulers = parent.Map.mapPawns.FreeColonistsSpawned.Where(colonist =>
|
||||
!colonist.Downed && colonist.CanReserveAndReach(p, PathEndMode.OnCell, Danger.Deadly) && colonist.CanReserveAndReach(parent, PathEndMode.InteractionCell, Danger.Deadly));
|
||||
var options = new List<FloatMenuOption>();
|
||||
var map = parent.Map;
|
||||
|
||||
if (!potentialHaulers.Any())
|
||||
{
|
||||
// If no one can haul, then it's unreachable.
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap + " (" + p.KindLabel + ", " + "CannotReach".Translate() + ")", null));
|
||||
}
|
||||
else
|
||||
{
|
||||
Action action = delegate
|
||||
{
|
||||
// Create a menu to select which colonist should do the hauling.
|
||||
var haulerOptions = new List<FloatMenuOption>();
|
||||
foreach (var hauler in potentialHaulers)
|
||||
{
|
||||
haulerOptions.Add(new FloatMenuOption(hauler.LabelCap, delegate
|
||||
{
|
||||
var haulJob = JobMaker.MakeJob(JobDefOf_WULA.WULA_HaulToMaintenancePod, p, parent);
|
||||
haulJob.count = 1;
|
||||
hauler.jobs.TryTakeOrderedJob(haulJob, JobTag.Misc);
|
||||
}));
|
||||
}
|
||||
Find.WindowStack.Add(new FloatMenu(haulerOptions));
|
||||
};
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap + " (" + p.KindLabel + ")", action));
|
||||
}
|
||||
}
|
||||
}
|
||||
// If the pawn is a free colonist and can walk, they can go on their own.
|
||||
else
|
||||
{
|
||||
if (!p.CanReach(parent, PathEndMode.InteractionCell, Danger.Deadly))
|
||||
{
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap + " (" + "CannotReach".Translate() + ")", null));
|
||||
}
|
||||
else
|
||||
{
|
||||
float required = RequiredComponents(p);
|
||||
if (refuelableComp.Fuel >= required)
|
||||
{
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap, () =>
|
||||
{
|
||||
Job job = JobMaker.MakeJob(JobDefOf_WULA.WULA_EnterMaintenancePod, parent);
|
||||
p.jobs.TryTakeOrderedJob(job, JobTag.Misc);
|
||||
}));
|
||||
}
|
||||
else
|
||||
{
|
||||
options.Add(new FloatMenuOption(p.LabelShortCap + " (" + "WULA_MaintenancePod_NotEnoughComponents".Translate(required.ToString("F0")) + ")", null));
|
||||
}
|
||||
}
|
||||
}
|
||||
foreach (var pawn in map.mapPawns.AllPawnsSpawned)
|
||||
{
|
||||
// 首先检查是否有维护需求
|
||||
var maintenanceNeed = pawn.needs?.TryGetNeed<Need_Maintenance>();
|
||||
if (maintenanceNeed == null)
|
||||
{
|
||||
// 这个Pawn没有维护需求,跳过
|
||||
continue;
|
||||
}
|
||||
|
||||
// 检查是否真的需要维护
|
||||
if (maintenanceNeed.CurLevel > Props.minNeedLevelToMaintain && !DebugSettings.godMode)
|
||||
continue;
|
||||
|
||||
// 创建选项
|
||||
var option = CreatePawnOption(pawn, maintenanceNeed);
|
||||
if (option != null)
|
||||
options.Add(option);
|
||||
}
|
||||
|
||||
return options;
|
||||
}
|
||||
}
|
||||
|
||||
private FloatMenuOption CreatePawnOption(Pawn pawn, Need_Maintenance need)
|
||||
{
|
||||
string label = $"{pawn.LabelShortCap} ({need.CurLevel.ToStringPercent()})";
|
||||
float requiredComponents = RequiredComponents;
|
||||
// 检查组件是否足够
|
||||
if (refuelableComp.Fuel < requiredComponents)
|
||||
{
|
||||
return new FloatMenuOption(label + " (" + "WULA_MaintenancePod_NotEnoughComponents".Translate(requiredComponents.ToString("F0")) + ")", null);
|
||||
}
|
||||
// 检查是否可以到达
|
||||
if (!pawn.CanReach(parent, PathEndMode.InteractionCell, Danger.Deadly))
|
||||
{
|
||||
return new FloatMenuOption(label + " (" + "CannotReach".Translate() + ")", null);
|
||||
}
|
||||
return new FloatMenuOption(label, () =>
|
||||
{
|
||||
if (pawn.Downed || !pawn.IsFreeColonist)
|
||||
{
|
||||
// 需要搬运
|
||||
var haulJob = JobMaker.MakeJob(JobDefOf_WULA.WULA_HaulToMaintenancePod, pawn, parent);
|
||||
var hauler = FindBestHauler(pawn);
|
||||
if (hauler != null)
|
||||
{
|
||||
hauler.jobs.TryTakeOrderedJob(haulJob);
|
||||
}
|
||||
else
|
||||
{
|
||||
Messages.Message("WULA_NoHaulerAvailable".Translate(), MessageTypeDefOf.RejectInput);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// 自己进入
|
||||
var enterJob = JobMaker.MakeJob(JobDefOf_WULA.WULA_EnterMaintenancePod, parent);
|
||||
pawn.jobs.TryTakeOrderedJob(enterJob);
|
||||
}
|
||||
});
|
||||
}
|
||||
private Pawn FindBestHauler(Pawn target)
|
||||
{
|
||||
return parent.Map.mapPawns.FreeColonistsSpawned
|
||||
.Where(colonist => !colonist.Downed &&
|
||||
colonist.CanReserveAndReach(target, PathEndMode.OnCell, Danger.Deadly) &&
|
||||
colonist.CanReserveAndReach(parent, PathEndMode.InteractionCell, Danger.Deadly))
|
||||
.OrderBy(colonist => colonist.Position.DistanceTo(target.Position))
|
||||
.FirstOrDefault();
|
||||
}
|
||||
}
|
||||
public enum MaintenancePodState
|
||||
{
|
||||
Idle,
|
||||
Running,
|
||||
Running
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user