LuatOS/luat/modules/luat_lib_i2s.c

/*
@module  i2s
@summary 数字音频
@version core V0007
@date    2022.05.26
@tag LUAT_USE_I2S
*/
#include "luat_base.h"
#include "luat_malloc.h"
#include "luat_i2s.h"
#include "luat_zbuff.h"
#include "c_common.h"
#define LUAT_LOG_TAG "i2s"
#include "luat_log.h"
#ifndef I2S_DEVICE_MAX_CNT
#define I2S_DEVICE_MAX_CNT 2
#endif
static int i2s_cbs[I2S_DEVICE_MAX_CNT];
static luat_zbuff_t *i2s_rx_buffer[I2S_DEVICE_MAX_CNT];


int l_i2s_handler(lua_State *L, void* ptr) {
    (void)ptr;
    //LLOGD("l_uart_handler");
    rtos_msg_t* msg = (rtos_msg_t*)lua_topointer(L, -1);
    lua_pop(L, 1);
    int i2s_id = msg->arg1;
    lua_geti(L, LUA_REGISTRYINDEX, i2s_cbs[i2s_id]);

    if (lua_isfunction(L, -1)) {
        if (msg->arg2 == 0) {
            //LLOGD("uart%ld sent callback", i2s_id);
        	lua_pushinteger(L, i2s_id);
    		lua_pushnil(L);
    		lua_call(L, 2, 0);
        }
        else {
        	lua_pushinteger(L, i2s_id);
        	lua_pushlightuserdata(L, i2s_rx_buffer[i2s_id]);
            lua_call(L, 2, 0);
        }
    }
    // 给rtos.recv方法返回个空数据
    lua_pushinteger(L, 0);
    return 1;
}

int32_t luat_i2s_rx_cb(void *pdata, void *param)
{
	Buffer_Struct *buffer = (Buffer_Struct *)pdata;
	int id = (int)param;
	if (!i2s_rx_buffer[id] || !i2s_cbs[id])
	{
		return -1;
	}
	if (buffer && (buffer->Pos))
	{
		uint32_t len = (buffer->Pos > i2s_rx_buffer[id]->len)?i2s_rx_buffer[id]->len:buffer->Pos;
		memcpy(i2s_rx_buffer[id]->addr, buffer->Data, len);
		i2s_rx_buffer[id]->used = len;
        rtos_msg_t msg;
        msg.handler = l_i2s_handler;
        msg.ptr = NULL;
        msg.arg1 = id;
        msg.arg2 = len;
        int re = luat_msgbus_put(&msg, 0);
		buffer->Pos = 0;

	}

	return 0;
}

/*
初始化i2s
@api i2s.setup(id, mode, sample, bitw, channel, format, mclk)
@int i2s通道号,与具体设备有关
@int 模式, 当前仅支持0, MASTER|TX|RX 模式, 暂不支持slave. 可选
@int 采样率,默认44100. 可选
@int 声道, 0 左声道, 1 右声道, 2 双声道. 可选
@int 格式, 可选MODE_I2S, MODE_LSB, MODE_MSB
@int mclk频率, 默认 8M. 可选
@return boolean 成功与否
@return int 底层返回值
@usage
-- 这个库处于开发阶段, 尚不可用
-- 以默认参数初始化i2s
i2s.setup(0)
-- 以详细参数初始化i2s, 示例为默认值
i2s.setup(0, 0, 44100, 16, 0, 0, 8000000)
*/
static int l_i2s_setup(lua_State *L) {
    luat_i2s_conf_t conf = {0};
    conf.id = luaL_checkinteger(L, 1);
    conf.mode = luaL_optinteger(L, 2, 0);
    conf.sample_rate = luaL_optinteger(L, 3, 44100); // 44.1k比较常见吧,待讨论
    conf.bits_per_sample = luaL_optinteger(L, 4, 16); // 通常就是16bit
    conf.channel_format = luaL_optinteger(L, 5, 0); // 1 右声道, 0 左声道
    conf.communication_format = luaL_optinteger(L, 6, 0); // 0 - I2S 标准, 当前只支持这种就行
    conf.mclk = luaL_optinteger(L, 7, 0);
    // conf.intr_alloc_flags = luaL_optinteger(L, 8, 0);
    int ret = luat_i2s_setup(&conf);
    lua_pushboolean(L, ret == 0 ? 1 : 0);
    lua_pushinteger(L, ret);
    return 2;
}

