use crate::bingpai::Bingpai;
pub(in super::super) fn calculate_replacement_number(bingpai: &Bingpai) -> u8 {
if bingpai.num_required_bingpai_mianzi() < 4 {
return u8::MAX;
}
let (num_kinds, num_duizi) = bingpai
.tile_counts()
.iter()
.filter(|&&count| count > 0)
.fold((0, 0), |(num_kinds, num_duizi), &count| {
(num_kinds + 1, num_duizi + (count >= 2) as u8)
});
7 - num_duizi + 7u8.saturating_sub(num_kinds)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_utils::FromTileCode;
use crate::tile::TileCounts;
#[test]
fn calculate_replacement_number_without_pair() {
let tile_counts = TileCounts::from_code("19m19p19s1234567z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 7);
}
#[test]
fn calculate_replacement_number_with_quadruple() {
let tile_counts = TileCounts::from_code("1188m288p55s1111z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 3);
}
#[test]
fn calculate_replacement_number_with_triplet() {
let tile_counts = TileCounts::from_code("1188m2388p55s111z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 2);
}
#[test]
fn calculate_replacement_number_with_2_triplets() {
let tile_counts = TileCounts::from_code("1188m288p555s111z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 3);
}
#[test]
fn calculate_replacement_number_tenpai() {
let tile_counts = TileCounts::from_code("1188m288p55s1177z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 1);
}
#[test]
fn calculate_replacement_number_win() {
let tile_counts = TileCounts::from_code("1188m2288p55s1177z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, 0);
}
#[test]
fn calculate_replacement_number_incomplete_hand() {
let tile_counts = TileCounts::from_code("1188m55s1122z");
let bingpai = Bingpai::new(&tile_counts).unwrap();
let replacement_number = calculate_replacement_number(&bingpai);
assert_eq!(replacement_number, u8::MAX);
}
}