Implement a bitrate option for the video encoder class

dimula73 · dimula73 · commit feed9359c7fd · 2016-06-08T22:52:21.000+03:00
You can just pass this option in a codec profile and it will
be set up in the codec context correctly.
diff --git a/src/AvTranscoder/encoder/VideoEncoder.cpp b/src/AvTranscoder/encoder/VideoEncoder.cpp
@@ -9,6 +9,16 @@ extern "C" {
 #include <stdexcept>
 #include <cstdlib>
 
+
+template <class T>
+bool from_string(T &t,
+                 const std::string &s,
+                 std::ios_base & (*f)(std::ios_base&))
+{
+   std::istringstream iss(s);
+   return !(iss>>f>>t).fail();
+}
+
 namespace avtranscoder
 {
 
@@ -59,7 +69,7 @@ void VideoEncoder::setupEncoder(const ProfileLoader::Profile& profile)
            (*it).first == constants::avProfileType || (*it).first == constants::avProfileCodec ||
            (*it).first == constants::avProfileWidth || (*it).first == constants::avProfileHeight ||
            (*it).first == constants::avProfilePixelFormat || (*it).first == constants::avProfileFrameRate ||
-           (*it).first == constants::avProfileThreads)
+           (*it).first == constants::avProfileThreads || (*it).first == constants::avProfileBitRate)
             continue;
 
         try
@@ -80,6 +90,17 @@ void VideoEncoder::setupEncoder(const ProfileLoader::Profile& profile)
         encoderFlags |= CODEC_FLAG_PSNR;
     }
     _codec.getAVCodecContext().flags |= encoderFlags;
+
+    if(profile.count(constants::avProfileBitRate)) {
+
+        int64_t bitRate = 0;
+        bool bconvert = from_string<int64_t>(bitRate, profile.at(constants::avProfileBitRate), std::dec);
+
+        if (bconvert) {
+            _codec.getAVCodecContext().bit_rate = bitRate;
+        }
+    }
+
     _codec.openCodec();
 
     // after open encoder, set specific encoder options
@@ -89,7 +110,7 @@ void VideoEncoder::setupEncoder(const ProfileLoader::Profile& profile)
            (*it).first == constants::avProfileType || (*it).first == constants::avProfileCodec ||
            (*it).first == constants::avProfileWidth || (*it).first == constants::avProfileHeight ||
            (*it).first == constants::avProfilePixelFormat || (*it).first == constants::avProfileFrameRate ||
-           (*it).first == constants::avProfileThreads)
+           (*it).first == constants::avProfileThreads || (*it).first == constants::avProfileBitRate)
             continue;
 
         try
diff --git a/src/AvTranscoder/profile/ProfileLoader.hpp b/src/AvTranscoder/profile/ProfileLoader.hpp
@@ -24,6 +24,7 @@ const std::string avProfileCodec = "codec";
 const std::string avProfilePixelFormat = "pix_fmt";
 const std::string avProfileSampleFormat = "sample_fmt";
 const std::string avProfileFrameRate = "r";
+const std::string avProfileBitRate = "bit_rate";
 const std::string avProfileWidth = "width";
 const std::string avProfileHeight = "height";
 const std::string avProfileSampleRate = "ar";
