From 43218230eb87fbad953c804016d7eafab3920075 Mon Sep 17 00:00:00 2001
From: Grigory Chudov <gchudov@gmail.com>
Date: Sat, 24 Feb 2018 11:32:35 -0500
Subject: [PATCH] eac3to gui: audio extraction working

---
 CUETools.Codecs.BDLPCM/MPLSReader.cs   |   6 +
 CUETools.eac3ui/CUETools.eac3ui.csproj |   4 +
 CUETools.eac3ui/MainWindow.xaml        | 120 ++++++++++---------
 CUETools.eac3ui/MainWindow.xaml.cs     | 156 +++++++++++++++++++++++--
 4 files changed, 218 insertions(+), 68 deletions(-)

diff --git a/CUETools.Codecs.BDLPCM/MPLSReader.cs b/CUETools.Codecs.BDLPCM/MPLSReader.cs
index bcba9b4..541adde 100644
--- a/CUETools.Codecs.BDLPCM/MPLSReader.cs
+++ b/CUETools.Codecs.BDLPCM/MPLSReader.cs
@@ -9,6 +9,12 @@ namespace CUETools.Codecs.BDLPCM
     [AudioDecoderClass("cuetools", "mpls", 2)]
     public class MPLSReader : IAudioSource
     {
+        public unsafe MPLSReader(string path, Stream IO, ushort pid)
+                : this(path, IO)
+        {
+            settings.Pid = pid;
+        }
+
         public unsafe MPLSReader(string path, Stream IO)
         {
             settings = new BDLPCMReaderSettings();
diff --git a/CUETools.eac3ui/CUETools.eac3ui.csproj b/CUETools.eac3ui/CUETools.eac3ui.csproj
index 0358cf8..95d581e 100644
--- a/CUETools.eac3ui/CUETools.eac3ui.csproj
+++ b/CUETools.eac3ui/CUETools.eac3ui.csproj
@@ -103,6 +103,10 @@
       <Project>{e75f7ccd-4266-42e1-a039-dc7eb5edd8f6}</Project>
       <Name>CUETools.Codecs.BDLPCM</Name>
     </ProjectReference>
+    <ProjectReference Include="..\CUETools.Codecs.FLAKE\CUETools.Codecs.FLAKE.csproj">
+      <Project>{082d6b9e-326e-4d15-9798-edae9ede70a6}</Project>
+      <Name>CUETools.Codecs.FLAKE</Name>
+    </ProjectReference>
     <ProjectReference Include="..\CUETools.Codecs\CUETools.Codecs.csproj">
       <Project>{6458a13a-30ef-45a9-9d58-e5031b17bee2}</Project>
       <Name>CUETools.Codecs</Name>
diff --git a/CUETools.eac3ui/MainWindow.xaml b/CUETools.eac3ui/MainWindow.xaml
index 948f4ad..6021275 100644
--- a/CUETools.eac3ui/MainWindow.xaml
+++ b/CUETools.eac3ui/MainWindow.xaml
@@ -7,65 +7,69 @@
         mc:Ignorable="d"
         MinWidth="300" MaxWidth="1200" MinHeight="300" MaxHeight="300"
         Title="CUETools eac3to" Height="300" Width="600" Closing="Window_Closing">
-	<StackPanel>
+    <StackPanel>
+        <StackPanel x:Name="stackParams">
+            <DockPanel>
+                <Label x:Name="labelSource" Content="Source" Margin="8" Width="80"/>
+                <Button x:Name="buttonBrowseSource" Content="Browse" Click="ButtonBrowseSource_Click" DockPanel.Dock="Right" Margin="8"/>
+                <TextBox x:Name="textBoxSource" TextWrapping="NoWrap" Margin="8" TextChanged="textBoxSource_TextChanged"/>
+            </DockPanel>
+            <DockPanel>
+                <Label x:Name="labelDestination" Content="Destination" Margin="8" Width="80"/>
+                <Button x:Name="buttonBrowseDestination" Content="Browse" Click="buttonBrowseDestination_Click" DockPanel.Dock="Right" Margin="8"/>
+                <TextBox x:Name="textBoxDestination" TextWrapping="NoWrap" Margin="8"/>
+            </DockPanel>
+            <DockPanel>
+                <Label x:Name="labelTitleSet" Content="Title Set" Margin="8" Width="80"/>
+                <ComboBox x:Name="cmbTitleSet" Margin="8" SelectionChanged="cmbTitleSet_SelectionChanged">
+                    <ComboBox.ItemTemplate>
+                        <DataTemplate>
+                            <StackPanel Orientation="Horizontal">
+                                <Rectangle Fill="{Binding Name}" Width="16" Height="16" Margin="0,2,5,2" />
+                                <TextBlock Text="{Binding FileName}" Margin="10,0" />
+                                <TextBlock Text="{Binding Duration, StringFormat={}({0:hh\\:mm\\:ss})}" />
+                                <TextBlock Text="{Binding Chapters.Count, StringFormat={}{0,2} chapters}" Margin="10,0" />
+                            </StackPanel>
+                        </DataTemplate>
+                    </ComboBox.ItemTemplate>
+                </ComboBox>
+            </DockPanel>
+            <DockPanel>
+                <Label x:Name="labelAudioTrack" Content="Audio Track" Margin="8" Width="80"/>
+                <ComboBox x:Name="cmbAudioTrack" Margin="8" SelectionChanged="cmbAudioTrack_SelectionChanged">
+                    <ComboBox.ItemTemplate>
+                        <DataTemplate>
+                            <StackPanel Orientation="Horizontal">
+                                <Rectangle Fill="{Binding Name}" Width="16" Height="16" Margin="0,2,5,2" />
+                                <TextBlock Text="{Binding CodecString}" Margin="10,0" />
+                                <TextBlock Text="{Binding LanguageString}" Margin="10,0" />
+                                <TextBlock Text="{Binding FormatString}" Margin="10,0" />
+                                <TextBlock Text="{Binding RateString}" Margin="10,0" />
+                            </StackPanel>
+                        </DataTemplate>
+                    </ComboBox.ItemTemplate>
+                </ComboBox>
+            </DockPanel>
+            <DockPanel>
+                <Label x:Name="labelMetadata" Content="Metadata" Margin="8" Width="80"/>
+                <ComboBox x:Name="cmbMetadata" Margin="8">
+                    <ComboBox.ItemTemplate>
+                        <DataTemplate>
+                            <StackPanel Orientation="Horizontal">
+                                <Rectangle Fill="{Binding source}" Width="16" Height="16" Margin="0,2,5,2" />
+                                <TextBlock Text="{Binding artist}" Margin="10,0" />
+                                <TextBlock Text="{Binding year}" Margin="10,0" />
+                                <TextBlock Text="{Binding album}" Margin="10,0" />
+                            </StackPanel>
+                        </DataTemplate>
+                    </ComboBox.ItemTemplate>
+                </ComboBox>
+            </DockPanel>
+        </StackPanel>
         <DockPanel>
-            <Label x:Name="labelSource" Content="Source" Margin="8" Width="80"/>
-            <Button x:Name="buttonBrowseSource" Content="Browse" Click="ButtonBrowseSource_Click" DockPanel.Dock="Right" Margin="8"/>
-            <TextBox x:Name="textBoxSource" TextWrapping="NoWrap" Margin="8" TextChanged="textBoxSource_TextChanged"/>
-		</DockPanel>
-        <DockPanel>
-            <Label x:Name="labelDestination" Content="Destination" Margin="8" Width="80"/>
-            <Button x:Name="buttonBrowseDestination" Content="Browse" Click="buttonBrowseDestination_Click" DockPanel.Dock="Right" Margin="8"/>
-            <TextBox x:Name="textBoxDestination" TextWrapping="NoWrap" Margin="8"/>
-        </DockPanel>
-        <DockPanel>
-            <Label x:Name="labelTitleSet" Content="Title Set" Margin="8" Width="80"/>
-            <ComboBox x:Name="cmbTitleSet" Margin="8" SelectionChanged="cmbTitleSet_SelectionChanged">
-                <ComboBox.ItemTemplate>
-                    <DataTemplate>
-                        <StackPanel Orientation="Horizontal">
-                            <Rectangle Fill="{Binding Name}" Width="16" Height="16" Margin="0,2,5,2" />
-                            <TextBlock Text="{Binding FileName}" Margin="10,0" />
-                            <TextBlock Text="{Binding Duration}" />
-                        </StackPanel>
-                    </DataTemplate>
-                </ComboBox.ItemTemplate>
-            </ComboBox>
-        </DockPanel>
-        <DockPanel>
-            <Label x:Name="labelAudioTrack" Content="Audio Track" Margin="8" Width="80"/>
-            <ComboBox x:Name="cmbAudioTrack" Margin="8" SelectionChanged="cmbAudioTrack_SelectionChanged">
-                <ComboBox.ItemTemplate>
-                    <DataTemplate>
-                        <StackPanel Orientation="Horizontal">
-                            <Rectangle Fill="{Binding Name}" Width="16" Height="16" Margin="0,2,5,2" />
-                            <TextBlock Text="{Binding CodecString}" Margin="10,0" />
-                            <TextBlock Text="{Binding LanguageString}" Margin="10,0" />
-                            <TextBlock Text="{Binding FormatString}" Margin="10,0" />
-                            <TextBlock Text="{Binding RateString}" Margin="10,0" />
-                        </StackPanel>
-                    </DataTemplate>
-                </ComboBox.ItemTemplate>
-            </ComboBox>
-        </DockPanel>
-        <DockPanel>
-            <Label x:Name="labelMetadata" Content="Metadata" Margin="8" Width="80"/>
-            <ComboBox x:Name="cmbMetadata" Margin="8">
-                <ComboBox.ItemTemplate>
-                    <DataTemplate>
-                        <StackPanel Orientation="Horizontal">
-                            <Rectangle Fill="{Binding source}" Width="16" Height="16" Margin="0,2,5,2" />
-                            <TextBlock Text="{Binding artist}" Margin="10,0" />
-                            <TextBlock Text="{Binding year}" Margin="10,0" />
-                            <TextBlock Text="{Binding album}" Margin="10,0" />
-                        </StackPanel>
-                    </DataTemplate>
-                </ComboBox.ItemTemplate>
-            </ComboBox>
-        </DockPanel>
-        <DockPanel>
-            <Button x:Name="buttonExtract" Content="Extract" DockPanel.Dock="Right" Margin="8"/>
-            <Label></Label>
+            <Button x:Name="buttonStop" Content="Stop" DockPanel.Dock="Left" Margin="8" Click="buttonStop_Click" Visibility="Hidden" Width="80"/>
+            <Button x:Name="buttonExtract" Content="Extract" DockPanel.Dock="Right" Margin="8" Click="buttonExtract_Click"/>
+            <ProgressBar Minimum="0" Maximum="100" Name="pbStatus" Margin="8" Visibility="Collapsed"/>
         </DockPanel>
     </StackPanel>
 </Window>
diff --git a/CUETools.eac3ui/MainWindow.xaml.cs b/CUETools.eac3ui/MainWindow.xaml.cs
index 57ac067..7f9f0b2 100644
--- a/CUETools.eac3ui/MainWindow.xaml.cs
+++ b/CUETools.eac3ui/MainWindow.xaml.cs
@@ -16,6 +16,10 @@ using System.Windows.Navigation;
 using CUETools.Codecs.BDLPCM;
 using CUETools.CDImage;
 using CUETools.CTDB;
+using System.ComponentModel;
+using Krystalware.UploadHelper;
+using CUETools.Codecs;
+using CUETools.Codecs.FLAKE;
 //using Microsoft.Win32;
 
 namespace CUETools.eac3ui
@@ -88,6 +92,7 @@ namespace CUETools.eac3ui
         private void cmbTitleSet_SelectionChanged(object sender, SelectionChangedEventArgs e)
         {
             cmbMetadata.ItemsSource = new List<CTDBResponseMeta>();
+            ctdb = null;
 
             var audios = new List<MPLSStream>();
             if (e.AddedItems.Count == 1)
@@ -96,21 +101,50 @@ namespace CUETools.eac3ui
                 rdr.MPLSHeader.play_item.ForEach(i => i.audio.ForEach(v => { if (!audios.Exists(v1 => v1.pid == v.pid)) audios.Add(v); }));
 
                 var chapters = rdr.Chapters;
-                string strtoc = "";
-                for (int i = 0; i < chapters.Count; i++)
-                    strtoc += string.Format(" {0}", chapters[i] / 600);
-                strtoc = strtoc.Substring(1);
-                CDImageLayout toc = new CDImageLayout(strtoc);
-                var ctdb = new CUEToolsDB(toc, null);
-                //Console.Error.WriteLine("Contacting CTDB...");
-                ctdb.ContactDB(null, "CUETools.eac3to 2.1.7", "", false, true, CTDBMetadataSearch.Extensive);
-                cmbMetadata.ItemsSource = ctdb.Metadata;
-                cmbMetadata.SelectedIndex = 0;
+                if (chapters.Count > 2)
+                {
+                    string strtoc = "";
+                    for (int i = 0; i < chapters.Count; i++)
+                        strtoc += string.Format(" {0}", chapters[i] / 600);
+                    strtoc = strtoc.Substring(1);
+                    CDImageLayout toc = new CDImageLayout(strtoc);
+                    ctdb = new CUEToolsDB(toc, null);
+                    workerCtdb = new BackgroundWorker();
+                    workerCtdb.DoWork += workerCtdb_DoWork;
+                    workerCtdb.RunWorkerAsync();
+                }
             }
             cmbAudioTrack.ItemsSource = audios;
             cmbAudioTrack.SelectedIndex = 0;
         }
 
+        void workerCtdb_DoWork(object sender, DoWorkEventArgs e)
+        {
+            //Console.Error.WriteLine("Contacting CTDB...");
+            this.Dispatcher.Invoke(() =>
+            {
+                pbStatus.Visibility = Visibility.Visible;
+                pbStatus.IsIndeterminate = true;
+            });
+            //ctdb.UploadHelper.onProgress += worker_ctdbProgress;
+            ctdb.ContactDB(null, "CUETools.eac3to 2.1.7", "", false, true, CTDBMetadataSearch.Extensive);
+            this.Dispatcher.Invoke(() =>
+            {
+                pbStatus.Visibility = Visibility.Collapsed;
+                pbStatus.IsIndeterminate = false;
+                cmbMetadata.ItemsSource = ctdb.Metadata;
+                cmbMetadata.SelectedIndex = 0;
+            });
+        }
+
+        //private void worker_ctdbProgress(object sender, UploadProgressEventArgs args)
+        //{
+        //    this.Dispatcher.Invoke(() =>
+        //    {
+        //        pbStatus.Value = (int)args.percent;
+        //    });
+        //}
+
         private void cmbAudioTrack_SelectionChanged(object sender, SelectionChangedEventArgs e)
         {
             if (e.AddedItems.Count == 1)
@@ -118,5 +152,107 @@ namespace CUETools.eac3ui
                 MPLSStream stream = (MPLSStream)(e.AddedItems[0]);
             }
         }
