CheckFunction.cpp

Go to the documentation of this file.

#include "thread/CheckFunction.h"
#include "common/Assert.h"
#include <thread>
 
NAMESPACE_SPH_BEGIN
 
const std::thread::id MAIN_THREAD_ID = std::this_thread::get_id();
 
FunctionChecker::FunctionChecker(std::atomic<Size>& reentrantCnt,
    std::atomic<Size>& totalCnt,
    const Flags<CheckFunction> flags)
    : reentrantCnt(reentrantCnt)
    , flags(flags) {
    reentrantCnt++;
    totalCnt++;
 
    if (flags.has(CheckFunction::MAIN_THREAD)) {
        SPH_ASSERT(std::this_thread::get_id() == MAIN_THREAD_ID, "Called from different thread");
    }
    if (flags.has(CheckFunction::NOT_MAIN_THREAD)) {
        SPH_ASSERT(std::this_thread::get_id() != MAIN_THREAD_ID, "Called from main thread");
    }
    if (flags.has(CheckFunction::NON_REENRANT)) {
        SPH_ASSERT(reentrantCnt == 1, "Reentrant " + std::to_string(reentrantCnt));
    }
    if (flags.has(CheckFunction::ONCE)) {
        SPH_ASSERT(totalCnt == 1, "Called more than once");
    }
}
 
FunctionChecker::~FunctionChecker() {
    reentrantCnt--;
 
    if (flags.has(CheckFunction::NO_THROW)) {
        SPH_ASSERT(!std::uncaught_exception(), "Function threw an exception");
    }
}
 
bool isMainThread() {
    return std::this_thread::get_id() == MAIN_THREAD_ID;
}
 
NAMESPACE_SPH_END