/*
@module  mcu
@summary 封装mcu一些特殊操作
@version core V0007
@date    2021.08.18
@tag LUAT_USE_MCU
*/
#include "luat_base.h"
#include "luat_mcu.h"
#include "luat_zbuff.h"
#include "luat_spi.h"
#include "luat_i2c.h"
#include "luat_uart.h"
#define LUAT_LOG_TAG "mcu"
#include "luat_log.h"
/*
设置主频,单位MHZ
@api mcu.setClk(mhz)
@int 主频,根据设备的不同有不同的有效值,请查阅手册
@return bool 成功返回true,否则返回false
@usage

-- 注意: 并非所有模块都支持主频调整,请查阅手册
-- Air101/Air103/Air601 支持设置成 2/40/80/160/240. 特别提醒, 设置到2M后, 如果要休眠, 必须先设置到80M
-- ESP32系列支持设置成 40/80/160/240 , 需要2024.1.1之后的固件
-- Air780系列, Air105, 不支持设置主频
-- Air780系列, 进入休眠模式时自动降频到24M

-- 设置到80MHZ
mcu.setClk(80)
sys.wait(1000)
-- 设置到240MHZ
mcu.setClk(240)
sys.wait(1000)
-- 设置到2MHZ
mcu.setClk(2)
sys.wait(1000)
*/
static int l_mcu_set_clk(lua_State* L) {
    int ret = luat_mcu_set_clk((size_t)luaL_checkinteger(L, 1));
    lua_pushboolean(L, ret == 0 ? 1 : 0);
    return 1;
}

/*
获取主频,单位MHZ
@api mcu.getClk()
@return int 若失败返回-1,否则返回主频数值,若等于0,可能处于32k晶振的省电模式
@usage
local mhz = mcu.getClk()
print("Boom", mhz)
*/
static int l_mcu_get_clk(lua_State* L) {
    int mhz = luat_mcu_get_clk();
    lua_pushinteger(L, mhz);
    return 1;
}

/*
获取设备唯一id. 注意,可能包含不可见字符,如需查看建议toHex()后打印
@api mcu.unique_id()
@return string 设备唯一id.若不支持, 会返回空字符串.
@usage
local unique_id = mcu.unique_id()
print("unique_id", unique_id)
*/
static int l_mcu_unique_id(lua_State* L) {
    size_t len = 0;
    const char* id = luat_mcu_unique_id(&len);
    lua_pushlstring(L, id, len);
    return 1;
}

/*
获取启动后的tick数,本身是无符号值,范围0~0xffffffff,lua是有符号计算,计算时超过0x7fffffff会变负数
@api mcu.ticks()
@return int 当前tick值
@usage
local tick = mcu.ticks()
print("ticks", tick)
-- 如需不会溢出的值, 可用mcu.ticks2(), 于2024.5.7新增
*/
static int l_mcu_ticks(lua_State* L) {
    long tick = luat_mcu_ticks();
    lua_pushinteger(L, tick);
    return 1;
}

/*
获取每秒的tick数量
@api mcu.hz()
@return int 每秒的tick数量,通常为1000
@usage
local hz = mcu.hz()
print("mcu.hz", hz)
*/
static int l_mcu_hz(lua_State* L) {
    uint32_t hz = luat_mcu_hz();
    lua_pushinteger(L, hz);
    return 1;
}

/*
读写mcu的32bit寄存器或者ram,谨慎使用写功能,请熟悉mcu的寄存器使用方法后再使用
@api mcu.reg32(address, value, mask)
@int 寄存器或者ram地址
@int 写入的值,如果没有,则直接返回当前值
@int 位掩码,可以对特定几个位置的bit做修改, 默认0xffffffff,修改全部32bit
@return int 返回当前寄存的值
@usage
local value = mcu.reg32(0x2009FFFC, 0x01, 0x01) --对0x2009FFFC地址上的值,修改bit0为1
*/
static int l_mcu_reg32(lua_State* L) {
	uint32_t addr = luaL_checkinteger(L, 1);
    volatile uint32_t *address = (uint32_t *)(addr & 0xfffffffc);
    if (lua_isinteger(L, 2)) {
    	volatile uint32_t value = lua_tointeger(L, 2);
    	volatile uint32_t mask = luaL_optinteger(L, 3, 0xffffffff);
    	volatile uint32_t org = *address;
    	*address = (org & ~mask)| (value & mask);
    	lua_pushinteger(L, *address);
    } else {
    	lua_pushinteger(L, *address);
    }
    return 1;
}

