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

#define MODULE_TAG "mpp_rt"

#include <fcntl.h>
#include <unistd.h>

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

#include "mpp_runtime.h"

#define MAX_DTS_PATH_LEN        256

#define get_srv_runtime() \
    ({ \
        MppRuntimeSrv *__tmp; \
        if (!srv_runtime) { \
            mpp_rt_srv_init(); \
        } \
        if (srv_runtime) { \
            __tmp = srv_runtime; \
        } else { \
            mpp_err("mpp mem pool srv not init at %s\n", __FUNCTION__); \
            __tmp = NULL; \
        } \
        __tmp; \
    })

static rk_u32 mpp_rt_debug = 0;

static const char *mpp_dts_base = "/proc/device-tree/";

static const char *mpp_vpu_names[] = {
    "vpu_service",
    "vpu-service",
    "vpu",
    //"hevc_service",
    //"hevc-service",
    //"rkvdec",
    //"rkvenc",
    //"vpu_combo",
};

#define mpp_rt_dbg(fmt, ...)    do { if (mpp_rt_debug) mpp_log(fmt, ## __VA_ARGS__); } while (0)

static const char *mpp_vpu_address[] = {
    "",                 /* old kernel   */
    "@10108000",        /* rk3036       */
    "@20020000",        /* rk322x       */
    "@30000000",        /* rv1108       */
    "@ff9a0000",        /* rk3288/3366  */
    "@ff350000",        /* rk322xh/3328 */
    "@ff650000",        /* rk3399       */
};

typedef struct MppRuntimeService_t {
    rk_u32  allocator_valid[MPP_BUFFER_TYPE_BUTT];
} MppRuntimeSrv;

static MppRuntimeSrv *srv_runtime;

static void mpp_rt_srv_init()
{
    MppRuntimeSrv *srv = srv_runtime;

    mpp_env_get_u32("mpp_rt_debug", &mpp_rt_debug, 0);

    if (srv)
        return;

    srv = mpp_calloc(MppRuntimeSrv, 1);
    if (!srv) {
        mpp_err_f("failed to allocate runtime service\n");
        return;
    }

    srv_runtime = srv;

    srv->allocator_valid[MPP_BUFFER_TYPE_NORMAL] = 1;
    srv->allocator_valid[MPP_BUFFER_TYPE_ION] = !access("/dev/ion", F_OK | R_OK | W_OK);
    srv->allocator_valid[MPP_BUFFER_TYPE_DRM] = !access("/dev/dri/card0", F_OK | R_OK | W_OK);
    srv->allocator_valid[MPP_BUFFER_TYPE_DMA_HEAP] = !access("/dev/dma_heap/system", F_OK | R_OK) &&
                                                     !access("/dev/dma_heap/system-dma32", F_OK | R_OK) &&
                                                     !access("/dev/dma_heap/system-uncached", F_OK | R_OK) &&
                                                     !access("/dev/dma_heap/system-uncached-dma32", F_OK | R_OK);

    if (!srv->allocator_valid[MPP_BUFFER_TYPE_ION] &&
        !srv->allocator_valid[MPP_BUFFER_TYPE_DRM] &&
        !srv->allocator_valid[MPP_BUFFER_TYPE_DMA_HEAP]) {
        mpp_err("can NOT found any allocator\n");
        return;
    }

    if (srv->allocator_valid[MPP_BUFFER_TYPE_DMA_HEAP]) {
        mpp_rt_dbg("use dma heap allocator\n");
        return;
    }

    if (srv->allocator_valid[MPP_BUFFER_TYPE_ION] && !srv->allocator_valid[MPP_BUFFER_TYPE_DRM]) {
        mpp_rt_dbg("use ion allocator\n");
        return;
    }

    if (!srv->allocator_valid[MPP_BUFFER_TYPE_ION] && srv->allocator_valid[MPP_BUFFER_TYPE_DRM]) {
        mpp_rt_dbg("use drm allocator\n");
        return;
    }

    if (!access("/dev/mpp_service", F_OK | R_OK | W_OK)) {
        srv->allocator_valid[MPP_BUFFER_TYPE_ION] = 0;

        mpp_rt_dbg("use drm allocator for mpp_service\n");
        return;
    }

    // If both ion and drm is enabled detect allocator in dts to choose one
    // TODO: When unify dma fd kernel is completed this part will be removed.
    if (srv->allocator_valid[MPP_BUFFER_TYPE_ION] &&
        srv->allocator_valid[MPP_BUFFER_TYPE_DRM]) {
        /* Detect hardware buffer type is ion or drm */
        rk_u32 i, j;
        char path[MAX_DTS_PATH_LEN];
        rk_u32 path_len = MAX_DTS_PATH_LEN - 1;
        rk_u32 dts_path_len = snprintf(path, path_len, "%s", mpp_dts_base);
        char *p = path + dts_path_len;
        rk_u32 allocator_found = 0;

        path_len -= dts_path_len;

        for (i = 0; i < MPP_ARRAY_ELEMS(mpp_vpu_names); i++) {
            for (j = 0; j < MPP_ARRAY_ELEMS(mpp_vpu_address); j++) {
                rk_u32 dev_path_len = snprintf(p, path_len, "%s%s",
                                               mpp_vpu_names[i], mpp_vpu_address[j]);
                int f_ok = access(path, F_OK);
                if (f_ok == 0) {
                    snprintf(p + dev_path_len, path_len - dev_path_len, "/%s", "allocator");
                    f_ok = access(path, F_OK);
                    if (f_ok == 0) {
                        rk_s32 val = 0;
                        FILE *fp = fopen(path, "rb");
                        if (fp) {
                            size_t len = fread(&val, 1, 4, fp);
                            // zero for ion non-zero for drm ->
                            // zero     - disable drm
                            // non-zero - disable ion
                            if (len != 4) {
                                mpp_err("failed to read dts allocator value default 0\n");
                                val = 0;
                            }

                            if (val == 0) {
                                srv->allocator_valid[MPP_BUFFER_TYPE_DRM] = 0;
                                mpp_rt_dbg("found ion allocator in dts\n");
                            } else {
                                srv->allocator_valid[MPP_BUFFER_TYPE_ION] = 0;
                                mpp_rt_dbg("found drm allocator in dts\n");
                            }
                            allocator_found = 1;
                        }
                    }
                }
                if (allocator_found)
                    break;
            }
            if (allocator_found)
                break;
        }

        if (!allocator_found)
            mpp_log("Can NOT found allocator in dts, enable both ion and drm\n");
    }
    return;
}

static void mpp_rt_srv_deinit()
{
    MPP_FREE(srv_runtime);
}

rk_u32 mpp_rt_allcator_is_valid(MppBufferType type)
{
    MppBufferType buffer_type = (MppBufferType)(type & MPP_BUFFER_TYPE_MASK);
    rk_u32 valid = 0;

    if (buffer_type < MPP_BUFFER_TYPE_BUTT) {
        MppRuntimeSrv *srv = get_srv_runtime();

        if (srv)
            valid = srv->allocator_valid[buffer_type];
    }

    return valid;
}

MPP_SINGLETON(MPP_SGLN_RUNTIME, mpp_runtime, mpp_rt_srv_init, mpp_rt_srv_deinit);