1 files changed, 17 insertions, 19 deletions

diff --git a/src/ch/epfl/xblast/client/XBlastComponent.java b/src/ch/epfl/xblast/client/XBlastComponent.java
index fe5bf5a..0ef8c5c 100644
--- a/src/ch/epfl/xblast/client/XBlastComponent.java
+++ b/src/ch/epfl/xblast/client/XBlastComponent.java
@@ -107,27 +107,25 @@ public final class XBlastComponent extends JComponent {
                 drawString(g, e.getKey(), e.getValue().toString());
         }
 
-    }
-
-    private static Comparator<GameState.Player> buildPlayerComparator(PlayerID priorityPlayer) {
-        Comparator<GameState.Player> coordinateComparator =
-                (p1, p2) -> p1.position().compareTo(p2.position());
-
-        Comparator<GameState.Player> idComparator =
-                (p1, p2) -> p1.id() == priorityPlayer ? -1 : 0;
+        /**
+         * Draws the players on the graphic context.
+         *
+         * @param g       the graphic context
+         * @param players the list of players to be displayed
+         */
+        private static void drawPlayers(Graphics2D g, List<GameState.Player> players) {
+            for (GameState.Player p : players)
+                drawImage(g, Grid.positionForPlayer(p), p.image());
+        }
 
-        return coordinateComparator.thenComparing(idComparator);
     }
 
-    /**
-     * Draws the players on the graphic context.
-     *
-     * @param g       the graphic context
-     * @param players the list of players to be displayed
-     */
-    private static void drawPlayers(Graphics2D g, List<GameState.Player> players) {
-        for (GameState.Player p : players)
-            Painter.drawImage(g, Grid.positionForPlayer(p), p.image());
+    private static List<GameState.Player> sortPlayers(List<GameState.Player> players, PlayerID currentPlayerID) {
+        return Lists.sorted(
+                players,
+                GameState.Player
+                        .POSITION_COMPARATOR
+                        .thenComparing(GameState.Player.idPushingComparator(currentPlayerID)));
     }
 
     private GameState gameState;
@@ -172,7 +170,7 @@ public final class XBlastComponent extends JComponent {
         Painter.drawImageMap(g, Lists.linearMap(Grid.SCORE_BG_POSITIONS, this.gameState.scores()));
         Painter.drawImageMap(g, Lists.linearMap(Grid.TIME_LINE_POSITIONS, this.gameState.ticks()));
         Painter.drawStringMap(g, Lists.traverseMaps(Lists.invertMap(Grid.SCORE_TXT_POSITIONS), this.gameState.playerScores()));
-        drawPlayers(g, Lists.sorted(this.gameState.players(), buildPlayerComparator(this.playerID)));
+        Painter.drawPlayers(g, sortPlayers(this.gameState.players(), this.playerID));
     }
 
 }