#include <cstdint>

#include <cstring>

#include <cstdio>

#include <cstdlib>

#include "cpuid.h"

#ifdef _WIN32

#define WIN32_LEAN_AND_MEAN

#include <windows.h>

typedef BOOL (WINAPI *LPFN_GLPI)(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION, PDWORD);

int countBitsSet(ULONG_PTR bitMask)

}

bool getProcessorInfoFromOS(int& cpus, int& cores, int& logicalCores, double& clockSpeed)

}

#else

bool getProcessorInfoFromOS(int& cpus, int& cores, int& logicalCores, double& clockSpeed)

{

return false;

}

#endif

#if defined(__x86_64__) || defined(_M_AMD64) || defined(__amd64__) || defined (_M_X64) || defined(_M_IX86) || defined(__i386__)

#define MOODYCAMEL_X86_OR_X64

#endif

#ifdef MOODYCAMEL_X86_OR_X64

struct CPUIDInfo

{

std::uint32_t data[4];

};

#ifdef _MSC_VER

#include <intrin.h>

inline CPUIDInfo cpuid(std::uint32_t eax)

}

#else

inline CPUIDInfo cpuid(std::uint32_t eax)

}

#endif

#endif // MOODYCAMEL_X86_OR_X64

namespace moodycamel

{

const char* getCPUString()

{

// TODO: Support non-x86/-x64 architectures

#ifdef MOODYCAMEL_X86_OR_X64

static char buf[128] = { 0 };

if (buf[0] != 0) {

return buf;

}

CPUIDInfo info = cpuid(0x80000000);

std::uint32_t ex = info.data[0];

for (std::uint32_t i = 0; i + 0x80000002 <= ex && i != 3; ++i) {

*(reinterpret_cast<CPUIDInfo*>(buf) + i) = cpuid(i + 0x80000002);

}

if (buf[0] == 0) {

strcpy(buf, UNKNOWN_CPU_STRING);

return buf;

}

info = cpuid(0);

if (info.data[0] < 1) {

// cpuid(1) not supported

return buf;

}

// Add number of CPUs, cores, HT, and GHz

info = cpuid(1);

bool ht = ((info.data[3] >> 28) & 1) == 1; // Note: This is also 1 on most multi-core systems, even if there's no HT

int cpus, cores, logicalCores;

double clockSpeed;

if (!getProcessorInfoFromOS(cpus, cores, logicalCores, clockSpeed)) {

return buf;

}

// Strip @ nGHz if any, since we re-calculate this ourselves

int atIndex;

for (atIndex = (int)std::strlen(buf) - 1; atIndex != -1; --atIndex) {

if (buf[atIndex] == '@') {

if (atIndex > 0 && buf[atIndex - 1] == ' ') {

--atIndex;

}

buf[atIndex] = '\0';

break;

}

}

char* str = buf + std::strlen(buf);

if (cpus > 1) {

// Assume identical CPUs

logicalCores /= cpus;

cores /= cpus;

std::sprintf(str, " x%d", cpus);

str += strlen(str);

}

ht = ht && logicalCores != cores;

std::sprintf(str, " with %d core%s%s @ %.1fGHz%s", cores, cores == 1 ? "" : "s", ht ? " (HyperThreaded)" : "", clockSpeed, cpus > 1 ? " each" : "");

return buf;

#else

return UNKNOWN_CPU_STRING;

#endif

}

}