From d6b93daf7a470a7aef5a5a1727d1e4f7082e02fa Mon Sep 17 00:00:00 2001
From: Rostyslav Hnatyshyn <rostyslav.hnatyshyn@gmail.com>
Date: Tue, 2 Jan 2024 16:12:32 -0700
Subject: [PATCH] queen can now infinitely lay eggs

---
 Cargo.toml  |  1 +
 src/main.rs | 28 +++++++++++++++++++---------
 2 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index ac2581e..4a3cb14 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -6,5 +6,6 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+downcast-rs = "1.2.0"
 ncurses = "5.101.0"
 rand = "0.8.5"
diff --git a/src/main.rs b/src/main.rs
index 2184a42..ab4d792 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,5 @@
 extern crate ncurses;
+extern crate downcast_rs;
 
 use ncurses::*;
 use rand::seq::SliceRandom;
@@ -75,7 +76,10 @@ impl World {
     }
 }
 
-trait Entity: AI + Renderable {}
+use downcast_rs::{Downcast, impl_downcast};
+
+trait Entity: AI + Renderable + Downcast {}
+impl_downcast!(Entity);
 
 struct Colony {
     ants: HashMap<u32, Box<dyn Entity>>,
@@ -149,6 +153,7 @@ impl Queen {
 struct Egg {
     pos: Point,
     counter: u8,
+    queen_id: u32
 }
 
 impl Renderable for Egg {
@@ -158,10 +163,11 @@ impl Renderable for Egg {
 }
 
 impl Egg {
-    fn new(x: i32, y: i32) -> Egg {
+    fn new(x: i32, y: i32, queen_id: u32) -> Egg {
         Egg {
             pos: Point(x, y),
             counter: 10,
+            queen_id
         }
     }
 }
@@ -172,7 +178,7 @@ impl AI for Egg {
             self.counter -= 1;
             return BoardCommand::Noop;
         } else {
-            BoardCommand::Hatch(id)
+            BoardCommand::Hatch(id, self.queen_id)
         }
     }
 
@@ -183,8 +189,8 @@ impl AI for Egg {
 
 enum BoardCommand {
     Dig(Point),
-    LayEgg(Point),
-    Hatch(u32),
+    LayEgg(Point, u32),
+    Hatch(u32, u32),
     Noop,
 }
 
@@ -234,7 +240,7 @@ impl AI for Queen {
                 // choose between laying an egg and moving
                 if self.egg_count < 3 {
                     self.egg_count += 1;
-                    return BoardCommand::LayEgg(pos);
+                    return BoardCommand::LayEgg(pos, id);
                 } else {
                     self.pos = pos;
                     return BoardCommand::Noop;
@@ -283,12 +289,16 @@ fn simulate(c: &mut Colony, w: &mut World, b: &mut Board) {
             BoardCommand::Dig(pos) => {
                 w.clear(pos);
             }
-            BoardCommand::LayEgg(pos) => {
-                c.add_entity(Egg::new(pos.0, pos.1));
+            BoardCommand::LayEgg(pos, id) => {
+                c.add_entity(Egg::new(pos.0, pos.1, id));
             }
-            BoardCommand::Hatch(egg_id) => {
+            BoardCommand::Hatch(egg_id, queen_id) => {
                 let egg = c.ants.remove(&egg_id);
                 let pos = egg.unwrap().get_position();
+
+                let q = c.ants.get_mut(&queen_id).unwrap();
+                let queen: &mut Queen = q.downcast_mut::<Queen>().unwrap();
+                queen.egg_count -= 1;
                 c.add_entity(Ant::new(pos.0, pos.1));
             }
             BoardCommand::Noop => {}