/*
 *  nextpnr -- Next Generation Place and Route
 *
 *  Copyright (C) 2018  gatecat <gatecat@ds0.me>
 *
 *  Permission to use, copy, modify, and/or distribute this software for any
 *  purpose with or without fee is hereby granted, provided that the above
 *  copyright notice and this permission notice appear in all copies.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 */

#include "cells.h"
#include <algorithm>
#include "design_utils.h"
#include "log.h"
#include "util.h"

NEXTPNR_NAMESPACE_BEGIN

std::unique_ptr<CellInfo> create_ecp5_cell(Context *ctx, IdString type, std::string name)
{
    static int auto_idx = 0;
    IdString name_id =
            name.empty() ? ctx->id("$nextpnr_" + type.str(ctx) + "_" + std::to_string(auto_idx++)) : ctx->id(name);
    std::unique_ptr<CellInfo> new_cell = std::make_unique<CellInfo>(ctx, name_id, type);

    auto copy_bel_ports = [&]() {
        // First find a Bel of the target type
        BelId tgt;
        for (auto bel : ctx->getBels()) {
            if (ctx->getBelType(bel) == type) {
                tgt = bel;
                break;
            }
        }
        NPNR_ASSERT(tgt != BelId());
        for (auto port : ctx->getBelPins(tgt)) {
            new_cell->ports[port] = PortInfo{port, nullptr, ctx->getBelPinType(tgt, port)};
        }
    };

    if (type == id_TRELLIS_SLICE) {
        new_cell->params[id_MODE] = std::string("LOGIC");
        new_cell->params[id_GSR] = std::string("DISABLED");
        new_cell->params[id_SRMODE] = std::string("LSR_OVER_CE");
        new_cell->params[id_CEMUX] = std::string("1");
        new_cell->params[id_CLKMUX] = std::string("CLK");
        new_cell->params[id_LSRMUX] = std::string("LSR");
        new_cell->params[id_LUT0_INITVAL] = Property(0, 16);
        new_cell->params[id_LUT1_INITVAL] = Property(0, 16);
        new_cell->params[id_REG0_SD] = std::string("0");
        new_cell->params[id_REG1_SD] = std::string("0");
        new_cell->params[id_REG0_REGSET] = std::string("RESET");
        new_cell->params[id_REG1_REGSET] = std::string("RESET");
        new_cell->params[id_CCU2_INJECT1_0] = std::string("NO");
        new_cell->params[id_CCU2_INJECT1_1] = std::string("NO");
        new_cell->params[id_WREMUX] = std::string("WRE");

        new_cell->addInput(id_A0);
        new_cell->addInput(id_B0);
        new_cell->addInput(id_C0);
        new_cell->addInput(id_D0);

        new_cell->addInput(id_A1);
        new_cell->addInput(id_B1);
        new_cell->addInput(id_C1);
        new_cell->addInput(id_D1);

        new_cell->addInput(id_M0);
        new_cell->addInput(id_M1);

        new_cell->addInput(id_FCI);
        new_cell->addInput(id_FXA);
        new_cell->addInput(id_FXB);

        new_cell->addInput(id_CLK);
        new_cell->addInput(id_LSR);
        new_cell->addInput(id_CE);

        new_cell->addInput(id_DI0);
        new_cell->addInput(id_DI1);

        new_cell->addInput(id_WD0);
        new_cell->addInput(id_WD1);
        new_cell->addInput(id_WAD0);
        new_cell->addInput(id_WAD1);
        new_cell->addInput(id_WAD2);
        new_cell->addInput(id_WAD3);
        new_cell->addInput(id_WRE);
        new_cell->addInput(id_WCK);

        new_cell->addOutput(id_F0);
        new_cell->addOutput(id_Q0);
        new_cell->addOutput(id_F1);
        new_cell->addOutput(id_Q1);

        new_cell->addOutput(id_FCO);
        new_cell->addOutput(id_OFX0);
        new_cell->addOutput(id_OFX1);

        new_cell->addOutput(id_WDO0);
        new_cell->addOutput(id_WDO1);
        new_cell->addOutput(id_WDO2);
        new_cell->addOutput(id_WDO3);
        new_cell->addOutput(id_WADO0);
        new_cell->addOutput(id_WADO1);
        new_cell->addOutput(id_WADO2);
        new_cell->addOutput(id_WADO3);
    } else if (type == id_TRELLIS_IO) {
        new_cell->params[id_DIR] = std::string("INPUT");
        new_cell->attrs[id_IO_TYPE] = std::string("LVCMOS33");
        new_cell->params[id_DATAMUX_ODDR] = std::string("PADDO");
        new_cell->params[id_DATAMUX_MDDR] = std::string("PADDO");

        new_cell->addInout(id_B);
        new_cell->addInput(id_I);
        new_cell->addInput(id_T);
        new_cell->addOutput(id_O);

        new_cell->addInput(id_IOLDO);
        new_cell->addInput(id_IOLTO);

    } else if (type == id_LUT4) {
        new_cell->params[id_INIT] = Property(0, 16);

        new_cell->addInput(id_A);
        new_cell->addInput(id_B);
        new_cell->addInput(id_C);
        new_cell->addInput(id_D);
        new_cell->addOutput(id_Z);
    } else if (type == id_CCU2C) {
        new_cell->params[id_INIT0] = Property(0, 16);
        new_cell->params[id_INIT1] = Property(0, 16);
        new_cell->params[id_INJECT1_0] = std::string("YES");
        new_cell->params[id_INJECT1_1] = std::string("YES");

        new_cell->addInput(id_CIN);

        new_cell->addInput(id_A0);
        new_cell->addInput(id_B0);
        new_cell->addInput(id_C0);
        new_cell->addInput(id_D0);

        new_cell->addInput(id_A1);
        new_cell->addInput(id_B1);
        new_cell->addInput(id_C1);
        new_cell->addInput(id_D1);

        new_cell->addOutput(id_S0);
        new_cell->addOutput(id_S1);
        new_cell->addOutput(id_COUT);

    } else if (type == id_DCCA) {
        new_cell->addInput(id_CLKI);
        new_cell->addOutput(id_CLKO);
        new_cell->addInput(id_CE);
    } else if (type == id_IOLOGIC || type == id_SIOLOGIC) {
        new_cell->params[id_MODE] = std::string("NONE");
        new_cell->params[id_GSR] = std::string("DISABLED");
        new_cell->params[id_CLKIMUX] = std::string("CLK");
        new_cell->params[id_CLKOMUX] = std::string("CLK");
        new_cell->params[id_LSRIMUX] = std::string("0");
        new_cell->params[id_LSROMUX] = std::string("0");
        new_cell->params[id_LSRMUX] = std::string("LSR");

        new_cell->params[ctx->id("DELAY.OUTDEL")] = std::string("DISABLED");
        new_cell->params[ctx->id("DELAY.DEL_VALUE")] = Property(0, 7);
        new_cell->params[ctx->id("DELAY.WAIT_FOR_EDGE")] = std::string("DISABLED");

        if (type == id_IOLOGIC) {
            new_cell->params[ctx->id("IDDRXN.MODE")] = std::string("NONE");
            new_cell->params[ctx->id("ODDRXN.MODE")] = std::string("NONE");

            new_cell->params[ctx->id("MIDDRX.MODE")] = std::string("NONE");
            new_cell->params[ctx->id("MODDRX.MODE")] = std::string("NONE");
            new_cell->params[ctx->id("MTDDRX.MODE")] = std::string("NONE");

            new_cell->params[id_IOLTOMUX] = std::string("NONE");
            new_cell->params[ctx->id("MTDDRX.DQSW_INVERT")] = std::string("DISABLED");
            new_cell->params[ctx->id("MTDDRX.REGSET")] = std::string("RESET");

            new_cell->params[ctx->id("MIDDRX_MODDRX.WRCLKMUX")] = std::string("NONE");
        }
        // Just copy ports from the Bel
        copy_bel_ports();
    } else if (type == id_TRELLIS_ECLKBUF) {
        new_cell->addInput(id_ECLKI);
        new_cell->addOutput(id_ECLKO);
    } else {
        log_error("unable to create ECP5 cell of type %s", type.c_str(ctx));
    }
    return new_cell;
}

