945 lines
33 KiB
C#
945 lines
33 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using RimWorld;
|
||
using RimWorld.Planet;
|
||
using UnityEngine;
|
||
using Verse;
|
||
using Verse.AI;
|
||
using Verse.Sound;
|
||
|
||
namespace WulaFallenEmpire
|
||
{
|
||
/// <summary>
|
||
/// 内置空间武装穿梭机 - 基于原版MapPortal机制的口袋空间实现
|
||
/// 结合了武装防御能力和口袋空间技术的复合型载具
|
||
/// </summary>
|
||
[StaticConstructorOnStartup]
|
||
public class Building_ArmedShuttleWithPocket : Building_ArmedShuttle, IThingHolder
|
||
{
|
||
#region 静态图标定义（使用原版MapPortal的图标）
|
||
|
||
/// <summary>查看口袋地图图标</summary>
|
||
private static readonly Texture2D ViewPocketMapTex = ContentFinder<Texture2D>.Get("UI/Commands/ViewCave");
|
||
|
||
/// <summary>取消进入图标</summary>
|
||
private static readonly Texture2D CancelEnterTex = ContentFinder<Texture2D>.Get("UI/Designators/Cancel");
|
||
|
||
/// <summary>默认进入图标</summary>
|
||
private static readonly Texture2D DefaultEnterTex = ContentFinder<Texture2D>.Get("UI/Commands/EnterCave");
|
||
|
||
#endregion
|
||
#region 口袋空间字段
|
||
|
||
/// <summary>内部口袋地图实例</summary>
|
||
private Map pocketMap;
|
||
|
||
/// <summary>口袋地图是否已生成</summary>
|
||
private bool pocketMapGenerated;
|
||
|
||
/// <summary>内部空间大小</summary>
|
||
private IntVec2 pocketMapSize = new IntVec2(80, 80);
|
||
|
||
/// <summary>地图生成器定义</summary>
|
||
private MapGeneratorDef mapGenerator;
|
||
|
||
/// <summary>退出点定义</summary>
|
||
private ThingDef exitDef;
|
||
|
||
/// <summary>允许直接访问（无需骇入）</summary>
|
||
private bool allowDirectAccess = true;
|
||
|
||
/// <summary>口袋空间内的物品容器</summary>
|
||
private ThingOwner innerContainer;
|
||
|
||
/// <summary>口袋地图退出点（模仿原版 MapPortal.exit）</summary>
|
||
public Building_PocketMapExit exit;
|
||
|
||
/// <summary>是否已经进入过（模仿原版 MapPortal.beenEntered）</summary>
|
||
protected bool beenEntered;
|
||
|
||
/// <summary>待加载物品列表（模仿原版 MapPortal.leftToLoad）</summary>
|
||
public List<TransferableOneWay> leftToLoad;
|
||
|
||
/// <summary>是否已通知无法加载更多（模仿原版 MapPortal.notifiedCantLoadMore）</summary>
|
||
public bool notifiedCantLoadMore;
|
||
|
||
#endregion
|
||
|
||
#region 属性
|
||
|
||
/// <summary>获取内部口袋地图</summary>
|
||
public Map PocketMap => pocketMap;
|
||
|
||
/// <summary>口袋地图是否已生成</summary>
|
||
public bool PocketMapGenerated => pocketMapGenerated;
|
||
|
||
/// <summary>是否允许直接访问口袋空间</summary>
|
||
public bool AllowDirectAccess => allowDirectAccess;
|
||
|
||
/// <summary>内部容器</summary>
|
||
public ThingOwner InnerContainer => innerContainer;
|
||
|
||
/// <summary>
|
||
/// 获取进入按钮的图标
|
||
/// </summary>
|
||
protected virtual Texture2D EnterTex => DefaultEnterTex;
|
||
|
||
/// <summary>
|
||
/// 获取进入按钮的文本
|
||
/// </summary>
|
||
public virtual string EnterString => "WULA.PocketSpace.Enter".Translate();
|
||
|
||
/// <summary>
|
||
/// 获取取消进入按钮的文本
|
||
/// </summary>
|
||
public virtual string CancelEnterString => "WULA.PocketSpace.CancelEnter".Translate();
|
||
|
||
/// <summary>
|
||
/// 获取进入中的文本
|
||
/// </summary>
|
||
public virtual string EnteringString => "WULA.PocketSpace.Entering".Translate();
|
||
|
||
/// <summary>加载是否正在进行（模仿原版 MapPortal.LoadInProgress）</summary>
|
||
public bool LoadInProgress
|
||
{
|
||
get
|
||
{
|
||
if (leftToLoad != null)
|
||
{
|
||
return leftToLoad.Any();
|
||
}
|
||
return false;
|
||
}
|
||
}
|
||
|
||
/// <summary>是否有Pawn可以加载任何东西（模仿原版 MapPortal.AnyPawnCanLoadAnythingNow）</summary>
|
||
public bool AnyPawnCanLoadAnythingNow
|
||
{
|
||
get
|
||
{
|
||
if (!LoadInProgress)
|
||
{
|
||
return false;
|
||
}
|
||
if (!Spawned)
|
||
{
|
||
return false;
|
||
}
|
||
// 简化版本，只检查基本条件
|
||
return Map.mapPawns.AllPawnsSpawned.Any(p => p.IsColonist && p.CanReach(this, PathEndMode.Touch, Danger.Deadly));
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 构造函数
|
||
|
||
public Building_ArmedShuttleWithPocket()
|
||
{
|
||
innerContainer = new ThingOwner<Thing>(this, oneStackOnly: false);
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 基础重写方法
|
||
|
||
|
||
|
||
public override void PostMake()
|
||
{
|
||
base.PostMake();
|
||
if (innerContainer == null)
|
||
{
|
||
innerContainer = new ThingOwner<Thing>(this, oneStackOnly: false);
|
||
}
|
||
}
|
||
|
||
public override void ExposeData()
|
||
{
|
||
base.ExposeData();
|
||
Scribe_Deep.Look(ref pocketMap, "pocketMap");
|
||
Scribe_Values.Look(ref pocketMapGenerated, "pocketMapGenerated", false);
|
||
Scribe_Values.Look(ref pocketMapSize, "pocketMapSize", new IntVec2(80, 80));
|
||
Scribe_Defs.Look(ref mapGenerator, "mapGenerator");
|
||
Scribe_Defs.Look(ref exitDef, "exitDef");
|
||
Scribe_Values.Look(ref allowDirectAccess, "allowDirectAccess", true);
|
||
Scribe_Deep.Look(ref innerContainer, "innerContainer", this);
|
||
|
||
if (Scribe.mode == LoadSaveMode.PostLoadInit)
|
||
{
|
||
if (innerContainer == null)
|
||
{
|
||
innerContainer = new ThingOwner<Thing>(this, oneStackOnly: false);
|
||
}
|
||
}
|
||
}
|
||
|
||
public override void DeSpawn(DestroyMode mode = DestroyMode.Vanish)
|
||
{
|
||
// 清理口袋地图
|
||
if (pocketMap != null && pocketMapGenerated)
|
||
{
|
||
try
|
||
{
|
||
// 将口袋空间中的物品和人员转移到主地图
|
||
TransferAllFromPocketToMainMap();
|
||
|
||
// 销毁口袋地图
|
||
PocketMapUtility.DestroyPocketMap(pocketMap);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Error cleaning up pocket map: {ex}");
|
||
}
|
||
}
|
||
base.DeSpawn(mode);
|
||
}
|
||
|
||
public override string GetInspectString()
|
||
{
|
||
StringBuilder sb = new StringBuilder(base.GetInspectString());
|
||
|
||
if (pocketMapGenerated)
|
||
{
|
||
sb.AppendLine("WULA.PocketSpace.Status".Translate() + ": " + "WULA.PocketSpace.Ready".Translate());
|
||
if (innerContainer.Count > 0)
|
||
{
|
||
sb.AppendLine("WULA.PocketSpace.ItemCount".Translate(innerContainer.Count));
|
||
}
|
||
|
||
// 显示口袋空间中的人员数量
|
||
if (pocketMap != null)
|
||
{
|
||
int pawnCount = pocketMap.mapPawns.AllPawnsSpawned.Where(p => p.IsColonist).Count();
|
||
if (pawnCount > 0)
|
||
{
|
||
sb.AppendLine("WULA.PocketSpace.PawnCount".Translate(pawnCount));
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
sb.AppendLine("WULA.PocketSpace.Status".Translate() + ": " + "WULA.PocketSpace.NotGenerated".Translate());
|
||
}
|
||
|
||
return sb.ToString().TrimEndNewlines();
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 口袋空间核心方法
|
||
|
||
/// <summary>
|
||
/// 检查是否可以进入口袋空间
|
||
/// </summary>
|
||
public bool CanEnterPocketSpace()
|
||
{
|
||
if (!allowDirectAccess)
|
||
{
|
||
return false; // 需要特殊权限
|
||
}
|
||
|
||
if (!Spawned)
|
||
{
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 进入口袋空间 - 基于原版PocketMapUtility实现
|
||
/// </summary>
|
||
public void EnterPocketSpace(IEnumerable<Pawn> pawns = null)
|
||
{
|
||
if (!CanEnterPocketSpace())
|
||
{
|
||
Messages.Message("WULA.PocketSpace.CannotEnter".Translate(), this, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 创建或获取口袋地图
|
||
if (pocketMap == null && !pocketMapGenerated)
|
||
{
|
||
CreatePocketMap();
|
||
}
|
||
|
||
if (pocketMap == null)
|
||
{
|
||
Messages.Message("WULA.PocketSpace.CreationFailed".Translate(), this, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 传送玩家到口袋空间
|
||
List<Pawn> pawnsToTransfer = new List<Pawn>();
|
||
|
||
if (pawns != null)
|
||
{
|
||
pawnsToTransfer.AddRange(pawns.Where(p => p != null && p.Spawned && p.IsColonist));
|
||
}
|
||
else
|
||
{
|
||
// 如果没有指定人员，传送选中的殖民者
|
||
pawnsToTransfer.AddRange(Find.Selector.SelectedPawns.Where(p => p.IsColonist));
|
||
}
|
||
|
||
if (pawnsToTransfer.Count == 0)
|
||
{
|
||
Messages.Message("WULA.PocketSpace.NoPawnsSelected".Translate(), this, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 执行传送
|
||
int transferredCount = 0;
|
||
foreach (Pawn pawn in pawnsToTransfer)
|
||
{
|
||
if (TransferPawnToPocketSpace(pawn))
|
||
{
|
||
transferredCount++;
|
||
}
|
||
}
|
||
|
||
if (transferredCount > 0)
|
||
{
|
||
Messages.Message("WULA.PocketSpace.TransferSuccess".Translate(transferredCount), MessageTypeDefOf.PositiveEvent);
|
||
|
||
// 切换到口袋地图
|
||
Current.Game.CurrentMap = pocketMap;
|
||
Find.CameraDriver.JumpToCurrentMapLoc(pocketMap.Center);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 切换到口袋空间视角
|
||
/// </summary>
|
||
public void SwitchToPocketSpace()
|
||
{
|
||
if (pocketMap == null)
|
||
{
|
||
if (!pocketMapGenerated)
|
||
{
|
||
CreatePocketMap();
|
||
}
|
||
|
||
if (pocketMap == null)
|
||
{
|
||
Messages.Message("WULA.PocketSpace.CreationFailed".Translate(), this, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
}
|
||
|
||
Current.Game.CurrentMap = pocketMap;
|
||
Find.CameraDriver.JumpToCurrentMapLoc(pocketMap.Center);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 创建口袋地图 - 使用原版PocketMapUtility（模仿 MapPortal.GeneratePocketMap）
|
||
/// </summary>
|
||
private void CreatePocketMap()
|
||
{
|
||
try
|
||
{
|
||
// 模仿原版 MapPortal.GeneratePocketMap 的实现
|
||
PocketMapUtility.currentlyGeneratingPortal = null; // 我们不是 MapPortal，但可以设为 null
|
||
pocketMap = GeneratePocketMapInt();
|
||
PocketMapUtility.currentlyGeneratingPortal = null;
|
||
|
||
if (pocketMap != null)
|
||
{
|
||
pocketMapGenerated = true;
|
||
|
||
// 在口袋地图中心放置退出点
|
||
CreateExitPoint();
|
||
|
||
Log.Message($"[WULA] Successfully created pocket map of size {pocketMapSize} for armed shuttle");
|
||
}
|
||
else
|
||
{
|
||
Log.Error("[WULA] Failed to create pocket map");
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Exception creating pocket map: {ex}");
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 生成口袋地图的内部实现（模仿 MapPortal.GeneratePocketMapInt）
|
||
/// </summary>
|
||
protected virtual Map GeneratePocketMapInt()
|
||
{
|
||
return PocketMapUtility.GeneratePocketMap(new IntVec3(pocketMapSize.x, 1, pocketMapSize.z), mapGenerator, GetExtraGenSteps(), this.Map);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取额外的生成步骤（模仿 MapPortal.GetExtraGenSteps）
|
||
/// </summary>
|
||
protected virtual IEnumerable<GenStepWithParams> GetExtraGenSteps()
|
||
{
|
||
return System.Linq.Enumerable.Empty<GenStepWithParams>();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 在口袋地图中创建退出点（模仿原版）
|
||
/// </summary>
|
||
private void CreateExitPoint()
|
||
{
|
||
if (pocketMap == null || exitDef == null) return;
|
||
|
||
try
|
||
{
|
||
// 在地图中心找一个合适的位置
|
||
IntVec3 exitPos = pocketMap.Center;
|
||
|
||
// 寻找可建造的位置
|
||
if (!exitPos.Standable(pocketMap) || exitPos.GetThingList(pocketMap).Any(t => t.def.category == ThingCategory.Building))
|
||
{
|
||
exitPos = CellFinder.RandomClosewalkCellNear(pocketMap.Center, pocketMap, 5,
|
||
p => p.Standable(pocketMap) && !p.GetThingList(pocketMap).Any(t => t.def.category == ThingCategory.Building));
|
||
}
|
||
|
||
if (exitPos.IsValid)
|
||
{
|
||
// 创建退出点建筑
|
||
Thing exitBuilding = ThingMaker.MakeThing(exitDef);
|
||
if (exitBuilding is Building_PocketMapExit exitPortal)
|
||
{
|
||
exitPortal.targetMap = this.Map;
|
||
exitPortal.targetPos = this.Position;
|
||
exitPortal.parentShuttle = this;
|
||
exit = exitPortal; // 设置 exit 引用，模仿原版 MapPortal
|
||
}
|
||
|
||
GenPlace.TryPlaceThing(exitBuilding, exitPos, pocketMap, ThingPlaceMode.Direct);
|
||
Log.Message($"[WULA] Created exit point at {exitPos} in pocket map");
|
||
}
|
||
else
|
||
{
|
||
Log.Warning("[WULA] Could not find valid position for exit point in pocket map");
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Error creating exit point: {ex}");
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 将单个Pawn传送到口袋空间
|
||
/// </summary>
|
||
private bool TransferPawnToPocketSpace(Pawn pawn)
|
||
{
|
||
if (pawn == null || !pawn.Spawned || pocketMap == null) return false;
|
||
|
||
try
|
||
{
|
||
// 找一个安全的位置
|
||
IntVec3 spawnPos = CellFinder.RandomClosewalkCellNear(pocketMap.Center, pocketMap, 10,
|
||
p => p.Standable(pocketMap) && !p.GetThingList(pocketMap).Any(t => t is Pawn));
|
||
|
||
if (spawnPos.IsValid)
|
||
{
|
||
pawn.DeSpawn();
|
||
GenPlace.TryPlaceThing(pawn, spawnPos, pocketMap, ThingPlaceMode.Near);
|
||
return true;
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Error transferring pawn {pawn?.LabelShort} to pocket space: {ex}");
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 将所有物品和人员从口袋空间转移到主地图
|
||
/// </summary>
|
||
private void TransferAllFromPocketToMainMap()
|
||
{
|
||
if (pocketMap == null || !Spawned) return;
|
||
|
||
try
|
||
{
|
||
// 转移所有殖民者
|
||
List<Pawn> pawnsToTransfer = pocketMap.mapPawns.AllPawnsSpawned
|
||
.Where(p => p.IsColonist).ToList();
|
||
|
||
foreach (Pawn pawn in pawnsToTransfer)
|
||
{
|
||
IntVec3 spawnPos = CellFinder.RandomClosewalkCellNear(this.Position, this.Map, 5,
|
||
p => p.Standable(this.Map) && !p.GetThingList(this.Map).Any(t => t is Pawn));
|
||
|
||
if (spawnPos.IsValid)
|
||
{
|
||
pawn.DeSpawn();
|
||
GenPlace.TryPlaceThing(pawn, spawnPos, this.Map, ThingPlaceMode.Near);
|
||
}
|
||
}
|
||
|
||
// 转移所有物品到内部容器
|
||
List<Thing> itemsToTransfer = pocketMap.listerThings.AllThings
|
||
.Where(t => t.def.category == ThingCategory.Item && t.def.EverHaulable).ToList();
|
||
|
||
foreach (Thing item in itemsToTransfer)
|
||
{
|
||
if (item.Spawned)
|
||
{
|
||
item.DeSpawn();
|
||
if (!innerContainer.TryAdd(item))
|
||
{
|
||
// 如果容器满了，丢到穿梭机附近
|
||
IntVec3 dropPos = CellFinder.RandomClosewalkCellNear(this.Position, this.Map, 3);
|
||
if (dropPos.IsValid)
|
||
{
|
||
GenPlace.TryPlaceThing(item, dropPos, this.Map, ThingPlaceMode.Near);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
Log.Message($"[WULA] Transferred {pawnsToTransfer.Count} pawns and {itemsToTransfer.Count} items from pocket space");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Error transferring from pocket map: {ex}");
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region Gizmo方法
|
||
|
||
public override IEnumerable<Gizmo> GetGizmos()
|
||
{
|
||
foreach (Gizmo gizmo in base.GetGizmos())
|
||
{
|
||
yield return gizmo;
|
||
}
|
||
|
||
if (allowDirectAccess)
|
||
{
|
||
// 进入口袋空间按钮（模仿原版MapPortal）
|
||
Command_Action enterCommand = new Command_Action();
|
||
enterCommand.action = delegate
|
||
{
|
||
// 使用自定义的殖民者选择对话框，模仿原版Dialog_EnterPortal的行为
|
||
OpenPawnSelectionDialog();
|
||
};
|
||
enterCommand.icon = EnterTex;
|
||
enterCommand.defaultLabel = EnterString + "...";
|
||
enterCommand.defaultDesc = "WULA.PocketSpace.EnterDesc".Translate();
|
||
|
||
// 检查是否可以进入（模仿原版MapPortal.IsEnterable）
|
||
string reason;
|
||
enterCommand.Disabled = !IsEnterable(out reason);
|
||
enterCommand.disabledReason = reason;
|
||
yield return enterCommand;
|
||
|
||
// 查看口袋地图按钮（模仿原版MapPortal）
|
||
if (pocketMap != null)
|
||
{
|
||
yield return new Command_Action
|
||
{
|
||
defaultLabel = "WULA.PocketSpace.ViewMap".Translate(),
|
||
defaultDesc = "WULA.PocketSpace.ViewMapDesc".Translate(),
|
||
icon = ViewPocketMapTex,
|
||
action = delegate
|
||
{
|
||
// 模仿原版，跳转到口袋地图并选中退出点
|
||
if (exit != null)
|
||
{
|
||
CameraJumper.TryJumpAndSelect(exit);
|
||
}
|
||
else
|
||
{
|
||
SwitchToPocketSpace();
|
||
}
|
||
}
|
||
};
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
|
||
#endregion
|
||
|
||
#region IThingHolder接口实现
|
||
|
||
public ThingOwner GetDirectlyHeldThings()
|
||
{
|
||
return innerContainer; // 使用我们自己的容器而不是 PortalContainerProxy
|
||
}
|
||
|
||
public void GetChildHolders(List<IThingHolder> outChildren)
|
||
{
|
||
ThingOwnerUtility.AppendThingHoldersFromThings(outChildren, GetDirectlyHeldThings());
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region MapPortal兼容接口（使Dialog_EnterPortal能正常工作）
|
||
|
||
/// <summary>
|
||
/// 检查是否可以进入（模仿原版MapPortal.IsEnterable）
|
||
/// </summary>
|
||
public virtual bool IsEnterable(out string reason)
|
||
{
|
||
if (!allowDirectAccess)
|
||
{
|
||
reason = "WULA.PocketSpace.AccessDenied".Translate();
|
||
return false;
|
||
}
|
||
|
||
if (!Spawned)
|
||
{
|
||
reason = "WULA.PocketSpace.NotSpawned".Translate();
|
||
return false;
|
||
}
|
||
|
||
reason = "";
|
||
return true;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取目标地图（模仿原版MapPortal.GetOtherMap）
|
||
/// </summary>
|
||
public virtual Map GetOtherMap()
|
||
{
|
||
if (pocketMap == null)
|
||
{
|
||
CreatePocketMap();
|
||
}
|
||
return pocketMap;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取目标位置（模仿原版MapPortal.GetDestinationLocation）
|
||
/// </summary>
|
||
public virtual IntVec3 GetDestinationLocation()
|
||
{
|
||
if (exit != null)
|
||
{
|
||
return exit.Position;
|
||
}
|
||
return pocketMap?.Center ?? IntVec3.Invalid;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 处理进入事件（模仿原版MapPortal.OnEntered）
|
||
/// </summary>
|
||
public virtual void OnEntered(Pawn pawn)
|
||
{
|
||
// 通知物品被添加（用于统计和管理）
|
||
Notify_ThingAdded(pawn);
|
||
|
||
// 播放传送音效（如果存在）
|
||
if (Find.CurrentMap == this.Map)
|
||
{
|
||
// 可以在这里添加音效播放
|
||
// def.portal?.traverseSound?.PlayOneShot(this);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 打开殖民者选择对话框（模仿原版Dialog_EnterPortal的功能）
|
||
/// </summary>
|
||
private void OpenPawnSelectionDialog()
|
||
{
|
||
// 获取所有可用的殖民者
|
||
List<Pawn> availablePawns = Map.mapPawns.AllPawnsSpawned
|
||
.Where(p => p.IsColonist && !p.Downed && p.CanReach(this, PathEndMode.Touch, Danger.Deadly))
|
||
.ToList();
|
||
|
||
if (availablePawns.Count == 0)
|
||
{
|
||
Messages.Message("WULA.PocketSpace.NoPawnsAvailable".Translate(), this, MessageTypeDefOf.RejectInput);
|
||
return;
|
||
}
|
||
|
||
// 创建选项列表
|
||
List<FloatMenuOption> options = new List<FloatMenuOption>();
|
||
|
||
// 添加单个殖民者选项
|
||
foreach (Pawn pawn in availablePawns)
|
||
{
|
||
FloatMenuOption option = new FloatMenuOption(
|
||
$"{pawn.LabelShort}",
|
||
delegate
|
||
{
|
||
EnterPocketSpace(new List<Pawn> { pawn });
|
||
}
|
||
);
|
||
options.Add(option);
|
||
}
|
||
|
||
// 添加“全部殖民者”选项
|
||
if (availablePawns.Count > 1)
|
||
{
|
||
FloatMenuOption allOption = new FloatMenuOption(
|
||
"WULA.PocketSpace.AllColonists".Translate(availablePawns.Count),
|
||
delegate
|
||
{
|
||
EnterPocketSpace(availablePawns);
|
||
}
|
||
);
|
||
options.Add(allOption);
|
||
}
|
||
|
||
// 添加“只切换视角”选项
|
||
FloatMenuOption viewOnlyOption = new FloatMenuOption(
|
||
"WULA.PocketSpace.ViewOnly".Translate(),
|
||
delegate
|
||
{
|
||
if (pocketMapGenerated)
|
||
{
|
||
SwitchToPocketSpace();
|
||
}
|
||
else
|
||
{
|
||
CreatePocketMap();
|
||
if (pocketMapGenerated)
|
||
{
|
||
SwitchToPocketSpace();
|
||
}
|
||
}
|
||
}
|
||
);
|
||
options.Add(viewOnlyOption);
|
||
|
||
// 显示浮动菜单
|
||
FloatMenu floatMenu = new FloatMenu(options);
|
||
Find.WindowStack.Add(floatMenu);
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 原版MapPortal的物品传送方法
|
||
|
||
/// <summary>
|
||
/// 通知有物品被添加（模仿原版 MapPortal.Notify_ThingAdded）
|
||
/// </summary>
|
||
public void Notify_ThingAdded(Thing t)
|
||
{
|
||
SubtractFromToLoadList(t, t.stackCount);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加到加载列表（模仿原版 MapPortal.AddToTheToLoadList）
|
||
/// </summary>
|
||
public void AddToTheToLoadList(TransferableOneWay t, int count)
|
||
{
|
||
if (!t.HasAnyThing || count <= 0)
|
||
{
|
||
return;
|
||
}
|
||
if (leftToLoad == null)
|
||
{
|
||
leftToLoad = new List<TransferableOneWay>();
|
||
}
|
||
TransferableOneWay transferableOneWay = TransferableUtility.TransferableMatching(t.AnyThing, leftToLoad, TransferAsOneMode.PodsOrCaravanPacking);
|
||
if (transferableOneWay != null)
|
||
{
|
||
for (int i = 0; i < t.things.Count; i++)
|
||
{
|
||
if (!transferableOneWay.things.Contains(t.things[i]))
|
||
{
|
||
transferableOneWay.things.Add(t.things[i]);
|
||
}
|
||
}
|
||
if (transferableOneWay.CanAdjustBy(count).Accepted)
|
||
{
|
||
transferableOneWay.AdjustBy(count);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
TransferableOneWay transferableOneWay2 = new TransferableOneWay();
|
||
leftToLoad.Add(transferableOneWay2);
|
||
transferableOneWay2.things.AddRange(t.things);
|
||
transferableOneWay2.AdjustTo(count);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 从加载列表中减去（模仿原版 MapPortal.SubtractFromToLoadList）
|
||
/// </summary>
|
||
public int SubtractFromToLoadList(Thing t, int count)
|
||
{
|
||
if (leftToLoad == null)
|
||
{
|
||
return 0;
|
||
}
|
||
TransferableOneWay transferableOneWay = TransferableUtility.TransferableMatchingDesperate(t, leftToLoad, TransferAsOneMode.PodsOrCaravanPacking);
|
||
if (transferableOneWay == null)
|
||
{
|
||
return 0;
|
||
}
|
||
if (transferableOneWay.CountToTransfer <= 0)
|
||
{
|
||
return 0;
|
||
}
|
||
int num = Mathf.Min(count, transferableOneWay.CountToTransfer);
|
||
transferableOneWay.AdjustBy(-num);
|
||
transferableOneWay.things.Remove(t);
|
||
if (transferableOneWay.CountToTransfer <= 0)
|
||
{
|
||
leftToLoad.Remove(transferableOneWay);
|
||
}
|
||
return num;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 取消加载（模仿原版 MapPortal.CancelLoad）
|
||
/// </summary>
|
||
public void CancelLoad()
|
||
{
|
||
// 简化版本，只清理列表
|
||
if (leftToLoad != null)
|
||
{
|
||
leftToLoad.Clear();
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
|
||
#region 穿梭机状态变化处理
|
||
|
||
/// <summary>
|
||
/// 更新口袋空间中退出点的目标位置（处理穿梭机位置变化）
|
||
/// </summary>
|
||
public void UpdateExitPointTarget()
|
||
{
|
||
if (pocketMap == null || exit == null) return;
|
||
|
||
try
|
||
{
|
||
// 如果退出点是我们的Building_PocketMapExit类型，更新其目标位置
|
||
if (exit is Building_PocketMapExit pocketExit)
|
||
{
|
||
// 更新目标地图和位置
|
||
if (this.Spawned)
|
||
{
|
||
// 穿梭机在地图上，更新目标位置
|
||
if (pocketExit.targetMap != this.Map || pocketExit.targetPos != this.Position)
|
||
{
|
||
pocketExit.targetMap = this.Map;
|
||
pocketExit.targetPos = this.Position;
|
||
pocketExit.parentShuttle = this;
|
||
Log.Message($"[WULA] Updated pocket map exit target to shuttle location: {this.Map?.uniqueID} at {this.Position}");
|
||
}
|
||
}
|
||
else
|
||
{
|
||
// 穿梭机不在地图上（可能在飞行中），记录警告但保持原有目标
|
||
Log.Warning($"[WULA] Shuttle not spawned, pocket map exit target may be outdated. Current target: {pocketExit.targetMap?.uniqueID} at {pocketExit.targetPos}");
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Log.Error($"[WULA] Error updating exit point target: {ex}");
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 重写Tick方法，定期检查穿梭机状态变化
|
||
/// </summary>
|
||
protected override void Tick()
|
||
{
|
||
base.Tick();
|
||
|
||
// 每隔一段时间检查退出点目标是否需要更新（处理穿梭机移动的情况）
|
||
if (this.IsHashIntervalTick(2500) && pocketMapGenerated && exit != null)
|
||
{
|
||
UpdateExitPointTarget();
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 重写 SpawnSetup，确保位置变化时更新退出点
|
||
/// </summary>
|
||
public override void SpawnSetup(Map map, bool respawningAfterLoad)
|
||
{
|
||
// 保存旧位置信息
|
||
Map oldMap = this.Map;
|
||
IntVec3 oldPos = this.Position;
|
||
|
||
base.SpawnSetup(map, respawningAfterLoad);
|
||
|
||
if (innerContainer == null)
|
||
{
|
||
innerContainer = new ThingOwner<Thing>(this, oneStackOnly: false);
|
||
}
|
||
|
||
// 简化实现，直接使用 innerContainer 而不是 PortalContainerProxy
|
||
// 因为 PortalContainerProxy 需要 MapPortal 类型，但我们不是继承自 MapPortal
|
||
|
||
// 更新退出点目标（处理穿梭机重新部署的情况）
|
||
UpdateExitPointTarget();
|
||
|
||
// 从 ThingDef 中读取 portal 配置
|
||
if (def.HasModExtension<PocketMapProperties>())
|
||
{
|
||
var portalProps = def.GetModExtension<PocketMapProperties>();
|
||
if (portalProps.pocketMapGenerator != null)
|
||
{
|
||
mapGenerator = portalProps.pocketMapGenerator;
|
||
}
|
||
if (portalProps.exitDef != null)
|
||
{
|
||
exitDef = portalProps.exitDef;
|
||
}
|
||
if (portalProps.pocketMapSize != IntVec2.Zero)
|
||
{
|
||
pocketMapSize = portalProps.pocketMapSize;
|
||
}
|
||
allowDirectAccess = portalProps.allowDirectAccess;
|
||
}
|
||
|
||
// 初始化地图生成器和退出点定义（如果 XML 中没有配置）
|
||
if (mapGenerator == null)
|
||
{
|
||
mapGenerator = DefDatabase<MapGeneratorDef>.GetNamed("AncientStockpile", false)
|
||
?? DefDatabase<MapGeneratorDef>.GetNamed("Base_Player", false)
|
||
?? MapGeneratorDefOf.Base_Player;
|
||
}
|
||
|
||
if (exitDef == null)
|
||
{
|
||
exitDef = DefDatabase<ThingDef>.GetNamed("WULA_PocketMapExit", false)
|
||
?? ThingDefOf.Door;
|
||
}
|
||
|
||
// 如果位置发生了变化，记录日志
|
||
if (oldMap != null && (oldMap != map || oldPos != this.Position))
|
||
{
|
||
Log.Message($"[WULA] Shuttle moved from {oldMap?.uniqueID}:{oldPos} to {map?.uniqueID}:{this.Position}, updating pocket map exit target");
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
}
|
||
|
||
/// <summary>
|
||
/// 口袋空间属性配置类
|
||
/// </summary>
|
||
public class PocketMapProperties : DefModExtension
|
||
{
|
||
/// <summary>口袋地图生成器</summary>
|
||
public MapGeneratorDef pocketMapGenerator;
|
||
|
||
/// <summary>退出点定义</summary>
|
||
public ThingDef exitDef;
|
||
|
||
/// <summary>口袋地图大小</summary>
|
||
public IntVec2 pocketMapSize = new IntVec2(13, 13);
|
||
|
||
/// <summary>允许直接访问</summary>
|
||
public bool allowDirectAccess = true;
|
||
}
|
||
} |