Changeset 14517 for trunk/sources/HeuristicLab.Optimization/3.3/Algorithms/BasicAlgorithm.cs

Timestamp:

12/22/16 10:07:46 (7 years ago)

Author:

jkarder

Message:

#2524: made BasicAlgorithm pausable

File:

• trunk/sources/HeuristicLab.Optimization/3.3/Algorithms/BasicAlgorithm.cs (modified) (9 diffs)

trunk/sources/HeuristicLab.Optimization/3.3/Algorithms/BasicAlgorithm.cs

@@ -24 +24 @@
 using System.Threading.Tasks;
 using HeuristicLab.Common;
+using HeuristicLab.Core;
 using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
 
@@ -29 +30 @@
   [StorableClass]
   public abstract class BasicAlgorithm : Algorithm, IStorableContent {
+    private bool initialized;
+    private bool pausePending;
+    private DateTime lastUpdateTime;
+    protected bool pausable;
+
     public string Filename { get; set; }
+    public bool Pausable { get { return pausable; } }
 
     [Storable]
@@ -47 +54 @@
     protected BasicAlgorithm(BasicAlgorithm original, Cloner cloner)
       : base(original, cloner) {
+      pausable = original.pausable;
       results = cloner.Clone(original.Results);
     }
@@ -58 +66 @@
       base.Prepare();
       results.Clear();
+      initialized = false;
       OnPrepared();
     }
@@ -64 +73 @@
       base.Start();
       CancellationTokenSource = new CancellationTokenSource();
-
+      pausePending = false;
       OnStarted();
       Task task = Task.Factory.StartNew(Run, cancellationTokenSource.Token, cancellationTokenSource.Token);
@@ -80 +89 @@
         CancellationTokenSource.Dispose();
         CancellationTokenSource = null;
-        OnStopped();
+        if (pausePending) OnPaused();
+        else OnStopped();
       });
     }
 
     public override void Pause() {
-      throw new NotSupportedException("Pause is not supported in basic algorithms.");
+      if (!Pausable)
+        throw new NotSupportedException("Pause is not supported by this algorithm.");
+
+      base.Pause();
+      pausePending = true;
+      cancellationTokenSource.Cancel();
     }
 
@@ -92 +107 @@
       // alternatively check the IsCancellationRequested property of the cancellation token
       base.Stop();
-      CancellationTokenSource.Cancel();
+      if (ExecutionState == ExecutionState.Paused) OnStopped();
+      else CancellationTokenSource.Cancel();
     }
 
-
-    private DateTime lastUpdateTime;
     private void Run(object state) {
       CancellationToken cancellationToken = (CancellationToken)state;
@@ -105 +119 @@
       timer.Start();
       try {
+        if (!initialized)
+          Initialize(cancellationToken);
+        initialized = true;
         Run(cancellationToken);
       } finally {
@@ -113 +130 @@
     }
 
+    protected virtual void Initialize(CancellationToken cancellationToken) { }
     protected abstract void Run(CancellationToken cancellationToken);