#include <sys/resource.h>
#include <sched.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/time.h>
#include <string.h>
#include <unistd.h>

extern "C"{
/* get total cores on system*/
int getNumCPU() {
cpu_set_t mask;
if (sched_getaffinity(0,sizeof(cpu_set_t),&mask)) {
fprintf (stderr, "Unable to retrieve affinity\n");
exit(1);
}
int nproc = 0,i;
for(i=0; i<CPU_SETSIZE; i++) 
   {
    if( CPU_ISSET(i,&mask) ) {
    nproc++;
    }
   }
return nproc;
}

void getCPU(int cpu) 
    {
    cpu_set_t mask;
    CPU_ZERO(&mask);
    CPU_SET(cpu, &mask);
    sched_setaffinity(0, sizeof(cpu_set_t), &mask);
    }
}//end extern