api_buffer.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <cstring>
#include <memory>
 
#include "esphome/core/defines.h"
#include "esphome/core/helpers.h"
 
namespace esphome::api {
 
inline std::unique_ptr<uint8_t[]> make_buffer(size_t n) {
#if defined(USE_ESP8266) || defined(USE_LIBRETINY)
  return std::make_unique<uint8_t[]>(n);
#else
  return std::make_unique_for_overwrite<uint8_t[]>(n);
#endif
}
 
class APIBuffer {
 public:
  void clear() { this->size_ = 0; }
  inline void reserve(size_t n) ESPHOME_ALWAYS_INLINE {
    if (n > this->capacity_)
      this->grow_(n);
  }
  inline void resize(size_t n) ESPHOME_ALWAYS_INLINE {
    this->reserve(n);
    this->size_ = n;  // no zero-fill
  }
  inline void reserve_and_resize(size_t reserve_size, size_t new_size) ESPHOME_ALWAYS_INLINE {
    this->reserve(std::max(reserve_size, new_size));
    this->size_ = new_size;
  }
  uint8_t *data() { return this->data_.get(); }
  const uint8_t *data() const { return this->data_.get(); }
  size_t size() const { return this->size_; }
  bool empty() const { return this->size_ == 0; }
  uint8_t &operator[](size_t i) { return this->data_[i]; }
  const uint8_t &operator[](size_t i) const { return this->data_[i]; }
  void release() {
    this->data_.reset();
    this->size_ = 0;
    this->capacity_ = 0;
  }
 
 protected:
  void grow_(size_t n);
  std::unique_ptr<uint8_t[]> data_;
  size_t size_{0};
  size_t capacity_{0};
};
 
}  // namespace esphome::api