LuatOS/luat/modules/luat_lib_rtc.c

/*
@module  rtc
@summary 实时时钟
@version 1.0
@date    2021.08.31
@demo rtc
@tag LUAT_USE_RTC
*/
#include "luat_base.h"
#include "luat_rtc.h"

#define LUAT_LOG_TAG "rtc"
#include "luat_log.h"

int Base_year = 1900;

#ifndef LUAT_COMPILER_NOWEAK
void LUAT_WEAK luat_rtc_set_tamp32(uint32_t tamp) {
    LLOGD("not support yet");
}
#endif

/*
设置时钟
@api rtc.set(val)
@table/int 时钟参数,见示例
@return bool 成功返回true,否则返回nil或false
@usage
rtc.set({year=2021,mon=8,day=31,hour=17,min=8,sec=43})
rtc.set(1652230554)
*/
static int l_rtc_set(lua_State *L){
    struct tm tblock = {0};
    int ret;
    if (!lua_istable(L, 1)) {
    	if (lua_isinteger(L, 1))
    	{
    		uint32_t tamp = lua_tointeger(L, 1);
    	    luat_rtc_set_tamp32(tamp);
    	    lua_pushboolean(L, 1);
    	    return 1;
    	}
        LLOGW("rtc time need table");
        return 0;
    }
    lua_settop(L, 1);

    lua_pushstring(L, "year");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_year = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss year");
        return 0;
    }

    lua_pushstring(L, "mon");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_mon = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss mon");
        return 0;
    }

    lua_pushstring(L, "day");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_mday = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss day");
        return 0;
    }

    lua_pushstring(L, "hour");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_hour = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss hour");
        return 0;
    }

    lua_pushstring(L, "min");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_min = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss min");
        return 0;
    }

    lua_pushstring(L, "sec");
    lua_gettable(L, 1);
    if (lua_isnumber(L, -1)) {
        tblock.tm_sec = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss sec");
        return 0;
    }

    tblock.tm_year -= Base_year;
    tblock.tm_mon -= 1;

    ret = luat_rtc_set(&tblock);
    lua_pushboolean(L, ret == 0 ? 1 : 0);
    return 1;
}

/*
获取时钟
@api rtc.get()
@return table 时钟参数,见示例
@usage
local t = rtc.get()
-- {year=2021,mon=8,day=31,hour=17,min=8,sec=43}
log.info("rtc", json.encode(t))
*/
static int l_rtc_get(lua_State *L){
    struct tm tblock = {0};
    int ret;
    ret = luat_rtc_get(&tblock);
    if (ret) {
        return 0;
    }

    tblock.tm_year += Base_year;
    tblock.tm_mon += 1;


    lua_newtable(L);

    lua_pushstring(L, "year");
    lua_pushinteger(L, tblock.tm_year );
    lua_settable(L, -3);

    lua_pushstring(L, "mon");
    lua_pushinteger(L, tblock.tm_mon );
    lua_settable(L, -3);

    lua_pushstring(L, "day");
    lua_pushinteger(L, tblock.tm_mday);
    lua_settable(L, -3);

    lua_pushstring(L, "hour");
    lua_pushinteger(L, tblock.tm_hour);
    lua_settable(L, -3);

    lua_pushstring(L, "min");
    lua_pushinteger(L, tblock.tm_min);
    lua_settable(L, -3);

    lua_pushstring(L, "sec");
    lua_pushinteger(L, tblock.tm_sec);
    lua_settable(L, -3);

    return 1;
}

/*
设置RTC唤醒时间
@api rtc.timerStart(id, tab)
@int 时钟id,通常只支持0
@table 时钟参数,见示例
@return bool 成功返回true,否则返回nil或false
@usage
-- 目前该接口不适用于移芯模块780E/700E/780EP系列，需要定时唤醒可使用pm.dtimerStart()
-- 使用前建议先rtc.set设置为正确的时间
rtc.timerStart(0, {year=2021,mon=9,day=1,hour=17,min=8,sec=43})
*/
static int l_rtc_timer_start(lua_State *L){
    int id;
    struct tm tblock = {0};
    int ret;

    id = luaL_checkinteger(L, 1);
    if (!lua_istable(L, 2)) {
        LLOGW("rtc time need table");
        return 0;
    }

    lua_settop(L, 2);
    lua_pushstring(L, "year");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_year = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss year");
        return 0;
    }

    lua_pushstring(L, "mon");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_mon = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss mon");
        return 0;
    }

    lua_pushstring(L, "day");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_mday = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss day");
        return 0;
    }

    lua_pushstring(L, "hour");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_hour = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss hour");
        return 0;
    }

    lua_pushstring(L, "min");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_min = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss min");
        return 0;
    }

    lua_pushstring(L, "sec");
    lua_gettable(L, 2);
    if (lua_isnumber(L, -1)) {
        tblock.tm_sec = luaL_checkinteger(L, -1);
    }
    else {
        LLOGW("rtc time miss sec");
        return 0;
    }

    tblock.tm_year -= Base_year;
    tblock.tm_mon -= 1;

    ret = luat_rtc_timer_start(id, &tblock);
    lua_pushboolean(L, ret == 0 ? 1 : 0);
    return 1;
}

/*
取消RTC唤醒时间
@api rtc.timerStop(id)
@int 时钟id,通常只支持0
@return bool 成功返回true,否则返回nil或false
@usage
rtc.timerStop(0)
*/
static int l_rtc_timer_stop(lua_State *L){
    int id;
    int ret;

    id = luaL_checkinteger(L, 1);
    ret = luat_rtc_timer_stop(id);
    lua_pushboolean(L, ret == 0 ? 1 : 0);
    return 1;
}

static int l_rtc_set_base_year(lua_State *L){
    Base_year = luaL_checkinteger(L, 1);
    return 0;
}

/*
读取或设置时区
@api rtc.timezone(tz)
@int 时区值,注意单位是1/4时区.例如东八区是 32,而非8. 可以不传
@return int 当前/设置后的时区值
@usage
-- 设置为东8区
rtc.timezone(32)
-- 设置为东3区
rtc.timezone(12)
-- 设置为西4区
rtc.timezone(-16)
-- 注意: 无论设置时区是多少, rtc.get/set总是UTC时间
-- 时区影响的是 os.date/os.time 函数
-- 只有部分模块支持设置时区, 且默认值一般为32, 即东八区
*/
static int l_rtc_timezone(lua_State *L){
    int timezone = 0;
    if (lua_isinteger(L, 1)) {
        timezone = luaL_checkinteger(L, 1);
        timezone = luat_rtc_timezone(&timezone);
    }
    else {
        timezone = luat_rtc_timezone(NULL);
    }
    lua_pushinteger(L, timezone);
    return 1;
}


#include "rotable2.h"
static const rotable_Reg_t reg_rtc[] =
{
    { "set",        ROREG_FUNC(l_rtc_set)},
    { "get",        ROREG_FUNC(l_rtc_get)},
    { "timerStart", ROREG_FUNC(l_rtc_timer_start)},
    { "timerStop",  ROREG_FUNC(l_rtc_timer_stop)},
    { "setBaseYear", ROREG_FUNC(l_rtc_set_base_year)},
    { "timezone",   ROREG_FUNC(l_rtc_timezone)},
	{ NULL,         ROREG_INT(0) }
};

LUAMOD_API int luaopen_rtc( lua_State *L ) {
    luat_newlib2(L, reg_rtc);
    return 1;
}