165 lines
4.4 KiB
C
165 lines
4.4 KiB
C
|
// Module for interfacing with the I2C interface
|
||
|
|
||
|
#include "module.h"
|
||
|
#include "lauxlib.h"
|
||
|
#include "platform.h"
|
||
|
|
||
|
// Lua: speed = i2c.setup( id, sda, scl, speed )
|
||
|
static int i2c_setup( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
unsigned sda = luaL_checkinteger( L, 2 );
|
||
|
unsigned scl = luaL_checkinteger( L, 3 );
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
MOD_CHECK_ID( gpio, sda );
|
||
|
MOD_CHECK_ID( gpio, scl );
|
||
|
|
||
|
if ( sda == 0 )
|
||
|
return luaL_error( L, "i2c SDA on D0 is not supported" );
|
||
|
|
||
|
s32 speed = ( s32 )luaL_checkinteger( L, 4 );
|
||
|
if ( speed <= 0 )
|
||
|
return luaL_error( L, "wrong arg range" );
|
||
|
speed = platform_i2c_setup( id, sda, scl, (u32)speed );
|
||
|
if ( speed == 0 )
|
||
|
return luaL_error( L, "failed to initialize i2c %d", id );
|
||
|
lua_pushinteger( L, speed );
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
// Lua: i2c.start( id )
|
||
|
static int i2c_start( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
if (platform_i2c_configured( id ) )
|
||
|
platform_i2c_send_start( id );
|
||
|
else
|
||
|
luaL_error( L, "i2c %d is not configured", id );
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
// Lua: i2c.stop( id )
|
||
|
static int i2c_stop( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
platform_i2c_send_stop( id );
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
// Lua: status = i2c.address( id, address, direction )
|
||
|
static int i2c_address( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
int address = luaL_checkinteger( L, 2 );
|
||
|
int direction = luaL_checkinteger( L, 3 );
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
if ( address < 0 || address > 127 )
|
||
|
return luaL_error( L, "wrong arg range" );
|
||
|
lua_pushboolean( L, platform_i2c_send_address( id, (u16)address, direction ) );
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
// Lua: wrote = i2c.write( id, data1, [data2], ..., [datan] )
|
||
|
// data can be either a string, a table or an 8-bit number
|
||
|
static int i2c_write( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
const char *pdata;
|
||
|
size_t datalen, i;
|
||
|
int numdata;
|
||
|
u32 wrote = 0;
|
||
|
unsigned argn;
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
if( lua_gettop( L ) < 2 )
|
||
|
return luaL_error( L, "wrong arg type" );
|
||
|
for( argn = 2; argn <= lua_gettop( L ); argn ++ )
|
||
|
{
|
||
|
// lua_isnumber() would silently convert a string of digits to an integer
|
||
|
// whereas here strings are handled separately.
|
||
|
if( lua_type( L, argn ) == LUA_TNUMBER )
|
||
|
{
|
||
|
numdata = ( int )luaL_checkinteger( L, argn );
|
||
|
if( numdata < 0 || numdata > 255 )
|
||
|
return luaL_error( L, "wrong arg range" );
|
||
|
if( platform_i2c_send_byte( id, numdata ) != 1 )
|
||
|
break;
|
||
|
wrote ++;
|
||
|
}
|
||
|
else if( lua_istable( L, argn ) )
|
||
|
{
|
||
|
datalen = lua_objlen( L, argn );
|
||
|
for( i = 0; i < datalen; i ++ )
|
||
|
{
|
||
|
lua_rawgeti( L, argn, i + 1 );
|
||
|
numdata = ( int )luaL_checkinteger( L, -1 );
|
||
|
lua_pop( L, 1 );
|
||
|
if( numdata < 0 || numdata > 255 )
|
||
|
return luaL_error( L, "wrong arg range" );
|
||
|
if( platform_i2c_send_byte( id, numdata ) == 0 )
|
||
|
break;
|
||
|
}
|
||
|
wrote += i;
|
||
|
if( i < datalen )
|
||
|
break;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
pdata = luaL_checklstring( L, argn, &datalen );
|
||
|
for( i = 0; i < datalen; i ++ )
|
||
|
if( platform_i2c_send_byte( id, pdata[ i ] ) == 0 )
|
||
|
break;
|
||
|
wrote += i;
|
||
|
if( i < datalen )
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
lua_pushinteger( L, wrote );
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
// Lua: read = i2c.read( id, size )
|
||
|
static int i2c_read( lua_State *L )
|
||
|
{
|
||
|
unsigned id = luaL_checkinteger( L, 1 );
|
||
|
u32 size = ( u32 )luaL_checkinteger( L, 2 ), i;
|
||
|
luaL_Buffer b;
|
||
|
int data;
|
||
|
|
||
|
MOD_CHECK_ID( i2c, id );
|
||
|
if( size == 0 )
|
||
|
return 0;
|
||
|
luaL_buffinit( L, &b );
|
||
|
for( i = 0; i < size; i ++ )
|
||
|
if( ( data = platform_i2c_recv_byte( id, i < size - 1 ) ) == -1 )
|
||
|
break;
|
||
|
else
|
||
|
luaL_addchar( &b, ( char )data );
|
||
|
luaL_pushresult( &b );
|
||
|
return 1;
|
||
|
}
|
||
|
|
||
|
// Module function map
|
||
|
LROT_BEGIN(i2c, NULL, 0)
|
||
|
LROT_FUNCENTRY( setup, i2c_setup )
|
||
|
LROT_FUNCENTRY( start, i2c_start )
|
||
|
LROT_FUNCENTRY( stop, i2c_stop )
|
||
|
LROT_FUNCENTRY( address, i2c_address )
|
||
|
LROT_FUNCENTRY( write, i2c_write )
|
||
|
LROT_FUNCENTRY( read, i2c_read )
|
||
|
LROT_NUMENTRY( FASTPLUS, PLATFORM_I2C_SPEED_FASTPLUS )
|
||
|
LROT_NUMENTRY( FAST, PLATFORM_I2C_SPEED_FAST )
|
||
|
LROT_NUMENTRY( SLOW, PLATFORM_I2C_SPEED_SLOW )
|
||
|
LROT_NUMENTRY( TRANSMITTER, PLATFORM_I2C_DIRECTION_TRANSMITTER )
|
||
|
LROT_NUMENTRY( RECEIVER, PLATFORM_I2C_DIRECTION_RECEIVER )
|
||
|
LROT_END(i2c, NULL, 0)
|
||
|
|
||
|
|
||
|
NODEMCU_MODULE(I2C, "i2c", i2c, NULL);
|