strarg.c

#include "strarg.h"
#include <stdint.h>
#include <string.h>

uint8_t __attribute__((hot)) IsNum(char ch)
{
    return IsInCloseRange(ch, '0', '9');
}

uint8_t __attribute__((hot)) IsBin(char ch)
{
    return IsInCloseRange(ch, '0', '1');
}

uint8_t __attribute__((hot)) IsHex(char ch)
{
    return IsUpper(ch) || IsLower(ch) || IsNum(ch);
}

/*
@brief 校验一个数是否合法(二进制,十进制,十六进制)
@param str 待校验的字符串
@return 校验结果 0-非法 1-二进制 2-十进制 3-十六进制
*/
uint8_t IsVaildNum(const char* str)
{
    uint8_t type = 0;

    uint8_t (*funs[])(char) = {IsBin, IsNum, IsHex};

    while (*str == '0' || *str == '-') // 跳过开头所有的0和符号前缀
        str++;

    if (*str == '\0') // 只有0的情况下
        return 2;     // 算十进制0

    // 根据开头定性格式
    if (*str == 'x' || *str == 'X') { // 十六进制
        type = 2;
    } else if (*str == 'b' || *str == 'B') { // 二进制
        type = 0;
    } else if (IsNum(*str)) { // 十进制
        type = 1;
    } else { // 非法字母
        return 0;
    }

    str++;

    // 校验其余字符是否合规
    while (*str != '\0') {
        if (!funs[type](*str++))
            return 0;
    }

    return type + 1;
}

/*
@brief 将字符串转为整数,自动识别格式(二进制,十进制,十六进制)
@param str 待转换的字符串
@param pNum 转换输出
@return 转换结果 0-转换失败 1-转换成功
*/
uint8_t getNum(const char* str, void* const _pNum)
{
    uint32_t* const pNum = _pNum;

    if (pNum == NULL)
        return 0;

    // 2-十进制 3-十六进制
    uint8_t type = IsVaildNum(str);
    *pNum = 0;

    if (type == 0) { // 非法字符,转换失败
        return 0;
    } else if (type == 1) { // 二进制,需手动转换
        str += 2;           // 跳过0b前缀
        while (*str != '\0') {
            *pNum <<= 1;
            *pNum |= *str - '0';
            str++;
        }
    } else if (type == 2) { // 十进制
        int32_t factor = 1;
        // 处理负数情况
        if ('-' == *str) {
            factor = -1;
            str++;
        }

        while (*str != '\0') {
            *pNum *= 10;
            *pNum += *str - '0';
            str++;
        }

        *pNum = (*pNum) * factor;
    } else if (type == 3) { // 十六进制
        str += 2;           // 跳过0x前缀
        while (*str != '\0') {
            *pNum <<= 4;

            if (*str >= 'a' && *str <= 'f') {
                *pNum += *str - ('a' - 10);
            } else if (*str >= 'A' && *str <= 'F') {
                *pNum += *str - ('A' - 10);
            } else if (*str >= '0' && *str <= '9') {
                *pNum += *str - '0';
            } else {
                *pNum = 0;
                return 0;
            }

            str++;
        }
    }

    return 1;
}

/*
@brief 将字符串转为浮点数
@param str 待转换的字符串
@param pNum 转换输出
@return 转换结果 0-转换失败 1-转换成功
*/
uint8_t getDouble(const char* str, double* const pNum)
{
    int32_t factor = 1;

    *pNum = 0;

    // skip spaces
    while (' ' == *str)
        str++;

    // then skip all zeros, must after skip spaces
    while ('0' == *str)
        str++;

    if ('-' == *str) {
        factor = -1;
        str++;
    }

    for (const char* s = str; *s != '\0'; s++) {
        if ((0 == IsNum(*s)) && ('.' != *s))
            return 0;
    }

    uint64_t val = 0;
    uint32_t factor_inc = 1;

    for (const char* s = str; *s != '\0'; s++) {
        if ('.' == *s) {
            factor_inc = 10;
            s++;
        }

        factor *= factor_inc;

        val *= 10;
        val += *s - '0';
    }

    *pNum = (double) val / factor;

    return 1;
}