Files
mr-library/device/i2c.c
2024-01-02 00:02:48 +08:00

441 lines
12 KiB
C

/*
* @copyright (c) 2023-2024, MR Development Team
*
* @license SPDX-License-Identifier: Apache-2.0
*
* @date 2023-11-09 MacRsh First version
*/
#include "include/device/mr_i2c.h"
#ifdef MR_USING_I2C
static int mr_i2c_bus_open(struct mr_dev *dev)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)dev;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)dev->drv->ops;
/* Reset the hold */
i2c_bus->hold = MR_FALSE;
/* Default address is 0x00 and 7-bit */
return ops->configure(i2c_bus, &i2c_bus->config, 0x00, MR_I2C_ADDR_BITS_7);
}
static int mr_i2c_bus_close(struct mr_dev *dev)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)dev;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)dev->drv->ops;
struct mr_i2c_config close_config = {0};
return ops->configure(i2c_bus, &close_config, 0x00, MR_I2C_ADDR_BITS_7);
}
static ssize_t mr_i2c_bus_read(struct mr_dev *dev, int off, void *buf, size_t size, int async)
{
return MR_ENOTSUP;
}
static ssize_t mr_i2c_bus_write(struct mr_dev *dev, int off, const void *buf, size_t size, int async)
{
return MR_ENOTSUP;
}
static ssize_t mr_i2c_bus_isr(struct mr_dev *dev, int event, void *args)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)dev;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
switch (event)
{
case MR_ISR_I2C_RD_INT:
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)i2c_bus->owner;
uint8_t data = ops->read(i2c_bus, MR_ENABLE);
/* Read data to FIFO. if callback is set, call it */
mr_ringbuf_push_force(&i2c_dev->rd_fifo, data);
if (i2c_dev->dev.rd_call.call != MR_NULL)
{
ssize_t size = (ssize_t)mr_ringbuf_get_data_size(&i2c_dev->rd_fifo);
i2c_dev->dev.rd_call.call(i2c_dev->dev.rd_call.desc, &size);
}
return MR_EOK;
}
default:
{
return MR_ENOTSUP;
}
}
}
/**
* @brief This function registers a i2c-bus.
*
* @param i2c_bus The i2c-bus.
* @param name The name of the i2c-bus.
* @param drv The driver of the i2c-bus.
*
* @return MR_EOK on success, otherwise an error code.
*/
int mr_i2c_bus_register(struct mr_i2c_bus *i2c_bus, const char *name, struct mr_drv *drv)
{
static struct mr_dev_ops ops =
{
mr_i2c_bus_open,
mr_i2c_bus_close,
mr_i2c_bus_read,
mr_i2c_bus_write,
MR_NULL,
mr_i2c_bus_isr
};
struct mr_i2c_config default_config = MR_I2C_CONFIG_DEFAULT;
MR_ASSERT(i2c_bus != MR_NULL);
MR_ASSERT(name != MR_NULL);
MR_ASSERT(drv != MR_NULL);
MR_ASSERT(drv->ops != MR_NULL);
/* Initialize the fields */
i2c_bus->config = default_config;
i2c_bus->owner = MR_NULL;
i2c_bus->hold = MR_FALSE;
/* Register the i2c-bus */
return mr_dev_register(&i2c_bus->dev, name, Mr_Dev_Type_I2C, MR_SFLAG_RDWR, &ops, drv);
}
MR_INLINE int i2c_dev_take_bus(struct mr_i2c_dev *i2c_dev)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
/* Check if the bus is busy */
if ((i2c_bus->hold == MR_TRUE) && (i2c_dev != i2c_bus->owner))
{
return MR_EBUSY;
}
/* If the owner changes, recheck the configuration */
if (i2c_dev != i2c_bus->owner)
{
/* Reconfigure the bus */
if (i2c_dev->config.baud_rate != i2c_bus->config.baud_rate
|| i2c_dev->config.host_slave != i2c_bus->config.host_slave)
{
int addr = (i2c_dev->config.host_slave == MR_I2C_SLAVE) ? i2c_dev->addr : 0x00;
int ret = ops->configure(i2c_bus, &i2c_dev->config, addr, i2c_dev->addr_bits);
if (ret != MR_EOK)
{
return ret;
}
}
i2c_bus->config = i2c_dev->config;
i2c_bus->owner = i2c_dev;
}
i2c_bus->hold = MR_TRUE;
return MR_EOK;
}
MR_INLINE int i2c_dev_release_bus(struct mr_i2c_dev *i2c_dev)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
if (i2c_dev != i2c_bus->owner)
{
return MR_EINVAL;
}
/* If it is a host, release the bus. The slave needs to hold the bus at all times */
if (i2c_dev->config.host_slave == MR_I2C_HOST)
{
i2c_bus->hold = MR_FALSE;
}
return MR_EOK;
}
#define MR_I2C_RD (0)
#define MR_I2C_WR (1)
MR_INLINE void i2c_dev_send_addr(struct mr_i2c_dev *i2c_dev, int rdwr)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
int addr = 0, addr_bits = MR_I2C_ADDR_BITS_7;
/* Get the address, otherwise use the 0x00 */
if (i2c_dev != MR_NULL)
{
addr = i2c_dev->addr;
addr_bits = i2c_dev->addr_bits;
}
/* Set the read command */
if (rdwr == MR_I2C_RD)
{
addr |= 0x01;
}
ops->start(i2c_bus);
ops->send_addr(i2c_bus, addr, addr_bits);
}
MR_INLINE void i2c_dev_send_stop(struct mr_i2c_dev *i2c_dev)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
ops->stop(i2c_bus);
}
MR_INLINE ssize_t i2c_dev_read(struct mr_i2c_dev *i2c_dev, uint8_t *buf, size_t size)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
uint8_t *rd_buf = (uint8_t *)buf;
ssize_t rd_size;
for (rd_size = 0; rd_size < size; rd_size += sizeof(*rd_buf))
{
*rd_buf = ops->read(i2c_bus, (size - rd_size) == sizeof(*rd_buf));
rd_buf++;
}
return rd_size;
}
MR_INLINE ssize_t i2c_dev_write(struct mr_i2c_dev *i2c_dev, const uint8_t *buf, size_t size)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)i2c_dev->dev.parent;
struct mr_i2c_bus_ops *ops = (struct mr_i2c_bus_ops *)i2c_bus->dev.drv->ops;
uint8_t *wr_buf = (uint8_t *)buf;
ssize_t wr_size;
for (wr_size = 0; wr_size < size; wr_size += sizeof(*wr_buf))
{
ops->write(i2c_bus, *wr_buf);
wr_buf++;
}
return wr_size;
}
static int mr_i2c_dev_open(struct mr_dev *dev)
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)dev;
return mr_ringbuf_allocate(&i2c_dev->rd_fifo, i2c_dev->rd_bufsz);
}
static int mr_i2c_dev_close(struct mr_dev *dev)
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)dev;
mr_ringbuf_free(&i2c_dev->rd_fifo);
return MR_EOK;
}
static ssize_t mr_i2c_dev_read(struct mr_dev *dev, int off, void *buf, size_t size, int async)
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)dev;
ssize_t ret = i2c_dev_take_bus(i2c_dev);
if (ret != MR_EOK)
{
return ret;
}
if (i2c_dev->config.host_slave == MR_I2C_HOST)
{
/* Send the address of the register that needs to be read */
if (off >= 0)
{
i2c_dev_send_addr(i2c_dev, MR_I2C_WR);
i2c_dev_write(i2c_dev, (uint8_t *)&off, (i2c_dev->config.reg_bits >> 3));
}
i2c_dev_send_addr(i2c_dev, MR_I2C_RD);
ret = i2c_dev_read(i2c_dev, (uint8_t *)buf, size);
i2c_dev_send_stop(i2c_dev);
} else
{
if (mr_ringbuf_get_bufsz(&i2c_dev->rd_fifo) == 0)
{
ret = i2c_dev_read(i2c_dev, (uint8_t *)buf, size);
} else
{
ret = (ssize_t)mr_ringbuf_read(&i2c_dev->rd_fifo, buf, size);
}
}
i2c_dev_release_bus(i2c_dev);
return ret;
}
static ssize_t mr_i2c_dev_write(struct mr_dev *dev, int off, const void *buf, size_t size, int async)
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)dev;
ssize_t ret = i2c_dev_take_bus(i2c_dev);
if (ret != MR_EOK)
{
return ret;
}
if (i2c_dev->config.host_slave == MR_I2C_HOST)
{
i2c_dev_send_addr(i2c_dev, MR_I2C_WR);
/* Send the address of the register that needs to be written */
if (off >= 0)
{
i2c_dev_write(i2c_dev, (uint8_t *)&off, (i2c_dev->config.reg_bits >> 3));
}
ret = i2c_dev_write(i2c_dev, (uint8_t *)buf, size);
i2c_dev_send_stop(i2c_dev);
} else
{
ret = i2c_dev_write(i2c_dev, (uint8_t *)buf, size);
}
i2c_dev_release_bus(i2c_dev);
return ret;
}
static int mr_i2c_dev_ioctl(struct mr_dev *dev, int off, int cmd, void *args)
{
struct mr_i2c_dev *i2c_dev = (struct mr_i2c_dev *)dev;
switch (cmd)
{
case MR_CTL_I2C_SET_CONFIG:
{
if (args != MR_NULL)
{
struct mr_i2c_bus *i2c_bus = (struct mr_i2c_bus *)dev->parent;
struct mr_i2c_config config = *(struct mr_i2c_config *)args;
/* If holding the bus, release it */
if (i2c_dev == i2c_bus->owner)
{
i2c_bus->hold = MR_FALSE;
i2c_bus->owner = MR_NULL;
}
/* Update the configuration and try again to get the bus */
i2c_dev->config = config;
if (config.host_slave == MR_I2C_SLAVE)
{
int ret = i2c_dev_take_bus(i2c_dev);
if (ret != MR_EOK)
{
return ret;
}
}
return MR_EOK;
}
return MR_EINVAL;
}
case MR_CTL_I2C_SET_RD_BUFSZ:
{
if (args != MR_NULL)
{
size_t bufsz = *(size_t *)args;
int ret = mr_ringbuf_allocate(&i2c_dev->rd_fifo, bufsz);
i2c_dev->rd_bufsz = 0;
if (ret == MR_EOK)
{
i2c_dev->rd_bufsz = bufsz;
}
return ret;
}
return MR_EINVAL;
}
case MR_CTL_I2C_CLR_RD_BUF:
{
mr_ringbuf_reset(&i2c_dev->rd_fifo);
return MR_EOK;
}
case MR_CTL_I2C_GET_CONFIG:
{
if (args != MR_NULL)
{
struct mr_i2c_config *config = (struct mr_i2c_config *)args;
*config = i2c_dev->config;
return MR_EOK;
}
return MR_EINVAL;
}
case MR_CTL_I2C_GET_RD_BUFSZ:
{
if (args != MR_NULL)
{
*(size_t *)args = i2c_dev->rd_bufsz;
return MR_EOK;
}
return MR_EINVAL;
}
case MR_CTL_I2C_GET_RD_DATASZ:
{
if (args != MR_NULL)
{
size_t *size = (size_t *)args;
*size = mr_ringbuf_get_bufsz(&i2c_dev->rd_fifo);
return MR_EOK;
}
}
default:
{
return MR_ENOTSUP;
}
}
}
/**
* @brief This function registers a i2c-device.
*
* @param i2c_dev The i2c-device.
* @param name The name of the i2c-device.
* @param addr The address of the i2c-device.
* @param addr_bits The number of address bits of the i2c-device.
*
* @return MR_EOK on success, otherwise an error code.
*/
int mr_i2c_dev_register(struct mr_i2c_dev *i2c_dev, const char *name, int addr, int addr_bits)
{
static struct mr_dev_ops ops =
{
mr_i2c_dev_open,
mr_i2c_dev_close,
mr_i2c_dev_read,
mr_i2c_dev_write,
mr_i2c_dev_ioctl,
MR_NULL
};
struct mr_i2c_config default_config = MR_I2C_CONFIG_DEFAULT;
MR_ASSERT(i2c_dev != MR_NULL);
MR_ASSERT(name != MR_NULL);
MR_ASSERT((addr_bits == MR_I2C_ADDR_BITS_7) || (addr_bits == MR_I2C_ADDR_BITS_10));
/* Initialize the fields */
i2c_dev->config = default_config;
mr_ringbuf_init(&i2c_dev->rd_fifo, MR_NULL, 0);
#ifndef MR_CFG_I2C_RD_BUFSZ
#define MR_CFG_I2C_RD_BUFSZ (0)
#endif /* MR_CFG_I2C_RD_BUFSZ */
i2c_dev->rd_bufsz = MR_CFG_I2C_RD_BUFSZ;
i2c_dev->addr = (addr_bits == MR_I2C_ADDR_BITS_7) ? addr : ((0xf0 | ((addr >> 7) & 0x06)) << 8) | (addr & 0xff);
i2c_dev->addr_bits = addr_bits;
/* Register the i2c-device */
return mr_dev_register(&i2c_dev->dev, name, Mr_Dev_Type_I2C, MR_SFLAG_RDWR | MR_SFLAG_NONDRV, &ops, MR_NULL);
}
#endif /* MR_USING_I2C */