binary.hpp

浏览该文件的文档.

/*
 * @Description: 二进制工具函数 头文件及其实现
 * @Version:
 * @Autor: dreamy-xay
 * @date: 2023-12-14
 * @LastEditors: dreamy-xay
 * @LastEditTime: 2024-03-13
 */
 
#ifndef COMMON_UTILS_BINARY_HPP
#define COMMON_UTILS_BINARY_HPP
 
#include <vector>
 
#include "common/enum.h"
#include "common/macro.h"
#include "common/type.h"
 
/* declare */
namespace cm {
 
bool ReadBit(uint number, ushort bit);
uint WriteBit(uint number, ushort bit, bool value);
 
std::vector<byte> IntToBytes(uint number, bool big_endian = true);
uint BytesToInt(const std::vector<byte>& bytes, bool big_endian = true);
 
}  // namespace cm
 
/* implement */
namespace cm {
 
inline bool ReadBit(uint number, ushort bit) {
    return number >> bit & 1;
}
 
inline uint WriteBit(uint number, ushort bit, bool value) {
    // 将第 bit 位设置为 1，或者设置为 0
    return value ? number | (1u << bit) : number & ~(1u << bit);
}
 
inline std::vector<byte> IntToBytes(uint number, bool big_endian) {
    std::vector<byte> bytes;
 
    if (big_endian) 
        for (int i = sizeof(number) - 1; i >= 0; --i) 
            bytes.emplace_back((number >> (8 * i)) & 0xFF);
    else
        for (int i = 0; i < sizeof(number); ++i)
            bytes.emplace_back((number >> (8 * i)) & 0xFF);
 
    return std::move(bytes);
}
 
inline uint BytesToInt(const std::vector<byte>& bytes, bool big_endian) {
    uint number = 0;
 
    // 字节数
    std::size_t num_bytes = bytes.size();
 
    if (big_endian)
        for (size_t i = 0; i < num_bytes; ++i)
            number = (number << 8) | bytes[i];
    else
        for (size_t i = num_bytes; i > 0; --i)
            number = (number << 8) | bytes[i - 1];
 
    return number;
}
 
}  // namespace cm
 
#endif