// SPDX-License-Identifier: MIT
//
// Copyright 2025 Advanced Micro Devices, Inc.

#include "dc_fused_io.h"

#include "dm_helpers.h"
#include "gpio.h"

static bool op_i2c_convert(
                union dmub_rb_cmd *cmd,
                const struct mod_hdcp_atomic_op_i2c *op,
                enum dmub_cmd_fused_request_type type,
                uint32_t ddc_line,
                bool over_aux
)
{
        struct dmub_cmd_fused_request *req = &cmd->fused_io.request;
        struct dmub_cmd_fused_request_location_i2c *loc = &req->u.i2c;

        if (!op || op->size > sizeof(req->buffer))
                return false;

        req->type = type;
        loc->is_aux = false;
        loc->ddc_line = ddc_line;
        loc->over_aux = over_aux;
        loc->address = op->address;
        loc->offset = op->offset;
        loc->length = op->size;
        memcpy(req->buffer, op->data, op->size);

        return true;
}

static bool op_aux_convert(
                union dmub_rb_cmd *cmd,
                const struct mod_hdcp_atomic_op_aux *op,
                enum dmub_cmd_fused_request_type type,
                uint32_t ddc_line
)
{
        struct dmub_cmd_fused_request *req = &cmd->fused_io.request;
        struct dmub_cmd_fused_request_location_aux *loc = &req->u.aux;

        if (!op || op->size > sizeof(req->buffer))
                return false;

        req->type = type;
        loc->is_aux = true;
        loc->ddc_line = ddc_line;
        loc->address = op->address;
        loc->length = op->size;
        memcpy(req->buffer, op->data, op->size);

        return true;
}

static bool atomic_write_poll_read(
                struct dc_link *link,
                union dmub_rb_cmd commands[3],
                uint32_t poll_timeout_us,
                uint8_t poll_mask_msb
)
{
        const uint8_t count = 3;
        const uint32_t timeout_per_request_us = 10000;
        const uint32_t timeout_per_aux_transaction_us = 10000;
        uint64_t timeout_us = 0;

        commands[1].fused_io.request.poll_mask_msb = poll_mask_msb;
        commands[1].fused_io.request.timeout_us = poll_timeout_us;

        for (uint8_t i = 0; i < count; i++) {
                struct dmub_rb_cmd_fused_io *io = &commands[i].fused_io;

                io->header.type = DMUB_CMD__FUSED_IO;
                io->header.sub_type = DMUB_CMD__FUSED_IO_EXECUTE;
                io->header.multi_cmd_pending = i != count - 1;
                io->header.payload_bytes = sizeof(commands[i].fused_io) - sizeof(io->header);

                timeout_us += timeout_per_request_us + io->request.timeout_us;
                if (!io->request.timeout_us && io->request.u.aux.is_aux)
                        timeout_us += timeout_per_aux_transaction_us * (io->request.u.aux.length / 16);
        }

        if (!dm_helpers_execute_fused_io(link->ctx, link, commands, count, timeout_us))
                return false;

        return commands[0].fused_io.request.status == FUSED_REQUEST_STATUS_SUCCESS;
}

bool dm_atomic_write_poll_read_i2c(
                struct dc_link *link,
                const struct mod_hdcp_atomic_op_i2c *write,
                const struct mod_hdcp_atomic_op_i2c *poll,
                struct mod_hdcp_atomic_op_i2c *read,
                uint32_t poll_timeout_us,
                uint8_t poll_mask_msb
)
{
        if (!link)
                return false;

        const bool over_aux = false;
        const uint32_t ddc_line = link->ddc->ddc_pin->pin_data->en;

        union dmub_rb_cmd commands[3] = { 0 };
        const bool converted = op_i2c_convert(&commands[0], write, FUSED_REQUEST_WRITE, ddc_line, over_aux)
                        && op_i2c_convert(&commands[1], poll, FUSED_REQUEST_POLL, ddc_line, over_aux)
                        && op_i2c_convert(&commands[2], read, FUSED_REQUEST_READ, ddc_line, over_aux);

        if (!converted)
                return false;

        const bool result = atomic_write_poll_read(link, commands, poll_timeout_us, poll_mask_msb);

        memcpy(read->data, commands[0].fused_io.request.buffer, read->size);
        return result;
}

bool dm_atomic_write_poll_read_aux(
                struct dc_link *link,
                const struct mod_hdcp_atomic_op_aux *write,
                const struct mod_hdcp_atomic_op_aux *poll,
                struct mod_hdcp_atomic_op_aux *read,
                uint32_t poll_timeout_us,
                uint8_t poll_mask_msb
)
{
        if (!link)
                return false;

        const uint32_t ddc_line = link->ddc->ddc_pin->pin_data->en;
        union dmub_rb_cmd commands[3] = { 0 };
        const bool converted = op_aux_convert(&commands[0], write, FUSED_REQUEST_WRITE, ddc_line)
                        && op_aux_convert(&commands[1], poll, FUSED_REQUEST_POLL, ddc_line)
                        && op_aux_convert(&commands[2], read, FUSED_REQUEST_READ, ddc_line);

        if (!converted)
                return false;

        const bool result = atomic_write_poll_read(link, commands, poll_timeout_us, poll_mask_msb);

        memcpy(read->data, commands[0].fused_io.request.buffer, read->size);
        return result;
}