2021-01-20 11:20:06 +00:00
|
|
|
// Example for calculating a drop rate:
|
|
|
|
|
//
|
|
|
|
|
// On every roll an f32 between 0 and 1 is created.
|
|
|
|
|
// For every loot table a total range is created by the sum of the individual
|
|
|
|
|
// ranges per item.
|
|
|
|
|
//
|
|
|
|
|
// This range is the sum of all single ranges defined per item in a table.
|
|
|
|
|
// // Individual Range
|
|
|
|
|
// (3, "common.items.food.cheese"), // 0.0..3.0
|
|
|
|
|
// (3, "common.items.food.apple"), // 3.0..6.0
|
|
|
|
|
// (3, "common.items.food.mushroom"), // 6.0..9.0
|
|
|
|
|
// (1, "common.items.food.coconut"), // 9.0..10.0
|
|
|
|
|
// (0.05, "common.items.food.apple_mushroom_curry"), // 10.0..10.05
|
|
|
|
|
// (0.10, "common.items.food.apple_stick"), // 10.05..10.15
|
|
|
|
|
// (0.10, "common.items.food.mushroom_stick"), // 10.15..10.25
|
|
|
|
|
//
|
|
|
|
|
// The f32 is multiplied by the max. value needed to drop an item in this
|
|
|
|
|
// particular table. X = max. value needed = 10.15
|
|
|
|
|
//
|
|
|
|
|
// Example roll
|
|
|
|
|
// [Random Value 0..1] * X = Number inside the table's total range
|
|
|
|
|
// 0.45777 * X = 4.65
|
|
|
|
|
// 4.65 is in the range of 3.0..6.0 => Apple drops
|
|
|
|
|
//
|
|
|
|
|
// Example drop chance calculation
|
|
|
|
|
// Cheese drop rate = 3/X = 29.6%
|
|
|
|
|
// Coconut drop rate = 1/X = 9.85%
|
|
|
|
|
|
2021-03-29 04:12:11 +00:00
|
|
|
use crate::{
|
|
|
|
|
assets::{self, AssetExt},
|
|
|
|
|
comp::{Body, Item},
|
|
|
|
|
};
|
2020-07-03 13:49:17 +00:00
|
|
|
use rand::prelude::*;
|
2021-03-28 21:41:14 +00:00
|
|
|
use serde::{de::DeserializeOwned, Deserialize, Serialize};
|
2020-07-03 13:49:17 +00:00
|
|
|
|
2020-08-28 01:02:17 +00:00
|
|
|
#[derive(Clone, Debug, PartialEq, Deserialize)]
|
2020-07-03 13:49:17 +00:00
|
|
|
pub struct Lottery<T> {
|
2020-07-03 20:23:24 +00:00
|
|
|
items: Vec<(f32, T)>,
|
2020-07-03 13:49:17 +00:00
|
|
|
total: f32,
|
|
|
|
|
}
|
|
|
|
|
|
2020-12-12 22:14:24 +00:00
|
|
|
impl<T: DeserializeOwned + Send + Sync + 'static> assets::Asset for Lottery<T> {
|
|
|
|
|
type Loader = assets::LoadFrom<Vec<(f32, T)>, assets::RonLoader>;
|
2020-12-13 01:09:57 +00:00
|
|
|
|
|
|
|
|
const EXTENSION: &'static str = "ron";
|
2020-07-03 13:49:17 +00:00
|
|
|
}
|
|
|
|
|
|
2020-12-12 22:14:24 +00:00
|
|
|
impl<T> From<Vec<(f32, T)>> for Lottery<T> {
|
|
|
|
|
fn from(mut items: Vec<(f32, T)>) -> Lottery<T> {
|
2020-07-03 13:49:17 +00:00
|
|
|
let mut total = 0.0;
|
2020-12-12 22:14:24 +00:00
|
|
|
|
|
|
|
|
for (rate, _) in &mut items {
|
|
|
|
|
total += *rate;
|
|
|
|
|
*rate = total - *rate;
|
|
|
|
|
}
|
|
|
|
|
|
2020-07-03 13:49:17 +00:00
|
|
|
Self { items, total }
|
|
|
|
|
}
|
2020-12-12 22:14:24 +00:00
|
|
|
}
|
2020-07-03 13:49:17 +00:00
|
|
|
|
2020-12-12 22:14:24 +00:00
|
|
|
impl<T> Lottery<T> {
|
2020-07-25 11:19:13 +00:00
|
|
|
pub fn choose_seeded(&self, seed: u32) -> &T {
|
|
|
|
|
let x = ((seed % 65536) as f32 / 65536.0) * self.total;
|
2020-07-03 20:23:24 +00:00
|
|
|
&self.items[self
|
|
|
|
|
.items
|
|
|
|
|
.binary_search_by(|(y, _)| y.partial_cmp(&x).unwrap())
|
|
|
|
|
.unwrap_or_else(|i| i.saturating_sub(1))]
|
|
|
|
|
.1
|
2020-07-03 13:49:17 +00:00
|
|
|
}
|
2020-07-05 16:49:15 +00:00
|
|
|
|
2020-08-12 14:10:12 +00:00
|
|
|
pub fn choose(&self) -> &T { self.choose_seeded(thread_rng().gen()) }
|
2020-07-25 11:19:13 +00:00
|
|
|
|
2020-07-05 16:49:15 +00:00
|
|
|
pub fn iter(&self) -> impl Iterator<Item = &(f32, T)> { self.items.iter() }
|
2021-03-22 09:39:35 +00:00
|
|
|
|
|
|
|
|
pub fn total(&self) -> f32 { self.total }
|
2020-07-05 16:49:15 +00:00
|
|
|
}
|
|
|
|
|
|
2021-03-28 21:41:14 +00:00
|
|
|
#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
|
|
|
|
|
pub enum LootSpec {
|
2021-03-28 23:17:23 +00:00
|
|
|
/// Asset specifier
|
2021-03-28 21:41:14 +00:00
|
|
|
Item(String),
|
2021-03-28 23:17:23 +00:00
|
|
|
/// Asset specifier, lower range, upper range
|
|
|
|
|
ItemQuantity(String, u32, u32),
|
|
|
|
|
/// Loot table
|
2021-03-28 21:41:14 +00:00
|
|
|
LootTable(String),
|
2021-03-28 23:17:23 +00:00
|
|
|
/// Matches on species to provide a crafting material
|
|
|
|
|
CreatureMaterial,
|
2021-03-28 21:41:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl LootSpec {
|
2021-03-29 04:12:11 +00:00
|
|
|
#[allow(unused_must_use)]
|
2021-03-28 23:17:23 +00:00
|
|
|
pub fn to_item(&self, body: Option<Body>) -> Item {
|
2021-03-28 21:41:14 +00:00
|
|
|
match self {
|
|
|
|
|
Self::Item(item) => Item::new_from_asset_expect(&item),
|
2021-03-28 23:17:23 +00:00
|
|
|
Self::ItemQuantity(item, lower, upper) => {
|
|
|
|
|
let range = *lower..=*upper;
|
|
|
|
|
let quantity = thread_rng().gen_range(range);
|
|
|
|
|
let mut item = Item::new_from_asset_expect(&item);
|
|
|
|
|
item.set_amount(quantity);
|
|
|
|
|
item
|
|
|
|
|
},
|
2021-03-29 04:12:11 +00:00
|
|
|
Self::LootTable(table) => Lottery::<LootSpec>::load_expect(&table)
|
|
|
|
|
.read()
|
|
|
|
|
.choose()
|
|
|
|
|
.to_item(body),
|
|
|
|
|
Self::CreatureMaterial => body.map_or(
|
|
|
|
|
Item::new_from_asset_expect("common.items.food.cheese"),
|
2021-03-30 15:10:17 +00:00
|
|
|
|b| b.get_material(),
|
2021-03-29 04:12:11 +00:00
|
|
|
),
|
2021-03-28 21:41:14 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-07-05 16:49:15 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
|
mod tests {
|
2020-08-12 17:29:51 +00:00
|
|
|
use super::*;
|
2020-12-12 22:14:24 +00:00
|
|
|
use crate::{assets::AssetExt, comp::Item};
|
2020-09-17 23:02:14 +00:00
|
|
|
|
2020-07-05 16:49:15 +00:00
|
|
|
#[test]
|
|
|
|
|
fn test_loot_table() {
|
2021-03-28 21:41:14 +00:00
|
|
|
let test = Lottery::<LootSpec>::load_expect("common.loot_tables.fallback");
|
2020-07-05 16:49:15 +00:00
|
|
|
|
2021-03-28 21:41:14 +00:00
|
|
|
for (_, to_itemifier) in test.read().iter() {
|
2020-07-05 16:49:15 +00:00
|
|
|
assert!(
|
2021-03-28 21:41:14 +00:00
|
|
|
Item::new_from_asset(to_itemifier).is_ok(),
|
2020-07-05 16:49:15 +00:00
|
|
|
"Invalid loot table item '{}'",
|
2021-03-28 21:41:14 +00:00
|
|
|
to_itemifier
|
2020-07-05 16:49:15 +00:00
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
2020-07-03 20:23:24 +00:00
|
|
|
}
|
