RPJosh
/
neatvnc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

h264-v4l2m2m: Implement key-frame coercion

h264-v4l2m2m
Andri Yngvason 2024-03-17 10:15:38 +00:00
parent fe3dc26441
commit 14a95b8620
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/h264-encoder-v4l2m2m-impl.c
View File

@ -443,6 +443,14 @@ static int alloc_src_buffers(struct h264_encoder_v4l2m2m* self)
return 0; return 0;
} }
static void force_key_frame(struct h264_encoder_v4l2m2m* self)
{
struct v4l2_control ctrl = { 0 };
ctrl.id = V4L2_CID_MPEG_VIDEO_FORCE_KEY_FRAME;
ctrl.value = 0;
ioctl(self->fd, VIDIOC_S_CTRL, &ctrl);
}
static void encode_buffer(struct h264_encoder_v4l2m2m* self, static void encode_buffer(struct h264_encoder_v4l2m2m* self,
struct nvnc_fb* fb) struct nvnc_fb* fb)
{ {
@ -485,6 +493,10 @@ static void encode_buffer(struct h264_encoder_v4l2m2m* self,
srcbuf->buffer.timestamp.tv_sec = fb->pts / UINT64_C(1000000); srcbuf->buffer.timestamp.tv_sec = fb->pts / UINT64_C(1000000);
srcbuf->buffer.timestamp.tv_usec = fb->pts % UINT64_C(1000000); srcbuf->buffer.timestamp.tv_usec = fb->pts % UINT64_C(1000000);
if (self->base.next_frame_should_be_keyframe)
force_key_frame(self);
self->base.next_frame_should_be_keyframe = false;
int rc = v4l2_qbuf(self->fd, &srcbuf->buffer); int rc = v4l2_qbuf(self->fd, &srcbuf->buffer);
if (rc < 0) { if (rc < 0) {
nvnc_log(NVNC_LOG_PANIC, "Failed to enqueue buffer: %m"); nvnc_log(NVNC_LOG_PANIC, "Failed to enqueue buffer: %m");