[PATCH v2 5/5] venus: vdec: Handle DRC after drain

[Stanimir Varbanov]

From: Fritz Koenig <frkoe...@chromium.org>

If the DRC is near the end of the stream the client
may send a V4L2_DEC_CMD_STOP before the DRC occurs.
V4L2_DEC_CMD_STOP puts the driver into the
VENUS_DEC_STATE_DRAIN state.  DRC must be aware so
that after the DRC event the state can be restored
correctly.

Signed-off-by: Fritz Koenig <frkoe...@chromium.org>
Signed-off-by: Stanimir Varbanov <stanimir.varba...@linaro.org>
---

Changes since v2 (from Fritz):
 - moved state transition from vdec_event_notify to vdec_event_change. 

 drivers/media/platform/qcom/venus/core.h |  1 +
 drivers/media/platform/qcom/venus/vdec.c | 14 ++++++++++++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/drivers/media/platform/qcom/venus/core.h 
b/drivers/media/platform/qcom/venus/core.h
index db0e6738281e..765ab7ed881b 100644
--- a/drivers/media/platform/qcom/venus/core.h
+++ b/drivers/media/platform/qcom/venus/core.h
@@ -413,6 +413,7 @@ struct venus_inst {
        unsigned int core_acquired: 1;
        unsigned int bit_depth;
        bool next_buf_last;
+       bool drain_active;
 };
 
 #define IS_V1(core)    ((core)->res->hfi_version == HFI_VERSION_1XX)
diff --git a/drivers/media/platform/qcom/venus/vdec.c 
b/drivers/media/platform/qcom/venus/vdec.c
index d27f4fd0ca01..6cc35ffe2d6e 100644
--- a/drivers/media/platform/qcom/venus/vdec.c
+++ b/drivers/media/platform/qcom/venus/vdec.c
@@ -519,8 +519,10 @@ vdec_decoder_cmd(struct file *file, void *fh, struct 
v4l2_decoder_cmd *cmd)
 
                ret = hfi_session_process_buf(inst, &fdata);
 
-               if (!ret && inst->codec_state == VENUS_DEC_STATE_DECODING)
+               if (!ret && inst->codec_state == VENUS_DEC_STATE_DECODING) {
                        inst->codec_state = VENUS_DEC_STATE_DRAIN;
+                       inst->drain_active = true;
+               }
        }
 
 unlock:
@@ -969,9 +971,13 @@ static int vdec_start_capture(struct venus_inst *inst)
 
        inst->codec_state = VENUS_DEC_STATE_DECODING;
 
+       if (inst->drain_active)
+               inst->codec_state = VENUS_DEC_STATE_DRAIN;
+
        inst->streamon_cap = 1;
        inst->sequence_cap = 0;
        inst->reconfig = false;
+       inst->drain_active = false;
 
        return 0;
 
@@ -1097,6 +1103,7 @@ static int vdec_stop_capture(struct venus_inst *inst)
                fallthrough;
        case VENUS_DEC_STATE_DRAIN:
                inst->codec_state = VENUS_DEC_STATE_STOPPED;
+               inst->drain_active = false;
                fallthrough;
        case VENUS_DEC_STATE_SEEK:
                vdec_cancel_dst_buffers(inst);
@@ -1296,8 +1303,10 @@ static void vdec_buf_done(struct venus_inst *inst, 
unsigned int buf_type,
 
                        v4l2_event_queue_fh(&inst->fh, &ev);
 
-                       if (inst->codec_state == VENUS_DEC_STATE_DRAIN)
+                       if (inst->codec_state == VENUS_DEC_STATE_DRAIN) {
+                               inst->drain_active = false;
                                inst->codec_state = VENUS_DEC_STATE_STOPPED;
+                       }
                }
 
                if (!bytesused)
@@ -1373,6 +1382,7 @@ static void vdec_event_change(struct venus_inst *inst,
                inst->codec_state = VENUS_DEC_STATE_CAPTURE_SETUP;
                break;
        case VENUS_DEC_STATE_DECODING:
+       case VENUS_DEC_STATE_DRAIN:
                inst->codec_state = VENUS_DEC_STATE_DRC;
                break;
        default:
-- 
2.17.1