/*
 * Copyright (c) 2024, Dan Klishch <danilklishch@gmail.com>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <AK/EnumBits.h>
#include <AK/Types.h>

namespace AK {

enum class CPUFeatures : u64 {
    None = 0ULL,
    Invalid = 1ULL << 63,

#if !defined(KERNEL) && (ARCH(I386) || ARCH(X86_64))
#    define AK_CAN_CODEGEN_FOR_X86_SSE42 1
    X86_SSE42 = 1ULL << 0,
#    define AK_CAN_CODEGEN_FOR_X86_SHA 1
    X86_SHA = 1ULL << 1,
#    define AK_CAN_CODEGEN_FOR_X86_AES 1
    X86_AES = 1ULL << 2,
#else
#    define AK_CAN_CODEGEN_FOR_X86_SSE42 0
    X86_SSE42 = Invalid,
#    define AK_CAN_CODEGEN_FOR_X86_SHA 0
    X86_SHA = Invalid,
#    define AK_CAN_CODEGEN_FOR_X86_AES 0
    X86_AES = Invalid,
#endif
};

AK_ENUM_BITWISE_OPERATORS(CPUFeatures);

#ifndef KERNEL
namespace Detail {
CPUFeatures detect_cpu_features_uncached();
}

inline CPUFeatures detect_cpu_features()
{
    static CPUFeatures const s_cached_features = Detail::detect_cpu_features_uncached();
    return s_cached_features;
}
#else
inline CPUFeatures detect_cpu_features()
{
    return CPUFeatures::None;
}
#endif

constexpr bool is_valid_feature(CPUFeatures feature)
{
    return !has_flag(feature, CPUFeatures::Invalid);
}

}

#ifdef USING_AK_GLOBALLY
using AK::CPUFeatures;
using AK::detect_cpu_features;
#endif