krABMaga · ddevin96 · Apr 16, 2026 · Apr 16, 2026 · Apr 16, 2026 · Apr 16, 2026
diff --git a/.github/workflows/install-deps/action.yml b/.github/workflows/install-deps/action.yml
@@ -5,7 +5,7 @@ runs:
   using: 'composite'
   steps:
     - name: Cache APT packages
-      uses: actions/cache@v4
+      uses: actions/cache@v5
       id: cache-apt
       with:
         path: /var/cache/apt/archives/*.deb
@@ -20,6 +20,7 @@ runs:
         sudo apt update
         sudo apt install -y --download-only \
           pkg-config libx11-dev libasound2-dev libudev-dev \
+          libwayland-dev libxkbcommon-dev \
           libxcb-randr0-dev libxcb-xtest0-dev libxcb-xinerama0-dev libxcb-shape0-dev libxcb-xkb-dev \
           libxcb-render0-dev libxcb-xfixes0-dev libxcb1-dev \
           portaudio19-dev build-essential libpulse-dev libdbus-1-dev \
@@ -31,6 +32,7 @@ runs:
         sudo apt update
         sudo apt install -y \
           pkg-config libx11-dev libasound2-dev libudev-dev \
+          libwayland-dev libxkbcommon-dev \
           libxcb-randr0-dev libxcb-xtest0-dev libxcb-xinerama0-dev libxcb-shape0-dev libxcb-xkb-dev \
           libxcb-render0-dev libxcb-xfixes0-dev libxcb1-dev \
           portaudio19-dev build-essential libpulse-dev libdbus-1-dev \

diff --git a/README.md b/README.md
@@ -9,10 +9,10 @@ This repository hosts working examples of agent-based simulations, based on the
 | [Ants Foraging](antsforaging)               | :heavy_check_mark: | <img src="antsforaging/ant1.gif" alt="drawing" width="200" height="200"/>      | 
 | [Flockers](flockers)                        | :heavy_check_mark: | <img src="flockers/flock.gif" alt="drawing" width="200" height="200"/>         |
 | [ForestFire](forestfire)                    | :heavy_check_mark: | <img src="forestfire/ff.gif" alt="drawing" width="200" height="200"/>   |
-| [ForestFire_bayesian](forestfire_bayesian)  | :heavy_check_mark: | 
+| [ForestFire_bayesian](forestfire_bayesian)  | :heavy_check_mark: | :x: |
 | [Schelling](schelling)                      | :heavy_check_mark: | <img src="schelling/schelling.gif" alt="drawing" width="200" height="200"/>    |
-| [Sir_bayesian](sir_bayesian)                  | :heavy_check_mark: |
-| [Sir_ga_exploration](sir_ga_exploration)               | :heavy_check_mark: |
+| [Sir_bayesian](sir_bayesian)                  | :heavy_check_mark: | :x: |
+| [Sir_ga_exploration](sir_ga_exploration)               | :heavy_check_mark: | :x: |
 | [Sugarscape](sugarscape)                        | :heavy_check_mark: | <img src="sugarscape/sugarscape.gif" alt="drawing" width="200" height="200"/>     |
 | [Template](template)                        | :heavy_check_mark: | <img src="template/template.gif" alt="drawing" width="200" height="200"/>     |
 | [Virus on a Network](virusnetwork)          | :heavy_check_mark: | <img src="virusnetwork/virus.gif" alt="drawing" width="200" height="200"/>     |

diff --git a/antsforaging/Cargo.toml b/antsforaging/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "antsforaging"
-version = "0.3.0"
+version = "0.6.0"
 authors = [
     "Carmine Spagnuolo <spagnuolocarmine@gmail.com>", 
     "Alessia Antelmi <aantelmi@unisa.it>", 
@@ -18,7 +18,7 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-krabmaga = "0.5.*"
+krabmaga = "0.6.*"
 
 [features]
 visualization = ["krabmaga/visualization"]

diff --git a/antsforaging/src/main.rs b/antsforaging/src/main.rs
@@ -4,15 +4,13 @@ use krabmaga::simulate;
 // Visualization specific imports
 #[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
 use {
-    crate::model::to_food_grid::ToFoodGrid, crate::model::to_home_grid::ToHomeGrid,
-    crate::visualization::vis_state::VisState, krabmaga::bevy::app::FixedUpdate,
-    krabmaga::bevy::prelude::Color, krabmaga::visualization::fields::number_grid_2d::BatchRender,
+    crate::model::state::ModelState, crate::model::to_food_grid::ToFoodGrid,
+    crate::model::to_home_grid::ToHomeGrid, crate::visualization::vis_state::VisState,
+    krabmaga::bevy::app::FixedUpdate, krabmaga::bevy::prelude::Color,
+    krabmaga::visualization::fields::number_grid_2d::BatchRender,
     krabmaga::visualization::visualization::Visualization,
 };
 
-// Global imports, required in all cases
-use crate::model::state::ModelState;
-
 pub mod model;
 
 // Constants
@@ -48,7 +46,7 @@ pub mod visualization;
 fn main() {
     let state = ModelState::new();
     let mut app = Visualization::default()
-        .with_background_color(Color::rgb(255., 255., 255.))
+        .with_background_color(Color::srgb(1., 1., 1.))
         .with_simulation_dimensions(WIDTH as f32, HEIGHT as f32)
         .with_window_dimensions(1280., 720.)
         .with_name("Ants foraging")
@@ -57,16 +55,10 @@ fn main() {
         FixedUpdate,
         (ToHomeGrid::batch_render, ToFoodGrid::batch_render),
     );
-    app.run()
+    app.run();
 }
 
-// #[cfg(not(any(feature = "visualization", feature = "visualization_wasm")))]
-// use {krabmaga::rand, krabmaga::rand::Rng};
-
-// Main used when only the simulation should run, without any visualization.
 #[cfg(not(any(feature = "visualization", feature = "visualization_wasm")))]
 fn main() {
-    let state = ModelState::new();
-
-    let _ = simulate!(state, STEP, 10);
+    println!("Visualization features are not enabled. Please enable one of the visualization features to see the simulation in action.");
 }
diff --git a/antsforaging/src/model/state.rs b/antsforaging/src/model/state.rs
@@ -7,6 +7,12 @@ use crate::{
 };
 use core::fmt;
 use core::hash::{Hash, Hasher};
+
+#[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
+use krabmaga::bevy::ecs as bevy_ecs;
+#[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
+use krabmaga::bevy::prelude::Component;
+
 use krabmaga::engine::fields::field::Field;
 use krabmaga::engine::fields::sparse_object_grid_2d::SparseGrid2D;
 use krabmaga::engine::location::Int2D;
@@ -35,6 +41,10 @@ impl fmt::Display for ItemType {
 }
 
 #[derive(Copy, Clone)]
+#[cfg_attr(
+    any(feature = "visualization", feature = "visualization_wasm"),
+    derive(Component)
+)]
 pub struct Item {
     pub id: u32,
     pub value: ItemType,

diff --git a/antsforaging/src/visualization/ant.rs b/antsforaging/src/visualization/ant.rs
@@ -1,7 +1,6 @@
 use crate::model::ant::Ant;
 use crate::model::state::ModelState;
-use krabmaga::bevy::ecs::component::TableStorage;
-use krabmaga::bevy::prelude::{Component, Quat, Transform, Visibility};
+use krabmaga::bevy::prelude::{Quat, Transform, Visibility};
 use krabmaga::engine::agent::Agent;
 use krabmaga::engine::location::Int2D;
 use krabmaga::engine::state::State;
@@ -10,9 +9,7 @@ use krabmaga::visualization::agent_render::{AgentRender, SpriteType};
 pub struct AntVis {
     pub id: u32,
 }
-impl Component for AntVis {
-    type Storage = TableStorage;
-}
+
 impl AgentRender for AntVis {
     fn sprite(&self, _agent: &Box<dyn Agent>, _state: &Box<&dyn State>) -> SpriteType {
         SpriteType::Emoji(String::from("ant"))

diff --git a/antsforaging/src/visualization/item.rs b/antsforaging/src/visualization/item.rs
@@ -1,8 +1 @@
-use krabmaga::bevy::ecs::component::TableStorage;
-use krabmaga::bevy::prelude::Component;
-
-use crate::model::state::Item;
-
-impl Component for Item {
-    type Storage = TableStorage;
-}
+// Item now derives Component in model/state.rs for Bevy 0.18 compatibility.
diff --git a/antsforaging/src/visualization/vis_state.rs b/antsforaging/src/visualization/vis_state.rs
@@ -15,7 +15,7 @@ use crate::model::ant::Ant;
 use crate::model::state::*;
 use crate::visualization::ant::AntVis;
 use krabmaga::bevy::ecs as bevy_ecs;
-use krabmaga::bevy::ecs::system::Resource;
+use krabmaga::bevy::prelude::Resource;
 
 #[derive(Clone, Resource)]
 pub struct VisState;
@@ -48,7 +48,7 @@ impl VisualizationState<ModelState> for VisState {
 
     fn get_agent(
         &self,
-        agent_render: &Box<dyn AgentRender>,
+        agent_render: &dyn AgentRender,
         state: &Box<&dyn StateTrait>,
     ) -> Option<Box<dyn Agent>> {
         let state = state.as_any().downcast_ref::<ModelState>().unwrap();

diff --git a/flockers/Cargo.toml b/flockers/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "flockers"
-version = "0.3.0"
+version = "0.6.0"
 authors = [
     "Carmine Spagnuolo <spagnuolocarmine@gmail.com>", 
     "Alessia Antelmi <aantelmi@unisa.it>", 
@@ -18,7 +18,7 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-krabmaga = "0.5.*"
+krabmaga = "0.6.*"
 
 [features]
 parallel = ["krabmaga/parallel"]

diff --git a/flockers/src/main.rs b/flockers/src/main.rs
@@ -1,19 +1,11 @@
 // No visualization specific imports
 #[cfg(not(any(feature = "visualization", feature = "visualization_wasm")))]
-use {
-    krabmaga::engine::schedule::Schedule,
-    // krabmaga::*,
-    krabmaga::engine::state::State,
-    krabmaga::simulate,
-    krabmaga::Info,
-    krabmaga::*,
-    std::time::Duration,
-};
+use {krabmaga::simulate, krabmaga::*};
 // Visualization specific imports
 #[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
 use {
     crate::visualization::vis_state::VisState, krabmaga::bevy::prelude::Color,
-    krabmaga::visualization::visualization::Visualization,
+    krabmaga::engine::state::State, krabmaga::visualization::visualization::Visualization,
 };
 
 use crate::model::state::Flocker;
@@ -35,16 +27,14 @@ pub static TOROIDAL: bool = true;
 // Main used when only the simulation should run, without any visualization.
 #[cfg(not(any(feature = "visualization", feature = "visualization_wasm")))]
 fn main() {
-    use krabmaga::simulate_old;
-
-    let step = 200;
+    let step = 1000;
 
-    let dim = (100., 100.);
-    let num_agents = 1000;
+    let dim = (1000., 1000.);
+    let num_agents = 10000;
     let state = Flocker::new(dim, num_agents);
     // let _ = simulate!(state, step, 1, Info::Normal);
-    // let _ = simulate!(state, step, 1);
-    let _ = simulate_old!(state, step, 1, krabmaga::Info::Normal);
+    simulate!(state, step, 1);
+    // let _ = simulate_old!(state, step, 1, krabmaga::Info::Normal);
 }
 
 // Main used when a visualization feature is applied.
@@ -56,7 +46,7 @@ fn main() {
     Visualization::default()
         .with_window_dimensions(1000., 700.)
         .with_simulation_dimensions(dim.0, dim.1)
-        .with_background_color(Color::rgb(0., 0., 0.))
+        .with_background_color(Color::srgb(0., 0., 0.))
         .with_name("Flockers")
         .start::<VisState, Flocker>(VisState, state);
 }
diff --git a/flockers/src/visualization/vis_state.rs b/flockers/src/visualization/vis_state.rs
@@ -1,4 +1,4 @@
-use krabmaga::bevy::prelude::Commands;
+use krabmaga::bevy::prelude::{Commands, Resource};
 use krabmaga::engine::agent::Agent;
 use krabmaga::engine::location::Real2D;
 use krabmaga::engine::schedule::Schedule;
@@ -12,7 +12,6 @@ use crate::model::bird::Bird;
 use crate::model::state::Flocker;
 use crate::visualization::bird_vis::BirdVis;
 use krabmaga::bevy::ecs as bevy_ecs;
-use krabmaga::bevy::ecs::system::Resource;
 
 #[derive(Clone, Resource)]
 pub struct VisState;
@@ -43,7 +42,7 @@ impl VisualizationState<Flocker> for VisState {
 
     fn get_agent(
         &self,
-        agent_render: &Box<dyn AgentRender>,
+        agent_render: &dyn AgentRender,
         state: &Box<&dyn State>,
     ) -> Option<Box<dyn Agent>> {
         // TODO we don't just need the agent associated to the agent render, we need the correct one so that we

diff --git a/flockers_mpi/Cargo.toml b/flockers_mpi/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "flockers_mpi"
-version = "0.4.1"
+version = "0.6.0"
 authors = [
     "Carmine Spagnuolo <spagnuolocarmine@gmail.com>", 
     "Alessia Antelmi <aantelmi@unisa.it>", 
@@ -18,7 +18,7 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-krabmaga = "0.5.*"
+krabmaga = "0.6.*"
 
 [features]
 distributed_mpi = ["krabmaga/distributed_mpi"]

diff --git a/forestfire/Cargo.toml b/forestfire/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "forestfire"
-version = "0.2.0"
+version = "0.6.0"
 authors = [
     "Carmine Spagnuolo <spagnuolocarmine@gmail.com>", 
     "Alessia Antelmi <aantelmi@unisa.it>", 
@@ -18,7 +18,7 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-krabmaga = "0.5.*"
+krabmaga = "0.6.*"
 
 [features]
 visualization = ["krabmaga/visualization"]

diff --git a/forestfire/src/main.rs b/forestfire/src/main.rs
@@ -45,7 +45,7 @@ fn main() {
     let mut app = Visualization::default()
         .with_simulation_dimensions(state.dim.0 as f32, state.dim.1 as f32)
         .with_window_dimensions(1000., 720.)
-        .with_background_color(Color::BLACK)
+        .with_background_color(Color::srgb(0., 0., 0.))
         .with_name("Forest Fire Model")
         .setup::<ForestVis, Forest>(ForestVis, state);
     app.add_systems(FixedUpdate, DenseGrid2D::<Tree>::render);

diff --git a/forestfire/src/model/forest.rs b/forestfire/src/model/forest.rs
@@ -1,5 +1,9 @@
 use crate::model::spread::Spread;
 use core::fmt;
+#[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
+use krabmaga::bevy::ecs as bevy_ecs;
+#[cfg(any(feature = "visualization", feature = "visualization_wasm"))]
+use krabmaga::bevy::prelude::Component;
 use krabmaga::engine::fields::dense_object_grid_2d::DenseGrid2D;
 use krabmaga::engine::fields::field::Field;
 use krabmaga::engine::location::Int2D;
@@ -29,6 +33,10 @@ impl fmt::Display for Status {
 }
 
 #[derive(Copy, Clone)]
+#[cfg_attr(
+    any(feature = "visualization", feature = "visualization_wasm"),
+    derive(Component)
+)]
 pub struct Tree {
     pub id: i32,
     pub status: Status,

diff --git a/forestfire/src/visualization/custom_systems.rs b/forestfire/src/visualization/custom_systems.rs
@@ -1,15 +1,9 @@
 use crate::model::forest::*;
-use krabmaga::bevy::ecs::component::TableStorage;
-use krabmaga::bevy::prelude::Component;
 use krabmaga::engine::fields::dense_object_grid_2d::DenseGrid2D;
 use krabmaga::engine::fields::sparse_object_grid_2d::SparseGrid2D;
 use krabmaga::engine::location::Int2D;
 use krabmaga::visualization::fields::object_grid_2d::RenderObjectGrid2D;
 
-impl Component for Tree {
-    type Storage = TableStorage;
-}
-
 impl RenderObjectGrid2D<Forest, Tree> for DenseGrid2D<Tree> {
     fn fetch_sparse_grid(_state: &Forest) -> Option<&SparseGrid2D<Tree>> {
         None

diff --git a/forestfire/src/visualization/forest_vis.rs b/forestfire/src/visualization/forest_vis.rs
@@ -1,8 +1,8 @@
 use crate::model::forest::Forest;
 use crate::Tree;
 use krabmaga::bevy::ecs as bevy_ecs;
-use krabmaga::bevy::ecs::system::Resource;
 use krabmaga::bevy::prelude::Commands;
+use krabmaga::bevy::prelude::Resource;
 use krabmaga::engine::agent::Agent;
 use krabmaga::engine::fields::dense_object_grid_2d::DenseGrid2D;
 use krabmaga::engine::fields::field::Field;
@@ -49,7 +49,7 @@ impl VisualizationState<Forest> for ForestVis {
 
     fn get_agent(
         &self,
-        _agent_render: &Box<dyn AgentRender>,
+        _agent_render: &dyn AgentRender,
         _state: &Box<&dyn State>,
     ) -> Option<Box<dyn Agent>> {
         None

diff --git a/forestfire_bayesian/Cargo.toml b/forestfire_bayesian/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "forestfire_bayesian"
-version = "0.2.0"
+version = "0.6.0"
 authors = [
     "Carmine Spagnuolo <spagnuolocarmine@gmail.com>", 
     "Alessia Antelmi <aantelmi@unisa.it>", 
@@ -18,7 +18,7 @@ readme = "README.md"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-krabmaga = "0.5.*"
+krabmaga = "0.6.*"
 
 [features]
 bayesian = ["krabmaga/bayesian"]
diff --git a/forestfire_bayesian/src/visualization/custom_systems.rs b/forestfire_bayesian/src/visualization/custom_systems.rs
@@ -1,13 +1,9 @@
-use krabmaga::bevy::ecs::component::TableStorage;
-use krabmaga::bevy::prelude::Component;
 use crate::model::forest::*;
 use krabmaga::engine::fields::dense_object_grid_2d::DenseGrid2D;
 use krabmaga::engine::fields::sparse_object_grid_2d::SparseGrid2D;
 use krabmaga::engine::location::Int2D;
 use krabmaga::visualization::fields::object_grid_2d::RenderObjectGrid2D;
 
-impl Component for Tree { type Storage = TableStorage; }
-
 impl RenderObjectGrid2D<Forest, Tree> for DenseGrid2D<Tree> {
     fn fetch_sparse_grid(_state: &Forest) -> Option<&SparseGrid2D<Tree>> {
         None