popcnt.h

// SPDX-FileCopyrightText: 2026 Edward Kmett <ekmett@gmail.com>
// SPDX-License-Identifier: BSD-2-Clause OR Apache-2.0
#pragma once
#include "native/config.h"
#include "native/attributes.h"
#include "native/isa.h"
#include <bit>
#include <cstdint>
#if NATIVE_HOST_X86
#include <immintrin.h>
 
namespace native {
  template<isa<x86> Arch> requires(Arch.has(x86_feature::popcnt))
  native_nodiscard native_inline native_const __attribute__((target("popcnt")))
  constexpr std::uint16_t popcnt(std::uint16_t value) noexcept {
    if (__builtin_is_constant_evaluated()) {
      return static_cast<std::uint16_t>(std::popcount(value));
    } else {
      return static_cast<std::uint16_t>(_mm_popcnt_u32(value));
    }
  }

  template<isa<x86> Arch> requires(Arch.has(x86_feature::popcnt))
  native_nodiscard native_inline native_const __attribute__((target("popcnt")))
  constexpr std::uint32_t popcnt(std::uint32_t value) noexcept {
    if (__builtin_is_constant_evaluated()) {
      return static_cast<std::uint32_t>(std::popcount(value));
    } else {
      return static_cast<std::uint32_t>(_mm_popcnt_u32(value));
    }
  }

  template<isa<x86> Arch> requires(Arch.has(x86_feature::popcnt))
  native_nodiscard native_inline native_const __attribute__((target("popcnt")))
  constexpr std::uint64_t popcnt(std::uint64_t value) noexcept {
    if (__builtin_is_constant_evaluated()) {
      return static_cast<std::uint64_t>(std::popcount(value));
    } else {
      return static_cast<std::uint64_t>(_mm_popcnt_u64(value));
    }
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::popcnt))
  native_nodiscard
  consteval std::uint16_t popcnt(std::uint16_t value) noexcept {
    return popcnt<isa<x86>{x86_feature::popcnt}>(value);
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::popcnt))
  native_nodiscard
  consteval std::uint32_t popcnt(std::uint32_t value) noexcept {
    return popcnt<isa<x86>{x86_feature::popcnt}>(value);
  }

  template<isa<x86> Arch> requires(!Arch.has(x86_feature::popcnt))
  native_nodiscard
  consteval std::uint64_t popcnt(std::uint64_t value) noexcept {
    return popcnt<isa<x86>{x86_feature::popcnt}>(value);
  }

}
#endif