#pragma once

#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif

#include <pthread.h>
#include <sched.h>

#include <set>
#include <unordered_map>
#include <vector>


class CPU
{
public:
  typedef std::vector<pthread_t> ThreadIdVec;
  typedef std::vector<int> CPUVec;
  typedef std::unordered_map<int,int> CPU2CoreMap;
  typedef std::unordered_map<int,std::set<int>> Core2CPUsMap;

public:
  static int count();
  static int getaffinity(const pthread_t thread_id, cpu_set_t *cpuset);
  static int setaffinity(const pthread_t thread_id, cpu_set_t *cpuset);
  static int setaffinity(const pthread_t thread_id, const int cpu);
  static int setaffinity(const pthread_t thread_id, const std::set<int> cpus);

  static CPU::CPUVec cpus();
  static CPU::CPU2CoreMap cpu2core();
  static CPU::Core2CPUsMap core2cpus();
};