Changeset 10462 for branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking/3.4/Analyzers/GenealogyAnalyzer.cs

Timestamp:

02/19/14 08:25:37 (10 years ago)

Author:

bburlacu

Message:

#1772: Fixed bug in configuration of the tracking operators, fixed null reference exception when trying to access fragments in SymbolicDataAnalysisGenealogyGraphView

File:

• branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking/3.4/Analyzers/GenealogyAnalyzer.cs (modified) (3 diffs)

branches/HeuristicLab.EvolutionTracking/HeuristicLab.EvolutionTracking/3.4/Analyzers/GenealogyAnalyzer.cs

@@ -1 @@
-using System.Collections.Generic;
-using System.Linq;
+using System.Linq;
 using HeuristicLab.Common;
 using HeuristicLab.Core;
@@ -167 +166 @@
     }
 
-    public override IOperation Apply() {
-      if (Generations.Value == 0) {
-        foreach (var individual in PopulationParameter.ActualValue) {
-          var vertex = new GenealogyGraphNode<T> {
-            Content = individual,
-            Rank = Generations.Value
-          };
-          GenealogyGraph.AddVertex(vertex);
-        }
-        // at the beginning we add the before/after operators to the instrumented operators
+    private void ConfigureTrackingOperators() {
+      // at the beginning we add the before/after operators to the instrumented operators
+      if (Crossover != null) {
+        var instrumentedCrossover = (InstrumentedOperator)Crossover;
+        instrumentedCrossover.AfterExecutionOperators.Clear();
+        instrumentedCrossover.BeforeExecutionOperators.Clear();
+
         if (EnableCrossoverTracking.Value) {
           if (BeforeCrossoverOperator != null) {
             BeforeCrossoverOperator.ParentsParameter.ActualName = CrossoverParentsParameterName;
             BeforeCrossoverOperator.ChildParameter.ActualName = CrossoverChildParameterName;
-            var instrumentedCrossover = (InstrumentedOperator)Crossover;
             instrumentedCrossover.BeforeExecutionOperators.Add(BeforeCrossoverOperator);
           }
@@ -187 +182 @@
             AfterCrossoverOperator.ParentsParameter.ActualName = CrossoverParentsParameterName;
             AfterCrossoverOperator.ChildParameter.ActualName = CrossoverChildParameterName;
-            var instrumentedCrossover = (InstrumentedOperator)Crossover;
             instrumentedCrossover.AfterExecutionOperators.Add(AfterCrossoverOperator);
           }
         }
-
+      }
+
+      if (Manipulator != null) {
+        var instrumentedManipulator = (InstrumentedOperator)Manipulator;
+        instrumentedManipulator.AfterExecutionOperators.Clear();
+        instrumentedManipulator.BeforeExecutionOperators.Clear();
         if (EnableManipulatorTracking.Value && Manipulator != null) {
-          if (BeforeManipulatorOperator == null) {
-            BeforeManipulatorOperator = new BeforeManipulatorOperator<T>();
+          if (BeforeManipulatorOperator != null) {
             BeforeManipulatorOperator.ChildParameter.ActualName = ManipulatorChildParameterName;
-            var instrumentedManipulator = (InstrumentedOperator)Manipulator;
             instrumentedManipulator.BeforeExecutionOperators.Add(BeforeManipulatorOperator);
           }
-          if (AfterManipulatorOperator == null) {
-            AfterManipulatorOperator = new AfterManipulatorOperator<T>();
+          if (AfterManipulatorOperator != null) {
             AfterManipulatorOperator.ChildParameter.ActualName = ManipulatorChildParameterName;
-            var instrumentedManipulator = (InstrumentedOperator)Manipulator;
             instrumentedManipulator.AfterExecutionOperators.Add(AfterManipulatorOperator);
           }
         }
+      }
+    }
+
+    public override IOperation Apply() {
+      var population = PopulationParameter.ActualValue.ToList();
+      var qualities = QualityParameter.ActualValue.ToList();
+
+      if (Generations.Value == 0) {
+        ConfigureTrackingOperators();
+
+        foreach (var individual in population) {
+          var vertex = new GenealogyGraphNode<T> { Content = individual, Rank = Generations.Value, };
+          GenealogyGraph.AddVertex(vertex);
+        }
       } else {
-        // add missing elite individuals in the current generation 
-        // TODO: optimize for speed
-        var currGen = new HashSet<T>(GenealogyGraph.Ranks[Generations.Value].Cast<IGenealogyGraphNode<T>>().Select(x => x.Content));
-        foreach (var individual in PopulationParameter.ActualValue) {
-          if (currGen.Contains(individual)) continue;
-          // it is an elite which was already added to the graph in the previous generation
+        var elite = population.FirstOrDefault(x => GenealogyGraph.Contains(x));
+        if (elite != null) {
           var vertex = new GenealogyGraphNode<T> {
-            Content = individual,
+            Content = elite,
             Rank = Generations.Value,
             IsElite = true
           };
+          var previousVertex = GenealogyGraph[elite].Last();
           GenealogyGraph.AddVertex(vertex);
-        }
-      }
-      // update qualities for the nodes in the graph
-      var population = PopulationParameter.ActualValue.ToList();
-      var qualities = QualityParameter.ActualValue.ToList();
+          previousVertex.AddForwardArc(vertex);
+          vertex.AddReverseArc(previousVertex);
+        }
+      }
+      // update qualities
       for (int i = 0; i < population.Count; ++i) {
-        foreach (var v in GenealogyGraph[population[i]].Cast<IGenealogyGraphNode<T>>()) {
-          v.Quality = qualities[i].Value;
+        var individual = population[i];
+        foreach (var vertex in GenealogyGraph[individual].Cast<IGenealogyGraphNode<T>>()) {
+          vertex.Quality = qualities[i].Value;
         }
       }