123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- // Module for RTC sample FIFO storage
- #include "module.h"
- #include "lauxlib.h"
- #include "user_modules.h"
- #include "rtc/rtctime.h"
- #define RTCTIME_SLEEP_ALIGNED rtctime_deep_sleep_until_aligned_us
- #include "rtc/rtcfifo.h"
- #include <string.h>
- // rtcfifo.prepare ([{sensor_count=n, interval_us=m, storage_begin=x, storage_end=y}])
- static int rtcfifo_prepare (lua_State *L)
- {
- uint32_t sensor_count = RTC_DEFAULT_TAGCOUNT;
- uint32_t interval_us = 0;
- int first = -1, last = -1;
- if (lua_istable (L, 1))
- {
- #ifdef LUA_USE_MODULES_RTCTIME
- lua_getfield (L, 1, "interval_us");
- if (lua_isnumber (L, -1))
- interval_us = lua_tointeger (L, -1);
- lua_pop (L, 1);
- #endif
- lua_getfield (L, 1, "sensor_count");
- if (lua_isnumber (L, -1))
- sensor_count = lua_tointeger (L, -1);
- lua_pop (L, 1);
- lua_getfield (L, 1, "storage_begin");
- if (lua_isnumber (L, -1))
- first = lua_tointeger (L, -1);
- lua_pop (L, 1);
- lua_getfield (L, 1, "storage_end");
- if (lua_isnumber (L, -1))
- last = lua_tointeger (L, -1);
- lua_pop (L, 1);
- }
- else if (!lua_isnone (L, 1))
- return luaL_error (L, "expected table as arg #1");
- rtc_fifo_prepare (0, interval_us, sensor_count);
- if (first != -1 && last != -1)
- rtc_fifo_put_loc (first, last, sensor_count);
- return 0;
- }
- // ready = rtcfifo.ready ()
- static int rtcfifo_ready (lua_State *L)
- {
- lua_pushinteger (L, rtc_fifo_check_magic ());
- return 1;
- }
- static void check_fifo_magic (lua_State *L)
- {
- if (!rtc_fifo_check_magic ())
- luaL_error (L, "rtcfifo not prepared!");
- }
- // rtcfifo.put (timestamp, value, decimals, sensor_name)
- static int rtcfifo_put (lua_State *L)
- {
- check_fifo_magic (L);
- sample_t s;
- s.timestamp = luaL_checkinteger (L, 1);
- s.value = luaL_checkinteger (L, 2);
- s.decimals = luaL_checkinteger (L, 3);
- size_t len;
- const char *str = luaL_checklstring (L, 4, &len);
- union {
- uint32_t u;
- char s[4];
- } conv = { 0 };
- strncpy (conv.s, str, len > 4 ? 4 : len);
- s.tag = conv.u;
- rtc_fifo_store_sample (&s);
- return 0;
- }
- static int extract_sample (lua_State *L, const sample_t *s)
- {
- lua_pushinteger (L, s->timestamp);
- lua_pushinteger (L, s->value);
- lua_pushinteger (L, s->decimals);
- union {
- uint32_t u;
- char s[4];
- } conv = { s->tag };
- if (conv.s[3] == 0)
- lua_pushstring (L, conv.s);
- else
- lua_pushlstring (L, conv.s, 4);
- return 4;
- }
- // timestamp, value, decimals, sensor_name = rtcfifo.pop ()
- static int rtcfifo_pop (lua_State *L)
- {
- check_fifo_magic (L);
- sample_t s;
- if (!rtc_fifo_pop_sample (&s))
- return 0;
- else
- return extract_sample (L, &s);
- }
- // timestamp, value, decimals, sensor_name = rtcfifo.peek ([offset])
- static int rtcfifo_peek (lua_State *L)
- {
- check_fifo_magic (L);
- sample_t s;
- uint32_t offs = 0;
- if (lua_isnumber (L, 1))
- offs = lua_tointeger (L, 1);
- if (!rtc_fifo_peek_sample (&s, offs))
- return 0;
- else
- return extract_sample (L, &s);
- }
- // rtcfifo.drop (num)
- static int rtcfifo_drop (lua_State *L)
- {
- check_fifo_magic (L);
- rtc_fifo_drop_samples (luaL_checkinteger (L, 1));
- return 0;
- }
- // num = rtcfifo.count ()
- static int rtcfifo_count (lua_State *L)
- {
- check_fifo_magic (L);
- lua_pushinteger (L, rtc_fifo_get_count ());
- return 1;
- }
- #ifdef LUA_USE_MODULES_RTCTIME
- // rtcfifo.dsleep_until_sample (min_sleep_us)
- static int rtcfifo_dsleep_until_sample (lua_State *L)
- {
- check_fifo_magic (L);
- uint32_t min_us = luaL_checkinteger (L, 1);
- rtc_fifo_deep_sleep_until_sample (min_us); // no return
- return 0;
- }
- #endif
- // Module function map
- LROT_BEGIN(rtcfifo, NULL, 0)
- LROT_FUNCENTRY( prepare, rtcfifo_prepare )
- LROT_FUNCENTRY( ready, rtcfifo_ready )
- LROT_FUNCENTRY( put, rtcfifo_put )
- LROT_FUNCENTRY( pop, rtcfifo_pop )
- LROT_FUNCENTRY( peek, rtcfifo_peek )
- LROT_FUNCENTRY( drop, rtcfifo_drop )
- LROT_FUNCENTRY( count, rtcfifo_count )
- #ifdef LUA_USE_MODULES_RTCTIME
- LROT_FUNCENTRY( dsleep_until_sample, rtcfifo_dsleep_until_sample )
- #endif
- LROT_END(rtcfifo, NULL, 0)
- NODEMCU_MODULE(RTCFIFO, "rtcfifo", rtcfifo, NULL);
|