/*
 * Copyright (C) 2009 David McPaul
 *
 * includes code from sysinfo.c which is
 * Copyright 2004-2008, Axel Dörfler, axeld@pinc-software.de.
 * Copyright (c) 2002, Carlos Hasan, for Haiku.
 *
 * All rights reserved. Distributed under the terms of the MIT License.
 */
 

#include "CpuCapabilities.h"

#include <string.h>
#include <cpu_type.h>


CPUCapabilities::~CPUCapabilities()
{
}


CPUCapabilities::CPUCapabilities()
        : fCapabilities(0)
{
#if defined(__i386__) || defined(__x86_64__)
        _SetIntelCapabilities();
#endif
}


#if defined(__i386__) || defined(__x86_64__)
void
CPUCapabilities::_SetIntelCapabilities()
{
        cpuid_info baseInfo;
        cpuid_info cpuInfo;
        int32 maxStandardFunction = 0;

        if (get_cpuid(&baseInfo, 0L, 0L) != B_OK) {
                // this CPU doesn't support cpuid
                return;
        }

        maxStandardFunction = baseInfo.eax_0.max_eax;
        if (maxStandardFunction >= 500) {
                maxStandardFunction = 0; /* old Pentium sample chips has cpu signature here */
        }
        
        /* Extended cpuid */

        get_cpuid(&cpuInfo, 0x80000000, 0L);

        if (maxStandardFunction > 0) {

                get_cpuid(&cpuInfo, 1L, 0L);
                if (cpuInfo.eax_1.features & (1UL << 23)) {
                        fCapabilities = CAPABILITY_MMX;
                }
        
                if (cpuInfo.eax_1.features & (1UL << 25)) {
                        fCapabilities = CAPABILITY_SSE1;
                }

                if (cpuInfo.eax_1.features & (1UL << 26)) {
                        fCapabilities = CAPABILITY_SSE2;
                }

                if (maxStandardFunction >= 1) {
                        /* Extended features */
                        if (cpuInfo.eax_1.extended_features & (1UL << 0)) {
                                fCapabilities = CAPABILITY_SSE3;
                        }
                        if (cpuInfo.eax_1.extended_features & (1UL << 9)) {
                                fCapabilities = CAPABILITY_SSSE3;
                        }
                        if (cpuInfo.eax_1.extended_features & (1UL << 19)) {
                                fCapabilities = CAPABILITY_SSE41;
                        }
                        if (cpuInfo.eax_1.extended_features & (1UL << 20)) {
                                fCapabilities = CAPABILITY_SSE42;
                        }
                }
        }
}
#endif // __i386__ || __x86_64__


bool
CPUCapabilities::HasMMX()
{
        return fCapabilities >= CAPABILITY_MMX;
}


bool
CPUCapabilities::HasSSE1()
{
        return fCapabilities >= CAPABILITY_SSE1;
}


bool
CPUCapabilities::HasSSE2()
{
        return fCapabilities >= CAPABILITY_SSE2;
}


bool
CPUCapabilities::HasSSE3()
{
        return fCapabilities >= CAPABILITY_SSE3;
}


bool
CPUCapabilities::HasSSSE3()
{
        return fCapabilities >= CAPABILITY_SSSE3;
}


bool
CPUCapabilities::HasSSE41()
{
        return fCapabilities >= CAPABILITY_SSE41;
}


bool
CPUCapabilities::HasSSE42()
{
        return fCapabilities >= CAPABILITY_SSE42;
}


void
CPUCapabilities::PrintCapabilities()
{
        static const char *CapArray[8] = {
                "", "MMX", "SSE1", "SSE2", "SSE3", "SSSE3", "SSE4.1", "SSE4.2"
        };

        printf("CPU is capable of running ");
        if (fCapabilities > 0) {
                for (uint32 i = 1; i <= fCapabilities; i++)
                        printf("%s ",CapArray[i]);
        } else {
                printf("no extensions");
        }
        printf("\n");
}