build from fen-like string

Author: ipax77

src/dsstats.db8cli/Program.cs

@@ -87,9 +87,18 @@ static void Main(string[] args)
                     from rp in r.ReplayPlayers
                     from s in rp.Spawns
                     from su in s.Units
-                    where r.GameTime > minDate && rp.Team == 1 && rp.Race == Commander.Zerg && s.Breakpoint == Breakpoint.All
+                    where r.GameTime > minDate && rp.Team == 2 && rp.Race == Commander.Terran && s.Breakpoint == Breakpoint.All
+                        && su.Unit.Name == "Thor"
                     select su.Poss;
-        var posString = query.ToList();
+        var posString = query
+            .OrderByDescending(o => o.Length)
+            .Take(4)
+            .ToList();
+
+        foreach (var pos in posString)
+        {
+            Console.WriteLine(pos);
+        }
 
         var allCoords = new List<(int x, int y)>();
 

src/dsstats.maui/dsstats.builder/dsstats.builder.tests/FenSharedTests.cs

@@ -0,0 +1,34 @@
+using dsstats.shared;
+using dsstats.shared.DsFen;
+
+namespace dsstats.builder.tests;
+
+[TestClass]
+public sealed class FenSharedTests
+{
+    [TestMethod]
+    public void FenRoundTrip_FromSpawnDto()
+    {
+        var spawn = new SpawnDto
+        {
+            Units = new List<SpawnUnitDto>
+        {
+            new() { Unit = new() { Name = "Zergling" }, Poss = "91,80" },
+            new() { Unit = new() { Name = "Mutalisk" }, Poss = "92,81" },
+        }
+        };
+
+        var cmdr = Commander.Zerg;
+        int team = 2;
+
+        var fen = DsFen.GetFen(spawn, cmdr, team);
+
+        var newSpawn = new SpawnDto { Units = [] };
+        DsFen.ApplyFen(fen, newSpawn, out cmdr, out team);
+
+        var newFen = DsFen.GetFen(newSpawn, cmdr, team);
+
+        Assert.AreEqual(fen, newFen);
+    }
+
+}

src/dsstats.maui/dsstats.builder/dsstats.builder.tests/FenTests.cs

@@ -0,0 +1,86 @@
+namespace dsstats.builder.tests;
+
+[TestClass]
+public sealed class FenTests
+{
+    [TestMethod]
+    public void CanCreateFen()
+    {
+        int team = 2;
+        var build = CmdrBuildFactory.Create(shared.Commander.Zerg);
+        Assert.IsNotNull(build);
+        var possString = "90,79";
+        var unitName = "Ultralisk";
+
+        var buildArea = new BuildArea(team);
+        buildArea.PlaceUnits(unitName, possString, team);
+
+        var fen = buildArea.ToFenString(build);
+
+        var fenBuildArea = new BuildArea(team);
+        fenBuildArea.FromFenString(fen, build);
+        var reverseFen = fenBuildArea.ToFenString(build);
+
+        Assert.AreEqual(fen, reverseFen);
+    }
+
+    [TestMethod]
+    public void CanCreateFen_WithAirUnit()
+    {
+        int team = 2;
+        var build = CmdrBuildFactory.Create(shared.Commander.Zerg);
+        Assert.IsNotNull(build);
+
+        var buildArea = new BuildArea(team);
+        buildArea.PlaceUnits("Mutalisk", "91,80", team); // air unit
+
+        var fen = buildArea.ToFenString(build);
+        var fenBuildArea = new BuildArea(team);
+        fenBuildArea.FromFenString(fen, build);
+        var reverseFen = fenBuildArea.ToFenString(build);
+
+        Assert.AreEqual(fen, reverseFen);
+    }
+
+    [TestMethod]
+    public void CanCreateFen_WithMultipleUnits()
+    {
+        int team = 2;
+        var build = CmdrBuildFactory.Create(shared.Commander.Zerg);
+        Assert.IsNotNull(build);
+
+        var buildArea = new BuildArea(team);
+        buildArea.PlaceUnits("Zergling", "91,80", team);
+        buildArea.PlaceUnits("Queen", "92,80", team);
+        buildArea.PlaceUnits("Overseer", "91,81", team); // air
+
+        var fen = buildArea.ToFenString(build);
+        var fenBuildArea = new BuildArea(team);
+        fenBuildArea.FromFenString(fen, build);
+        var reverseFen = fenBuildArea.ToFenString(build);
+
+        Assert.AreEqual(fen, reverseFen);
+    }
+
+    [TestMethod]
+    public void CanCreateFen_WithToggledUnit()
+    {
+        int team = 2;
+        var build = CmdrBuildFactory.Create(shared.Commander.Zerg);
+        Assert.IsNotNull(build);
+
+        // Assume "Lurker" shares key 'a' with Hydralisk, but Hydralisk is active,
+        // so Lurker should appear as 'A' (uppercase)
+        var buildArea = new BuildArea(team);
+        buildArea.PlaceUnits("Lurker", "93,82", team);
+
+        var fen = buildArea.ToFenString(build);
+        Assert.IsTrue(fen.Contains('A'), $"fen does not contain upper case A: {fen}");
+
+        var fenBuildArea = new BuildArea(team);
+        fenBuildArea.FromFenString(fen, build);
+        var reverseFen = fenBuildArea.ToFenString(build);
+
+        Assert.AreEqual(fen, reverseFen);
+    }
+}

