avfilter/scale_npp: fix logic used in previous patch

7 years ago · 62b75537db
parent f4ebbda566
commit 62b75537db
1 changed files with 5 additions and 1 deletions
--- a/libavfilter/vf_scale_npp.c
+++ b/libavfilter/vf_scale_npp.c
@ -319,7 +319,11 @@ static int init_processing_chain(AVFilterContext *ctx, int in_width, int in_heig
        last_stage = i;
    }

-    ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(s->stages[last_stage > 0 ? last_stage : 0].frames_ctx);
+    if (last_stage < 0)
+        ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(ctx->inputs[0]->hw_frames_ctx);
+    else
+        ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(s->stages[last_stage].frames_ctx);
+
    if (!ctx->outputs[0]->hw_frames_ctx)
        return AVERROR(ENOMEM);