assert.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_DETAIL_ASSERT_HPP_
#define PLSSVM_DETAIL_ASSERT_HPP_
#pragma once
 
#include "plssvm/exceptions/source_location.hpp"  // plssvm::source_location
 
#include "fmt/color.h"                            // fmt::emphasis, fmt::fg, fmt::color
#include "fmt/core.h"                             // fmt::format
 
#include <cstdlib>                                // std::abort
#include <iostream>                               // std::cerr, std::endl
#include <string_view>                            // std::string_view
#include <utility>                                // std::forward
 
namespace plssvm::detail {
 
template <typename... Args>
inline void check_assertion(const bool cond, const std::string_view cond_str, const source_location &loc, const std::string_view msg, Args &&...args) {
    // check if assertion holds
    if (!cond) {
        // print assertion error message
        std::cerr << fmt::format(
            "Assertion '{}' failed!\n"
            "  in file      {}\n"
            "  in function  {}\n"
            "  @ line       {}\n\n"
            "{}\n",
            fmt::format(fmt::emphasis::bold | fmt::fg(fmt::color::green), "{}", cond_str),
            loc.file_name(),
            loc.function_name(),
            loc.line(),
            fmt::format(fmt::emphasis::bold | fmt::fg(fmt::color::red), msg, std::forward<Args>(args)...))
                  << std::endl;
 
        // abort further execution
        std::abort();
    }
}
 
#if defined(PLSSVM_ENABLE_ASSERTS) || !defined(NDEBUG)
    #define PLSSVM_ASSERT_ENABLED
#endif
 
#if defined(PLSSVM_ASSERT_ENABLED)
    #define PLSSVM_ASSERT(cond, msg, ...) plssvm::detail::check_assertion((cond), (#cond), plssvm::source_location::current(), (msg), ##__VA_ARGS__)
#else
    #define PLSSVM_ASSERT(cond, msg, ...)
#endif
 
}  // namespace plssvm::detail
 
#endif  // PLSSVM_DETAIL_ASSERT_HPP_