OptionLoader: add getSampleFormats

Clement Champetier · Clement Champetier · commit 39314e1f0ed1 · 2014-07-16T16:10:04.000+02:00
diff --git a/src/AvTranscoder/OptionLoader.cpp b/src/AvTranscoder/OptionLoader.cpp
@@ -7,6 +7,7 @@ extern "C" {
 #include <libavutil/opt.h>
 #include <libavutil/pixfmt.h>
 #include <libavutil/pixdesc.h>
+#include <libavutil/samplefmt.h>
 }
 
 #ifndef AV_OPT_FLAG_FILTERING_PARAM
@@ -322,4 +323,34 @@ std::vector<std::string> OptionLoader::getPixelFormats( const std::string& video
 	return pixelFormats;
 }
 
+std::vector<std::string> OptionLoader::getSampleFormats( const std::string& audioCodecName ) const
+{
+	std::vector<std::string> sampleFormats;
+	
+	if( audioCodecName.empty() )
+	{
+		for( size_t sampleFormat = 0; sampleFormat < AV_SAMPLE_FMT_NB; ++sampleFormat)
+		{
+			sampleFormats.push_back( av_get_sample_fmt_name( static_cast<AVSampleFormat>( sampleFormat ) ) );
+		}
+	}
+	else
+	{
+		const AVCodec* audioCodec = avcodec_find_encoder_by_name( audioCodecName.c_str() );
+		if( audioCodec && audioCodec->sample_fmts != NULL )
+		{
+			size_t sample_fmt = 0;
+			while( audioCodec->sample_fmts[sample_fmt] != -1 )
+			{
+				const char* sampleFormatName = av_get_sample_fmt_name( audioCodec->sample_fmts[sample_fmt] );
+				if( sampleFormatName )
+					sampleFormats.push_back( std::string( sampleFormatName ) );
+				sample_fmt++;
+			}
+		}
+	}
+	
+	return sampleFormats;
+}
+
 }
diff --git a/src/AvTranscoder/OptionLoader.hpp b/src/AvTranscoder/OptionLoader.hpp
@@ -67,6 +67,12 @@ class OptionLoader
 	 */
 	std::vector<std::string> getPixelFormats( const std::string& videoCodecName = "" ) const;
 	
+	/**
+	 *  Get array of sample format supported by an audio codec.
+	 *  @param audioCodecName: the audio codec name (empty if not indicated, and so get all sample formats supported by all audio codecs).
+	 */
+	std::vector<std::string> getSampleFormats( const std::string& audioCodecName = "" ) const;
+	
 private:
 	/**
 	 * @brief: load array of Option depending on the flags.
