knight fixing, custom piece images

README.md

@@ -31,7 +31,7 @@ use fenix::*;
 // 4. The ChessAssets instance
 // This returns an result(), or an error if it fails to convert the fen to image.
 // The board also auto rotates depending on whose turn it is.
-let result = fen_to_board_img("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w", "chess_board.png", 1, ChessAssets::default());
+let result = fen_to_board_img("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w", "chess_board.png", 1, &ChessAssets::default());
 
 result.unwrap();
 

src/chess_assets/asset_manager.rs

@@ -2,6 +2,8 @@ use std::collections::HashMap;
 
 use image::{ImageFormat, RgbaImage};
 
+use crate::errors::FenToImgError;
+
 pub struct ChessAssets {
  pub piece_images: HashMap<char, RgbaImage>,
  pub board_image: RgbaImage,
@@ -137,3 +139,78 @@ impl Default for ChessAssets {
  }
  }
 }
+
+impl ChessAssets {
+ pub fn new(piece_images: HashMap<char, RgbaImage>, board_image: RgbaImage) -> Self {
+ ChessAssets {
+ piece_images,
+ board_image,
+ }
+ }
+ // Black Pieces
+ /// Piece size requirements: 16x16 Canvas size, Minimum of 1 pixel of border
+ /// image_path: Path to the image file
+ pub fn set_black_king(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('k', image);
+ Ok(())
+ }
+ pub fn set_black_queen(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('q', image);
+ Ok(())
+ }
+ pub fn set_black_rook(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('r', image);
+ Ok(())
+ }
+ pub fn set_black_bishop(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('b', image);
+ Ok(())
+ }
+ pub fn set_black_knight(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('n', image);
+ Ok(())
+ }
+ pub fn set_black_pawn(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('p', image);
+ Ok(())
+ }
+ // White Pieces
+ pub fn set_white_king(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('K', image);
+ Ok(())
+ }
+ pub fn set_white_queen(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('Q', image);
+ Ok(())
+ }
+ pub fn set_white_rook(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('R', image);
+ Ok(())
+ }
+ pub fn set_white_bishop(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('B', image);
+ Ok(())
+ }
+ pub fn set_white_knight(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('N', image);
+ Ok(())
+ }
+ pub fn set_white_pawn(&mut self, image_path: &str) -> Result<(), FenToImgError> {
+ let image = image::open(image_path).map_err(|err| FenToImgError::ImageError(err))?.to_rgba8();
+ self.piece_images.insert('P', image);
+ Ok(())
+ }
+
+
+}

src/errors.rs

@@ -16,4 +16,4 @@ impl From<&str> for FenToImgError {
  fn from(err: &str) -> FenToImgError {
  FenToImgError::FenError(err.to_string())
  }
-}
+}