/*
转换10进制数为16进制字符串输出
@api mcu.x32(value)
@int 需要转换的值
@return string 16进制字符串
@usage
local value = mcu.x32(0x2009FFFC) --输出"0x2009fffc"
*/
static int l_mcu_x32(lua_State* L) {
    uint32_t value = luaL_checkinteger(L, 1);
    char c[16];
    sprintf_(c, "0x%lx", value);
    lua_pushstring(L, c);
    return 1;
}

// #ifdef __LUATOS_TICK_64BIT__
/*
获取启动后的高精度tick，如果支持bit64库，可以直接输出转换好的bit64结构
@api mcu.tick64()
@boolean 是否输出bit64结构,true是,其他都是false,留空也是false,用于兼容旧的demo
@return string 当前tick值,8个字节的uint64,如果支持64bit库,同时要求输出64bit结构的话,会输出9字节的string
@return int 1us有几个tick,0表示未知
@usage
local tick_str, tick_per = mcu.tick64()
print("ticks", tick_str, tick_per)
*/
static int l_mcu_hw_tick64(lua_State* L) {

    uint64_t tick = luat_mcu_tick64();
    uint32_t us_period = luat_mcu_us_period();
#ifdef LUAT_USE_BIT64
	if (lua_isboolean(L, 1) && lua_toboolean(L, 1))
	{
		uint8_t data[9] = {0};
		memcpy(data, &tick, 8);
		lua_pushlstring(L, (const char*)data, 9);
	}
	else
	{
		lua_pushlstring(L, (const char*)&tick, 8);
	}
#else
    lua_pushlstring(L, (const char*)&tick, 8);
#endif
    lua_pushinteger(L, us_period);
    return 2;
}

/*
计算2个64bit tick的差值
@api mcu.dtick64(tick1, tick2, check_value)
@string 64bit的string
@string 64bit的string
@int 参考值,可选项,如果为0,则返回结果中第一个项目为true
@return boolean 与参考值比较,如果大于等于为true,反之为false
@return int 差值tick1 - tick2,如果超过了0x7fffffff,结果可能是错的
@usage
local result, diff_tick = mcu.dtick64(tick1, tick2)
print("ticks", result, diff_tick)
*/
static int l_mcu_hw_diff_tick64(lua_State* L) {
	uint64_t tick1, tick2;
	int64_t diff;
	int check_value = 0;
    size_t len1;
    const char *data1 = luaL_checklstring(L, 1, &len1);
    size_t len2;
    const char *data2 = luaL_checklstring(L, 2, &len2);
    check_value = luaL_optinteger(L, 3, 0);

    memcpy(&tick1, data1, len1);
    memcpy(&tick2, data2, len2);
    diff = tick1 - tick2;
    lua_pushboolean(L, (diff >= (int64_t)check_value)?1:0);
    lua_pushinteger(L, diff);
    return 2;
}

