|
| 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 GLOBAL_CONTROL_REG_ADDR (0x11) |
| 27 | +#define RESET_REG_ADDR (0x7F) |
| 28 | + |
| 29 | +/* Default register value on power-up */ |
| 30 | +#define DIR_REG_DEFAULT_VAL (0) |
| 31 | +#define OUT_REG_DEFAULT_VAL (0) |
| 32 | + |
| 33 | +/** |
| 34 | + * @brief Device Structure Type |
| 35 | + * |
| 36 | + */ |
| 37 | +typedef struct { |
| 38 | + esp_io_expander_t base; |
| 39 | + i2c_master_dev_handle_t i2c_handle; |
| 40 | + struct { |
| 41 | + uint16_t direction; |
| 42 | + uint16_t output; |
| 43 | + } regs; |
| 44 | +} esp_io_expander_aw9523_t; |
| 45 | + |
| 46 | +static char *TAG = "aw9523"; |
| 47 | + |
| 48 | +static esp_err_t read_input_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 49 | +static esp_err_t write_output_reg(esp_io_expander_handle_t handle, uint32_t value); |
| 50 | +static esp_err_t read_output_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 51 | +static esp_err_t write_direction_reg(esp_io_expander_handle_t handle, uint32_t value); |
| 52 | +static esp_err_t read_direction_reg(esp_io_expander_handle_t handle, uint32_t *value); |
| 53 | +static esp_err_t reset(esp_io_expander_t *handle); |
| 54 | +static esp_err_t del(esp_io_expander_t *handle); |
| 55 | + |
| 56 | +esp_err_t esp_io_expander_new_i2c_aw9523(i2c_master_bus_handle_t i2c_bus, uint32_t dev_addr, |
| 57 | + esp_io_expander_handle_t *handle_ret) |
| 58 | +{ |
| 59 | + ESP_RETURN_ON_FALSE(handle_ret != NULL, ESP_ERR_INVALID_ARG, TAG, "Invalid handle_ret"); |
| 60 | + |
| 61 | + // Allocate memory for driver object |
| 62 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)calloc(1, sizeof(esp_io_expander_aw9523_t)); |
| 63 | + ESP_RETURN_ON_FALSE(aw9523 != NULL, ESP_ERR_NO_MEM, TAG, "Malloc failed"); |
| 64 | + |
| 65 | + // Add new I2C device |
| 66 | + esp_err_t ret = ESP_OK; |
| 67 | + const i2c_device_config_t i2c_dev_cfg = { |
| 68 | + .device_address = dev_addr, |
| 69 | + .scl_speed_hz = I2C_CLK_SPEED, |
| 70 | + }; |
| 71 | + ESP_GOTO_ON_ERROR(i2c_master_bus_add_device(i2c_bus, &i2c_dev_cfg, &aw9523->i2c_handle), err, TAG, |
| 72 | + "Add new I2C device failed"); |
| 73 | + |
| 74 | + aw9523->base.config.io_count = IO_COUNT; |
| 75 | + aw9523->base.config.flags.dir_out_bit_zero = 1; |
| 76 | + aw9523->base.read_input_reg = read_input_reg; |
| 77 | + aw9523->base.write_output_reg = write_output_reg; |
| 78 | + aw9523->base.read_output_reg = read_output_reg; |
| 79 | + aw9523->base.write_direction_reg = write_direction_reg; |
| 80 | + aw9523->base.read_direction_reg = read_direction_reg; |
| 81 | + aw9523->base.del = del; |
| 82 | + aw9523->base.reset = reset; |
| 83 | + |
| 84 | + /* Reset configuration and register status */ |
| 85 | + ESP_GOTO_ON_ERROR(reset(&aw9523->base), err1, TAG, "Reset failed"); |
| 86 | + |
| 87 | + *handle_ret = &aw9523->base; |
| 88 | + return ESP_OK; |
| 89 | +err1: |
| 90 | + i2c_master_bus_rm_device(aw9523->i2c_handle); |
| 91 | +err: |
| 92 | + free(aw9523); |
| 93 | + return ret; |
| 94 | +} |
| 95 | + |
| 96 | +static esp_err_t read_input_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 97 | +{ |
| 98 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 99 | + base); |
| 100 | + |
| 101 | + uint8_t temp[2] = {0, 0}; |
| 102 | + ESP_RETURN_ON_ERROR(i2c_master_transmit_receive(aw9523->i2c_handle, (uint8_t[]) { |
| 103 | + INPUT_REG_ADDR |
| 104 | + }, 1, temp, sizeof(temp), I2C_TIMEOUT_MS), TAG, "Read input reg failed"); |
| 105 | + *value = (((uint32_t)temp[1]) << 8) | (temp[0]); |
| 106 | + return ESP_OK; |
| 107 | +} |
| 108 | + |
| 109 | +static esp_err_t write_output_reg(esp_io_expander_handle_t handle, uint32_t value) |
| 110 | +{ |
| 111 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 112 | + base); |
| 113 | + value &= 0xffff; |
| 114 | + |
| 115 | + uint8_t data[] = {OUTPUT_REG_ADDR, value & 0xff, value >> 8}; |
| 116 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 117 | + "Write output reg failed"); |
| 118 | + aw9523->regs.output = value; |
| 119 | + return ESP_OK; |
| 120 | +} |
| 121 | + |
| 122 | +static esp_err_t read_output_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 123 | +{ |
| 124 | + ESP_RETURN_ON_FALSE(value != NULL, ESP_ERR_INVALID_ARG, TAG, "Pointer to the value must not be NULL"); |
| 125 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 126 | + base); |
| 127 | + |
| 128 | + *value = aw9523->regs.output; |
| 129 | + return ESP_OK; |
| 130 | +} |
| 131 | + |
| 132 | +static esp_err_t write_direction_reg(esp_io_expander_handle_t handle, uint32_t value) |
| 133 | +{ |
| 134 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 135 | + base); |
| 136 | + value &= 0xffff; |
| 137 | + |
| 138 | + uint8_t data[] = {DIRECTION_REG_ADDR, value & 0xff, value >> 8}; |
| 139 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 140 | + "Write direction reg failed"); |
| 141 | + aw9523->regs.direction = value; |
| 142 | + return ESP_OK; |
| 143 | +} |
| 144 | + |
| 145 | +static esp_err_t read_direction_reg(esp_io_expander_handle_t handle, uint32_t *value) |
| 146 | +{ |
| 147 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 148 | + base); |
| 149 | + |
| 150 | + *value = aw9523->regs.direction; |
| 151 | + return ESP_OK; |
| 152 | +} |
| 153 | + |
| 154 | +static esp_err_t reset(esp_io_expander_t *handle) |
| 155 | +{ |
| 156 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 157 | + base); |
| 158 | + |
| 159 | + uint8_t data[] = {RESET_REG_ADDR, 0x00}; |
| 160 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 161 | + "Write direction reg failed"); |
| 162 | + |
| 163 | + /* AW9523 P0 is in push-pull mode */ |
| 164 | + data[0] = GLOBAL_CONTROL_REG_ADDR; |
| 165 | + data[1] = 0x10; |
| 166 | + ESP_RETURN_ON_ERROR(i2c_master_transmit(aw9523->i2c_handle, data, sizeof(data), I2C_TIMEOUT_MS), TAG, |
| 167 | + "Write direction reg failed"); |
| 168 | + |
| 169 | + ESP_RETURN_ON_ERROR(write_direction_reg(handle, DIR_REG_DEFAULT_VAL), TAG, "Write dir reg failed"); |
| 170 | + ESP_RETURN_ON_ERROR(write_output_reg(handle, OUT_REG_DEFAULT_VAL), TAG, "Write output reg failed"); |
| 171 | + |
| 172 | + return ESP_OK; |
| 173 | +} |
| 174 | + |
| 175 | +static esp_err_t del(esp_io_expander_t *handle) |
| 176 | +{ |
| 177 | + esp_io_expander_aw9523_t *aw9523 = (esp_io_expander_aw9523_t *)__containerof(handle, esp_io_expander_aw9523_t, |
| 178 | + base); |
| 179 | + |
| 180 | + ESP_RETURN_ON_ERROR(i2c_master_bus_rm_device(aw9523->i2c_handle), TAG, "Remove I2C device failed"); |
| 181 | + free(aw9523); |
| 182 | + return ESP_OK; |
| 183 | +} |
0 commit comments