xref: /freebsd/lib/libc/stdbit/stdc_trailing_ones.c (revision 6296500a85c8474e3ff3fe2f8e4a9d56dd0acd64)

/*
 * Copyright (c) 2025 Robert Clausecker <fuz@FreeBSD.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <assert.h>
#include <limits.h>
#include <stdbit.h>

/* Avoid triggering undefined behavior if x == ~0. */
static_assert(UCHAR_WIDTH < UINT_WIDTH,
    "stdc_trailing_ones_uc needs UCHAR_WIDTH < UINT_WIDTH");

unsigned int
stdc_trailing_ones_uc(unsigned char x)
{
	return (__builtin_ctz(~x));
}

/* Avoid triggering undefined behavior if x == ~0. */
static_assert(USHRT_WIDTH < UINT_WIDTH,
    "stdc_trailing_ones_uc needs USHRT_WIDTH < UINT_WIDTH");

unsigned int
stdc_trailing_ones_us(unsigned short x)
{
	return (__builtin_ctz(~x));
}

unsigned int
stdc_trailing_ones_ui(unsigned int x)
{
	if (x == ~0U)
		return (UINT_WIDTH);

	return (__builtin_ctz(~x));
}

unsigned int
stdc_trailing_ones_ul(unsigned long x)
{
	if (x == ~0UL)
		return (ULONG_WIDTH);

	return (__builtin_ctzl(~x));
}

unsigned int
stdc_trailing_ones_ull(unsigned long long x)
{
	if (x == ~0ULL)
		return (ULLONG_WIDTH);

	return (__builtin_ctzll(~x));
}