Implement analogRead() and hallRead()

pandax381 · pandax381 · commit ed48ba42ec6a · 2017-05-18T15:51:20.000+09:00
diff --git a/src/gpio.c b/src/gpio.c
@@ -2,6 +2,7 @@
 #include <mruby/value.h>
 
 #include "driver/gpio.h"
+#include "driver/adc.h"
 
 #define GPIO_MODE_DEF_PULLUP (BIT3)
 #define GPIO_MODE_INPUT_PULLUP ((GPIO_MODE_INPUT)|(GPIO_MODE_DEF_PULLUP))
@@ -54,6 +55,26 @@ mrb_esp32_gpio_digital_read(mrb_state *mrb, mrb_value self) {
   return mrb_fixnum_value(gpio_get_level(mrb_fixnum(pin)));
 }
 
+static mrb_value
+mrb_esp32_gpio_analog_read(mrb_state *mrb, mrb_value self) {
+  mrb_value ch;
+
+  mrb_get_args(mrb, "o", &ch);
+
+  if (!mrb_fixnum_p(ch)) {
+    return mrb_nil_value();
+  }
+
+  adc1_config_channel_atten(mrb_fixnum(ch), ADC_ATTEN_11db);
+
+  return mrb_fixnum_value(adc1_get_voltage(mrb_fixnum(ch)));
+}
+
+static mrb_value
+mrb_esp32_gpio_hall_read(mrb_state *mrb, mrb_value self) {
+  return mrb_fixnum_value(hall_sensor_read());
+}
+
 void
 mrb_mruby_esp32_gpio_gem_init(mrb_state* mrb)
 {
@@ -65,6 +86,10 @@ mrb_mruby_esp32_gpio_gem_init(mrb_state* mrb)
   mrb_define_module_function(mrb, gpio, "pinMode", mrb_esp32_gpio_pin_mode, MRB_ARGS_REQ(2));
   mrb_define_module_function(mrb, gpio, "digitalWrite", mrb_esp32_gpio_digital_write, MRB_ARGS_REQ(2));
   mrb_define_module_function(mrb, gpio, "digitalRead", mrb_esp32_gpio_digital_read, MRB_ARGS_REQ(1));
+  mrb_define_module_function(mrb, gpio, "analogRead", mrb_esp32_gpio_analog_read, MRB_ARGS_REQ(1));
+  mrb_define_module_function(mrb, gpio, "hallRead", mrb_esp32_gpio_hall_read, MRB_ARGS_NONE());
+
+  adc1_config_width(ADC_WIDTH_12Bit);
 
   constants = mrb_define_module_under(mrb, gpio, "Constants");
 
@@ -112,6 +137,16 @@ mrb_mruby_esp32_gpio_gem_init(mrb_state* mrb)
   define_const(GPIO_NUM_39);
   define_const(GPIO_NUM_MAX);
 
+  define_const(ADC1_CHANNEL_0);
+  define_const(ADC1_CHANNEL_1);
+  define_const(ADC1_CHANNEL_2);
+  define_const(ADC1_CHANNEL_3);
+  define_const(ADC1_CHANNEL_4);
+  define_const(ADC1_CHANNEL_5);
+  define_const(ADC1_CHANNEL_6);
+  define_const(ADC1_CHANNEL_7);
+  define_const(ADC1_CHANNEL_MAX);
+
   mrb_define_const(mrb, constants, "LOW", mrb_fixnum_value(0));
   mrb_define_const(mrb, constants, "HIGH", mrb_fixnum_value(1));
 
