VideoProperties: refactored the methods that use gop_size

Clement Champetier · Clement Champetier · commit a3c82098171d · 2016-05-13T11:18:39.000+02:00
* Use VideoProperties::getGopSize() instead.
* Add doc to this method.
diff --git a/src/AvTranscoder/properties/VideoProperties.cpp b/src/AvTranscoder/properties/VideoProperties.cpp
@@ -340,7 +340,7 @@ size_t VideoProperties::getBitRate() const
     avcodec_open2(_codecContext, _codec, NULL);
 
     int gotFrame = 0;
-    int count = 0;
+    size_t count = 0;
     int gopFramesSize = 0;
 
     while(!av_read_frame(const_cast<AVFormatContext*>(_formatContext), &pkt))
@@ -359,12 +359,12 @@ size_t VideoProperties::getBitRate() const
             }
         }
         av_free_packet(&pkt);
-        if(_codecContext->gop_size == count)
+        if(getGopSize() == count)
             break;
     }
 
     int bitsPerByte = 8;
-    return (gopFramesSize / _codecContext->gop_size) * bitsPerByte * getFps();
+    return (gopFramesSize / getGopSize()) * bitsPerByte * getFps();
 }
 
 size_t VideoProperties::getMaxBitRate() const
@@ -484,7 +484,7 @@ void VideoProperties::analyseGopStructure(IProgress& progress)
             avcodec_open2(_codecContext, _codec, NULL);
 
             Frame frame;
-            int count = 0;
+            size_t count = 0;
             int gotFrame = 0;
             bool stopAnalyse = false;
 
@@ -501,14 +501,14 @@ void VideoProperties::analyseGopStructure(IProgress& progress)
                         _isInterlaced = avFrame.interlaced_frame;
                         _isTopFieldFirst = avFrame.top_field_first;
                         ++count;
-                        if(progress.progress(count, _codecContext->gop_size) == eJobStatusCancel)
+                        if(progress.progress(count, getGopSize()) == eJobStatusCancel)
                             stopAnalyse = true;
                     }
                 }
 
                 av_free_packet(&pkt);
 
-                if(_codecContext->gop_size == count)
+                if(getGopSize() == count)
                 {
                     stopAnalyse = true;
                 }
diff --git a/src/AvTranscoder/properties/VideoProperties.hpp b/src/AvTranscoder/properties/VideoProperties.hpp
@@ -47,7 +47,13 @@ class AvExport VideoProperties : public StreamProperties
     size_t getTicksPerFrame() const;
     size_t getWidth() const;
     size_t getHeight() const;
+
+    /**
+     * @return the distance between two nearest I frame
+     * @note it returns 0 for intra_only
+     */
     size_t getGopSize() const;
+
     size_t getDtgActiveFormat() const;
     size_t getReferencesFrames() const;
     int getProfile() const;

Original file line number	Diff line number	Diff line change
`@@ -340,7 +340,7 @@ size_t VideoProperties::getBitRate() const`
`340`	`340`	`avcodec_open2(_codecContext, _codec, NULL);`
`341`	`341`
`342`	`342`	`int gotFrame = 0;`
`343`		`- int count = 0;`
	`343`	`+ size_t count = 0;`
`344`	`344`	`int gopFramesSize = 0;`
`345`	`345`
`346`	`346`	`while(!av_read_frame(const_cast<AVFormatContext*>(_formatContext), &pkt))`
`@@ -359,12 +359,12 @@ size_t VideoProperties::getBitRate() const`
`359`	`359`	`}`
`360`	`360`	`}`
`361`	`361`	`av_free_packet(&pkt);`
`362`		`- if(_codecContext->gop_size == count)`
	`362`	`+ if(getGopSize() == count)`
`363`	`363`	`break;`
`364`	`364`	`}`
`365`	`365`
`366`	`366`	`int bitsPerByte = 8;`
`367`		`- return (gopFramesSize / _codecContext->gop_size) * bitsPerByte * getFps();`
	`367`	`+ return (gopFramesSize / getGopSize()) * bitsPerByte * getFps();`
`368`	`368`	`}`
`369`	`369`
`370`	`370`	`size_t VideoProperties::getMaxBitRate() const`
`@@ -484,7 +484,7 @@ void VideoProperties::analyseGopStructure(IProgress& progress)`
`484`	`484`	`avcodec_open2(_codecContext, _codec, NULL);`
`485`	`485`
`486`	`486`	`Frame frame;`
`487`		`- int count = 0;`
	`487`	`+ size_t count = 0;`
`488`	`488`	`int gotFrame = 0;`
`489`	`489`	`bool stopAnalyse = false;`
`490`	`490`
`@@ -501,14 +501,14 @@ void VideoProperties::analyseGopStructure(IProgress& progress)`
`501`	`501`	`_isInterlaced = avFrame.interlaced_frame;`
`502`	`502`	`_isTopFieldFirst = avFrame.top_field_first;`
`503`	`503`	`++count;`
`504`		`- if(progress.progress(count, _codecContext->gop_size) == eJobStatusCancel)`
	`504`	`+ if(progress.progress(count, getGopSize()) == eJobStatusCancel)`
`505`	`505`	`stopAnalyse = true;`
`506`	`506`	`}`
`507`	`507`	`}`
`508`	`508`
`509`	`509`	`av_free_packet(&pkt);`
`510`	`510`
`511`		`- if(_codecContext->gop_size == count)`
	`511`	`+ if(getGopSize() == count)`
`512`	`512`	`{`
`513`	`513`	`stopAnalyse = true;`
`514`	`514`	`}`