PkmnLib_rs/src/static_data/species_data/form.rs

159 lines
5.1 KiB
Rust
Raw Normal View History

2022-07-01 15:07:22 +00:00
use hashbrown::HashSet;
2022-06-11 18:51:37 +00:00
use crate::static_data::LearnableMoves;
use crate::static_data::Statistic;
use crate::static_data::{Ability, StaticStatisticSet};
2022-07-01 15:07:22 +00:00
use crate::static_data::{AbilityIndex, TypeIdentifier};
2022-06-11 18:51:37 +00:00
use crate::Random;
use crate::StringKey;
2022-07-01 15:07:22 +00:00
/// A form is a variant of a specific species. A species always has at least one form, but can have
/// many more.
#[derive(Debug)]
pub struct Form {
2022-07-01 15:07:22 +00:00
/// The name of the form.
name: StringKey,
2022-07-01 15:07:22 +00:00
/// The height of the form in meters.
height: f32,
2022-07-01 15:07:22 +00:00
/// The weight of the form in kilograms.
weight: f32,
2022-07-01 15:07:22 +00:00
/// The base amount of experience that is gained when beating a Pokemon with this form.
base_experience: u32,
2022-07-01 15:07:22 +00:00
/// The normal types a Pokemon with this form has.
types: Vec<TypeIdentifier>,
/// The inherent values of a form of species that are used for the stats of a Pokemon.
base_stats: StaticStatisticSet<u16>,
2022-07-01 15:07:22 +00:00
/// The possible abilities a Pokemon with this form can have.
abilities: Vec<StringKey>,
2022-07-01 15:07:22 +00:00
/// The possible hidden abilities a Pokemon with this form can have.
hidden_abilities: Vec<StringKey>,
2022-07-01 15:07:22 +00:00
/// The moves a Pokemon with this form can learn.
moves: LearnableMoves,
2022-07-01 15:07:22 +00:00
/// Arbitrary flags can be set on a form for scripting use.
flags: HashSet<StringKey>,
}
impl Form {
2022-07-01 15:07:22 +00:00
/// Instantiates a new form.
pub fn new(
name: &StringKey,
height: f32,
weight: f32,
base_experience: u32,
2022-07-01 15:07:22 +00:00
types: Vec<TypeIdentifier>,
base_stats: StaticStatisticSet<u16>,
abilities: Vec<StringKey>,
hidden_abilities: Vec<StringKey>,
moves: LearnableMoves,
flags: HashSet<StringKey>,
) -> Form {
Form {
name: name.clone(),
height,
weight,
base_experience,
types,
base_stats,
abilities,
hidden_abilities,
moves,
flags,
}
}
2022-07-01 15:07:22 +00:00
/// The name of the form.
pub fn name(&self) -> &StringKey {
&self.name
}
2022-07-01 15:07:22 +00:00
/// The height of the form in meters.
pub fn height(&self) -> f32 {
self.height
}
2022-07-01 15:07:22 +00:00
/// The weight of the form in kilograms.
pub fn weight(&self) -> f32 {
self.weight
}
2022-07-01 15:07:22 +00:00
/// The base amount of experience that is gained when beating a Pokemon with this form.
pub fn base_experience(&self) -> u32 {
self.base_experience
}
2022-07-01 15:07:22 +00:00
/// The normal types a Pokemon with this form has.
pub fn types(&self) -> &Vec<TypeIdentifier> {
&self.types
}
2022-07-01 15:07:22 +00:00
/// The inherent values of a form of species that are used for the stats of a Pokemon.
pub fn base_stats(&self) -> &StaticStatisticSet<u16> {
&self.base_stats
}
2022-07-01 15:07:22 +00:00
/// The possible abilities a Pokemon with this form can have.
pub fn abilities(&self) -> &Vec<StringKey> {
&self.abilities
}
2022-07-01 15:07:22 +00:00
/// The possible hidden abilities a Pokemon with this form can have.
pub fn hidden_abilities(&self) -> &Vec<StringKey> {
&self.hidden_abilities
}
2022-07-01 15:07:22 +00:00
/// The moves a Pokemon with this form can learn.
pub fn moves(&self) -> &LearnableMoves {
&self.moves
}
2022-07-01 15:07:22 +00:00
/// Arbitrary flags can be set on a form for scripting use.
pub fn flags(&self) -> &HashSet<StringKey> {
&self.flags
}
2022-07-01 15:07:22 +00:00
/// Get a type of the move at a certain index.
pub fn get_type(&self, index: usize) -> TypeIdentifier {
self.types[index]
}
2022-07-01 15:07:22 +00:00
/// Gets a single base stat value.
pub fn get_base_stat(&self, stat: Statistic) -> u16 {
self.base_stats.get_stat(stat)
}
2022-07-01 15:07:22 +00:00
/// Find the index of an ability that can be on this form.
pub fn find_ability_index(&self, ability: &Ability) -> Option<AbilityIndex> {
for (index, a) in self.abilities.iter().enumerate() {
if a == ability.name() {
return Some(AbilityIndex {
hidden: false,
index: index as u8,
});
}
}
for (index, a) in self.hidden_abilities.iter().enumerate() {
if a == ability.name() {
return Some(AbilityIndex {
hidden: true,
index: index as u8,
});
}
}
None
}
2022-07-01 15:07:22 +00:00
/// Gets an ability from the form.
pub fn get_ability(&self, index: AbilityIndex) -> &StringKey {
if index.hidden {
&self.hidden_abilities[index.index as usize]
} else {
&self.abilities[index.index as usize]
}
}
2022-07-01 15:07:22 +00:00
/// Gets a random ability from the form.
pub fn get_random_ability(&self, rand: &mut Random) -> &StringKey {
&self.abilities[rand.get_between_unsigned(0, self.abilities.len() as u32) as usize]
}
2022-07-01 15:07:22 +00:00
/// Gets a random hidden ability from the form.
pub fn get_random_hidden_ability(&self, rand: &mut Random) -> &StringKey {
&self.hidden_abilities[rand.get_between_unsigned(0, self.hidden_abilities.len() as u32) as usize]
}
2022-07-01 15:07:22 +00:00
/// Check if the form has a specific flag set.
pub fn has_flag(&self, key: &StringKey) -> bool {
self.flags.contains(key)
}
}