/*
 * CPUFamily.cpp
 *
 *  Created on: Sep 23, 2010
 */

/** @file CPUFamily.cpp
 *
 *  Python module to be able to check the processor family (Core/Nehalem)
 *
 */

#include <boost/python.hpp>

#define cpuid(func,ax,bx,cx,dx) __asm__ __volatile__ ("cpuid": "=a" (ax), "=b" (bx), "=c" (cx), "=d" (dx) : "a" (func));

bool is_nehalem() {
    int a,b,c,d; 
    cpuid(1,a,b,c,d); 
    int sse4_2_mask = 1 << 20; 
    return (c & sse4_2_mask);
}

const char* CPUFamily() {
    if (is_nehalem()) {
        return "nehalem";
    } else {
        return "core";
    }
}

BOOST_PYTHON_MODULE(PyCPUFamily)
{
    using namespace boost::python;
    def("CPUFamily", CPUFamily);
}