InputVideo: add setProfile

Clement Champetier · Clement Champetier · commit aa7c90b48dd5 · 2014-07-28T15:18:22.000+02:00
Can set the profile of an InputStream.
diff --git a/src/AvTranscoder/EssenceStream/InputVideo.cpp b/src/AvTranscoder/EssenceStream/InputVideo.cpp
@@ -14,6 +14,7 @@ extern "C" {
 #include <AvTranscoder/EssenceStructures/VideoFrame.hpp>
 
 #include <stdexcept>
+#include <iostream>
 
 namespace avtranscoder
 {
@@ -146,4 +147,26 @@ void InputVideo::flushDecoder()
 	avcodec_flush_buffers( _codecContext );
 }
 
+void InputVideo::setProfile( const Profile::ProfileDesc& desc )
+{
+	VideoDesc videoDesc = CodedDesc( *_codec, *_codecContext );
+
+	for( Profile::ProfileDesc::const_iterator it = desc.begin(); it != desc.end(); ++it )
+	{
+		if( (*it).first == Profile::avProfileIdentificator ||
+			(*it).first == Profile::avProfileIdentificatorHuman ||
+			(*it).first == Profile::avProfileType )
+			continue;
+
+		try
+		{
+			videoDesc.set( (*it).first, (*it).second );
+		}
+		catch( std::exception& e )
+		{
+			std::cout << "warning: " << e.what() << std::endl;
+		}
+	}
+}
+
 }
diff --git a/src/AvTranscoder/EssenceStream/InputVideo.hpp b/src/AvTranscoder/EssenceStream/InputVideo.hpp
@@ -3,6 +3,8 @@
 
 #include "InputEssence.hpp"
 
+#include <AvTranscoder/Profile.hpp>
+
 class AVCodec;
 class AVCodecContext;
 class AVFrame;
@@ -24,7 +26,9 @@ class AvExport InputVideo : public InputEssence
 	bool readNextFrame( Frame& frameBuffer, const size_t subStreamIndex );
 
 	void flushDecoder();
-
+	
+	void setProfile( const Profile::ProfileDesc& desc );
+	
 private:
 	AvInputStream*     _inputStream;
 	AVCodec*           _codec;
