Merge pull request #331 from valnoel/feature/add_decoded_frames_getters

valnoel · web-flow · commit 9e1b5180b4f7 · 2023-05-02T15:17:48.000+02:00
Add getters for decoded frames
diff --git a/src/AvTranscoder/transcoder/StreamTranscoder.hpp b/src/AvTranscoder/transcoder/StreamTranscoder.hpp
@@ -100,6 +100,23 @@ class AvExport StreamTranscoder
     /// Returns a reference to the stream which wraps data
     IOutputStream& getOutputStream() const { return *_outputStream; }
 
+    /// Returns the total number of generated frames for this processed stream
+    size_t getNumberOfGeneratedFrames() const {
+        size_t generatedFrames = 0;
+        for (IDecoder* generator : _generators) {
+            generatedFrames += generator->getNbDecodedFrames();
+        }
+        return generatedFrames;
+    }
+    /// Returns the total number of decoded frames for this processed stream
+    size_t getNumberOfDecodedFrames() const {
+        size_t decodedFrames = 0;
+        for (IDecoder* inputDecoder : _inputDecoders) {
+            decodedFrames += inputDecoder->getNbDecodedFrames();
+        }
+        return decodedFrames;
+    }
+
     /**
      * @brief Returns if the stream has the ability to switch to a generator.
      */
diff --git a/src/AvTranscoder/transcoder/Transcoder.hpp b/src/AvTranscoder/transcoder/Transcoder.hpp
@@ -139,6 +139,24 @@ class AvExport Transcoder
     void setProcessMethod(const EProcessMethod eProcessMethod, const size_t indexBasedStream = 0,
                           const double outputDuration = 0);
 
+    /**
+     * @brief Returns the total number of generated frames for a specific stream
+     * @param streamIndex: the index of the stream
+     * @return The total number of generated frames
+     */
+    size_t getNumberOfGeneratedFrames(const size_t streamIndex) const {
+        return _streamTranscoders.at(streamIndex)->getNumberOfGeneratedFrames();
+    }
+
+    /**
+     * @brief Returns the total number of decoded frames for a specific stream
+     * @param streamIndex: the index of the stream
+     * @return The total number of decoded frames
+     */
+    size_t getNumberOfDecodedFrames(const size_t streamIndex) const {
+        return _streamTranscoders.at(streamIndex)->getNumberOfDecodedFrames();
+    }
+
 private:
     void addRewrapStream(const InputStreamDesc& inputStreamDesc, const float offset);
     void addTranscodeStream(const std::vector<InputStreamDesc>& inputStreamDescArray, const ProfileLoader::Profile& profile,
