From 98eabf2989cb3706ee05b6481b0a7221a6c23a51 Mon Sep 17 00:00:00 2001
From: sanine <sanine.not@pm.me>
Date: Wed, 29 Nov 2023 17:11:07 -0600
Subject: add mutateGenome

---
 src/Genome.hs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'src')

diff --git a/src/Genome.hs b/src/Genome.hs
index e0091b9..cb90ed1 100644
--- a/src/Genome.hs
+++ b/src/Genome.hs
@@ -10,6 +10,8 @@ module Genome
   , mutateGenomeRemoveInternal
   , mutateGenomeAddGene 
   , mutateGenomeRemoveGene 
+
+  , mutateGenome
   ) where
 
 
@@ -172,3 +174,28 @@ mutateGenomeRemoveGene g r =
     (idx, r') = randomR (0, (length $ genes g) - 1) r
     (front, _:back) = splitAt idx (genes g)
   in (g {genes = front ++ back}, r')
+
+
+-- mutate a random gene
+mutateGenomeMutateGene :: RandomGen a => Genome -> a -> (Genome, a)
+mutateGenomeMutateGene g r =
+  let
+    (idx, r') = randomR (0, length (genes g) - 1) r
+    gene = (genes g) !! idx
+    (gene', r'') = mutateGene g gene r'
+    (front, _:back) = splitAt idx (genes g)
+  in (g {genes = front ++ (gene':back)}, r'')
+
+
+-- mutate a genome
+mutateGenome :: RandomGen a => Genome -> a -> (Genome, a)
+mutateGenome g r =
+  let 
+    (f, r') = randomChoice (concat 
+      [ replicate 9 mutateGenomeMutateGene
+      , replicate 2 mutateGenomeAddGene
+      , replicate 2 mutateGenomeRemoveGene
+      , [mutateGenomeAddInternal, mutateGenomeRemoveInternal]
+      ]
+      ) r
+  in f g r'
-- 
cgit v1.2.1