1
0
mirror of https://github.com/esphome/esphome.git synced 2025-03-15 15:18:16 +00:00
Igor Mammedov 2f78308de0 uart: add drain() API
it would allow to empty RX buffer when necessary

Signed-off-by: Igor Mammedov <imammedo@redhat.com>
2019-05-23 19:15:21 +02:00

127 lines
3.1 KiB
C++

#pragma once
#include <HardwareSerial.h>
#include "esphome/core/esphal.h"
#include "esphome/core/component.h"
namespace esphome {
namespace uart {
#ifdef ARDUINO_ARCH_ESP8266
class ESP8266SoftwareSerial {
public:
void setup(int8_t tx_pin, int8_t rx_pin, uint32_t baud_rate);
uint8_t read_byte();
uint8_t peek_byte();
void flush();
void drain();
void write_byte(uint8_t data);
int available();
protected:
static void gpio_intr(ESP8266SoftwareSerial *arg);
inline void wait_(uint32_t *wait, const uint32_t &start);
inline bool read_bit_(uint32_t *wait, const uint32_t &start);
inline void write_bit_(bool bit, uint32_t *wait, const uint32_t &start);
uint32_t bit_time_{0};
uint8_t *rx_buffer_{nullptr};
size_t rx_buffer_size_{64};
volatile size_t rx_in_pos_{0};
size_t rx_out_pos_{0};
ISRInternalGPIOPin *tx_pin_{nullptr};
ISRInternalGPIOPin *rx_pin_{nullptr};
};
#endif
class UARTComponent : public Component, public Stream {
public:
void set_baud_rate(uint32_t baud_rate) { baud_rate_ = baud_rate; }
void setup() override;
void dump_config() override;
void write_byte(uint8_t data);
void write_array(const uint8_t *data, size_t len);
void write_str(const char *str);
bool peek_byte(uint8_t *data);
bool read_byte(uint8_t *data);
bool read_array(uint8_t *data, size_t len);
int available() override;
void flush() override;
void drain();
float get_setup_priority() const override { return setup_priority::BUS; }
size_t write(uint8_t data) override;
int read() override;
int peek() override;
void set_tx_pin(uint8_t tx_pin) { this->tx_pin_ = tx_pin; }
void set_rx_pin(uint8_t rx_pin) { this->rx_pin_ = rx_pin; }
protected:
bool check_read_timeout_(size_t len = 1);
HardwareSerial *hw_serial_{nullptr};
#ifdef ARDUINO_ARCH_ESP8266
ESP8266SoftwareSerial *sw_serial_{nullptr};
#endif
optional<uint8_t> tx_pin_;
optional<uint8_t> rx_pin_;
uint32_t baud_rate_;
};
#ifdef ARDUINO_ARCH_ESP32
extern uint8_t next_uart_num;
#endif
class UARTDevice : public Stream {
public:
UARTDevice() = default;
UARTDevice(UARTComponent *parent) : parent_(parent) {}
void set_uart_parent(UARTComponent *parent) { this->parent_ = parent; }
void write_byte(uint8_t data) { this->parent_->write_byte(data); }
void write_array(const uint8_t *data, size_t len) { this->parent_->write_array(data, len); }
void write_str(const char *str) { this->parent_->write_str(str); }
bool read_byte(uint8_t *data) { return this->parent_->read_byte(data); }
bool peek_byte(uint8_t *data) { return this->parent_->peek_byte(data); }
bool read_array(uint8_t *data, size_t len) { return this->parent_->read_array(data, len); }
int available() override { return this->parent_->available(); }
void flush() override { return this->parent_->flush(); }
void drain() { return this->parent_->drain(); }
size_t write(uint8_t data) override { return this->parent_->write(data); }
int read() override { return this->parent_->read(); }
int peek() override { return this->parent_->peek(); }
protected:
UARTComponent *parent_{nullptr};
};
} // namespace uart
} // namespace esphome