Merge pull request #52 from cchampet/dev_addMetadata

MarcAntoine-Arnaud · MarcAntoine-Arnaud · commit daeb051360ba · 2014-08-12T16:11:19.000+02:00
OutputFile: add addMetadata
diff --git a/src/AvTranscoder/File/OutputFile.cpp b/src/AvTranscoder/File/OutputFile.cpp
@@ -204,6 +204,24 @@ bool OutputFile::endWrap( )
 	return true;
 }
 
+void OutputFile::addMetadata( const MetadatasMap& dataMap )
+{
+	for( MetadatasMap::const_iterator it = dataMap.begin(); it != dataMap.end(); ++it )
+	{
+		addMetadata( it->first, it->second );
+	}
+}
+
+void OutputFile::addMetadata( const std::string& key, const std::string& value )
+{
+	int ret = av_dict_set( &_formatContext->metadata, key.c_str(), value.c_str(), 0 );
+	if( ret < 0 )
+	{
+		char err[250];
+		av_strerror( ret, err, 250 );
+		std::cout << err << std::endl;
+	}
+}
 
 void OutputFile::setProfile( const Profile::ProfileDesc& desc )
 {
diff --git a/src/AvTranscoder/File/OutputFile.hpp b/src/AvTranscoder/File/OutputFile.hpp
@@ -3,6 +3,8 @@
 
 #include <AvTranscoder/common.hpp>
 
+#include <AvTranscoder/Metadatas/MediaMetadatasStructures.hpp>
+
 #include <AvTranscoder/CodedStructures/DataStream.hpp>
 #include <AvTranscoder/CodedStructures/VideoDesc.hpp>
 #include <AvTranscoder/CodedStructures/AudioDesc.hpp>
@@ -98,6 +100,13 @@ class AvExport OutputFile
 	 * @param desc: the profile of the output format
 	 */
 	virtual void setProfile( const Profile::ProfileDesc& desc );
+	
+	/**
+	 * @brief Add metadata to the output file.
+	 * @note Depending on the format, you are not sure to find your metadata after the transcode.
+     */
+	virtual void addMetadata( const MetadatasMap& dataMap );
+	virtual void addMetadata( const std::string& key, const std::string& value );
 
 	virtual void setVerbose( bool verbose = false ){ _verbose = verbose; }
 
