From 0ebfb13e047a71323a17e3f14b4d58cab65722f5 Mon Sep 17 00:00:00 2001
From: Rowan Torbitzky-Lane <rowan.a.tl@protonmail.com>
Date: Thu, 20 Feb 2025 01:49:38 -0600
Subject: [PATCH] less IO functions, pre-loop done, next is the main loop

---
 src/HushGP/GP.hs     | 32 +++++++++++++++++++++++---------
 src/HushGP/Genome.hs | 13 +++++++++----
 2 files changed, 32 insertions(+), 13 deletions(-)

diff --git a/src/HushGP/GP.hs b/src/HushGP/GP.hs
index c3a61c0..90a4815 100644
--- a/src/HushGP/GP.hs
+++ b/src/HushGP/GP.hs
@@ -1,23 +1,37 @@
+-- | The main file containing information about the GP loop and various population transformation functions.
 module HushGP.GP where
 
-import HushGP.State
 import HushGP.Genome
 import HushGP.GP.PushArgs
-import Control.Monad
 import Control.Parallel.Strategies
+import Control.Monad
+import Data.List (sort)
 -- import Debug.Trace (trace, traceStack)
 
 -- | Using a PushArgs object, generates a population of the specified size with the
 -- specified instructions in parallel.
-generatePopulation :: PushArgs -> [Individual]
-generatePopulation pushArgs =
-  replicate (populationSize pushArgs) (makeRandomIndividual pushArgs) `using` rpar
+generatePopulation :: PushArgs -> IO [Individual]
+generatePopulation pushArgs = do
+  pop <- replicateM (populationSize pushArgs) (makeRandomIndividual pushArgs)
+  return (pop `using` evalList rpar) -- Does this work? Need to test this with the HEC viewing tool.
 
-evaluatePopulation :: PushArgs -> [Individual] -> IO [Individual]
-evaluatePopulation pushArgs population = map (fmap (errorFunction pushArgs pushArgs (trainingData pushArgs)) . plushy) population
+-- | Evaluates a population of plushies with the error function passed in via PushArgs and sorts them.
+-- TODO: Need to make this runnable in parallel too.
+evaluatePopulation :: PushArgs -> [Individual] -> [Individual]
+evaluatePopulation pushArgs population = sort $ zipWith updateIndividual (map (errorFunction pushArgs pushArgs (trainingData pushArgs) . plushy) population) population
+
+-- | A helper function used in evaluatePopulation. Takes a [Double] as the error scores and an individual.
+-- Updates the error fields in an individual, and returns it.
+updateIndividual :: [Double] -> Individual -> Individual
+updateIndividual errors ind = ind{totalFitness = Just (sum errors), fitnessCases = Just errors}
 
 -- | The start of the gp loop. TODO: Make this more accurate later.
 gpLoop :: PushArgs -> IO ()
 gpLoop pushArgs = do
-  let unEvaledPopulation =  generatePopulation pushArgs
-  print "gamer"
+  unEvaledPopulation <- generatePopulation pushArgs
+  let evaledPop = evaluatePopulation pushArgs unEvaledPopulation
+  print evaledPop
+
+-- | The guts of the GP loop. Where the work gets done after the initialization happens
+-- in the main gpLoop function.
+-- gpLoop' 
diff --git a/src/HushGP/Genome.hs b/src/HushGP/Genome.hs
index f57e7ab..62a719f 100644
--- a/src/HushGP/Genome.hs
+++ b/src/HushGP/Genome.hs
@@ -13,14 +13,19 @@ import HushGP.GP.PushArgs
 -- | The structure for an individual containing the genome, the totalFitness, and
 -- the individual fitness cases for lexicase.
 data Individual = Individual {
-  plushy :: IO [Gene],
+  plushy :: [Gene],
   totalFitness :: Maybe Double,
   fitnessCases :: Maybe [Double]
-}
+} deriving (Show, Eq)
+
+instance Ord Individual where
+  ind0 <= ind1 = totalFitness ind0 <= totalFitness ind1
 
 -- | Makes a random individual based on the variables in a passed PushArgs.
-makeRandomIndividual :: PushArgs -> Individual
-makeRandomIndividual pushArgs = Individual {plushy = makeRandomPlushy pushArgs, totalFitness = Nothing, fitnessCases = Nothing}
+makeRandomIndividual :: PushArgs -> IO Individual
+makeRandomIndividual pushArgs = do
+  randomPlushy <- makeRandomPlushy pushArgs
+  return Individual {plushy = randomPlushy, totalFitness = Nothing, fitnessCases = Nothing}
 
 -- | Makes a random plushy from variables in a passed PushArgs.
 makeRandomPlushy :: PushArgs -> IO [Gene]