/* all about gd factory */

#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <pthread.h>
#include <errno.h>
#include <unistd.h>
#include <hal_timer.h>

#include "inter.h"

#define NOR_GD_QE_BIT BIT(1)
#define NOR_GD_CMP_BIT BIT(6)
#define NOR_GD_CMD_RDSR2 0x35
#define NOR_GD_CMD_WRSR2 0x31

static struct nor_info idt_gd[] =
{
    {
        .model = "GD25Q127C",
        .id = {0xc8, 0x40, 0x18},
        .total_size = SZ_16M,
        .flag = SUPPORT_GENERAL,
    },
    {
        .model = "GD25Q256",
        .id = {0xc8, 0x40, 0x19},
        .total_size = SZ_16M,
        .flag = SUPPORT_GENERAL,
    },
};

static int nor_gd_quad_mode(struct nor_flash *unused)
{
    int ret;
    unsigned char cmd[3];
    char reg[2] = {0};

    cmd[0] = NOR_GD_CMD_RDSR2;
    ret = nor_transfer(1, cmd, 1, reg, 2);
    if (ret) {
        SPINOR_ERR("read status register2 fail\n");
        return ret;
    }

    if (reg[1] & NOR_GD_QE_BIT)
        return 0;

    ret = nor_write_enable();
    if (ret)
        return ret;

    cmd[0] = NOR_GD_CMD_WRSR2;
    cmd[1] = reg[1] | NOR_GD_QE_BIT;
    ret = nor_transfer(2, cmd, 2, NULL, 0);
    if (ret) {
        SPINOR_ERR("set status register fail\n");
        return ret;
    }

    if (nor_wait_ready(0, 500)) {
        SPINOR_ERR("wait set qd mode failed\n");
        return -EBUSY;
    }

    cmd[0] = NOR_GD_CMD_RDSR2;
    ret = nor_transfer(1, cmd, 1, reg, 2);
    if (ret) {
        SPINOR_ERR("read status register2 fail\n");
        return ret;
    }

    if (!(reg[1] & NOR_GD_QE_BIT)) {
        SPINOR_ERR("set gd QE failed\n");
        return -EINVAL;
    }
    return 0;
}

static struct nor_factory nor_gd = {
    .factory = FACTORY_GD,
    .idt = idt_gd,
    .idt_cnt = sizeof(idt_gd),

    .init = NULL,
    .deinit = NULL,
    .init_lock = NULL,
    .deinit_lock = NULL,
    .lock = NULL,
    .unlock = NULL,
    .islock = NULL,
    .set_quad_mode = nor_gd_quad_mode,
    .set_4bytes_addr = NULL,
};

int nor_register_factory_gd(void)
{
    return nor_register_factory(&nor_gd);
}