static void set_param_safe(bool has_ff, CellInfo *lc, IdString name, const std::string &value)
{
    NPNR_ASSERT(!has_ff || lc->params.at(name) == value);
    lc->params[name] = value;
}

static void replace_port_safe(bool has_ff, CellInfo *ff, IdString ff_port, CellInfo *lc, IdString lc_port)
{
    if (has_ff) {
        NPNR_ASSERT(lc->ports.at(lc_port).net == ff->ports.at(ff_port).net);
        NetInfo *ffnet = ff->ports.at(ff_port).net;
        if (ffnet != nullptr)
            ffnet->users.erase(
                    std::remove_if(ffnet->users.begin(), ffnet->users.end(),
                                   [ff, ff_port](PortRef port) { return port.cell == ff && port.port == ff_port; }),
                    ffnet->users.end());
    } else {
        replace_port(ff, ff_port, lc, lc_port);
    }
}

void ff_to_slice(Context *ctx, CellInfo *ff, CellInfo *lc, int index, bool driven_by_lut)
{
    if (lc->hierpath == IdString())
        lc->hierpath = ff->hierpath;
    bool has_ff = lc->ports.at(id_Q0).net != nullptr || lc->ports.at(id_Q1).net != nullptr;
    std::string reg = "REG" + std::to_string(index);
    set_param_safe(has_ff, lc, id_SRMODE, str_or_default(ff->params, id_SRMODE, "LSR_OVER_CE"));
    set_param_safe(has_ff, lc, id_GSR, str_or_default(ff->params, id_GSR, "DISABLED"));
    set_param_safe(has_ff, lc, id_CEMUX, str_or_default(ff->params, id_CEMUX, "1"));
    set_param_safe(has_ff, lc, id_LSRMUX, str_or_default(ff->params, id_LSRMUX, "LSR"));
    set_param_safe(has_ff, lc, id_CLKMUX, str_or_default(ff->params, id_CLKMUX, "CLK"));

    lc->params[ctx->id(reg + "_SD")] = std::string(driven_by_lut ? "1" : "0");
    lc->params[ctx->id(reg + "_REGSET")] = str_or_default(ff->params, id_REGSET, "RESET");
    lc->params[ctx->id(reg + "_LSRMODE")] = str_or_default(ff->params, id_LSRMODE, "LSR");
    replace_port_safe(has_ff, ff, id_CLK, lc, id_CLK);
    if (ff->ports.find(id_LSR) != ff->ports.end())
        replace_port_safe(has_ff, ff, id_LSR, lc, id_LSR);
    if (ff->ports.find(id_CE) != ff->ports.end())
        replace_port_safe(has_ff, ff, id_CE, lc, id_CE);

    replace_port(ff, id_Q, lc, ctx->id("Q" + std::to_string(index)));
    if (get_net_or_empty(ff, id_M) != nullptr) {
        // PRLD FFs that use both M and DI
        NPNR_ASSERT(!driven_by_lut);
        // As M is used; must route DI through a new LUT
        lc->params[ctx->id(reg + "_SD")] = std::string("1");
        lc->params[ctx->id("LUT" + std::to_string(index) + "_INITVAL")] = Property(0xFF00, 16);
        replace_port(ff, id_DI, lc, ctx->id("D" + std::to_string(index)));
        replace_port(ff, id_M, lc, ctx->id("M" + std::to_string(index)));
        connect_ports(ctx, lc, ctx->id("F" + std::to_string(index)), lc, ctx->id("DI" + std::to_string(index)));
    } else {
        if (driven_by_lut) {
            replace_port(ff, id_DI, lc, ctx->id("DI" + std::to_string(index)));
        } else {
            replace_port(ff, id_DI, lc, ctx->id("M" + std::to_string(index)));
        }
    }
}

