Added PGNDatabase and PGNGame classes with support for PGN parsing
This commit is contained in:
parent
63317eaa0b
commit
f949a0132b
29
src/dev/kske/chess/pgn/PGNDatabase.java
Normal file
29
src/dev/kske/chess/pgn/PGNDatabase.java
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
package dev.kske.chess.pgn;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Scanner;
|
||||||
|
|
||||||
|
import dev.kske.chess.exception.ChessException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Project: <strong>Chess</strong><br>
|
||||||
|
* File: <strong>PGNDatabase.java</strong><br>
|
||||||
|
* Created: <strong>4 Oct 2019</strong><br>
|
||||||
|
* Author: <strong>Kai S. K. Engelbart</strong>
|
||||||
|
*/
|
||||||
|
public class PGNDatabase {
|
||||||
|
|
||||||
|
private final List<PGNGame> games = new ArrayList<>();
|
||||||
|
|
||||||
|
public void load(File pgnFile) {
|
||||||
|
try (Scanner sc = new Scanner(pgnFile)) {
|
||||||
|
while (sc.hasNext())
|
||||||
|
games.add(PGNGame.parse(sc));
|
||||||
|
} catch (FileNotFoundException | ChessException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
77
src/dev/kske/chess/pgn/PGNGame.java
Normal file
77
src/dev/kske/chess/pgn/PGNGame.java
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
package dev.kske.chess.pgn;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Scanner;
|
||||||
|
import java.util.regex.MatchResult;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
import dev.kske.chess.board.Board;
|
||||||
|
import dev.kske.chess.exception.ChessException;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Project: <strong>Chess</strong><br>
|
||||||
|
* File: <strong>PGNGame.java</strong><br>
|
||||||
|
* Created: <strong>22 Sep 2019</strong><br>
|
||||||
|
* Author: <strong>Kai S. K. Engelbart</strong>
|
||||||
|
*/
|
||||||
|
public class PGNGame {
|
||||||
|
|
||||||
|
private final Map<String, String> tagPairs = new HashMap<>(7);
|
||||||
|
private final Board board = new Board();
|
||||||
|
|
||||||
|
public static PGNGame parse(Scanner sc) throws ChessException {
|
||||||
|
PGNGame game = new PGNGame();
|
||||||
|
|
||||||
|
MatchResult matchResult;
|
||||||
|
Pattern tagPairPattern = Pattern.compile("\\[(\\w+) \"(.*)\"]"),
|
||||||
|
movePattern = Pattern.compile("\\d+\\. (\\S+)\\s(\\S+)"),
|
||||||
|
nagPattern = Pattern.compile("(\\$\\d{1,3})*"),
|
||||||
|
terminationMarkerPattern = Pattern.compile("1-0|0-1|1\\/2-1\\/2|\\*");
|
||||||
|
|
||||||
|
// Parse tag pairs
|
||||||
|
while (true) {
|
||||||
|
if (sc.findInLine(tagPairPattern) != null) {
|
||||||
|
matchResult = sc.match();
|
||||||
|
if (matchResult.groupCount() == 2) game.setTag(matchResult.group(0), matchResult.group(1));
|
||||||
|
else break;
|
||||||
|
} else break;
|
||||||
|
sc.nextLine();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse movetext
|
||||||
|
while (true) {
|
||||||
|
// Skip NAG (Numeric Annotation Glyph)
|
||||||
|
sc.skip(nagPattern);
|
||||||
|
|
||||||
|
// TODO: Parse RAV (Recursive Annotation Variation)
|
||||||
|
|
||||||
|
sc.findWithinHorizon(movePattern, 20);
|
||||||
|
matchResult = sc.match();
|
||||||
|
if (matchResult.groupCount() > 0) for (int i = 1; i < matchResult.groupCount() + 1; i++) {
|
||||||
|
game.board.move(matchResult.group(i));
|
||||||
|
System.out.println(game.board.toFEN());
|
||||||
|
}
|
||||||
|
else break;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse game termination marker
|
||||||
|
if (sc.hasNext(terminationMarkerPattern)) {
|
||||||
|
sc.next(terminationMarkerPattern);
|
||||||
|
} else throw new ChessException("Game termination marker expected");
|
||||||
|
|
||||||
|
return game;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getTag(String tagName) {
|
||||||
|
return tagPairs.get(tagName);
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean hasTag(String tagName) {
|
||||||
|
return tagPairs.containsKey(tagName);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setTag(String tagName, String tagValue) {
|
||||||
|
tagPairs.put(tagName, tagValue);
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user