Add initial code

This commit is contained in:
Roscoe 2025-05-04 03:31:37 +01:00
commit 5e60c2deb1
Signed by: RoscoeDaWah
SSH key fingerprint: SHA256:Hqn452XQ1ETzUt/FthJu6+OFkS4NBxCv5VQSEvuk7CE
7 changed files with 250 additions and 0 deletions

41
.gitignore vendored Normal file
View file

@ -0,0 +1,41 @@
target/
!.mvn/wrapper/maven-wrapper.jar
!**/src/main/**/target/
!**/src/test/**/target/
### IntelliJ IDEA ###
.idea/modules.xml
.idea/jarRepositories.xml
.idea/compiler.xml
.idea/libraries/
*.iws
*.iml
*.ipr
### Eclipse ###
.apt_generated
.classpath
.factorypath
.project
.settings
.springBeans
.sts4-cache
### NetBeans ###
/nbproject/private/
/nbbuild/
/dist/
/nbdist/
/.nb-gradle/
build/
!**/src/main/**/build/
!**/src/test/**/build/
### VS Code ###
.vscode/
### Mac OS ###
.DS_Store
/tmp/
*.png

10
.idea/.gitignore generated vendored Normal file
View file

@ -0,0 +1,10 @@
# Default ignored files
/shelf/
/workspace.xml
# Editor-based HTTP Client requests
/httpRequests/
# Datasource local storage ignored files
/dataSources/
/dataSources.local.xml
/discord.xml

7
.idea/encodings.xml generated Normal file
View file

@ -0,0 +1,7 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="Encoding">
<file url="file://$PROJECT_DIR$/src/main/java" charset="UTF-8" />
<file url="file://$PROJECT_DIR$/src/main/resources" charset="UTF-8" />
</component>
</project>

14
.idea/misc.xml generated Normal file
View file

@ -0,0 +1,14 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ExternalStorageConfigurationManager" enabled="true" />
<component name="MavenProjectsManager">
<option name="originalFiles">
<list>
<option value="$PROJECT_DIR$/pom.xml" />
</list>
</option>
</component>
<component name="ProjectRootManager" version="2" languageLevel="JDK_21" default="true" project-jdk-name="homebrew-23" project-jdk-type="JavaSDK">
<output url="file://$PROJECT_DIR$/out" />
</component>
</project>

6
.idea/vcs.xml generated Normal file
View file

@ -0,0 +1,6 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="$PROJECT_DIR$" vcs="Git" />
</component>
</project>

42
pom.xml Normal file
View file

@ -0,0 +1,42 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>moe.wah</groupId>
<artifactId>PGN-ator</artifactId>
<version>1.0-SNAPSHOT</version>
<properties>
<maven.compiler.source>21</maven.compiler.source>
<maven.compiler.target>21</maven.compiler.target>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>
<repositories>
<repository>
<id>jitpack.io</id>
<url>https://jitpack.io</url>
</repository>
</repositories>
<dependencies>
<dependency>
<groupId>com.github.bhlangonijr</groupId>
<artifactId>chesslib</artifactId>
<version>1.3.4</version>
</dependency>
<dependency>
<groupId>com.github.alexandreroman</groupId>
<artifactId>chessimage</artifactId>
<version>1.0.0</version>
</dependency>
<dependency>
<groupId>com.sksamuel.scrimage</groupId>
<artifactId>scrimage-core</artifactId>
<version>4.3.1</version>
</dependency>
</dependencies>
</project>

View file

