PkmnLib_rs/src/static_data/species_data/species.rs

use self::super::form::Form;
use crate::static_data::species_data::gender::Gender;
use crate::utils::random::Random;
use derive_getters::Getters;
use hashbrown::{HashMap, HashSet};

#[derive(Debug, Getters)]
pub struct Species<'a> {
    id: u16,
    name: String,
    gender_rate: f32,
    growth_rate: String,
    capture_rate: u8,
    forms: HashMap<String, Form<'a>>,
    flags: HashSet<String>,
}

impl<'a> Species<'a> {
    pub fn new(
        id: u16,
        name: &str,
        gender_rate: f32,
        growth_rate: &str,
        capture_rate: u8,
        default_form: Form<'a>,
        flags: HashSet<String>,
    ) -> Species<'a> {
        let mut forms = HashMap::with_capacity(1);
        forms.insert("default".to_string(), default_form);
        Species {
            id,
            name: name.to_string(),
            gender_rate,
            growth_rate: growth_rate.to_string(),
            capture_rate,
            forms,
            flags,
        }
    }

    pub fn add_form(&mut self, id: String, form: Form<'a>) {
        self.forms.insert(id, form);
    }

    pub fn get_form(&self, id: &str) -> Option<&Form> {
        self.forms.get(id)
    }

    pub fn get_random_gender(&self, rand: &mut Random) -> Gender {
        if self.gender_rate < 0.0 {
            Gender::Genderless
        } else if rand.get_float() >= self.gender_rate {
            Gender::Female
        } else {
            Gender::Male
        }
    }

    pub fn has_flag(&self, key: &str) -> bool {
        self.flags.contains(key)
    }
}