/*
选择时钟源,当前仅air105支持
@api mcu.setXTAL(source_main, source_32k, delay)
@boolean 高速时钟是否使用外部时钟源,如果为空则不改变
@boolean 低速32K是否使用外部时钟源,如果为空则不改变
@int PLL稳定时间,在切换高速时钟的时候,根据硬件环境,需要delay一段时间等待PLL稳定,默认是1200,建议不小于1024
@usage
mcu.setXTAL(true, true, 1248)	--高速时钟使用外部时钟,低速32K使用外部晶振, delay1248
*/
static int l_mcu_set_xtal(lua_State* L) {
	int source_main = 255;
	int source_32k = 255;
	int delay = luaL_optinteger(L, 3, 1200);
	if (lua_isboolean(L, 1)) {
		source_main = lua_toboolean(L, 1);
	}
	if (lua_isboolean(L, 2)) {
		source_32k = lua_toboolean(L, 2);
	}
	luat_mcu_set_clk_source(source_main, source_32k, delay);
    return 0;
}
// #endif
#ifdef LUAT_COMPILER_NOWEAK
#else
LUAT_WEAK void luat_mcu_set_hardfault_mode(int mode) {;}
LUAT_WEAK void luat_mcu_xtal_ref_output(uint8_t main_enable, uint8_t slow_32k_enable) {;}
LUAT_WEAK int luat_uart_pre_setup(int uart_id, uint8_t use_alt_type){return -1;}
LUAT_WEAK int luat_i2c_set_iomux(int id, uint8_t value){return -1;}
#endif
/*
mcu死机时处理模式
@api mcu.hardfault(mode)
@int 处理模式，0死机停机，1死机后重启，2死机后尽量将错误信息提交给外部工具后重启 3.死机时写入关键信息到flash后立刻重启
@usage
mcu.hardfault(0)	--死机后停机，一般用于调试状态
mcu.hardfault(1)	--死机后重启，一般用于正式产品
mcu.hardfault(2)	--死机后尽量将错误信息提交给外部工具后重启，一般用于压力测试或者正式产品
*/
static int l_mcu_set_hardfault_mode(lua_State* L)
{
	luat_mcu_set_hardfault_mode(luaL_optinteger(L, 1, 0));
	return 0;
}

/*
在外设打开前，将外设IO复用到非默认配置上，目前只支持Air780E的部分外设复用到其他配置，这是一个临时接口，如果后续有更合适的api，本接口将不再更新
@api mcu.iomux(type, channel, value)
@int 外设类型，目前只有mcu.UART,mcu.I2C
@int 总线序号，0~N，
@int 新的配置，这个需要根据具体平台决定
@usage
mcu.iomux(mcu.UART, 2, 1)	-- Air780E的UART2复用到gpio12和gpio13(Air780EG默认是这个复用，不要动)
mcu.iomux(mcu.UART, 2, 2)	-- Air780E的UART2复用到gpio6和gpio7
mcu.iomux(mcu.I2C, 0, 1)	-- Air780E的I2C0复用到gpio12和gpio13
mcu.iomux(mcu.I2C, 0, 2)	-- Air780E的I2C0复用到gpio16和gpio17
mcu.iomux(mcu.I2C, 1, 1)	-- Air780E的I2C1复用到gpio4和gpio5
 */
static int l_mcu_iomux(lua_State* L)
{
	int type = luaL_optinteger(L, 1, 0xff);
	int channel = luaL_optinteger(L, 2, 0xff);
	int value = luaL_optinteger(L, 3, 0);
	LLOGD("mcu iomux %d,%d,%d", type, channel, value);
	switch(type)
	{
	case LUAT_MCU_PERIPHERAL_UART:
		luat_uart_pre_setup(channel, value);
		break;
	case LUAT_MCU_PERIPHERAL_I2C:
		luat_i2c_set_iomux(channel, value);
		break;
	}
	return 0;
}

/*
IO外设功能复用选择，注意普通MCU通常是以GPIO号为唯一ID号，但是专用SOC，比如CAT1的，可能以PAD号或者模块pin脚号(pin.xxx后续支持)为唯一ID号。本函数不是所有平台适用
@api mcu.altfun(type, sn, pad_index, alt_fun, is_input)
@int 外设类型，目前有mcu.UART,mcu.I2C,mcu.SPI,mcu.PWM,mcu.GPIO,mcu.I2S,mcu.LCD,mcu.CAM，具体需要看平台
@int 总线序号，0~N，如果是mcu.GPIO，则是GPIO号。具体看平台的IOMUX复用表
@int 唯一ID号，如果留空不写，则表示清除配置，使用平台的默认配置。具体看平台的IOMUX复用表
@int 复用功能序号，0~N。具体看平台的IOMUX复用表
@boolean 是否是输入功能，true是，留空是false
@usage
-- UART2复用到paddr 25/26 alt 3
mcu.altfun(mcu.UART,2,  25, 3, true)
mcu.altfun(mcu.UART,2,  26, 3, false)

-- Air8101的SDIO复用演示, 1线模式, 暂不支持4线
mcu.altfun(mcu.SDI0,0,14)
mcu.altfun(mcu.SDI0,0,15)
mcu.altfun(mcu.SDI0,0,16)
// mcu.altfun(mcu.SDI0,0,17)
// mcu.altfun(mcu.SDI0,0,18)
// mcu.altfun(mcu.SDI0,0,19)
 */
