VideoDuration: override getDuration method

Clement Champetier · Clement Champetier · commit a1cc1388c1d2 · 2016-06-20T17:06:02.000+02:00
If it is a raw stream, get the file size to compute the stream duration.
diff --git a/src/AvTranscoder/properties/VideoProperties.cpp b/src/AvTranscoder/properties/VideoProperties.cpp
@@ -2,6 +2,7 @@
 
 #include <AvTranscoder/data/decoded/Frame.hpp>
 #include <AvTranscoder/properties/util.hpp>
+#include <AvTranscoder/properties/FileProperties.hpp>
 #include <AvTranscoder/progress/NoDisplayProgress.hpp>
 
 extern "C" {
@@ -464,6 +465,22 @@ float VideoProperties::getFps() const
     return av_q2d(_formatContext->streams[_streamIndex]->avg_frame_rate);
 }
 
+float VideoProperties::getDuration() const
+{
+    const float duration = StreamProperties::getDuration();
+    if(duration == 0)
+    {
+        LOG_WARN("The duration of the stream '" << _streamIndex << "' of file '" << _formatContext->filename << "' is unknown.")
+        if(_fileProperties->isRawFormat())
+        {
+            LOG_INFO("Get the file size to compute the duration.")
+            return _fileProperties->getFileSize() / getBitRate() * 8;
+        }
+        return 0;
+    }
+    return duration;
+}
+
 bool VideoProperties::hasBFrames() const
 {
     if(!_codecContext)
diff --git a/src/AvTranscoder/properties/VideoProperties.hpp b/src/AvTranscoder/properties/VideoProperties.hpp
@@ -61,6 +61,12 @@ class AvExport VideoProperties : public StreamProperties
      */
     float getFps() const;
 
+    /**
+     * @brief Override StreamProperties method.
+     * @return The stream duration in seconds, 0 if not available
+     */
+    float getDuration() const;
+
     bool hasBFrames() const;
     // bool isClosedGop() const;
 
