Transcoder: add getter to a specific stream

Clement Champetier · Clement Champetier · commit 6946e6c84105 · 2014-10-31T19:18:18.000+01:00
* An external library could need to update a specific stream during
process.
* Also add a getter to the current Essence in StreamTranscoder, for the
same reason above.
diff --git a/src/AvTranscoder/transcoder/StreamTranscoder.hpp b/src/AvTranscoder/transcoder/StreamTranscoder.hpp
@@ -57,18 +57,23 @@ class AvExport StreamTranscoder
 	void switchToGeneratorEssence();
 	void switchToInputEssence();
 
-	void setVerbose( bool verbose = true ){ _verbose = verbose; }
-
-	void setInfinityStream( bool isInfinity ) { _infinityStream = isInfinity; }
-
-	void setOffset( bool offset = true ){ _offset = offset; }
-
 	/**
 	 * @brief Get the duration of the stream.
 	 * @note if it's a generated stream, return limit of double.
 	 */
 	double getDuration() const;
 
+	/**
+	 * @return a reference to the current essence processed in this stream.
+	 */
+	IInputEssence& getCurrentEssence() const { return *_currentEssence; }
+
+	void setVerbose( bool verbose = true ){ _verbose = verbose; }
+
+	void setInfinityStream( bool isInfinity ) { _infinityStream = isInfinity; }
+
+	void setOffset( bool offset = true ){ _offset = offset; }
+
 private:
 	bool processRewrap();
 	bool processRewrap( const int subStreamIndex );
diff --git a/src/AvTranscoder/transcoder/Transcoder.hpp b/src/AvTranscoder/transcoder/Transcoder.hpp
@@ -113,6 +113,12 @@ class AvExport Transcoder
 	 */
 	void process( IProgress& progress );
 
+	/**
+	 * @param streamIndex: careful about the order of stream insertion of the Transcoder.
+	 * @return a reference to a stream manage by the Transcoder.
+	 */
+	StreamTranscoder& getStream( size_t streamIndex ) const { return *_streamTranscoders.at( streamIndex ); }
+
 	/**
 	 * @brief Set the transcodage politic.
 	 * @note Call it after adding the streams.
