/* SPDX-License-Identifier: Apache-2.0 OR MIT */
/*
 * Copyright (c) 2015 Rockchip Electronics Co., Ltd.
 */

#define MODULE_TAG "mpp_task_impl"

#include <string.h>

#include "mpp_env.h"
#include "mpp_mem.h"
#include "mpp_debug.h"

#include "mpp_task_impl.h"
#include "mpp_meta_impl.h"

#define MAX_TASK_COUNT      8

#define MPP_TASK_DBG_FUNCTION       (0x00000001)
#define MPP_TASK_DBG_FLOW           (0x00000002)

#define mpp_task_dbg(flag, fmt, ...)     _mpp_dbg(mpp_task_debug, flag, fmt, ## __VA_ARGS__)
#define mpp_task_dbg_f(flag, fmt, ...)   _mpp_dbg_f(mpp_task_debug, flag, fmt, ## __VA_ARGS__)

#define mpp_task_dbg_func(fmt, ...)      mpp_task_dbg_f(MPP_TASK_DBG_FUNCTION, fmt, ## __VA_ARGS__)
#define mpp_task_dbg_flow(fmt, ...)      mpp_task_dbg(MPP_TASK_DBG_FLOW, fmt, ## __VA_ARGS__)

typedef struct MppTaskStatusInfo_t {
    struct list_head    list;
    RK_S32              count;
    MppTaskStatus       status;
    MppCond             cond;
} MppTaskStatusInfo;

typedef struct MppTaskQueueImpl_t {
    char                name[32];
    void                *mpp;
    MppMutex            lock;
    RK_S32              task_count;
    RK_S32              ready;          // flag for deinit

    // two ports inside of task queue
    MppPort             input;
    MppPort             output;

    MppTaskImpl         *tasks;

    MppTaskStatusInfo   info[MPP_TASK_STATUS_BUTT];
} MppTaskQueueImpl;

typedef struct MppPortImpl_t {
    MppPortType         type;
    MppTaskQueueImpl    *queue;

    MppTaskStatus       status_curr;
    MppTaskStatus       next_on_dequeue;
    MppTaskStatus       next_on_enqueue;
} MppPortImpl;

static const char *module_name = MODULE_TAG;
static const char *port_type_str[] = {
    "input",
    "output",
    "NULL",
};

static const char *task_status_str[] = {
    "input_port",
    "input_hold",
    "output_port",
    "output_hold",
    "NULL",
};

RK_U32 mpp_task_debug = 0;

static inline void setup_mpp_task_name(MppTaskImpl *task)
{
    task->name = module_name;
}

MPP_RET check_mpp_task_name(MppTask task)
{
    if (task && ((MppTaskImpl *)task)->name == module_name)
        return MPP_OK;

    mpp_err_f("pointer %p failed on check\n", task);
    mpp_abort();
    return MPP_NOK;
}

static MPP_RET mpp_port_init(MppTaskQueueImpl *queue, MppPortType type, MppPort *port)
{
    MppPortImpl *impl = mpp_malloc(MppPortImpl, 1);
    if (!impl) {
        mpp_err_f("failed to malloc MppPort type %d\n", type);
        return MPP_ERR_MALLOC;
    }

    mpp_task_dbg_func("enter queue %p type %d\n", queue, type);

    impl->type  = type;
    impl->queue = queue;

    if (MPP_PORT_INPUT == type) {
        impl->status_curr     = MPP_INPUT_PORT;
        impl->next_on_dequeue = MPP_INPUT_HOLD;
        impl->next_on_enqueue = MPP_OUTPUT_PORT;
    } else {
        impl->status_curr     = MPP_OUTPUT_PORT;
        impl->next_on_dequeue = MPP_OUTPUT_HOLD;
        impl->next_on_enqueue = MPP_INPUT_PORT;
    }

    *port = (MppPort *)impl;

    mpp_task_dbg_func("leave queue %p port %p\n", queue, impl);

    return MPP_OK;
}

static MPP_RET mpp_port_deinit(MppPort port)
{
    mpp_task_dbg_func("enter port %p\n", port);
    mpp_free(port);
    mpp_task_dbg_func("leave\n");
    return MPP_OK;
}