static int l_mcu_alt_ctrl(lua_State* L)
{
#ifdef LUAT_MCU_IOMUX_CTRL
	int type = luaL_optinteger(L, 1, 0xff);
	int sn = luaL_optinteger(L, 2, 0xff);
	int pad = luaL_optinteger(L, 3, -1);
	int alt_fun = luaL_optinteger(L, 4, 0);
	uint8_t is_input = 0;
	if (lua_isboolean(L, 5))
	{
		is_input = lua_toboolean(L, 5);
	}
	LLOGD("mcu altfun %d,%d,%d,%d,%d", type, sn, pad, alt_fun, is_input);
	luat_mcu_iomux_ctrl(type, sn, pad, alt_fun, is_input);
#else
	LLOGW("no support mcu.altfun");
#endif
	return 0;
}

/*
获取高精度的计数
@api mcu.ticks2(mode)
@int 模式, 看后面的用法说明
@return int 根据mode的不同,返回值的含义不同
@return int 根据mode的不同,返回值的含义不同
@usage
-- 本函数于2024.5.7新增
-- 与mcu.ticks()的区别是,底层计数器是64bit的, 在可预计的将来不会溢出
-- 所以本函数返回的值总是递增的, 而且32bit固件也能处理

-- 模式可选值 及 对应的返回值
-- 0: 返回微秒数, 以秒为分割, 例如 1234567890us 返回2个值: 1234, 567890
-- 1: 返回毫秒数, 以千秒为分割, 例如 1234567890ms 返回2个值: 1234, 567890
-- 2: 返回秒数,   以百万秒为分割, 例如 1234567890s  返回2个值: 1234, 567890

local us_h, us_l = mcu.ticks2(0)
local ms_h, ms_l = mcu.ticks2(1)
local sec_h, sec_l = mcu.ticks2(2)
log.info("us_h", us_h, "us_l", us_l)
log.info("ms_h", ms_h, "ms_l", ms_l)
log.info("sec_h", sec_h, "sec_l", sec_l)
*/
static int l_mcu_ticks2(lua_State* L) {
	int mode = luaL_optinteger(L, 1, 0);
    uint64_t tick = luat_mcu_tick64();
    uint32_t us_period = luat_mcu_us_period();
	uint64_t us = tick / us_period;
	uint64_t ms = us / 1000;
	uint64_t sec = ms / 1000;
    switch (mode)
	{
	case 0:
		// 模式0, 返回微秒数
		lua_pushinteger(L, (uint32_t)(us / 1000000));
		lua_pushinteger(L, (uint32_t)(us % 1000000));
		return 2;
	case 1:
		// 模式1, 返回毫秒数
		lua_pushinteger(L, (uint32_t)(ms / 1000000));
		lua_pushinteger(L, (uint32_t)(ms % 1000000));
		return 2;
	case 2:
		// 模式2, 返回秒数
		lua_pushinteger(L, (uint32_t)(sec / 1000000));
		lua_pushinteger(L, (uint32_t)(sec % 1000000));
		return 2;
	default:
		break;
	}
    return 0;
}


/*
晶振参考时钟输出
@api mcu.XTALRefOutput(source_main, source_32k)
@boolean 高速晶振参考时钟是否输出
@boolean 低速32K晶振参考时钟是否输出
@usage
-- 本函数于2024.5.17新增
-- 当前仅Air780EXXX系列支持
mcu.XTALRefOutput(true, false)	--高速晶振参考时钟输出,低速32K不输出
*/
static int l_mcu_xtal_ref_output(lua_State* L) {
	int source_main = 0;
	int source_32k = 0;
	// int delay = luaL_optinteger(L, 3, 1200);
	if (lua_isboolean(L, 1)) {
		source_main = lua_toboolean(L, 1);
	}
	if (lua_isboolean(L, 2)) {
		source_32k = lua_toboolean(L, 2);
	}
	luat_mcu_xtal_ref_output(source_main, source_32k);
    return 0;
}

LUAT_WEAK int luat_mcu_muid(char* buf) {return -1;}
static int l_mcu_muid(lua_State* L) {
	char muid[33] = {0};
	luat_mcu_muid(muid);
	// LLOGD("mcu muid %s", muid);
	lua_pushstring(L, muid);
    return 1;
}

