Changeset 13814 for branches/HeuristicLab.VariableInteractionNetworks/HeuristicLab.VariableInteractionNetworks/3.3/CreateTargetVariationExperimentDialog.cs

Timestamp:

04/28/16 15:58:06 (8 years ago)

Author:

bburlacu

Message:

#2288: Improve the CreateTargetVariationExperimentDialog

File:

• branches/HeuristicLab.VariableInteractionNetworks/HeuristicLab.VariableInteractionNetworks/3.3/CreateTargetVariationExperimentDialog.cs (modified) (8 diffs)

branches/HeuristicLab.VariableInteractionNetworks/HeuristicLab.VariableInteractionNetworks/3.3/CreateTargetVariationExperimentDialog.cs

@@ -22 +22 @@
 using System;
 using System.Collections.Generic;
+using System.ComponentModel;
 using System.Linq;
 using System.Windows.Forms;
@@ -32 +33 @@
   public partial class CreateTargetVariationExperimentDialog : Form {
     public Experiment Experiment { get; private set; }
-    private int repetitions;
-    private int combinationGroupSize;
 
     public CreateTargetVariationExperimentDialog() : this(null) { }
     public CreateTargetVariationExperimentDialog(IAlgorithm alg) {
       InitializeComponent();
-      repetitions = (int)repetitionsNumericUpDown.Value;
-      combinationGroupSize = (int)combinationGroupSizeNumericUpDown.Value;
 
       var problem = alg.Problem as IDataAnalysisProblem;
-      var inputCount = problem.ProblemData.AllowedInputVariables.Count();
-      if (combinationGroupSize > inputCount)
-        combinationGroupSize = inputCount;
-      warningLabel.Text = string.Format("Warning: this will create {0} combinations.", EnumerableExtensions.BinomialCoefficient(inputCount, combinationGroupSize));
+      var inputCount = problem.ProblemData.AllowedInputVariables.Count() - 1; // -1 because one will be the target
+      if (combinationGroupSizeNumericUpDown.Value > inputCount)
+        combinationGroupSizeNumericUpDown.Value = inputCount;
+      var combinationGroupSize = (int)combinationGroupSizeNumericUpDown.Value;
+      combinationCountLabel.Text = ((inputCount + 1) * EnumerableExtensions.BinomialCoefficient(inputCount, combinationGroupSize)).ToString();
 
       Experiment = null;
@@ -51 +49 @@
     }
 
-    public Experiment CreateVariableCombinations(IAlgorithm algorithm) {
-      var experiment = new Experiment("Target Variation Experiment");
-
+    public static IEnumerable<IOptimizer> CreateVariableCombinations(IAlgorithm algorithm, int combinationGroupSize, int repetitions) {
       if (algorithm == null)
         throw new ArgumentNullException("The algorithm parameter must be a valid IAlgorithm instance.");
@@ -61 +57 @@
 
       var variables = problemData.AllowedInputVariables.ToList();
-
-      for (int i = 0; i < variables.Count; i++) {
-        var target = variables[i];
+      int count = 1;
+      foreach (var target in variables) {
         var inputs = variables.Where(x => x != target).ToList();
-        if (combinationGroupSize > inputs.Count)
-          combinationGroupSize = inputs.Count;
-
         var combinations = inputs.Combinations(combinationGroupSize);
 
@@ -81 +73 @@
           SetTargetName(problemData, target);
 
-          var batchrun = new BatchRun(string.Format("Batchrun {0}", i + 1)) {
+          var batchRun = new BatchRun(string.Format("Batchrun {0}", count++)) {
             Repetitions = repetitions,
             Optimizer = alg
           };
-
-          experiment.Optimizers.Add(batchrun);
+          yield return batchRun;
         }
       }
-      return experiment;
     }
 
@@ -117 +107 @@
 
     #region events
-    private void okButton_Click(object sender, EventArgs e) {
-      IContentView activeView = MainFormManager.MainForm.ActiveView as IContentView;
-      if ((activeView != null) && (activeView.Content != null) && (activeView.Content is IAlgorithm) && !activeView.Locked) {
-        Experiment = CreateVariableCombinations((IAlgorithm)activeView.Content);
-        DialogResult = DialogResult.OK;
-        Close();
-      }
+    private void CreateTargetVariationExperimentDialog_FormClosing(object sender, FormClosingEventArgs e) {
+      if (worker.IsBusy) {
+        if (DialogResult != DialogResult.OK) {
+          if (worker.IsBusy) worker.CancelAsync();
+        }
+        e.Cancel = true;
+      }
+    }
+
+    private void repetitionsNumericUpDown_Validated(object sender, EventArgs e) {
+      if (repetitionsNumericUpDown.Text == string.Empty)
+        repetitionsNumericUpDown.Text = repetitionsNumericUpDown.Value.ToString();
     }
 
@@ -132 +127 @@
       var problem = algorithm.Problem as IDataAnalysisProblem;
       if (problem == null) return;
-      var inputCount = problem.ProblemData.AllowedInputVariables.Count();
+      var inputCount = problem.ProblemData.AllowedInputVariables.Count() - 1;
       if (combinationGroupSizeNumericUpDown.Value > inputCount)
         combinationGroupSizeNumericUpDown.Value = inputCount;
-    }
-
-    private void repetitionsNumericUpDown_Validated(object sender, EventArgs e) {
-      if (repetitionsNumericUpDown.Text == string.Empty)
-        repetitionsNumericUpDown.Text = repetitionsNumericUpDown.Value.ToString();
-      repetitions = (int)repetitionsNumericUpDown.Value;
     }
 
@@ -146 +135 @@
       if (combinationGroupSizeNumericUpDown.Text == string.Empty)
         combinationGroupSizeNumericUpDown.Text = combinationGroupSizeNumericUpDown.Value.ToString();
-      combinationGroupSize = (int)combinationGroupSizeNumericUpDown.Value;
+
       IContentView activeView = MainFormManager.MainForm.ActiveView as IContentView;
       if (activeView == null) return;
       var algorithm = (IAlgorithm)activeView.Content;
       var problem = (IDataAnalysisProblem)algorithm.Problem;
-      var inputCount = problem.ProblemData.AllowedInputVariables.Count();
-      warningLabel.Text = string.Format("Warning: this will create {0} combinations.", EnumerableExtensions.BinomialCoefficient(inputCount, combinationGroupSize));
+      var inputCount = problem.ProblemData.AllowedInputVariables.Count() - 1;
+      if (combinationGroupSizeNumericUpDown.Value > inputCount)
+        combinationGroupSizeNumericUpDown.Value = inputCount;
+      combinationCountLabel.Text = ((inputCount + 1) * EnumerableExtensions.BinomialCoefficient(inputCount, (int)combinationGroupSizeNumericUpDown.Value)).ToString();
+    }
+
+    private void worker_DoWork(object sender, DoWorkEventArgs e) {
+      IContentView activeView = (IContentView)MainFormManager.MainForm.ActiveView;
+      var algorithm = (IAlgorithm)activeView.Content;
+      var problem = (IDataAnalysisProblem)algorithm.Problem;
+      var experiment = new Experiment("Target Variation Experiment");
+      var repetitions = (int)repetitionsNumericUpDown.Value;
+      var inputCount = problem.ProblemData.AllowedInputVariables.Count() - 1;
+      var combinationGroupSize = (int)combinationGroupSizeNumericUpDown.Value;
+      var totalNumberOfCombinations = (inputCount + 1) * EnumerableExtensions.BinomialCoefficient(inputCount, combinationGroupSize);
+      int progress = 0;
+      foreach (var optimizer in CreateVariableCombinations(algorithm, combinationGroupSize, repetitions)) {
+        experiment.Optimizers.Add(optimizer);
+        progress++;
+        worker.ReportProgress((int)Math.Round(100d * progress / totalNumberOfCombinations));
+      }
+      Experiment = experiment;
+    }
+
+    private void worker_ProgressChanged(object sender, ProgressChangedEventArgs e) {
+      if (InvokeRequired) {
+        Invoke((Action<int>)UpdateProgress, e.ProgressPercentage);
+      } else {
+        UpdateProgress(e.ProgressPercentage);
+      }
+    }
+
+    private void UpdateProgress(int percentage) {
+      progressBar.Value = percentage;
+    }
+
+    private void worker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e) {
+      if (e.Cancelled || e.Error != null)
+        return;
+      DialogResult = DialogResult.OK;
+      Close();
+    }
+
+    private void okButton_Click(object sender, EventArgs args) {
+      IContentView activeView = MainFormManager.MainForm.ActiveView as IContentView;
+      if ((activeView != null) && (activeView.Content != null) && (activeView.Content is IAlgorithm) && !activeView.Locked) {
+        okButton.Enabled = false;
+        combinationsLabel.Visible = false;
+        combinationCountLabel.Visible = false;
+        progressBar.Visible = true;
+        worker.RunWorkerAsync();
+      }
+    }
+
+    private void cancelButton_Click(object sender, EventArgs e) {
+      if (worker.IsBusy)
+        worker.CancelAsync();
     }
     #endregion