void lut_to_slice(Context *ctx, CellInfo *lut, CellInfo *lc, int index)
{
    if (lc->hierpath == IdString())
        lc->hierpath = lut->hierpath;
    lc->params[ctx->id("LUT" + std::to_string(index) + "_INITVAL")] =
            get_or_default(lut->params, id_INIT, Property(0, 16));
    replace_port(lut, id_A, lc, ctx->id("A" + std::to_string(index)));
    replace_port(lut, id_B, lc, ctx->id("B" + std::to_string(index)));
    replace_port(lut, id_C, lc, ctx->id("C" + std::to_string(index)));
    replace_port(lut, id_D, lc, ctx->id("D" + std::to_string(index)));
    replace_port(lut, id_Z, lc, ctx->id("F" + std::to_string(index)));
}

void ccu2c_to_slice(Context *ctx, CellInfo *ccu, CellInfo *lc)
{
    if (lc->hierpath == IdString())
        lc->hierpath = ccu->hierpath;
    lc->params[id_MODE] = std::string("CCU2");
    lc->params[id_LUT0_INITVAL] = get_or_default(ccu->params, id_INIT0, Property(0, 16));
    lc->params[id_LUT1_INITVAL] = get_or_default(ccu->params, id_INIT1, Property(0, 16));

    lc->params[id_CCU2_INJECT1_0] = str_or_default(ccu->params, id_INJECT1_0, "YES");
    lc->params[id_CCU2_INJECT1_1] = str_or_default(ccu->params, id_INJECT1_1, "YES");

    replace_port(ccu, id_CIN, lc, id_FCI);

    replace_port(ccu, id_A0, lc, id_A0);
    replace_port(ccu, id_B0, lc, id_B0);
    replace_port(ccu, id_C0, lc, id_C0);
    replace_port(ccu, id_D0, lc, id_D0);

    replace_port(ccu, id_A1, lc, id_A1);
    replace_port(ccu, id_B1, lc, id_B1);
    replace_port(ccu, id_C1, lc, id_C1);
    replace_port(ccu, id_D1, lc, id_D1);

    replace_port(ccu, id_S0, lc, id_F0);
    replace_port(ccu, id_S1, lc, id_F1);

    replace_port(ccu, id_COUT, lc, id_FCO);
}