MPP_RET _mpp_port_poll(const char *caller, MppPort port, MppPollType timeout)
{
    MppPortImpl *port_impl = (MppPortImpl *)port;
    MppTaskQueueImpl *queue = port_impl->queue;
    MppTaskStatusInfo *curr = NULL;
    MPP_RET ret = MPP_NOK;

    mpp_mutex_lock(&queue->lock);

    mpp_task_dbg_func("enter port %p\n", port);
    if (!queue->ready) {
        mpp_err("try to query when %s queue is not ready\n",
                port_type_str[port_impl->type]);
        goto RET;
    }

    curr = &queue->info[port_impl->status_curr];
    if (curr->count) {
        mpp_assert(!list_empty(&curr->list));
        ret = (MPP_RET)curr->count;
        mpp_task_dbg_flow("mpp %p %s from %s poll %s port timeout %d count %d\n",
                          queue->mpp, queue->name, caller,
                          port_type_str[port_impl->type],
                          timeout, curr->count);
    } else {
        mpp_assert(list_empty(&curr->list));

        /* timeout
         * zero     - non-block
         * negtive  - block
         * positive - timeout value
         */
        if (timeout) {
            MppCond *cond = &curr->cond;

            if (timeout < 0) {
                mpp_task_dbg_flow("mpp %p %s from %s poll %s port block wait start\n",
                                  queue->mpp, queue->name, caller,
                                  port_type_str[port_impl->type]);

                ret = (MPP_RET)mpp_cond_wait(cond, &queue->lock);
            } else {
                mpp_task_dbg_flow("mpp %p %s from %s poll %s port %d timeout wait start\n",
                                  queue->mpp, queue->name, caller,
                                  port_type_str[port_impl->type], timeout);
                ret = (MPP_RET)mpp_cond_timedwait(cond, &queue->lock, timeout);
            }

            if (curr->count) {
                mpp_assert(!list_empty(&curr->list));
                ret = (MPP_RET)curr->count;
            } else if (ret > 0)
                ret = MPP_NOK;
        }

        mpp_task_dbg_flow("mpp %p %s from %s poll %s port timeout %d ret %d\n",
                          queue->mpp, queue->name, caller,
                          port_type_str[port_impl->type], timeout, ret);
    }
RET:
    mpp_mutex_unlock(&queue->lock);
    mpp_task_dbg_func("leave\n");
    return ret;
}

MPP_RET _mpp_port_move(const char *caller, MppPort port, MppTask task,
                       MppTaskStatus status)
{
    MppTaskImpl *task_impl = (MppTaskImpl *)task;
    MppPortImpl *port_impl = (MppPortImpl *)port;
    MppTaskQueueImpl *queue = port_impl->queue;
    MppTaskStatusInfo *curr = NULL;
    MppTaskStatusInfo *next = NULL;
    MPP_RET ret = MPP_NOK;

    mpp_mutex_lock(&queue->lock);

    mpp_task_dbg_func("caller %s enter port %p task %p\n", caller, port, task);

    if (!queue->ready) {
        mpp_err("try to move task when %s queue is not ready\n",
                port_type_str[port_impl->type]);
        goto RET;
    }

    check_mpp_task_name(task);

    mpp_assert(task_impl->queue == (MppTaskQueue)queue);

    curr = &queue->info[task_impl->status];
    next = &queue->info[status];

    list_del_init(&task_impl->list);
    curr->count--;
    list_add_tail(&task_impl->list, &next->list);
    next->count++;

    mpp_task_dbg_flow("mpp %p %s from %s move %s port task %p %s -> %s done\n",
                      queue->mpp, queue->name, caller,
                      port_type_str[port_impl->type], task_impl,
                      task_status_str[task_impl->status],
                      task_status_str[status]);
    task_impl->status = status;

    mpp_cond_signal(&next->cond);
    mpp_task_dbg_func("signal port %p\n", next);
    ret = MPP_OK;
RET:
    mpp_task_dbg_func("caller %s leave port %p task %p ret %d\n", caller, port, task, ret);
    mpp_mutex_unlock(&queue->lock);

    return ret;
}

MPP_RET _mpp_port_dequeue(const char *caller, MppPort port, MppTask *task)
{
    MppPortImpl *port_impl = (MppPortImpl *)port;
    MppTaskQueueImpl *queue = port_impl->queue;
    MppTaskStatusInfo *curr = NULL;
    MppTaskStatusInfo *next = NULL;
    MppTaskImpl *task_impl = NULL;
    MppTask p = NULL;
    MPP_RET ret = MPP_NOK;

    mpp_mutex_lock(&queue->lock);

    mpp_task_dbg_func("caller %s enter port %p\n", caller, port);

    if (!queue->ready) {
        mpp_err("try to dequeue when %s queue is not ready\n",
                port_type_str[port_impl->type]);
        goto RET;
    }

    curr = &queue->info[port_impl->status_curr];
    next = &queue->info[port_impl->next_on_dequeue];

    *task = NULL;
    if (curr->count == 0) {
        mpp_assert(list_empty(&curr->list));
        mpp_task_dbg_flow("mpp %p %s from %s dequeue %s port task %s -> %s failed\n",
                          queue->mpp, queue->name, caller,
                          port_type_str[port_impl->type],
                          task_status_str[port_impl->status_curr],
                          task_status_str[port_impl->next_on_dequeue]);
        goto RET;
    }

    mpp_assert(!list_empty(&curr->list));
    task_impl = list_entry(curr->list.next, MppTaskImpl, list);
    p = (MppTask)task_impl;
    check_mpp_task_name(p);
    list_del_init(&task_impl->list);
    curr->count--;
    mpp_assert(curr->count >= 0);

    list_add_tail(&task_impl->list, &next->list);
    next->count++;
    task_impl->status = next->status;

    mpp_task_dbg_flow("mpp %p %s from %s dequeue %s port task %p %s -> %s done\n",
                      queue->mpp, queue->name, caller,
                      port_type_str[port_impl->type], task_impl,
                      task_status_str[port_impl->status_curr],
                      task_status_str[port_impl->next_on_dequeue]);

    *task = p;
    ret = MPP_OK;
RET:
    mpp_task_dbg_func("caller %s leave port %p task %p ret %d\n", caller, port, *task, ret);
    mpp_mutex_unlock(&queue->lock);

    return ret;
}

