X-Git-Url: https://git.hedgecode.org/?p=chesshog.git;a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fhedgecode%2Fchess%2Fposition%2FDiagramPosition.java;h=1d2875956e18c6ce36c1031b8ae7ed1dbc1b83ba;hp=e12b477678edec971737305217782b6320fab670;hb=c30c1d1230ec2771527b172a4f2ee8a44e7d4bda;hpb=fe0dd25e16055463de62d3c43f5b0c05a1eaaa75

diff --git a/src/main/java/org/hedgecode/chess/position/DiagramPosition.java b/src/main/java/org/hedgecode/chess/position/DiagramPosition.java
index e12b477..1d28759 100644
--- a/src/main/java/org/hedgecode/chess/position/DiagramPosition.java
+++ b/src/main/java/org/hedgecode/chess/position/DiagramPosition.java
@@ -18,6 +18,7 @@ package org.hedgecode.chess.position;
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.TreeMap;
 
 /**
  *
@@ -55,6 +56,19 @@ public class DiagramPosition implements Position {
     }
 
     @Override
+    public Map<Square, ColorPiece> getSquares(SquareSort sorting) {
+        if (sorting == null)
+            sorting = SquareSort.defaultSort();
+
+        TreeMap<Square, ColorPiece> sortedSquares = new TreeMap<>(
+                sorting.comparator()
+        );
+        sortedSquares.putAll(squares);
+
+        return sortedSquares;
+    }
+
+    @Override
     public Map<Square, ColorPiece> getSquarePieces(Color color) {
         return getSquarePieces(color, null);
     }