luban-lite-t3e-pro/packages/third-party/i2c-tools/src/i2c_tools.cpp

/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2019-2023 Wu Han
 */

#include "i2c_utils.h"

static void i2c(int argc,char *argv[])
{
    if(argc > 2)
    {
        #ifdef I2C_TOOLS_USE_SW_I2C
            if( i2c_init(strtonum(argv[ARG_SDA_POS]), strtonum(argv[ARG_SCL_POS])) )
            {
                rt_kprintf("[i2c] failed to find bus with sda=%s scl=%s\n", argv[ARG_SDA_POS], argv[ARG_SCL_POS]);
                return;
            }
        #else
            if(i2c_init(argv[ARG_BUS_NAME_POS]))
            {
                rt_kprintf("[i2c] failed to find bus %s\n", argv[ARG_BUS_NAME_POS]);
                return;
            }
        #endif

        if(!strcmp(argv[ARG_CMD_POS], "scan"))
        {
            rt_uint32_t start_addr = 0x00;
            rt_uint32_t stop_addr  = 0x80;
        #ifdef I2C_TOOLS_USE_SW_I2C
            if(argc >= 5)
            {
                start_addr = strtonum(argv[4]);
                if(argc > 5)
                {
                    stop_addr = strtonum(argv[5]);
                }
            }
        #else
            if(argc >= 4)
            {
                start_addr = strtonum(argv[3]);
                if(argc > 4)
                {
                    stop_addr = strtonum(argv[4]);
                }
            }
        #endif
            if (start_addr >= 0x80 || stop_addr > 0x80) {
                rt_kprintf("[i2c] The addresses only range from 0x00 to 0x7F\n");
                return;
            }
            if (start_addr >= stop_addr) {
                rt_kprintf("[i2c] The stop address should be higher than the start address\n");
                return;
            }
            i2c_scan(start_addr, stop_addr);
            return;
        }

        if(argc < 5)
        {
            i2c_help();
            return;
        }

        if(!strcmp(argv[1], "read"))
        {
            rt_uint8_t buffer[I2C_TOOLS_BUFFER_SIZE];
            rt_uint8_t reg = strtonum(argv[ARG_DATA_POS]);
            rt_uint8_t len = 1;

            if (argc == ARG_READ_MAX)
            {
                len = atoi(argv[ARG_READ_MAX - 1]);
            }
            if (len > I2C_TOOLS_BUFFER_SIZE)
            {
                rt_kprintf("The length %d is too long than the max %d\n",
                           len, I2C_TOOLS_BUFFER_SIZE);
                len = I2C_TOOLS_BUFFER_SIZE;
            }
            if (i2c_read(strtonum(argv[ARG_ADDR_POS]), reg, buffer, len) == len)
            {
                for (rt_uint8_t i = 0; i < len; i++, reg++)
                {
                    if (i && (i % 16 == 0))
                        rt_kprintf("\n");
                    if (i % 16 == 0)
                        rt_kprintf("0x%02X: ", reg);
                    else if (i % 8 == 0)
                        rt_kprintf("   ");

                    rt_kprintf("%02X ", buffer[i]);
                }
                rt_kprintf("\n");
            }
            else
            {
                #ifdef I2C_TOOLS_USE_SW_I2C
                    rt_kprintf("[i2c] read from bus failed with sda=%s scl=%s\n", argv[ARG_SDA_POS], argv[ARG_SCL_POS]);
                #else
                    rt_kprintf("[i2c] read from %s failed\n", argv[ARG_BUS_NAME_POS]);
                #endif
            }
        }

        else if(!strcmp(argv[1], "write"))
        {
            unsigned char buffer[I2C_TOOLS_BUFFER_SIZE];
            for(int i=0; i<(argc-ARG_DATA_POS); i++)
            {
                buffer[i] = strtonum(argv[i+ARG_DATA_POS]);
            }
            if(!i2c_write(strtonum(argv[ARG_DATA_POS-1]), buffer, argc-ARG_DATA_POS))
            {
                #ifdef I2C_TOOLS_USE_SW_I2C
                    rt_kprintf("[i2c] write to bus failed with sda=%s scl=%s\n", argv[ARG_SDA_POS], argv[ARG_SCL_POS]);
                #else
                    rt_kprintf("[i2c] write to %s failed\n", argv[ARG_BUS_NAME_POS]);
                #endif
            }
        }
    }
    else
    {
        i2c_help();
    }
}
MSH_CMD_EXPORT(i2c, i2c tools);