Frame: added isVideo/AudioFrame methods

Clement Champetier · Clement Champetier · commit 11b72e85af29 · 2016-01-21T16:42:46.000+01:00
diff --git a/src/AvTranscoder/data/decoded/Frame.cpp b/src/AvTranscoder/data/decoded/Frame.cpp
@@ -80,4 +80,18 @@ void Frame::allocateAVFrame()
         throw std::runtime_error("Unable to allocate an empty Frame.");
     }
 }
+
+bool Frame::isAudioFrame() const
+{
+    if(_frame->sample_rate && _frame->channels && _frame->channel_layout && _frame->format != -1)
+        return true;
+    return false;
+}
+
+bool Frame::isVideoFrame() const
+{
+    if(_frame->width && _frame->height && _frame->format != -1)
+        return true;
+    return false;
+}
 }
diff --git a/src/AvTranscoder/data/decoded/Frame.hpp b/src/AvTranscoder/data/decoded/Frame.hpp
@@ -63,6 +63,18 @@ class AvExport Frame
      */
     void clear();
 
+    /**
+     * @return If it corresponds to a valid audio frame.
+     * @see AudioFrame
+     */
+    bool isAudioFrame() const;
+
+    /**
+     * @return If it corresponds to a valid video frame.
+     * @see VideoFrame
+     */
+    bool isVideoFrame() const;
+
 #ifndef SWIG
     AVFrame& getAVFrame() { return *_frame; }
     const AVFrame& getAVFrame() const { return *_frame; }

Original file line number	Diff line number	Diff line change
`@@ -80,4 +80,18 @@ void Frame::allocateAVFrame()`
`80`	`80`	`throw std::runtime_error("Unable to allocate an empty Frame.");`
`81`	`81`	`}`
`82`	`82`	`}`
	`83`	`+`
	`84`	`+bool Frame::isAudioFrame() const`
	`85`	`+{`
	`86`	`+ if(_frame->sample_rate && _frame->channels && _frame->channel_layout && _frame->format != -1)`
	`87`	`+ return true;`
	`88`	`+ return false;`
	`89`	`+}`
	`90`	`+`
	`91`	`+bool Frame::isVideoFrame() const`
	`92`	`+{`
	`93`	`+ if(_frame->width && _frame->height && _frame->format != -1)`
	`94`	`+ return true;`
	`95`	`+ return false;`
	`96`	`+}`
`83`	`97`	`}`