GeneratorVideo: throw exception if no codec

Clement Champetier · Clement Champetier · commit 4ca62566b1d4 · 2014-10-31T18:48:54.000+01:00
diff --git a/src/AvTranscoder/essenceStream/GeneratorVideo.cpp b/src/AvTranscoder/essenceStream/GeneratorVideo.cpp
@@ -2,6 +2,8 @@
 
 #include <AvTranscoder/transform/VideoTransform.hpp>
 
+#include <stdexcept>
+
 namespace avtranscoder
 {
 
@@ -32,8 +34,14 @@ void GeneratorVideo::setFrame( Frame& inputFrame )
 
 bool GeneratorVideo::readNextFrame( Frame& frameBuffer )
 {
+	// Generate black image
 	if( ! _inputFrame )
 	{
+		if( ! _codec )
+		{
+			throw std::runtime_error( "Can't readNextFrame of video generator without knowing codec." );
+		}
+
 		// @todo support PAL (0 to 255) and NTFS (16 to 235)
 		int fillChar = 0;
 
@@ -55,7 +63,8 @@ bool GeneratorVideo::readNextFrame( Frame& frameBuffer )
 
 		return true;
 	}
-	
+
+	// Take image from _inputFrame
 	if( frameBuffer.getSize() != _inputFrame->getSize() )
 		frameBuffer.getBuffer().resize( _inputFrame->getSize() );
 	std::memcpy( frameBuffer.getPtr(), _inputFrame->getPtr(), _inputFrame->getSize() );
