FFI for Pokemon Party, make Pokemon Party use interior mutability.
This commit is contained in:
@@ -45,7 +45,7 @@ impl BattleParty {
|
||||
}
|
||||
|
||||
/// Gets a Pokemon at an index.
|
||||
pub fn get_pokemon(&self, index: usize) -> &Option<Arc<Pokemon>> {
|
||||
pub fn get_pokemon(&self, index: usize) -> Option<Arc<Pokemon>> {
|
||||
self.party.at(index)
|
||||
}
|
||||
|
||||
|
||||
@@ -1,13 +1,18 @@
|
||||
use parking_lot::lock_api::RwLockReadGuard;
|
||||
use parking_lot::{RawRwLock, RwLock};
|
||||
use std::sync::Arc;
|
||||
|
||||
use crate::dynamic_data::models::pokemon::Pokemon;
|
||||
use crate::{ValueIdentifiable, ValueIdentifier};
|
||||
|
||||
/// A list of Pokemon belonging to a trainer.
|
||||
#[derive(Debug)]
|
||||
#[cfg_attr(feature = "wasm", derive(unique_type_id_derive::UniqueTypeId))]
|
||||
pub struct PokemonParty {
|
||||
/// A unique identifier so we know what value this is.
|
||||
identifier: ValueIdentifier,
|
||||
/// The underlying list of Pokemon.
|
||||
pokemon: Vec<Option<Arc<Pokemon>>>,
|
||||
pokemon: RwLock<Vec<Option<Arc<Pokemon>>>>,
|
||||
}
|
||||
|
||||
impl PokemonParty {
|
||||
@@ -17,42 +22,50 @@ impl PokemonParty {
|
||||
for _i in 0..size {
|
||||
pokemon.push(None);
|
||||
}
|
||||
Self { pokemon }
|
||||
Self {
|
||||
identifier: Default::default(),
|
||||
pokemon: RwLock::new(pokemon),
|
||||
}
|
||||
}
|
||||
|
||||
/// Instantiates a party with a list.
|
||||
pub fn new_from_vec(pokemon: Vec<Option<Arc<Pokemon>>>) -> Self {
|
||||
Self { pokemon }
|
||||
Self {
|
||||
identifier: Default::default(),
|
||||
pokemon: RwLock::new(pokemon),
|
||||
}
|
||||
}
|
||||
|
||||
/// Gets a Pokemon at an index in the party.
|
||||
pub fn at(&self, index: usize) -> &Option<Arc<Pokemon>> {
|
||||
let opt = self.pokemon.get(index);
|
||||
pub fn at(&self, index: usize) -> Option<Arc<Pokemon>> {
|
||||
let read_lock = self.pokemon.read();
|
||||
let opt = read_lock.get(index);
|
||||
if let Some(v) = opt {
|
||||
v
|
||||
v.clone()
|
||||
} else {
|
||||
&None
|
||||
None
|
||||
}
|
||||
}
|
||||
|
||||
/// Swaps two Pokemon in the party around.
|
||||
pub fn switch(&mut self, a: usize, b: usize) {
|
||||
self.pokemon.swap(a, b);
|
||||
pub fn switch(&self, a: usize, b: usize) {
|
||||
self.pokemon.write().swap(a, b);
|
||||
}
|
||||
|
||||
/// Sets the Pokemon at an index to a Pokemon, returning the old Pokemon.
|
||||
pub fn swap_into(&mut self, index: usize, pokemon: Option<Arc<Pokemon>>) -> Option<Arc<Pokemon>> {
|
||||
if index >= self.pokemon.len() {
|
||||
pub fn swap_into(&self, index: usize, pokemon: Option<Arc<Pokemon>>) -> Option<Arc<Pokemon>> {
|
||||
let mut party = self.pokemon.write();
|
||||
if index >= party.len() {
|
||||
return pokemon;
|
||||
}
|
||||
let old = self.pokemon[index].as_ref().cloned();
|
||||
self.pokemon[index] = pokemon;
|
||||
let old = party[index].as_ref().cloned();
|
||||
party[index] = pokemon;
|
||||
old
|
||||
}
|
||||
|
||||
/// Whether or not the party still has Pokemon that can be used in battle.
|
||||
pub fn has_usable_pokemon(&self) -> bool {
|
||||
for pokemon in self.pokemon.iter().flatten() {
|
||||
for pokemon in self.pokemon.read().iter().flatten() {
|
||||
if pokemon.is_usable() {
|
||||
return true;
|
||||
}
|
||||
@@ -62,30 +75,31 @@ impl PokemonParty {
|
||||
|
||||
/// Get the length of the underlying list of Pokemon.
|
||||
pub fn length(&self) -> usize {
|
||||
self.pokemon.len()
|
||||
self.pokemon.read().len()
|
||||
}
|
||||
|
||||
/// Gets the underlying list of Pokemon.
|
||||
pub fn pokemon(&self) -> &Vec<Option<Arc<Pokemon>>> {
|
||||
&self.pokemon
|
||||
pub fn pokemon(&self) -> RwLockReadGuard<'_, RawRwLock, Vec<Option<Arc<Pokemon>>>> {
|
||||
self.pokemon.read()
|
||||
}
|
||||
|
||||
/// Makes sure there are no empty spots in the party anymore, leaving the length the same.
|
||||
pub fn pack_party(&mut self) {
|
||||
pub fn pack_party(&self) {
|
||||
let mut first_empty = None;
|
||||
let mut i = 0;
|
||||
let mut party = self.pokemon.write();
|
||||
loop {
|
||||
if self.pokemon[i].is_none() {
|
||||
if party[i].is_none() {
|
||||
if first_empty.is_none() {
|
||||
first_empty = Some(i)
|
||||
}
|
||||
} else if first_empty.is_some() {
|
||||
self.pokemon.swap(first_empty.unwrap(), i);
|
||||
party.swap(first_empty.unwrap(), i);
|
||||
i = first_empty.unwrap();
|
||||
first_empty = None;
|
||||
}
|
||||
i += 1;
|
||||
if i >= self.pokemon.len() {
|
||||
if i >= party.len() {
|
||||
break;
|
||||
}
|
||||
}
|
||||
@@ -93,7 +107,7 @@ impl PokemonParty {
|
||||
|
||||
/// Checks if the party contains a given pokemon.
|
||||
pub fn has_pokemon(&self, pokemon: &Pokemon) -> bool {
|
||||
for p in self.pokemon.iter().flatten() {
|
||||
for p in self.pokemon.read().iter().flatten() {
|
||||
if std::ptr::eq(p.as_ref(), pokemon) {
|
||||
return true;
|
||||
}
|
||||
@@ -101,3 +115,9 @@ impl PokemonParty {
|
||||
false
|
||||
}
|
||||
}
|
||||
|
||||
impl ValueIdentifiable for PokemonParty {
|
||||
fn value_identifier(&self) -> ValueIdentifier {
|
||||
self.identifier
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user