IOutputStream: add getNbFrames function

Clement Champetier · Clement Champetier · commit e35d620a7b9a · 2015-06-08T12:40:35.000+02:00
diff --git a/src/AvTranscoder/stream/IOutputStream.hpp b/src/AvTranscoder/stream/IOutputStream.hpp
@@ -26,6 +26,7 @@ class AvExport IOutputStream
 
 	virtual size_t getStreamIndex() const = 0;
 	virtual double getStreamDuration() const = 0;
+	virtual size_t getNbFrames() const = 0;
 
 	virtual EWrappingStatus wrap( const CodedData& data ) = 0;
 };
diff --git a/src/AvTranscoder/stream/OutputStream.cpp b/src/AvTranscoder/stream/OutputStream.cpp
@@ -25,6 +25,12 @@ double OutputStream::getStreamDuration() const
 #endif
 }
 
+size_t OutputStream::getNbFrames() const
+{
+	AVStream& outputStream = _outputFile->getFormatContext().getAVStream( _streamIndex );
+	return outputStream.nb_frames;
+}
+
 IOutputStream::EWrappingStatus OutputStream::wrap( const CodedData& data )
 {
 	assert( _outputFile != NULL );
diff --git a/src/AvTranscoder/stream/OutputStream.hpp b/src/AvTranscoder/stream/OutputStream.hpp
@@ -15,6 +15,7 @@ class AvExport OutputStream : public IOutputStream
 
 	size_t getStreamIndex() const { return _streamIndex; }
 	double getStreamDuration() const;
+	size_t getNbFrames() const;
 
 	IOutputStream::EWrappingStatus wrap( const CodedData& data );
 

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,12 @@ double OutputStream::getStreamDuration() const`
`25`	`25`	`#endif`
`26`	`26`	`}`
`27`	`27`
	`28`	`+size_t OutputStream::getNbFrames() const`
	`29`	`+{`
	`30`	`+ AVStream& outputStream = _outputFile->getFormatContext().getAVStream( _streamIndex );`
	`31`	`+ return outputStream.nb_frames;`
	`32`	`+}`
	`33`	`+`
`28`	`34`	`IOutputStream::EWrappingStatus OutputStream::wrap( const CodedData& data )`
`29`	`35`	`{`
`30`	`36`	`assert( _outputFile != NULL );`