Add Repeat button

nguyenquyhy · Feb 22, 2022 · ad41c62 · ad41c62
1 parent 369c992
commit ad41c62
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 4 deletions.
diff --git a/FlightRecorder.Client.Logics/IReplayLogic.cs b/FlightRecorder.Client.Logics/IReplayLogic.cs
@@ -18,6 +18,7 @@ public interface IReplayLogic
         void Seek(int value);
         bool StopReplay();
         void ChangeRate(double rate);
+        void SetRepeat(bool repeat);
         void Unfreeze();
         void NotifyPosition(AircraftPositionStruct? value);
 

diff --git a/FlightRecorder.Client.Logics/ReplayLogic.cs b/FlightRecorder.Client.Logics/ReplayLogic.cs
@@ -34,6 +34,7 @@ public class ReplayLogic : IReplayLogic, IDisposable
         private int currentFrame;
 
         private double rate = 1;
+        private bool repeat = false;
         private int? pausedFrame;
         private double? pausedRate;
         private bool isReplayStopping;
@@ -237,6 +238,11 @@ public void ChangeRate(double rate)
             this.rate = rate;
         }
 
+        public void SetRepeat(bool repeat)
+        {
+            this.repeat = repeat;
+        }
+
         public void Unfreeze()
         {
             if (replayMilliseconds != null)
@@ -338,7 +344,7 @@ private async Task RunReplay()
 
                 if (isReplayStopping)
                 {
-                    FinishReplay();
+                    FinishReplay(false);
                     return;
                 }
 
@@ -396,7 +402,7 @@ private async Task RunReplay()
                         else
                         {
                             // Last frame
-                            FinishReplay();
+                            FinishReplay(true);
                             return;
                         }
                     }
@@ -414,7 +420,7 @@ private async Task RunReplay()
             }
         }
 
-        private void FinishReplay()
+        private void FinishReplay(bool reachedLastFrame)
         {
             logger.LogInformation("Replay finished.");
 
@@ -443,7 +449,14 @@ private void FinishReplay()
             replayMilliseconds = null;
             offsetStartMilliseconds = null;
 
-            ReplayFinished?.Invoke(this, new EventArgs());
+            if (reachedLastFrame && repeat)
+            {
+                Replay();
+            }
+            else
+            {
+                ReplayFinished?.Invoke(this, new EventArgs());
+            }
         }
 
         private void MoveAircraft(long nextElapsed, AircraftPositionStruct position, long? lastElapsed, AircraftPositionStruct? lastPosition, long currentElapsed)

diff --git a/FlightRecorder.Client/AIWindow.xaml b/FlightRecorder.Client/AIWindow.xaml
@@ -45,6 +45,7 @@
                 </Grid.ColumnDefinitions>
 
                 <StackPanel Orientation="Horizontal" Grid.Column="0">
+                    <ToggleButton x:Name="ButtonRepeat" Padding="5,0" Click="ButtonRepeat_Click">Repeat</ToggleButton>
                     <Button x:Name="ButtonSpeed" Content="x1" Padding="5,0" Margin="0,0,5,0" ContextMenuService.Placement="Bottom"
                             IsEnabled="{Binding State, Converter={StaticResource ValueToTrueConverter}, ConverterParameter=IdleSaved|IdleUnsaved|PausingSaved|PausingUnsaved}"
                             Click="ButtonContext_Click">

diff --git a/FlightRecorder.Client/BaseWindow.cs b/FlightRecorder.Client/BaseWindow.cs
@@ -2,6 +2,7 @@
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Controls;
+using System.Windows.Controls.Primitives;
 
 namespace FlightRecorder.Client
 {
@@ -99,6 +100,14 @@ protected void Slider_MouseWheel(object sender, System.Windows.Input.MouseWheelE
             }
         }
 
+        protected void ButtonRepeat_Click(object sender, RoutedEventArgs e)
+        {
+            if (sender is ToggleButton buttonRepeat)
+            {
+                replayLogic.SetRepeat(buttonRepeat.IsChecked == true);
+            }
+        }
+
         #endregion
     }
 }
diff --git a/FlightRecorder.Client/MainWindow.xaml b/FlightRecorder.Client/MainWindow.xaml
@@ -62,6 +62,7 @@
                 </Grid.ColumnDefinitions>
 
                 <StackPanel Orientation="Horizontal" Grid.Column="0">
+                    <ToggleButton x:Name="ButtonRepeat" Padding="5,0" Click="ButtonRepeat_Click">Repeat</ToggleButton>
                     <Button x:Name="ButtonSpeed" Content="x1" Padding="5,0" Margin="0,0,5,0" ContextMenuService.Placement="Bottom"
                             IsEnabled="{Binding State, Converter={StaticResource ValueToTrueConverter}, ConverterParameter=IdleSaved|IdleUnsaved|PausingSaved|PausingUnsaved}"
                             Click="ButtonContext_Click">