change input media to be an enum over string

Author: Linnea May

Samples/StyleTransfer/AppModel.cs

@@ -16,6 +16,12 @@
 
 namespace StyleTransfer
 {
+    enum InputMediaType
+    {
+        LiveStream,
+        AcquireImage,
+        FilePick
+    }
     class AppModel : INotifyPropertyChanged
     {
         public AppModel()
@@ -51,8 +57,8 @@ public string ModelSource
         }
 
         // Type of input media to use
-        private string _inputMedia;
-        public string InputMedia
+        private InputMediaType _inputMedia;
+        public InputMediaType InputMedia
         {
             get { return _inputMedia; }
             set
@@ -137,7 +143,7 @@ public bool StreamingControlsVisible
         {
             get
             {
-                return _inputMedia == "LiveStream";
+                return _inputMedia == InputMediaType.LiveStream;
             }
         }
 

Samples/StyleTransfer/AppViewModel.cs

@@ -65,7 +65,7 @@ public AppViewModel()
         {
             _appModel = new AppModel();
             SaveCommand = new RelayCommand(SaveOutput);
-            SetMediaSourceCommand = new RelayCommand<string>(async (x) => await SetMediaSource(x));
+            SetMediaSourceCommand = new RelayCommand<InputMediaType>(async (x) => await SetMediaSource(x));
             SetModelSourceCommand = new RelayCommand(async () => await SetModelSource());
             ChangeLiveStreamCommand = new RelayCommand(async () => await ChangeLiveStream());
 
@@ -130,9 +130,9 @@ public bool UseGpu
             set
             {
                 _useGpu = value;
-                if (_appModel.InputMedia == "LiveStream")
+                if (_appModel.InputMedia == InputMediaType.LiveStream)
                 {
-                    SetMediaSourceCommand.Execute("LiveStream");
+                    SetMediaSourceCommand.Execute(InputMediaType.LiveStream);
                 }
                 OnPropertyChanged();
             }
@@ -209,13 +209,13 @@ public string StatusMessage
         public async void SaveOutput()
         {
             Debug.WriteLine("UIButtonSaveImage_Click");
-            if (_appModel.InputMedia != "LiveStream")
+            if (_appModel.InputMedia != InputMediaType.LiveStream)
             {
                 await ImageHelper.SaveVideoFrameToFilePickedAsync(_appModel.OutputFrame);
             }
         }
 
-        public async Task SetMediaSource(string src)
+        public async Task SetMediaSource(InputMediaType src)
         {
             _appModel.InputMedia = src;
             await LoadModelAsync();
@@ -227,13 +227,13 @@ public async Task SetMediaSource(string src)
 
             switch (_appModel.InputMedia)
             {
-                case "LiveStream":
+                case InputMediaType.LiveStream:
                     await StartLiveStream();
                     break;
-                case "AcquireImage":
+                case InputMediaType.AcquireImage:
                     await StartAcquireImage();
                     break;
-                case "FilePick":
+                case InputMediaType.FilePick:
                     await StartFilePick();
                     break;
                 default:
@@ -249,13 +249,13 @@ public async Task SetModelSource()
 
             switch (_appModel.InputMedia)
             {
-                case "LiveStream":
+                case InputMediaType.LiveStream:
                     await ChangeLiveStream();
                     break;
-                case "AcquireImage":
+                case InputMediaType.AcquireImage:
                     await ChangeImage();
                     break;
-                case "FilePick":
+                case InputMediaType.FilePick:
                     await ChangeImage();
                     break;
                 default:

Samples/StyleTransfer/MainPage.xaml

@@ -140,40 +140,36 @@
           <Button
               Name="UIButtonLiveStream"
               Command="{Binding SetMediaSourceCommand}"
-              CommandParameter="LiveStream"
               ToolTipService.ToolTip="Camera preview">
             <Button.Content>
               <SymbolIcon Symbol="Video" />
             </Button.Content>
+            <Button.CommandParameter>
+              <local:InputMediaType>LiveStream</local:InputMediaType>
+            </Button.CommandParameter>
           </Button>
           <Button
               Name="UIButtonAcquireImage"
               Command="{Binding SetMediaSourceCommand}"
-              CommandParameter="AcquireImage"
               ToolTipService.ToolTip="Take a photo">
             <Button.Content>
               <SymbolIcon Symbol="Camera" />
             </Button.Content>
+            <Button.CommandParameter>
+              <local:InputMediaType>AcquireImage</local:InputMediaType>
+            </Button.CommandParameter>
           </Button>
           <Button
               Name="UIButtonFilePick"
               Command="{Binding SetMediaSourceCommand}"
-              CommandParameter="FilePick"
               ToolTipService.ToolTip="Select an image from a file">
             <Button.Content>
               <SymbolIcon Symbol="OpenFile" />
             </Button.Content>
+            <Button.CommandParameter>
+              <local:InputMediaType>FilePick</local:InputMediaType>
+            </Button.CommandParameter>
           </Button>
-          <Button
-              Name="UIButtonInking"
-              Command="{Binding SetMediaSourceCommand}"
-              CommandParameter="Inking"
-              ToolTipService.ToolTip="Draw on a canvas">
-            <Button.Content>
-              <SymbolIcon Symbol="Edit" />
-            </Button.Content>
-          </Button>
-
         </StackPanel>
 
         <!--  Inference device  -->