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
102
103
104
105
106
107
108
109

use crate::{make_case_elim, make_proj_elim};
use rand::{seq::SliceRandom, thread_rng};
use serde::{Deserialize, Serialize};

make_proj_elim!(
    body,
    #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
    pub struct Body {
        pub species: Species,
        pub body_type: BodyType,
    }
);

impl Body {
    pub fn random() -> Self {
        let mut rng = thread_rng();
        let species = *ALL_SPECIES.choose(&mut rng).unwrap();
        Self::random_with(&mut rng, &species)
    }

    #[inline]
    pub fn random_with(rng: &mut impl rand::Rng, &species: &Species) -> Self {
        let body_type = *ALL_BODY_TYPES.choose(rng).unwrap();
        Self { species, body_type }
    }
}

impl From<Body> for super::Body {
    fn from(body: Body) -> Self { super::Body::BirdLarge(body) }
}

make_case_elim!(
    species,
    #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
    #[repr(u32)]
    pub enum Species {
        Phoenix = 0,
        Cockatrice = 1,
        Roc = 2,
        FlameWyvern = 3,
        CloudWyvern = 4,
        FrostWyvern = 5,
        SeaWyvern = 6,
        WealdWyvern = 7,
    }
);

/// Data representing per-species generic data.
///
/// NOTE: Deliberately don't (yet?) implement serialize.
#[derive(Clone, Debug, Deserialize)]
pub struct AllSpecies<SpeciesMeta> {
    pub phoenix: SpeciesMeta,
    pub cockatrice: SpeciesMeta,
    pub roc: SpeciesMeta,
    pub wyvern_flame: SpeciesMeta,
    pub wyvern_cloud: SpeciesMeta,
    pub wyvern_frost: SpeciesMeta,
    pub wyvern_sea: SpeciesMeta,
    pub wyvern_weald: SpeciesMeta,
}

impl<'a, SpeciesMeta> core::ops::Index<&'a Species> for AllSpecies<SpeciesMeta> {
    type Output = SpeciesMeta;

    #[inline]
    fn index(&self, &index: &'a Species) -> &Self::Output {
        match index {
            Species::Phoenix => &self.phoenix,
            Species::Cockatrice => &self.cockatrice,
            Species::Roc => &self.roc,
            Species::FlameWyvern => &self.wyvern_flame,
            Species::CloudWyvern => &self.wyvern_cloud,
            Species::FrostWyvern => &self.wyvern_frost,
            Species::SeaWyvern => &self.wyvern_sea,
            Species::WealdWyvern => &self.wyvern_weald,
        }
    }
}

pub const ALL_SPECIES: [Species; 8] = [
    Species::Phoenix,
    Species::Cockatrice,
    Species::Roc,
    Species::FlameWyvern,
    Species::CloudWyvern,
    Species::FrostWyvern,
    Species::SeaWyvern,
    Species::WealdWyvern,
];

impl<'a, SpeciesMeta: 'a> IntoIterator for &'a AllSpecies<SpeciesMeta> {
    type IntoIter = std::iter::Copied<std::slice::Iter<'static, Self::Item>>;
    type Item = Species;

    fn into_iter(self) -> Self::IntoIter { ALL_SPECIES.iter().copied() }
}

make_case_elim!(
    body_type,
    #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
    #[repr(u32)]
    pub enum BodyType {
        Female = 0,
        Male = 1,
    }
);

pub const ALL_BODY_TYPES: [BodyType; 2] = [BodyType::Female, BodyType::Male];