#include "i2c_utils.h"

#ifdef I2C_TOOLS_USE_SW_I2C
    #include "SoftwareI2C.h"
    SoftwareI2C softwarei2c;
#else
    struct rt_i2c_bus_device *i2c_bus;
#endif

long strtonum(const char* str)
{
    long data;
    if (strcmp(str, "0x"))
    {
        data = (rt_uint8_t)strtol(str, NULL, 0);
    }
    else
    {
        data = atoi(str);
    }
    return data;
}

void i2c_help()
{
    rt_kprintf("---------------\n");
    rt_kprintf("i2c tools help:\n");
    rt_kprintf("---------------\n");
#ifdef I2C_TOOLS_USE_SW_I2C
    rt_kprintf("i2c scan  pin_sda pin_scl\n");
    rt_kprintf("i2c read  pin_sda pin_scl address  register  [len=1]\n");
    rt_kprintf("i2c write pin_sda pin_scl address [register] data_0 data_1 ...\n\n");
#else
    rt_kprintf("i2c scan  bus_name\n");
    rt_kprintf("i2c read  bus_name address  register  [len=1]\n");
    rt_kprintf("i2c write bus_name address [register] data_0 data_1 ...\n\n");
#endif
}

#ifdef I2C_TOOLS_USE_SW_I2C
rt_bool_t i2c_init(int sda, int scl)
{
    softwarei2c.begin(sda, scl);
    return RT_EOK;
}
#else
rt_bool_t i2c_init(const char* busname)
{
    i2c_bus = rt_i2c_bus_device_find(busname);
    if(i2c_bus == RT_NULL)
    {
        return RT_ERROR;
    }
    else
    {
        return RT_EOK;
    }
}
#endif

int i2c_probe(char addr)
{
#ifdef I2C_TOOLS_USE_SW_I2C
    int ret = softwarei2c.beginTransmission(addr);
    softwarei2c.endTransmission();
    return ret;
#else
    unsigned char cmd[1];
    cmd[0] = 0;

    struct rt_i2c_msg msgs;
    msgs.addr = addr;
    msgs.flags = RT_I2C_WR;
    msgs.buf = cmd;
    msgs.len = 0;

    return rt_i2c_transfer(i2c_bus, &msgs, 1);
#endif
}

rt_bool_t i2c_write(char addr, rt_uint8_t* data, int len)
{
    #ifdef I2C_TOOLS_USE_SW_I2C
        softwarei2c.beginTransmission(addr);
        int ret = softwarei2c.write(data, len);
        softwarei2c.endTransmission();
        return ret;
    #else
        struct rt_i2c_msg msgs;
        msgs.addr = addr;
        msgs.flags = RT_I2C_WR;
        msgs.buf = data;
        msgs.len = len;
        return (rt_i2c_transfer(i2c_bus, &msgs, 1) == 1);
    #endif
}

rt_uint8_t i2c_read(rt_uint8_t addr, rt_uint8_t reg, rt_uint8_t* buffer, rt_uint8_t len)
{
    #ifdef I2C_TOOLS_USE_SW_I2C
        rt_uint8_t ret = 0; 
        softwarei2c.requestFrom(addr, len);
        while (softwarei2c.available()) {
            buffer[ret] = softwarei2c.read();
            ret++;
        }
        return ret;
    #else
        struct rt_i2c_msg msgs[2];

        msgs[0].addr = addr;
        msgs[0].flags = RT_I2C_WR;
        msgs[0].buf = ®
        msgs[0].len = 1;

        msgs[1].addr = addr;
        msgs[1].flags = RT_I2C_RD;
        msgs[1].buf = buffer;
        msgs[1].len = len;

        if (rt_i2c_transfer(i2c_bus, msgs, 2) == 2)
        {
            return len;
        }
        else
        {
            return 0;
        }
    #endif
}

void i2c_scan()
{
    rt_kprintf("     0  1  2  3  4  5  6  7  8  9  a  b  c  d  e  f\n");
    for(unsigned char i=0; i<8; i++)
    {
        rt_kprintf("%d0: ", i);
        for(unsigned char j=0; j<16; j++)
        {
            char addr = i*16+j;
            if( i2c_probe(addr) == 1)
            {
                rt_kprintf("%02X", addr);
            }
            else
            {
                rt_kprintf("--");
            }
            rt_kprintf(" ");
        }
        rt_kprintf("\n");
    }
}