Threading.hpp

#pragma once
 
#include <pthread.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
 
#define TESTRUN(cmd, msg)                                                                                              \
    int res = cmd;                                                                                                     \
    if (res != 0) {                                                                                                    \
        printf(msg, res);                                                                                              \
        exit(res);                                                                                                     \
    }
 
class Mutex
{
  public:
    Mutex() { TESTRUN(pthread_mutex_init(&mtx, NULL), "Mutex init failed with msg %d") }
 
    virtual ~Mutex() { TESTRUN(pthread_mutex_destroy(&mtx), "Mutex destroy failed with msg %d") }
 
    void lock() { TESTRUN(pthread_mutex_lock(&mtx), "Mutex lock failed with msg %d") }
 
    void unlock() { TESTRUN(pthread_mutex_unlock(&mtx), "Mutex unlock failed with msg %d") }
 
    bool tryLock()
    {
        // return true if lock acquired
        return pthread_mutex_trylock(&mtx) == 0;
    }
 
  protected:
    pthread_mutex_t mtx;
};
 
class Thread
{
  public:
    Thread(void* (*main)(void*), void* arg) { pthread_create(&myTid, NULL, main, arg); }
 
    void join() { pthread_join(myTid, NULL); }
 
    void setThreadAffinity(int coreId)
    {
        cpu_set_t cpuset;
        CPU_ZERO(&cpuset);
        CPU_SET(coreId, &cpuset);
 
        pthread_setaffinity_np(this->myTid, sizeof(cpu_set_t), &cpuset);
    }
 
  protected:
    pthread_t myTid;
};