mirror of
https://github.com/opentx/opentx.git
synced 2025-07-26 01:35:21 +03:00
1413 lines
40 KiB
C++
1413 lines
40 KiB
C++
/*
|
|
* Authors (alphabetical order)
|
|
* - Andre Bernet <bernet.andre@gmail.com>
|
|
* - Andreas Weitl
|
|
* - Bertrand Songis <bsongis@gmail.com>
|
|
* - Bryan J. Rentoul (Gruvin) <gruvin@gmail.com>
|
|
* - Cameron Weeks <th9xer@gmail.com>
|
|
* - Erez Raviv
|
|
* - Gabriel Birkus
|
|
* - Jean-Pierre Parisy
|
|
* - Karl Szmutny
|
|
* - Michael Blandford
|
|
* - Michal Hlavinka
|
|
* - Pat Mackenzie
|
|
* - Philip Moss
|
|
* - Rob Thomson
|
|
* - Romolo Manfredini <romolo.manfredini@gmail.com>
|
|
* - Thomas Husterer
|
|
*
|
|
* opentx is based on code named
|
|
* gruvin9x by Bryan J. Rentoul: http://code.google.com/p/gruvin9x/,
|
|
* er9x by Erez Raviv: http://code.google.com/p/er9x/,
|
|
* and the original (and ongoing) project by
|
|
* Thomas Husterer, th9x: http://code.google.com/p/th9x/
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
*/
|
|
|
|
#include "opentx.h"
|
|
#include "stamp-opentx.h"
|
|
|
|
#if !defined(SIMU)
|
|
extern "C" {
|
|
#endif
|
|
#include <lua.h>
|
|
#include <lauxlib.h>
|
|
#include <lualib.h>
|
|
#if !defined(SIMU)
|
|
}
|
|
#endif
|
|
|
|
#define lua_registernumber(L, n, i) (lua_pushnumber(L, (i)), lua_setglobal(L, (n)))
|
|
#define lua_registerint(L, n, i) (lua_pushinteger(L, (i)), lua_setglobal(L, (n)))
|
|
#define lua_pushtablenil(L, k) (lua_pushstring(L, (k)), lua_pushnil(L), lua_settable(L, -3))
|
|
#define lua_pushtableboolean(L, k, v) (lua_pushstring(L, (k)), lua_pushboolean(L, (v)), lua_settable(L, -3))
|
|
#define lua_pushtablenumber(L, k, v) (lua_pushstring(L, (k)), lua_pushinteger(L, (v)), lua_settable(L, -3))
|
|
#define lua_pushtablestring(L, k, v) (lua_pushstring(L, (k)), lua_pushstring(L, (v)), lua_settable(L, -3))
|
|
#define lua_pushtablezstring(L, k, v) { char _zz[sizeof(v)+1]; zchar2str(_zz, (v), sizeof(v)); lua_pushstring(L, (k)); lua_pushstring(L, _zz); lua_settable(L, -3); }
|
|
#define lua_registerlib(L, name, tab) (luaL_newmetatable(L, name), luaL_setfuncs(L, tab, 0), lua_setglobal(L, name))
|
|
|
|
lua_State *L = NULL;
|
|
uint8_t luaState = 0;
|
|
ScriptInternalData scriptInternalData[MAX_SCRIPTS] = { { SCRIPT_NOFILE, 0 } };
|
|
ScriptInternalData standaloneScript = { SCRIPT_NOFILE, 0 };
|
|
uint16_t maxLuaInterval = 0;
|
|
uint16_t maxLuaDuration = 0;
|
|
|
|
#define PERMANENT_SCRIPTS_MAX_INSTRUCTIONS (1000/100)
|
|
#define MANUAL_SCRIPTS_MAX_INSTRUCTIONS (10000/100)
|
|
#define SCRIPTS_MAX_HEAP 50
|
|
#define SET_LUA_INSTRUCTIONS_COUNT(x) (instructionsPercent=0, lua_sethook(L, hook, LUA_MASKCOUNT, x))
|
|
|
|
static int instructionsPercent = 0;
|
|
void hook(lua_State* L, lua_Debug *ar)
|
|
{
|
|
instructionsPercent++;
|
|
if (instructionsPercent > 100) {
|
|
// From now on, as soon as a line is executed, error
|
|
// keep erroring until you're script reaches the top
|
|
lua_sethook(L, hook, LUA_MASKLINE, 0);
|
|
luaL_error(L, "");
|
|
}
|
|
}
|
|
|
|
static int luaGetVersion(lua_State *L)
|
|
{
|
|
lua_pushstring(L, VERS_STR);
|
|
return 1;
|
|
}
|
|
|
|
static int luaGetTime(lua_State *L)
|
|
{
|
|
lua_pushunsigned(L, get_tmr10ms());
|
|
return 1;
|
|
}
|
|
|
|
static void luaGetValueAndPush(int src)
|
|
{
|
|
/*
|
|
Hint about dividers are taken from putsTelemetryChannel()
|
|
*/
|
|
if (!TELEMETRY_STREAMING() && src>=MIXSRC_FIRST_TELEM && src<=MIXSRC_LAST_TELEM) {
|
|
//telemetry not working, return zero for telemetry sources
|
|
lua_pushinteger(L, (int)0);
|
|
return;
|
|
}
|
|
switch (src) {
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_TX_VOLTAGE:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_VFAS:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MIN_VFAS:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_CELLS_SUM:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MIN_CELLS_SUM:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_CURRENT:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MAX_CURRENT:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_VSPEED:
|
|
//theese need to be divided by 10
|
|
lua_pushnumber(L, getValue(src)/10.0);
|
|
break;
|
|
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_A1:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_A2:
|
|
//convert raw A1/2 values to calibrated values
|
|
lua_pushnumber(L, applyChannelRatio(src-(MIXSRC_FIRST_TELEM-1+TELEM_A1), getValue(src))/100.0);
|
|
break;
|
|
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MIN_A1:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MIN_A2:
|
|
//convert raw A1/2 values to calibrated values
|
|
lua_pushnumber(L, applyChannelRatio(src-(MIXSRC_FIRST_TELEM-1+TELEM_MIN_A1), getValue(src))/100.0);
|
|
break;
|
|
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_CELL:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_MIN_CELL:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_ALT:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_ACCx:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_ACCy:
|
|
case MIXSRC_FIRST_TELEM-1+TELEM_ACCz:
|
|
//theese need to be divided by 100
|
|
lua_pushnumber(L, getValue(src)/100.0);
|
|
break;
|
|
|
|
//TODO: add other values that need special treatment
|
|
|
|
default:
|
|
lua_pushinteger(L, getValue(src));
|
|
}
|
|
}
|
|
|
|
static int luaGetValue(lua_State *L)
|
|
{
|
|
if (lua_isnumber(L, 1)) {
|
|
int src = luaL_checkinteger(L, 1);
|
|
luaGetValueAndPush(src);
|
|
return 1;
|
|
}
|
|
else {
|
|
const char *what = luaL_checkstring(L, 1);
|
|
if (!strcmp(what, "altitude")) {
|
|
lua_pushnumber(L, double(frskyData.hub.baroAltitude)/100);
|
|
return 1;
|
|
}
|
|
else if (!strcmp(what, "vario")) {
|
|
lua_pushnumber(L, double(frskyData.hub.varioSpeed)/100);
|
|
return 1;
|
|
}
|
|
else if (frskyData.hub.gpsFix) {
|
|
if (!strcmp(what, "latitude")) {
|
|
lua_pushnumber(L, gpsToDouble(frskyData.hub.gpsLatitudeNS=='S', frskyData.hub.gpsLatitude_bp, frskyData.hub.gpsLatitude_ap));
|
|
return 1;
|
|
}
|
|
else if (!strcmp(what, "longitude")) {
|
|
lua_pushnumber(L, gpsToDouble(frskyData.hub.gpsLongitudeEW=='W', frskyData.hub.gpsLongitude_bp, frskyData.hub.gpsLongitude_ap));
|
|
return 1;
|
|
}
|
|
else if (!strcmp(what, "pilot latitude")) {
|
|
lua_pushnumber(L, pilotLatitude);
|
|
return 1;
|
|
}
|
|
else if (!strcmp(what, "pilot longitude")) {
|
|
lua_pushnumber(L, pilotLongitude);
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int luaPlayFile(lua_State *L)
|
|
{
|
|
const char * filename = luaL_checkstring(L, 1);
|
|
PLAY_FILE(filename, 0, 0);
|
|
return 0;
|
|
}
|
|
|
|
static int luaPlayNumber(lua_State *L)
|
|
{
|
|
int number = luaL_checkinteger(L, 1);
|
|
int unit = luaL_checkinteger(L, 2);
|
|
int att = luaL_checkinteger(L, 3);
|
|
playNumber(number, unit, att, 0);
|
|
return 0;
|
|
}
|
|
|
|
static int luaKillEvents(lua_State *L)
|
|
{
|
|
int event = luaL_checkinteger(L, 1);
|
|
killEvents(event);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdLock(lua_State *L)
|
|
{
|
|
lcd_locked = true;
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdClear(lua_State *L)
|
|
{
|
|
lcd_clear();
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawPoint(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
lcd_plot(x, y);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawLine(lua_State *L)
|
|
{
|
|
int x1 = luaL_checkinteger(L, 1);
|
|
int y1 = luaL_checkinteger(L, 2);
|
|
int x2 = luaL_checkinteger(L, 3);
|
|
int y2 = luaL_checkinteger(L, 4);
|
|
int pat = luaL_checkinteger(L, 5);
|
|
int flags = luaL_checkinteger(L, 6);
|
|
lcd_line(x1, y1, x2, y2, pat, flags);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawText(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
const char * s = luaL_checkstring(L, 3);
|
|
int att = luaL_checkinteger(L, 4);
|
|
lcd_putsAtt(x, y, s, att);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawNumber(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int n = luaL_checkinteger(L, 3);
|
|
int att = luaL_checkinteger(L, 4);
|
|
lcd_outdezAtt(x, y, n, att);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawSwitch(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int s = luaL_checkinteger(L, 3);
|
|
int att = luaL_checkinteger(L, 4);
|
|
putsSwitches(x, y, s, att);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawSource(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int s = luaL_checkinteger(L, 3);
|
|
int att = luaL_checkinteger(L, 4);
|
|
putsMixerSource(x, y, s, att);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawPixmap(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
const char * filename = luaL_checkstring(L, 3);
|
|
uint8_t bitmap[2+LCD_W/2*LCD_H/2]; // width max is LCD_W/2 pixels for saving stack and avoid a malloc here
|
|
const pm_char * error = bmpLoad(bitmap, filename, LCD_W/2, LCD_H);
|
|
if (!error) {
|
|
lcd_bmp(x, y, bitmap);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawRectangle(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int w = luaL_checkinteger(L, 3);
|
|
int h = luaL_checkinteger(L, 4);
|
|
lcd_rect(x, y, w, h, 0xff, 0);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawFilledRectangle(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int w = luaL_checkinteger(L, 3);
|
|
int h = luaL_checkinteger(L, 4);
|
|
int flags = luaL_checkinteger(L, 5);
|
|
lcd_filled_rect(x, y, w, h, SOLID, flags);
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawScreenTitle(lua_State *L)
|
|
{
|
|
const char * str = luaL_checkstring(L, 1);
|
|
int idx = luaL_checkinteger(L, 2);
|
|
int cnt = luaL_checkinteger(L, 3);
|
|
|
|
if (cnt) displayScreenIndex(idx-1, cnt, 0);
|
|
lcd_filled_rect(0, 0, LCD_W, FH, SOLID, FILL_WHITE|GREY_DEFAULT);
|
|
title(str);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaLcdDrawCombobox(lua_State *L)
|
|
{
|
|
int x = luaL_checkinteger(L, 1);
|
|
int y = luaL_checkinteger(L, 2);
|
|
int w = luaL_checkinteger(L, 3);
|
|
luaL_checktype(L, 4, LUA_TTABLE);
|
|
int count = luaL_len(L, 4); /* get size of table */
|
|
int idx = luaL_checkinteger(L, 5);
|
|
int flags = luaL_checkinteger(L, 6);
|
|
if (idx >= count) {
|
|
// TODO error
|
|
}
|
|
if (flags & BLINK) {
|
|
lcd_filled_rect(x, y, w-9, count*9+2, SOLID, ERASE);
|
|
lcd_rect(x, y, w-9, count*9+2);
|
|
for (int i=0; i<count; i++) {
|
|
lua_rawgeti(L, 4, i+1);
|
|
const char * item = luaL_checkstring(L, -1);
|
|
lcd_putsAtt(x+2, y+2+9*i, item, 0);
|
|
}
|
|
lcd_filled_rect(x+1, y+1+9*idx, w-11, 9);
|
|
lcd_filled_rect(x+w-10, y, 10, 11, SOLID, ERASE);
|
|
lcd_rect(x+w-10, y, 10, 11);
|
|
}
|
|
else if (flags & INVERS) {
|
|
lcd_filled_rect(x, y, w, 11);
|
|
lcd_filled_rect(x+w-9, y+1, 8, 9, SOLID, ERASE);
|
|
lua_rawgeti(L, 4, idx+1);
|
|
const char * item = luaL_checkstring(L, -1);
|
|
lcd_putsAtt(x+2, y+2, item, INVERS);
|
|
}
|
|
else {
|
|
lcd_filled_rect(x, y, w, 11, SOLID, ERASE);
|
|
lcd_rect(x, y, w, 11);
|
|
lcd_filled_rect(x+w-10, y+1, 9, 9, SOLID);
|
|
lua_rawgeti(L, 4, idx+1);
|
|
const char * item = luaL_checkstring(L, -1);
|
|
lcd_putsAtt(x+2, y+2, item, 0);
|
|
}
|
|
|
|
lcd_hline(x+w-8, y+3, 6);
|
|
lcd_hline(x+w-8, y+5, 6);
|
|
lcd_hline(x+w-8, y+7, 6);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelGetTimer(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < MAX_TIMERS) {
|
|
TimerData & timer = g_model.timers[idx];
|
|
lua_newtable(L);
|
|
lua_pushtablenumber(L, "mode", timer.mode);
|
|
lua_pushtablenumber(L, "start", timer.start);
|
|
lua_pushtablenumber(L, "value", timersStates[idx].val);
|
|
lua_pushtablenumber(L, "countdownBeep", timer.countdownBeep);
|
|
lua_pushtableboolean(L, "minuteBeep", timer.minuteBeep);
|
|
lua_pushtableboolean(L, "persistent", timer.persistent);
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelSetTimer(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
|
|
if (idx < MAX_TIMERS) {
|
|
TimerData & timer = g_model.timers[idx];
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "mode")) {
|
|
timer.mode = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "start")) {
|
|
timer.start = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "value")) {
|
|
timersStates[idx].val = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "countdownBeep")) {
|
|
timer.countdownBeep = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "minuteBeep")) {
|
|
timer.minuteBeep = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "persistent")) {
|
|
timer.persistent = luaL_checkinteger(L, -1);
|
|
}
|
|
}
|
|
eeDirty(EE_MODEL);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int getFirstInput(int chn)
|
|
{
|
|
for (int i=0; i<MAX_INPUTS; i++) {
|
|
ExpoData * expo = expoAddress(i);
|
|
if (!expo->srcRaw || expo->chn>chn) break;
|
|
if (expo->chn == chn) {
|
|
return i;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
static int getInputsCountFromFirst(int chn, int first)
|
|
{
|
|
int count = 0;
|
|
if (first >= 0) {
|
|
for (int i=first; i<MAX_INPUTS; i++) {
|
|
ExpoData * expo = expoAddress(i);
|
|
if (!expo->srcRaw || expo->chn!=chn) break;
|
|
count++;
|
|
}
|
|
}
|
|
return count;
|
|
}
|
|
|
|
static int getInputsCount(int chn)
|
|
{
|
|
int first = getFirstInput(chn);
|
|
return getInputsCountFromFirst(chn, first);
|
|
}
|
|
|
|
static int luaModelGetInputsCount(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int count = getInputsCount(chn);
|
|
lua_pushinteger(L, count);
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelGetInput(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
int first = getFirstInput(chn);
|
|
int count = getInputsCountFromFirst(chn, first);
|
|
if (first>=0 && idx<count) {
|
|
ExpoData * expo = expoAddress(first+idx);
|
|
lua_newtable(L);
|
|
lua_pushtablezstring(L, "name", expo->name);
|
|
lua_pushtablenumber(L, "source", expo->srcRaw);
|
|
lua_pushtablenumber(L, "weight", expo->weight);
|
|
lua_pushtablenumber(L, "offset", expo->offset);
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelInsertInput(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
|
|
int first = getFirstInput(chn);
|
|
int count = getInputsCountFromFirst(chn, first);
|
|
|
|
if (chn<MAX_INPUTS && getExpoMixCount(1)<MAX_INPUTS && idx<=count) {
|
|
idx = first+idx;
|
|
s_currCh = chn+1;
|
|
insertExpoMix(1, idx);
|
|
ExpoData * expo = expoAddress(idx);
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "name")) {
|
|
const char * name = luaL_checkstring(L, -1);
|
|
str2zchar(expo->name, name, sizeof(expo->name));
|
|
}
|
|
else if (!strcmp(key, "source")) {
|
|
expo->srcRaw = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "weight")) {
|
|
expo->weight = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "offset")) {
|
|
expo->offset = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "switch")) {
|
|
expo->swtch = luaL_checkinteger(L, -1);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelDeleteInput(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
|
|
int first = getFirstInput(chn);
|
|
int count = getInputsCountFromFirst(chn, first);
|
|
|
|
if (first>=0 && idx<count) {
|
|
deleteExpoMix(1, first+idx);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelDeleteInputs(lua_State *L)
|
|
{
|
|
clearInputs();
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelDefaultInputs(lua_State *L)
|
|
{
|
|
defaultInputs();
|
|
return 0;
|
|
}
|
|
|
|
static int getFirstMix(int chn)
|
|
{
|
|
for (int i=0; i<MAX_MIXERS; i++) {
|
|
MixData * mix = mixAddress(i);
|
|
if (!mix->srcRaw || mix->destCh>=chn) {
|
|
return i;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int getMixesCountFromFirst(int chn, int first)
|
|
{
|
|
int count = 0;
|
|
if (first >= 0) {
|
|
for (int i=first; i<MAX_MIXERS; i++) {
|
|
MixData * mix = mixAddress(i);
|
|
if (!mix->srcRaw || mix->destCh!=chn) break;
|
|
count++;
|
|
}
|
|
}
|
|
return count;
|
|
}
|
|
|
|
static int getMixesCount(int chn)
|
|
{
|
|
int first = getFirstMix(chn);
|
|
return getMixesCountFromFirst(chn, first);
|
|
}
|
|
|
|
static int luaModelGetMixesCount(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int count = getMixesCount(chn);
|
|
lua_pushinteger(L, count);
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelGetMix(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
int first = getFirstMix(chn);
|
|
int count = getMixesCountFromFirst(chn, first);
|
|
if (idx<count) {
|
|
MixData * mix = mixAddress(first+idx);
|
|
lua_newtable(L);
|
|
lua_pushtablezstring(L, "name", mix->name);
|
|
lua_pushtablenumber(L, "source", mix->srcRaw);
|
|
lua_pushtablenumber(L, "weight", mix->weight);
|
|
lua_pushtablenumber(L, "offset", mix->offset);
|
|
lua_pushtablenumber(L, "switch", mix->swtch);
|
|
lua_pushtablenumber(L, "multiplex", mix->mltpx);
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelInsertMix(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
|
|
int first = getFirstMix(chn);
|
|
int count = getMixesCountFromFirst(chn, first);
|
|
|
|
if (chn<NUM_CHNOUT && getExpoMixCount(0)<MAX_MIXERS && idx<=count) {
|
|
idx += first;
|
|
s_currCh = chn+1;
|
|
insertExpoMix(0, idx);
|
|
MixData *mix = mixAddress(idx);
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "name")) {
|
|
const char * name = luaL_checkstring(L, -1);
|
|
str2zchar(mix->name, name, sizeof(mix->name));
|
|
}
|
|
else if (!strcmp(key, "source")) {
|
|
mix->srcRaw = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "weight")) {
|
|
mix->weight = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "offset")) {
|
|
mix->offset = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "switch")) {
|
|
mix->swtch = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "multiplex")) {
|
|
mix->mltpx = luaL_checkinteger(L, -1);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelDeleteMix(lua_State *L)
|
|
{
|
|
int chn = luaL_checkunsigned(L, 1);
|
|
int idx = luaL_checkunsigned(L, 2);
|
|
|
|
int first = getFirstMix(chn);
|
|
int count = getMixesCountFromFirst(chn, first);
|
|
|
|
if (idx<count) {
|
|
deleteExpoMix(0, first+idx);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelDeleteMixes(lua_State *L)
|
|
{
|
|
memset(g_model.mixData, 0, sizeof(g_model.mixData));
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelGetLogicalSwitch(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_LOGICAL_SWITCH) {
|
|
LogicalSwitchData * sw = lswAddress(idx);
|
|
lua_newtable(L);
|
|
lua_pushtablenumber(L, "function", sw->func);
|
|
lua_pushtablenumber(L, "v1", sw->v1);
|
|
lua_pushtablenumber(L, "v2", sw->v2);
|
|
lua_pushtablenumber(L, "v3", sw->v3);
|
|
lua_pushtablenumber(L, "and", sw->andsw);
|
|
lua_pushtablenumber(L, "delay", sw->delay);
|
|
lua_pushtablenumber(L, "duration", sw->duration);
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelSetLogicalSwitch(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_LOGICAL_SWITCH) {
|
|
LogicalSwitchData * sw = lswAddress(idx);
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "function")) {
|
|
sw->func = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "v1")) {
|
|
sw->v1 = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "v2")) {
|
|
sw->v2 = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "v3")) {
|
|
sw->v3 = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "and")) {
|
|
sw->andsw = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "delay")) {
|
|
sw->delay = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "duration")) {
|
|
sw->duration = luaL_checkinteger(L, -1);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelGetCustomFunction(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_CFN) {
|
|
CustomFnData * cfn = &g_model.funcSw[idx];
|
|
lua_newtable(L);
|
|
lua_pushtablenumber(L, "switch", CFN_SWITCH(cfn));
|
|
lua_pushtablenumber(L, "function", CFN_FUNC(cfn));
|
|
lua_pushtablezstring(L, "name", cfn->play.name);
|
|
lua_pushtablenumber(L, "value", cfn->all.val);
|
|
lua_pushtablenumber(L, "mode", cfn->all.mode);
|
|
lua_pushtablenumber(L, "active", CFN_ACTIVE(cfn));
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelSetCustomFunction(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_CFN) {
|
|
CustomFnData * cfn = &g_model.funcSw[idx];
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "switch")) {
|
|
CFN_SWITCH(cfn) = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "function")) {
|
|
CFN_FUNC(cfn) = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "name")) {
|
|
const char * name = luaL_checkstring(L, -1);
|
|
str2zchar(cfn->play.name, name, sizeof(cfn->play.name));
|
|
}
|
|
else if (!strcmp(key, "value")) {
|
|
cfn->all.val = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "mode")) {
|
|
cfn->all.mode = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "active")) {
|
|
CFN_ACTIVE(cfn) = luaL_checkinteger(L, -1);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelGetOutput(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_CHNOUT) {
|
|
LimitData * limit = limitAddress(idx);
|
|
lua_newtable(L);
|
|
lua_pushtablezstring(L, "name", limit->name);
|
|
lua_pushtablenumber(L, "min", limit->min-1000);
|
|
lua_pushtablenumber(L, "max", limit->max+1000);
|
|
lua_pushtablenumber(L, "offset", limit->offset);
|
|
lua_pushtablenumber(L, "ppmCenter", limit->ppmCenter);
|
|
lua_pushtablenumber(L, "symetrical", limit->symetrical);
|
|
lua_pushtablenumber(L, "revert", limit->revert);
|
|
if (limit->curve)
|
|
lua_pushtablenumber(L, "curve", limit->curve-1);
|
|
else
|
|
lua_pushtablenil(L, "curve");
|
|
}
|
|
else {
|
|
lua_pushnil(L);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelSetOutput(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
if (idx < NUM_CHNOUT) {
|
|
LimitData * limit = limitAddress(idx);
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TSTRING); // key is string
|
|
const char * key = luaL_checkstring(L, -2);
|
|
if (!strcmp(key, "name")) {
|
|
const char * name = luaL_checkstring(L, -1);
|
|
str2zchar(limit->name, name, sizeof(limit->name));
|
|
}
|
|
else if (!strcmp(key, "min")) {
|
|
limit->min = luaL_checkinteger(L, -1)+1000;
|
|
}
|
|
else if (!strcmp(key, "max")) {
|
|
limit->max = luaL_checkinteger(L, -1)-1000;
|
|
}
|
|
else if (!strcmp(key, "offset")) {
|
|
limit->offset = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "ppmCenter")) {
|
|
limit->ppmCenter = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "symetrical")) {
|
|
limit->symetrical = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "revert")) {
|
|
limit->revert = luaL_checkinteger(L, -1);
|
|
}
|
|
else if (!strcmp(key, "curve")) {
|
|
if (lua_isnil(L, -1))
|
|
limit->curve = 0;
|
|
else
|
|
limit->curve = luaL_checkinteger(L, -1) + 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int luaModelGetGlobalVariable(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
int phase = luaL_checkunsigned(L, 2);
|
|
if (phase < MAX_FLIGHT_MODES && idx < MAX_GVARS)
|
|
lua_pushinteger(L, g_model.phaseData[phase].gvars[idx]);
|
|
else
|
|
lua_pushnil(L);
|
|
return 1;
|
|
}
|
|
|
|
static int luaModelSetGlobalVariable(lua_State *L)
|
|
{
|
|
int idx = luaL_checkunsigned(L, 1);
|
|
int phase = luaL_checkunsigned(L, 2);
|
|
int value = luaL_checkinteger(L, 3);
|
|
if (phase < MAX_FLIGHT_MODES && idx < MAX_GVARS && value >= -GVAR_LIMIT && value <= GVAR_LIMIT) {
|
|
g_model.phaseData[phase].gvars[idx] = value;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int luaPopupInput(lua_State *L)
|
|
{
|
|
uint8_t event = luaL_checkinteger(L, 2);
|
|
s_warning_input_value = luaL_checkinteger(L, 3);
|
|
s_warning_input_min = luaL_checkinteger(L, 4);
|
|
s_warning_input_max = luaL_checkinteger(L, 5);
|
|
s_warning = luaL_checkstring(L, 1);
|
|
s_warning_type = WARNING_TYPE_INPUT;
|
|
displayWarning(event);
|
|
if (s_warning_result) {
|
|
s_warning_result = 0;
|
|
lua_pushstring(L, "OK");
|
|
}
|
|
else if (!s_warning) {
|
|
lua_pushstring(L, "CANCEL");
|
|
}
|
|
else {
|
|
lua_pushinteger(L, s_warning_input_value);
|
|
}
|
|
s_warning = NULL;
|
|
return 1;
|
|
}
|
|
|
|
static int luaDefaultStick(lua_State *L)
|
|
{
|
|
uint8_t channel = luaL_checkinteger(L, 1);
|
|
lua_pushinteger(L, channel_order(channel+1)-1);
|
|
return 1;
|
|
}
|
|
|
|
static int luaDefaultChannel(lua_State *L)
|
|
{
|
|
uint8_t stick = luaL_checkinteger(L, 1);
|
|
for (int i=1; i<=4; i++) {
|
|
int tmp = channel_order(i) - 1;
|
|
if (tmp == stick) {
|
|
lua_pushinteger(L, i-1);
|
|
return 1;
|
|
}
|
|
}
|
|
lua_pushnil(L);
|
|
return 1;
|
|
}
|
|
|
|
int luaGetInputs(ScriptInternalData & sid)
|
|
{
|
|
if (!lua_istable(L, -1))
|
|
return -1;
|
|
|
|
sid.inputsCount = 0;
|
|
for (lua_pushnil(L); lua_next(L, -2) && sid.inputsCount<MAX_SCRIPT_INPUTS; lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TTABLE); // value is table
|
|
uint8_t field = 0;
|
|
for (lua_pushnil(L); lua_next(L, -2) && field<5; lua_pop(L, 1), field++) {
|
|
switch (field) {
|
|
case 0:
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TSTRING); // value is string
|
|
sid.inputs[sid.inputsCount].name = lua_tostring(L, -1);
|
|
break;
|
|
case 1:
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TNUMBER); // value is number
|
|
sid.inputs[sid.inputsCount].type = lua_tointeger(L, -1);
|
|
if (sid.inputs[sid.inputsCount].type == 0) {
|
|
sid.inputs[sid.inputsCount].min = -100;
|
|
sid.inputs[sid.inputsCount].max = 100;
|
|
}
|
|
else {
|
|
sid.inputs[sid.inputsCount].max = MIXSRC_LAST_TELEM;
|
|
}
|
|
break;
|
|
case 2:
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TNUMBER); // value is number
|
|
sid.inputs[sid.inputsCount].min = lua_tointeger(L, -1);
|
|
break;
|
|
case 3:
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TNUMBER); // value is number
|
|
sid.inputs[sid.inputsCount].max = lua_tointeger(L, -1);
|
|
break;
|
|
case 4:
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TNUMBER); // value is number
|
|
sid.inputs[sid.inputsCount].def = lua_tointeger(L, -1);
|
|
break;
|
|
}
|
|
}
|
|
|
|
sid.inputsCount++;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int luaGetOutputs(ScriptInternalData & sid)
|
|
{
|
|
if (!lua_istable(L, -1))
|
|
return -1;
|
|
|
|
sid.outputsCount = 0;
|
|
for (lua_pushnil(L); lua_next(L, -2) && sid.outputsCount<MAX_SCRIPT_OUTPUTS; lua_pop(L, 1)) {
|
|
luaL_checktype(L, -2, LUA_TNUMBER); // key is number
|
|
luaL_checktype(L, -1, LUA_TSTRING); // value is string
|
|
sid.outputs[sid.outputsCount++].name = lua_tostring(L, -1);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static const luaL_Reg modelLib[] = {
|
|
{ "getTimer", luaModelGetTimer },
|
|
{ "setTimer", luaModelSetTimer },
|
|
{ "getInputsCount", luaModelGetInputsCount },
|
|
{ "getInput", luaModelGetInput },
|
|
{ "insertInput", luaModelInsertInput },
|
|
{ "deleteInput", luaModelDeleteInput },
|
|
{ "deleteInputs", luaModelDeleteInputs },
|
|
{ "defaultInputs", luaModelDefaultInputs },
|
|
{ "getMixesCount", luaModelGetMixesCount },
|
|
{ "getMix", luaModelGetMix },
|
|
{ "insertMix", luaModelInsertMix },
|
|
{ "deleteMix", luaModelDeleteMix },
|
|
{ "deleteMixes", luaModelDeleteMixes },
|
|
{ "getLogicalSwitch", luaModelGetLogicalSwitch },
|
|
{ "setLogicalSwitch", luaModelSetLogicalSwitch },
|
|
{ "getCustomFunction", luaModelGetCustomFunction },
|
|
{ "setCustomFunction", luaModelSetCustomFunction },
|
|
{ "getOutput", luaModelGetOutput },
|
|
{ "setOutput", luaModelSetOutput },
|
|
{ "getGlobalVariable", luaModelGetGlobalVariable },
|
|
{ "setGlobalVariable", luaModelSetGlobalVariable },
|
|
{ NULL, NULL } /* sentinel */
|
|
};
|
|
|
|
static const luaL_Reg lcdLib[] = {
|
|
{ "lock", luaLcdLock },
|
|
{ "clear", luaLcdClear },
|
|
{ "drawPoint", luaLcdDrawPoint },
|
|
{ "drawLine", luaLcdDrawLine },
|
|
{ "drawRectangle", luaLcdDrawRectangle },
|
|
{ "drawFilledRectangle", luaLcdDrawFilledRectangle },
|
|
{ "drawText", luaLcdDrawText },
|
|
{ "drawNumber", luaLcdDrawNumber },
|
|
{ "drawSwitch", luaLcdDrawSwitch },
|
|
{ "drawSource", luaLcdDrawSource },
|
|
{ "drawPixmap", luaLcdDrawPixmap },
|
|
{ "drawScreenTitle", luaLcdDrawScreenTitle },
|
|
{ "drawCombobox", luaLcdDrawCombobox },
|
|
{ NULL, NULL } /* sentinel */
|
|
};
|
|
|
|
void luaInit()
|
|
{
|
|
if (L) {
|
|
lua_close(L);
|
|
}
|
|
|
|
L = luaL_newstate();
|
|
|
|
// Init lua
|
|
luaL_openlibs(L);
|
|
|
|
// Push OpenTX libs
|
|
lua_registerlib(L, "model", modelLib);
|
|
lua_registerlib(L, "lcd", lcdLib);
|
|
|
|
// Push OpenTX functions
|
|
lua_register(L, "getTime", luaGetTime);
|
|
lua_register(L, "getVersion", luaGetVersion);
|
|
lua_register(L, "getValue", luaGetValue);
|
|
lua_register(L, "playFile", luaPlayFile);
|
|
lua_register(L, "playNumber", luaPlayNumber);
|
|
lua_register(L, "popupInput", luaPopupInput);
|
|
lua_register(L, "defaultStick", luaDefaultStick);
|
|
lua_register(L, "defaultChannel", luaDefaultChannel);
|
|
lua_register(L, "killEvents", luaKillEvents);
|
|
|
|
// Push OpenTX constants
|
|
lua_registerint(L, "FULLSCALE", RESX);
|
|
lua_registerint(L, "DBLSIZE", DBLSIZE);
|
|
lua_registerint(L, "MIDSIZE", MIDSIZE);
|
|
lua_registerint(L, "SMLSIZE", SMLSIZE);
|
|
lua_registerint(L, "INVERS", INVERS);
|
|
lua_registerint(L, "PREC1", PREC1);
|
|
lua_registerint(L, "BLINK", BLINK);
|
|
lua_registerint(L, "VALUE", 0);
|
|
lua_registerint(L, "SOURCE", 1);
|
|
lua_registerint(L, "REPLACE", MLTPX_REP);
|
|
lua_registerint(L, "MIXSRC_FIRST_INPUT", MIXSRC_FIRST_INPUT);
|
|
lua_registerint(L, "MIXSRC_Rud", MIXSRC_Rud);
|
|
lua_registerint(L, "MIXSRC_Ele", MIXSRC_Ele);
|
|
lua_registerint(L, "MIXSRC_Thr", MIXSRC_Thr);
|
|
lua_registerint(L, "MIXSRC_Ail", MIXSRC_Ail);
|
|
lua_registerint(L, "MIXSRC_SA", MIXSRC_SA);
|
|
lua_registerint(L, "MIXSRC_SB", MIXSRC_SB);
|
|
lua_registerint(L, "MIXSRC_SC", MIXSRC_SC);
|
|
lua_registerint(L, "MIXSRC_SD", MIXSRC_SD);
|
|
lua_registerint(L, "MIXSRC_SE", MIXSRC_SE);
|
|
lua_registerint(L, "MIXSRC_SF", MIXSRC_SF);
|
|
lua_registerint(L, "MIXSRC_CH1", MIXSRC_CH1);
|
|
lua_registerint(L, "SWSRC_LAST", SWSRC_LAST_LOGICAL_SWITCH);
|
|
lua_registerint(L, "EVT_MENU_BREAK", EVT_KEY_BREAK(KEY_MENU));
|
|
lua_registerint(L, "EVT_PAGE_BREAK", EVT_KEY_BREAK(KEY_PAGE));
|
|
lua_registerint(L, "EVT_PAGE_LONG", EVT_KEY_LONG(KEY_PAGE));
|
|
lua_registerint(L, "EVT_ENTER_BREAK", EVT_KEY_BREAK(KEY_ENTER));
|
|
lua_registerint(L, "EVT_ENTER_LONG", EVT_KEY_LONG(KEY_ENTER));
|
|
lua_registerint(L, "EVT_EXIT_BREAK", EVT_KEY_BREAK(KEY_EXIT));
|
|
lua_registerint(L, "EVT_PLUS_BREAK", EVT_KEY_BREAK(KEY_PLUS));
|
|
lua_registerint(L, "EVT_MINUS_BREAK", EVT_KEY_BREAK(KEY_MINUS));
|
|
lua_registerint(L, "EVT_PLUS_FIRST", EVT_KEY_FIRST(KEY_PLUS));
|
|
lua_registerint(L, "EVT_MINUS_FIRST", EVT_KEY_FIRST(KEY_MINUS));
|
|
lua_registerint(L, "EVT_PLUS_REPT", EVT_KEY_REPT(KEY_PLUS));
|
|
lua_registerint(L, "EVT_MINUS_REPT", EVT_KEY_REPT(KEY_MINUS));
|
|
lua_registerint(L, "FILL_WHITE", FILL_WHITE);
|
|
lua_registerint(L, "GREY_DEFAULT", GREY_DEFAULT);
|
|
lua_registerint(L, "SOLID", SOLID);
|
|
lua_registerint(L, "DOTTED", DOTTED);
|
|
lua_registerint(L, "LCD_W", LCD_W);
|
|
lua_registerint(L, "LCD_H", LCD_H);
|
|
}
|
|
|
|
void luaFree(ScriptInternalData & sid)
|
|
{
|
|
if (sid.run) {
|
|
luaL_unref(L, LUA_REGISTRYINDEX, sid.run);
|
|
sid.run = 0;
|
|
lua_gc(L, LUA_GCCOLLECT, 0);
|
|
}
|
|
}
|
|
|
|
int luaLoad(const char *filename, ScriptInternalData & sid)
|
|
{
|
|
int init = 0;
|
|
|
|
uint8_t prev_mem = lua_gc(L, LUA_GCCOUNT, 0);
|
|
|
|
sid.instructions = 0;
|
|
sid.state = SCRIPT_OK;
|
|
|
|
luaFree(sid);
|
|
|
|
SET_LUA_INSTRUCTIONS_COUNT(MANUAL_SCRIPTS_MAX_INSTRUCTIONS);
|
|
|
|
if (luaL_loadfile(L, filename) == 0 &&
|
|
lua_pcall(L, 0, 1, 0) == 0 &&
|
|
lua_istable(L, -1)) {
|
|
|
|
luaL_checktype(L, -1, LUA_TTABLE);
|
|
|
|
for (lua_pushnil(L); lua_next(L, -2); lua_pop(L, 1)) {
|
|
const char *key = lua_tostring(L, -2);
|
|
if (!strcmp(key, "init")) {
|
|
init = luaL_ref(L, LUA_REGISTRYINDEX);
|
|
lua_pushnil(L);
|
|
}
|
|
else if (!strcmp(key, "run")) {
|
|
sid.run = luaL_ref(L, LUA_REGISTRYINDEX);
|
|
lua_pushnil(L);
|
|
}
|
|
else if (!strcmp(key, "input")) {
|
|
luaGetInputs(sid);
|
|
}
|
|
else if (!strcmp(key, "output")) {
|
|
luaGetOutputs(sid);
|
|
}
|
|
}
|
|
|
|
if (init) {
|
|
lua_rawgeti(L, LUA_REGISTRYINDEX, init);
|
|
luaL_unref(L, LUA_REGISTRYINDEX, init);
|
|
if (lua_pcall(L, 0, 0, 0) != 0) {
|
|
TRACE("Error in script %s init: %s", filename, lua_tostring(L, -1));
|
|
sid.state = SCRIPT_SYNTAX_ERROR;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
TRACE("Error in script %s: %s", filename, lua_tostring(L, -1));
|
|
sid.state = SCRIPT_SYNTAX_ERROR;
|
|
}
|
|
|
|
sid.memory = lua_gc(L, LUA_GCCOUNT, 0) - prev_mem;
|
|
|
|
if (sid.state != SCRIPT_OK) {
|
|
luaFree(sid);
|
|
}
|
|
|
|
return sid.state;
|
|
}
|
|
|
|
void luaLoadModelScript(uint8_t index)
|
|
{
|
|
ScriptData & sd = g_model.scriptsData[index];
|
|
ScriptInternalData & sid = scriptInternalData[index];
|
|
|
|
memset(&sid, 0, sizeof(sid));
|
|
sid.state = SCRIPT_NOFILE;
|
|
|
|
if (ZEXIST(sd.file)) {
|
|
char filename[sizeof(SCRIPTS_PATH)+sizeof(sd.file)+sizeof(SCRIPTS_EXT)] = SCRIPTS_PATH "/";
|
|
strncpy(filename+sizeof(SCRIPTS_PATH), sd.file, sizeof(sd.file));
|
|
filename[sizeof(SCRIPTS_PATH)+sizeof(sd.file)] = '\0';
|
|
strcat(filename+sizeof(SCRIPTS_PATH), SCRIPTS_EXT);
|
|
luaLoad(filename, sid);
|
|
}
|
|
}
|
|
|
|
void luaLoadModelScripts()
|
|
{
|
|
// Load model scripts
|
|
for (int i=0; i<MAX_SCRIPTS; i++) {
|
|
luaLoadModelScript(i);
|
|
}
|
|
}
|
|
|
|
char lua_warning_str[WARNING_LINE_LEN+1];
|
|
char lua_warning_info[WARNING_LINE_LEN+1];
|
|
|
|
void luaError(uint8_t error)
|
|
{
|
|
const char *msg = lua_tostring(L, -1);
|
|
if (msg) {
|
|
#if defined(SIMU)
|
|
if (!strncmp(msg, "./", 2)) msg += 2;
|
|
#else
|
|
if (!strncmp(msg, "/SCRIPTS/", 9)) msg += 9;
|
|
#endif
|
|
strncpy(lua_warning_str, msg, WARNING_LINE_LEN);
|
|
lua_warning_str[WARNING_LINE_LEN] = '\0';
|
|
POPUP_WARNING(lua_warning_str);
|
|
for (int i=0; i<WARNING_LINE_LEN; i++) {
|
|
if (msg[i] == ':' && msg[i+1] == ' ') {
|
|
lua_warning_str[i] = '\0';
|
|
strncpy(lua_warning_info, &msg[i+2], WARNING_LINE_LEN);
|
|
lua_warning_info[WARNING_LINE_LEN] = '\0';
|
|
SET_WARNING_INFO(lua_warning_info, WARNING_LINE_LEN, 0);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
switch (error) {
|
|
case SCRIPT_SYNTAX_ERROR:
|
|
msg = "Script syntax error";
|
|
break;
|
|
case SCRIPT_KILLED:
|
|
msg = "Script killed";
|
|
break;
|
|
case SCRIPT_LEAK:
|
|
msg = "Script memory leak";
|
|
break;
|
|
}
|
|
POPUP_WARNING(msg);
|
|
}
|
|
}
|
|
|
|
void luaExec(const char *filename)
|
|
{
|
|
LUA_RESET();
|
|
standaloneScript.state = SCRIPT_NOFILE;
|
|
int result = luaLoad(filename, standaloneScript);
|
|
// TODO the same with run ...
|
|
if (result == SCRIPT_OK) {
|
|
luaState = LUASTATE_STANDALONE_SCRIPT_RUNNING;
|
|
}
|
|
else {
|
|
luaError(result);
|
|
}
|
|
}
|
|
|
|
void luaTask(uint8_t evt)
|
|
{
|
|
lcd_locked = false;
|
|
static uint8_t luaDisplayStatistics = false;
|
|
|
|
uint32_t t0 = get_tmr10ms();
|
|
static uint32_t lastLuaTime = 0;
|
|
uint16_t interval = (lastLuaTime == 0 ? 0 : (t0 - lastLuaTime));
|
|
lastLuaTime = t0;
|
|
if (interval > maxLuaInterval) {
|
|
maxLuaInterval = interval;
|
|
}
|
|
|
|
if (luaState & LUASTATE_STANDALONE_SCRIPT_RUNNING) {
|
|
// standalone script
|
|
if (standaloneScript.state == SCRIPT_OK && standaloneScript.run) {
|
|
SET_LUA_INSTRUCTIONS_COUNT(MANUAL_SCRIPTS_MAX_INSTRUCTIONS);
|
|
lua_rawgeti(L, LUA_REGISTRYINDEX, standaloneScript.run);
|
|
lua_pushinteger(L, evt);
|
|
if (lua_pcall(L, 1, 1, 0) == 0) {
|
|
if (!lua_isnumber(L, -1)) {
|
|
if (instructionsPercent > 100) {
|
|
TRACE("Script killed");
|
|
standaloneScript.state = SCRIPT_KILLED;
|
|
luaState = LUASTATE_RELOAD_MODEL_SCRIPTS;
|
|
}
|
|
else if (lua_isstring(L, -1)) {
|
|
char nextScript[_MAX_LFN+1];
|
|
strncpy(nextScript, lua_tostring(L, -1), _MAX_LFN);
|
|
nextScript[_MAX_LFN] = '\0';
|
|
luaExec(nextScript);
|
|
}
|
|
else {
|
|
TRACE("Script error");
|
|
standaloneScript.state = SCRIPT_SYNTAX_ERROR;
|
|
luaState = LUASTATE_RELOAD_MODEL_SCRIPTS;
|
|
}
|
|
}
|
|
else {
|
|
int scriptResult = lua_tointeger(L, -1);
|
|
lua_pop(L, 1); /* pop returned value */
|
|
if (scriptResult != 0) {
|
|
TRACE("Script finished with status %d", scriptResult);
|
|
standaloneScript.state = SCRIPT_NOFILE;
|
|
luaState = LUASTATE_RELOAD_MODEL_SCRIPTS;
|
|
}
|
|
else if (luaDisplayStatistics) {
|
|
int gc = 1000*lua_gc(L, LUA_GCCOUNT, 0) + lua_gc(L, LUA_GCCOUNTB, 0);
|
|
lcd_hline(0, 7*FH-1, lcdLastPos+FW, ERASE);
|
|
lcd_puts(0, 7*FH, "GV Use: ");
|
|
lcd_outdezAtt(lcdLastPos, 7*FH, gc, LEFT);
|
|
lcd_putc(lcdLastPos, 7*FH, 'b');
|
|
lcd_hline(0, 7*FH-2, lcdLastPos+FW, FORCE);
|
|
lcd_vlineStip(lcdLastPos+FW, 7*FH-2, FH+2, SOLID, FORCE);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
TRACE("Script error: %s", lua_tostring(L, -1));
|
|
standaloneScript.state = (instructionsPercent > 100 ? SCRIPT_KILLED : SCRIPT_SYNTAX_ERROR);
|
|
luaState = LUASTATE_RELOAD_MODEL_SCRIPTS;
|
|
}
|
|
|
|
if (standaloneScript.state != SCRIPT_OK) {
|
|
luaError(standaloneScript.state);
|
|
}
|
|
|
|
if (evt == EVT_KEY_LONG(KEY_EXIT)) {
|
|
TRACE("Script force exit");
|
|
killEvents(evt);
|
|
standaloneScript.state = SCRIPT_NOFILE;
|
|
luaState = LUASTATE_RELOAD_MODEL_SCRIPTS;
|
|
}
|
|
else if (evt == EVT_KEY_LONG(KEY_MENU)) {
|
|
killEvents(evt);
|
|
luaDisplayStatistics = !luaDisplayStatistics;
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
// model scripts
|
|
if (luaState & LUASTATE_RELOAD_MODEL_SCRIPTS) {
|
|
luaState = 0;
|
|
LUA_RESET();
|
|
luaLoadModelScripts();
|
|
}
|
|
|
|
for (int i=0; i<MAX_SCRIPTS; i++) {
|
|
ScriptData & sd = g_model.scriptsData[i];
|
|
ScriptInternalData & sid = scriptInternalData[i];
|
|
if (sid.state == SCRIPT_OK) {
|
|
uint8_t prev_mem = lua_gc(L, LUA_GCCOUNT, 0);
|
|
SET_LUA_INSTRUCTIONS_COUNT(PERMANENT_SCRIPTS_MAX_INSTRUCTIONS);
|
|
lua_rawgeti(L, LUA_REGISTRYINDEX, sid.run);
|
|
for (int j=0; j<sid.inputsCount; j++) {
|
|
if (sid.inputs[j].type == 1)
|
|
luaGetValueAndPush((uint8_t)sd.inputs[j]);
|
|
else
|
|
lua_pushinteger(L, sd.inputs[j] + sid.inputs[j].def);
|
|
}
|
|
if (lua_pcall(L, sid.inputsCount, sid.outputsCount, 0) == 0) {
|
|
for (int j=sid.outputsCount-1; j>=0; j--) {
|
|
if (!lua_isnumber(L, -1)) {
|
|
sid.state = (instructionsPercent > 100 ? SCRIPT_KILLED : SCRIPT_SYNTAX_ERROR);
|
|
TRACE("Script %10s disabled", sd.file);
|
|
break;
|
|
}
|
|
sid.outputs[j].value = lua_tointeger(L, -1);
|
|
lua_pop(L, 1);
|
|
}
|
|
}
|
|
else {
|
|
if (instructionsPercent > 100) {
|
|
TRACE("Script %10s killed", sd.file);
|
|
sid.state = SCRIPT_KILLED;
|
|
}
|
|
else {
|
|
TRACE("Script %10s error: %s", sd.file, lua_tostring(L, -1));
|
|
sid.state = SCRIPT_SYNTAX_ERROR;
|
|
}
|
|
}
|
|
|
|
if (sid.state != SCRIPT_OK) {
|
|
luaFree(sid);
|
|
}
|
|
else {
|
|
sid.memory += lua_gc(L, LUA_GCCOUNT, 0) - prev_mem;
|
|
if (instructionsPercent > sid.instructions) {
|
|
sid.instructions = instructionsPercent;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (lua_gc(L, LUA_GCCOUNT, 0) > SCRIPTS_MAX_HEAP) {
|
|
uint8_t max_memory = 0;
|
|
int8_t max_idx = -1;
|
|
for (int i=0; i<MAX_SCRIPTS; i++) {
|
|
ScriptInternalData & sid = scriptInternalData[i];
|
|
if (sid.state == SCRIPT_OK && sid.memory > max_memory && sid.memory > 15) {
|
|
max_idx = i;
|
|
}
|
|
}
|
|
if (max_idx >= 0) {
|
|
ScriptInternalData & sid = scriptInternalData[max_idx];
|
|
TRACE("Script %d killed", max_idx);
|
|
sid.state = SCRIPT_LEAK;
|
|
luaFree(sid);
|
|
}
|
|
}
|
|
}
|
|
|
|
lua_gc(L, LUA_GCCOLLECT, 0);
|
|
|
|
#if defined(SIMU) || defined(DEBUG)
|
|
static int lastgc = 0;
|
|
int gc = 1000*lua_gc(L, LUA_GCCOUNT, 0) + lua_gc(L, LUA_GCCOUNTB, 0);
|
|
if (gc != lastgc) {
|
|
lastgc = gc;
|
|
TRACE("GC Use: %dbytes", gc);
|
|
}
|
|
#endif
|
|
|
|
t0 = get_tmr10ms() - t0;
|
|
if (t0 > maxLuaDuration) {
|
|
maxLuaDuration = t0;
|
|
}
|
|
}
|