mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-10 09:48:32 +00:00
Thomas Krijnen
255 lines
9.1 KiB
C++
255 lines
9.1 KiB
C++
/********************************************************************************
|
|
* *
|
|
* This file is part of IfcOpenShell. *
|
|
* *
|
|
* IfcOpenShell is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the Lesser GNU General Public License as published by *
|
|
* the Free Software Foundation, either version 3.0 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
* IfcOpenShell 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 *
|
|
* Lesser GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the Lesser GNU General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
********************************************************************************/
|
|
|
|
#include "IfcLogger.h"
|
|
|
|
#include "../ifcparse/IfcException.h"
|
|
#include "../ifcparse/Argument.h"
|
|
|
|
#include <boost/algorithm/string/replace.hpp>
|
|
#include <boost/optional.hpp>
|
|
|
|
#include <boost/property_tree/ptree.hpp>
|
|
#include <boost/property_tree/json_parser.hpp>
|
|
#include <boost/version.hpp>
|
|
|
|
#include <mutex>
|
|
#include <iostream>
|
|
#include <algorithm>
|
|
#include <ctime>
|
|
#include <iomanip>
|
|
#include <chrono>
|
|
|
|
namespace {
|
|
|
|
std::string get_time(bool with_milliseconds=false) {
|
|
std::ostringstream oss;
|
|
time_t now = time(nullptr);
|
|
oss << std::put_time(localtime(&now), "%F %T");
|
|
|
|
if (with_milliseconds) {
|
|
auto now_chrono = std::chrono::system_clock::now();
|
|
auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(now_chrono.time_since_epoch()) % 1000;
|
|
oss << '.' << std::setfill('0') << std::setw(3) << ms.count();
|
|
}
|
|
|
|
return oss.str();
|
|
}
|
|
|
|
template <typename T>
|
|
struct severity_strings {
|
|
static const std::array<std::basic_string<T>, 5> value;
|
|
};
|
|
|
|
template <>
|
|
const std::array<std::basic_string<char>, 5> severity_strings<char>::value = { "Performance", "Debug", "Notice", "Warning", "Error" };
|
|
|
|
template <>
|
|
const std::array<std::basic_string<wchar_t>, 5> severity_strings<wchar_t>::value = { L"Performance", L"Debug", L"Notice", L"Warning", L"Error" };
|
|
|
|
template <typename T>
|
|
void plain_text_message(T& os, const boost::optional<IfcUtil::IfcBaseClass*>& current_product, Logger::Severity type, const std::string& message, const IfcUtil::IfcBaseInterface* instance) {
|
|
os << "[" << severity_strings<typename T::char_type>::value[type] << "] ";
|
|
os << "[" << get_time(type <= Logger::LOG_PERF).c_str() << "] ";
|
|
if (current_product) {
|
|
std::string global_id = *((IfcUtil::IfcBaseEntity*)*current_product)->get("GlobalId");
|
|
os << "{" << global_id.c_str() << "} ";
|
|
}
|
|
os << message.c_str() << std::endl;
|
|
if (instance) {
|
|
std::string instance_string = instance->data().toString();
|
|
if (instance_string.size() > 259) {
|
|
instance_string = instance_string.substr(0, 256) + "...";
|
|
}
|
|
os << instance_string.c_str() << std::endl;
|
|
}
|
|
}
|
|
|
|
template <typename T>
|
|
std::basic_string<T> string_as(const std::string& s) {
|
|
std::basic_string<T> v;
|
|
v.assign(s.begin(), s.end());
|
|
return v;
|
|
}
|
|
|
|
template <typename T>
|
|
void json_message(T& os, const boost::optional<IfcUtil::IfcBaseClass*>& current_product, Logger::Severity type, const std::string& message, const IfcUtil::IfcBaseInterface* instance) {
|
|
boost::property_tree::basic_ptree<std::basic_string<typename T::char_type>, std::basic_string<typename T::char_type> > pt;
|
|
|
|
// @todo this is crazy
|
|
static const typename T::char_type time_string[] = { 't', 'i', 'm', 'e', 0 };
|
|
static const typename T::char_type level_string[] = { 'l', 'e', 'v', 'e', 'l', 0 };
|
|
static const typename T::char_type product_string[] = { 'p', 'r', 'o', 'd', 'u', 'c', 't', 0 };
|
|
static const typename T::char_type message_string[] = { 'm', 'e', 's', 's', 'a', 'g', 'e', 0 };
|
|
static const typename T::char_type instance_string[] = { 'i', 'n', 's', 't', 'a', 'n', 'c', 'e', 0 };
|
|
|
|
pt.put(level_string, severity_strings<typename T::char_type>::value[type]);
|
|
if (current_product) {
|
|
pt.put(product_string, string_as<typename T::char_type>((**current_product).data().toString()));
|
|
}
|
|
pt.put(message_string, string_as<typename T::char_type>(message));
|
|
if (instance) {
|
|
pt.put(instance_string, string_as<typename T::char_type>(instance->data().toString()));
|
|
}
|
|
|
|
pt.put(time_string, string_as<typename T::char_type>(get_time()));
|
|
|
|
boost::property_tree::write_json(os, pt, false);
|
|
}
|
|
}
|
|
|
|
void Logger::SetProduct(boost::optional<IfcUtil::IfcBaseClass*> product) {
|
|
if (verbosity <= LOG_DEBUG && product) {
|
|
Message(LOG_DEBUG, "Begin processing", *product);
|
|
}
|
|
if (!product && print_perf_stats_on_element) {
|
|
PrintPerformanceStats();
|
|
performance_statistics.clear();
|
|
}
|
|
current_product = product;
|
|
}
|
|
|
|
void Logger::SetOutput(std::ostream* l1, std::ostream* l2) {
|
|
wlog1 = wlog2 = 0;
|
|
log1 = l1;
|
|
log2 = l2;
|
|
if (!log2) {
|
|
log2 = &log_stream;
|
|
}
|
|
}
|
|
|
|
void Logger::SetOutput(std::wostream* l1, std::wostream* l2) {
|
|
log1 = log2 = 0;
|
|
wlog1 = l1;
|
|
wlog2 = l2;
|
|
if (!wlog2) {
|
|
log2 = &log_stream;
|
|
}
|
|
}
|
|
|
|
void Logger::Message(Logger::Severity type, const std::string& message, const IfcUtil::IfcBaseInterface* instance) {
|
|
static std::mutex m;
|
|
std::lock_guard<std::mutex> lk(m);
|
|
|
|
if (type == LOG_PERF) {
|
|
if (!first_timepoint) {
|
|
first_timepoint = std::chrono::time_point_cast<std::chrono::nanoseconds>(std::chrono::high_resolution_clock::now()).time_since_epoch().count();
|
|
}
|
|
double t0 = (std::chrono::time_point_cast<std::chrono::nanoseconds>(std::chrono::high_resolution_clock::now()).time_since_epoch().count() - *first_timepoint) / 1.e9;
|
|
if (message.substr(0, 5) == "done ") {
|
|
auto orig = message.substr(5);
|
|
performance_statistics[orig] += t0 - performance_signal_start[orig];
|
|
} else {
|
|
performance_signal_start[message] = t0;
|
|
}
|
|
}
|
|
|
|
if (type > max_severity) {
|
|
max_severity = type;
|
|
}
|
|
if ((log2 || wlog2) && type >= verbosity) {
|
|
if (format == FMT_PLAIN) {
|
|
if (log2) {
|
|
plain_text_message(*log2, current_product, type, message, instance);
|
|
} else if (wlog2) {
|
|
plain_text_message(*wlog2, current_product, type, message, instance);
|
|
}
|
|
} else if (format == FMT_JSON) {
|
|
if (log2) {
|
|
json_message(*log2, current_product, type, message, instance);
|
|
} else if (wlog2) {
|
|
json_message(*wlog2, current_product, type, message, instance);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void Logger::Message(Logger::Severity type, const std::exception& exception, const IfcUtil::IfcBaseInterface* instance) {
|
|
Message(type, std::string(exception.what()), instance);
|
|
}
|
|
|
|
template <typename T>
|
|
void status(T& log1, const std::string& message, bool new_line) {
|
|
log1 << message.c_str();
|
|
if (new_line) {
|
|
log1 << std::endl;
|
|
} else {
|
|
log1 << std::flush;
|
|
}
|
|
}
|
|
|
|
void Logger::Status(const std::string& message, bool new_line) {
|
|
if (log1) {
|
|
status(*log1, message, new_line);
|
|
} else if (wlog1) {
|
|
status(*wlog1, message, new_line);
|
|
}
|
|
}
|
|
|
|
void Logger::ProgressBar(int progress) {
|
|
Status("\r[" + std::string(progress,'#') + std::string(50 - progress,' ') + "]", false);
|
|
}
|
|
|
|
std::string Logger::GetLog() {
|
|
return log_stream.str();
|
|
}
|
|
|
|
void Logger::PrintPerformanceStats() {
|
|
std::vector<std::pair<double, std::string>> items;
|
|
for (auto& p : performance_statistics) {
|
|
items.push_back({ p.second, p.first });
|
|
}
|
|
|
|
std::sort(items.begin(), items.end());
|
|
std::reverse(items.begin(), items.end());
|
|
|
|
size_t max_size = 0;
|
|
for (auto& p : items) {
|
|
if (p.second.size() > max_size) {
|
|
max_size = p.second.size();
|
|
}
|
|
}
|
|
|
|
for (auto& p : items) {
|
|
auto s = p.second + std::string(max_size - p.second.size(), ' ') + ": " + std::to_string(p.first);
|
|
Message(LOG_PERF, s);
|
|
}
|
|
}
|
|
|
|
void Logger::Verbosity(Logger::Severity v) { verbosity = v; }
|
|
Logger::Severity Logger::Verbosity() { return verbosity; }
|
|
|
|
Logger::Severity Logger::MaxSeverity() { return max_severity; }
|
|
|
|
void Logger::OutputFormat(Format f) { format = f; }
|
|
Logger::Format Logger::OutputFormat() { return format; }
|
|
|
|
std::ostream* Logger::log1 = 0;
|
|
std::ostream* Logger::log2 = 0;
|
|
std::wostream* Logger::wlog1 = 0;
|
|
std::wostream* Logger::wlog2 = 0;
|
|
std::stringstream Logger::log_stream;
|
|
Logger::Severity Logger::verbosity = Logger::LOG_NOTICE;
|
|
Logger::Severity Logger::max_severity = Logger::LOG_NOTICE;
|
|
Logger::Format Logger::format = Logger::FMT_PLAIN;
|
|
boost::optional<IfcUtil::IfcBaseClass*> Logger::current_product;
|
|
boost::optional<long long> Logger::first_timepoint;
|
|
std::map<std::string, double> Logger::performance_statistics;
|
|
std::map<std::string, double> Logger::performance_signal_start;
|
|
bool Logger::print_perf_stats_on_element = false; |