Fix featurecat#820 (playouts "1.5k" is wrongly read as 15000 in SGF)

xiaoyifang · Nov 28, 2020 · 17ce537 · 17ce537
1 parent ca8691b
commit 17ce537
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/main/java/featurecat/lizzie/rules/SGFParser.java b/src/main/java/featurecat/lizzie/rules/SGFParser.java
@@ -291,12 +291,12 @@ private static BoardHistoryList parseValue(
             line1[1] =
                 line1[1].replaceAll(",", "."); // fix a decimal representation localization issue
             Lizzie.board.getData().winrate = 100 - Double.parseDouble(line1[1]);
+            String playoutsStr = line1[2];
+            int kilo = 1000;
+            int playoutsUnit =
+                playoutsStr.endsWith("m") ? kilo * kilo : playoutsStr.endsWith("k") ? kilo : 1;
             int numPlayouts =
-                Integer.parseInt(
-                    line1[2]
-                        .replaceAll("k", "000")
-                        .replaceAll("m", "000000")
-                        .replaceAll("[^0-9]", ""));
+            (int) (Double.parseDouble(playoutsStr.replaceAll("[^0-9.]", "")) * playoutsUnit);
             Lizzie.board.getData().setPlayouts(numPlayouts);
             if (numPlayouts > 0 && !line2.isEmpty()) {
               Lizzie.board.getData().bestMoves = Lizzie.leelaz.parseInfo(line2);