Add entity rotation to map view

Workbench-Team · Aug 7, 2023 · ceeea53 · ceeea53
1 parent 9ab100a
commit ceeea53
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/Content.MapRenderer/Painters/EntityData.cs b/Content.MapRenderer/Painters/EntityData.cs
@@ -3,7 +3,7 @@
 
 namespace Content.MapRenderer.Painters;
 
-public readonly record struct EntityData(EntityUid Owner, SpriteComponent Sprite, float X, float Y)
+public readonly record struct EntityData(EntityUid Owner, SpriteComponent Sprite, float X, float Y, Angle Rotation)
 {
     public readonly EntityUid Owner = Owner;
 
@@ -12,4 +12,6 @@ public readonly record struct EntityData(EntityUid Owner, SpriteComponent Sprite
     public readonly float X = X;
 
     public readonly float Y = Y;
+
+    public readonly Angle Rotation = Rotation;
 }
diff --git a/Content.MapRenderer/Painters/EntityPainter.cs b/Content.MapRenderer/Painters/EntityPainter.cs
@@ -135,6 +135,7 @@ public void Run(Image canvas, EntityData entity, SharedTransformSystem xformSyst
 
             var pointX = (int) entity.X - (imgX / 2) + EyeManager.PixelsPerMeter / 2;
             var pointY = (int) entity.Y - (imgY / 2) + EyeManager.PixelsPerMeter / 2;
+            image.Mutate(o => o.Rotate((float) -entity.Rotation.Degrees));
             canvas.Mutate(o => o.DrawImage(image, new Point(pointX, pointY), 1));
         }
     }

diff --git a/Content.MapRenderer/Painters/GridPainter.cs b/Content.MapRenderer/Painters/GridPainter.cs
@@ -92,7 +92,7 @@ private ConcurrentDictionary<EntityUid, List<EntityData>> GetEntities()
                     var position = transform.LocalPosition;
 
                     var (x, y) = TransformLocalPosition(position, grid);
-                    var data = new EntityData(entity, sprite, x, y);
+                    var data = new EntityData(entity, sprite, x, y, transform.LocalRotation);
 
                     components.GetOrAdd(transform.GridUid.Value, _ => new List<EntityData>()).Add(data);
                 }