388 lines
9.6 KiB
Rust
388 lines
9.6 KiB
Rust
// https://red-bean.com/sgf/
|
|
// https://red-bean.com/sgf/user_guide/index.html
|
|
// https://red-bean.com/sgf/sgf4.html
|
|
|
|
// todo: support collections in a file
|
|
// Properties to support. Remove each one as it gets support.
|
|
// B
|
|
// KO
|
|
// MN
|
|
// W
|
|
// AB
|
|
// AE
|
|
// AW
|
|
// PL
|
|
// C
|
|
// DM
|
|
// GB
|
|
// GW
|
|
// HO
|
|
// N
|
|
// UC
|
|
// V
|
|
// BM
|
|
// DO
|
|
// IT
|
|
// TE
|
|
// AR
|
|
// CR
|
|
// DD
|
|
// LB
|
|
// LN
|
|
// MA
|
|
// SL
|
|
// SQ
|
|
// TR
|
|
// AP
|
|
// CA
|
|
// FF
|
|
// GM
|
|
// ST
|
|
// SZ
|
|
// AN
|
|
// BR
|
|
// BT
|
|
// CP
|
|
// DT
|
|
// EV
|
|
// GN
|
|
// GC
|
|
// ON
|
|
// OT
|
|
// PB
|
|
// PC
|
|
// PW
|
|
// RE
|
|
// RO
|
|
// RU
|
|
// SO
|
|
// TM
|
|
// US
|
|
// WR
|
|
// WT
|
|
// BL
|
|
// OB
|
|
// OW
|
|
// WL
|
|
// FG
|
|
// PM
|
|
// VW
|
|
|
|
use crate::{
|
|
date::{parse_date_field, Date},
|
|
tree::{Size, Tree},
|
|
Error,
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
use typeshare::typeshare;
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct Game {
|
|
pub board_size: Size,
|
|
pub info: GameInfo,
|
|
|
|
pub tree: Tree,
|
|
}
|
|
|
|
impl TryFrom<Tree> for Game {
|
|
type Error = Error;
|
|
|
|
fn try_from(tree: Tree) -> Result<Self, Self::Error> {
|
|
let board_size = match tree.sequence[0].find_prop("SZ") {
|
|
Some(prop) => Size::try_from(prop.values[0].as_str())?,
|
|
None => Size {
|
|
width: 19,
|
|
height: 19,
|
|
},
|
|
};
|
|
let mut info = GameInfo::default();
|
|
info.app_name = tree.sequence[0]
|
|
.find_prop("AP")
|
|
.map(|prop| prop.values[0].clone());
|
|
info.black_player = tree.sequence[0]
|
|
.find_prop("PB")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
info.black_rank = tree.sequence[0]
|
|
.find_prop("BR")
|
|
.and_then(|prop| Rank::try_from(prop.values[0].as_str()).ok());
|
|
|
|
info.white_player = tree.sequence[0]
|
|
.find_prop("PW")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
info.white_rank = tree.sequence[0]
|
|
.find_prop("WR")
|
|
.and_then(|prop| Rank::try_from(prop.values[0].as_str()).ok());
|
|
|
|
info.result = tree.sequence[0]
|
|
.find_prop("RE")
|
|
.and_then(|prop| GameResult::try_from(prop.values[0].as_str()).ok());
|
|
|
|
info.time_limits = tree.sequence[0]
|
|
.find_prop("TM")
|
|
.and_then(|prop| prop.values[0].parse::<u64>().ok())
|
|
.and_then(|seconds| Some(std::time::Duration::from_secs(seconds)));
|
|
|
|
info.date = tree.sequence[0]
|
|
.find_prop("DT")
|
|
.and_then(|prop| {
|
|
let v = prop
|
|
.values
|
|
.iter()
|
|
.map(|val| parse_date_field(val))
|
|
.fold(Ok(vec![]), |acc, v| match (acc, v) {
|
|
(Ok(mut acc), Ok(mut values)) => {
|
|
acc.append(&mut values);
|
|
Ok(acc)
|
|
}
|
|
(Ok(_), Err(err)) => Err(err),
|
|
(Err(err), _) => Err(err),
|
|
})
|
|
.ok()?;
|
|
Some(v)
|
|
})
|
|
.unwrap_or(vec![]);
|
|
|
|
info.event = tree.sequence[0]
|
|
.find_prop("EV")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
info.round = tree.sequence[0]
|
|
.find_prop("RO")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
info.source = tree.sequence[0]
|
|
.find_prop("SO")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
info.game_keeper = tree.sequence[0]
|
|
.find_prop("US")
|
|
.map(|prop| prop.values.join(", "));
|
|
|
|
Ok(Game {
|
|
board_size,
|
|
info,
|
|
|
|
tree,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
|
|
#[typeshare]
|
|
pub enum Rank {
|
|
Kyu(u8),
|
|
Dan(u8),
|
|
Pro(u8),
|
|
}
|
|
|
|
impl TryFrom<&str> for Rank {
|
|
type Error = String;
|
|
fn try_from(r: &str) -> Result<Rank, Self::Error> {
|
|
let parts = r.split(" ").map(|s| s.to_owned()).collect::<Vec<String>>();
|
|
let cnt = parts[0].parse::<u8>().map_err(|err| format!("{:?}", err))?;
|
|
match parts[1].to_ascii_lowercase().as_str() {
|
|
"kyu" => Ok(Rank::Kyu(cnt)),
|
|
"dan" => Ok(Rank::Dan(cnt)),
|
|
"pro" => Ok(Rank::Pro(cnt)),
|
|
_ => Err("unparsable".to_owned()),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ToString for Rank {
|
|
fn to_string(&self) -> String {
|
|
unimplemented!()
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, Default)]
|
|
pub struct GameInfo {
|
|
pub app_name: Option<String>,
|
|
pub annotator: Option<String>,
|
|
pub copyright: Option<String>,
|
|
pub event: Option<String>,
|
|
// Games can be played across multiple days, even multiple years. The format specifies
|
|
// shortcuts.
|
|
pub date: Vec<Date>,
|
|
pub location: Option<String>,
|
|
// special rules for the round-number and type
|
|
pub round: Option<String>,
|
|
pub ruleset: Option<String>,
|
|
pub source: Option<String>,
|
|
pub time_limits: Option<std::time::Duration>,
|
|
pub game_keeper: Option<String>,
|
|
pub komi: Option<f32>,
|
|
|
|
pub game_name: Option<String>,
|
|
pub game_comments: Option<String>,
|
|
|
|
pub black_player: Option<String>,
|
|
pub black_rank: Option<Rank>,
|
|
pub black_team: Option<String>,
|
|
|
|
pub white_player: Option<String>,
|
|
pub white_rank: Option<Rank>,
|
|
pub white_team: Option<String>,
|
|
|
|
pub opening: Option<String>,
|
|
pub overtime: Option<String>,
|
|
pub result: Option<GameResult>,
|
|
}
|
|
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
pub enum GameResult {
|
|
Annulled,
|
|
Draw,
|
|
Black(Win),
|
|
White(Win),
|
|
}
|
|
|
|
impl TryFrom<&str> for GameResult {
|
|
type Error = String;
|
|
fn try_from(s: &str) -> Result<GameResult, Self::Error> {
|
|
if s == "0" {
|
|
Ok(GameResult::Draw)
|
|
} else if s == "Void" {
|
|
Ok(GameResult::Annulled)
|
|
} else {
|
|
let parts = s.split("+").collect::<Vec<&str>>();
|
|
let res = match parts[0].to_ascii_lowercase().as_str() {
|
|
"b" => GameResult::Black,
|
|
"w" => GameResult::White,
|
|
_ => panic!("unknown result format"),
|
|
};
|
|
match parts[1].to_ascii_lowercase().as_str() {
|
|
"r" | "resign" => Ok(res(Win::Resignation)),
|
|
"t" | "time" => Ok(res(Win::Time)),
|
|
"f" | "forfeit" => Ok(res(Win::Forfeit)),
|
|
_ => {
|
|
let score = parts[1].parse::<f32>().unwrap();
|
|
Ok(res(Win::Score(score)))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug, PartialEq)]
|
|
pub enum Win {
|
|
Score(f32),
|
|
Resignation,
|
|
Forfeit,
|
|
Time,
|
|
}
|
|
|
|
/*
|
|
enum PropType {
|
|
Move,
|
|
Setup,
|
|
Root,
|
|
GameInfo,
|
|
}
|
|
|
|
enum PropValue {
|
|
Empty,
|
|
Number,
|
|
Real,
|
|
Double,
|
|
Color,
|
|
SimpleText,
|
|
Text,
|
|
Point,
|
|
Move,
|
|
Stone,
|
|
}
|
|
*/
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::{
|
|
date::Date,
|
|
tree::{parse_collection, Size},
|
|
};
|
|
use std::fs::File;
|
|
use std::io::Read;
|
|
|
|
const EXAMPLE: &'static str = "(;FF[4]C[root](;C[a];C[b](;C[c])
|
|
(;C[d];C[e]))
|
|
(;C[f](;C[g];C[h];C[i])
|
|
(;C[j])))";
|
|
|
|
fn with_text(text: &str, f: impl FnOnce(Vec<Game>)) {
|
|
let (_, games) = parse_collection::<nom::error::VerboseError<&str>>(text).unwrap();
|
|
let games = games
|
|
.into_iter()
|
|
.map(|game| Game::try_from(game).expect("game to parse"))
|
|
.collect::<Vec<Game>>();
|
|
f(games);
|
|
}
|
|
|
|
fn with_file(path: &std::path::Path, f: impl FnOnce(Vec<Game>)) {
|
|
let mut file = File::open(path).unwrap();
|
|
let mut text = String::new();
|
|
let _ = file.read_to_string(&mut text);
|
|
with_text(&text, f);
|
|
}
|
|
|
|
#[test]
|
|
fn it_parses_game_root() {
|
|
with_text(EXAMPLE, |trees| {
|
|
assert_eq!(trees.len(), 1);
|
|
let tree = &trees[0];
|
|
assert_eq!(tree.info.app_name, None);
|
|
assert_eq!(
|
|
tree.board_size,
|
|
Size {
|
|
width: 19,
|
|
height: 19
|
|
}
|
|
);
|
|
// assert_eq!(tree.text, EXAMPLE.to_owned());
|
|
});
|
|
|
|
with_file(std::path::Path::new("test_data/print1.sgf"), |trees| {
|
|
assert_eq!(trees.len(), 1);
|
|
let tree = &trees[0];
|
|
assert_eq!(tree.info.app_name, None);
|
|
assert_eq!(
|
|
tree.board_size,
|
|
Size {
|
|
width: 19,
|
|
height: 19
|
|
}
|
|
);
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
fn it_parses_game_info() {
|
|
with_file(std::path::Path::new("test_data/print1.sgf"), |trees| {
|
|
assert_eq!(trees.len(), 1);
|
|
let tree = &trees[0];
|
|
assert_eq!(tree.info.black_player, Some("Takemiya Masaki".to_owned()));
|
|
assert_eq!(tree.info.black_rank, Some(Rank::Dan(9)));
|
|
assert_eq!(tree.info.white_player, Some("Cho Chikun".to_owned()));
|
|
assert_eq!(tree.info.white_rank, Some(Rank::Dan(9)));
|
|
assert_eq!(tree.info.result, Some(GameResult::White(Win::Resignation)));
|
|
assert_eq!(
|
|
tree.info.time_limits,
|
|
Some(std::time::Duration::from_secs(28800))
|
|
);
|
|
assert_eq!(
|
|
tree.info.date,
|
|
vec![
|
|
Date::Date(chrono::NaiveDate::from_ymd_opt(1996, 10, 18).unwrap()),
|
|
Date::Date(chrono::NaiveDate::from_ymd_opt(1996, 10, 19).unwrap()),
|
|
]
|
|
);
|
|
assert_eq!(tree.info.event, Some("21st Meijin".to_owned()));
|
|
assert_eq!(tree.info.round, Some("2 (final)".to_owned()));
|
|
assert_eq!(tree.info.source, Some("Go World #78".to_owned()));
|
|
assert_eq!(tree.info.game_keeper, Some("Arno Hollosi".to_owned()));
|
|
});
|
|
}
|
|
}
|