void dram_to_ramw(Context *ctx, CellInfo *ram, CellInfo *lc)
{
    if (lc->hierpath == IdString())
        lc->hierpath = ram->hierpath;
    lc->params[id_MODE] = std::string("RAMW");
    replace_port(ram, ctx->id("WAD[0]"), lc, id_D0);
    replace_port(ram, ctx->id("WAD[1]"), lc, id_B0);
    replace_port(ram, ctx->id("WAD[2]"), lc, id_C0);
    replace_port(ram, ctx->id("WAD[3]"), lc, id_A0);

    replace_port(ram, ctx->id("DI[0]"), lc, id_C1);
    replace_port(ram, ctx->id("DI[1]"), lc, id_A1);
    replace_port(ram, ctx->id("DI[2]"), lc, id_D1);
    replace_port(ram, ctx->id("DI[3]"), lc, id_B1);
}

static unsigned get_dram_init(const Context *ctx, const CellInfo *ram, int bit)
{
    auto init_prop = get_or_default(ram->params, id_INITVAL, Property(0, 64));
    NPNR_ASSERT(!init_prop.is_string);
    const std::string &idata = init_prop.str;
    NPNR_ASSERT(idata.length() == 64);
    unsigned value = 0;
    for (int i = 0; i < 16; i++) {
        char c = idata.at(4 * i + bit);
        if (c == '1')
            value |= (1 << i);
        else
            NPNR_ASSERT(c == '0' || c == 'x');
    }
    return value;
}

