Fix capture incorrect alignment with cpu encoding for some window sizes

3 files changed, 8 insertions, 41 deletions

diff --git a/src/capture/kms.c b/src/capture/kms.c
index 5b39805..2601c5b 100644
--- a/src/capture/kms.c
+++ b/src/capture/kms.c
@@ -152,25 +152,8 @@ static int gsr_capture_kms_start(gsr_capture *cap, AVCodecContext *video_codec_c
     /* Disable vsync */
     self->params.egl->eglSwapInterval(self->params.egl->egl_display, 0);
 
-    // TODO: Move this and xcomposite equivalent to a common section unrelated to capture method
-    if(self->params.egl->gpu_info.vendor == GSR_GPU_VENDOR_AMD && video_codec_context->codec_id == AV_CODEC_ID_HEVC) {
-        // TODO: dont do this if using ffmpeg reports that this is not needed (AMD driver bug that was fixed recently)
-        video_codec_context->width = FFALIGN(self->capture_size.x, 64);
-        video_codec_context->height = FFALIGN(self->capture_size.y, 16);
-    } else if(self->params.egl->gpu_info.vendor == GSR_GPU_VENDOR_AMD && video_codec_context->codec_id == AV_CODEC_ID_AV1) {
-        // TODO: Dont do this for VCN 5 and forward which should fix this hardware bug
-        video_codec_context->width = FFALIGN(self->capture_size.x, 64);
-        // AMD driver has special case handling for 1080 height to set it to 1082 instead of 1088 (1080 aligned to 16).
-        // TODO: Set height to 1082 in this case, but it wont work because it will be aligned to 1088.
-        if(self->capture_size.y == 1080) {
-            video_codec_context->height = 1080;
-        } else {
-            video_codec_context->height = FFALIGN(self->capture_size.y, 16);
-        }
-    } else {
-        video_codec_context->width = FFALIGN(self->capture_size.x, 2);
-        video_codec_context->height = FFALIGN(self->capture_size.y, 2);
-    }
+    video_codec_context->width = FFALIGN(self->capture_size.x, 2);
+    video_codec_context->height = FFALIGN(self->capture_size.y, 2);
 
     frame->width = video_codec_context->width;
     frame->height = video_codec_context->height;
diff --git a/src/capture/nvfbc.c b/src/capture/nvfbc.c
index b619250..9c7a041 100644
--- a/src/capture/nvfbc.c
+++ b/src/capture/nvfbc.c
@@ -359,11 +359,11 @@ static int gsr_capture_nvfbc_start(gsr_capture *cap, AVCodecContext *video_codec
     }
 
     if(cap_nvfbc->capture_region) {
-        video_codec_context->width = cap_nvfbc->width & ~1;
-        video_codec_context->height = cap_nvfbc->height & ~1;
+        video_codec_context->width = FFALIGN(cap_nvfbc->width, 2);
+        video_codec_context->height = FFALIGN(cap_nvfbc->height, 2);
     } else {
-        video_codec_context->width = cap_nvfbc->tracking_width & ~1;
-        video_codec_context->height = cap_nvfbc->tracking_height & ~1;
+        video_codec_context->width = FFALIGN(cap_nvfbc->tracking_width, 2);
+        video_codec_context->height = FFALIGN(cap_nvfbc->tracking_height, 2);
     }
 
     frame->width = video_codec_context->width;
diff --git a/src/capture/xcomposite.c b/src/capture/xcomposite.c
index eb49d15..4f89e11 100644
--- a/src/capture/xcomposite.c
+++ b/src/capture/xcomposite.c
@@ -168,24 +168,8 @@ static int gsr_capture_xcomposite_start(gsr_capture *cap, AVCodecContext *video_
     if(self->params.region_size.x > 0 && self->params.region_size.y > 0)
         video_size = self->params.region_size;
 
-    if(self->params.egl->gpu_info.vendor == GSR_GPU_VENDOR_AMD && video_codec_context->codec_id == AV_CODEC_ID_HEVC) {
-        // TODO: dont do this if using ffmpeg reports that this is not needed (AMD driver bug that was fixed recently)
-        video_codec_context->width = FFALIGN(video_size.x, 64);
-        video_codec_context->height = FFALIGN(video_size.y, 16);
-    } else if(self->params.egl->gpu_info.vendor == GSR_GPU_VENDOR_AMD && video_codec_context->codec_id == AV_CODEC_ID_AV1) {
-        // TODO: Dont do this for VCN 5 and forward which should fix this hardware bug
-        video_codec_context->width = FFALIGN(video_size.x, 64);
-        // AMD driver has special case handling for 1080 height to set it to 1082 instead of 1088 (1080 aligned to 16).
-        // TODO: Set height to 1082 in this case, but it wont work because it will be aligned to 1088.
-        if(video_size.y == 1080) {
-            video_codec_context->height = 1080;
-        } else {
-            video_codec_context->height = FFALIGN(video_size.y, 16);
-        }
-    } else {
-        video_codec_context->width = FFALIGN(video_size.x, 2);
-        video_codec_context->height = FFALIGN(video_size.y, 2);
-    }
+    video_codec_context->width = FFALIGN(video_size.x, 2);
+    video_codec_context->height = FFALIGN(video_size.y, 2);
 
     frame->width = video_codec_context->width;
     frame->height = video_codec_context->height;