Ghost z-levels move actions + mapping z-levels combine command (#654)

* some setup * fast ghost zlevels-moving * zlevel combine command for mapping
2024-12-09 20:39:46 +03:00
parent 5384adb2b3
commit bec8d7a7d4
17 changed files with 320 additions and 165 deletions
--- a/Content.Server/_CP14/ZLevels/EntitySystems/CP14StationZLevelsSystem.cs
+++ b/Content.Server/_CP14/ZLevels/EntitySystems/CP14StationZLevelsSystem.cs
@@ -0,0 +1,108 @@
+using Content.Server._CP14.ZLevels.Components;
+using Content.Server.Station.Components;
+using Content.Server.Station.Events;
+using Content.Server.Station.Systems;
+using Content.Shared.Maps;
+using Content.Shared.Station.Components;
+using Robust.Server.GameObjects;
+using Robust.Server.Maps;
+using Robust.Shared.Map;
+
+namespace Content.Server._CP14.ZLevels.EntitySystems;
+
+public sealed partial class CP14StationZLevelsSystem : EntitySystem
+{
+    [Dependency] private readonly MapSystem _map = default!;
+    [Dependency] private readonly StationSystem _station = default!;
+    [Dependency] private readonly MapLoaderSystem _mapLoader = default!;
+    [Dependency] private readonly TransformSystem _transform = default!;
+    [Dependency] private readonly ITileDefinitionManager _tileDefManager = default!;
+    [Dependency] private readonly TileSystem _tile = default!;
+    [Dependency] private readonly SharedMapSystem _maps = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+        InitializePortals();
+        InitActions();
+
+        SubscribeLocalEvent<CP14StationZLevelsComponent, StationPostInitEvent>(OnStationPostInit);
+    }
+
+    private void OnStationPostInit(Entity<CP14StationZLevelsComponent> ent, ref StationPostInitEvent args)
+    {
+        if (ent.Comp.Initialized)
+            return;
+
+        if (!TryComp(ent, out StationDataComponent? dataComp))
+        {
+            Log.Error($"Failed to init CP14StationZLevelsSystem: no StationData");
+            return;
+        }
+
+        var defaultMap = _station.GetLargestGrid(dataComp);
+
+        if (defaultMap is null)
+        {
+            Log.Error($"Failed to init CP14StationZLevelsSystem: defaultMap is null");
+            return;
+        }
+
+        ent.Comp.LevelEntities.Add(Transform(defaultMap.Value).MapID, ent.Comp.DefaultMapLevel);
+
+        ent.Comp.Initialized = true;
+
+        foreach (var (map, level) in ent.Comp.Levels)
+        {
+            if (ent.Comp.LevelEntities.ContainsValue(map))
+            {
+                Log.Error($"Key duplication for CP14StationZLevelsSystem at level {map}!");
+                continue;
+            }
+
+            var path = level.Path.ToString();
+            if (path is null)
+            {
+                Log.Error($"path {path} for CP14StationZLevelsSystem at level {map} don't exist!");
+                continue;
+            }
+
+            var mapUid = _map.CreateMap(out var mapId);
+            var member = EnsureComp<StationMemberComponent>(mapUid);
+            member.Station = ent;
+
+            Log.Info($"Created map {mapId} for CP14StationZLevelsSystem at level {map}");
+            var options = new MapLoadOptions { LoadMap = true };
+
+            if (!_mapLoader.TryLoad(mapId, path, out var grids, options))
+            {
+                Log.Error($"Failed to load map for CP14StationZLevelsSystem at level {map}!");
+                Del(mapUid);
+                continue;
+            }
+            ent.Comp.LevelEntities.Add(mapId, map);
+        }
+    }
+
+    public MapId? GetMapOffset(EntityUid mapUid, int offset)
+    {
+        var query = EntityQueryEnumerator<CP14StationZLevelsComponent>();
+        while (query.MoveNext(out var uid, out var zLevel))
+        {
+            if (!zLevel.LevelEntities.TryGetValue(Transform(mapUid).MapID, out var currentLevel))
+                continue;
+
+            var targetLevel = currentLevel + offset;
+
+            if (!zLevel.LevelEntities.ContainsValue(targetLevel))
+                continue;
+
+            foreach (var (key, value) in zLevel.LevelEntities)
+            {
+                if (value == targetLevel && _map.MapExists(key))
+                    return key;
+            }
+        }
+        return null;
+    }
+}