x-iztro 0.2.0

Zi Wei Dou Shu (Purple Star Astrology) chart engine, field-for-field identical to iztro
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

//! 主星安放
//!
//! 根据紫微和天府的起始宫位索引，将14颗主星安放到12宫中。

use crate::data::stars::StarKey;
use crate::data::types::*;
use crate::models::star::Star;
use crate::utils::fix_index;

/// 获取14颗主星的安放结果
///
/// 紫微系6星（逆时针安放）+ 天府系8星（顺时针安放）
pub fn get_major_stars(
    ziwei_index: usize,
    tianfu_index: usize,
    yearly_stem: HeavenlyStem,
    lang: Language,
    config: &Config,
) -> [Vec<Star>; 12] {
    let mut result: [Vec<Star>; 12] = Default::default();

    // 紫微系（逆时针，从 ziwei_index 减 i）
    // i=0: 紫微, i=1: 天机, i=2: SKIP, i=3: 太阳, i=4: 武曲, i=5: 天同,
    // i=6: SKIP, i=7: SKIP, i=8: 廉贞
    let ziwei_group: [(usize, StarKey); 6] = [
        (0, StarKey::ZiweiMaj),
        (1, StarKey::TianjiMaj),
        (3, StarKey::TaiyangMaj),
        (4, StarKey::WuquMaj),
        (5, StarKey::TiantongMaj),
        (8, StarKey::LianzhenMaj),
    ];

    for (offset, key) in ziwei_group {
        let idx = fix_index(ziwei_index as i32 - offset as i32, 12);
        let star = crate::star::make_star(
            key,
            StarType::Major,
            Scope::Origin,
            idx,
            Some(yearly_stem),
            lang,
            config,
        );
        result[idx].push(star);
    }

    // 天府系（顺时针，从 tianfu_index 加 i）
    // i=0: 天府, i=1: 太阴, i=2: 贪狼, i=3: 巨门, i=4: 天相, i=5: 天梁,
    // i=6: 七杀, i=7: SKIP, i=8: SKIP, i=9: SKIP, i=10: 破军
    let tianfu_group: [(usize, StarKey); 8] = [
        (0, StarKey::TianfuMaj),
        (1, StarKey::TaiyinMaj),
        (2, StarKey::TanlangMaj),
        (3, StarKey::JumenMaj),
        (4, StarKey::TianxiangMaj),
        (5, StarKey::TianliangMaj),
        (6, StarKey::QishaMaj),
        (10, StarKey::PojunMaj),
    ];

    for (offset, key) in tianfu_group {
        let idx = fix_index(tianfu_index as i32 + offset as i32, 12);
        let star = crate::star::make_star(
            key,
            StarType::Major,
            Scope::Origin,
            idx,
            Some(yearly_stem),
            lang,
            config,
        );
        result[idx].push(star);
    }

    result
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_major_stars_count() {
        let result = get_major_stars(0, 0, HeavenlyStem::Jia, Language::ZhCN, &Config::default());
        let total: usize = result.iter().map(|v| v.len()).sum();
        // 14 major stars total
        assert_eq!(total, 14);
    }

    #[test]
    fn test_all_major_type() {
        let result = get_major_stars(3, 9, HeavenlyStem::Yi, Language::ZhCN, &Config::default());
        for palace in &result {
            for star in palace {
                assert_eq!(star.star_type, StarType::Major);
                assert_eq!(star.scope, Scope::Origin);
            }
        }
    }
}