VideoProperties: add analyseFull private method which decodes all the stream

Clement Champetier · Clement Champetier · commit 525f0aa8f143 · 2016-11-02T18:45:36.000+01:00
And compute the exact number of frames.
diff --git a/src/AvTranscoder/properties/VideoProperties.cpp b/src/AvTranscoder/properties/VideoProperties.cpp
@@ -28,6 +28,7 @@ VideoProperties::VideoProperties(const FileProperties& fileProperties, const siz
     , _isTopFieldFirst(false)
     , _gopSize(0)
     , _gopStructure()
+    , _nbFrames(0)
     , _firstGopTimeCode(-1)
 {
     if(_codecContext)
@@ -38,6 +39,8 @@ VideoProperties::VideoProperties(const FileProperties& fileProperties, const siz
 
     if(_levelAnalysis == eAnalyseLevelFirstGop)
         analyseGopStructure(progress);
+    else if(_levelAnalysis == eAnalyseLevelFull)
+        analyseFull(progress);
 }
 
 std::string VideoProperties::getProfileName() const
@@ -365,6 +368,9 @@ size_t VideoProperties::getNbFrames() const
     size_t nbFrames = _formatContext->streams[_streamIndex]->nb_frames;
     if(nbFrames == 0)
     {
+        if(_levelAnalysis == eAnalyseLevelFull)
+            return _nbFrames;
+
         LOG_WARN("The number of frames in the stream '" << _streamIndex << "' of file '" << _formatContext->filename
                                                         << "' is unknown.")
         const float duration = getDuration();
@@ -554,6 +560,38 @@ void VideoProperties::analyseGopStructure(IProgress& progress)
     }
 }
 
+void VideoProperties::analyseFull(IProgress& progress)
+{
+    analyseGopStructure(progress);
+
+    if(! _formatContext || ! _codecContext || ! _codec)
+        return;
+    if(! _codecContext->width || ! _codecContext->height)
+        return;
+
+    InputFile& file = const_cast<InputFile&>(_fileProperties->getInputFile());
+    // Get the stream
+    IInputStream& stream = file.getStream(_streamIndex);
+    stream.activate();
+    // Create a decoder
+    VideoDecoder decoder(static_cast<InputStream&>(stream));
+
+    VideoFrame frame(VideoFrameDesc(getWidth(), getHeight(), getPixelProperties().getAVPixelFormat()));
+    while(decoder.decodeNextFrame(frame))
+    {
+        ++_nbFrames;
+    }
+
+    // Returns at the beginning of the stream
+    file.seekAtFrame(0, AVSEEK_FLAG_BYTE);
+
+    // Check GOP size
+    if(_nbFrames <= 0)
+    {
+        throw std::runtime_error("Invalid number of frames when decoding the video stream.");
+    }
+}
+
 PropertyVector& VideoProperties::fillVector(PropertyVector& data) const
 {
     // Add properties of base class
diff --git a/src/AvTranscoder/properties/VideoProperties.hpp b/src/AvTranscoder/properties/VideoProperties.hpp
@@ -101,11 +101,15 @@ class AvExport VideoProperties : public StreamProperties
 
 private:
     /**
-     *  @brief frame type / is key frame
-     *  @param progress: callback to get analysis progression
+     * @param progress: callback to get analysis progression
      */
     void analyseGopStructure(IProgress& progress);
 
+    /**
+     * @param progress: callback to get analysis progression
+     */
+    void analyseFull(IProgress& progress);
+
 #ifndef SWIG
     template <typename T>
     void addProperty(PropertyVector& dataVector, const std::string& key, T (VideoProperties::*getter)(void) const) const
@@ -133,13 +137,20 @@ class AvExport VideoProperties : public StreamProperties
     PixelProperties _pixelProperties;
 
     //@{
-    // Can acces these data when analyse first gop
+    // @brief Can access these data when analysing the first GOP.
+    // @see eAnalyseLevelFirstGOP
     bool _isInterlaced;
     bool _isTopFieldFirst;
     size_t _gopSize;
     std::vector<std::pair<char, int> > _gopStructure; ///< picture type, encoded frame size in bytes
                                                       //@}
 
+    //@{
+    // @brief Can access these data when analysing all the stream.
+    // @see eAnalyseLevelFull
+    size_t _nbFrames;
+    //}
+
     /**
      * @brief GOP timecode of the first frame
      * @note  AVCodecContext stores the GOP timecode of the last decoded frame