src/dsstats.maui/dsstats.builder/dsstats.builder/BuildArea.cs

@@ -1,3 +1,5 @@
+using dsstats.shared.DsFen;
+
 namespace dsstats.builder;
 
 /// <summary>
@@ -50,6 +52,7 @@ public List<InputEvent> GetBuildEvents(ScreenArea screenArea, CmdrBuild build)
         List<BuildUnit> topUnits = [];
         List<BuildUnit> centerUnits = [];
         List<BuildUnit> bottomUnits = [];
+        WorkerMenu workerMenu = new();
 
         foreach (var unit in allUnits)
         {
@@ -71,15 +74,15 @@ public List<InputEvent> GetBuildEvents(ScreenArea screenArea, CmdrBuild build)
 
         foreach (var centerUnit in centerUnits)
         {
-            events.AddRange(build.GetBuildEvents(centerUnit.UnitName, centerUnit.Pos, screenArea));
+            events.AddRange(build.GetBuildEvents(centerUnit.UnitName, centerUnit.Pos, screenArea, workerMenu));
         }
         if (topUnits.Count > 0)
         {
             events.AddRange(DsBuilder.ScrollY(Convert.ToInt32(250 * screenArea._scaleY), screenArea.GetCenter()));
             foreach (var topUnit in topUnits)
             {
                 events.AddRange(build.GetBuildEvents(topUnit.UnitName,
-                    topUnit.Pos with { Y = topUnit.Pos.Y + Convert.ToInt32(125 * screenArea._scaleY) }, screenArea));
+                    topUnit.Pos with { Y = topUnit.Pos.Y + Convert.ToInt32(125 * screenArea._scaleY) }, screenArea, workerMenu));
             }
         }
         if (bottomUnits.Count > 0)
@@ -91,7 +94,7 @@ public List<InputEvent> GetBuildEvents(ScreenArea screenArea, CmdrBuild build)
             foreach (var bottomUnit in bottomUnits)
             {
                 events.AddRange(build.GetBuildEvents(bottomUnit.UnitName,
-                    bottomUnit.Pos with { Y = bottomUnit.Pos.Y - Convert.ToInt32(300 * screenArea._scaleY) }, screenArea));
+                    bottomUnit.Pos with { Y = bottomUnit.Pos.Y - Convert.ToInt32(300 * screenArea._scaleY) }, screenArea, workerMenu));
             }
         }
 
@@ -136,7 +139,10 @@ private List<RlPoint> GetUnitPositions(string unitString, int team)
         for (int i = 0; i < stringPoints.Length; i += 2)
         {
             RlPoint mapPoint = new(int.Parse(stringPoints[i]), int.Parse(stringPoints[i + 1]));
-            mapPoints.Add(NormalizeToTop(mapPoint));
+            if (IsPointInPolygon(mapPoint, polygon))
+            {
+                mapPoints.Add(NormalizeToTop(mapPoint));
+            }
         }
         return mapPoints;
     }
