Skip to content

Commit

Permalink
AP_HAL_ESP32: Add GPIO driver and buzzer
Browse files Browse the repository at this point in the history
  • Loading branch information
sblaksono committed Oct 1, 2024
1 parent 1bdc635 commit cdbdd93
Show file tree
Hide file tree
Showing 4 changed files with 192 additions and 1 deletion.
141 changes: 141 additions & 0 deletions libraries/AP_HAL_ESP32/GPIO.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,141 @@
/*
* This file is free software: you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This file is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program. If not, see <http://www.gnu.org/licenses/>.
*
* Code by Bayu Laksono
*/

#include "AP_HAL_ESP32.h"
#include "GPIO.h"

#include "hal/gpio_types.h"
#include "driver/gpio.h"

using namespace ESP32;

static gpio_num_t gpio_by_pin_num(uint8_t pin)
{
if (pin < GPIO_NUM_MAX) {
return (gpio_num_t)pin;
}
return GPIO_NUM_NC;
}

GPIO::GPIO()
{}

void GPIO::init()
{}

void GPIO::pinMode(uint8_t pin, uint8_t output)
{
gpio_num_t g = gpio_by_pin_num(pin);
if (g != GPIO_NUM_NC) {
gpio_config_t io_conf = {};
io_conf.intr_type = GPIO_INTR_DISABLE;
io_conf.mode = output ? GPIO_MODE_OUTPUT : GPIO_MODE_INPUT;
io_conf.pin_bit_mask = 1ULL<<g;
io_conf.pull_down_en = output ? GPIO_PULLDOWN_DISABLE : GPIO_PULLDOWN_ENABLE;
io_conf.pull_up_en = GPIO_PULLUP_DISABLE;
gpio_config(&io_conf);
}
}

uint8_t GPIO::read(uint8_t pin)
{
gpio_num_t g = gpio_by_pin_num(pin);
if (g != GPIO_NUM_NC) {
return gpio_get_level(g);
}
return 0;
}

void GPIO::write(uint8_t pin, uint8_t value)
{
gpio_num_t g = gpio_by_pin_num(pin);
if (g != GPIO_NUM_NC) {
gpio_set_level(g, value);
}
}

void GPIO::toggle(uint8_t pin)
{
gpio_num_t g = gpio_by_pin_num(pin);
if (g != GPIO_NUM_NC) {
if (gpio_get_level(g)) {
gpio_set_level(g, 0);
}
else {
gpio_set_level(g, 1);
}
}
}

/* Alternative interface: */
AP_HAL::DigitalSource* GPIO::channel(uint16_t pin) {
gpio_num_t g = gpio_by_pin_num(pin);
if (g != GPIO_NUM_NC) {
return NEW_NOTHROW DigitalSource(pin);
}
return nullptr;
}

bool GPIO::usb_connected(void)
{
return false;
}

DigitalSource::DigitalSource(uint8_t pin) :
_pin(pin)
{}

void DigitalSource::mode(uint8_t output)
{
gpio_num_t g = gpio_by_pin_num(_pin);
if (g != GPIO_NUM_NC) {
gpio_config_t io_conf = {};
io_conf.intr_type = GPIO_INTR_DISABLE;
io_conf.mode = output ? GPIO_MODE_OUTPUT : GPIO_MODE_INPUT;
io_conf.pin_bit_mask = 1ULL<<g;
io_conf.pull_down_en = output ? GPIO_PULLDOWN_DISABLE : GPIO_PULLDOWN_ENABLE;
io_conf.pull_up_en = GPIO_PULLUP_DISABLE;
gpio_config(&io_conf);
}
}

uint8_t DigitalSource::read() {
gpio_num_t g = gpio_by_pin_num(_pin);
if (g != GPIO_NUM_NC) {
return gpio_get_level(g);
}
return 0;
}

void DigitalSource::write(uint8_t value) {
gpio_num_t g = gpio_by_pin_num(_pin);
if (g != GPIO_NUM_NC) {
gpio_set_level(g, value);
}
}

void DigitalSource::toggle() {
gpio_num_t g = gpio_by_pin_num(_pin);
if (g != GPIO_NUM_NC) {
if (gpio_get_level(g)) {
gpio_set_level(g, 0);
}
else {
gpio_set_level(g, 1);
}
}
}
45 changes: 45 additions & 0 deletions libraries/AP_HAL_ESP32/GPIO.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
/*
* This file is free software: you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This file is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include "AP_HAL_ESP32.h"

class ESP32::GPIO : public AP_HAL::GPIO {
public:
GPIO();
void init() override;
void pinMode(uint8_t pin, uint8_t output) override;
uint8_t read(uint8_t pin) override;
void write(uint8_t pin, uint8_t value) override;
void toggle(uint8_t pin) override;

/* Alternative interface: */
AP_HAL::DigitalSource* channel(uint16_t n) override;

/* return true if USB cable is connected */
bool usb_connected(void) override;
};

class ESP32::DigitalSource : public AP_HAL::DigitalSource {
public:
DigitalSource(uint8_t pin);
void mode(uint8_t output) override;
uint8_t read() override;
void write(uint8_t value) override;
void toggle() override;
private:
uint8_t _pin;
};
3 changes: 2 additions & 1 deletion libraries/AP_HAL_ESP32/HAL_ESP32_Class.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
#include "WiFiUdpDriver.h"
#include "RCInput.h"
#include "RCOutput.h"
#include "GPIO.h"
#include "Storage.h"
#include "AnalogIn.h"
#include "Util.h"
Expand Down Expand Up @@ -69,7 +70,7 @@ static Empty::Storage storageDriver;
#else
static ESP32::Storage storageDriver;
#endif
static Empty::GPIO gpioDriver;
static ESP32::GPIO gpioDriver;
#if AP_SIM_ENABLED
static Empty::RCOutput rcoutDriver;
#else
Expand Down
4 changes: 4 additions & 0 deletions libraries/AP_Notify/AP_Notify.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -415,6 +415,10 @@ void AP_Notify::add_backends(void)
ADD_BACKEND(NEW_NOTHROW AP_ToneAlarm());
#endif

// ESP32 noise makers
#elif CONFIG_HAL_BOARD == HAL_BOARD_ESP32
ADD_BACKEND(NEW_NOTHROW Buzzer());

// Linux noise makers
#elif CONFIG_HAL_BOARD == HAL_BOARD_LINUX
#if CONFIG_HAL_BOARD_SUBTYPE == HAL_BOARD_SUBTYPE_LINUX_NAVIO || \
Expand Down

0 comments on commit cdbdd93

Please sign in to comment.