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 { /// /// 内置空间武装穿梭机 - 基于原版MapPortal机制的口袋空间实现 /// 结合了武装防御能力和口袋空间技术的复合型载具 /// [StaticConstructorOnStartup] public class Building_ArmedShuttleWithPocket : Building_ArmedShuttle, IThingHolder { #region 静态图标定义(使用原版MapPortal的图标) /// 查看口袋地图图标 private static readonly Texture2D ViewPocketMapTex = ContentFinder.Get("UI/Commands/ViewCave"); /// 取消进入图标 private static readonly Texture2D CancelEnterTex = ContentFinder.Get("UI/Designators/Cancel"); /// 默认进入图标 private static readonly Texture2D DefaultEnterTex = ContentFinder.Get("UI/Commands/EnterCave"); #endregion #region 口袋空间字段 /// 内部口袋地图实例 private Map pocketMap; /// 口袋地图是否已生成 private bool pocketMapGenerated; /// 内部空间大小 private IntVec2 pocketMapSize = new IntVec2(80, 80); /// 地图生成器定义 private MapGeneratorDef mapGenerator; /// 退出点定义 private ThingDef exitDef; /// 允许直接访问(无需骇入) private bool allowDirectAccess = true; /// 口袋空间内的物品容器 private ThingOwner innerContainer; /// 口袋地图退出点(模仿原版 MapPortal.exit) public Building_PocketMapExit exit; /// 是否已经进入过(模仿原版 MapPortal.beenEntered) protected bool beenEntered; /// 待加载物品列表(模仿原版 MapPortal.leftToLoad) public List leftToLoad; /// 是否已通知无法加载更多(模仿原版 MapPortal.notifiedCantLoadMore) public bool notifiedCantLoadMore; #endregion #region 属性 /// 获取内部口袋地图 public Map PocketMap => pocketMap; /// 口袋地图是否已生成 public bool PocketMapGenerated => pocketMapGenerated; /// 是否允许直接访问口袋空间 public bool AllowDirectAccess => allowDirectAccess; /// 内部容器 public ThingOwner InnerContainer => innerContainer; /// /// 获取进入按钮的图标 /// protected virtual Texture2D EnterTex => DefaultEnterTex; /// /// 获取进入按钮的文本 /// public virtual string EnterString => "WULA.PocketSpace.Enter".Translate(); /// /// 获取取消进入按钮的文本 /// public virtual string CancelEnterString => "WULA.PocketSpace.CancelEnter".Translate(); /// /// 获取进入中的文本 /// public virtual string EnteringString => "WULA.PocketSpace.Entering".Translate(); /// 加载是否正在进行(模仿原版 MapPortal.LoadInProgress) public bool LoadInProgress { get { if (leftToLoad != null) { return leftToLoad.Any(); } return false; } } /// 是否有Pawn可以加载任何东西(模仿原版 MapPortal.AnyPawnCanLoadAnythingNow) 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(this, oneStackOnly: false); } #endregion #region 基础重写方法 public override void PostMake() { base.PostMake(); if (innerContainer == null) { innerContainer = new ThingOwner(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(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 口袋空间核心方法 /// /// 检查是否可以进入口袋空间 /// public bool CanEnterPocketSpace() { if (!allowDirectAccess) { return false; // 需要特殊权限 } if (!Spawned) { return false; } return true; } /// /// 进入口袋空间 - 基于原版PocketMapUtility实现 /// public void EnterPocketSpace(IEnumerable 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 pawnsToTransfer = new List(); 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); } } /// /// 切换到口袋空间视角 /// 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); } /// /// 创建口袋地图 - 使用原版PocketMapUtility(模仿 MapPortal.GeneratePocketMap) /// 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}"); } } /// /// 生成口袋地图的内部实现(模仿 MapPortal.GeneratePocketMapInt) /// protected virtual Map GeneratePocketMapInt() { return PocketMapUtility.GeneratePocketMap(new IntVec3(pocketMapSize.x, 1, pocketMapSize.z), mapGenerator, GetExtraGenSteps(), this.Map); } /// /// 获取额外的生成步骤(模仿 MapPortal.GetExtraGenSteps) /// protected virtual IEnumerable GetExtraGenSteps() { return System.Linq.Enumerable.Empty(); } /// /// 在口袋地图中创建退出点(模仿原版) /// 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}"); } } /// /// 将单个Pawn传送到口袋空间 /// 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; } /// /// 将所有物品和人员从口袋空间转移到主地图 /// private void TransferAllFromPocketToMainMap() { if (pocketMap == null || !Spawned) return; try { // 转移所有殖民者 List 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 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 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 outChildren) { ThingOwnerUtility.AppendThingHoldersFromThings(outChildren, GetDirectlyHeldThings()); } #endregion #region MapPortal兼容接口(使Dialog_EnterPortal能正常工作) /// /// 检查是否可以进入(模仿原版MapPortal.IsEnterable) /// 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; } /// /// 获取目标地图(模仿原版MapPortal.GetOtherMap) /// public virtual Map GetOtherMap() { if (pocketMap == null) { CreatePocketMap(); } return pocketMap; } /// /// 获取目标位置(模仿原版MapPortal.GetDestinationLocation) /// public virtual IntVec3 GetDestinationLocation() { if (exit != null) { return exit.Position; } return pocketMap?.Center ?? IntVec3.Invalid; } /// /// 处理进入事件(模仿原版MapPortal.OnEntered) /// public virtual void OnEntered(Pawn pawn) { // 通知物品被添加(用于统计和管理) Notify_ThingAdded(pawn); // 播放传送音效(如果存在) if (Find.CurrentMap == this.Map) { // 可以在这里添加音效播放 // def.portal?.traverseSound?.PlayOneShot(this); } } /// /// 打开殖民者选择对话框(模仿原版Dialog_EnterPortal的功能) /// private void OpenPawnSelectionDialog() { // 获取所有可用的殖民者 List 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 options = new List(); // 添加单个殖民者选项 foreach (Pawn pawn in availablePawns) { FloatMenuOption option = new FloatMenuOption( $"{pawn.LabelShort}", delegate { EnterPocketSpace(new List { 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的物品传送方法 /// /// 通知有物品被添加(模仿原版 MapPortal.Notify_ThingAdded) /// public void Notify_ThingAdded(Thing t) { SubtractFromToLoadList(t, t.stackCount); } /// /// 添加到加载列表(模仿原版 MapPortal.AddToTheToLoadList) /// public void AddToTheToLoadList(TransferableOneWay t, int count) { if (!t.HasAnyThing || count <= 0) { return; } if (leftToLoad == null) { leftToLoad = new List(); } 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); } } /// /// 从加载列表中减去(模仿原版 MapPortal.SubtractFromToLoadList) /// 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; } /// /// 取消加载(模仿原版 MapPortal.CancelLoad) /// public void CancelLoad() { // 简化版本,只清理列表 if (leftToLoad != null) { leftToLoad.Clear(); } } #endregion #region 穿梭机状态变化处理 /// /// 更新口袋空间中退出点的目标位置(处理穿梭机位置变化) /// 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}"); } } /// /// 重写Tick方法,定期检查穿梭机状态变化 /// protected override void Tick() { base.Tick(); // 每隔一段时间检查退出点目标是否需要更新(处理穿梭机移动的情况) if (this.IsHashIntervalTick(2500) && pocketMapGenerated && exit != null) { UpdateExitPointTarget(); } } /// /// 重写 SpawnSetup,确保位置变化时更新退出点 /// 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(this, oneStackOnly: false); } // 简化实现,直接使用 innerContainer 而不是 PortalContainerProxy // 因为 PortalContainerProxy 需要 MapPortal 类型,但我们不是继承自 MapPortal // 更新退出点目标(处理穿梭机重新部署的情况) UpdateExitPointTarget(); // 从 ThingDef 中读取 portal 配置 if (def.HasModExtension()) { var portalProps = def.GetModExtension(); 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.GetNamed("AncientStockpile", false) ?? DefDatabase.GetNamed("Base_Player", false) ?? MapGeneratorDefOf.Base_Player; } if (exitDef == null) { exitDef = DefDatabase.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 } /// /// 口袋空间属性配置类 /// public class PocketMapProperties : DefModExtension { /// 口袋地图生成器 public MapGeneratorDef pocketMapGenerator; /// 退出点定义 public ThingDef exitDef; /// 口袋地图大小 public IntVec2 pocketMapSize = new IntVec2(13, 13); /// 允许直接访问 public bool allowDirectAccess = true; } }