New templates

docs/algorithms.md

@@ -8,7 +8,7 @@ Documented generation process is easier to visualize.
 ### Peashooter example
 1. Create [generator] root
 2. Add [transfer] stem
-3. Add [trigger] head
+3. Add [transfer] head
 4. Add [modifier] leaf
 5. Return to head
 6. Add [output] face

rust-pvz-genetics/src/genetics/edge.rs

@@ -1,31 +0,0 @@
-use std::cmp::Ordering;
-use crate::genetics::gene::Gene;
-
-#[derive(PartialEq,Eq,Clone)]
-pub struct Edge {
-    pub from: Gene,
-    pub to: Gene
-}
-
-impl Edge{
-    pub fn new(from: Gene, to: Gene) -> Self {
-        Self {
-            from,
-            to
-        }
-    }
-}
-
-impl PartialOrd for Edge {
-    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
-        Some(self.cmp(other))
-    }
-}
-
-impl Ord for Edge {
-    fn cmp(&self, other: &Self) -> Ordering {
-        if self.from == other.to {return Ordering::Greater;}
-        else if self.to == other.from {return Ordering::Less;}
-        else {return Ordering::Equal;}
-    }
-}

rust-pvz-genetics/src/genetics/gene.rs

@@ -6,6 +6,7 @@ pub struct Gene {
 
 #[derive(PartialEq,Eq,Clone)]
 pub enum GeneType {
+    Dummy,
     PureProducer {
         create_flow: fn()
     },

rust-pvz-genetics/src/genetics/genome.rs

@@ -1,8 +1,8 @@
-use crate::genetics::{gene::Gene, edge::Edge};
+use crate::genetics::gene::Gene;
 
 pub struct PlantGenome {
     pub(crate) genes: Vec<Gene>,
-    pub(crate) edges: Vec<Edge>
+    pub(crate) edges: Vec<(usize,usize)>
 }
 
 impl PlantGenome {
@@ -12,43 +12,30 @@ impl PlantGenome {
             edges: Vec::new()
         }
     }
-    fn insert_gene(&mut self, gene: Gene, parent: Option<Gene>) -> Option<PlantGenomeInsertError> {
-        if let None = parent && self.genes.len() > 0 {
-            return Some(PlantGenomeInsertError::RootExists);
-        }
-        if self.genes.contains(&gene){
-            return Some(PlantGenomeInsertError::GeneExists);
+    pub fn from_edges(nodes: Vec<Gene>,edges: Vec<(usize,usize)>) -> Result<Self,PlantCreationError> {
+        // Check for nodes amount and edges amount mismatch
+        if nodes.len() != edges.len() {
+            return Err(PlantCreationError::AmountMismatch);
         }
 
-        if let Some(parent_gene) = parent {
-            let edge = Edge::new(parent_gene,gene.clone());
-            if self.edges.contains(&edge) {
-                return Some(PlantGenomeInsertError::EdgeExists);
-            }
-            self.edges.push(edge);
+        if nodes.len() == 0 {
+            return Err(PlantCreationError::EmptyNodes);
         }
 
-        self.genes.push(gene);
-        self.update_graph();
-        return None;
+        if edges.len() == 0 {
+            return Err(PlantCreationError::EmptyEdges);
         }
-    pub(crate) fn update_graph(&mut self) {
-        // Sort by parent to guarantee root at [0]
-        self.edges.sort();
-    }
-    pub fn crossingover(&mut self, other: &mut Self) {
 
-        self.update_graph();
-        other.update_graph();
-    }
-    pub fn mutation(&mut self) {
-        self.update_graph();
+        return Ok(Self {
+            genes: nodes,
+            edges
+        });
     }
 }
 
 #[derive(Debug)]
-pub enum PlantGenomeInsertError {
-    GeneExists,
-    EdgeExists,
-    RootExists,
+pub enum PlantCreationError {
+    EmptyNodes,
+    EmptyEdges,
+    AmountMismatch,
 }

rust-pvz-genetics/src/genetics/genome_builder.rs

@@ -1,88 +0,0 @@
-use crate::genetics::prelude::*;
-
-pub struct GenomeBuilder {
-    genome: PlantGenome,
-    current_gene: usize,
-}
-
-impl GenomeBuilder {
-
-    pub fn new() -> Self {
-        Self {
-            genome: PlantGenome::new(),
-            current_gene: 0
-        }
-    }
-
-    pub fn finish(mut self) -> PlantGenome {
-        self.genome.update_graph();
-        return self.genome;
-    }
-
-    pub fn create_root(mut self, root: Gene) -> Result<Self,GenomeBuilderError> {
-        // Check for root existance
-        if self.genome.genes.len() > 0 {
-            return Err(GenomeBuilderError::RootExists);
-        }
-        self.genome.genes.push(root);
-
-        return Ok(self);
-    }
-
-    pub fn insert_gene(mut self, gene: Gene, parent: Gene) -> Result<Self,GenomeBuilderError> {
-        // Check for gene collision
-        if self.genome.genes.contains(&gene){
-            return Err(GenomeBuilderError::GeneExists);
-        }
-
-        let edge = Edge::new(parent,gene.clone());
-        self.genome.edges.push(edge);
-
-        self.current_gene = self.genome.genes.len();
-        self.genome.genes.push(gene);
-
-        Ok(self)
-    }
-    pub fn add_gene(mut self, gene: Gene) -> Result<Self,GenomeBuilderError> {
-        // Check for root existance
-        if self.genome.genes.len() == 0 {
-           return Err(GenomeBuilderError::RootDoesntExist);
-        }
-        // Check for gene collision
-        if self.genome.genes.contains(&gene){
-            return Err(GenomeBuilderError::GeneExists);
-        }
-        let edge = Edge::new(self.get_current_gene().clone(),gene.clone());
-        self.genome.edges.push(edge);
-
-        self.current_gene = self.genome.genes.len();
-        self.genome.genes.push(gene);
-
-        Ok(self)
-
-    }
-    pub fn back(mut self, amount: Option<usize>) -> Self {
-        for _ in 0..amount.unwrap_or(0) {
-            
-        }
-        self
-    }
-    #[inline(always)]
-    pub fn get_current_gene(&self) -> &Gene {
-        &self.genome.genes[self.current_gene]
-    }
-
-    #[inline(always)]
-    pub fn get_current_gene_mut(&mut self) -> &mut Gene {
-        &mut self.genome.genes[self.current_gene]
-    }
-    
-}
-
-#[derive(Debug)]
-pub enum GenomeBuilderError {
-    RootExists,
-    RootDoesntExist,
-    GeneExists,
-    EdgeExists,
-}

rust-pvz-genetics/src/genetics/mod.rs

@@ -1,14 +1,11 @@
 #[cfg(test)]
 mod tests;
 pub mod plant_templates;
-pub mod edge;
 pub mod gene;
 pub mod genome;
-pub mod genome_builder;
 pub mod arguments;
 
 pub mod prelude {
-    pub use crate::genetics::edge::*;
     pub use crate::genetics::genome::*;
     pub use crate::genetics::gene::*;
 }

rust-pvz-genetics/src/genetics/plant_templates.rs

@@ -1,12 +1,45 @@
-use crate::genetics::{genome_builder::GenomeBuilder, prelude::*};
+use crate::genetics::prelude::*;
 
 pub fn peashooter_template() -> PlantGenome {
-    GenomeBuilder::new()
-        .create_root(Gene::new("root", GeneType::pure_producer(||{}))).unwrap()
-        .add_gene(Gene::new("stem", GeneType::fill_all())).unwrap()
-        .add_gene(Gene::new("head",GeneType::random_distribution())).unwrap()
-        .add_gene(Gene::new("leaf",GeneType::modifier(||{}))).unwrap()
-        .back(None)
-        .add_gene(Gene::new("face",GeneType::consumer(||{}))).unwrap()
-        .finish()
+    PlantGenome::from_edges(vec![
+        Gene::new("root", GeneType::pure_producer(||{})),
+        Gene::new("stem", GeneType::fill_all()),
+        Gene::new("head",GeneType::random_distribution()),
+        Gene::new("leaf",GeneType::modifier(||{})),
+        Gene::new("face",GeneType::consumer(||{}))
+    ],vec![
+        (0,1),
+        (1,2),
+        (2,3),
+        (2,4)
+    ]).unwrap()
 }
+
+pub fn sunflower_template() -> PlantGenome {
+    PlantGenome::from_edges(vec![
+        Gene::new("root", GeneType::Dummy),
+        Gene::new("stem", GeneType::Dummy),
+        Gene::new("head", GeneType::modifier(||{})),
+        Gene::new("face", GeneType::consumer(||{})),
+    ], vec![
+        (0,1),
+        (1,2),
+        (2,3),
+    ]).unwrap()
+}
+
+pub fn cherry_bomb_template() -> PlantGenome {
+    PlantGenome::from_edges(vec![
+        Gene::new("root", GeneType::modifier(||{})),
+        Gene::new("head", GeneType::consumer(||{})),
+        Gene::new("head", GeneType::consumer(||{})),
+        Gene::new("face", GeneType::Dummy),
+        Gene::new("face", GeneType::Dummy),
+    ], vec![
+        (0,1),
+        (0,2),
+        (1,3),
+        (2,4),
+    ]).unwrap()
+}
+