OptionLoader: add getters for short/long names of video/audio codecs

Clement Champetier · Clement Champetier · commit 4690467e77f7 · 2014-06-16T13:43:35.000+02:00
diff --git a/src/AvTranscoder/OptionLoader.cpp b/src/AvTranscoder/OptionLoader.cpp
@@ -22,6 +22,10 @@ OptionLoader::OptionLoader()
 	, m_codec( NULL )
 	, m_formatsLongNames()
 	, m_formatsShortNames()
+	, m_videoCodecsLongNames()
+	, m_videoCodecsShortNames()
+	, m_audioCodecsLongNames()
+	, m_audioCodecsShortNames()
 {
 	// Alloc format context
 	m_avFormatContext = avformat_alloc_context();
@@ -50,6 +54,42 @@ OptionLoader::OptionLoader()
 			}
 		}
 	}
+	
+	// fill video and audio codec short and long names
+	AVCodec* c = NULL;
+	while( ( c = av_codec_next( c ) ) )
+	{
+#if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT( 53, 34, 0 )
+		if( c->encode2 )
+#else
+		if( c->encode )
+#endif
+		{
+			switch( c->type )
+			{
+				case AVMEDIA_TYPE_VIDEO:
+				{
+					if( c->long_name )
+					{
+						m_videoCodecsLongNames.push_back( std::string( c->long_name ) );
+						m_videoCodecsShortNames.push_back( std::string( c->name ) );
+					}
+					break;
+				}
+				case AVMEDIA_TYPE_AUDIO:
+				{
+					if( c->long_name )
+					{
+						m_audioCodecsLongNames.push_back( std::string( c->long_name ) );
+						m_audioCodecsShortNames.push_back( std::string( c->name ) );
+					}
+					break;
+				}
+				default:
+					break;
+			}
+		}
+	}
 }
 
 OptionLoader::~OptionLoader()
diff --git a/src/AvTranscoder/OptionLoader.hpp b/src/AvTranscoder/OptionLoader.hpp
@@ -54,6 +54,13 @@ class OptionLoader
 	
 	std::vector<std::string>& getFormatsLongNames() { return m_formatsLongNames; }
 	std::vector<std::string>& getFormatsShortNames() { return m_formatsShortNames; }
+	
+	std::vector<std::string>& getVideoCodecsLongNames() { return m_videoCodecsLongNames; }
+	std::vector<std::string>& getVideoCodecsShortNames() { return m_videoCodecsShortNames; }
+	
+	std::vector<std::string>& getAudioCodecsLongNames() { return m_audioCodecsLongNames; }
+	std::vector<std::string>& getAudioCodecsShortNames() { return m_audioCodecsShortNames; }
+	
 private:
 	/**
 	 * @brief: load array of Option depending on the flags.
@@ -69,6 +76,12 @@ class OptionLoader
 	
 	std::vector<std::string> m_formatsLongNames;
 	std::vector<std::string> m_formatsShortNames;
+	
+	std::vector<std::string> m_videoCodecsLongNames;
+	std::vector<std::string> m_videoCodecsShortNames;
+
+	std::vector<std::string> m_audioCodecsLongNames;
+	std::vector<std::string> m_audioCodecsShortNames;
 
 };
 
