Initial community commit
This commit is contained in:
Jef
parent
537bcbc862
commit
fc06254474
16440 changed files with 4239995 additions and 2 deletions
|
|
@ -0,0 +1,262 @@
|
|||
/*
|
||||
* StreamEncoder.cpp
|
||||
* -----------------
|
||||
* Purpose: Exporting streamed music files.
|
||||
* Notes : none
|
||||
* Authors: Joern Heusipp
|
||||
* OpenMPT Devs
|
||||
* The OpenMPT source code is released under the BSD license. Read LICENSE for more details.
|
||||
*/
|
||||
|
||||
#include "stdafx.h"
|
||||
|
||||
#include "StreamEncoder.h"
|
||||
#include "StreamEncoderVorbis.h"
|
||||
|
||||
#include "Mptrack.h"
|
||||
|
||||
#ifdef MPT_WITH_OGG
|
||||
#include <ogg/ogg.h>
|
||||
#endif
|
||||
#ifdef MPT_WITH_VORBIS
|
||||
#include <vorbis/codec.h>
|
||||
#endif
|
||||
#ifdef MPT_WITH_VORBISENC
|
||||
#include <vorbis/vorbisenc.h>
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
OPENMPT_NAMESPACE_BEGIN
|
||||
|
||||
|
||||
|
||||
static Encoder::Traits VorbisBuildTraits()
|
||||
{
|
||||
Encoder::Traits traits;
|
||||
#if defined(MPT_WITH_OGG) && defined(MPT_WITH_VORBIS) && defined(MPT_WITH_VORBISENC)
|
||||
traits.fileExtension = P_("ogg");
|
||||
traits.fileShortDescription = U_("Vorbis");
|
||||
traits.fileDescription = U_("Ogg Vorbis");
|
||||
traits.encoderSettingsName = U_("Vorbis");
|
||||
traits.canTags = true;
|
||||
traits.maxChannels = 4;
|
||||
traits.samplerates = mpt::make_vector(vorbis_samplerates);
|
||||
traits.modes = Encoder::ModeABR | Encoder::ModeQuality;
|
||||
traits.bitrates = mpt::make_vector(vorbis_bitrates);
|
||||
traits.defaultSamplerate = 48000;
|
||||
traits.defaultChannels = 2;
|
||||
traits.defaultMode = Encoder::ModeQuality;
|
||||
traits.defaultBitrate = 160;
|
||||
traits.defaultQuality = 0.5;
|
||||
#endif
|
||||
return traits;
|
||||
}
|
||||
|
||||
#if defined(MPT_WITH_OGG) && defined(MPT_WITH_VORBIS) && defined(MPT_WITH_VORBISENC)
|
||||
|
||||
class VorbisStreamWriter : public StreamWriterBase
|
||||
{
|
||||
private:
|
||||
ogg_stream_state os;
|
||||
ogg_page og;
|
||||
ogg_packet op;
|
||||
vorbis_info vi;
|
||||
vorbis_comment vc;
|
||||
vorbis_dsp_state vd;
|
||||
vorbis_block vb;
|
||||
int vorbis_channels;
|
||||
private:
|
||||
void WritePage()
|
||||
{
|
||||
buf.resize(og.header_len);
|
||||
std::memcpy(buf.data(), og.header, og.header_len);
|
||||
WriteBuffer();
|
||||
buf.resize(og.body_len);
|
||||
std::memcpy(buf.data(), og.body, og.body_len);
|
||||
WriteBuffer();
|
||||
}
|
||||
void AddCommentField(const std::string &field, const mpt::ustring &data)
|
||||
{
|
||||
if(!field.empty() && !data.empty())
|
||||
{
|
||||
vorbis_comment_add_tag(&vc, field.c_str(), mpt::ToCharset(mpt::Charset::UTF8, data).c_str());
|
||||
}
|
||||
}
|
||||
public:
|
||||
VorbisStreamWriter(std::ostream &stream, const Encoder::Settings &settings, const FileTags &tags)
|
||||
: StreamWriterBase(stream)
|
||||
{
|
||||
vorbis_channels = 0;
|
||||
|
||||
vorbis_channels = settings.Channels;
|
||||
|
||||
vorbis_info_init(&vi);
|
||||
vorbis_comment_init(&vc);
|
||||
|
||||
if(settings.Mode == Encoder::ModeQuality)
|
||||
{
|
||||
vorbis_encode_init_vbr(&vi, vorbis_channels, settings.Samplerate, settings.Quality);
|
||||
} else
|
||||
{
|
||||
vorbis_encode_init(&vi, vorbis_channels, settings.Samplerate, -1, settings.Bitrate * 1000, -1);
|
||||
}
|
||||
|
||||
vorbis_analysis_init(&vd, &vi);
|
||||
vorbis_block_init(&vd, &vb);
|
||||
ogg_stream_init(&os, mpt::random<uint32>(theApp.PRNG()));
|
||||
|
||||
if(settings.Tags)
|
||||
{
|
||||
AddCommentField("ENCODER", tags.encoder);
|
||||
AddCommentField("SOURCEMEDIA", U_("tracked music file"));
|
||||
AddCommentField("TITLE", tags.title );
|
||||
AddCommentField("ARTIST", tags.artist );
|
||||
AddCommentField("ALBUM", tags.album );
|
||||
AddCommentField("DATE", tags.year );
|
||||
AddCommentField("COMMENT", tags.comments );
|
||||
AddCommentField("GENRE", tags.genre );
|
||||
AddCommentField("CONTACT", tags.url );
|
||||
AddCommentField("BPM", tags.bpm ); // non-standard
|
||||
AddCommentField("TRACKNUMBER", tags.trackno );
|
||||
}
|
||||
|
||||
ogg_packet header;
|
||||
ogg_packet header_comm;
|
||||
ogg_packet header_code;
|
||||
vorbis_analysis_headerout(&vd, &vc, &header, &header_comm, &header_code);
|
||||
ogg_stream_packetin(&os, &header);
|
||||
while(ogg_stream_flush(&os, &og))
|
||||
{
|
||||
WritePage();
|
||||
}
|
||||
ogg_stream_packetin(&os, &header_comm);
|
||||
ogg_stream_packetin(&os, &header_code);
|
||||
while(ogg_stream_flush(&os, &og))
|
||||
{
|
||||
WritePage();
|
||||
}
|
||||
|
||||
}
|
||||
void WriteInterleaved(size_t count, const float *interleaved) override
|
||||
{
|
||||
size_t countTotal = count;
|
||||
while(countTotal > 0)
|
||||
{
|
||||
int countChunk = mpt::saturate_cast<int>(countTotal);
|
||||
countTotal -= countChunk;
|
||||
float **buffer = vorbis_analysis_buffer(&vd, countChunk);
|
||||
for(int frame = 0; frame < countChunk; ++frame)
|
||||
{
|
||||
for(int channel = 0; channel < vorbis_channels; ++channel)
|
||||
{
|
||||
buffer[channel][frame] = interleaved[frame*vorbis_channels+channel];
|
||||
}
|
||||
}
|
||||
vorbis_analysis_wrote(&vd, countChunk);
|
||||
while(vorbis_analysis_blockout(&vd, &vb) == 1)
|
||||
{
|
||||
vorbis_analysis(&vb, NULL);
|
||||
vorbis_bitrate_addblock(&vb);
|
||||
while(vorbis_bitrate_flushpacket(&vd, &op))
|
||||
{
|
||||
ogg_stream_packetin(&os, &op);
|
||||
while(ogg_stream_pageout(&os, &og))
|
||||
{
|
||||
WritePage();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
void WriteFinalize() override
|
||||
{
|
||||
vorbis_analysis_wrote(&vd, 0);
|
||||
while(vorbis_analysis_blockout(&vd, &vb) == 1)
|
||||
{
|
||||
vorbis_analysis(&vb, NULL);
|
||||
vorbis_bitrate_addblock(&vb);
|
||||
while(vorbis_bitrate_flushpacket(&vd, &op))
|
||||
{
|
||||
ogg_stream_packetin(&os, &op);
|
||||
while(ogg_stream_flush(&os, &og))
|
||||
{
|
||||
WritePage();
|
||||
if(ogg_page_eos(&og))
|
||||
{
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
virtual ~VorbisStreamWriter()
|
||||
{
|
||||
ogg_stream_clear(&os);
|
||||
vorbis_block_clear(&vb);
|
||||
vorbis_dsp_clear(&vd);
|
||||
vorbis_comment_clear(&vc);
|
||||
vorbis_info_clear(&vi);
|
||||
}
|
||||
};
|
||||
|
||||
#endif // MPT_WITH_OGG && MPT_WITH_VORBIS && MPT_WITH_VORBISENC
|
||||
|
||||
|
||||
|
||||
VorbisEncoder::VorbisEncoder()
|
||||
{
|
||||
SetTraits(VorbisBuildTraits());
|
||||
}
|
||||
|
||||
|
||||
bool VorbisEncoder::IsAvailable() const
|
||||
{
|
||||
#if defined(MPT_WITH_OGG) && defined(MPT_WITH_VORBIS) && defined(MPT_WITH_VORBISENC)
|
||||
return true;
|
||||
#else
|
||||
return false;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
std::unique_ptr<IAudioStreamEncoder> VorbisEncoder::ConstructStreamEncoder(std::ostream &file, const Encoder::Settings &settings, const FileTags &tags) const
|
||||
{
|
||||
#if defined(MPT_WITH_OGG) && defined(MPT_WITH_VORBIS) && defined(MPT_WITH_VORBISENC)
|
||||
return std::make_unique<VorbisStreamWriter>(file, settings, tags);
|
||||
#else
|
||||
MPT_UNREFERENCED_PARAMETER(file);
|
||||
MPT_UNREFERENCED_PARAMETER(settings);
|
||||
MPT_UNREFERENCED_PARAMETER(tags);
|
||||
return nullptr;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
bool VorbisEncoder::IsBitrateSupported(int samplerate, int channels, int bitrate) const
|
||||
{
|
||||
#if defined(MPT_WITH_OGG) && defined(MPT_WITH_VORBIS) && defined(MPT_WITH_VORBISENC)
|
||||
vorbis_info vi;
|
||||
vorbis_info_init(&vi);
|
||||
bool result = (0 == vorbis_encode_init(&vi, channels, samplerate, -1, bitrate * 1000, -1));
|
||||
vorbis_info_clear(&vi);
|
||||
return result;
|
||||
#else
|
||||
MPT_UNREFERENCED_PARAMETER(samplerate);
|
||||
MPT_UNREFERENCED_PARAMETER(channels);
|
||||
MPT_UNREFERENCED_PARAMETER(bitrate);
|
||||
return false;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
mpt::ustring VorbisEncoder::DescribeQuality(float quality) const
|
||||
{
|
||||
static constexpr int q_table[11] = { 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 500 }; // http://wiki.hydrogenaud.io/index.php?title=Recommended_Ogg_Vorbis
|
||||
int q = Clamp(mpt::saturate_round<int>(quality * 10.0f), 0, 10);
|
||||
return MPT_UFORMAT("Q{} (~{} kbit)")(mpt::ufmt::fix(quality * 10.0f, 1), q_table[q]);
|
||||
}
|
||||
|
||||
|
||||
|
||||
OPENMPT_NAMESPACE_END
|
||||
Loading…
Add table
Add a link
Reference in a new issue