MPP_RET _mpp_port_enqueue(const char *caller, MppPort port, MppTask task)
{
    MppTaskImpl *task_impl = (MppTaskImpl *)task;
    MppPortImpl *port_impl = (MppPortImpl *)port;
    MppTaskQueueImpl *queue = port_impl->queue;
    MppTaskStatusInfo *curr = NULL;
    MppTaskStatusInfo *next = NULL;
    MPP_RET ret = MPP_NOK;

    mpp_mutex_lock(&queue->lock);

    mpp_task_dbg_func("caller %s enter port %p task %p\n", caller, port, task);

    if (!queue->ready) {
        mpp_err("try to enqueue when %s queue is not ready\n",
                port_type_str[port_impl->type]);
        mpp_mutex_unlock(&queue->lock);
        goto RET;
    }

    check_mpp_task_name(task);

    mpp_assert(task_impl->queue  == (MppTaskQueue)queue);
    mpp_assert(task_impl->status == port_impl->next_on_dequeue);

    curr = &queue->info[task_impl->status];
    next = &queue->info[port_impl->next_on_enqueue];

    list_del_init(&task_impl->list);
    curr->count--;
    list_add_tail(&task_impl->list, &next->list);
    next->count++;
    task_impl->status = next->status;

    mpp_task_dbg_flow("mpp %p %s from %s enqueue %s port task %p %s -> %s done\n",
                      queue->mpp, queue->name, caller,
                      port_type_str[port_impl->type], task_impl,
                      task_status_str[port_impl->next_on_dequeue],
                      task_status_str[port_impl->next_on_enqueue]);

    mpp_cond_signal(&next->cond);
    mpp_task_dbg_func("signal port %p\n", next);
    ret = MPP_OK;
RET:
    mpp_task_dbg_func("caller %s leave port %p task %p ret %d\n", caller, port, task, ret);
    mpp_mutex_unlock(&queue->lock);

    return ret;
}

MPP_RET _mpp_port_awake(const char *caller, MppPort port)
{
    if (port == NULL)
        return MPP_NOK;

    mpp_task_dbg_func("caller %s enter port %p\n", caller, port);
    MppPortImpl *port_impl = (MppPortImpl *)port;
    MppTaskQueueImpl *queue = port_impl->queue;
    MppTaskStatusInfo *curr = NULL;
    if (queue) {
        mpp_mutex_lock(&queue->lock);
        curr = &queue->info[port_impl->status_curr];
        if (curr) {
            mpp_cond_signal(&curr->cond);
        }
        mpp_mutex_unlock(&queue->lock);
    }

    mpp_task_dbg_func("caller %s leave port %p\n", caller, port);
    return MPP_OK;
}

