From cf079fa333daa51d3d9e4ddce6b8acfb833ac72b Mon Sep 17 00:00:00 2001
From: chris <chris@>
Date: Thu, 8 Oct 2015 20:34:24 -0700
Subject: [PATCH] Basic hx711 functionality

---
 app/include/user_modules.h |  1 +
 app/modules/hx711.c        | 81 ++++++++++++++++++++++++++++++++++++++
 app/modules/modules.h      | 11 +++++-
 3 files changed, 92 insertions(+), 1 deletion(-)
 create mode 100644 app/modules/hx711.c

diff --git a/app/include/user_modules.h b/app/include/user_modules.h
index c9e06f13..eb8d21b1 100644
--- a/app/include/user_modules.h
+++ b/app/include/user_modules.h
@@ -43,6 +43,7 @@
 #define LUA_USE_MODULES_SNTP
 //#define LUA_USE_MODULES_BMP085
 #define LUA_USE_MODULES_TSL2561
+#define LUA_USE_MODULES_HX711
 
 #endif /* LUA_USE_MODULES */
 
diff --git a/app/modules/hx711.c b/app/modules/hx711.c
new file mode 100644
index 00000000..c95cfff3
--- /dev/null
+++ b/app/modules/hx711.c
@@ -0,0 +1,81 @@
+#include "lualib.h"
+#include "lauxlib.h"
+#include "platform.h"
+#include "auxmods.h"
+#include "lrotable.h"
+#include "c_stdlib.h"
+#include "c_string.h"
+#include "user_interface.h"
+static uint8_t data_pin;
+static uint8_t clk_pin;
+
+/*Lua: init(clk_pin,data_pin)*/
+static int hx711_init(lua_State* L) {
+  clk_pin = luaL_checkinteger(L,1);
+  data_pin = luaL_checkinteger(L,2);
+  MOD_CHECK_ID( gpio, clk_pin );
+  MOD_CHECK_ID( gpio, data_pin );
+
+  platform_gpio_mode(clk_pin, PLATFORM_GPIO_OUTPUT, PLATFORM_GPIO_FLOAT);
+  platform_gpio_mode(data_pin, PLATFORM_GPIO_INPUT, PLATFORM_GPIO_FLOAT);
+  platform_gpio_write(clk_pin,1);//put chip to sleep.
+  return 0;
+}
+
+#define HX711_MAX_WAIT 1000000
+/*will only read chA@128gain*/
+/*Lua: result = hx711.read()*/
+static int ICACHE_FLASH_ATTR hx711_read(lua_State* L) {
+  uint32_t i;
+  int32_t data = 0;
+  //TODO: double check init has happened first.
+
+  //wakeup hx711
+  platform_gpio_write(clk_pin,0);
+  
+  //wait for data ready.  or time out.
+  //TODO: set pin inturrupt and come back to it.  This may take up to 1/10 sec
+  //        or maybe just make an async version too and have both available.
+  for (i = 0; i<HX711_MAX_WAIT && platform_gpio_read(data_pin)==1;i++){
+    asm ("nop"); //don't optimize away this loop.
+  }
+  
+  //Handle timeout error
+  if (i>=HX711_MAX_WAIT) {
+    return luaL_error( L, "ADC timeout!", ( unsigned )0 );
+  }
+
+  for (i = 0; i<24 ; i++){  //clock in the 24 bits
+    platform_gpio_write(clk_pin,1);
+    platform_gpio_write(clk_pin,0);
+    data = data<<1;
+    if (platform_gpio_read(data_pin)==1) {
+      data = i==0 ? -1 : data|1; //signextend the first bit
+    }
+  }
+  //add 25th clock pulse to prevent protocol error (probably not needed 
+  // since we'll go to sleep immediately after and reset on wakeup.)
+  platform_gpio_write(clk_pin,1);
+  platform_gpio_write(clk_pin,0);
+  //sleep
+  platform_gpio_write(clk_pin,1);
+  lua_pushinteger( L, data );
+  return 1; 
+}
+
+#define MIN_OPT_LEVEL 2
+#include "lrodefs.h"
+const LUA_REG_TYPE hx711_map[] =
+{
+  { LSTRKEY( "init" ), LFUNCVAL( hx711_init )},
+  { LSTRKEY( "read" ), LFUNCVAL( hx711_read )},
+  //{ LSTRKEY( "write" ), LFUNCVAL( ws2812_writegrb )},
+  { LNILKEY, LNILVAL}
+};
+
+LUALIB_API int luaopen_hx711(lua_State *L) {
+  // TODO: the below todo was inherited from the ws2812 code but is still valid.
+  // TODO: Make sure that the GPIO system is initialized
+  LREGISTER(L, "hx711", hx711_map);
+  return 1;
+}
diff --git a/app/modules/modules.h b/app/modules/modules.h
index 48781cc9..7292b6e9 100644
--- a/app/modules/modules.h
+++ b/app/modules/modules.h
@@ -245,6 +245,14 @@
 #define ROM_MODULES_TSL2561
 #endif
 
+#if defined(LUA_USE_MODULES_HX711)
+#define MODULES_HX711      "hx711"
+#define ROM_MODULES_HX711  \
+    _ROM(MODULES_HX711, luaopen_hx711, hx711_map)
+#else
+#define ROM_MODULES_HX711
+#endif
+
 #define LUA_MODULES_ROM     \
         ROM_MODULES_GPIO    \
         ROM_MODULES_PWM		\
@@ -275,6 +283,7 @@
         ROM_MODULES_RTCFIFO \
         ROM_MODULES_SNTP    \
         ROM_MODULES_BMP085  \
-        ROM_MODULES_TSL2561
+        ROM_MODULES_TSL2561 \
+        ROM_MODULES_HX711  
 
 #endif