Merge pull request #11 from Toyz/master

Added the option to set tick value when calling Tick()
Mpdreamz · Sep 21, 2017 · 66bb030 · 66bb030
2 parents 2637e72 + e0bf557
commit 66bb030
Showing 1 changed file with 24 additions and 12 deletions.
diff --git a/src/ShellProgressBar/ProgressBarBase.cs b/src/ShellProgressBar/ProgressBarBase.cs
@@ -64,28 +64,40 @@ protected virtual void OnDone()
 
 		public void Tick(string message = null)
 		{
-			Interlocked.Increment(ref _currentTick);
-			if (message != null)
-				Interlocked.Exchange(ref _message, message);
-
-			if (_currentTick >= _maxTicks)
-			{
-				this.EndTime = DateTime.Now;
-				this.OnDone();
-			}
-			DisplayProgress();
+            FinishTick(message);
 		}
 
-		public void UpdateMaxTicks(int maxTicks)
+        public void Tick(string message, int prec)
+        {
+            Interlocked.Exchange(ref _currentTick, prec);
+
+            FinishTick(message);
+        }
+
+        public void UpdateMaxTicks(int maxTicks)
 		{
 			Interlocked.Exchange(ref _maxTicks, maxTicks);
 		}
 
 		public void UpdateMessage(string message)
 		{
-			Interlocked.Exchange(ref _message, message);
+            Interlocked.Exchange(ref _message, message);
 
 			DisplayProgress();
 		}
+
+        private void FinishTick(string message)
+        {
+            Interlocked.Increment(ref _currentTick);
+            if (message != null)
+                Interlocked.Exchange(ref _message, message);
+
+            if (_currentTick >= _maxTicks)
+            {
+                this.EndTime = DateTime.Now;
+                this.OnDone();
+            }
+            DisplayProgress();
+        }
 	}
 }