MPP_RET mpp_task_queue_init(MppTaskQueue *queue, void *mpp, const char *name)
{
    if (!queue) {
        mpp_err_f("invalid NULL input\n");
        return MPP_ERR_NULL_PTR;
    }

    MPP_RET ret = MPP_NOK;
    MppTaskQueueImpl *p = NULL;
    RK_S32 i;

    mpp_env_get_u32("mpp_task_debug", &mpp_task_debug, 0);
    mpp_task_dbg_func("enter\n");

    *queue = NULL;

    p = mpp_calloc(MppTaskQueueImpl, 1);
    if (!p) {
        mpp_err_f("malloc queue failed\n");
        goto RET;
    }

    for (i = 0; i < MPP_TASK_STATUS_BUTT; i++) {
        INIT_LIST_HEAD(&p->info[i].list);
        p->info[i].count  = 0;
        p->info[i].status = (MppTaskStatus)i;
        if (i == MPP_INPUT_PORT || i == MPP_OUTPUT_PORT)
            mpp_cond_init(&p->info[i].cond);
    }

    mpp_mutex_init(&p->lock);

    if (mpp_port_init(p, MPP_PORT_INPUT, &p->input))
        goto RET;

    if (mpp_port_init(p, MPP_PORT_OUTPUT, &p->output)) {
        mpp_port_deinit(p->input);
        goto RET;
    }

    p->mpp = mpp;
    if (name)
        strncpy(p->name, name, sizeof(p->name) - 1);
    else
        strncpy(p->name, "none", sizeof(p->name) - 1);

    ret = MPP_OK;
RET:
    if (ret) {
        mpp_mutex_destroy(&p->lock);
        mpp_cond_destroy(&p->info[MPP_INPUT_PORT].cond);
        mpp_cond_destroy(&p->info[MPP_OUTPUT_PORT].cond);
        MPP_FREE(p);
    }

    *queue = p;

    mpp_task_dbg_func("leave ret %d queue %p\n", ret, p);
    return ret;
}

MPP_RET mpp_task_queue_setup(MppTaskQueue queue, RK_S32 task_count)
{
    MppTaskQueueImpl *impl = (MppTaskQueueImpl *)queue;
    MppTaskStatusInfo *info;
    MppTaskImpl *tasks;
    RK_S32 i;

    mpp_mutex_lock(&impl->lock);

    // NOTE: queue can only be setup once
    mpp_assert(impl->tasks == NULL);
    mpp_assert(impl->task_count == 0);
    tasks = mpp_calloc(MppTaskImpl, task_count);
    if (!tasks) {
        mpp_err_f("malloc tasks list failed\n");
        mpp_mutex_unlock(&impl->lock);
        return MPP_ERR_MALLOC;
    }

    impl->tasks = tasks;
    impl->task_count = task_count;

    info = &impl->info[MPP_INPUT_PORT];

    for (i = 0; i < task_count; i++) {
        setup_mpp_task_name(&tasks[i]);
        INIT_LIST_HEAD(&tasks[i].list);
        tasks[i].index  = i;
        tasks[i].queue  = queue;
        tasks[i].status = MPP_INPUT_PORT;
        mpp_meta_get(&tasks[i].meta);

        list_add_tail(&tasks[i].list, &info->list);
        info->count++;
    }
    impl->ready = 1;

    mpp_mutex_unlock(&impl->lock);

    return MPP_OK;
}

MPP_RET mpp_task_queue_deinit(MppTaskQueue queue)
{
    MppTaskQueueImpl *p = (MppTaskQueueImpl *)queue;
    RK_S32 i;

    if (!p) {
        mpp_err_f("found NULL input queue\n");
        return MPP_ERR_NULL_PTR;
    }

    mpp_mutex_lock(&p->lock);

    p->ready = 0;
    mpp_cond_signal(&p->info[MPP_INPUT_PORT].cond);
    mpp_cond_signal(&p->info[MPP_OUTPUT_PORT].cond);

    if (p->tasks) {
        for (i = 0; i < p->task_count; i++) {
            MppMeta meta = p->tasks[i].meta;

            /* we must ensure that all task return to init status */
            if (mpp_meta_size(meta)) {
                mpp_err_f("%s queue idx %d task %p status %d meta size %d\n",
                          p->name, i, &p->tasks[i], p->tasks[i].status,
                          mpp_meta_size(meta));
                mpp_meta_dump(meta);
            }
            mpp_meta_put(p->tasks[i].meta);
        }
        mpp_free(p->tasks);
    }

    if (p->input) {
        mpp_port_deinit(p->input);
        p->input = NULL;
    }
    if (p->output) {
        mpp_port_deinit(p->output);
        p->output = NULL;
    }
    mpp_mutex_unlock(&p->lock);
    mpp_mutex_destroy(&p->lock);

    mpp_cond_destroy(&p->info[MPP_INPUT_PORT].cond);
    mpp_cond_destroy(&p->info[MPP_OUTPUT_PORT].cond);

    mpp_free(p);
    return MPP_OK;
}

MppPort mpp_task_queue_get_port(MppTaskQueue queue, MppPortType type)
{
    if (!queue || type >= MPP_PORT_BUTT) {
        mpp_err_f("invalid input queue %p type %d\n", queue, type);
        return NULL;
    }

    MppTaskQueueImpl *impl = (MppTaskQueueImpl *)queue;
    return (type == MPP_PORT_INPUT) ? (impl->input) : (impl->output);
}

MppMeta mpp_task_get_meta(MppTask task)
{
    MppMeta meta = NULL;

    if (task)
        meta = ((MppTaskImpl *)task)->meta;

    return meta;
}
