Model: rename grids field

This commit is contained in:
Simon Gardling 2025-03-24 17:03:26 -04:00
parent 5d574c9674
commit 56f3eae156
Signed by: titaniumtown
GPG Key ID: 9AB28AC10ECE533D

View File

@ -13,8 +13,8 @@ use std::{path::Path, time::Instant};
// Top-level simulation class.
pub struct Model {
// The grid they move on.
grids: Vec<Grid>,
// per-population grid (one for each population)
population_grids: Vec<Grid>,
// Attraction table governs interaction across populations
attraction_table: Vec<Vec<f32>>,
@ -39,7 +39,7 @@ impl Model {
const REPULSION_FACTOR_STD: f32 = 0.1;
pub fn print_configurations(&self) {
for (i, grid) in self.grids.iter().enumerate() {
for (i, grid) in self.population_grids.iter().enumerate() {
println!("Grid {}: {}", i, grid.config);
}
println!("Attraction table: {:#?}", self.attraction_table);
@ -84,7 +84,7 @@ impl Model {
}
Model {
grids,
population_grids: grids,
attraction_table,
diffusivity,
iteration: 0,
@ -108,14 +108,18 @@ impl Model {
let mut time_per_agent_list: Vec<f64> = Vec::new();
let mut time_per_step_list: Vec<f64> = Vec::new();
let agents_num: usize = self.grids.iter().map(|grid| grid.agents.len()).sum();
let agents_num: usize = self
.population_grids
.iter()
.map(|grid| grid.agents.len())
.sum();
(0..steps).for_each(|_| {
// Combine grids
combine(&mut self.grids, &self.attraction_table);
combine(&mut self.population_grids, &self.attraction_table);
let agents_tick_time = Instant::now();
// Tick agents
self.grids.par_iter_mut().for_each(|grid| {
self.population_grids.par_iter_mut().for_each(|grid| {
grid.tick();
grid.diffuse(self.diffusivity); // Diffuse + Decay
});
@ -144,7 +148,7 @@ impl Model {
}
fn save_image_data(&mut self) {
let grids = ThinGridData::new_from_grid_vec(self.grids.clone());
let grids = ThinGridData::new_from_grid_vec(self.population_grids.clone());
let img_data = ImgData::new(grids, self.palette);
self.img_data_vec.push((self.iteration + 1, img_data));
let size: usize = std::mem::size_of_val(&self.img_data_vec);