@@ -52,6 +52,26 @@ static int rkvdec_get_valid_fmt(struct rkvdec_ctx *ctx)
return ctx->coded_fmt_desc->decoded_fmts[0];
}
+static int rkvdec_sps_check(struct rkvdec_ctx *ctx, struct v4l2_pix_format_mplane *pix_mp)
+{
+ const struct rkvdec_coded_fmt_desc *coded_desc = ctx->coded_fmt_desc;
+
+ /*
+ * When a codec doesn't implement the SPS check, handle it as if no
+ * SPS exists for the codec.
+ */
+ if (coded_desc->ops->sps_check) {
+ if (!ctx->sps)
+ return -EINVAL;
+
+ if (!pix_mp)
+ pix_mp = &ctx->decoded_fmt.fmt.pix_mp;
+ return coded_desc->ops->sps_check(ctx, ctx->sps, pix_mp);
+ }
+
+ return 0;
+}
+
static int rkvdec_get_sps_attributes(struct rkvdec_ctx *ctx, void *sps,
struct sps_attributes *attributes)
{
@@ -375,6 +395,13 @@ static int rkvdec_s_output_fmt(struct file *file, void *priv,
struct vb2_queue *vq;
int ret;
+ /*
+ * When the new output format doesn't match the existing SPS stored in
+ * the context, then the SPS needs to be reset by user-space.
+ */
+ if (rkvdec_sps_check(ctx, &f->fmt.pix_mp) < 0)
+ ctx->sps = NULL;
+
/*
* In order to support dynamic resolution change, the decoder admits
* a resolution change, as long as the pixelformat remains. Can't be
@@ -603,6 +630,14 @@ static int rkvdec_start_streaming(struct vb2_queue *q, unsigned int count)
if (V4L2_TYPE_IS_CAPTURE(q->type))
return 0;
+ /*
+ * An incompatible SPS at this point is invalid, abort the process.
+ */
+ if (rkvdec_sps_check(ctx, NULL) < 0) {
+ ctx->sps = NULL;
+ return -EINVAL;
+ }
+
/*
* Make sure that both the output and the capture queue are running
*/
@@ -75,6 +75,7 @@ struct rkvdec_coded_fmt_ops {
int (*adjust_fmt)(struct rkvdec_ctx *ctx,
struct v4l2_format *f);
u32 (*valid_fmt)(struct rkvdec_ctx *ctx);
+ int (*sps_check)(struct rkvdec_ctx *ctx, void *sps, struct v4l2_pix_format_mplane *pix_mp);
int (*get_sps_attributes)(struct rkvdec_ctx *ctx, void *sps,
struct sps_attributes *attributes);
int (*start)(struct rkvdec_ctx *ctx);
Add a callback for each codec variant, that ensures that a given format is compatible with the SPS that is stored in the context of the instance. This is used for two scenarios: - New output queue format is set, which is incompatible with the currently stored SPS - Start streaming is called but the SPS doesn't match the current format anymore Signed-off-by: Sebastian Fricke <sebastian.fricke@collabora.com> --- drivers/staging/media/rkvdec/rkvdec.c | 35 +++++++++++++++++++++++++++++++++++ drivers/staging/media/rkvdec/rkvdec.h | 1 + 2 files changed, 36 insertions(+)