initial_code
This commit is contained in:
Spectre
parent
a57debd087
commit
393b3395a5
14 changed files with 2808 additions and 0 deletions
19
.direnv/bin/nix-direnv-reload
Executable file
19
.direnv/bin/nix-direnv-reload
Executable file
|
|
@ -0,0 +1,19 @@
|
|||
#!/usr/bin/env bash
|
||||
set -e
|
||||
if [[ ! -d "/home/spectre/code/path_of_exile" ]]; then
|
||||
echo "Cannot find source directory; Did you move it?"
|
||||
echo "(Looking for "/home/spectre/code/path_of_exile")"
|
||||
echo 'Cannot force reload with this script - use "direnv reload" manually and then try again'
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# rebuild the cache forcefully
|
||||
_nix_direnv_force_reload=1 direnv exec "/home/spectre/code/path_of_exile" true
|
||||
|
||||
# Update the mtime for .envrc.
|
||||
# This will cause direnv to reload again - but without re-building.
|
||||
touch "/home/spectre/code/path_of_exile/.envrc"
|
||||
|
||||
# Also update the timestamp of whatever profile_rc we have.
|
||||
# This makes sure that we know we are up to date.
|
||||
touch -r "/home/spectre/code/path_of_exile/.envrc" "/home/spectre/code/path_of_exile/.direnv"/*.rc
|
||||
1
.direnv/flake-inputs/42c5i4afkf4qcq3ssn06s5awm844dv9q-source
Symbolic link
1
.direnv/flake-inputs/42c5i4afkf4qcq3ssn06s5awm844dv9q-source
Symbolic link
|
|
@ -0,0 +1 @@
|
|||
/nix/store/42c5i4afkf4qcq3ssn06s5awm844dv9q-source
|
||||
1
.direnv/flake-inputs/d17n12f96zm6ycv9wkkmql31av8ar928-source
Symbolic link
1
.direnv/flake-inputs/d17n12f96zm6ycv9wkkmql31av8ar928-source
Symbolic link
|
|
@ -0,0 +1 @@
|
|||
/nix/store/d17n12f96zm6ycv9wkkmql31av8ar928-source
|
||||
1
.direnv/flake-inputs/lryfc8mhk1czqsa421di2y5nzz5c3b8m-source
Symbolic link
1
.direnv/flake-inputs/lryfc8mhk1czqsa421di2y5nzz5c3b8m-source
Symbolic link
|
|
@ -0,0 +1 @@
|
|||
/nix/store/lryfc8mhk1czqsa421di2y5nzz5c3b8m-source
|
||||
1
.direnv/flake-profile-a5d5b61aa8a61b7d9d765e1daf971a9a578f1cfa
Symbolic link
1
.direnv/flake-profile-a5d5b61aa8a61b7d9d765e1daf971a9a578f1cfa
Symbolic link
|
|
@ -0,0 +1 @@
|
|||
/nix/store/mapfkjj4mxiv2ibv3ra8ch4hcjw16wm1-nix-shell-env
|
||||
2049
.direnv/flake-profile-a5d5b61aa8a61b7d9d765e1daf971a9a578f1cfa.rc
Normal file
2049
.direnv/flake-profile-a5d5b61aa8a61b7d9d765e1daf971a9a578f1cfa.rc
Normal file
File diff suppressed because it is too large
Load diff
1
.envrc
Normal file
1
.envrc
Normal file
|
|
@ -0,0 +1 @@
|
|||
use flake
|
||||
1
crop_rot/.gitignore
vendored
Normal file
1
crop_rot/.gitignore
vendored
Normal file
|
|
@ -0,0 +1 @@
|
|||
/target
|
||||
327
crop_rot/Cargo.lock
generated
Normal file
327
crop_rot/Cargo.lock
generated
Normal file
|
|
@ -0,0 +1,327 @@
|
|||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26"
|
||||
|
||||
[[package]]
|
||||
name = "byteorder"
|
||||
version = "1.5.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "crop_rot"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"rand",
|
||||
"rand_distr",
|
||||
"rayon",
|
||||
"serde",
|
||||
"toml",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-deque"
|
||||
version = "0.8.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "613f8cc01fe9cf1a3eb3d7f488fd2fa8388403e97039e2f73692932e291a770d"
|
||||
dependencies = [
|
||||
"crossbeam-epoch",
|
||||
"crossbeam-utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-epoch"
|
||||
version = "0.9.18"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5b82ac4a3c2ca9c3460964f020e1402edd5753411d7737aa39c3714ad1b5420e"
|
||||
dependencies = [
|
||||
"crossbeam-utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-utils"
|
||||
version = "0.8.20"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "22ec99545bb0ed0ea7bb9b8e1e9122ea386ff8a48c0922e43f36d45ab09e0e80"
|
||||
|
||||
[[package]]
|
||||
name = "either"
|
||||
version = "1.13.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "60b1af1c220855b6ceac025d3f6ecdd2b7c4894bfe9cd9bda4fbb4bc7c0d4cf0"
|
||||
|
||||
[[package]]
|
||||
name = "equivalent"
|
||||
version = "1.0.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5443807d6dff69373d433ab9ef5378ad8df50ca6298caf15de6e52e24aaf54d5"
|
||||
|
||||
[[package]]
|
||||
name = "getrandom"
|
||||
version = "0.2.15"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c4567c8db10ae91089c99af84c68c38da3ec2f087c3f82960bcdbf3656b6f4d7"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"libc",
|
||||
"wasi",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "hashbrown"
|
||||
version = "0.15.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1e087f84d4f86bf4b218b927129862374b72199ae7d8657835f1e89000eea4fb"
|
||||
|
||||
[[package]]
|
||||
name = "indexmap"
|
||||
version = "2.6.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "707907fe3c25f5424cce2cb7e1cbcafee6bdbe735ca90ef77c29e84591e5b9da"
|
||||
dependencies = [
|
||||
"equivalent",
|
||||
"hashbrown",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.159"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "561d97a539a36e26a9a5fad1ea11a3039a67714694aaa379433e580854bc3dc5"
|
||||
|
||||
[[package]]
|
||||
name = "libm"
|
||||
version = "0.2.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4ec2a862134d2a7d32d7983ddcdd1c4923530833c9f2ea1a44fc5fa473989058"
|
||||
|
||||
[[package]]
|
||||
name = "memchr"
|
||||
version = "2.7.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
|
||||
|
||||
[[package]]
|
||||
name = "num-traits"
|
||||
version = "0.2.19"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "071dfc062690e90b734c0b2273ce72ad0ffa95f0c74596bc250dcfd960262841"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"libm",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "ppv-lite86"
|
||||
version = "0.2.20"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "77957b295656769bb8ad2b6a6b09d897d94f05c41b069aede1fcdaa675eaea04"
|
||||
dependencies = [
|
||||
"zerocopy",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "proc-macro2"
|
||||
version = "1.0.87"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b3e4daa0dcf6feba26f985457cdf104d4b4256fc5a09547140f3631bb076b19a"
|
||||
dependencies = [
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "quote"
|
||||
version = "1.0.37"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand"
|
||||
version = "0.8.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
|
||||
dependencies = [
|
||||
"libc",
|
||||
"rand_chacha",
|
||||
"rand_core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand_chacha"
|
||||
version = "0.3.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
|
||||
dependencies = [
|
||||
"ppv-lite86",
|
||||
"rand_core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand_core"
|
||||
version = "0.6.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
|
||||
dependencies = [
|
||||
"getrandom",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rand_distr"
|
||||
version = "0.4.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "32cb0b9bc82b0a0876c2dd994a7e7a2683d3e7390ca40e6886785ef0c7e3ee31"
|
||||
dependencies = [
|
||||
"num-traits",
|
||||
"rand",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rayon"
|
||||
version = "1.10.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b418a60154510ca1a002a752ca9714984e21e4241e804d32555251faf8b78ffa"
|
||||
dependencies = [
|
||||
"either",
|
||||
"rayon-core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rayon-core"
|
||||
version = "1.12.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1465873a3dfdaa8ae7cb14b4383657caab0b3e8a0aa9ae8e04b044854c8dfce2"
|
||||
dependencies = [
|
||||
"crossbeam-deque",
|
||||
"crossbeam-utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde"
|
||||
version = "1.0.210"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c8e3592472072e6e22e0a54d5904d9febf8508f65fb8552499a1abc7d1078c3a"
|
||||
dependencies = [
|
||||
"serde_derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde_derive"
|
||||
version = "1.0.210"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "243902eda00fad750862fc144cea25caca5e20d615af0a81bee94ca738f1df1f"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "serde_spanned"
|
||||
version = "0.6.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "87607cb1398ed59d48732e575a4c28a7a8ebf2454b964fe3f224f2afc07909e1"
|
||||
dependencies = [
|
||||
"serde",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "syn"
|
||||
version = "2.0.79"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "89132cd0bf050864e1d38dc3bbc07a0eb8e7530af26344d3d2bbbef83499f590"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"unicode-ident",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "toml"
|
||||
version = "0.8.19"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a1ed1f98e3fdc28d6d910e6737ae6ab1a93bf1985935a1193e68f93eeb68d24e"
|
||||
dependencies = [
|
||||
"serde",
|
||||
"serde_spanned",
|
||||
"toml_datetime",
|
||||
"toml_edit",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "toml_datetime"
|
||||
version = "0.6.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0dd7358ecb8fc2f8d014bf86f6f638ce72ba252a2c3a2572f2a795f1d23efb41"
|
||||
dependencies = [
|
||||
"serde",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "toml_edit"
|
||||
version = "0.22.22"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4ae48d6208a266e853d946088ed816055e556cc6028c5e8e2b84d9fa5dd7c7f5"
|
||||
dependencies = [
|
||||
"indexmap",
|
||||
"serde",
|
||||
"serde_spanned",
|
||||
"toml_datetime",
|
||||
"winnow",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "unicode-ident"
|
||||
version = "1.0.13"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e91b56cd4cadaeb79bbf1a5645f6b4f8dc5bde8834ad5894a8db35fda9efa1fe"
|
||||
|
||||
[[package]]
|
||||
name = "wasi"
|
||||
version = "0.11.0+wasi-snapshot-preview1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
|
||||
|
||||
[[package]]
|
||||
name = "winnow"
|
||||
version = "0.6.20"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "36c1fec1a2bb5866f07c25f68c26e565c4c200aebb96d7e55710c19d3e8ac49b"
|
||||
dependencies = [
|
||||
"memchr",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy"
|
||||
version = "0.7.35"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1b9b4fd18abc82b8136838da5d50bae7bdea537c574d8dc1a34ed098d6c166f0"
|
||||
dependencies = [
|
||||
"byteorder",
|
||||
"zerocopy-derive",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "zerocopy-derive"
|
||||
version = "0.7.35"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "fa4f8080344d4671fb4e831a13ad1e68092748387dfc4f55e356242fae12ce3e"
|
||||
dependencies = [
|
||||
"proc-macro2",
|
||||
"quote",
|
||||
"syn",
|
||||
]
|
||||
11
crop_rot/Cargo.toml
Normal file
11
crop_rot/Cargo.toml
Normal file
|
|
@ -0,0 +1,11 @@
|
|||
[package]
|
||||
name = "crop_rot"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
rand = "0.8.5"
|
||||
rand_distr = "0.4.3"
|
||||
rayon = "1.10.0"
|
||||
serde = { version = "1.0.210", features=["derive"] }
|
||||
toml = "0.8.19"
|
||||
20
crop_rot/config.toml
Normal file
20
crop_rot/config.toml
Normal file
|
|
@ -0,0 +1,20 @@
|
|||
expected_lifeforce = [3.220000000000082, 11.733855400000088, 70.33212248000017, 162.32843617999993, 275.6750786199999, 402.55435975999916, 536.7185175399989, 673.7040835000004, 811.9261996200031, 948.5246767200055]
|
||||
[lifeforce_value]
|
||||
yellow = 2.5
|
||||
blue = 1.0
|
||||
red = 1.0
|
||||
|
||||
[seed_lifeforce]
|
||||
t1 = 0.14
|
||||
t2 = 1.62
|
||||
t3 = 47.0
|
||||
t4 = 230.0
|
||||
|
||||
[map_mods]
|
||||
pack_size = 1.0
|
||||
|
||||
[seed_upgrade_chance]
|
||||
t1 = 0.25
|
||||
t2 = 0.2
|
||||
t3 = 0.03
|
||||
|
||||
75
crop_rot/src/main.rs
Normal file
75
crop_rot/src/main.rs
Normal file
|
|
@ -0,0 +1,75 @@
|
|||
// mod harvest;
|
||||
pub mod model;
|
||||
pub mod weights;
|
||||
|
||||
use model::*;
|
||||
use weights::{ColorWeights, Config};
|
||||
|
||||
use std::fs;
|
||||
|
||||
use rayon::prelude::*;
|
||||
|
||||
fn main() {
|
||||
let config = parse_config("config.toml").unwrap();
|
||||
|
||||
let mut res: Vec<(ColorWeights, Vec<f64>)> = Vec::new();
|
||||
|
||||
for color in gen_possible_color_weights().into_iter() {
|
||||
let val = run_sim(100, &config, &color);
|
||||
println!("{:?} {:?}", color, val);
|
||||
res.push((color, val));
|
||||
}
|
||||
|
||||
println!("{}", res);
|
||||
|
||||
// for i in 0..10 {
|
||||
// let res = lifeforce_after_upgrades(i, 1000000, &config);
|
||||
|
||||
// println!("{i}: {res}");
|
||||
// }
|
||||
|
||||
|
||||
}
|
||||
|
||||
fn parse_config(path: &str) -> Result<Config, Box<dyn std::error::Error>> {
|
||||
let file = fs::read_to_string(path)?;
|
||||
let config: Config = toml::from_str(&file)?;
|
||||
Ok(config)
|
||||
}
|
||||
|
||||
fn run_sim(n: usize, config: &Config, color_weight: &ColorWeights) -> Vec<f64> {
|
||||
(0..n)
|
||||
.into_par_iter()
|
||||
.map(|_| Harvest::<Field>::new(color_weight, config).recursive_harvest(config))
|
||||
.collect()
|
||||
}
|
||||
|
||||
fn gen_possible_color_weights() -> Vec<ColorWeights> {
|
||||
let allowed = [0.55, 0.65, 0.75, 0.80, 0.90, 1.00];
|
||||
|
||||
let mut all: Vec<ColorWeights> = Vec::new();
|
||||
|
||||
for &yellow in &allowed {
|
||||
for &blue in &allowed {
|
||||
for &red in &allowed {
|
||||
if yellow == 0.55 || blue == 0.55 || red == 0.55 {
|
||||
all.push(ColorWeights { yellow, blue, red });
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
all
|
||||
}
|
||||
|
||||
fn lifeforce_after_upgrades(n: usize, tries: usize, config: &Config) -> f64 {
|
||||
(0..tries)
|
||||
.into_par_iter()
|
||||
.map(|_| {
|
||||
let mut field: Field = Field { color: weights::Color::Yellow, wilted: false, t1_seeds: 23, t2_seeds: 0, t3_seeds: 0, t4_seeds: 0};
|
||||
for i in 0..n {
|
||||
field.upgrade(config);
|
||||
}
|
||||
field.harvest(config).unwrap()
|
||||
}).sum::<f64>() / (tries as f64)
|
||||
}
|
||||
194
crop_rot/src/model.rs
Normal file
194
crop_rot/src/model.rs
Normal file
|
|
@ -0,0 +1,194 @@
|
|||
use std::fmt::{self, Debug, Display};
|
||||
use rand::prelude::*;
|
||||
|
||||
use crate::weights::{Config, ColorWeights, Color};
|
||||
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Field {
|
||||
pub color: Color,
|
||||
pub wilted: bool,
|
||||
pub t1_seeds: i8,
|
||||
pub t2_seeds: i8,
|
||||
pub t3_seeds: i8,
|
||||
pub t4_seeds: i8,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct FakeField {
|
||||
color: Color,
|
||||
wilted: bool,
|
||||
upgrades: usize,
|
||||
}
|
||||
|
||||
pub trait Harvestable {
|
||||
fn new(color_weights: &ColorWeights) -> Self;
|
||||
|
||||
fn new_ungenerated() -> Self;
|
||||
|
||||
fn harvest(&mut self, config: &Config) -> Result<f64, &'static str>;
|
||||
|
||||
fn upgrade(&mut self, config: &Config);
|
||||
|
||||
fn wilt(&mut self);
|
||||
|
||||
fn is_wilted(&self) -> bool;
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Harvest<T: Harvestable> {
|
||||
fields: [T; 10],
|
||||
value: f64,
|
||||
}
|
||||
|
||||
impl Harvestable for Field {
|
||||
fn new(color_weights: &ColorWeights) -> Self {
|
||||
Field { color: color_weights.color(), wilted: false, t1_seeds: 23, t2_seeds: 0, t3_seeds: 0, t4_seeds: 0 }
|
||||
}
|
||||
|
||||
fn new_ungenerated() -> Self {
|
||||
Field { color: Color::NotGenerated, wilted: true, t1_seeds: 0, t2_seeds: 0, t3_seeds: 0, t4_seeds: 0 }
|
||||
}
|
||||
|
||||
fn harvest(&mut self, config: &Config) -> Result<f64, &'static str> {
|
||||
match self.wilted {
|
||||
false => {
|
||||
self.wilted = true;
|
||||
Ok(self.t1_seeds as f64 * config.seed_lifeforce.t1 + self.t2_seeds as f64 * config.seed_lifeforce.t2 + self.t3_seeds as f64 * config.seed_lifeforce.t3 + self.t4_seeds as f64 * config.seed_lifeforce.t4)
|
||||
},
|
||||
true => Err("cannot harvest wilted crop")
|
||||
}
|
||||
}
|
||||
|
||||
fn upgrade(&mut self, config: &Config) {
|
||||
if !self.wilted {
|
||||
let t2 = config.seed_upgrade_chance.t1_to_t2(self.t1_seeds);
|
||||
let t3 = config.seed_upgrade_chance.t2_to_t3(self.t2_seeds);
|
||||
let t4 = config.seed_upgrade_chance.t3_to_t4(self.t3_seeds);
|
||||
self.t1_seeds += - t2;
|
||||
self.t2_seeds += t2 - t3;
|
||||
self.t3_seeds += t3 - t4;
|
||||
self.t4_seeds += t4;
|
||||
}
|
||||
}
|
||||
|
||||
fn wilt(&mut self) {
|
||||
self.wilted = true;
|
||||
}
|
||||
|
||||
fn is_wilted(&self) -> bool {
|
||||
self.wilted
|
||||
}
|
||||
}
|
||||
|
||||
impl Harvestable for FakeField {
|
||||
fn new(color_weights: &ColorWeights) -> FakeField {
|
||||
FakeField { color: color_weights.color(), wilted: false, upgrades: 0 }
|
||||
}
|
||||
|
||||
fn new_ungenerated() -> Self {
|
||||
FakeField {color: Color::NotGenerated, wilted: true, upgrades: 0}
|
||||
}
|
||||
|
||||
fn harvest(&mut self, config: &Config) -> Result<f64, &'static str> {
|
||||
match self.wilted {
|
||||
false => {
|
||||
self.wilted = true;
|
||||
Ok(config.expected_lifeforce[self.upgrades] * config.lifeforce_value.from_color(&self.color))
|
||||
}
|
||||
true => Err("cannot harvest wilted crop"),
|
||||
}
|
||||
}
|
||||
|
||||
fn upgrade(&mut self, config: &Config) {
|
||||
if !self.wilted { self.upgrades += 1; }
|
||||
}
|
||||
|
||||
fn wilt(&mut self) {
|
||||
self.wilted = true;
|
||||
}
|
||||
|
||||
fn is_wilted(&self) -> bool {
|
||||
self.wilted
|
||||
}
|
||||
}
|
||||
|
||||
impl<T> Harvest<T> where
|
||||
T: Harvestable + Debug + Clone + Display
|
||||
{
|
||||
pub fn new(color_weights: &ColorWeights, config: &Config) -> Harvest<T> {
|
||||
let rolled_fields: usize = rand::distributions::Uniform::new(3, 6).sample(&mut thread_rng()) * 2;
|
||||
|
||||
let mut fields: Vec<T> = Vec::new();
|
||||
|
||||
for _ in 0..rolled_fields {
|
||||
fields.push(T::new(color_weights));
|
||||
}
|
||||
for _ in rolled_fields..10 {
|
||||
fields.push(T::new_ungenerated())
|
||||
}
|
||||
|
||||
Harvest { value: 0.0, fields: fields.try_into().unwrap() }
|
||||
}
|
||||
|
||||
pub fn harvest(&mut self, n: usize, config: &Config) -> Result<f64, &'static str> {
|
||||
match self.fields[n].harvest(config) {
|
||||
Err(x) => Err(x),
|
||||
Ok(gained_val) => {
|
||||
self.value += gained_val;
|
||||
let neighbor = n + 1 - n % 2;
|
||||
|
||||
if thread_rng().gen_bool(0.4) {self.fields[neighbor].wilt();}
|
||||
|
||||
self.fields.iter_mut().for_each(|field| field.upgrade(config));
|
||||
|
||||
Ok(gained_val)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn recursive_harvest(self, config: &Config) -> f64 {
|
||||
if self.fields.iter().all(|field| field.is_wilted()) {
|
||||
return self.value;
|
||||
} else {
|
||||
return self.fields.iter().enumerate().filter(|(_, field)| !field.is_wilted()).map(|(i, field)| {
|
||||
let mut clone = self.clone();
|
||||
clone.harvest(i, config);
|
||||
clone.recursive_harvest(config)
|
||||
}).reduce(f64::max).unwrap()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for Field {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
match self.wilted {
|
||||
false => write!(f, "{}:{}:{}:{}:{}", match self.color {
|
||||
Color::Yellow => "Y",
|
||||
Color::Blue => "B",
|
||||
Color::Red => "R",
|
||||
Color::NotGenerated => "N/A"
|
||||
}, self.t1_seeds, self.t2_seeds, self.t3_seeds, self.t4_seeds),
|
||||
true => write!(f, "Wilted")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for FakeField {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
write!(f, "Up:{}", self.upgrades)
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: fmt::Display + Harvestable> fmt::Display for Harvest<T> {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
for (i, field) in self.fields.iter().enumerate() {
|
||||
write!(f, "{}", field)?;
|
||||
if i % 2 == 0 {
|
||||
write!(f, " <-> ")?;
|
||||
}
|
||||
else { write!(f, " ")?; }
|
||||
}
|
||||
write!(f, "")
|
||||
}
|
||||
}
|
||||
107
crop_rot/src/weights.rs
Normal file
107
crop_rot/src/weights.rs
Normal file
|
|
@ -0,0 +1,107 @@
|
|||
use rand::prelude::Distribution;
|
||||
use rand_distr::Binomial;
|
||||
use rand::distributions::WeightedIndex;
|
||||
use rand::thread_rng;
|
||||
|
||||
use serde::Deserialize;
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct UpgradeProbabilities {
|
||||
pub t1_t2: f64,
|
||||
pub t2_t3: f64,
|
||||
pub t3_t4: f64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct LifeforceValue {
|
||||
pub yellow: f64,
|
||||
pub blue: f64,
|
||||
pub red: f64,
|
||||
}
|
||||
|
||||
impl LifeforceValue {
|
||||
pub fn from_color(&self, color: &Color) -> f64 {
|
||||
match color {
|
||||
Color::Yellow => self.yellow,
|
||||
Color::Blue => self.blue,
|
||||
Color::Red => self.red,
|
||||
_ => 0.0
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct SeedLifeforce {
|
||||
pub t1: f64,
|
||||
pub t2: f64,
|
||||
pub t3: f64,
|
||||
pub t4: f64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct MapMods {
|
||||
pub pack_size: f64,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct Weights {
|
||||
pub upgrade: UpgradeProbabilities,
|
||||
pub value: LifeforceValue,
|
||||
pub lifeforce: SeedLifeforce,
|
||||
pub map_mods: MapMods,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct SeedUpgradeChances {
|
||||
pub t1: f64,
|
||||
pub t2: f64,
|
||||
pub t3: f64,
|
||||
}
|
||||
|
||||
impl SeedUpgradeChances {
|
||||
pub fn t1_to_t2(&self, n: i8) -> i8 {
|
||||
Binomial::new(n as u64, self.t1).unwrap().sample(&mut thread_rng()) as i8
|
||||
}
|
||||
pub fn t2_to_t3(&self, n: i8) -> i8 {
|
||||
Binomial::new(n as u64, self.t2).unwrap().sample(&mut thread_rng()) as i8
|
||||
}
|
||||
pub fn t3_to_t4(&self, n: i8) -> i8 {
|
||||
Binomial::new(n as u64, self.t3).unwrap().sample(&mut thread_rng()) as i8
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize)]
|
||||
pub struct Config {
|
||||
pub lifeforce_value: LifeforceValue,
|
||||
pub seed_lifeforce: SeedLifeforce,
|
||||
pub map_mods: MapMods,
|
||||
pub seed_upgrade_chance: SeedUpgradeChances,
|
||||
pub expected_lifeforce: [f64; 10],
|
||||
}
|
||||
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub enum Color {
|
||||
Yellow,
|
||||
Blue,
|
||||
Red,
|
||||
NotGenerated
|
||||
}
|
||||
|
||||
#[derive(Debug)]
|
||||
pub struct ColorWeights {
|
||||
pub yellow: f64,
|
||||
pub blue: f64,
|
||||
pub red: f64,
|
||||
}
|
||||
|
||||
impl ColorWeights {
|
||||
pub fn color(&self) -> Color {
|
||||
match WeightedIndex::new([self.yellow, self.blue, self.red]).unwrap().sample(&mut thread_rng()) {
|
||||
0 => Color::Yellow,
|
||||
1 => Color::Blue,
|
||||
2 => Color::Red,
|
||||
_ => unreachable!(),
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue