winamp/Src/external_dependencies/openmpt-trunk/openmpt123/openmpt123_flac.hpp
2024-09-24 14:54:57 +02:00

143 lines
5.4 KiB
C++

/*
* openmpt123_flac.hpp
* -------------------
* Purpose: libopenmpt command line player
* Notes : (currently none)
* Authors: OpenMPT Devs
* The OpenMPT source code is released under the BSD license. Read LICENSE for more details.
*/
#ifndef OPENMPT123_FLAC_HPP
#define OPENMPT123_FLAC_HPP
#include "openmpt123_config.hpp"
#include "openmpt123.hpp"
#if defined(MPT_WITH_FLAC)
#if defined(_MSC_VER) && defined(__clang__) && defined(__c2__)
#include <sys/types.h>
#if __STDC__
typedef _off_t off_t;
#endif
#endif
#if defined(__clang__)
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wreserved-id-macro"
#endif
#include <FLAC/metadata.h>
#include <FLAC/format.h>
#include <FLAC/stream_encoder.h>
#if defined(__clang__)
#pragma clang diagnostic pop
#endif
namespace openmpt123 {
class flac_stream_raii : public file_audio_stream_base {
private:
commandlineflags flags;
std::string filename;
bool called_init;
std::vector< std::pair< std::string, std::string > > tags;
FLAC__StreamMetadata * flac_metadata[1];
FLAC__StreamEncoder * encoder;
std::vector<FLAC__int32> interleaved_buffer;
void add_vorbiscomment_field( FLAC__StreamMetadata * vorbiscomment, const std::string & field, const std::string & value ) {
if ( !value.empty() ) {
FLAC__StreamMetadata_VorbisComment_Entry entry;
FLAC__metadata_object_vorbiscomment_entry_from_name_value_pair( &entry, field.c_str(), value.c_str() );
FLAC__metadata_object_vorbiscomment_append_comment( vorbiscomment, entry, false );
}
}
public:
flac_stream_raii( const std::string & filename_, const commandlineflags & flags_, std::ostream & /*log*/ ) : flags(flags_), filename(filename_), called_init(false), encoder(0) {
flac_metadata[0] = 0;
encoder = FLAC__stream_encoder_new();
if ( !encoder ) {
throw exception( "error creating flac encoder" );
}
FLAC__stream_encoder_set_channels( encoder, flags.channels );
FLAC__stream_encoder_set_bits_per_sample( encoder, flags.use_float ? 24 : 16 );
FLAC__stream_encoder_set_sample_rate( encoder, flags.samplerate );
FLAC__stream_encoder_set_compression_level( encoder, 8 );
}
~flac_stream_raii() {
if ( encoder ) {
FLAC__stream_encoder_finish( encoder );
FLAC__stream_encoder_delete( encoder );
encoder = 0;
}
if ( flac_metadata[0] ) {
FLAC__metadata_object_delete( flac_metadata[0] );
flac_metadata[0] = 0;
}
}
void write_metadata( std::map<std::string,std::string> metadata ) override {
if ( called_init ) {
return;
}
tags.clear();
tags.push_back( std::make_pair( "TITLE", metadata[ "title" ] ) );
tags.push_back( std::make_pair( "ARTIST", metadata[ "artist" ] ) );
tags.push_back( std::make_pair( "DATE", metadata[ "date" ] ) );
tags.push_back( std::make_pair( "COMMENT", metadata[ "message" ] ) );
if ( !metadata[ "type" ].empty() && !metadata[ "tracker" ].empty() ) {
tags.push_back( std::make_pair( "SOURCEMEDIA", std::string() + "'" + metadata[ "type" ] + "' tracked music file, made with '" + metadata[ "tracker" ] + "', rendered with '" + get_encoder_tag() + "'" ) );
} else if ( !metadata[ "type_long" ].empty() ) {
tags.push_back( std::make_pair( "SOURCEMEDIA", std::string() + "'" + metadata[ "type" ] + "' tracked music file, rendered with '" + get_encoder_tag() + "'" ) );
} else if ( !metadata[ "tracker" ].empty() ) {
tags.push_back( std::make_pair( "SOURCEMEDIA", std::string() + "tracked music file, made with '" + metadata[ "tracker" ] + "', rendered with '" + get_encoder_tag() + "'" ) );
} else {
tags.push_back( std::make_pair( "SOURCEMEDIA", std::string() + "tracked music file, rendered with '" + get_encoder_tag() + "'" ) );
}
tags.push_back( std::make_pair( "ENCODER", get_encoder_tag() ) );
flac_metadata[0] = FLAC__metadata_object_new( FLAC__METADATA_TYPE_VORBIS_COMMENT );
for ( std::vector< std::pair< std::string, std::string > >::iterator tag = tags.begin(); tag != tags.end(); ++tag ) {
add_vorbiscomment_field( flac_metadata[0], tag->first, tag->second );
}
FLAC__stream_encoder_set_metadata( encoder, flac_metadata, 1 );
}
void write( const std::vector<float*> buffers, std::size_t frames ) override {
if ( !called_init ) {
FLAC__stream_encoder_init_file( encoder, filename.c_str(), NULL, 0 );
called_init = true;
}
interleaved_buffer.clear();
for ( std::size_t frame = 0; frame < frames; frame++ ) {
for ( std::size_t channel = 0; channel < buffers.size(); channel++ ) {
float in = buffers[channel][frame];
if ( in <= -1.0f ) {
in = -1.0f;
} else if ( in >= 1.0f ) {
in = 1.0f;
}
FLAC__int32 out = mpt_lround( in * (1<<23) );
out = std::max( 0 - (1<<23), out );
out = std::min( out, 0 + (1<<23) - 1 );
interleaved_buffer.push_back( out );
}
}
FLAC__stream_encoder_process_interleaved( encoder, interleaved_buffer.data(), static_cast<unsigned int>( frames ) );
}
void write( const std::vector<std::int16_t*> buffers, std::size_t frames ) override {
if ( !called_init ) {
FLAC__stream_encoder_init_file( encoder, filename.c_str(), NULL, 0 );
called_init = true;
}
interleaved_buffer.clear();
for ( std::size_t frame = 0; frame < frames; frame++ ) {
for ( std::size_t channel = 0; channel < buffers.size(); channel++ ) {
interleaved_buffer.push_back( buffers[channel][frame] );
}
}
FLAC__stream_encoder_process_interleaved( encoder, interleaved_buffer.data(), static_cast<unsigned int>( frames ) );
}
};
} // namespace openmpt123
#endif // MPT_WITH_FLAC
#endif // OPENMPT123_FLAC_HPP