b180223d18
* Remove recovery and MPI speed tests. * Remove readme. * Remove Python binding. * Add checks in C API.
155 lines
3.7 KiB
C++
155 lines
3.7 KiB
C++
/*!
|
|
* Copyright (c) 2014 by Contributors
|
|
* \file utils.h
|
|
* \brief simple utils to support the code
|
|
* \author Tianqi Chen
|
|
*/
|
|
#ifndef RABIT_INTERNAL_UTILS_H_
|
|
#define RABIT_INTERNAL_UTILS_H_
|
|
|
|
#include <rabit/base.h>
|
|
#include <cstring>
|
|
#include <cstdio>
|
|
#include <string>
|
|
#include <cstdlib>
|
|
#include <stdexcept>
|
|
#include <vector>
|
|
#include "dmlc/io.h"
|
|
#include "xgboost/logging.h"
|
|
#include <cstdarg>
|
|
|
|
#if !defined(__GNUC__) || defined(__FreeBSD__)
|
|
#define fopen64 std::fopen
|
|
#endif // !defined(__GNUC__) || defined(__FreeBSD__)
|
|
|
|
#ifdef _MSC_VER
|
|
// NOTE: sprintf_s is not equivalent to snprintf,
|
|
// they are equivalent when success, which is sufficient for our case
|
|
#define snprintf sprintf_s
|
|
#define vsnprintf vsprintf_s
|
|
|
|
#else
|
|
|
|
#ifdef _FILE_OFFSET_BITS
|
|
#if _FILE_OFFSET_BITS == 32
|
|
#pragma message("Warning: FILE OFFSET BITS defined to be 32 bit")
|
|
#endif // _FILE_OFFSET_BITS == 32
|
|
#endif // _FILE_OFFSET_BITS
|
|
|
|
#ifdef __APPLE__
|
|
#define off64_t off_t
|
|
#define fopen64 std::fopen
|
|
#endif // __APPLE__
|
|
|
|
extern "C" {
|
|
#include <sys/types.h>
|
|
}
|
|
#endif // _MSC_VER
|
|
|
|
#include <cinttypes>
|
|
|
|
namespace rabit {
|
|
/*! \brief namespace for helper utils of the project */
|
|
namespace utils {
|
|
|
|
/*! \brief error message buffer length */
|
|
const int kPrintBuffer = 1 << 12;
|
|
|
|
/* \brief Case-insensitive string comparison */
|
|
inline int CompareStringsCaseInsensitive(const char* s1, const char* s2) {
|
|
#ifdef _MSC_VER
|
|
return _stricmp(s1, s2);
|
|
#else // _MSC_VER
|
|
return strcasecmp(s1, s2);
|
|
#endif // _MSC_VER
|
|
}
|
|
|
|
/* \brief parse config string too bool*/
|
|
inline bool StringToBool(const char* s) {
|
|
return CompareStringsCaseInsensitive(s, "true") == 0 || atoi(s) != 0;
|
|
}
|
|
|
|
inline void HandlePrint(const char *msg) {
|
|
printf("%s", msg);
|
|
}
|
|
|
|
/*! \brief printf, prints messages to the console */
|
|
inline void Printf(const char *fmt, ...) {
|
|
std::string msg(kPrintBuffer, '\0');
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vsnprintf(&msg[0], kPrintBuffer, fmt, args);
|
|
va_end(args);
|
|
HandlePrint(msg.c_str());
|
|
}
|
|
|
|
/*! \brief assert a condition is true, use this to handle debug information */
|
|
inline void Assert(bool exp, const char *fmt, ...) {
|
|
if (!exp) {
|
|
std::string msg(kPrintBuffer, '\0');
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vsnprintf(&msg[0], kPrintBuffer, fmt, args);
|
|
va_end(args);
|
|
LOG(FATAL) << msg;
|
|
}
|
|
}
|
|
|
|
/*!\brief same as assert, but this is intended to be used as a message for users */
|
|
inline void Check(bool exp, const char *fmt, ...) {
|
|
if (!exp) {
|
|
std::string msg(kPrintBuffer, '\0');
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vsnprintf(&msg[0], kPrintBuffer, fmt, args);
|
|
va_end(args);
|
|
LOG(FATAL) << msg;
|
|
}
|
|
}
|
|
|
|
/*! \brief report error message, same as check */
|
|
inline void Error(const char *fmt, ...) {
|
|
{
|
|
std::string msg(kPrintBuffer, '\0');
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
vsnprintf(&msg[0], kPrintBuffer, fmt, args);
|
|
va_end(args);
|
|
LOG(FATAL) << msg;
|
|
}
|
|
}
|
|
} // namespace utils
|
|
|
|
// Can not use std::min on Windows with msvc due to:
|
|
// error C2589: '(': illegal token on right side of '::'
|
|
template <typename T>
|
|
auto Min(T const& l, T const& r) {
|
|
return l < r ? l : r;
|
|
}
|
|
// same with Min
|
|
template <typename T>
|
|
auto Max(T const& l, T const& r) {
|
|
return l > r ? l : r;
|
|
}
|
|
|
|
// easy utils that can be directly accessed in xgboost
|
|
/*! \brief get the beginning address of a vector */
|
|
template<typename T>
|
|
inline T *BeginPtr(std::vector<T> &vec) { // NOLINT(*)
|
|
if (vec.size() == 0) {
|
|
return nullptr;
|
|
} else {
|
|
return &vec[0];
|
|
}
|
|
}
|
|
inline char* BeginPtr(std::string &str) { // NOLINT(*)
|
|
if (str.length() == 0) return nullptr;
|
|
return &str[0];
|
|
}
|
|
inline const char* BeginPtr(const std::string &str) {
|
|
if (str.length() == 0) return nullptr;
|
|
return &str[0];
|
|
}
|
|
} // namespace rabit
|
|
#endif // RABIT_INTERNAL_UTILS_H_
|