transcoder: add ProcessStat class, returned by Transcoder::process method

Clement Champetier · Clement Champetier · commit 9d475b83859f · 2015-06-08T12:41:41.000+02:00
diff --git a/src/AvTranscoder/transcoder/ProcessStat.cpp b/src/AvTranscoder/transcoder/ProcessStat.cpp
@@ -0,0 +1,13 @@
+#include "ProcessStat.hpp"
+
+#include <utility>
+
+namespace avtranscoder
+{
+
+void ProcessStat::addVideoStat( const size_t streamIndex, const VideoStat& videoStat )
+{
+	_videoStats.insert( std::make_pair( streamIndex, videoStat ) );
+}
+
+}
diff --git a/src/AvTranscoder/transcoder/ProcessStat.hpp b/src/AvTranscoder/transcoder/ProcessStat.hpp
@@ -0,0 +1,58 @@
+#ifndef  _AV_TRANSCODER_PROCESSSTAT_HPP
+#define	 _AV_TRANSCODER_PROCESSSTAT_HPP
+
+#include <AvTranscoder/common.hpp>
+
+#include <map>
+
+namespace avtranscoder
+{
+
+/**
+ * @brief Statistic related to a video stream.
+ */
+class VideoStat
+{
+public:
+	VideoStat( const double duration, const size_t nbFrames )
+	: _duration( duration )
+	, _nbFrames( nbFrames )
+	, _quality( 0 )
+	, _psnr( 0 )
+	{}
+
+public:
+	static double psnr( double d )
+	{
+		return -10.0 * log(d) / log(10.0);
+	}
+
+public:
+	double _duration;
+	size_t _nbFrames;
+	size_t _quality;  ///< Between 1 (good) and FF_LAMBDA_MAX (bad). 0 if unknown.
+	double _psnr;  ///< 0 if unknown.
+};
+
+/**
+ * @brief ProcessStat contains statistic results given after the process.
+ * @see Transcoder::process methods
+ */
+class ProcessStat
+{
+public:
+	ProcessStat()
+	: _videoStats()
+	{}
+
+	void addVideoStat( const size_t streamIndex, const VideoStat& videoStat );
+
+	VideoStat& getVideoStat( const size_t streamIndex ) { return _videoStats.at(streamIndex); }
+
+private:
+	std::map<size_t, VideoStat> _videoStats;  ///< Key: streamIndex, Value: statistic video results
+};
+
+}
+
+#endif
diff --git a/src/AvTranscoder/transcoder/Transcoder.cpp b/src/AvTranscoder/transcoder/Transcoder.cpp
@@ -221,13 +221,13 @@ bool Transcoder::processFrame()
 	return true;
 }
 
-void Transcoder::process()
+ProcessStat Transcoder::process()
 {
 	NoDisplayProgress progress;
-	process( progress );
+	return process( progress );
 }
 
-void Transcoder::process( IProgress& progress )
+ProcessStat Transcoder::process( IProgress& progress )
 {
 	if( _streamTranscoders.size() == 0 )
 		throw std::runtime_error( "missing input streams in transcoder" );
@@ -266,6 +266,12 @@ void Transcoder::process( IProgress& progress )
 	_outputFile.endWrap();
 
 	LOG_INFO( "End of process" )
+
+	LOG_INFO( "Get process statistics" )
+	ProcessStat processStat;
+	fillProcessStat( processStat );
+
+	return processStat;
 }
 
 void Transcoder::setProcessMethod( const EProcessMethod eProcessMethod, const size_t indexBasedStream, const double outputDuration )
@@ -511,4 +517,29 @@ void Transcoder::manageSwitchToGenerator()
 	}
 }
 
+void Transcoder::fillProcessStat( ProcessStat& processStat )
+{
+	for( size_t streamIndex = 0; streamIndex < _streamTranscoders.size(); ++streamIndex )
+	{
+		IOutputStream& stream = _streamTranscoders.at( streamIndex )->getOutputStream();
+		AVCodecContext& encoderContext = _streamTranscoders.at( streamIndex )->getEncoder().getCodec().getAVCodecContext();
+		switch( encoderContext.codec_type )
+		{
+			case AVMEDIA_TYPE_VIDEO:
+			{
+				VideoStat videoStat( stream.getStreamDuration(), stream.getNbFrames() );
+				if( encoderContext.coded_frame && ( encoderContext.flags & CODEC_FLAG_PSNR) )
+				{
+					videoStat._quality = encoderContext.coded_frame->quality;
+					videoStat._psnr = VideoStat::psnr(encoderContext.coded_frame->error[0] / (encoderContext.width * encoderContext.height * 255.0 * 255.0));
+				}
+				processStat.addVideoStat( streamIndex, videoStat );
+				break;
+			}
+			default:
+				break;
+		}
+	}
+}
+
 }
diff --git a/src/AvTranscoder/transcoder/Transcoder.hpp b/src/AvTranscoder/transcoder/Transcoder.hpp
@@ -6,6 +6,7 @@
 #include <AvTranscoder/file/IOutputFile.hpp>
 #include <AvTranscoder/stream/IInputStream.hpp>
 #include <AvTranscoder/profile/ProfileLoader.hpp>
+#include <AvTranscoder/transcoder/ProcessStat.hpp>
 
 #include "StreamTranscoder.hpp"
 
@@ -119,10 +120,11 @@ class AvExport Transcoder
 	 * @brief Process all the streams, and ended the process depending on the transcode politic.
 	 * @note The function manages all process: init(), beginWrap(), processFrame()s, and endWrap().
 	 * @param progress: choose a progress, or create your own in C++ or in bindings by inherit IProgress class.
+	 * @return ProcessStat: object with statistics of the process for each stream.
 	 * @see IProgress
 	 */
-	void process( IProgress& progress );
-	void process();  ///< Call process with no display of progression
+	ProcessStat process( IProgress& progress );
+	ProcessStat process();  ///< Call process with no display of progression
 
 	/**
 	 * @brief Return the list of streams added to the transcoder.
@@ -188,6 +190,11 @@ class AvExport Transcoder
          */
 	void manageSwitchToGenerator();
 
+	/**
+	 * @brief Fill the given ProcessStat to summarize the process.
+	 */
+	void fillProcessStat( ProcessStat& processStat );
+
 private:
 	IOutputFile& _outputFile;  ///< The output media file after process (has link)
 	std::vector< InputFile* > _inputFiles;  ///< The list of input files which contain added streams (has ownership)
