/* cpuid.c
 *
 * Author           : Alexander J. Yee
 * Date Created     : 01/19/2012
 * Last Modified    : 01/25/2012
 *
 *
 *
 * And of course... The typical copyright stuff...
 *
 *      Redistribution of this program in both source or binary, regardless of
 *  form, with or without modification is permitted as long as the following
 *  conditions are met:
 *          1.  This copyright notice is maintained either inline in the source
 *              or distributed with the binary.
 *          2.  A list of all contributing authors along with their contributions
 *              is included either inline in the source or distributed with the
 *              binary.
 *          3.  The following disclaimer is maintained either inline in the
 *              source or distributed with the binary.
 *
 *  Disclaimer:
 *  This software is provided "as is", without any guarantee made to its
 *  suitability or fitness for any particular use. It may contain bugs so use
 *  of this program is at your own risk. I take no responsibility for any
 *  damage that may unintentionally be caused through its use.
 */

// https://github.com/Mysticial/Flops/blob/e571da6e94f7b6d2d1a90e87b19398c5c4de4375/version1/source/cpuid.c
// http://www.sandpile.org/x86/cpuid.htm

#ifndef _cpuid_c
#define _cpuid_c
#include <stdio.h>
//#include <omp.h>
#include "cpuid.h"
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
#ifdef WIN32
#else
void cpuid(int *info,int x){
  int ax,bx,cx,dx;

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

  info[0] = ax;
  info[1] = bx;
  info[2] = cx;
  info[3] = dx;
}
#endif
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
void cpuid_print_name(){
  int name[13];
  cpuid(name + 0,0x80000002);
  cpuid(name + 4,0x80000003);
  cpuid(name + 8,0x80000004);
  name[12] = '\0';

  printf("CPU Name = %s\n",(char*)name);
//  printf("Threads  = %d\n",omp_get_max_threads());
  printf("\n");
}
void cpuid_print_exts(){
  int x64     = 0;
  int MMX     = 0;
  int SSE     = 0;
  int SSE2    = 0;
  int SSE3    = 0;
  int SSSE3   = 0;
  int SSE41   = 0;
  int SSE42   = 0;
  int SSE4a   = 0;
  int AVX     = 0;
  int AVX2    = 0;
  int XOP     = 0;
  int FMA3    = 0;
  int FMA4    = 0;

  int info[4];
  cpuid(info, 0);
  int nIds = info[0];

  cpuid(info, 0x80000000);
  int nExIds = info[0];

  //  Detect Instruction Set
  if (nIds >= 1){
    cpuid(info,0x00000001);
    MMX   = (info[3] & ((int)1 << 23)) != 0;
    SSE   = (info[3] & ((int)1 << 25)) != 0;
    SSE2  = (info[3] & ((int)1 << 26)) != 0;
    SSE3  = (info[2] & ((int)1 <<  0)) != 0;

    SSSE3 = (info[2] & ((int)1 <<  9)) != 0;
    SSE41 = (info[2] & ((int)1 << 19)) != 0;
    SSE42 = (info[2] & ((int)1 << 20)) != 0;

    AVX   = (info[2] & ((int)1 << 28)) != 0;
    FMA3  = (info[2] & ((int)1 << 12)) != 0;
  }

  if (nExIds >= 0x00000007){
    cpuid(info,0x00000007);
    AVX2  = (info[2] & ((int)1 <<  5)) != 0;
  }

  if (nExIds >= 0x80000001){
    cpuid(info,0x80000001);
    x64   = (info[3] & ((int)1 << 29)) != 0;
    SSE4a = (info[2] & ((int)1 <<  6)) != 0;
    FMA4  = (info[2] & ((int)1 << 16)) != 0;
    XOP   = (info[2] & ((int)1 << 11)) != 0;
  }

  printf("Hardware Features:\n");
  printf("x64   = %d\n",x64);
  printf("MMX   = %d\n",MMX);
  printf("SSE   = %d\n",SSE);
  printf("SSE2  = %d\n",SSE2);
  printf("SSE3  = %d\n",SSE3);
  printf("SSSE3 = %d\n",SSSE3);
  printf("SSE4a = %d\n",SSE4a);
  printf("SSE41 = %d\n",SSE41);
  printf("SSE42 = %d\n",SSE42);
  printf("AVX   = %d\n",AVX);
  printf("AVX2  = %d\n",AVX2);
  printf("FMA3  = %d\n",FMA3);
  printf("FMA4  = %d\n",FMA4);
  printf("XOP   = %d\n",XOP);
  printf("\n");
}
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////
#endif