riichienv-core 0.3.6

Japanese Mahjong (Riichi) game engine with MJAI protocol support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

/// Compact tile dimension for 3-player mahjong (sanma).
/// 27 valid tile types: 1m, 9m, 1-9p, 1-9s, 4 winds, 3 dragons (no 2m-8m).
pub(crate) const TILE_DIM_3P: usize = 27;

/// Map tile34 index (0-33) to compact 27-tile index.
/// Returns None for tile34 in 1..=7 (2m-8m, excluded in sanma).
#[inline]
pub(crate) fn tile34_to_compact(tile34: usize) -> Option<usize> {
    match tile34 {
        0 => Some(0),
        1..=7 => None,
        8..=33 => Some(tile34 - 7),
        _ => None,
    }
}

/// Helper: write a scalar value broadcast across 27 compact tile positions into a flat buffer.
/// buf layout: channel-major, i.e. buf[(ch_offset + ch) * TILE_DIM_3P + tile] = val
#[inline]
pub(crate) fn broadcast_scalar(buf: &mut [f32], ch_offset: usize, ch: usize, val: f32) {
    let start = (ch_offset + ch) * TILE_DIM_3P;
    for j in 0..TILE_DIM_3P {
        buf[start + j] = val;
    }
}

/// Helper: set a single value in the flat buffer.
#[inline]
pub(crate) fn set_val(buf: &mut [f32], ch_offset: usize, ch: usize, tile: usize, val: f32) {
    buf[(ch_offset + ch) * TILE_DIM_3P + tile] = val;
}

/// Helper: add a value in the flat buffer.
#[inline]
pub(crate) fn add_val(buf: &mut [f32], ch_offset: usize, ch: usize, tile: usize, val: f32) {
    buf[(ch_offset + ch) * TILE_DIM_3P + tile] += val;
}

/// Sanma dora indicator -> dora tile mapping (tile IDs, i.e. tile/4 in 0..34).
/// In sanma: 1m(0)->9m(8), 9m(8)->1m(0), no 2m-8m exist.
pub(crate) fn get_next_tile_sanma(tile: u32) -> u8 {
    let tile34 = tile / 4;
    match tile34 {
        0 => (8 * 4) as u8,  // 1m -> 9m
        8 => 0u8,            // 9m -> 1m (tile34=0, times 4 = 0)
        1..=7 => tile as u8, // shouldn't appear in sanma
        _ => get_next_tile(tile),
    }
}

/// Standard next tile for dora calculation (non-manzu suits and honors).
fn get_next_tile(tile: u32) -> u8 {
    let tile_type = (tile / 4) / 9;
    let tile_num = (tile / 4) % 9;

    if tile_type < 3 {
        let next_num = if tile_num == 8 { 0 } else { tile_num + 1 };
        ((tile_type * 9 + next_num) * 4) as u8
    } else {
        let base = tile / 4;
        if (27..31).contains(&base) {
            let wind_idx = base - 27;
            let next_wind = (wind_idx + 1) % 4;
            ((27 + next_wind) * 4) as u8
        } else if (31..34).contains(&base) {
            let dragon_idx = base - 31;
            let next_dragon = (dragon_idx + 1) % 3;
            ((31 + next_dragon) * 4) as u8
        } else {
            tile as u8
        }
    }
}