stat: add AudioStat class

Clement Champetier · Clement Champetier · commit 7339f506c806 · 2015-06-08T16:48:13.000+02:00
diff --git a/src/AvTranscoder/stat/AudioStat.hpp b/src/AvTranscoder/stat/AudioStat.hpp
@@ -0,0 +1,27 @@
+#ifndef  _AV_TRANSCODER_AUDIOSTAT_HPP
+#define  _AV_TRANSCODER_AUDIOSTAT_HPP
+
+#include <AvTranscoder/common.hpp>
+
+namespace avtranscoder
+{
+
+/**
+ * @brief Statistics related to an audio stream.
+ */
+class AvExport AudioStat
+{
+public:
+	AudioStat( const double duration, const size_t nbPackets )
+	: _duration( duration )
+	, _nbPackets( nbPackets )
+	{}
+
+public:
+	double _duration;
+	size_t _nbPackets;
+};
+
+}
+
+#endif
diff --git a/src/AvTranscoder/stat/ProcessStat.cpp b/src/AvTranscoder/stat/ProcessStat.cpp
@@ -10,4 +10,9 @@ void ProcessStat::addVideoStat( const size_t streamIndex, const VideoStat& video
 	_videoStats.insert( std::make_pair( streamIndex, videoStat ) );
 }
 
+void ProcessStat::addAudioStat( const size_t streamIndex, const AudioStat& audioStat )
+{
+	_audioStats.insert( std::make_pair( streamIndex, audioStat ) );
+}
+
 }
diff --git a/src/AvTranscoder/stat/ProcessStat.hpp b/src/AvTranscoder/stat/ProcessStat.hpp
@@ -3,6 +3,7 @@
 
 #include <AvTranscoder/common.hpp>
 #include <AvTranscoder/stat/VideoStat.hpp>
+#include <AvTranscoder/stat/AudioStat.hpp>
 
 #include <map>
 
@@ -21,11 +22,14 @@ class AvExport ProcessStat
 	{}
 
 	void addVideoStat( const size_t streamIndex, const VideoStat& videoStat );
+	void addAudioStat( const size_t streamIndex, const AudioStat& audioStat );
 
 	VideoStat& getVideoStat( const size_t streamIndex ) { return _videoStats.at(streamIndex); }
+	AudioStat& getAudioStat( const size_t streamIndex ) { return _audioStats.at(streamIndex); }
 
 private:
 	std::map<size_t, VideoStat> _videoStats;  ///< Key: streamIndex, Value: statistic video results
+	std::map<size_t, AudioStat> _audioStats;  ///< Key: streamIndex, Value: statistic audio results
 };
 
 }
diff --git a/src/AvTranscoder/stream/OutputStream.hpp b/src/AvTranscoder/stream/OutputStream.hpp
@@ -15,7 +15,7 @@ class AvExport OutputStream : public IOutputStream
 
 	size_t getStreamIndex() const { return _streamIndex; }
 	double getStreamDuration() const;
-	size_t getNbFrames() const;
+	size_t getNbFrames() const;  ///< If audio stream, returns number of packets
 
 	IOutputStream::EWrappingStatus wrap( const CodedData& data );
 
diff --git a/src/AvTranscoder/transcoder/Transcoder.cpp b/src/AvTranscoder/transcoder/Transcoder.cpp
@@ -537,6 +537,12 @@ void Transcoder::fillProcessStat( ProcessStat& processStat )
 				processStat.addVideoStat( streamIndex, videoStat );
 				break;
 			}
+			case AVMEDIA_TYPE_AUDIO:
+			{
+				AudioStat audioStat( stream.getStreamDuration(), stream.getNbFrames() );
+				processStat.addAudioStat( streamIndex, audioStat );
+				break;
+			}
 			default:
 				break;
 		}

Original file line number	Diff line number	Diff line change
`@@ -10,4 +10,9 @@ void ProcessStat::addVideoStat( const size_t streamIndex, const VideoStat& video`
`10`	`10`	`_videoStats.insert( std::make_pair( streamIndex, videoStat ) );`
`11`	`11`	`}`
`12`	`12`
	`13`	`+void ProcessStat::addAudioStat( const size_t streamIndex, const AudioStat& audioStat )`
	`14`	`+{`
	`15`	`+ _audioStats.insert( std::make_pair( streamIndex, audioStat ) );`
	`16`	`+}`
	`17`	`+`
`13`	`18`	`}`
Original file line number	Diff line number	Diff line change
`@@ -537,6 +537,12 @@ void Transcoder::fillProcessStat( ProcessStat& processStat )`
`537`	`537`	`processStat.addVideoStat( streamIndex, videoStat );`
`538`	`538`	`break;`
`539`	`539`	`}`
	`540`	`+ case AVMEDIA_TYPE_AUDIO:`
	`541`	`+ {`
	`542`	`+ AudioStat audioStat( stream.getStreamDuration(), stream.getNbFrames() );`
	`543`	`+ processStat.addAudioStat( streamIndex, audioStat );`
	`544`	`+ break;`
	`545`	`+ }`
`540`	`546`	`default:`
`541`	`547`	`break;`
`542`	`548`	`}`