@@ -157,6 +163,134 @@ public RlPoint GetCenter()
         return new(x1 + ((x2 - x1) / 2), y1 + ((y2 - y1) / 2));
     }
 
+    public string ToFenString(CmdrBuild build)
+    {
+        var groundUnits = new Dictionary<RlPoint, char>();
+        var airUnits = new Dictionary<RlPoint, char>();
+
+        foreach (var kv in units)
+        {
+            string unitName = kv.Key;
+            var buildOption = build.GetUnitBuildOption(unitName);
+            if (buildOption == null) continue;
+
+            char key = buildOption.RequiresToggle && !buildOption.IsActive
+                ? char.ToUpper(buildOption.Key)
+                : buildOption.Key;
+
+            foreach (var pos in kv.Value)
+            {
+                if (buildOption.IsAir)
+                    airUnits[pos] = key;
+                else
+                    groundUnits[pos] = key;
+            }
+        }
+
+        if (groundUnits.Count == 0 && airUnits.Count == 0)
+            return "";
+
+        // Determine grid bounds (combined for both layers)
+        var allPoints = groundUnits.Keys.Concat(airUnits.Keys).ToList();
+        int minX = allPoints.Min(p => p.X);
+        int maxX = allPoints.Max(p => p.X);
+        int minY = allPoints.Min(p => p.Y);
+        int maxY = allPoints.Max(p => p.Y);
+
+        string EncodeLayer(Dictionary<RlPoint, char> layer)
+        {
+            var rows = new List<string>();
+            for (int y = minY; y <= maxY; y++)
+            {
+                string row = "";
+                int emptyCount = 0;
+
+                for (int x = minX; x <= maxX; x++)
+                {
+                    var pt = new RlPoint(x, y);
+                    if (layer.TryGetValue(pt, out char key))
+                    {
+                        if (emptyCount > 0)
+                        {
+                            row += emptyCount.ToString();
+                            emptyCount = 0;
+                        }
+                        row += key;
+                    }
+                    else
+                    {
+                        emptyCount++;
+                    }
+                }
+
+                if (emptyCount > 0)
+                    row += emptyCount.ToString();
+
+                rows.Add(row);
+            }
+
+            return string.Join("/", rows);
+        }
+
+        var groundFen = EncodeLayer(groundUnits);
+        var airFen = EncodeLayer(airUnits);
+
+        return $"{groundFen}|{airFen}";
+    }
+
+    public void FromFenString(string fen, CmdrBuild build)
+    {
+        if (string.IsNullOrWhiteSpace(fen))
+            return;
+
+        // Split into ground and air layers
+        var layers = fen.Split('|');
+        string groundLayer = layers.Length > 0 ? layers[0] : "";
+        string airLayer = layers.Length > 1 ? layers[1] : "";
+
+        ParseLayer(groundLayer, build, isAir: false);
+        if (!string.IsNullOrWhiteSpace(airLayer))
+            ParseLayer(airLayer, build, isAir: true);
+    }
+
+    private void ParseLayer(string fenLayer, CmdrBuild build, bool isAir)
+    {
+        if (string.IsNullOrWhiteSpace(fenLayer))
+            return;
+
+        int y = 0;
+        var rows = fenLayer.Split('/');
+        foreach (var row in rows)
+        {
+            int x = 0;
+            foreach (char ch in row)
+            {
+                if (char.IsDigit(ch))
+                {
+                    x += ch - '0';
+                }
+                else
+                {
+                    bool isUpper = char.IsUpper(ch);
+                    char key = char.ToLower(ch);
+
+                    var unitName = build.GetUnitNameFromKey(key, isAir, isUpper);
+                    if (unitName != null)
+                    {
+                        if (!units.TryGetValue(unitName, out var unitPositions))
+                            unitPositions = units[unitName] = [];
+
+                        unitPositions.Add(new RlPoint(x, y)); // normalized pos
+                    }
+
+                    x++;
+                }
+            }
+
+            y++;
+        }
+    }
+
     private bool IsPointInsideOrOnEdge(RlPoint p)
     {
         if (IsPointInPolygon(p, polygon))