DummyAudio: add setFrame

Clement Champetier · Clement Champetier · commit 3bcbc868345f · 2014-07-22T11:47:25.000+02:00
* Can create a dummyVideo, and set its Frame (like DummyVideo).
* When read the next frame of the object, read silent if no specific
Frame was set, or read the data of the Frame specified.
diff --git a/src/AvTranscoder/EssenceStream/DummyAudio.cpp b/src/AvTranscoder/EssenceStream/DummyAudio.cpp
@@ -5,6 +5,7 @@ namespace avtranscoder
 
 DummyAudio::DummyAudio( )
 	: InputEssence( )
+	, _inputFrame( NULL )
 {
 }
 
@@ -27,21 +28,33 @@ AudioDesc DummyAudio::getAudioDesc() const
 	return _audioDesc;
 }
 
+void DummyAudio::setFrame( Frame& inputFrame )
+{
+	_inputFrame = &inputFrame;
+}
+
 bool DummyAudio::readNextFrame( Frame& frameBuffer )
 {
 	frameBuffer.getBuffer().resize( _frameDesc.getDataSize() );
 
-	AudioFrame& audioFrameBuffer = static_cast<AudioFrame&>( frameBuffer );
-	audioFrameBuffer.setNbSamples( 1.0 * _frameDesc.getSampleRate() / _frameDesc.getFps() );
+	if( ! _inputFrame )
+	{
+		AudioFrame& audioFrameBuffer = static_cast<AudioFrame&>( frameBuffer );
+		audioFrameBuffer.setNbSamples( 1.0 * _frameDesc.getSampleRate() / _frameDesc.getFps() );
 
-	//av_samples_set_silence( data.getPtr(), offset, nb_samples, nb_channels, sample_fmt );
-	int fill_char = (
-		_frameDesc.getSampleFormat() == AV_SAMPLE_FMT_U8 ||
-		_frameDesc.getSampleFormat() == AV_SAMPLE_FMT_U8P
-		) ? 0x80 : 0x00;
-
-	memset( frameBuffer.getPtr(), fill_char, frameBuffer.getSize() );
+		//av_samples_set_silence( data.getPtr(), offset, nb_samples, nb_channels, sample_fmt );
+		int fill_char = (
+			_frameDesc.getSampleFormat() == AV_SAMPLE_FMT_U8 ||
+			_frameDesc.getSampleFormat() == AV_SAMPLE_FMT_U8P
+			) ? 0x80 : 0x00;
 
+		memset( frameBuffer.getPtr(), fill_char, frameBuffer.getSize() );
+		return true;
+	}
+	
+	if( frameBuffer.getSize() != _inputFrame->getSize() )
+		frameBuffer.getBuffer().resize( _inputFrame->getSize() );
+	std::memcpy( frameBuffer.getPtr(), _inputFrame->getPtr(), _inputFrame->getSize() );
 	return true;
 }
 
diff --git a/src/AvTranscoder/EssenceStream/DummyAudio.hpp b/src/AvTranscoder/EssenceStream/DummyAudio.hpp
@@ -23,13 +23,17 @@ class AvExport DummyAudio : public InputEssence
 	AudioDesc getAudioDesc() const;
 	
 	void setup() {}
+	
+	void setFrame( Frame& inputFrame );
 
 	bool readNextFrame( Frame& frameBuffer );
 	bool readNextFrame( Frame& frameBuffer, const size_t subStreamIndex );
 
 private:
 	AudioDesc      _audioDesc;
 	AudioFrameDesc _frameDesc;
+	
+	Frame*    _inputFrame;
 };
 
 }