+
+        BackgroundWorker workerCtdb;
+        CUEToolsDB ctdb;
+
+        BackgroundWorker workerExtract;
+        CTDBResponseMeta meta;
+        MPLSReader reader;
+        ushort pid;
+        string outputPath;
+
+        private void buttonExtract_Click(object sender, RoutedEventArgs e)
+        {
+            if (cmbTitleSet.SelectedItem == null) return;
+            pid = ((MPLSStream)cmbAudioTrack.SelectedItem).pid;
+            reader = new MPLSReader((cmbTitleSet.SelectedItem as MPLSReader).Path, null, pid);
+            meta = cmbMetadata.SelectedItem as CTDBResponseMeta;
+            outputPath = Path.Combine(textBoxDestination.Text, meta != null ? meta.artist + " - " + meta.year + " - " + meta.album + ".flac" : "image.flac");
+
+            pbStatus.Visibility = Visibility.Visible;
+            pbStatus.Value = 0.0;
+            //pbStatus.IsIndeterminate = true;
+            stackParams.IsEnabled = false;
+            buttonExtract.IsEnabled = false;
+            buttonStop.Visibility = Visibility.Visible;
+            buttonStop.IsEnabled = true;
+
+            workerExtract = new BackgroundWorker();
+            workerExtract.WorkerSupportsCancellation = true;
+            workerExtract.DoWork += workerExtract_DoWork;
+            workerExtract.RunWorkerAsync();
+        }
+
+        void workerExtract_DoWork(object sender, DoWorkEventArgs e)
+        {
+            try
+            {
+                AudioBuffer buff = new AudioBuffer(reader, 0x10000);
+                FlakeWriterSettings settings = new FlakeWriterSettings()
+                {
+                    PCM = reader.PCM,
+                    Padding = 16536,
+                    EncoderMode = "5"
+                };
+                var start = DateTime.Now;
+                TimeSpan lastPrint = TimeSpan.FromMilliseconds(0);
+                var writer = new FlakeWriter(outputPath, settings);
+                try
+                {
+                    while (reader.Read(buff, -1) != 0)
+                    {
+                        writer.Write(buff);
+                        TimeSpan elapsed = DateTime.Now - start;
+                        if ((elapsed - lastPrint).TotalMilliseconds > 60)
+                        {
+                            long length = Math.Max((long)(reader.Duration.TotalSeconds * reader.PCM.SampleRate), Math.Max(reader.Position, 1));
+                            this.Dispatcher.Invoke(() =>
+                            {
+                                pbStatus.Value = 100.0 * reader.Position / length;
+                            });
+                            lastPrint = elapsed;
+                        }
+                        if (workerExtract.CancellationPending)
+                        {
+                            throw new Exception("aborted");
+                        }
+                    }
+                }
+                finally
+                {
+                    writer.Close();
+                }
+            }
+            catch (Exception ex)
+            {
+                this.Dispatcher.Invoke(() =>
+                {
+                    MessageBox.Show(this, ex.Message, "Extraction failed");
+                });
+            }
+            finally
+            {
+                reader.Close();
+                reader = null;
+            }
+
+            this.Dispatcher.Invoke(() =>
+            {
+                pbStatus.Visibility = Visibility.Collapsed;
+                //pbStatus.IsIndeterminate = false;
+                stackParams.IsEnabled = true;
+                buttonExtract.IsEnabled = true;
+                buttonStop.Visibility = Visibility.Hidden;
+                buttonStop.IsEnabled = false;
+            });
+        }
+
+        private void buttonStop_Click(object sender, RoutedEventArgs e)
+        {
+            workerExtract.CancelAsync();
+            buttonStop.Visibility = Visibility.Hidden;
+            buttonStop.IsEnabled = false;
+        }
     }
 }