void dram_to_ram_slice(Context *ctx, CellInfo *ram, CellInfo *lc, CellInfo *ramw, int index)
{
    if (lc->hierpath == IdString())
        lc->hierpath = ram->hierpath;
    lc->params[id_MODE] = std::string("DPRAM");
    lc->params[id_WREMUX] = str_or_default(ram->params, id_WREMUX, "WRE");
    lc->params[id_WCKMUX] = str_or_default(ram->params, id_WCKMUX, "WCK");

    unsigned permuted_init0 = 0, permuted_init1 = 0;
    unsigned init0 = get_dram_init(ctx, ram, index * 2), init1 = get_dram_init(ctx, ram, index * 2 + 1);

    for (int i = 0; i < 16; i++) {
        int permuted_addr = 0;
        if (i & 1)
            permuted_addr |= 8;
        if (i & 2)
            permuted_addr |= 2;
        if (i & 4)
            permuted_addr |= 4;
        if (i & 8)
            permuted_addr |= 1;
        if (init0 & (1 << permuted_addr))
            permuted_init0 |= (1 << i);
        if (init1 & (1 << permuted_addr))
            permuted_init1 |= (1 << i);
    }

    lc->params[id_LUT0_INITVAL] = Property(permuted_init0, 16);
    lc->params[id_LUT1_INITVAL] = Property(permuted_init1, 16);

    if (ram->ports.count(ctx->id("RAD[0]"))) {
        connect_port(ctx, ram->ports.at(ctx->id("RAD[0]")).net, lc, id_D0);
        connect_port(ctx, ram->ports.at(ctx->id("RAD[0]")).net, lc, id_D1);
    }
    if (ram->ports.count(ctx->id("RAD[1]"))) {
        connect_port(ctx, ram->ports.at(ctx->id("RAD[1]")).net, lc, id_B0);
        connect_port(ctx, ram->ports.at(ctx->id("RAD[1]")).net, lc, id_B1);
    }
    if (ram->ports.count(ctx->id("RAD[2]"))) {
        connect_port(ctx, ram->ports.at(ctx->id("RAD[2]")).net, lc, id_C0);
        connect_port(ctx, ram->ports.at(ctx->id("RAD[2]")).net, lc, id_C1);
    }
    if (ram->ports.count(ctx->id("RAD[3]"))) {
        connect_port(ctx, ram->ports.at(ctx->id("RAD[3]")).net, lc, id_A0);
        connect_port(ctx, ram->ports.at(ctx->id("RAD[3]")).net, lc, id_A1);
    }

    if (ram->ports.count(id_WRE))
        connect_port(ctx, ram->ports.at(id_WRE).net, lc, id_WRE);
    if (ram->ports.count(id_WCK))
        connect_port(ctx, ram->ports.at(id_WCK).net, lc, id_WCK);

    connect_ports(ctx, ramw, id_WADO0, lc, id_WAD0);
    connect_ports(ctx, ramw, id_WADO1, lc, id_WAD1);
    connect_ports(ctx, ramw, id_WADO2, lc, id_WAD2);
    connect_ports(ctx, ramw, id_WADO3, lc, id_WAD3);

    if (index == 0) {
        connect_ports(ctx, ramw, id_WDO0, lc, id_WD0);
        connect_ports(ctx, ramw, id_WDO1, lc, id_WD1);

        replace_port(ram, ctx->id("DO[0]"), lc, id_F0);
        replace_port(ram, ctx->id("DO[1]"), lc, id_F1);

    } else if (index == 1) {
        connect_ports(ctx, ramw, id_WDO2, lc, id_WD0);
        connect_ports(ctx, ramw, id_WDO3, lc, id_WD1);

        replace_port(ram, ctx->id("DO[2]"), lc, id_F0);
        replace_port(ram, ctx->id("DO[3]"), lc, id_F1);
    } else {
        NPNR_ASSERT_FALSE("bad DPRAM index");
    }
}

