From ad1682b0a7eb2fe2971ee5a744fee21f250ec53b Mon Sep 17 00:00:00 2001
From: tmokmss <tomookam@live.jp>
Date: Sat, 20 Oct 2018 01:46:38 +0900
Subject: [PATCH] add selectors to choose which game mode to convert

---
 Osu2Saber/Model/BatchProcessor.cs           | 13 +++++++++----
 Osu2Saber/Model/OszProcessor.cs             |  3 +--
 Osu2Saber/View/ConfigPanel.xaml             |  4 ++++
 Osu2Saber/ViewModel/ConfigPanelViewModel.cs | 20 +++++++++++++++++++-
 4 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/Osu2Saber/Model/BatchProcessor.cs b/Osu2Saber/Model/BatchProcessor.cs
index fc4fe60..99083fb 100644
--- a/Osu2Saber/Model/BatchProcessor.cs
+++ b/Osu2Saber/Model/BatchProcessor.cs
@@ -8,6 +8,10 @@ namespace Osu2Saber.Model
 {
     class BatchProcessor : BindableBase
     {
+        public static bool IncludeTaiko { set; get; } = false;
+        public static bool IncludeCtB { set; get; } = true;
+        public static bool IncludeMania { set; get; } = true;
+
         object progressLock = new object();
         double progress;
         Logger logger;
@@ -75,10 +79,11 @@ Osu2BsConverter ConvertBeatmap(OszProcessor oszp)
             OutputDir = Osu2BsConverter.WorkDir;
             var o2b = new Osu2BsConverter(oszp.OutDir, oszp.OszName);
 
-            foreach (var osufile in oszp.LoadOsuFiles())
-            {
-                o2b.AddBeatmap(osufile);
-            }
+            oszp.LoadOsuFiles()
+                .Where(osuFile => IncludeTaiko || osuFile.Mode != 1)
+                .Where(osuFile => IncludeCtB || osuFile.Mode != 2)
+                .Where(osuFile => IncludeMania || osuFile.Mode != 3)
+                .ToList().ForEach(osufile => o2b.AddBeatmap(osufile));
 
             o2b.ProcessAll();
 
diff --git a/Osu2Saber/Model/OszProcessor.cs b/Osu2Saber/Model/OszProcessor.cs
index 53ac03e..5caaf10 100644
--- a/Osu2Saber/Model/OszProcessor.cs
+++ b/Osu2Saber/Model/OszProcessor.cs
@@ -78,8 +78,7 @@ public IEnumerable<Beatmap> LoadOsuFiles()
         {
             var beatmaps = OsuFiles
                 .Select(file => LoadOsuFile(file))
-                .Where(map => map != null)
-                .Where(map => map.Mode != 1); // exclude taiko map
+                .Where(map => map != null);
             return beatmaps;
         }
     }
diff --git a/Osu2Saber/View/ConfigPanel.xaml b/Osu2Saber/View/ConfigPanel.xaml
index 5c7ae67..069c375 100644
--- a/Osu2Saber/View/ConfigPanel.xaml
+++ b/Osu2Saber/View/ConfigPanel.xaml
@@ -23,6 +23,10 @@
         <CheckBox IsChecked="{Binding PreferHarder}" Content="Prefer harder maps" HorizontalAlignment="Left" Margin="10,103,0,0" VerticalAlignment="Top" ToolTip="Prioritize harder beatmaps when there're more than 5 beatmaps"/>
         <CheckBox IsChecked="{Binding HandleHitSlider}" Content="Don't ignore Hit Slider" HorizontalAlignment="Left" Margin="10,134,0,0" VerticalAlignment="Top" ToolTip="By default, hit sliders are ignored because they make a map too difficult. However, some maps get better with them. Just try it!"/>
         <CheckBox IsChecked="{Binding NoDirectionAndPlacement}" Content="No cut-direction &amp; scattared-placement" HorizontalAlignment="Left" Margin="10,164,0,0" VerticalAlignment="Top" ToolTip="For mappers, this option generates only notes without setting any cut direction or scattared placement"/>
+        <Label Content="Includes:" HorizontalAlignment="Left" Margin="10,184,0,0" VerticalAlignment="Top" Width="102" Height="27"/>
+        <CheckBox IsChecked="{Binding IncludeTaiko}" Content="Taiko" HorizontalAlignment="Left" Margin="70,190,0,0" VerticalAlignment="Top"/>
+        <CheckBox IsChecked="{Binding IncludeCtB}" Content="CtB" HorizontalAlignment="Left" Margin="128,190,0,0" VerticalAlignment="Top"/>
+        <CheckBox IsChecked="{Binding IncludeMania}" Content="mania" HorizontalAlignment="Left" Margin="178,190,0,0" VerticalAlignment="Top"/>
 
     </Grid>
 </UserControl>
diff --git a/Osu2Saber/ViewModel/ConfigPanelViewModel.cs b/Osu2Saber/ViewModel/ConfigPanelViewModel.cs
index 6befa56..66607d3 100644
--- a/Osu2Saber/ViewModel/ConfigPanelViewModel.cs
+++ b/Osu2Saber/ViewModel/ConfigPanelViewModel.cs
@@ -30,7 +30,7 @@ public double MinimumDifficulty
             }
             get => Osu2BsConverter.MinimumDifficulty;
         }
-        
+
         public bool PreferHarder
         {
             set { Osu2BsConverter.PreferHarder = value; }
@@ -48,5 +48,23 @@ public bool NoDirectionAndPlacement
             set { ConvertAlgorithm.NoDirectionAndPlacement = value; }
             get => ConvertAlgorithm.NoDirectionAndPlacement;
         }
+
+        public bool IncludeTaiko
+        {
+            set { BatchProcessor.IncludeTaiko = value; }
+            get => BatchProcessor.IncludeTaiko;
+        }
+
+        public bool IncludeCtB
+        {
+            set { BatchProcessor.IncludeCtB = value; }
+            get => BatchProcessor.IncludeCtB;
+        }
+
+        public bool IncludeMania
+        {
+            set { BatchProcessor.IncludeMania = value; }
+            get => BatchProcessor.IncludeMania;
+        }
     }
 }