Changeset 13485 for branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem/3.3/ExpertSystem.cs

Timestamp:

12/18/15 12:42:13 (8 years ago)

Author:

abeham

Message:

#2457:

• fixed bugs in iterated sampling view
• worked on download runs from okb feature

File:

• branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem/3.3/ExpertSystem.cs (modified) (2 diffs)

branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem/3.3/ExpertSystem.cs

@@ -21 +21 @@
 
 using System;
+using System.Collections.Generic;
 using System.ComponentModel;
-using System.Collections.Generic;
 using System.Drawing;
+using System.IO;
 using System.Linq;
+using System.Threading.Tasks;
 using HeuristicLab.Analysis;
+using HeuristicLab.Common;
+using HeuristicLab.Common.Resources;
 using HeuristicLab.Core;
+using HeuristicLab.Data;
+using HeuristicLab.MainForm;
 using HeuristicLab.Optimization;
 using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
-using HeuristicLab.Common;
-using HeuristicLab.Common.Resources;
-using HeuristicLab.Data;
+using HeuristicLab.Persistence.Default.Xml;
 
 namespace HeuristicLab.OptimizationExpertSystem {
@@ -222 +226 @@
         }
       } finally { ProblemInstances.UpdateOfRunsInProgress = false; }
+    }
+
+    private static readonly HashSet<string> InterestingValueNames = new HashSet<string>() {
+      "QualityPerEvaluations", "Problem Name", "Problem Type", "Algorithm Name", "Algorithm Type", "Maximization"
+    };
+
+    public async void UpdateKnowledgeBaseAsync(IProgress progress) {
+      progress.Start("Updating Knowledge Base from OKB");
+      await Task.Factory.StartNew(() => { DoUpdateKnowledgeBase(progress); }, TaskCreationOptions.LongRunning);
+    }
+
+    public void UpdateKnowledgeBase() {
+      DoUpdateKnowledgeBase(new Progress(string.Empty, ProgressState.Started));
+    }
+
+    private void DoUpdateKnowledgeBase(IProgress progress) {
+      var queryClient = Clients.OKB.Query.QueryClient.Instance;
+      var adminClient = Clients.OKB.Administration.AdministrationClient.Instance;
+      try {
+        progress.Status = "Refreshing query client...";
+        queryClient.Refresh();
+        var interestingValues = queryClient.ValueNames.Where(x => InterestingValueNames.Contains(x.Name)).ToList();
+
+        var problemTypeFilter = (Clients.OKB.Query.StringComparisonAvailableValuesFilter)queryClient.Filters.Single(x => x.Label == "Problem Data Type Name");
+        problemTypeFilter.Value = Problem.GetType().Name;
+
+        progress.Status = "Obtaining number of runs...";
+        var count = queryClient.GetNumberOfRuns(problemTypeFilter);
+        if (count == 0) return;
+
+        var runIds = queryClient.GetRunIds(problemTypeFilter).ToList();
+        adminClient.Refresh();
+        var i = 0;
+        var conversions = new List<Task>();
+        var runGroup = new Dictionary<Tuple<long, string>, List<IRun>>();
+        while (i < count) {
+          var nextIds = runIds.Skip(i).Take(500).ToList();
+          progress.Status = string.Format("Downloading runs {0} to {1} of {2}...", i, i + nextIds.Count, count);
+          var okbRuns = queryClient.GetRunsWithValues(nextIds, true, interestingValues);
+          conversions.Add(Task.Factory.StartNew(() => {
+            var hlRuns = okbRuns.AsParallel().Select(x => new { Id = x.Algorithm.Id, Name = x.Algorithm.Name, Run = queryClient.ConvertToOptimizationRun(x) })
+                                .GroupBy(x => Tuple.Create(x.Id, x.Name)).ToList();
+            lock (runGroup) {
+              foreach (var hr in hlRuns) {
+                List<IRun> runList;
+                if (!runGroup.TryGetValue(hr.Key, out runList)) {
+                  runList = new List<IRun>();
+                  runGroup[hr.Key] = runList;
+                }
+                runList.AddRange(hr.Select(x => x.Run));
+              }
+            }
+          }));
+          i += nextIds.Count;
+          progress.ProgressValue = 0.8 * (i / (double)count);
+        }
+        Task.WaitAll(conversions.ToArray());
+        var list = new ItemList<IAlgorithm>();
+        i = 0;
+        foreach (var rGroup in runGroup) {
+          progress.Status = string.Format("Downloading algorithm {0}...", rGroup.Key.Item2);
+          var data = Clients.OKB.Administration.AdministrationClient.GetAlgorithmData(rGroup.Key.Item1);
+          if (data != null) {
+            using (var stream = new MemoryStream(data)) {
+              try {
+                var alg = (IAlgorithm)XmlParser.Deserialize<IContent>(stream);
+                alg.Runs.AddRange(rGroup.Value);
+                list.Add(alg);
+              } catch (Exception) { }
+              stream.Close();
+            }
+          }
+          i++;
+          progress.ProgressValue = 0.8 + 0.2 * (i / (double)runGroup.Count);
+        }
+        AlgorithmInstances = list;
+      } finally { progress.Finish(); }
     }