i2c.c 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233
  1. // Module for interfacing with the I2C interface
  2. #include "module.h"
  3. #include "lauxlib.h"
  4. #include "lextra.h"
  5. #include "platform.h"
  6. #include "i2c_common.h"
  7. // Lua: i2c.setup( id, sda, scl, speed )
  8. static int i2c_setup( lua_State *L )
  9. {
  10. unsigned id = luaL_checkinteger( L, 1 );
  11. unsigned sda = luaL_checkinteger( L, 2 );
  12. unsigned scl = luaL_checkinteger( L, 3 );
  13. uint32_t speed = (uint32_t)luaL_checkinteger( L, 4 );
  14. luaL_argcheck( L, id < I2C_ID_MAX, 1, "invalid id" );
  15. MOD_CHECK_ID( gpio, sda );
  16. MOD_CHECK_ID( gpio, scl );
  17. if (id == I2C_ID_SW) {
  18. luaL_argcheck( L, speed <= PLATFORM_I2C_SPEED_SLOW, 4, "wrong arg range" );
  19. if (!platform_i2c_setup( id, sda, scl, (uint32_t)speed ))
  20. luaL_error( L, "setup failed" );
  21. } else {
  22. luaL_argcheck( L, speed <= PLATFORM_I2C_SPEED_FASTPLUS, 4, "wrong arg range" );
  23. li2c_hw_master_setup( L, id, sda, scl, speed );
  24. }
  25. return 0;
  26. }
  27. // Lua: i2c.start( id )
  28. static int i2c_start( lua_State *L )
  29. {
  30. unsigned id = luaL_checkinteger( L, 1 );
  31. luaL_argcheck( L, id < I2C_ID_MAX, 1, "invalid id" );
  32. if (id == I2C_ID_SW) {
  33. if (!platform_i2c_send_start( id ))
  34. luaL_error( L, "command failed");
  35. } else {
  36. li2c_hw_master_start( L, id );
  37. }
  38. return 0;
  39. }
  40. // Lua: i2c.stop( id )
  41. static int i2c_stop( lua_State *L )
  42. {
  43. unsigned id = luaL_checkinteger( L, 1 );
  44. luaL_argcheck( L, id < I2C_ID_MAX, 1, "invalid id" );
  45. if (id == I2C_ID_SW) {
  46. if (!platform_i2c_send_stop( id ))
  47. luaL_error( L, "command failed" );
  48. } else {
  49. li2c_hw_master_stop( L, id );
  50. }
  51. return 0;
  52. }
  53. // Lua: status = i2c.address( id, address, direction )
  54. static int i2c_address( lua_State *L )
  55. {
  56. int stack = 0;
  57. unsigned id = luaL_checkinteger( L, ++stack );
  58. luaL_argcheck( L, id < I2C_ID_MAX, stack, "invalid id" );
  59. int address = luaL_checkinteger( L, ++stack );
  60. luaL_argcheck( L, address >= 0 && address <= 127, stack, "wrong arg range" );
  61. int direction = luaL_checkinteger( L, ++stack );
  62. luaL_argcheck( L,
  63. direction == PLATFORM_I2C_DIRECTION_TRANSMITTER ||
  64. direction == PLATFORM_I2C_DIRECTION_RECEIVER,
  65. stack, "wrong arg range" );
  66. bool ack_check_en = luaL_optbool( L, ++stack, true );
  67. int ret;
  68. if (id == I2C_ID_SW) {
  69. ret = platform_i2c_send_address( id, (uint16_t)address, direction, ack_check_en ? 1 : 0 );
  70. } else {
  71. ret = li2c_hw_master_address( L, id, (uint16_t)address, direction, ack_check_en );
  72. }
  73. lua_pushboolean( L, ret > 0 );
  74. return 1;
  75. }
  76. // Lua: wrote = i2c.write( id, data1, [data2], ..., [datan] )
  77. // data can be either a string, a table or an 8-bit number
  78. static int i2c_write( lua_State *L )
  79. {
  80. unsigned id = luaL_checkinteger( L, 1 );
  81. luaL_argcheck( L, id < I2C_ID_MAX, 1, "invalid id" );
  82. const char *pdata;
  83. size_t datalen, i;
  84. int numdata;
  85. uint32_t wrote = 0;
  86. unsigned argn;
  87. bool ack_check_en = true;
  88. if (lua_isboolean( L, -1 )) {
  89. ack_check_en = lua_toboolean( L, -1 );
  90. lua_pop( L, 1 );
  91. }
  92. if( lua_gettop( L ) < 2 )
  93. return luaL_error( L, "wrong arg type" );
  94. for( argn = 2; argn <= lua_gettop( L ); argn ++ )
  95. {
  96. // lua_isnumber() would silently convert a string of digits to an integer
  97. // whereas here strings are handled separately.
  98. if( lua_type( L, argn ) == LUA_TNUMBER )
  99. {
  100. numdata = ( int )luaL_checkinteger( L, argn );
  101. if( numdata < 0 || numdata > 255 )
  102. return luaL_error( L, "wrong arg range" );
  103. if (id == I2C_ID_SW) {
  104. if( platform_i2c_send_byte( id, numdata, 1) != 1 && ack_check_en )
  105. break;
  106. } else {
  107. li2c_hw_master_write( L, id, numdata, ack_check_en );
  108. }
  109. wrote ++;
  110. }
  111. else if( lua_istable( L, argn ) )
  112. {
  113. datalen = lua_objlen( L, argn );
  114. for( i = 0; i < datalen; i ++ )
  115. {
  116. lua_rawgeti( L, argn, i + 1 );
  117. numdata = ( int )luaL_checkinteger( L, -1 );
  118. lua_pop( L, 1 );
  119. if( numdata < 0 || numdata > 255 )
  120. return luaL_error( L, "wrong arg range" );
  121. if (id == I2C_ID_SW) {
  122. if( platform_i2c_send_byte( id, numdata, 1 ) == 0 && ack_check_en)
  123. break;
  124. } else {
  125. li2c_hw_master_write( L, id, numdata, ack_check_en );
  126. }
  127. }
  128. wrote += i;
  129. if( i < datalen )
  130. break;
  131. }
  132. else
  133. {
  134. pdata = luaL_checklstring( L, argn, &datalen );
  135. for( i = 0; i < datalen; i ++ )
  136. if (id == I2C_ID_SW) {
  137. if( platform_i2c_send_byte( id, pdata[ i ], 1 ) == 0 && ack_check_en)
  138. break;
  139. } else {
  140. li2c_hw_master_write( L, id, pdata[ i ], ack_check_en );
  141. }
  142. wrote += i;
  143. if( i < datalen )
  144. break;
  145. }
  146. }
  147. lua_pushinteger( L, wrote );
  148. return 1;
  149. }
  150. // Lua: read = i2c.read( id, size )
  151. static int i2c_read( lua_State *L )
  152. {
  153. unsigned id = luaL_checkinteger( L, 1 );
  154. luaL_argcheck( L, id < I2C_ID_MAX, 1, "invalid id" );
  155. uint32_t size = ( uint32_t )luaL_checkinteger( L, 2 ), i;
  156. if (id == I2C_ID_SW) {
  157. luaL_Buffer b;
  158. int data;
  159. if( size == 0 )
  160. return 0;
  161. luaL_buffinit( L, &b );
  162. for( i = 0; i < size; i ++ )
  163. if( ( data = platform_i2c_recv_byte( id, i < size - 1 ) ) == -1 )
  164. break;
  165. else
  166. luaL_addchar( &b, ( char )data );
  167. luaL_pushresult( &b );
  168. return 1;
  169. } else {
  170. li2c_hw_master_read( L, id, size );
  171. return 0;
  172. }
  173. }
  174. static const LUA_REG_TYPE i2c_map[] = {
  175. { LSTRKEY( "setup" ), LFUNCVAL( i2c_setup ) },
  176. { LSTRKEY( "start" ), LFUNCVAL( i2c_start ) },
  177. { LSTRKEY( "stop" ), LFUNCVAL( i2c_stop ) },
  178. { LSTRKEY( "address" ), LFUNCVAL( i2c_address ) },
  179. { LSTRKEY( "write" ), LFUNCVAL( i2c_write ) },
  180. { LSTRKEY( "read" ), LFUNCVAL( i2c_read ) },
  181. { LSTRKEY( "transfer" ), LFUNCVAL( li2c_hw_master_transfer ) },
  182. { LSTRKEY( "slave"), LROVAL( li2c_slave_map ) },
  183. { LSTRKEY( "FASTPLUS" ), LNUMVAL( PLATFORM_I2C_SPEED_FASTPLUS ) },
  184. { LSTRKEY( "FAST" ), LNUMVAL( PLATFORM_I2C_SPEED_FAST ) },
  185. { LSTRKEY( "SLOW" ), LNUMVAL( PLATFORM_I2C_SPEED_SLOW ) },
  186. { LSTRKEY( "TRANSMITTER" ), LNUMVAL( PLATFORM_I2C_DIRECTION_TRANSMITTER ) },
  187. { LSTRKEY( "RECEIVER" ), LNUMVAL( PLATFORM_I2C_DIRECTION_RECEIVER ) },
  188. { LSTRKEY( "SW" ), LNUMVAL( I2C_ID_SW ) },
  189. { LSTRKEY( "HW0" ), LNUMVAL( I2C_ID_HW0 ) },
  190. { LSTRKEY( "HW1" ), LNUMVAL( I2C_ID_HW1 ) },
  191. {LNILKEY, LNILVAL}
  192. };
  193. int luaopen_i2c( lua_State *L ) {
  194. li2c_hw_master_init( L );
  195. li2c_hw_slave_init( L );
  196. return 0;
  197. }
  198. NODEMCU_MODULE(I2C, "i2c", i2c_map, luaopen_i2c);