OptionLoader: add getters for short/long names of format

Clement Champetier · Clement Champetier · commit 2489b7f608d3 · 2014-06-16T13:42:26.000+02:00
diff --git a/src/AvTranscoder/OptionLoader.cpp b/src/AvTranscoder/OptionLoader.cpp
@@ -20,6 +20,8 @@ OptionLoader::OptionLoader()
 	, m_avCodecContext( NULL )
 	, m_outputFormat( NULL )
 	, m_codec( NULL )
+	, m_formatsLongNames()
+	, m_formatsShortNames()
 {
 	// Alloc format context
 	m_avFormatContext = avformat_alloc_context();
@@ -33,6 +35,21 @@ OptionLoader::OptionLoader()
 	AVCodec* avCodec = NULL;
 	m_avCodecContext = avcodec_alloc_context3( avCodec );
 #endif
+	
+	// fill format short and long names
+	AVOutputFormat* fmt = NULL;
+	while( ( fmt = av_oformat_next( fmt ) ) )
+	{
+		// add only format with video track
+		if( fmt->video_codec != AV_CODEC_ID_NONE )
+		{
+			if( fmt->long_name )
+			{
+				m_formatsLongNames.push_back( std::string( fmt->long_name ) + std::string( " (" ) + std::string( fmt->name ) + std::string( ")" ) );
+				m_formatsShortNames.push_back( std::string( fmt->name ) );
+			}
+		}
+	}
 }
 
 OptionLoader::~OptionLoader()
diff --git a/src/AvTranscoder/OptionLoader.hpp b/src/AvTranscoder/OptionLoader.hpp
@@ -51,6 +51,9 @@ class OptionLoader
 	OptionMap loadOutputFormatOptions();
 	OptionMap loadVideoCodecOptions();
 	OptionMap loadAudioCodecOptions();
+	
+	std::vector<std::string>& getFormatsLongNames() { return m_formatsLongNames; }
+	std::vector<std::string>& getFormatsShortNames() { return m_formatsShortNames; }
 private:
 	/**
 	 * @brief: load array of Option depending on the flags.
@@ -63,6 +66,9 @@ class OptionLoader
 	
 	AVOutputFormat* m_outputFormat;
 	AVCodec* m_codec;
+	
+	std::vector<std::string> m_formatsLongNames;
+	std::vector<std::string> m_formatsShortNames;
 
 };
 
