From 68e222afdce21dca0f0bfaa19ef1c3e8b2c51a7e Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Thu, 26 May 2016 15:42:27 +0100
Subject: [PATCH] esp8266: Add mp_hal_pin_input() and mp_hal_pin_output()
 functions.

---
 esp8266/esp_mphal.h |  2 ++
 esp8266/modpybpin.c | 29 +++++++++++++++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/esp8266/esp_mphal.h b/esp8266/esp_mphal.h
index 377bbeb6a4..0de494300a 100644
--- a/esp8266/esp_mphal.h
+++ b/esp8266/esp_mphal.h
@@ -68,6 +68,8 @@ void ets_event_poll(void);
 #include "esp8266/modpyb.h"
 #define mp_hal_pin_obj_t uint32_t
 #define mp_hal_get_pin_obj(o) mp_obj_get_pin(o)
+void mp_hal_pin_input(mp_hal_pin_obj_t pin);
+void mp_hal_pin_output(mp_hal_pin_obj_t pin);
 void mp_hal_pin_config_od(mp_hal_pin_obj_t pin);
 #define mp_hal_pin_low(p) do { \
         if ((p) == 16) { WRITE_PERI_REG(RTC_GPIO_ENABLE, (READ_PERI_REG(RTC_GPIO_ENABLE) & ~1) | 1); } \
diff --git a/esp8266/modpybpin.c b/esp8266/modpybpin.c
index 6b9e20268b..166d6f566f 100644
--- a/esp8266/modpybpin.c
+++ b/esp8266/modpybpin.c
@@ -36,6 +36,7 @@
 #include "py/nlr.h"
 #include "py/runtime.h"
 #include "py/gc.h"
+#include "py/mphal.h"
 #include "modpyb.h"
 
 #define GET_TRIGGER(phys_port) \
@@ -123,6 +124,34 @@ uint mp_obj_get_pin(mp_obj_t pin_in) {
     return mp_obj_get_pin_obj(pin_in)->phys_port;
 }
 
+void mp_hal_pin_input(mp_hal_pin_obj_t pin_id) {
+    pin_mode[pin_id] = GPIO_MODE_INPUT;
+    if (pin_id == 16) {
+        WRITE_PERI_REG(PAD_XPD_DCDC_CONF, (READ_PERI_REG(PAD_XPD_DCDC_CONF) & 0xffffffbc) | 1);
+        WRITE_PERI_REG(RTC_GPIO_CONF, READ_PERI_REG(RTC_GPIO_CONF) & ~1);
+        WRITE_PERI_REG(RTC_GPIO_ENABLE, (READ_PERI_REG(RTC_GPIO_ENABLE) & ~1)); // input
+    } else {
+        const pyb_pin_obj_t *self = &pyb_pin_obj[pin_id];
+        PIN_FUNC_SELECT(self->periph, self->func);
+        PIN_PULLUP_DIS(self->periph);
+        gpio_output_set(0, 0, 0, 1 << self->phys_port);
+    }
+}
+
+void mp_hal_pin_output(mp_hal_pin_obj_t pin_id) {
+    pin_mode[pin_id] = GPIO_MODE_OUTPUT;
+    if (pin_id == 16) {
+        WRITE_PERI_REG(PAD_XPD_DCDC_CONF, (READ_PERI_REG(PAD_XPD_DCDC_CONF) & 0xffffffbc) | 1);
+        WRITE_PERI_REG(RTC_GPIO_CONF, READ_PERI_REG(RTC_GPIO_CONF) & ~1);
+        WRITE_PERI_REG(RTC_GPIO_ENABLE, (READ_PERI_REG(RTC_GPIO_ENABLE) & ~1) | 1); // output
+    } else {
+        const pyb_pin_obj_t *self = &pyb_pin_obj[pin_id];
+        PIN_FUNC_SELECT(self->periph, self->func);
+        PIN_PULLUP_DIS(self->periph);
+        gpio_output_set(0, 0, 1 << self->phys_port, 0);
+    }
+}
+
 int pin_get(uint pin) {
     if (pin == 16) {
         return READ_PERI_REG(RTC_GPIO_IN_DATA) & 1;