void nxio_to_tr(Context *ctx, CellInfo *nxio, CellInfo *trio, std::vector<std::unique_ptr<CellInfo>> &created_cells,
                pool<IdString> &todelete_cells)
{
    if (nxio->type == ctx->id("$nextpnr_ibuf")) {
        trio->params[id_DIR] = std::string("INPUT");
        replace_port(nxio, id_O, trio, id_O);
    } else if (nxio->type == ctx->id("$nextpnr_obuf")) {
        trio->params[id_DIR] = std::string("OUTPUT");
        replace_port(nxio, id_I, trio, id_I);
    } else if (nxio->type == ctx->id("$nextpnr_iobuf")) {
        // N.B. tristate will be dealt with below
        NetInfo *i = get_net_or_empty(nxio, id_I);
        if (i == nullptr || i->driver.cell == nullptr)
            trio->params[id_DIR] = std::string("INPUT");
        else {
            log_info("%s: %s.%s\n", ctx->nameOf(i), ctx->nameOf(i->driver.cell), ctx->nameOf(i->driver.port));
            trio->params[id_DIR] = std::string("BIDIR");
        }
        replace_port(nxio, id_I, trio, id_I);
        replace_port(nxio, id_O, trio, id_O);
    } else {
        NPNR_ASSERT(false);
    }
    NetInfo *donet = trio->ports.at(id_I).net, *dinet = trio->ports.at(id_O).net;

    // Rename I/O nets to avoid conflicts
    if (donet != nullptr && donet->name == nxio->name)
        rename_net(ctx, donet, ctx->id(donet->name.str(ctx) + "$TRELLIS_IO_OUT"));
    if (dinet != nullptr && dinet->name == nxio->name)
        rename_net(ctx, dinet, ctx->id(dinet->name.str(ctx) + "$TRELLIS_IO_IN"));

    if (ctx->nets.count(nxio->name)) {
        int i = 0;
        IdString new_name;
        do {
            new_name = ctx->id(nxio->name.str(ctx) + "$rename$" + std::to_string(i++));
        } while (ctx->nets.count(new_name));
        rename_net(ctx, ctx->nets.at(nxio->name).get(), new_name);
    }

    // Create a new top port net for accurate IO timing analysis and simulation netlists
    if (ctx->ports.count(nxio->name)) {
        IdString tn_netname = nxio->name;
        NPNR_ASSERT(!ctx->nets.count(tn_netname));
        ctx->net_aliases.erase(tn_netname);
        NetInfo *toplevel_net = ctx->createNet(tn_netname);
        toplevel_net->name = tn_netname;
        connect_port(ctx, toplevel_net, trio, id_B);
        ctx->ports[nxio->name].net = toplevel_net;
    }

    CellInfo *tbuf = net_driven_by(
            ctx, donet, [](const Context *ctx, const CellInfo *cell) { return cell->type == ctx->id("$_TBUF_"); },
            id_Y);
    if (tbuf) {
        replace_port(tbuf, id_A, trio, id_I);
        // Need to invert E to form T
        std::unique_ptr<CellInfo> inv_lut = create_ecp5_cell(ctx, id_LUT4, trio->name.str(ctx) + "$invert_T");
        replace_port(tbuf, id_E, inv_lut.get(), id_A);
        inv_lut->params[id_INIT] = Property(21845, 16);
        connect_ports(ctx, inv_lut.get(), id_Z, trio, id_T);
        created_cells.push_back(std::move(inv_lut));

        if (donet->users.size() > 1) {
            for (auto user : donet->users)
                log_info("     remaining tristate user: %s.%s\n", user.cell->name.c_str(ctx), user.port.c_str(ctx));
            log_error("unsupported tristate IO pattern for IO buffer '%s', "
                      "instantiate SB_IO manually to ensure correct behaviour\n",
                      nxio->name.c_str(ctx));
        }
        ctx->nets.erase(donet->name);
        todelete_cells.insert(tbuf->name);
    }
}

NEXTPNR_NAMESPACE_END