use crate::{make_case_elim, make_proj_elim}; use rand::{seq::SliceRandom, thread_rng}; use serde::{Deserialize, Serialize}; use strum::{Display, EnumString}; make_proj_elim!( body, #[derive(Copy, Clone, Debug, PartialEq, Eq, 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 for super::Body { fn from(body: Body) -> Self { super::Body::BirdMedium(body) } } make_case_elim!( species, #[derive( Copy, Clone, Debug, Display, EnumString, PartialEq, Eq, Hash, Serialize, Deserialize, )] #[repr(u32)] pub enum Species { SnowyOwl = 0, HornedOwl = 1, Duck = 2, Cockatiel = 3, Chicken = 4, Bat = 5, Penguin = 6, Goose = 7, Peacock = 8, Eagle = 9, Parrot = 10, Crow = 11, Dodo = 12, Parakeet = 13, Puffin = 14, Toucan = 15, } ); /// Data representing per-species generic data. /// /// NOTE: Deliberately don't (yet?) implement serialize. #[derive(Clone, Debug, Deserialize)] pub struct AllSpecies { pub snowy_owl: SpeciesMeta, pub horned_owl: SpeciesMeta, pub duck: SpeciesMeta, pub cockatiel: SpeciesMeta, pub chicken: SpeciesMeta, pub bat: SpeciesMeta, pub penguin: SpeciesMeta, pub goose: SpeciesMeta, pub peacock: SpeciesMeta, pub eagle: SpeciesMeta, pub parrot: SpeciesMeta, pub crow: SpeciesMeta, pub dodo: SpeciesMeta, pub parakeet: SpeciesMeta, pub puffin: SpeciesMeta, pub toucan: SpeciesMeta, } impl<'a, SpeciesMeta> core::ops::Index<&'a Species> for AllSpecies { type Output = SpeciesMeta; #[inline] fn index(&self, &index: &'a Species) -> &Self::Output { match index { Species::SnowyOwl => &self.snowy_owl, Species::HornedOwl => &self.horned_owl, Species::Duck => &self.duck, Species::Cockatiel => &self.cockatiel, Species::Chicken => &self.chicken, Species::Bat => &self.bat, Species::Penguin => &self.penguin, Species::Goose => &self.goose, Species::Peacock => &self.peacock, Species::Eagle => &self.eagle, Species::Parrot => &self.parrot, Species::Crow => &self.crow, Species::Dodo => &self.dodo, Species::Parakeet => &self.parakeet, Species::Puffin => &self.puffin, Species::Toucan => &self.toucan, } } } pub const ALL_SPECIES: [Species; 16] = [ Species::SnowyOwl, Species::HornedOwl, Species::Duck, Species::Cockatiel, Species::Chicken, Species::Bat, Species::Penguin, Species::Goose, Species::Peacock, Species::Eagle, Species::Parrot, Species::Crow, Species::Dodo, Species::Parakeet, Species::Puffin, Species::Toucan, ]; impl<'a, SpeciesMeta: 'a> IntoIterator for &'a AllSpecies { type IntoIter = std::iter::Copied>; type Item = Species; fn into_iter(self) -> Self::IntoIter { ALL_SPECIES.iter().copied() } } make_case_elim!( body_type, #[derive( Copy, Clone, Debug, Display, EnumString, PartialEq, Eq, Hash, Serialize, Deserialize, )] #[repr(u32)] pub enum BodyType { Female = 0, Male = 1, } ); pub const ALL_BODY_TYPES: [BodyType; 2] = [BodyType::Female, BodyType::Male];