logs.hpp

Go to the documentation of this file.

// Copyright (C) 2024 Miguel Ángel González Santamarta
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
#ifndef LLAMA_UTILS__LOGS_HPP
#define LLAMA_UTILS__LOGS_HPP
 
#include <cstdarg>
#include <cstdio>
#include <cstring>
 
namespace llama_utils {

typedef void (*LogFunction)(const char *file, const char *function, int line,
                            const char *text, ...);
 
// Declare function pointers for logging at different severity levels
extern LogFunction log_error; 
extern LogFunction log_warn;  
extern LogFunction log_info;  
extern LogFunction log_debug; 

enum LogLevel {
  ERROR = 0,
  WARN,
  INFO,
  DEBUG
};

extern LogLevel log_level;

inline const char *extract_filename(const char *path) {
  const char *filename = std::strrchr(path, '/');
  if (!filename) {
    filename = std::strrchr(path, '\\'); // handle Windows-style paths
  }
  return filename ? filename + 1 : path;
}
 
#define LLAMA_LOG_ERROR(text, ...)                                             \
  if (llama_utils::log_level >= llama_utils::ERROR)                            \
  llama_utils::log_error(llama_utils::extract_filename(__FILE__),              \
                         __FUNCTION__, __LINE__, text, ##__VA_ARGS__)
 
#define LLAMA_LOG_WARN(text, ...)                                              \
  if (llama_utils::log_level >= llama_utils::WARN)                             \
  llama_utils::log_warn(llama_utils::extract_filename(__FILE__), __FUNCTION__, \
                        __LINE__, text, ##__VA_ARGS__)
 
#define LLAMA_LOG_INFO(text, ...)                                              \
  if (llama_utils::log_level >= llama_utils::INFO)                             \
  llama_utils::log_info(llama_utils::extract_filename(__FILE__), __FUNCTION__, \
                        __LINE__, text, ##__VA_ARGS__)
 
#define LLAMA_LOG_DEBUG(text, ...)                                             \
  if (llama_utils::log_level >= llama_utils::DEBUG)                            \
  llama_utils::log_debug(llama_utils::extract_filename(__FILE__),              \
                         __FUNCTION__, __LINE__, text, ##__VA_ARGS__)

void set_log_level(LogLevel log_level);
 
} // namespace llama_utils
 
#endif // llama_utils__LOGS_HPP