|
| 1 | +/* |
| 2 | + * SPDX-FileCopyrightText: 2026 Espressif Systems (Shanghai) CO LTD |
| 3 | + * |
| 4 | + * SPDX-License-Identifier: Apache-2.0 |
| 5 | + */ |
| 6 | + |
| 7 | +#include <inttypes.h> |
| 8 | +#include <string.h> |
| 9 | +#include <stdlib.h> |
| 10 | +#include "esp_bit_defs.h" |
| 11 | +#include "esp_check.h" |
| 12 | +#include "esp_log.h" |
| 13 | +#include "esp_io_expander.h" |
| 14 | +#include "esp_io_expander_aw9523.h" |
| 15 | + |
| 16 | +/* I2C communication related */ |
| 17 | +#define I2C_TIMEOUT_MS (1000) |
| 18 | +#define I2C_CLK_SPEED (400000) |
| 19 | + |
| 20 | +#define IO_COUNT (16) |
| 21 | + |
| 22 | +/* Register address */ |
| 23 | +#define INPUT_REG_ADDR (0x00) |
| 24 | +#define OUTPUT_REG_ADDR (0x02) |
| 25 | +#define DIRECTION_REG_ADDR (0x04) |
| 26 | +#define RESET_REG_ADDR (0x7F) |
| 27 | + |
| 28 | +/* Default register value on power-up */ |
| 29 | +#define DIR_REG_DEFAULT_VAL (0xffff) |
| 30 | +#define OUT_REG_DEFAULT_VAL (0xffff) |
| 31 | + |
| 32 | +/** |
| 33 | + * @brief Device Structure Type |
| 34 | + * |
| 35 | + */ |
| 36 | +typedef struct { |
| 37 | + esp_io_expander_t base; |
| 38 | + i2c_master_dev_handle_t i2c_handle; |
| 39 | + struct { |
| 40 | + uint16_t direction; |
| 41 | + uint16_t output; |
| 42 | + } regs; |
| 43 | +} esp_io_expander_aw9523_t; |
| 44 | + |
| 45 | +static char *TAG = "aw9523"; |
| 46 | + |
| 47 | +static esp_err_t read_input_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 48 | +static esp_err_t write_output_reg(esp_io_expander_handle_t handle, uint32_t value); |
| 49 | +static esp_err_t read_output_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 50 | +static esp_err_t write_direction_reg(esp_io_expander_handle_t handle, uint32_t value); |
| 51 | +static esp_err_t read_direction_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 52 | +static esp_err_t reset(esp_io_expander_t *handle); |
| 53 | +static esp_err_t del(esp_io_expander_t *handle); |
| 54 | + |
| 55 | +esp_err_t esp_io_expander_new_i2c_aw9523(i2c_master_bus_handle_t i2c_bus, uint32_t dev_addr, |
| 56 | + esp_io_expander_handle_t *handle_ret) |
| 57 | +{ |
| 58 | + ESP_RETURN_ON_FALSE(handle_ret != NULL, ESP_ERR_INVALID_ARG, TAG, "Invalid handle_ret"); |
| 59 | + |
| 60 | + // Allocate memory for driver object |
| 61 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)calloc(1, sizeof(esp_io_expander_aw9523_t)); |
| 62 | + ESP_RETURN_ON_FALSE(aw9523 != NULL, ESP_ERR_NO_MEM, TAG, "Malloc failed"); |
| 63 | + |
| 64 | + // Add new I2C device |
| 65 | + esp_err_t ret = ESP_OK; |
| 66 | + const i2c_device_config_t i2c_dev_cfg = { |
| 67 | + .device_address = dev_addr, |
| 68 | + .scl_speed_hz = I2C_CLK_SPEED, |
| 69 | + }; |
| 70 | + ESP_GOTO_ON_ERROR(i2c_master_bus_add_device(i2c_bus, &i2c_dev_cfg, &aw9523->i2c_handle), err, TAG, |
| 71 | + "Add new I2C device failed"); |
| 72 | + |
| 73 | + aw9523->base.config.io_count = IO_COUNT; |
| 74 | + aw9523->base.config.flags.dir_out_bit_zero = 1; |
| 75 | + aw9523->base.read_input_reg = read_input_reg; |
| 76 | + aw9523->base.write_output_reg = write_output_reg; |
| 77 | + aw9523->base.read_output_reg = read_output_reg; |
| 78 | + aw9523->base.write_direction_reg = write_direction_reg; |
| 79 | + aw9523->base.read_direction_reg = read_direction_reg; |
| 80 | + aw9523->base.del = del; |
| 81 | + aw9523->base.reset = reset; |
| 82 | + |
| 83 | + /* Reset configuration and register status */ |
| 84 | + ESP_GOTO_ON_ERROR(reset(&aw9523->base), err1, TAG, "Reset failed"); |
| 85 | + |
| 86 | + *handle_ret = &aw9523->base; |
| 87 | + return ESP_OK; |
| 88 | +err1: |
| 89 | + i2c_master_bus_rm_device(aw9523->i2c_handle); |
| 90 | +err: |
| 91 | + free(aw9523); |
| 92 | + return ret; |
| 93 | +} |
| 94 | + |
| 95 | +static esp_err_t read_input_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 96 | +{ |
| 97 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 98 | + base); |
| 99 | + |
| 100 | + uint8_t temp[2] = {0, 0}; |
| 101 | + ESP_RETURN_ON_ERROR(i2c_master_transmit_receive(aw9523->i2c_handle, (uint8_t[]) { |
| 102 | + INPUT_REG_ADDR |
| 103 | + }, 1, temp, sizeof(temp), I2C_TIMEOUT_MS), TAG, "Read input reg failed"); |
| 104 | + *value = (((uint32_t)temp[1]) << 8) | (temp[0]); |
| 105 | + return ESP_OK; |
| 106 | +} |
| 107 | + |
| 108 | +static esp_err_t write_output_reg(esp_io_expander_handle_t handle, uint32_t value) |
| 109 | +{ |
| 110 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 111 | + base); |
| 112 | + value &= 0xffff; |
| 113 | + |
| 114 | + uint8_t data[] = {OUTPUT_REG_ADDR, value & 0xff, value >> 8}; |
| 115 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 116 | + "Write output reg failed"); |
| 117 | + aw9523->regs.output = value; |
| 118 | + return ESP_OK; |
| 119 | +} |
| 120 | + |
| 121 | +static esp_err_t read_output_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 122 | +{ |
| 123 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 124 | + base); |
| 125 | + |
| 126 | + *value = aw9523->regs.output; |
| 127 | + return ESP_OK; |
| 128 | +} |
| 129 | + |
| 130 | +static esp_err_t write_direction_reg(esp_io_expander_handle_t handle, uint32_t value) |
| 131 | +{ |
| 132 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 133 | + base); |
| 134 | + value &= 0xffff; |
| 135 | + |
| 136 | + uint8_t data[] = {DIRECTION_REG_ADDR, value & 0xff, value >> 8}; |
| 137 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 138 | + "Write direction reg failed"); |
| 139 | + aw9523->regs.direction = value; |
| 140 | + return ESP_OK; |
| 141 | +} |
| 142 | + |
| 143 | +static esp_err_t read_direction_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 144 | +{ |
| 145 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 146 | + base); |
| 147 | + |
| 148 | + *value = aw9523->regs.direction; |
| 149 | + return ESP_OK; |
| 150 | +} |
| 151 | + |
| 152 | +static esp_err_t reset(esp_io_expander_t *handle) |
| 153 | +{ |
| 154 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 155 | + base); |
| 156 | + |
| 157 | + uint8_t data[] = {RESET_REG_ADDR, 0x00}; |
| 158 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 159 | + "Write direction reg failed"); |
| 160 | + return ESP_OK; |
| 161 | +} |
| 162 | + |
| 163 | +static esp_err_t del(esp_io_expander_t *handle) |
| 164 | +{ |
| 165 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 166 | + base); |
| 167 | + |
| 168 | + ESP_RETURN_ON_ERROR(i2c_master_bus_rm_device(aw9523->i2c_handle), TAG, "Remove I2C device failed"); |
| 169 | + free(aw9523); |
| 170 | + return ESP_OK; |
| 171 | +} |
0 commit comments