Changeset 9754 for branches/HeuristicLab.Analysis.AlgorithmBehavior/HeuristicLab.Analysis.AlgorithmBehavior.Analyzers/3.3/SolutionsCaching/PermutationSolutionDictionary.cs

Timestamp:

07/25/13 11:14:56 (11 years ago)

Author:

ascheibe

Message:

#1886 added a caching analyzer for crossover

File:

• branches/HeuristicLab.Analysis.AlgorithmBehavior/HeuristicLab.Analysis.AlgorithmBehavior.Analyzers/3.3/SolutionsCaching/PermutationSolutionDictionary.cs (modified) (5 diffs)

branches/HeuristicLab.Analysis.AlgorithmBehavior/HeuristicLab.Analysis.AlgorithmBehavior.Analyzers/3.3/SolutionsCaching/PermutationSolutionDictionary.cs

@@ -20 +20 @@
 #endregion
 
+using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -30 +31 @@
   [Item("PermutationSolutionDictionary", "Stores permutations generated by an algorithm.")]
   [StorableClass]
-  public class PermutationSolutionDictionary<TKey, TValue> : SolutionDictionary<TKey, TValue>
-    where TKey : PermutationWrapper
-    where TValue : PermutationInformation {
+  public class PermutationSolutionDictionary : SolutionDictionary<PermutationWrapper, PermutationInformation> {
+    [Storable]
+    protected Dictionary<PermutationWrapper, List<PermutationInformation>> solutionDictionary;
+    public Dictionary<PermutationWrapper, List<PermutationInformation>> SolutionDictionary { get { return solutionDictionary; } }
 
     public PermutationSolutionDictionary() {
-      solutionDictionary = new Dictionary<TKey, List<TValue>>(new PermutationWrapperEqualityComparer());
+      solutionDictionary = new Dictionary<PermutationWrapper, List<PermutationInformation>>(new PermutationWrapperEqualityComparer());
     }
 
     [StorableConstructor]
     protected PermutationSolutionDictionary(bool deserializing) : base(deserializing) { }
-    protected PermutationSolutionDictionary(PermutationSolutionDictionary<TKey, TValue> original, Cloner cloner)
+    protected PermutationSolutionDictionary(PermutationSolutionDictionary original, Cloner cloner)
       : base(original, cloner) {
+      this.solutionDictionary = new Dictionary<PermutationWrapper, List<PermutationInformation>>(original.solutionDictionary, new PermutationWrapperEqualityComparer());
     }
 
-    protected override void InsertKeyValue(TKey key, TValue value) {
+    public override int Size() {
+      return solutionDictionary.Count;
+    }
+
+    public override void Add(PermutationWrapper key, PermutationInformation value) {
+      if (solutionDictionary.Count == 0) {
+        var lst = new List<PermutationInformation>();
+        lst.Add(value);
+        solutionDictionary.Add(key, lst);
+        return;
+      }
+
+      if (ContainsKey(key)) {
+        if (!solutionDictionary[key].Contains(value))
+          solutionDictionary[key].Add(value);
+      } else {
+        InsertKeyValue(key, value);
+      }
+    }
+
+    protected override void InsertKeyValue(PermutationWrapper key, PermutationInformation value) {
       int diffCnt;
       Permutation p = key.GetPermutation();
@@ -53 +76 @@
       }
 
-      var lst = new List<TValue>();
+      var lst = new List<PermutationInformation>();
       lst.Add(value);
       solutionDictionary.Add(key, lst);
@@ -81 +104 @@
     }
 
+    public PermutationWrapper GetPermutationWrapperOfPermutation(Permutation permutation) {
+      PermutationEqualityComparer peq = new PermutationEqualityComparer();
+      var res = solutionDictionary.Where(x => peq.GetHashCode(x.Key.GetPermutation()) == peq.GetHashCode(permutation));
+
+      //TODO: fix this
+      if (res.Count() > 1) {
+        throw new Exception("ERROR: Cannot conatin duplicates!");
+      }
+
+      return res.First().Key;
+    }
+
     public override IDeepCloneable Clone(Cloner cloner) {
-      return new PermutationSolutionDictionary<TKey, TValue>(this, cloner);
+      return new PermutationSolutionDictionary(this, cloner);
     }
 
@@ -96 +131 @@
       return solutionDictionary.Where(x => x.Value.Where(y => y.Generation == generation).Count() != 0).Select(x => x.Key).ToList<PermutationWrapper>();
     }
+
+    public override bool ContainsKey(PermutationWrapper key) {
+      return solutionDictionary.ContainsKey(key);
+    }
   }
 }