@ -0,0 +1,130 @@
package moe.wah;
import com.github.alexandreroman.chessimage.ChessRenderer;
import com.github.alexandreroman.chessimage.ChessThemeLibrary;
import com.github.bhlangonijr.chesslib.Board;
import com.github.bhlangonijr.chesslib.game.Game;
import com.github.bhlangonijr.chesslib.move.Move;
import com.github.bhlangonijr.chesslib.move.MoveList;
import com.github.bhlangonijr.chesslib.pgn.PgnHolder;
import com.sksamuel.scrimage.ImmutableImage;
import com.sksamuel.scrimage.nio.StreamingGifWriter;
import com.sksamuel.scrimage.nio.StreamingGifWriter.GifStream;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.Duration;
import java.time.Instant;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Main {
public static void main(String[] args) throws Exception {
Files.createDirectories(Paths.get("./temp/"));
// Load PGN from file
PgnHolder pgn = new PgnHolder("game.pgn");
pgnToGif(pgn);
}
private static void pgnToGif(PgnHolder pgn) throws Exception {
ChessRenderer renderer = new ChessRenderer(ChessThemeLibrary.GREEN_THEME, 60);
Game game = pgnToGame(pgn);
String boardID = String.valueOf(Instant.now().toEpochMilli());
game.loadMoveText();
System.out.println(game.getTermination());
MoveList moves = game.getHalfMoves();
Board board = new Board();
int index = 1;
for (Move move : moves) {
board.doMove(move);
renderBoard(board, renderer, boardID, index);
// Add 5 extra frames of final board
if (index == moves.size()) {
for (int x = 0; x < 5; x++) {
index++;
renderBoard(board, renderer, boardID, index);
}
}
index++;
}
System.out.println("Rendered " + moves.size() + " moves");
boardToGif(boardID);
}
private static void renderBoard(Board board, ChessRenderer renderer, String boardID, int index) {
String index_padded = String.format("%03d", index);
File targetFile = new File("./temp/", "board_" + boardID + "_" + index_padded + ".png");
try (FileOutputStream out = new FileOutputStream(targetFile)) {
renderer.render(board.getFen(), out);
} catch (IOException e) {
System.out.println("Error rendering move " + index_padded + ", " + e.getMessage());
}
}
private static Game pgnToGame(PgnHolder pgn) throws Exception {
pgn.loadPgn();
return pgn.getGames().getFirst();
}
/**
* Converts a chess board to a GIF
* Output file is board_x.gif where x is the boardID
* @param boardID ID of the board (UNIX epoch)
*/
public static void boardToGif(String boardID) throws Exception {
List<File> boardFiles = getBoardFiles(new File("./temp/"), boardID);
if (boardFiles.isEmpty()) {
throw new RuntimeException("No PNG files found for board with ID: " + boardID);
}
List<BufferedImage> images = new ArrayList<>();
for (File file : boardFiles) {
BufferedImage image = ImageIO.read(file);
images.add(image);
}
System.out.println("There are " + (long) images.size() + " frames for board " + boardID);
String gif_name = "board_" + boardID + ".gif";
StreamingGifWriter writer = new StreamingGifWriter(Duration.ofMillis(500), true, false);
GifStream gif = writer.prepareStream(gif_name, BufferedImage.TYPE_INT_ARGB);
for (BufferedImage image : images) {
gif.writeFrame(ImmutableImage.fromAwt(image));
}
gif.close();
System.out.println("Rendered " + gif_name);
for (File file : boardFiles) {
boolean delete_success = file.delete();
if (!delete_success) {
System.out.println("Failed to delete " + file.getName() + "!");
}
}
System.out.println("Removed " + boardFiles.size() + " temp files");
}
/**
* Gets the PNG files for the specified board
* @param directory Directory to search
* @param boardID Board ID (UNIX epoch)
* @return A list of board Files
*/
private static List<File> getBoardFiles(File directory, String boardID) {
List<File> boardFiles = new ArrayList<>();
File[] files = directory.listFiles();
assert files != null;
Arrays.sort(files);
for (File file : files) {
if (file.isFile() && file.getName().toLowerCase().endsWith(".png")
&& file.getName().startsWith("board_" + boardID)) {
boardFiles.add(file);
}
}
return boardFiles;
}
}