mirror of
https://github.com/claunia/cuetools.net.git
synced 2025-12-16 18:14:25 +00:00
opencl flac encoder
This commit is contained in:
chudov
331
CUETools.FLACCL.cmd/Program.cs
Normal file
331
CUETools.FLACCL.cmd/Program.cs
Normal file
@@ -0,0 +1,331 @@
|
||||
/**
|
||||
* CUETools.FLACCL: FLAC audio encoder using CUDA
|
||||
* Copyright (c) 2009 Gregory S. Chudov
|
||||
*
|
||||
* This library is free software; you can redistribute it and/or
|
||||
* modify it under the terms of the GNU Lesser General Public
|
||||
* License as published by the Free Software Foundation; either
|
||||
* version 2.1 of the License, or (at your option) any later version.
|
||||
*
|
||||
* This library is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public
|
||||
* License along with this library; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
*/
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Text;
|
||||
using System.IO;
|
||||
using CUETools.Codecs;
|
||||
using CUETools.Codecs.FLAKE;
|
||||
using CUETools.Codecs.FLACCL;
|
||||
|
||||
namespace CUETools.FLACCL.cmd
|
||||
{
|
||||
class Program
|
||||
{
|
||||
static void Usage()
|
||||
{
|
||||
Console.WriteLine("Usage : CUETools.FLACCL.exe [options] <input.wav>");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Options:");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine(" -0 .. -11 Compression level, default 7; 9..11 are non-subset");
|
||||
Console.WriteLine(" -o <file> Output filename, or \"-\" for stdout, or nul");
|
||||
Console.WriteLine(" -p # Padding bytes");
|
||||
Console.WriteLine(" -q --quiet Quiet mode");
|
||||
Console.WriteLine(" --verify Verify during encoding");
|
||||
Console.WriteLine(" --no-md5 Don't compute MD5 hash");
|
||||
Console.WriteLine(" --no-seektable Don't generate a seektable");
|
||||
Console.WriteLine(" --slow-gpu Some encoding stages are done on CPU");
|
||||
Console.WriteLine(" --cpu-threads Use additional CPU threads");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Advanced Options:");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine(" -b # Block size");
|
||||
Console.WriteLine(" -v # Variable block size mode (0,4)");
|
||||
Console.WriteLine(" -s <method> Stereo decorrelation (independent,search)");
|
||||
Console.WriteLine(" -r #[,#] Rice partition order {max} or {min},{max} (0..8)");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("LPC options:");
|
||||
Console.WriteLine();
|
||||
Console.WriteLine(" -w <func>[,<func>] Window functions (bartlett,welch,hann,flattop,tukey)");
|
||||
Console.WriteLine(" -l #[,#] Prediction order {max} or {min},{max} (1..32)");
|
||||
Console.WriteLine(" --max-precision Coefficients precision search (0..1)");
|
||||
Console.WriteLine();
|
||||
}
|
||||
|
||||
static int Main(string[] args)
|
||||
{
|
||||
TextWriter stdout = Console.Out;
|
||||
Console.SetOut(Console.Error);
|
||||
|
||||
DateTime start = DateTime.Now;
|
||||
TimeSpan lastPrint = TimeSpan.FromMilliseconds(0);
|
||||
bool debug = false, quiet = false;
|
||||
string stereo_method = null;
|
||||
string window_function = null;
|
||||
string input_file = null;
|
||||
string output_file = null;
|
||||
int min_partition_order = -1, max_partition_order = -1,
|
||||
min_lpc_order = -1, max_lpc_order = -1,
|
||||
min_fixed_order = -1, max_fixed_order = -1,
|
||||
min_precision = -1, max_precision = -1,
|
||||
orders_per_window = -1,
|
||||
blocksize = -1;
|
||||
int level = -1, padding = -1, vbr_mode = -1;
|
||||
bool do_md5 = true, do_seektable = true, do_verify = false, gpu_only = true;
|
||||
bool buffered = false;
|
||||
int cpu_threads = 0;
|
||||
bool ok = true;
|
||||
|
||||
for (int arg = 0; arg < args.Length; arg++)
|
||||
{
|
||||
if (args[arg].Length == 0)
|
||||
ok = false;
|
||||
else if (args[arg] == "--debug")
|
||||
debug = true;
|
||||
else if ((args[arg] == "-q" || args[arg] == "--quiet"))
|
||||
quiet = true;
|
||||
else if (args[arg] == "--verify")
|
||||
do_verify = true;
|
||||
else if (args[arg] == "--no-seektable")
|
||||
do_seektable = false;
|
||||
else if (args[arg] == "--slow-gpu")
|
||||
gpu_only = false;
|
||||
else if (args[arg] == "--no-md5")
|
||||
do_md5 = false;
|
||||
else if (args[arg] == "--buffered")
|
||||
buffered = true;
|
||||
else if (args[arg] == "--cpu-threads")
|
||||
ok = (++arg < args.Length) && int.TryParse(args[arg], out cpu_threads);
|
||||
else if ((args[arg] == "-o" || args[arg] == "--output") && ++arg < args.Length)
|
||||
output_file = args[arg];
|
||||
else if ((args[arg] == "-s" || args[arg] == "--stereo") && ++arg < args.Length)
|
||||
stereo_method = args[arg];
|
||||
else if ((args[arg] == "-w" || args[arg] == "--window") && ++arg < args.Length)
|
||||
window_function = args[arg];
|
||||
else if ((args[arg] == "-r" || args[arg] == "--partition-order") && ++arg < args.Length)
|
||||
{
|
||||
ok = (args[arg].Split(',').Length == 2 &&
|
||||
int.TryParse(args[arg].Split(',')[0], out min_partition_order) &&
|
||||
int.TryParse(args[arg].Split(',')[1], out max_partition_order)) ||
|
||||
int.TryParse(args[arg], out max_partition_order);
|
||||
}
|
||||
else if ((args[arg] == "-l" || args[arg] == "--lpc-order") && ++arg < args.Length)
|
||||
{
|
||||
ok = (args[arg].Split(',').Length == 2 &&
|
||||
int.TryParse(args[arg].Split(',')[0], out min_lpc_order) &&
|
||||
int.TryParse(args[arg].Split(',')[1], out max_lpc_order)) ||
|
||||
int.TryParse(args[arg], out max_lpc_order);
|
||||
}
|
||||
else if (args[arg] == "--fixed-order" && ++arg < args.Length)
|
||||
{
|
||||
ok = (args[arg].Split(',').Length == 2 &&
|
||||
int.TryParse(args[arg].Split(',')[0], out min_fixed_order) &&
|
||||
int.TryParse(args[arg].Split(',')[1], out max_fixed_order)) ||
|
||||
int.TryParse(args[arg], out max_fixed_order);
|
||||
}
|
||||
else if ((args[arg] == "-c" || args[arg] == "--max-precision") && ++arg < args.Length)
|
||||
{
|
||||
ok = (args[arg].Split(',').Length == 2 &&
|
||||
int.TryParse(args[arg].Split(',')[0], out min_precision) &&
|
||||
int.TryParse(args[arg].Split(',')[1], out max_precision)) ||
|
||||
int.TryParse(args[arg], out max_precision);
|
||||
}
|
||||
else if ((args[arg] == "-v" || args[arg] == "--vbr"))
|
||||
ok = (++arg < args.Length) && int.TryParse(args[arg], out vbr_mode);
|
||||
else if (args[arg] == "--orders-per-window")
|
||||
ok = (++arg < args.Length) && int.TryParse(args[arg], out orders_per_window);
|
||||
else if ((args[arg] == "-b" || args[arg] == "--blocksize") && ++arg < args.Length)
|
||||
ok = int.TryParse(args[arg], out blocksize);
|
||||
else if ((args[arg] == "-p" || args[arg] == "--padding") && ++arg < args.Length)
|
||||
ok = int.TryParse(args[arg], out padding);
|
||||
else if (args[arg] != "-" && args[arg][0] == '-' && int.TryParse(args[arg].Substring(1), out level))
|
||||
ok = level >= 0 && level <= 11;
|
||||
else if ((args[arg][0] != '-' || args[arg] == "-") && input_file == null)
|
||||
input_file = args[arg];
|
||||
else
|
||||
ok = false;
|
||||
if (!ok)
|
||||
break;
|
||||
}
|
||||
if (!quiet)
|
||||
{
|
||||
Console.WriteLine("{0}, Copyright (C) 2009 Gregory S. Chudov.", FLACCLWriter.vendor_string);
|
||||
Console.WriteLine("This is free software under the GNU GPLv3+ license; There is NO WARRANTY, to");
|
||||
Console.WriteLine("the extent permitted by law. <http://www.gnu.org/licenses/> for details.");
|
||||
}
|
||||
if (!ok || input_file == null)
|
||||
{
|
||||
Usage();
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (((input_file == "-" || Path.GetExtension(input_file) == ".flac") && output_file == null))
|
||||
{
|
||||
Console.WriteLine();
|
||||
Console.WriteLine("Output file not specified.");
|
||||
Console.WriteLine();
|
||||
Usage();
|
||||
return 2;
|
||||
}
|
||||
|
||||
IAudioSource audioSource;
|
||||
if (input_file == "-")
|
||||
audioSource = new WAVReader("", Console.OpenStandardInput());
|
||||
else if (File.Exists(input_file) && Path.GetExtension(input_file) == ".wav")
|
||||
audioSource = new WAVReader(input_file, null);
|
||||
else if (File.Exists(input_file) && Path.GetExtension(input_file) == ".flac")
|
||||
audioSource = new FlakeReader(input_file, null);
|
||||
else
|
||||
{
|
||||
Usage();
|
||||
return 2;
|
||||
}
|
||||
if (buffered)
|
||||
audioSource = new AudioPipe(audioSource, FLACCLWriter.MAX_BLOCKSIZE);
|
||||
if (output_file == null)
|
||||
output_file = Path.ChangeExtension(input_file, "flac");
|
||||
FLACCLWriter encoder = new FLACCLWriter((output_file == "-" || output_file == "nul") ? "" : output_file,
|
||||
output_file == "-" ? Console.OpenStandardOutput() :
|
||||
output_file == "nul" ? new NullStream() : null,
|
||||
audioSource.PCM);
|
||||
encoder.FinalSampleCount = audioSource.Length;
|
||||
IAudioDest audioDest = encoder;
|
||||
AudioBuffer buff = new AudioBuffer(audioSource, FLACCLWriter.MAX_BLOCKSIZE);
|
||||
|
||||
try
|
||||
{
|
||||
(encoder.Settings as FLACCLWriterSettings).GPUOnly = gpu_only;
|
||||
(encoder.Settings as FLACCLWriterSettings).CPUThreads = cpu_threads;
|
||||
(encoder.Settings as FLACCLWriterSettings).DoVerify = do_verify;
|
||||
(encoder.Settings as FLACCLWriterSettings).DoMD5 = do_md5;
|
||||
if (level >= 0)
|
||||
encoder.CompressionLevel = level;
|
||||
if (stereo_method != null)
|
||||
encoder.StereoMethod = Flake.LookupStereoMethod(stereo_method);
|
||||
if (window_function != null)
|
||||
encoder.WindowFunction = Flake.LookupWindowFunction(window_function);
|
||||
if (min_partition_order >= 0)
|
||||
encoder.MinPartitionOrder = min_partition_order;
|
||||
if (max_partition_order >= 0)
|
||||
encoder.MaxPartitionOrder = max_partition_order;
|
||||
if (min_lpc_order >= 0)
|
||||
encoder.MinLPCOrder = min_lpc_order;
|
||||
if (max_lpc_order >= 0)
|
||||
encoder.MaxLPCOrder = max_lpc_order;
|
||||
if (min_fixed_order >= 0)
|
||||
encoder.MinFixedOrder = min_fixed_order;
|
||||
if (max_fixed_order >= 0)
|
||||
encoder.MaxFixedOrder = max_fixed_order;
|
||||
if (max_precision >= 0)
|
||||
encoder.MaxPrecisionSearch = max_precision;
|
||||
if (min_precision >= 0)
|
||||
encoder.MinPrecisionSearch = min_precision;
|
||||
if (blocksize >= 0)
|
||||
encoder.BlockSize = blocksize;
|
||||
if (padding >= 0)
|
||||
encoder.Padding = padding;
|
||||
if (vbr_mode >= 0)
|
||||
encoder.VBRMode = vbr_mode;
|
||||
if (orders_per_window >= 0)
|
||||
encoder.OrdersPerWindow = orders_per_window;
|
||||
encoder.DoSeekTable = do_seektable;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Usage();
|
||||
Console.WriteLine("");
|
||||
Console.WriteLine("Error: {0}.", ex.Message);
|
||||
return 3;
|
||||
}
|
||||
|
||||
if (!quiet)
|
||||
{
|
||||
Console.WriteLine("Filename : {0}", input_file);
|
||||
Console.WriteLine("File Info : {0}kHz; {1} channel; {2} bit; {3}", audioSource.PCM.SampleRate, audioSource.PCM.ChannelCount, audioSource.PCM.BitsPerSample, TimeSpan.FromSeconds(audioSource.Length * 1.0 / audioSource.PCM.SampleRate));
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
while (audioSource.Read(buff, -1) != 0)
|
||||
{
|
||||
audioDest.Write(buff);
|
||||
TimeSpan elapsed = DateTime.Now - start;
|
||||
if (!quiet)
|
||||
{
|
||||
if ((elapsed - lastPrint).TotalMilliseconds > 60)
|
||||
{
|
||||
Console.Error.Write("\rProgress : {0:00}%; {1:0.00}x; {2}/{3}",
|
||||
100.0 * audioSource.Position / audioSource.Length,
|
||||
audioSource.Position / elapsed.TotalSeconds / audioSource.PCM.SampleRate,
|
||||
elapsed,
|
||||
TimeSpan.FromMilliseconds(elapsed.TotalMilliseconds / audioSource.Position * audioSource.Length)
|
||||
);
|
||||
lastPrint = elapsed;
|
||||
}
|
||||
}
|
||||
}
|
||||
audioDest.Close();
|
||||
}
|
||||
#if DEBUG
|
||||
catch (OpenCLNet.OpenCLBuildException ex)
|
||||
{
|
||||
Console.Error.Write("\r \r");
|
||||
Console.WriteLine("Error : {0}", ex.Message);
|
||||
Console.WriteLine("{0}", ex.BuildLogs[0]);
|
||||
audioDest.Delete();
|
||||
audioSource.Close();
|
||||
return 4;
|
||||
}
|
||||
#else
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.Error.Write("\r \r");
|
||||
Console.WriteLine("Error : {0}", ex.Message);
|
||||
audioDest.Delete();
|
||||
audioSource.Close();
|
||||
return 4;
|
||||
}
|
||||
#endif
|
||||
|
||||
TimeSpan totalElapsed = DateTime.Now - start;
|
||||
if (!quiet)
|
||||
{
|
||||
Console.Error.Write("\r \r");
|
||||
Console.WriteLine("Results : {0:0.00}x; {2} bytes in {1} seconds;",
|
||||
audioSource.Position / totalElapsed.TotalSeconds / audioSource.PCM.SampleRate,
|
||||
totalElapsed,
|
||||
encoder.TotalSize
|
||||
);
|
||||
}
|
||||
audioSource.Close();
|
||||
|
||||
if (debug)
|
||||
{
|
||||
Console.SetOut(stdout);
|
||||
Console.Out.WriteLine("{0}\t{1}\t{2}\t{3}\t{4}..{5}\t{6}..{7}\t{8}..{9}\t{10}\t{11}",
|
||||
encoder.TotalSize,
|
||||
encoder.UserProcessorTime.TotalSeconds > 0 ? encoder.UserProcessorTime.TotalSeconds : totalElapsed.TotalSeconds,
|
||||
encoder.StereoMethod.ToString().PadRight(15),
|
||||
encoder.WindowFunction.ToString().PadRight(15),
|
||||
encoder.MinPartitionOrder,
|
||||
encoder.MaxPartitionOrder,
|
||||
encoder.MinLPCOrder,
|
||||
encoder.MaxLPCOrder,
|
||||
encoder.MinPrecisionSearch,
|
||||
encoder.MaxPrecisionSearch,
|
||||
encoder.BlockSize,
|
||||
encoder.VBRMode
|
||||
);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user