mirror of
https://github.com/claunia/cuetools.net.git
synced 2025-12-16 18:14:25 +00:00
support lame_enc.dll
This commit is contained in:
chudov
343
CUETools.Codecs.LAME/Encoder.cs
Normal file
343
CUETools.Codecs.LAME/Encoder.cs
Normal file
@@ -0,0 +1,343 @@
|
||||
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;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user