l3g4200d.c 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  1. /*
  2. * Driver for STM L3G4200D three axis gyroscope
  3. *
  4. * Code based on BMP085 driver.
  5. */
  6. #include "module.h"
  7. #include "lauxlib.h"
  8. #include "platform.h"
  9. #include "c_stdlib.h"
  10. #include "c_string.h"
  11. static const uint32_t i2c_id = 0;
  12. static const uint8_t i2c_addr = 0x69;
  13. static uint8_t r8u(uint32_t id, uint8_t reg) {
  14. uint8_t ret;
  15. platform_i2c_send_start(id);
  16. platform_i2c_send_address(id, i2c_addr, PLATFORM_I2C_DIRECTION_TRANSMITTER);
  17. platform_i2c_send_byte(id, reg);
  18. platform_i2c_send_stop(id);
  19. platform_i2c_send_start(id);
  20. platform_i2c_send_address(id, i2c_addr, PLATFORM_I2C_DIRECTION_RECEIVER);
  21. ret = platform_i2c_recv_byte(id, 0);
  22. platform_i2c_send_stop(id);
  23. return ret;
  24. }
  25. static void w8u(uint32_t id, uint8_t reg, uint8_t val) {
  26. platform_i2c_send_start(i2c_id);
  27. platform_i2c_send_address(i2c_id, i2c_addr, PLATFORM_I2C_DIRECTION_TRANSMITTER);
  28. platform_i2c_send_byte(i2c_id, reg);
  29. platform_i2c_send_byte(i2c_id, val);
  30. platform_i2c_send_stop(i2c_id);
  31. }
  32. static int l3g4200d_setup(lua_State* L) {
  33. uint8_t devid;
  34. devid = r8u(i2c_id, 0xF);
  35. if (devid != 0xD3) {
  36. return luaL_error(L, "device not found");
  37. }
  38. w8u(i2c_id, 0x20, 0xF);
  39. return 0;
  40. }
  41. static int l3g4200d_init(lua_State* L) {
  42. uint32_t sda;
  43. uint32_t scl;
  44. platform_print_deprecation_note("l3g4200d.init() is replaced by l3g4200d.setup()", "in the next version");
  45. sda = luaL_checkinteger(L, 1);
  46. scl = luaL_checkinteger(L, 2);
  47. luaL_argcheck(L, sda > 0 && scl > 0, 1, "no i2c for D0");
  48. platform_i2c_setup(i2c_id, sda, scl, PLATFORM_I2C_SPEED_SLOW);
  49. return l3g4200d_setup(L);
  50. }
  51. static int l3g4200d_read(lua_State* L) {
  52. uint8_t data[6];
  53. int x,y,z;
  54. int i;
  55. platform_i2c_send_start(i2c_id);
  56. platform_i2c_send_address(i2c_id, i2c_addr, PLATFORM_I2C_DIRECTION_TRANSMITTER);
  57. platform_i2c_send_byte(i2c_id, 0xA8);
  58. platform_i2c_send_start(i2c_id);
  59. platform_i2c_send_address(i2c_id, i2c_addr, PLATFORM_I2C_DIRECTION_RECEIVER);
  60. for (i=0; i<5; i++) {
  61. data[i] = platform_i2c_recv_byte(i2c_id, 1);
  62. }
  63. data[5] = platform_i2c_recv_byte(i2c_id, 0);
  64. platform_i2c_send_stop(i2c_id);
  65. x = (int16_t) ((data[1] << 8) | data[0]);
  66. y = (int16_t) ((data[3] << 8) | data[2]);
  67. z = (int16_t) ((data[5] << 8) | data[4]);
  68. lua_pushinteger(L, x);
  69. lua_pushinteger(L, y);
  70. lua_pushinteger(L, z);
  71. return 3;
  72. }
  73. static const LUA_REG_TYPE l3g4200d_map[] = {
  74. { LSTRKEY( "read" ), LFUNCVAL( l3g4200d_read )},
  75. { LSTRKEY( "setup" ), LFUNCVAL( l3g4200d_setup )},
  76. // init() is deprecated
  77. { LSTRKEY( "init" ), LFUNCVAL( l3g4200d_init )},
  78. { LNILKEY, LNILVAL}
  79. };
  80. NODEMCU_MODULE(L3G4200D, "l3g4200d", l3g4200d_map, NULL);