Added runtime bitrate and fps control

ReferenceType · Sep 9, 2023 · 815b4aa · 815b4aa
1 parent 361d593
commit 815b4aa
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 8 deletions.
diff --git a/EncoderTest/Program.cs b/EncoderTest/Program.cs
@@ -52,12 +52,14 @@ static void Main(string[] args)
                     foreach (var frame in frames)
                     {
 
-                        byte[] b = frame.ToByteArray();
-                        frame.CopyTo(buffer, 0);
+                        //byte[] b = frame.ToByteArray();
+                        //frame.CopyTo(buffer, 0);
                         Decode(frame.Data, frame.Length, frame.Type);
                     }
 
                 }
+                encoder.SetMaxBitrate((1+i)*100000);
+                encoder.SetTargetFps(22);
             }
            Console.WriteLine("\n Time: "+sw.ElapsedMilliseconds);
 

diff --git a/H264Sharp/Decoder.cpp b/H264Sharp/Decoder.cpp
@@ -8,7 +8,6 @@
 using namespace System::Drawing;
 using namespace System::Drawing::Imaging;
 using namespace System::Runtime::InteropServices;
-using namespace std::chrono;
 
 namespace H264Sharp {
 

diff --git a/H264Sharp/Encoder.cpp b/H264Sharp/Encoder.cpp
@@ -5,7 +5,6 @@
 #include <iostream>
 #include "Encoder.h"
 
-using namespace std::chrono;
 using namespace System::Drawing;
 using namespace System::Drawing::Imaging;
 using namespace System::Runtime::InteropServices;
@@ -323,9 +322,9 @@ namespace H264Sharp {
 
 	}
 
-	bool Encoder::Encode(array<Byte>^ i420, [Out]array<EncodedFrame^>^% frame)
+	bool Encoder::Encode(array<Byte>^ i420, int startIndex, [Out]array<EncodedFrame^>^% frame)
 	{
-		pin_ptr<Byte> ptr = &i420[0];
+		pin_ptr<Byte> ptr = &i420[startIndex];
 		bool res = Encode(innerBuffer, frame);
 
 		ptr = nullptr; // unpin
@@ -381,6 +380,23 @@ namespace H264Sharp {
 		return encoder->ForceIntraFrame(true);
 	}
 
+	void H264Sharp::Encoder::SetMaxBitrate(int target)
+	{
+		SBitrateInfo param;
+
+		memset(&param, 0, sizeof(SBitrateInfo));
+		param.iBitrate = target;
+		param.iLayer = SPATIAL_LAYER_ALL;
+		encoder->SetOption(ENCODER_OPTION_MAX_BITRATE, &param);
+		encoder->SetOption(ENCODER_OPTION_BITRATE, &param);
+	}
+	void H264Sharp::Encoder::SetTargetFps(float target)
+	{
+
+		encoder->SetOption(ENCODER_OPTION_FRAME_RATE, &target);
+	}
+
+
 	Encoder::~Encoder()
 	{
 		this->!Encoder();

diff --git a/H264Sharp/Encoder.h b/H264Sharp/Encoder.h
@@ -25,12 +25,13 @@ namespace H264Sharp {
 			bool Encode(BgraImage ^bgra, array<EncodedFrame^>^% frame);
 			bool Encode(RgbImage ^rgb, array<EncodedFrame^>^% frame);
 			bool Encode(RgbaImage ^rgba, array<EncodedFrame^>^% frame);
-			bool Encode(array<Byte> ^i420, array<EncodedFrame^>^% frame);
+			bool Encode(array<Byte> ^i420, int startIndex, array<EncodedFrame^>^% frame);
 			bool Encode(unsigned char* i420, array<EncodedFrame^>^% frame);
 			bool Encode(IntPtr^ i420, array<EncodedFrame^>^% frame);
 
 			int ForceIntraFrame();
-
+			void SetMaxBitrate(int target);
+			void SetTargetFps(float target);
 
 	private:
 		int buffer_size;