/*
发送i2s数据
@api i2s.send(id, data, len)
@int 通道id
@string 数据, 可以是字符串或zbuff
@int 数据长度,单位字节, 字符串默认为字符串全长, zbuff默认为指针位置
@return boolean 成功与否
@return int 底层返回值,供调试用
@usage
local f = io.open("/luadb/abc.wav")
while 1 do
    local data = f:read(4096)
    if not data or #data == 0 then
        break
    end
    i2s.send(0, data)
    sys.wait(100)
end
*/
static int l_i2s_send(lua_State *L) {
    char* buff;
    size_t len;
    int id = luaL_checkinteger(L, 1);
    if (lua_isuserdata(L, 2)) {
        luat_zbuff_t* zbuff = ((luat_zbuff_t *)luaL_checkudata(L, 2, LUAT_ZBUFF_TYPE));
        buff = (char*)zbuff->addr;
        len = zbuff->cursor;
    }
    else {
        buff = (char*)luaL_checklstring(L, 2, &len);
    }
    if (lua_type(L, 3) == LUA_TINTEGER) {
        len = luaL_checkinteger(L, 3);
    }
    int ret = luat_i2s_send(id, buff, len);
    lua_pushboolean(L, ret == len ? 1 : 0);
    lua_pushinteger(L, ret);
    return 2;
}

/*
接收i2s数据，注意在数据在回调时已经存放在zbuff里，目前只有air780e系列支持
@api i2s.recv(id, buffer, len)
@int 通道id
@zbuff 数据缓存区
@int 单次返回的数据长度,单位字节,必须与传入的zbuff的大小一致
@return boolean 成功与否
@usage
local buffer = zbuff.create(3200)
local succ = i2s.recv(0, buffer, 3200);
*/
static int l_i2s_recv(lua_State *L) {
    luaL_Buffer buff;
    int id = luaL_checkinteger(L, 1);
    if (id >= I2S_DEVICE_MAX_CNT)
    {
    	lua_pushboolean(L, 0);
        return 1;
    }
    if (lua_isuserdata(L, 2)) {
        luat_zbuff_t* zbuff = ((luat_zbuff_t *)luaL_checkudata(L, 2, LUAT_ZBUFF_TYPE));
        i2s_rx_buffer[id] = zbuff;
    }
    else {
    	i2s_rx_buffer[id] = NULL;
    }

    size_t len = luaL_checkinteger(L, 3);
	lua_pushboolean(L, !luat_i2s_recv(id, NULL, len));
    return 1;
}

/*
关闭i2s
@api i2s.close(id)
@int 通道id
@return nil 无返回值
@usage
i2s.close(0)
*/
static int l_i2s_close(lua_State *L) {
    int id = luaL_checkinteger(L, 1);
    luat_i2s_close(id);
    return 0;
}

/*
注册I2S事件回调
@api    i2s.on(id, func)
@int i2s id, i2s0写0, i2s1写1
@function 回调方法
@return nil 无返回值
@usage
i2s.on(0, function(id, buff)
	if buff then
		log.info("i2s get data in zbuff")
	else
		log.info("i2s tx one block done")
	end
end)
*/
static int l_i2s_on(lua_State *L) {
    int i2s_id = luaL_checkinteger(L, 1);
    if (i2s_id >= I2S_DEVICE_MAX_CNT)
    {
        lua_pushliteral(L, "no such i2s id");
        return 1;
    }
    if (i2s_cbs[i2s_id] != 0)
    {
    	luaL_unref(L, LUA_REGISTRYINDEX, i2s_cbs[i2s_id]);
    }
    if (lua_isfunction(L, 2)) {
        lua_pushvalue(L, 2);
        i2s_cbs[i2s_id] = luaL_ref(L, LUA_REGISTRYINDEX);
    }
    return 0;
}

int l_i2s_play(lua_State *L);
int l_i2s_pause(lua_State *L);
int l_i2s_stop(lua_State *L);

#ifndef LUAT_COMPILER_NOWEAK
LUAT_WEAK int l_i2s_play(lua_State *L) {
    LLOGE("not support yet");
    return 0;
}

LUAT_WEAK int l_i2s_pause(lua_State *L) {
    LLOGE("not support yet");
    return 0;
}

LUAT_WEAK int l_i2s_stop(lua_State *L) {
    LLOGE("not support yet");
    return 0;
}
#endif

#include "rotable2.h"
static const rotable_Reg_t reg_i2s[] =
{
    { "setup",      ROREG_FUNC(l_i2s_setup)},
    { "send",       ROREG_FUNC(l_i2s_send)},
    { "recv",       ROREG_FUNC(l_i2s_recv)},
    { "close",      ROREG_FUNC(l_i2s_close)},
	{ "on",       ROREG_FUNC(l_i2s_on)},
    // 以下为兼容扩展功能,待定
    { "play",       ROREG_FUNC(l_i2s_play)},
    { "pause",      ROREG_FUNC(l_i2s_pause)},
    { "stop",       ROREG_FUNC(l_i2s_stop)},
	//@const MODE_I2S number I2S标准，比如ES7149
	{ "MODE_I2S", 	ROREG_INT(0)},
	//@const MODE_LSB number LSB格式
	{ "MODE_LSB", 	ROREG_INT(1)},
	//@const MODE_MSB number MSB格式，比如TM8211
	{ "MODE_MSB", 	ROREG_INT(2)},
	{ NULL,         ROREG_INT(0) }
};

LUAMOD_API int luaopen_i2s(lua_State *L)
{
    luat_newlib2(L, reg_i2s);
    return 1;
}