refactor: remove mut_shared_buffer function and streamline buffer handling in SPI transactions

This commit is contained in:
2025-08-07 00:26:53 +08:00
parent 6269bb99c4
commit ffcaba10fd
2 changed files with 66 additions and 60 deletions

View File

@ -35,8 +35,6 @@ constexpr uint8_t SPI_READ_BUFFER_CMD = RADIOLIB_SX126X_CMD_READ_BUFFER;
constexpr size_t DEFAULT_TIMEOUT_MS = 1000; constexpr size_t DEFAULT_TIMEOUT_MS = 1000;
std::span<uint8_t> mut_shared_buffer();
struct spi_config_t { struct spi_config_t {
spi_host_device_t host_id; spi_host_device_t host_id;
int clock_speed_hz; int clock_speed_hz;

View File

@ -14,29 +14,21 @@
// https://docs.espressif.com/projects/esp-idf/en/v5.3.2/esp32h2/api-reference/peripherals/spi_master.html // https://docs.espressif.com/projects/esp-idf/en/v5.3.2/esp32h2/api-reference/peripherals/spi_master.html
// https://github.com/espressif/esp-idf/blob/v5.3.2/examples/peripherals/spi_master/lcd/main/spi_master_example_main.c // https://github.com/espressif/esp-idf/blob/v5.3.2/examples/peripherals/spi_master/lcd/main/spi_master_example_main.c
namespace app::driver::hal::spi { namespace app::driver::hal::spi {
constexpr auto SPI_CMD_BIT_SIZE = 8; static constexpr uint8_t CMD_BIT_SIZE = 8;
constexpr auto SPI_BUFFER_OFFSET_BIT_SIZE = 8; static constexpr uint8_t OFFSET_BIT_SIZE = 8;
constexpr auto SPI_REGISTER_ADDR_BIT_SIZE = 16; static constexpr uint8_t REG_ADDR_BIT_SIZE = 16;
constexpr auto MAX_BUFFER_SIZE = 256;
static constexpr auto MAX_BUFFER_SIZE = 256;
namespace details { namespace details {
static bool is_initialized = false; static bool is_initialized = false;
static spi_device_handle_t spi_device; static spi_device_handle_t spi_device;
/// should satisfy most of the control register write/read static uint8_t _data_buffer[MAX_BUFFER_SIZE];
/// without overlap with the data buffer
constexpr auto DATA_BUFFER_OFFSET = 24;
static_assert(DATA_BUFFER_OFFSET < MAX_BUFFER_SIZE,
"DATA_BUFFER_OFFSET must be less than MAX_BUFFER_SIZE");
static uint8_t _shared_buffer[MAX_BUFFER_SIZE];
/**
* @brief used in generic stream io
*/
static std::span<uint8_t> shared_buffer = std::span{_shared_buffer};
/** /**
* @brief only used in read/write buffer io * @brief only used in read/write buffer io
*/ */
static std::span<uint8_t> data_buffer = std::span{_shared_buffer + DATA_BUFFER_OFFSET, MAX_BUFFER_SIZE - DATA_BUFFER_OFFSET}; static std::span<uint8_t> data_buffer = std::span{_data_buffer};
} }
/** /**
@ -59,10 +51,6 @@ static constexpr auto verify_statuses = [](std::span<const uint8_t> statuses) {
return error::OK; return error::OK;
}; };
std::span<uint8_t> mut_shared_buffer() {
return std::span{details::shared_buffer};
}
void init(spi_config_t config) { void init(spi_config_t config) {
if (details::is_initialized) { if (details::is_initialized) {
return; return;
@ -162,9 +150,15 @@ inline bool wait_for_not_busy(const size_t timeout_ms) {
/*! READ */ /*! READ */
constexpr auto STACK_BUFFER_SIZE = 32;
constexpr auto PADDING = 0xFE;
Result<Unit, error_t> Result<Unit, error_t>
read_stream(uint8_t cmd, std::span<uint8_t> data, const size_t timeout_ms) { read_stream(uint8_t cmd, std::span<uint8_t> data, const size_t timeout_ms) {
if (data.size() > MAX_BUFFER_SIZE - 1) { std::array<uint8_t, STACK_BUFFER_SIZE> _rx_buf;
std::array<uint8_t, STACK_BUFFER_SIZE> _tx_buf;
if (data.size() > STACK_BUFFER_SIZE - 1) {
return ue_t{error::INVALID_SIZE}; return ue_t{error::INVALID_SIZE};
} }
@ -172,7 +166,9 @@ read_stream(uint8_t cmd, std::span<uint8_t> data, const size_t timeout_ms) {
return ue_t{error::TIMEOUT}; return ue_t{error::TIMEOUT};
} }
auto rx_buffer = details::shared_buffer.subspan(0, data.size() + 1); auto rx_buffer = std::span{_rx_buf}.subspan(0, data.size() + 1);
auto tx_buffer = std::span{_tx_buf}.subspan(0, rx_buffer.size());
std::ranges::fill(tx_buffer, PADDING);
spi_transaction_ext_t transaction; spi_transaction_ext_t transaction;
transaction = spi_transaction_ext_t{ transaction = spi_transaction_ext_t{
@ -181,16 +177,16 @@ read_stream(uint8_t cmd, std::span<uint8_t> data, const size_t timeout_ms) {
.cmd = cmd, .cmd = cmd,
.length = static_cast<size_t>(rx_buffer.size() * 8), .length = static_cast<size_t>(rx_buffer.size() * 8),
.rxlength = static_cast<size_t>(rx_buffer.size() * 8), .rxlength = static_cast<size_t>(rx_buffer.size() * 8),
.tx_buffer = nullptr, .tx_buffer = tx_buffer.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = CMD_BIT_SIZE,
.address_bits = 0, .address_bits = 0,
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }
@ -208,7 +204,9 @@ read_stream(uint8_t cmd, std::span<uint8_t> data, const size_t timeout_ms) {
Result<Unit, error_t> Result<Unit, error_t>
write_stream(uint8_t cmd, std::span<const uint8_t> data, const size_t timeout_ms) { write_stream(uint8_t cmd, std::span<const uint8_t> data, const size_t timeout_ms) {
if (data.size() > MAX_BUFFER_SIZE) { std::array<uint8_t, STACK_BUFFER_SIZE> _rx_buf;
if (data.size() > STACK_BUFFER_SIZE - 1) {
return ue_t{error::INVALID_SIZE}; return ue_t{error::INVALID_SIZE};
} }
@ -216,7 +214,7 @@ write_stream(uint8_t cmd, std::span<const uint8_t> data, const size_t timeout_ms
return ue_t{error::TIMEOUT}; return ue_t{error::TIMEOUT};
} }
const auto rx_buffer = details::shared_buffer.subspan(0, data.size()); const auto rx_buffer = std::span{_rx_buf}.subspan(0, data.size());
spi_transaction_ext_t transaction; spi_transaction_ext_t transaction;
transaction = spi_transaction_ext_t{ transaction = spi_transaction_ext_t{
.base = { .base = {
@ -227,13 +225,13 @@ write_stream(uint8_t cmd, std::span<const uint8_t> data, const size_t timeout_ms
.tx_buffer = data.data(), .tx_buffer = data.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = CMD_BIT_SIZE,
.address_bits = 0, .address_bits = 0,
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }
@ -249,7 +247,10 @@ write_stream(uint8_t cmd, std::span<const uint8_t> data, const size_t timeout_ms
Result<Unit, error_t> Result<Unit, error_t>
read_register(uint16_t reg, std::span<uint8_t> data, const size_t timeout_ms) { read_register(uint16_t reg, std::span<uint8_t> data, const size_t timeout_ms) {
if (data.size() > MAX_BUFFER_SIZE - 1) { std::array<uint8_t, STACK_BUFFER_SIZE> _rx_buf;
std::array<uint8_t, STACK_BUFFER_SIZE> _tx_buf;
if (data.size() > STACK_BUFFER_SIZE - 1) {
return ue_t{error::INVALID_SIZE}; return ue_t{error::INVALID_SIZE};
} }
@ -264,7 +265,10 @@ read_register(uint16_t reg, std::span<uint8_t> data, const size_t timeout_ms) {
// if the tx_buffer is nullptr. // if the tx_buffer is nullptr.
spi_transaction_ext_t transaction; spi_transaction_ext_t transaction;
auto rx_buffer = details::shared_buffer.subspan(0, data.size() + 1); auto rx_buffer = std::span{_rx_buf}.subspan(0, data.size() + 1);
auto tx_buffer = std::span{_tx_buf}.subspan(0, rx_buffer.size());
std::ranges::fill(tx_buffer, PADDING);
transaction = spi_transaction_ext_t{ transaction = spi_transaction_ext_t{
.base = { .base = {
.flags = SPI_TRANS_VARIABLE_CMD | SPI_TRANS_VARIABLE_ADDR, .flags = SPI_TRANS_VARIABLE_CMD | SPI_TRANS_VARIABLE_ADDR,
@ -272,16 +276,16 @@ read_register(uint16_t reg, std::span<uint8_t> data, const size_t timeout_ms) {
.addr = reg, .addr = reg,
.length = static_cast<size_t>(rx_buffer.size() * 8), .length = static_cast<size_t>(rx_buffer.size() * 8),
.rxlength = static_cast<size_t>(rx_buffer.size() * 8), .rxlength = static_cast<size_t>(rx_buffer.size() * 8),
.tx_buffer = nullptr, .tx_buffer = tx_buffer.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = static_cast<uint8_t>(CMD_BIT_SIZE),
.address_bits = static_cast<uint8_t>(SPI_REGISTER_ADDR_BIT_SIZE), .address_bits = static_cast<uint8_t>(REG_ADDR_BIT_SIZE),
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }
@ -297,7 +301,8 @@ read_register(uint16_t reg, std::span<uint8_t> data, const size_t timeout_ms) {
Result<Unit, error_t> Result<Unit, error_t>
write_register(uint16_t offset, std::span<const uint8_t> data, const size_t timeout_ms) { write_register(uint16_t offset, std::span<const uint8_t> data, const size_t timeout_ms) {
if (data.size() > MAX_BUFFER_SIZE) { std::array<uint8_t, STACK_BUFFER_SIZE> _rx_buf;
if (data.size() > STACK_BUFFER_SIZE - 1) {
return ue_t{error::INVALID_SIZE}; return ue_t{error::INVALID_SIZE};
} }
@ -305,7 +310,7 @@ write_register(uint16_t offset, std::span<const uint8_t> data, const size_t time
return ue_t{error::TIMEOUT}; return ue_t{error::TIMEOUT};
} }
auto rx_buffer = details::shared_buffer.subspan(0, data.size()); auto rx_buffer = std::span{_rx_buf}.subspan(0, data.size());
spi_transaction_ext_t transaction; spi_transaction_ext_t transaction;
transaction = spi_transaction_ext_t{ transaction = spi_transaction_ext_t{
@ -318,14 +323,14 @@ write_register(uint16_t offset, std::span<const uint8_t> data, const size_t time
.tx_buffer = data.data(), .tx_buffer = data.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = static_cast<uint8_t>(CMD_BIT_SIZE),
.address_bits = static_cast<uint8_t>(SPI_REGISTER_ADDR_BIT_SIZE), .address_bits = static_cast<uint8_t>(REG_ADDR_BIT_SIZE),
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }
@ -346,6 +351,9 @@ read_buffer(uint8_t offset, uint8_t size, const size_t timeout_ms) {
} }
auto rx_buffer = details::data_buffer.subspan(0, size_required); auto rx_buffer = details::data_buffer.subspan(0, size_required);
auto _tx_buffer = std::make_unique<uint8_t[]>(rx_buffer.size());
auto tx_buffer = std::span{_tx_buffer.get(), rx_buffer.size()};
std::ranges::fill(tx_buffer, PADDING);
spi_transaction_ext_t transaction; spi_transaction_ext_t transaction;
transaction = spi_transaction_ext_t{ transaction = spi_transaction_ext_t{
@ -355,16 +363,16 @@ read_buffer(uint8_t offset, uint8_t size, const size_t timeout_ms) {
.addr = offset, .addr = offset,
.length = static_cast<size_t>(rx_buffer.size() * 8), .length = static_cast<size_t>(rx_buffer.size() * 8),
.rxlength = static_cast<size_t>(rx_buffer.size() * 8), .rxlength = static_cast<size_t>(rx_buffer.size() * 8),
.tx_buffer = nullptr, .tx_buffer = tx_buffer.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = static_cast<uint8_t>(CMD_BIT_SIZE),
.address_bits = static_cast<uint8_t>(SPI_BUFFER_OFFSET_BIT_SIZE), .address_bits = static_cast<uint8_t>(OFFSET_BIT_SIZE),
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }
@ -401,14 +409,14 @@ write_buffer(uint8_t offset, std::span<const uint8_t> data_from_host, const size
.tx_buffer = data_from_host.data(), .tx_buffer = data_from_host.data(),
.rx_buffer = rx_buffer.data(), .rx_buffer = rx_buffer.data(),
}, },
.command_bits = static_cast<uint8_t>(SPI_CMD_BIT_SIZE), .command_bits = static_cast<uint8_t>(CMD_BIT_SIZE),
.address_bits = static_cast<uint8_t>(SPI_BUFFER_OFFSET_BIT_SIZE), .address_bits = static_cast<uint8_t>(OFFSET_BIT_SIZE),
.dummy_bits = 0, .dummy_bits = 0,
}; };
esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base); esp_err_t err = spi_device_transmit(details::spi_device, &transaction.base);
if (err != ESP_OK) { if (err != ESP_OK) {
ESP_LOGE(TAG, "failed to transfer %s (%d)", esp_err_to_name(err), err); ESP_LOGE(TAG, "spi transmit: %s (%d)", esp_err_to_name(err), err);
return ue_t{error::FAILED}; return ue_t{error::FAILED};
} }