// SPDX-License-Identifier: GPL-2.0
/*
* Copyright (c) 2016 MediaTek Inc.
* Author: Tiffany Lin <tiffany.lin@mediatek.com>
*/

#include <linux/errno.h>
#include <linux/wait.h>

#include "../decoder/mtk_vcodec_dec_drv.h"
#include "../encoder/mtk_vcodec_enc_drv.h"
#include "mtk_vcodec_intr.h"

int mtk_vcodec_wait_for_done_ctx(void *priv, int command, unsigned int timeout_ms,
                                 unsigned int hw_id)
{
        int instance_type = *((int *)priv);
        long timeout_jiff, ret;
        int ctx_id, ctx_type, status = 0;
        int *ctx_int_cond, *ctx_int_type;
        wait_queue_head_t *ctx_queue;
        struct platform_device *pdev;

        if (instance_type == DECODER) {
                struct mtk_vcodec_dec_ctx *ctx;

                ctx = priv;
                ctx_id = ctx->id;
                ctx_type = ctx->type;
                ctx_int_cond = ctx->int_cond;
                ctx_int_type = ctx->int_type;
                ctx_queue = ctx->queue;
                pdev = ctx->dev->plat_dev;
        } else {
                struct mtk_vcodec_enc_ctx *ctx;

                ctx = priv;
                ctx_id = ctx->id;
                ctx_type = ctx->type;
                ctx_int_cond = ctx->int_cond;
                ctx_int_type = ctx->int_type;
                ctx_queue = ctx->queue;
                pdev = ctx->dev->plat_dev;
        }

        timeout_jiff = msecs_to_jiffies(timeout_ms);
        ret = wait_event_interruptible_timeout(ctx_queue[hw_id],
                                               ctx_int_cond[hw_id],
                                               timeout_jiff);

        if (!ret) {
                status = -1;    /* timeout */
                dev_err(&pdev->dev, "[%d] cmd=%d, type=%d, dec timeout=%ums (%d %d)",
                        ctx_id, command, ctx_type, timeout_ms,
                        ctx_int_cond[hw_id], ctx_int_type[hw_id]);
        } else if (-ERESTARTSYS == ret) {
                status = -1;
                dev_err(&pdev->dev, "[%d] cmd=%d, type=%d, dec inter fail (%d %d)",
                        ctx_id, command, ctx_type,
                        ctx_int_cond[hw_id], ctx_int_type[hw_id]);
        }

        ctx_int_cond[hw_id] = 0;
        ctx_int_type[hw_id] = 0;

        return status;
}
EXPORT_SYMBOL(mtk_vcodec_wait_for_done_ctx);