#include "rotable2.h"
static const rotable_Reg_t reg_mcu[] =
{
    { "setClk" ,        ROREG_FUNC(l_mcu_set_clk)},
    { "getClk",         ROREG_FUNC(l_mcu_get_clk)},
    { "unique_id",      ROREG_FUNC(l_mcu_unique_id)},
    { "ticks",          ROREG_FUNC(l_mcu_ticks)},
    { "hz",             ROREG_FUNC(l_mcu_hz)},
	{ "reg32",          ROREG_FUNC(l_mcu_reg32)},
	{ "x32",            ROREG_FUNC(l_mcu_x32)},
// #ifdef __LUATOS_TICK_64BIT__
	{ "tick64",			ROREG_FUNC(l_mcu_hw_tick64)},
	{ "dtick64",		ROREG_FUNC(l_mcu_hw_diff_tick64)},
	{ "setXTAL",		ROREG_FUNC(l_mcu_set_xtal)},
	{ "hardfault",		ROREG_FUNC(l_mcu_set_hardfault_mode)},
#ifdef LUAT_MCU_IOMUX_CTRL
	{ "altfun",			ROREG_FUNC(l_mcu_alt_ctrl)},
#else
	{ "iomux",			ROREG_FUNC(l_mcu_iomux)},
#endif
    { "ticks2",         ROREG_FUNC(l_mcu_ticks2)},
	{ "XTALRefOutput",         ROREG_FUNC(l_mcu_xtal_ref_output)},
	{ "muid",         	ROREG_FUNC(l_mcu_muid)},
// #endif
	//@const UART number 外设类型-串口
	{ "UART",             ROREG_INT(LUAT_MCU_PERIPHERAL_UART) },
	//@const I2C number 外设类型-I2C
	{ "I2C",             ROREG_INT(LUAT_MCU_PERIPHERAL_I2C) },
	//@const SPI number 外设类型-SPI
	{ "SPI",             ROREG_INT(LUAT_MCU_PERIPHERAL_SPI) },
	//@const PWM number 外设类型-PWM
	{ "PWM",             ROREG_INT(LUAT_MCU_PERIPHERAL_PWM) },
	//@const PWM number 外设类型-CAN
	{ "CAN",             ROREG_INT(LUAT_MCU_PERIPHERAL_CAN) },
	//@const GPIO number 外设类型-GPIO
	{ "GPIO",             ROREG_INT(LUAT_MCU_PERIPHERAL_GPIO) },
	//@const I2S number 外设类型-I2S, 音频总线
	{ "I2S",             ROREG_INT(LUAT_MCU_PERIPHERAL_I2S) },
	//@const LCD number 外设类型-LCD, LCD专用总线
	{ "LCD",             ROREG_INT(LUAT_MCU_PERIPHERAL_LCD) },
	//@const CAM number 外设类型-CAM,与 mcu.CAMERA一样
	{ "CAM",             ROREG_INT(LUAT_MCU_PERIPHERAL_CAMERA) },
	//@const CAMERA number 外设类型-CAMERA，就是CAM,摄像头
	{ "CAMERA",          ROREG_INT(LUAT_MCU_PERIPHERAL_CAMERA) },
	//@const ONEWIRE number 外设类型-ONEWIRE，单总线协议
	{ "ONEWIRE",         ROREG_INT(LUAT_MCU_PERIPHERAL_ONEWIRE) },
	//@const SDIO number 外设类型-SDIO，接TF卡
	{ "SDIO",            ROREG_INT(LUAT_MCU_PERIPHERAL_SDIO) },
	//@const KEYBORAD number 外设类型-KEYBORAD，键盘
	{ "KEYBORAD",        ROREG_INT(LUAT_MCU_PERIPHERAL_KEYBORAD) },
	//@const ETH number 外设类型-ETH，网口
	{ "ETH",             ROREG_INT(LUAT_MCU_PERIPHERAL_ETH) },
	{ NULL,              ROREG_INT(0) }
};

LUAMOD_API int luaopen_mcu( lua_State *L ) {
    luat_newlib2(L, reg_mcu);
    return 1;
}