mirror of
https://github.com/claunia/cuetools.net.git
synced 2025-12-16 18:14:25 +00:00
344 lines
7.7 KiB
C#
344 lines
7.7 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.IO;
|
|||
|
|
using CUETools.Codecs;
|
|||
|
|
|
|||
|
|
namespace CUETools.Codecs.LAME
|
|||
|
|
{
|
|||
|
|
public class LAMEEncoder : IAudioDest
|
|||
|
|
{
|
|||
|
|
private bool closed = false;
|
|||
|
|
private BE_CONFIG m_Mp3Config = null;
|
|||
|
|
private uint m_hLameStream = 0;
|
|||
|
|
private uint m_InputSamples = 0;
|
|||
|
|
private uint m_OutBufferSize = 0;
|
|||
|
|
private byte[] m_InBuffer = null;
|
|||
|
|
private int m_InBufferPos = 0;
|
|||
|
|
private byte[] m_OutBuffer = null;
|
|||
|
|
private AudioPCMConfig _pcm;
|
|||
|
|
private string _path;
|
|||
|
|
private Stream _IO;
|
|||
|
|
private long position = 0, sample_count = -1;
|
|||
|
|
|
|||
|
|
public LAMEEncoder(string path, Stream IO, AudioPCMConfig pcm)
|
|||
|
|
{
|
|||
|
|
_pcm = pcm;
|
|||
|
|
_path = path;
|
|||
|
|
_IO = IO;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public LAMEEncoder(string path, AudioPCMConfig pcm)
|
|||
|
|
: this(path, null, pcm)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void DeInit()
|
|||
|
|
{
|
|||
|
|
if (!inited || closed)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
uint EncodedSize = 0;
|
|||
|
|
if (m_InBufferPos > 0)
|
|||
|
|
{
|
|||
|
|
if (Lame_encDll.EncodeChunk(m_hLameStream, m_InBuffer, 0, (uint)m_InBufferPos, m_OutBuffer, ref EncodedSize) == Lame_encDll.BE_ERR_SUCCESSFUL)
|
|||
|
|
{
|
|||
|
|
if (EncodedSize > 0)
|
|||
|
|
{
|
|||
|
|
_IO.Write(m_OutBuffer, 0, (int)EncodedSize);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
EncodedSize = 0;
|
|||
|
|
if (Lame_encDll.beDeinitStream(m_hLameStream, m_OutBuffer, ref EncodedSize) == Lame_encDll.BE_ERR_SUCCESSFUL)
|
|||
|
|
{
|
|||
|
|
if (EncodedSize > 0)
|
|||
|
|
{
|
|||
|
|
_IO.Write(m_OutBuffer, 0, (int)EncodedSize);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
Lame_encDll.beCloseStream(m_hLameStream);
|
|||
|
|
_IO.Close();
|
|||
|
|
}
|
|||
|
|
closed = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Close()
|
|||
|
|
{
|
|||
|
|
bool needTag = !closed && _path != null && _path != "";
|
|||
|
|
DeInit();
|
|||
|
|
if (needTag)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
Lame_encDll.beWriteInfoTag(m_hLameStream, _path);
|
|||
|
|
}
|
|||
|
|
catch
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Delete()
|
|||
|
|
{
|
|||
|
|
if (!closed)
|
|||
|
|
{
|
|||
|
|
DeInit();
|
|||
|
|
if (_path != "")
|
|||
|
|
File.Delete(_path);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected virtual BE_CONFIG MakeConfig()
|
|||
|
|
{
|
|||
|
|
return new BE_CONFIG(_pcm, 128);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private bool inited = false;
|
|||
|
|
private void Init()
|
|||
|
|
{
|
|||
|
|
if (inited)
|
|||
|
|
return;
|
|||
|
|
|
|||
|
|
m_Mp3Config = MakeConfig();
|
|||
|
|
|
|||
|
|
uint LameResult = Lame_encDll.beInitStream(m_Mp3Config, ref m_InputSamples, ref m_OutBufferSize, ref m_hLameStream);
|
|||
|
|
if (LameResult != Lame_encDll.BE_ERR_SUCCESSFUL)
|
|||
|
|
throw new ApplicationException(string.Format("Lame_encDll.beInitStream failed with the error code {0}", LameResult));
|
|||
|
|
|
|||
|
|
m_InBuffer = new byte[m_InputSamples * 2]; //Input buffer is expected as short[]
|
|||
|
|
m_OutBuffer = new byte[m_OutBufferSize];
|
|||
|
|
|
|||
|
|
if (_IO == null)
|
|||
|
|
_IO = new FileStream(_path, FileMode.Create, FileAccess.Write, FileShare.Read);
|
|||
|
|
|
|||
|
|
inited = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void Write(AudioBuffer buff)
|
|||
|
|
{
|
|||
|
|
buff.Prepare(this);
|
|||
|
|
|
|||
|
|
Init();
|
|||
|
|
|
|||
|
|
byte[] buffer = buff.Bytes;
|
|||
|
|
int index = 0;
|
|||
|
|
int count = buff.ByteLength;
|
|||
|
|
|
|||
|
|
int ToCopy = 0;
|
|||
|
|
uint EncodedSize = 0;
|
|||
|
|
uint LameResult;
|
|||
|
|
while (count > 0)
|
|||
|
|
{
|
|||
|
|
if (m_InBufferPos > 0)
|
|||
|
|
{
|
|||
|
|
ToCopy = Math.Min(count, m_InBuffer.Length - m_InBufferPos);
|
|||
|
|
Buffer.BlockCopy(buffer, index, m_InBuffer, m_InBufferPos, ToCopy);
|
|||
|
|
m_InBufferPos += ToCopy;
|
|||
|
|
index += ToCopy;
|
|||
|
|
count -= ToCopy;
|
|||
|
|
if (m_InBufferPos >= m_InBuffer.Length)
|
|||
|
|
{
|
|||
|
|
m_InBufferPos = 0;
|
|||
|
|
if ((LameResult = Lame_encDll.EncodeChunk(m_hLameStream, m_InBuffer, m_OutBuffer, ref EncodedSize)) == Lame_encDll.BE_ERR_SUCCESSFUL)
|
|||
|
|
{
|
|||
|
|
if (EncodedSize > 0)
|
|||
|
|
{
|
|||
|
|
_IO.Write(m_OutBuffer, 0, (int)EncodedSize);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
throw new ApplicationException(string.Format("Lame_encDll.EncodeChunk failed with the error code {0}", LameResult));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
if (count >= m_InBuffer.Length)
|
|||
|
|
{
|
|||
|
|
if ((LameResult = Lame_encDll.EncodeChunk(m_hLameStream, buffer, index, (uint)m_InBuffer.Length, m_OutBuffer, ref EncodedSize)) == Lame_encDll.BE_ERR_SUCCESSFUL)
|
|||
|
|
{
|
|||
|
|
if (EncodedSize > 0)
|
|||
|
|
{
|
|||
|
|
_IO.Write(m_OutBuffer, 0, (int)EncodedSize);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
throw new ApplicationException(string.Format("Lame_encDll.EncodeChunk failed with the error code {0}", LameResult));
|
|||
|
|
}
|
|||
|
|
count -= m_InBuffer.Length;
|
|||
|
|
index += m_InBuffer.Length;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
Buffer.BlockCopy(buffer, index, m_InBuffer, 0, count);
|
|||
|
|
m_InBufferPos = count;
|
|||
|
|
index += count;
|
|||
|
|
count = 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public virtual int CompressionLevel
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (value != 0)
|
|||
|
|
throw new Exception("unsupported compression level");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public virtual string Options
|
|||
|
|
{
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (value == null || value == "") return;
|
|||
|
|
throw new Exception("Unsupported options " + value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public long Position
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return position;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public long FinalSampleCount
|
|||
|
|
{
|
|||
|
|
set { sample_count = (int)value; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public long BlockSize
|
|||
|
|
{
|
|||
|
|
set { }
|
|||
|
|
get { return 0; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public AudioPCMConfig PCM
|
|||
|
|
{
|
|||
|
|
get { return _pcm; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public string Path { get { return _path; } }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
[AudioEncoderClass("lame VBR", "mp3", false, "V9 V8 V7 V6 V5 V4 V3 V2 V1 V0", "V2", 2)]
|
|||
|
|
public class LAMEEncoderVBR : LAMEEncoder
|
|||
|
|
{
|
|||
|
|
private int quality = 0;
|
|||
|
|
|
|||
|
|
public LAMEEncoderVBR(string path, Stream IO, AudioPCMConfig pcm)
|
|||
|
|
: base(path, IO, pcm)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public LAMEEncoderVBR(string path, AudioPCMConfig pcm)
|
|||
|
|
: base(path, null, pcm)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected override BE_CONFIG MakeConfig()
|
|||
|
|
{
|
|||
|
|
BE_CONFIG Mp3Config = new BE_CONFIG(PCM, 128);
|
|||
|
|
Mp3Config.format.lhv1.bWriteVBRHeader = 1;
|
|||
|
|
Mp3Config.format.lhv1.nMode = MpegMode.JOINT_STEREO;
|
|||
|
|
Mp3Config.format.lhv1.bEnableVBR = 1;
|
|||
|
|
Mp3Config.format.lhv1.nVBRQuality = quality;
|
|||
|
|
Mp3Config.format.lhv1.nVbrMethod = VBRMETHOD.VBR_METHOD_NEW; // --vbr-new
|
|||
|
|
return Mp3Config;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override int CompressionLevel
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return 9 - quality;
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (value < 0 || value > 9)
|
|||
|
|
throw new Exception("unsupported compression level");
|
|||
|
|
quality = 9 - value;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override string Options
|
|||
|
|
{
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (value == null || value == "") return;
|
|||
|
|
string[] args = value.Split();
|
|||
|
|
for (int i = 0; i < args.Length; i++)
|
|||
|
|
{
|
|||
|
|
//if (args[i] == "--padding-length" && (++i) < args.Length)
|
|||
|
|
//{
|
|||
|
|
// PaddingLength = int.Parse(args[i]);
|
|||
|
|
// continue;
|
|||
|
|
//}
|
|||
|
|
//if (args[i] == "--verify")
|
|||
|
|
//{
|
|||
|
|
// DoVerify = true;
|
|||
|
|
// continue;
|
|||
|
|
//}
|
|||
|
|
throw new Exception("Unsupported options " + value);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[AudioEncoderClass("lame CBR", "mp3", false, "96 128 192 256 320", "256", 2)]
|
|||
|
|
public class LAMEEncoderCBR : LAMEEncoder
|
|||
|
|
{
|
|||
|
|
private int bps_index;
|
|||
|
|
private static readonly uint[] bps_table = new uint[] {96, 128, 192, 256, 320};
|
|||
|
|
|
|||
|
|
public LAMEEncoderCBR(string path, Stream IO, AudioPCMConfig pcm)
|
|||
|
|
: base(path, IO, pcm)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public LAMEEncoderCBR(string path, AudioPCMConfig pcm)
|
|||
|
|
: base(path, null, pcm)
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public override int CompressionLevel
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return bps_index;
|
|||
|
|
}
|
|||
|
|
set
|
|||
|
|
{
|
|||
|
|
if (value < 0 || value > bps_table.Length)
|
|||
|
|
throw new Exception("unsupported compression level");
|
|||
|
|
bps_index = value;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
protected override BE_CONFIG MakeConfig()
|
|||
|
|
{
|
|||
|
|
BE_CONFIG Mp3Config = new BE_CONFIG(PCM, bps_table[bps_index]);
|
|||
|
|
Mp3Config.format.lhv1.bWriteVBRHeader = 1;
|
|||
|
|
//Mp3Config.format.lhv1.nVbrMethod = VBRMETHOD.VBR_METHOD_NONE; // --cbr
|
|||
|
|
//Mp3Config.format.lhv1.nPreset = LAME_QUALITY_PRESET.LQP_NORMAL_QUALITY;
|
|||
|
|
return Mp3Config;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|