#pragma once
#include <stdint.h>
#ifdef _MSC_VER
#include <intrin.h>
#endif
namespace Luau
{
namespace CodeGen
{
inline int countlz(uint32_t n)
{
#ifdef _MSC_VER
unsigned long rl;
return _BitScanReverse(&rl, n) ? 31 - int(rl) : 32;
#else
return n == 0 ? 32 : __builtin_clz(n);
#endif
}
inline int countrz(uint32_t n)
{
#ifdef _MSC_VER
unsigned long rl;
return _BitScanForward(&rl, n) ? int(rl) : 32;
#else
return n == 0 ? 32 : __builtin_ctz(n);
#endif
}
inline int countlz(uint64_t n)
{
#ifdef _MSC_VER
#ifdef _WIN64
unsigned long rl;
return _BitScanReverse64(&rl, n) ? 63 - int(rl) : 64;
#else
unsigned long rl;
if (_BitScanReverse(&rl, uint32_t(n >> 32)))
return 31 - int(rl);
return _BitScanReverse(&rl, uint32_t(n)) ? 63 - int(rl) : 64;
#endif
#else
return n == 0 ? 64 : __builtin_clzll(n);
#endif
}
inline int countrz(uint64_t n)
{
#ifdef _MSC_VER
#ifdef _WIN64
unsigned long rl;
return _BitScanForward64(&rl, n) ? int(rl) : 64;
#else
unsigned long rl;
if (_BitScanForward(&rl, uint32_t(n)))
return rl;
return _BitScanForward(&rl, uint32_t(n >> 32)) ? int(rl) + 32 : 64;
#endif
#else
return n == 0 ? 64 : __builtin_ctzll(n);
#endif
}
inline int lrotate(uint32_t u, int s)
{
#ifdef _MSC_VER
return _rotl(u, s);
#else
return (u << (s & 31)) | (u >> ((32 - s) & 31));
#endif
}
inline int rrotate(uint32_t u, int s)
{
#ifdef _MSC_VER
return _rotr(u, s);
#else
return (u >> (s & 31)) | (u << ((32 - s) & 31));
#endif
}
inline uint64_t byteswap(uint64_t a)
{
#if defined(__GNUC__) || defined(__clang__)
return __builtin_bswap64(a);
#elif defined(_MSC_VER)
return _byteswap_uint64(a);
#else
return (a >> 56) | ((a & 0x00FF000000000000ull) >> 40) | ((a & 0x0000FF0000000000ull) >> 24) | ((a & 0x000000FF00000000ull) >> 8) |
((a & 0x00000000FF000000ull) << 8) | ((a & 0x0000000000FF0000ull) << 24) | ((a & 0x000000000000FF00ull) << 40) |
((a & 0x00000000000000FFull) << 56);
#endif
}
} }