OptionLoader: add getPixelFormats

Clement Champetier · Clement Champetier · commit a269316afe84 · 2014-06-18T18:50:14.000+02:00
* Returns the list of supported pixel formats depending on the video
codec.
diff --git a/src/AvTranscoder/OptionLoader.cpp b/src/AvTranscoder/OptionLoader.cpp
@@ -5,6 +5,8 @@ extern "C" {
 	#define __STDC_CONSTANT_MACROS
 #endif
 #include <libavutil/opt.h>
+#include <libavutil/pixfmt.h>
+#include <libavutil/pixdesc.h>
 }
 
 #include <string>
@@ -270,4 +272,33 @@ OptionLoader::OptionArray OptionLoader::loadOptions( void* av_class, int req_fla
 	return options;
 }
 
+std::vector<std::string> OptionLoader::getPixelFormats ( const std::string& videoCodecName ) const
+{
+	std::vector<std::string> pixelFormats;
+	
+	AVCodec* videoCodec = avcodec_find_encoder_by_name( videoCodecName.c_str() );
+	
+	if( videoCodec->pix_fmts == 0 )
+	{
+		for( int pix_fmt = 0; pix_fmt < PIX_FMT_NB; pix_fmt++ )
+		{
+			const AVPixFmtDescriptor* pix_desc = &av_pix_fmt_descriptors[pix_fmt];
+			if( ! pix_desc->name )
+				continue;
+			pixelFormats.push_back( std::string( pix_desc->name ) );
+		}
+		return pixelFormats;
+	}
+	
+	int i = 0;
+	while( videoCodec->pix_fmts[i] != -1 )
+	{
+		const AVPixFmtDescriptor* pix_desc = &av_pix_fmt_descriptors[ videoCodec->pix_fmts[i] ];
+		pixelFormats.push_back( std::string( pix_desc->name ) );
+		i++;
+	}
+	
+	return pixelFormats;
+}
+
 }
diff --git a/src/AvTranscoder/OptionLoader.hpp b/src/AvTranscoder/OptionLoader.hpp
@@ -61,6 +61,11 @@ class OptionLoader
 	std::vector<std::string>& getAudioCodecsLongNames() { return m_audioCodecsLongNames; }
 	std::vector<std::string>& getAudioCodecsShortNames() { return m_audioCodecsShortNames; }
 	
+	/**
+	 *  Get array of pixel format supported by video codec.
+	 */
+	std::vector<std::string> getPixelFormats( const std::string& videoCodecName ) const;
+	
 private:
 	/**
 	 * @brief: load array of Option depending on the flags.
