VideoEncoder: refactored encodeFrame method

Clement Champetier · Clement Champetier · commit bd2cf1528c64 · 2016-04-21T14:54:18.000+02:00
Add private method encode.
diff --git a/src/AvTranscoder/encoder/VideoEncoder.cpp b/src/AvTranscoder/encoder/VideoEncoder.cpp
@@ -109,9 +109,6 @@ bool VideoEncoder::encodeFrame(const Frame& sourceFrame, CodedData& codedFrame)
     AVCodecContext& avCodecContext = _codec.getAVCodecContext();
 
     AVPacket& packet = codedFrame.getAVPacket();
-    packet.data = NULL;
-    packet.stream_index = 0;
-
     if((avCodecContext.coded_frame) && (avCodecContext.coded_frame->pts != (int)AV_NOPTS_VALUE))
     {
         packet.pts = avCodecContext.coded_frame->pts;
@@ -122,51 +119,38 @@ bool VideoEncoder::encodeFrame(const Frame& sourceFrame, CodedData& codedFrame)
         packet.flags |= AV_PKT_FLAG_KEY;
     }
 
-#if LIBAVCODEC_VERSION_MAJOR > 53
-    int gotPacket = 0;
-    int ret = avcodec_encode_video2(&avCodecContext, &packet, &sourceFrame.getAVFrame(), &gotPacket);
-    if(ret != 0)
-    {
-        throw std::runtime_error("Encode video frame error: avcodec encode video frame - " +
-                                 getDescriptionFromErrorCode(ret));
-    }
-    return ret == 0 && gotPacket == 1;
-#else
-    int ret = avcodec_encode_video(&avCodecContext, packet.data, packet.size, &sourceFrame.getAVFrame());
-    if(ret < 0)
-    {
-        throw std::runtime_error("Encode video frame error: avcodec encode video frame - " +
-                                 getDescriptionFromErrorCode(ret));
-    }
-    return ret == 0;
-#endif
+    return encode(&sourceFrame.getAVFrame(), packet);
 }
 
 bool VideoEncoder::encodeFrame(CodedData& codedFrame)
 {
-    AVCodecContext& avCodecContext = _codec.getAVCodecContext();
+    return encode(NULL, codedFrame.getAVPacket());
+}
 
-    AVPacket& packet = codedFrame.getAVPacket();
-    packet.data = NULL;
-    packet.stream_index = 0;
+bool VideoEncoder::encode(const AVFrame* decodedData, AVPacket& encodedData)
+{
+    encodedData.data = NULL;
+    encodedData.stream_index = 0;
 
+    AVCodecContext& avCodecContext = _codec.getAVCodecContext();
 #if LIBAVCODEC_VERSION_MAJOR > 53
     int gotPacket = 0;
-    int ret = avcodec_encode_video2(&avCodecContext, &packet, NULL, &gotPacket);
+    int ret = avcodec_encode_video2(&avCodecContext, &encodedData, decodedData, &gotPacket);
     if(ret != 0)
     {
-        throw std::runtime_error("Encode video frame error: avcodec encode last video frame - " +
+        throw std::runtime_error("Encode video frame error: avcodec encode video frame - " +
                                  getDescriptionFromErrorCode(ret));
     }
     return ret == 0 && gotPacket == 1;
 #else
-    int ret = avcodec_encode_video(&avCodecContext, packet.data, packet.size, NULL);
+    int ret = avcodec_encode_video(&avCodecContext, encodedData.data, encodedData.size, decodedData);
     if(ret < 0)
     {
-        throw std::runtime_error("Encode video frame error: avcodec encode last video frame - " +
+        throw std::runtime_error("Encode video frame error: avcodec encode video frame - " +
                                  getDescriptionFromErrorCode(ret));
     }
     return ret == 0;
 #endif
 }
+
 }
diff --git a/src/AvTranscoder/encoder/VideoEncoder.hpp b/src/AvTranscoder/encoder/VideoEncoder.hpp
@@ -24,6 +24,9 @@ class AvExport VideoEncoder : public IEncoder
     ICodec& getCodec() { return _codec; }
     VideoCodec& getVideoCodec() { return _codec; }
 
+private:
+    bool encode(const AVFrame* decodedData, AVPacket& encodedData);
+
 private:
     VideoCodec _codec;
 };
