From 5c99e80670be7326967f9c89fcae9c3965aa81da Mon Sep 17 00:00:00 2001
From: Rendo <geometicalplayer@gmail.com>
Date: Sat, 8 Nov 2025 20:57:55 +0500
Subject: [PATCH] Use of children count and simple error handling

---
 src/node/mod.rs           |  7 ++++++-
 src/node/node_modifier.rs | 27 ++++++++++++++++++++++++---
 2 files changed, 30 insertions(+), 4 deletions(-)

diff --git a/src/node/mod.rs b/src/node/mod.rs
index c73ec60..0e67b0f 100644
--- a/src/node/mod.rs
+++ b/src/node/mod.rs
@@ -12,10 +12,15 @@ pub struct Node {
 }
 
 impl Node {
-    pub fn new(children: Option<Vec<Node>>, handler: NodeHandler) -> Self {
+    pub fn new(
+        children: Option<Vec<Node>>,
+        handler: NodeHandler,
+        max_children_count: Option<usize>,
+    ) -> Self {
         Self {
             children: children.unwrap_or(vec![]),
             handler: handler,
+            max_children_count,
         }
     }
     pub fn get_value(&self) -> f64 {
diff --git a/src/node/node_modifier.rs b/src/node/node_modifier.rs
index 5f8fd33..ac2f932 100644
--- a/src/node/node_modifier.rs
+++ b/src/node/node_modifier.rs
@@ -1,10 +1,14 @@
 use crate::node::Node;
+use crate::node::NodeHandler;
 use crate::node::functions;
-use crate::node::handler::NodeHandler;
 use rand::random_range;
 
 const PICK_STOP_PROBABILITY: u8 = 2;
 
+pub enum NodeManipulationError {
+    TooMuchChildren(Node),
+}
+
 pub struct NodeModifier<'a> {
     picked_node: &'a mut Node,
     function_mutation_pool: Vec<fn(Vec<f64>) -> f64>,
@@ -57,16 +61,33 @@ impl<'a> NodeModifier<'a> {
         }
     }
 
-    pub fn add_node(&mut self, node: Node) {
+    pub fn add_node(&mut self, node: Node) -> Result<(), NodeManipulationError> {
+        if let Some(x) = self.picked_node.max_children_count {
+            if self.picked_node.children.len() + 1 >= x {
+                return Err(NodeManipulationError::TooMuchChildren(node));
+            }
+        }
         self.picked_node.children.push(node);
+        Ok(())
     }
 
-    pub fn insert_node(&mut self, mut node: Node, between: Option<usize>) {
+    pub fn insert_node(
+        &mut self,
+        mut node: Node,
+        between: Option<usize>,
+    ) -> Result<(), NodeManipulationError> {
+        if let Some(x) = self.picked_node.max_children_count {
+            if self.picked_node.children.len() + 1 >= x {
+                return Err(NodeManipulationError::TooMuchChildren(node));
+            }
+        }
         let children_count = self.picked_node.children.len();
         let operated_index = between.unwrap_or(random_range(0..children_count));
         let moved = self.picked_node.children.remove(operated_index);
         node.children.push(moved);
         self.picked_node.children.insert(operated_index, node);
+
+        Ok(())
     }
 
     pub fn mutate_node(&mut self) {