From d8c37337d9ade654e29251bf6067f95ac96a243d Mon Sep 17 00:00:00 2001
From: nick black <dankamongmen@gmail.com>
Date: Fri, 2 Jul 2021 22:09:58 -0400
Subject: [PATCH] allow different backends to specify their own padding
 requirements #1722

---
 src/lib/internal.h |  1 +
 src/lib/visual.c   | 22 +++++++++++-----------
 src/media/ffmpeg.c |  1 +
 3 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/src/lib/internal.h b/src/lib/internal.h
index b287b2049..15c942f3f 100644
--- a/src/lib/internal.h
+++ b/src/lib/internal.h
@@ -1759,6 +1759,7 @@ typedef struct ncvisual_implementation {
   int (*visual_stream)(notcurses* nc, struct ncvisual* ncv, float timescale,
                        ncstreamcb streamer, const struct ncvisual_options* vopts, void* curry);
   char* (*visual_subtitle)(const struct ncvisual* ncv);
+  int rowalign; // rowstride base, can be 0 for no padding
   // do a persistent resize, changing the ncv itself
   int (*visual_resize)(struct ncvisual* ncv, int rows, int cols);
   void (*visual_destroy)(struct ncvisual* ncv);
diff --git a/src/lib/visual.c b/src/lib/visual.c
index e6c12afe8..9efa43c8b 100644
--- a/src/lib/visual.c
+++ b/src/lib/visual.c
@@ -562,16 +562,16 @@ int ncvisual_rotate(ncvisual* ncv, double rads){
   return 0;
 }
 
-// ffmpeg wants rows to be multiples of IMGALIGN (64)
-#define IMGALIGN 64
 static inline size_t
-pad_for_image(size_t stride){
-  if(stride % IMGALIGN == 0){
+pad_for_image(size_t stride, int cols){
+  if(visual_implementation.rowalign == 0){
+    return 4 * cols;
+  }else if(stride % visual_implementation.rowalign == 0){
     return stride;
   }
-  return (stride + IMGALIGN) / IMGALIGN * IMGALIGN;
+  return (stride + visual_implementation.rowalign) /
+          visual_implementation.rowalign * visual_implementation.rowalign;
 }
-#undef IMGALIGN
 
 ncvisual* ncvisual_from_rgba(const void* rgba, int rows, int rowstride, int cols){
   if(rowstride % 4){
@@ -581,7 +581,7 @@ ncvisual* ncvisual_from_rgba(const void* rgba, int rows, int rowstride, int cols
   ncvisual* ncv = ncvisual_create();
   if(ncv){
     // ffmpeg needs inputs with rows aligned on 192-byte boundaries
-    ncv->rowstride = pad_for_image(rowstride);
+    ncv->rowstride = pad_for_image(rowstride, cols);
     ncv->pixx = cols;
     ncv->pixy = rows;
     uint32_t* data = malloc(ncv->rowstride * ncv->pixy);
@@ -603,7 +603,7 @@ ncvisual* ncvisual_from_rgb_packed(const void* rgba, int rows, int rowstride,
                                    int cols, int alpha){
   ncvisual* ncv = ncvisual_create();
   if(ncv){
-    ncv->rowstride = pad_for_image(cols * 4);
+    ncv->rowstride = pad_for_image(cols * 4, cols);
     ncv->pixx = cols;
     ncv->pixy = rows;
     uint32_t* data = malloc(ncv->rowstride * ncv->pixy);
@@ -640,7 +640,7 @@ ncvisual* ncvisual_from_rgb_loose(const void* rgba, int rows, int rowstride,
   }
   ncvisual* ncv = ncvisual_create();
   if(ncv){
-    ncv->rowstride = pad_for_image(cols * 4);
+    ncv->rowstride = pad_for_image(cols * 4, cols);
     ncv->pixx = cols;
     ncv->pixy = rows;
     uint32_t* data = malloc(ncv->rowstride * ncv->pixy);
@@ -667,7 +667,7 @@ ncvisual* ncvisual_from_bgra(const void* bgra, int rows, int rowstride, int cols
   }
   ncvisual* ncv = ncvisual_create();
   if(ncv){
-    ncv->rowstride = pad_for_image(rowstride);
+    ncv->rowstride = pad_for_image(rowstride, cols);
     ncv->pixx = cols;
     ncv->pixy = rows;
     uint32_t* data = malloc(ncv->rowstride * ncv->pixy);
@@ -704,7 +704,7 @@ int ncvisual_resize(ncvisual* n, int rows, int cols){
 }
 
 int ncvisual_resize_noninterpolative(ncvisual* n, int rows, int cols){
-  size_t dstride = pad_for_image(cols * 4);
+  size_t dstride = pad_for_image(cols * 4, cols);
   uint32_t* r = resize_bitmap(n->data, n->pixy, n->pixx, n->rowstride,
                               rows, cols, dstride);
   if(r == NULL){
diff --git a/src/media/ffmpeg.c b/src/media/ffmpeg.c
index 61640e2ad..93baf9eaa 100644
--- a/src/media/ffmpeg.c
+++ b/src/media/ffmpeg.c
@@ -633,6 +633,7 @@ const ncvisual_implementation local_visual_implementation = {
   .visual_subtitle = ffmpeg_subtitle,
   .visual_resize = ffmpeg_resize,
   .visual_destroy = ffmpeg_destroy,
+  .rowalign = 64, // ffmpeg wants multiples of IMGALIGN (64)
   .canopen_images = true,
   .canopen_videos = true,
 };