如何编写avr I2C主机读取函数

#include <avr/io.h>
#include <util/twi.h>

#define I2C_ADDRESS 0x50 // 从设备地址

void i2c_init() {
    // 初始化I2C总线
    TWBR = 0x48; // 设置I2C通信速率
    TWSR = 0x00; // 设置预分频因子为1
    TWCR = (1 << TWEN); // 使能I2C总线
}

void i2c_start() {
    // 发送起始信号
    TWCR = (1 << TWINT) | (1 << TWSTA) | (1 << TWEN);
    while (!(TWCR & (1 << TWINT))); // 等待起始信号发送完成
}

void i2c_send_address(uint8_t address, uint8_t read) {
    // 发送从设备地址和读取位
    TWDR = (address << 1) | read;
    TWCR = (1 << TWINT) | (1 << TWEN);
    while (!(TWCR & (1 << TWINT))); // 等待地址发送完成
}

uint8_t i2c_read_data() {
    // 读取数据
    TWCR = (1 << TWINT) | (1 << TWEN) | (1 << TWEA);
    while (!(TWCR & (1 << TWINT))); // 等待数据接收完成
    return TWDR;
}

void i2c_stop() {
    // 发送停止信号
    TWCR = (1 << TWINT) | (1 << TWSTO) | (1 << TWEN);
}

uint8_t i2c_read_byte(uint8_t address) {
    i2c_start(); // 发送起始信号
    i2c_send_address(address, 1); // 发送从设备地址和读取位
    uint8_t data = i2c_read_data(); // 读取数据
    i2c_stop(); // 发送停止信号
    return data;
}

int main() {
    i2c_init(); // 初始化I2C总线
    uint8_t data = i2c_read_byte(I2C_ADDRESS); // 读取从设备的数据
    // 处理读取到的数据
    return 0;
}