123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240 |
- // Module for RTC time keeping
- #include "module.h"
- #include "lauxlib.h"
- #include "rtc/rtctime_internal.h"
- #include "rtc/rtctime.h"
- /* seconds per day */
- #define SPD 24*60*60
- /* days per month -- nonleap! */
- static const short __spm[13] =
- { 0,
- (31),
- (31+28),
- (31+28+31),
- (31+28+31+30),
- (31+28+31+30+31),
- (31+28+31+30+31+30),
- (31+28+31+30+31+30+31),
- (31+28+31+30+31+30+31+31),
- (31+28+31+30+31+30+31+31+30),
- (31+28+31+30+31+30+31+31+30+31),
- (31+28+31+30+31+30+31+31+30+31+30),
- (31+28+31+30+31+30+31+31+30+31+30+31),
- };
- static int __isleap (int year) {
- /* every fourth year is a leap year except for century years that are
- * not divisible by 400. */
- /* return (year % 4 == 0 && (year % 100 != 0 || year % 400 == 0)); */
- return (!(year % 4) && ((year % 100) || !(year % 400)));
- }
- // ******* C API functions *************
- void __attribute__((noreturn)) TEXT_SECTION_ATTR rtc_time_enter_deep_sleep_final (void)
- {
- ets_intr_lock();
- Cache_Read_Disable();
- rtc_reg_write(0x18,8);
- rtc_reg_write_and_loop(0x08,0x00100000); // go to sleep
- __builtin_unreachable();
- }
- void rtctime_early_startup (void)
- {
- // Cache_Read_Enable (0, 0, 1);
- rtc_time_register_bootup ();
- rtc_time_switch_clocks ();
- // Cache_Read_Disable ();
- }
- void rtctime_late_startup (void)
- {
- rtc_time_switch_system ();
- }
- void rtctime_adjust_rate (int rate)
- {
- rtc_time_set_rate (rate);
- }
- void rtctime_gettimeofday (struct rtc_timeval *tv)
- {
- rtc_time_gettimeofday (tv);
- }
- void rtctime_settimeofday (const struct rtc_timeval *tv)
- {
- if (!rtc_time_check_magic ())
- rtc_time_prepare ();
- int32_t rate = rtc_time_get_rate();
- rtc_time_settimeofday (tv);
- rtc_time_set_rate(rate);
- }
- bool rtctime_have_time (void)
- {
- return rtc_time_have_time ();
- }
- void rtctime_deep_sleep_us (uint32_t us)
- {
- rtc_time_deep_sleep_us (us);
- }
- void rtctime_deep_sleep_until_aligned_us (uint32_t align_us, uint32_t min_us)
- {
- rtc_time_deep_sleep_until_aligned (align_us, min_us);
- }
- void rtctime_gmtime (const int32 stamp, struct rtc_tm *r)
- {
- int32_t i;
- int32_t work = stamp % (SPD);
- r->tm_sec = work % 60; work /= 60;
- r->tm_min = work % 60; r->tm_hour = work / 60;
- work = stamp / (SPD);
- r->tm_wday = (4 + work) % 7;
- for (i = 1970; ; ++i) {
- int32_t k = __isleap (i) ? 366 : 365;
- if (work >= k) {
- work -= k;
- } else {
- break;
- }
- }
- r->tm_year = i - 1900;
- r->tm_yday = work;
- r->tm_mday = 1;
- if (__isleap (i) && (work > 58)) {
- if (work == 59) r->tm_mday = 2; /* 29.2. */
- work -= 1;
- }
- for (i = 11; i && (__spm[i] > work); --i) ;
- r->tm_mon = i;
- r->tm_mday += work - __spm[i];
- }
- // ******* Lua API functions *************
- // rtctime.set (sec, usec)
- static int rtctime_set (lua_State *L)
- {
- if (!rtc_time_check_magic ())
- rtc_time_prepare ();
- uint32_t sec = luaL_checkinteger (L, 1);
- uint32_t usec = 0;
- if (lua_isnumber (L, 2))
- usec = lua_tointeger (L, 2);
- struct rtc_timeval tv = { sec, usec };
- rtctime_settimeofday (&tv);
- if (lua_isnumber(L, 3))
- rtc_time_set_rate(lua_tointeger(L, 3));
- return 0;
- }
- // sec, usec = rtctime.get ()
- static int rtctime_get (lua_State *L)
- {
- struct rtc_timeval tv;
- rtctime_gettimeofday (&tv);
- lua_pushinteger (L, tv.tv_sec);
- lua_pushinteger (L, tv.tv_usec);
- lua_pushinteger (L, rtc_time_get_rate());
- return 3;
- }
- static void do_sleep_opt (lua_State *L, int idx)
- {
- if (lua_isnumber (L, idx))
- {
- uint32_t opt = lua_tointeger (L, idx);
- if (opt < 0 || opt > 4)
- luaL_error (L, "unknown sleep option");
- system_deep_sleep_set_option (opt);
- }
- }
- // rtctime.adjust_delta (usec)
- static int rtctime_adjust_delta (lua_State *L)
- {
- uint32_t us = luaL_checkinteger (L, 1);
- lua_pushinteger(L, rtc_time_adjust_delta_by_rate(us));
- return 1;
- }
- // rtctime.dsleep (usec, option)
- static int rtctime_dsleep (lua_State *L)
- {
- uint32_t us = luaL_checkinteger (L, 1);
- do_sleep_opt (L, 2);
- rtctime_deep_sleep_us (us); // does not return
- return 0;
- }
- // rtctime.dsleep_aligned (aligned_usec, min_usec, option)
- static int rtctime_dsleep_aligned (lua_State *L)
- {
- if (!rtctime_have_time ())
- return luaL_error (L, "time not available, unable to align");
- uint32_t align_us = luaL_checkinteger (L, 1);
- uint32_t min_us = luaL_checkinteger (L, 2);
- do_sleep_opt (L, 3);
- rtctime_deep_sleep_until_aligned_us (align_us, min_us); // does not return
- return 0;
- }
- #define ADD_TABLE_ITEM(L, key, val) \
- lua_pushinteger (L, val); \
- lua_setfield (L, -2, key);
- // rtctime.epoch2cal (stamp)
- static int rtctime_epoch2cal (lua_State *L)
- {
- struct rtc_tm date;
- int32_t stamp = luaL_checkint (L, 1);
- luaL_argcheck (L, stamp >= 0, 1, "wrong arg range");
- rtctime_gmtime (stamp, &date);
- /* construct Lua table */
- lua_createtable (L, 0, 8);
- ADD_TABLE_ITEM (L, "yday", date.tm_yday + 1);
- ADD_TABLE_ITEM (L, "wday", date.tm_wday + 1);
- ADD_TABLE_ITEM (L, "year", date.tm_year + 1900);
- ADD_TABLE_ITEM (L, "mon", date.tm_mon + 1);
- ADD_TABLE_ITEM (L, "day", date.tm_mday);
- ADD_TABLE_ITEM (L, "hour", date.tm_hour);
- ADD_TABLE_ITEM (L, "min", date.tm_min);
- ADD_TABLE_ITEM (L, "sec", date.tm_sec);
- return 1;
- }
- // Module function map
- LROT_BEGIN(rtctime, NULL, 0)
- LROT_FUNCENTRY( set, rtctime_set )
- LROT_FUNCENTRY( get, rtctime_get )
- LROT_FUNCENTRY( adjust_delta, rtctime_adjust_delta )
- LROT_FUNCENTRY( dsleep, rtctime_dsleep )
- LROT_FUNCENTRY( dsleep_aligned, rtctime_dsleep_aligned )
- LROT_FUNCENTRY( epoch2cal, rtctime_epoch2cal )
- LROT_END(rtctime, NULL, 0)
- NODEMCU_MODULE(RTCTIME, "rtctime", rtctime, NULL);
|