CatStation/Content.Shared/Coordinates/Helpers/SnapgridHelper.cs

using System.Numerics;
using Robust.Shared.Map;
using Robust.Shared.Map.Components;

namespace Content.Shared.Coordinates.Helpers
{
    public static class SnapgridHelper
    {
        public static EntityCoordinates SnapToGrid(this EntityCoordinates coordinates, IEntityManager? entMan = null)
        {
            IoCManager.Resolve(ref entMan);
            var xformSys = entMan.System<SharedTransformSystem>();

            var gridId = xformSys.GetGrid(coordinates.EntityId);

            if (gridId == null)
            {
                var mapPos = xformSys.ToMapCoordinates(coordinates);
                var mapX = (int)Math.Floor(mapPos.X) + 0.5f;
                var mapY = (int)Math.Floor(mapPos.Y) + 0.5f;
                mapPos = new MapCoordinates(new Vector2(mapX, mapY), mapPos.MapId);
                return xformSys.ToCoordinates(coordinates.EntityId, mapPos);
            }

            var grid = entMan.GetComponent<MapGridComponent>(gridId.Value);
            var tileSize = grid.TileSize;
            var localPos = xformSys.WithEntityId(coordinates, gridId.Value).Position;
            var x = (int)Math.Floor(localPos.X / tileSize) + tileSize / 2f;
            var y = (int)Math.Floor(localPos.Y / tileSize) + tileSize / 2f;
            var gridPos = new EntityCoordinates(gridId.Value, new Vector2(x, y));
            return xformSys.WithEntityId(gridPos, coordinates.EntityId);
        }

        public static EntityCoordinates SnapToGrid(this EntityCoordinates coordinates, MapGridComponent grid)
        {
            var tileSize = grid.TileSize;

            var localPos = coordinates.Position;

            var x = (int)Math.Floor(localPos.X / tileSize) + tileSize / 2f;
            var y = (int)Math.Floor(localPos.Y / tileSize) + tileSize / 2f;

            return new EntityCoordinates(coordinates.EntityId, x, y);
        }
    }
}