209 lines
6.8 KiB
C++
209 lines
6.8 KiB
C++
|
/*
|
||
|
|
* openmpt123_sndfile.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_SNDFILE_HPP
|
||
|
|
#define OPENMPT123_SNDFILE_HPP
|
||
|
|
|
||
|
|
#include "openmpt123_config.hpp"
|
||
|
|
#include "openmpt123.hpp"
|
||
|
|
|
||
|
|
#if defined(MPT_WITH_SNDFILE)
|
||
|
|
|
||
|
|
#include <sndfile.h>
|
||
|
|
|
||
|
|
namespace openmpt123 {
|
||
|
|
|
||
|
|
class sndfile_stream_raii : public file_audio_stream_base {
|
||
|
|
private:
|
||
|
|
commandlineflags flags;
|
||
|
|
std::ostream & log;
|
||
|
|
SNDFILE * sndfile;
|
||
|
|
std::vector<float> interleaved_float_buffer;
|
||
|
|
std::vector<std::int16_t> interleaved_int_buffer;
|
||
|
|
private:
|
||
|
|
enum match_mode_enum {
|
||
|
|
match_print,
|
||
|
|
match_recurse,
|
||
|
|
match_exact,
|
||
|
|
match_better,
|
||
|
|
match_any
|
||
|
|
};
|
||
|
|
std::string match_mode_to_string( match_mode_enum match_mode ) {
|
||
|
|
switch ( match_mode ) {
|
||
|
|
case match_print : return "print" ; break;
|
||
|
|
case match_recurse: return "recurse"; break;
|
||
|
|
case match_exact : return "exact" ; break;
|
||
|
|
case match_better : return "better" ; break;
|
||
|
|
case match_any : return "any" ; break;
|
||
|
|
}
|
||
|
|
return "";
|
||
|
|
}
|
||
|
|
int matched_result( const SF_FORMAT_INFO & format_info, const SF_FORMAT_INFO & subformat_info, match_mode_enum match_mode ) {
|
||
|
|
if ( flags.verbose ) {
|
||
|
|
log << "sndfile: using format '"
|
||
|
|
<< format_info.name << " (" << format_info.extension << ")" << " / " << subformat_info.name
|
||
|
|
<< "', "
|
||
|
|
<< "match: " << match_mode_to_string( match_mode )
|
||
|
|
<< std::endl;
|
||
|
|
}
|
||
|
|
return ( format_info.format & SF_FORMAT_TYPEMASK ) | subformat_info.format;
|
||
|
|
}
|
||
|
|
int find_format( const std::string & extension, match_mode_enum match_mode ) {
|
||
|
|
|
||
|
|
if ( match_mode == match_recurse ) {
|
||
|
|
int result = 0;
|
||
|
|
result = find_format( extension, match_exact );
|
||
|
|
if ( result ) {
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
result = find_format( extension, match_better );
|
||
|
|
if ( result ) {
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
result = find_format( extension, match_any );
|
||
|
|
if ( result ) {
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
if ( result ) {
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int format = 0;
|
||
|
|
int major_count;
|
||
|
|
sf_command( 0, SFC_GET_FORMAT_MAJOR_COUNT, &major_count, sizeof( int ) );
|
||
|
|
for ( int m = 0; m < major_count; m++ ) {
|
||
|
|
|
||
|
|
SF_FORMAT_INFO format_info;
|
||
|
|
format_info.format = m;
|
||
|
|
sf_command( 0, SFC_GET_FORMAT_MAJOR, &format_info, sizeof( SF_FORMAT_INFO ) );
|
||
|
|
format = format_info.format;
|
||
|
|
|
||
|
|
int subtype_count;
|
||
|
|
sf_command( 0, SFC_GET_FORMAT_SUBTYPE_COUNT, &subtype_count, sizeof( int ) );
|
||
|
|
for ( int s = 0; s < subtype_count; s++ ) {
|
||
|
|
|
||
|
|
SF_FORMAT_INFO subformat_info;
|
||
|
|
subformat_info.format = s;
|
||
|
|
sf_command( 0, SFC_GET_FORMAT_SUBTYPE, &subformat_info, sizeof( SF_FORMAT_INFO ) );
|
||
|
|
format = ( format & SF_FORMAT_TYPEMASK ) | subformat_info.format;
|
||
|
|
|
||
|
|
SF_INFO sfinfo;
|
||
|
|
std::memset( &sfinfo, 0, sizeof( SF_INFO ) );
|
||
|
|
sfinfo.channels = flags.channels;
|
||
|
|
sfinfo.format = format;
|
||
|
|
if ( sf_format_check( &sfinfo ) ) {
|
||
|
|
|
||
|
|
switch ( match_mode ) {
|
||
|
|
case match_print:
|
||
|
|
log << "sndfile: "
|
||
|
|
<< ( format_info.name ? format_info.name : "" ) << " (" << ( format_info.extension ? format_info.extension : "" ) << ")"
|
||
|
|
<< " / "
|
||
|
|
<< ( subformat_info.name ? subformat_info.name : "" )
|
||
|
|
<< " ["
|
||
|
|
<< std::setbase(16) << std::setw(8) << std::setfill('0') << format_info.format
|
||
|
|
<< "|"
|
||
|
|
<< std::setbase(16) << std::setw(8) << std::setfill('0') << subformat_info.format
|
||
|
|
<< "]"
|
||
|
|
<< std::endl;
|
||
|
|
break;
|
||
|
|
case match_recurse:
|
||
|
|
break;
|
||
|
|
case match_exact:
|
||
|
|
if ( extension == format_info.extension ) {
|
||
|
|
if ( flags.use_float && ( subformat_info.format == SF_FORMAT_FLOAT ) ) {
|
||
|
|
return matched_result( format_info, subformat_info, match_mode );
|
||
|
|
} else if ( !flags.use_float && ( subformat_info.format == SF_FORMAT_PCM_16 ) ) {
|
||
|
|
return matched_result( format_info, subformat_info, match_mode );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case match_better:
|
||
|
|
if ( extension == format_info.extension ) {
|
||
|
|
if ( flags.use_float && ( subformat_info.format == SF_FORMAT_FLOAT || subformat_info.format == SF_FORMAT_DOUBLE ) ) {
|
||
|
|
return matched_result( format_info, subformat_info, match_mode );
|
||
|
|
} else if ( !flags.use_float && ( subformat_info.format & ( subformat_info.format == SF_FORMAT_PCM_16 || subformat_info.format == SF_FORMAT_PCM_24 || subformat_info.format == SF_FORMAT_PCM_32 ) ) ) {
|
||
|
|
return matched_result( format_info, subformat_info, match_mode );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case match_any:
|
||
|
|
if ( extension == format_info.extension ) {
|
||
|
|
return matched_result( format_info, subformat_info, match_mode );
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
|
||
|
|
}
|
||
|
|
void write_metadata_field( int str_type, const std::string & str ) {
|
||
|
|
if ( !str.empty() ) {
|
||
|
|
sf_set_string( sndfile, str_type, str.c_str() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
public:
|
||
|
|
sndfile_stream_raii( const std::string & filename, const commandlineflags & flags_, std::ostream & log_ ) : flags(flags_), log(log_), sndfile(0) {
|
||
|
|
if ( flags.verbose ) {
|
||
|
|
find_format( "", match_print );
|
||
|
|
log << std::endl;
|
||
|
|
}
|
||
|
|
int format = find_format( flags.output_extension, match_recurse );
|
||
|
|
if ( !format ) {
|
||
|
|
throw exception( "unknown file type" );
|
||
|
|
}
|
||
|
|
SF_INFO info;
|
||
|
|
std::memset( &info, 0, sizeof( SF_INFO ) );
|
||
|
|
info.samplerate = flags.samplerate;
|
||
|
|
info.channels = flags.channels;
|
||
|
|
info.format = format;
|
||
|
|
sndfile = sf_open( filename.c_str(), SFM_WRITE, &info );
|
||
|
|
}
|
||
|
|
~sndfile_stream_raii() {
|
||
|
|
sf_close( sndfile );
|
||
|
|
sndfile = 0;
|
||
|
|
}
|
||
|
|
void write_metadata( std::map<std::string,std::string> metadata ) override {
|
||
|
|
write_metadata_field( SF_STR_TITLE, metadata[ "title" ] );
|
||
|
|
write_metadata_field( SF_STR_ARTIST, metadata[ "artist" ] );
|
||
|
|
write_metadata_field( SF_STR_DATE, metadata[ "date" ] );
|
||
|
|
write_metadata_field( SF_STR_COMMENT, metadata[ "message" ] );
|
||
|
|
write_metadata_field( SF_STR_SOFTWARE, append_software_tag( metadata[ "tracker" ] ) );
|
||
|
|
}
|
||
|
|
void write( const std::vector<float*> buffers, std::size_t frames ) override {
|
||
|
|
interleaved_float_buffer.clear();
|
||
|
|
for ( std::size_t frame = 0; frame < frames; frame++ ) {
|
||
|
|
for ( std::size_t channel = 0; channel < buffers.size(); channel++ ) {
|
||
|
|
interleaved_float_buffer.push_back( buffers[channel][frame] );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
sf_writef_float( sndfile, interleaved_float_buffer.data(), frames );
|
||
|
|
}
|
||
|
|
void write( const std::vector<std::int16_t*> buffers, std::size_t frames ) override {
|
||
|
|
interleaved_int_buffer.clear();
|
||
|
|
for ( std::size_t frame = 0; frame < frames; frame++ ) {
|
||
|
|
for ( std::size_t channel = 0; channel < buffers.size(); channel++ ) {
|
||
|
|
interleaved_int_buffer.push_back( buffers[channel][frame] );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
sf_writef_short( sndfile, interleaved_int_buffer.data(), frames );
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace openmpt123
|
||
|
|
|
||
|
|
#endif // MPT_WITH_SNDFILE
|
||
|
|
|
||
|
|
#endif // OPENMPT123_SNDFILE_HPP
|