Use the actual plane sizes instead.

Signed-off-by: James Almer <jamr...@gmail.com>
---
 libavcodec/hevc_refs.c | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/libavcodec/hevc_refs.c b/libavcodec/hevc_refs.c
index 35b8e5e696..b0035356ac 100644
--- a/libavcodec/hevc_refs.c
+++ b/libavcodec/hevc_refs.c
@@ -22,6 +22,7 @@
  */
 
 #include "libavutil/avassert.h"
+#include "libavutil/imgutils.h"
 
 #include "thread.h"
 #include "hevc.h"
@@ -401,9 +402,16 @@ static HEVCFrame *generate_missing_ref(HEVCContext *s, int 
poc)
 
     if (!s->avctx->hwaccel) {
         if (!s->ps.sps->pixel_shift) {
-            for (i = 0; frame->frame->buf[i]; i++)
-                memset(frame->frame->buf[i]->data, 1 << (s->ps.sps->bit_depth 
- 1),
-                       frame->frame->buf[i]->size);
+            size_t sizes[4];
+            ptrdiff_t linesizes[4];
+            for (i = 0; i < 4; i++)
+                linesizes[i] = frame->frame->linesize[i];
+            if (av_image_fill_plane_sizes(sizes, frame->frame->format, 
frame->frame->height, linesizes) < 0) {
+                ff_hevc_unref_frame(s, frame, ~0);
+                return NULL;
+            }
+            for (i = 0; frame->frame->data[i]; i++)
+                memset(frame->frame->data[i], 1 << (s->ps.sps->bit_depth - 1), 
sizes[i]);
         } else {
             for (i = 0; frame->frame->data[i]; i++)
                 for (y = 0; y < (s->ps.sps->height >> s->ps.sps->vshift[i]); 
y++) {
-- 
2.35.1

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".

Reply via email to