Browse Source
* Use SPI for sending serial data. * Supports 1-4 shift registers chained together by setting `ngios` property appropriately.xmkb
Peter Johanson
3 years ago
committed by
Pete Johanson
5 changed files with 274 additions and 1 deletions
@ -1 +1,2 @@
@@ -1 +1,2 @@
|
||||
rsource "Kconfig.mcp23017" |
||||
rsource "Kconfig.595" |
@ -0,0 +1,24 @@
@@ -0,0 +1,24 @@
|
||||
# 595 GPIO configuration options |
||||
|
||||
# Copyright (c) 2022 The ZMK Contributors |
||||
# SPDX-License-Identifier: MIT |
||||
|
||||
DT_COMPAT_ZMK_GPIO_595 := zmk,gpio-595 |
||||
|
||||
menuconfig GPIO_595 |
||||
bool "595 Shift Register SPI driver" |
||||
default $(dt_compat_enabled,$(DT_COMPAT_ZMK_GPIO_595)) |
||||
depends on SPI |
||||
select HAS_DTS_GPIO |
||||
help |
||||
Enable driver for 595 shift register chip using SPI. |
||||
|
||||
if GPIO_595 |
||||
|
||||
config GPIO_595_INIT_PRIORITY |
||||
int "Init priority" |
||||
default 75 |
||||
help |
||||
Device driver initialization priority. |
||||
|
||||
endif #GPIO_595 |
@ -0,0 +1,215 @@
@@ -0,0 +1,215 @@
|
||||
/*
|
||||
* Copyright (c) 2022 The ZMK Contributors |
||||
* |
||||
* SPDX-License-Identifier: MIT |
||||
*/ |
||||
|
||||
#define DT_DRV_COMPAT zmk_gpio_595 |
||||
|
||||
/**
|
||||
* @file Driver for 595 SPI-based GPIO driver. |
||||
*/ |
||||
|
||||
#include <errno.h> |
||||
|
||||
#include <kernel.h> |
||||
#include <device.h> |
||||
#include <init.h> |
||||
#include <sys/byteorder.h> |
||||
#include <drivers/gpio.h> |
||||
#include <drivers/spi.h> |
||||
|
||||
#define LOG_LEVEL CONFIG_GPIO_LOG_LEVEL |
||||
#include <logging/log.h> |
||||
LOG_MODULE_REGISTER(gpio_595); |
||||
|
||||
/** Configuration data */ |
||||
struct reg_595_config { |
||||
/* gpio_driver_data needs to be first */ |
||||
struct gpio_driver_config common; |
||||
|
||||
struct spi_dt_spec bus; |
||||
|
||||
uint8_t ngpios; |
||||
}; |
||||
|
||||
/** Runtime driver data */ |
||||
struct reg_595_drv_data { |
||||
/* gpio_driver_data needs to be first */ |
||||
struct gpio_driver_config data; |
||||
|
||||
struct k_sem lock; |
||||
|
||||
uint32_t gpio_cache; |
||||
}; |
||||
|
||||
static int reg_595_write_registers(const struct device *dev, uint32_t value) { |
||||
const struct reg_595_config *config = dev->config; |
||||
struct reg_595_drv_data *const drv_data = (struct reg_595_drv_data *const)dev->data; |
||||
int ret = 0; |
||||
|
||||
uint8_t nwrite = config->ngpios / 8; |
||||
uint32_t reg_data = sys_cpu_to_be32(value); |
||||
|
||||
/* Allow a sequence of 1-4 registers in sequence, lowest byte is for the first in the chain */ |
||||
const struct spi_buf tx_buf[1] = {{ |
||||
.buf = ((uint8_t *)®_data) + (4 - nwrite), |
||||
.len = nwrite, |
||||
}}; |
||||
|
||||
const struct spi_buf_set tx = { |
||||
.buffers = tx_buf, |
||||
.count = ARRAY_SIZE(tx_buf), |
||||
}; |
||||
|
||||
ret = spi_write_dt(&config->bus, &tx); |
||||
if (ret < 0) { |
||||
LOG_ERR("spi_write FAIL %d\n", ret); |
||||
return ret; |
||||
} |
||||
|
||||
drv_data->gpio_cache = value; |
||||
return 0; |
||||
} |
||||
|
||||
/**
|
||||
* @brief Setup the pin direction (input or output) |
||||
* |
||||
* @param dev Device struct of the 595 |
||||
* @param pin The pin number |
||||
* @param flags Flags of pin or port |
||||
* |
||||
* @return 0 if successful, failed otherwise |
||||
*/ |
||||
static int setup_pin_dir(const struct device *dev, uint32_t pin, int flags) { |
||||
if ((flags & GPIO_OUTPUT) == 0U) { |
||||
return -ENOTSUP; |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int reg_595_pin_config(const struct device *dev, gpio_pin_t pin, gpio_flags_t flags) { |
||||
int ret; |
||||
|
||||
/* Can't do SPI bus operations from an ISR */ |
||||
if (k_is_in_isr()) { |
||||
return -EWOULDBLOCK; |
||||
} |
||||
|
||||
if ((flags & GPIO_OPEN_DRAIN) != 0U) { |
||||
return -ENOTSUP; |
||||
}; |
||||
|
||||
ret = setup_pin_dir(dev, pin, flags); |
||||
if (ret) { |
||||
LOG_ERR("595: error setting pin direction (%d)", ret); |
||||
} |
||||
|
||||
return ret; |
||||
} |
||||
|
||||
static int reg_595_port_get_raw(const struct device *dev, uint32_t *value) { return -ENOTSUP; } |
||||
|
||||
static int reg_595_port_set_masked_raw(const struct device *dev, uint32_t mask, uint32_t value) { |
||||
struct reg_595_drv_data *const drv_data = (struct reg_595_drv_data *const)dev->data; |
||||
uint32_t buf; |
||||
int ret; |
||||
|
||||
/* Can't do SPI bus operations from an ISR */ |
||||
if (k_is_in_isr()) { |
||||
return -EWOULDBLOCK; |
||||
} |
||||
|
||||
k_sem_take(&drv_data->lock, K_FOREVER); |
||||
|
||||
buf = drv_data->gpio_cache; |
||||
buf = (buf & ~mask) | (mask & value); |
||||
|
||||
ret = reg_595_write_registers(dev, buf); |
||||
|
||||
k_sem_give(&drv_data->lock); |
||||
return ret; |
||||
} |
||||
|
||||
static int reg_595_port_set_bits_raw(const struct device *dev, uint32_t mask) { |
||||
return reg_595_port_set_masked_raw(dev, mask, mask); |
||||
} |
||||
|
||||
static int reg_595_port_clear_bits_raw(const struct device *dev, uint32_t mask) { |
||||
return reg_595_port_set_masked_raw(dev, mask, 0); |
||||
} |
||||
|
||||
static int reg_595_port_toggle_bits(const struct device *dev, uint32_t mask) { |
||||
struct reg_595_drv_data *const drv_data = (struct reg_595_drv_data *const)dev->data; |
||||
uint32_t buf; |
||||
int ret; |
||||
|
||||
/* Can't do SPI bus operations from an ISR */ |
||||
if (k_is_in_isr()) { |
||||
return -EWOULDBLOCK; |
||||
} |
||||
|
||||
k_sem_take(&drv_data->lock, K_FOREVER); |
||||
|
||||
buf = drv_data->gpio_cache; |
||||
buf ^= mask; |
||||
|
||||
ret = reg_595_write_registers(dev, buf); |
||||
|
||||
k_sem_give(&drv_data->lock); |
||||
return ret; |
||||
} |
||||
|
||||
static const struct gpio_driver_api api_table = { |
||||
.pin_configure = reg_595_pin_config, |
||||
.port_get_raw = reg_595_port_get_raw, |
||||
.port_set_masked_raw = reg_595_port_set_masked_raw, |
||||
.port_set_bits_raw = reg_595_port_set_bits_raw, |
||||
.port_clear_bits_raw = reg_595_port_clear_bits_raw, |
||||
.port_toggle_bits = reg_595_port_toggle_bits, |
||||
}; |
||||
|
||||
/**
|
||||
* @brief Initialization function of 595 |
||||
* |
||||
* @param dev Device struct |
||||
* @return 0 if successful, failed otherwise. |
||||
*/ |
||||
static int reg_595_init(const struct device *dev) { |
||||
const struct reg_595_config *const config = dev->config; |
||||
struct reg_595_drv_data *const drv_data = (struct reg_595_drv_data *const)dev->data; |
||||
|
||||
if (!device_is_ready(config->bus.bus)) { |
||||
LOG_ERR("Unable to get SPI bus device"); |
||||
return -ENODEV; |
||||
} |
||||
|
||||
k_sem_init(&drv_data->lock, 1, 1); |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
#define GPIO_PORT_PIN_MASK_FROM_NGPIOS(ngpios) ((gpio_port_pins_t)(((uint64_t)1 << (ngpios)) - 1U)) |
||||
|
||||
#define GPIO_PORT_PIN_MASK_FROM_DT_INST(inst) \ |
||||
GPIO_PORT_PIN_MASK_FROM_NGPIOS(DT_INST_PROP(inst, ngpios)) |
||||
|
||||
#define REG_595_INIT(n) \ |
||||
static struct reg_595_config reg_595_##n##_config = { \
|
||||
.common = \
|
||||
{ \
|
||||
.port_pin_mask = GPIO_PORT_PIN_MASK_FROM_DT_INST(n), \
|
||||
}, \
|
||||
.bus = \
|
||||
SPI_DT_SPEC_INST_GET(n, SPI_OP_MODE_MASTER | SPI_TRANSFER_MSB | SPI_WORD_SET(8), 0), \
|
||||
.ngpios = DT_INST_PROP(n, ngpios), \
|
||||
}; \
|
||||
\
|
||||
static struct reg_595_drv_data reg_595_##n##_drvdata = {}; \
|
||||
\
|
||||
/* This has to init after SPI master */ \
|
||||
DEVICE_DT_INST_DEFINE(n, reg_595_init, NULL, ®_595_##n##_drvdata, ®_595_##n##_config, \
|
||||
POST_KERNEL, CONFIG_GPIO_595_INIT_PRIORITY, &api_table); |
||||
|
||||
DT_INST_FOREACH_STATUS_OKAY(REG_595_INIT) |
@ -0,0 +1,33 @@
@@ -0,0 +1,33 @@
|
||||
# |
||||
# Copyright (c) 2022 The ZMK Contributors |
||||
# |
||||
# SPDX-License-Identifier: MIT |
||||
# |
||||
|
||||
description: > |
||||
This is a representation of the 595 Shift Register. |
||||
|
||||
compatible: "zmk,gpio-595" |
||||
|
||||
include: [gpio-controller.yaml, spi-device.yaml] |
||||
|
||||
properties: |
||||
label: |
||||
required: true |
||||
|
||||
"#gpio-cells": |
||||
const: 2 |
||||
|
||||
ngpios: |
||||
type: int |
||||
required: true |
||||
enum: |
||||
- 8 |
||||
- 16 |
||||
- 24 |
||||
- 32 |
||||
description: Number of gpios supported |
||||
|
||||
gpio-cells: |
||||
- pin |
||||
- flags |
Loading…
Reference in new issue