qt_sinks.h 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304
  1. // Copyright(c) 2015-present, Gabi Melman, mguludag and spdlog contributors.
  2. // Distributed under the MIT License (http://opensource.org/licenses/MIT)
  3. #pragma once
  4. //
  5. // Custom sink for QPlainTextEdit or QTextEdit and its children (QTextBrowser...
  6. // etc) Building and using requires Qt library.
  7. //
  8. // Warning: the qt_sink won't be notified if the target widget is destroyed.
  9. // If the widget's lifetime can be shorter than the logger's one, you should provide some permanent
  10. // QObject, and then use a standard signal/slot.
  11. //
  12. #include "spdlog/common.h"
  13. #include "spdlog/details/log_msg.h"
  14. #include "spdlog/details/synchronous_factory.h"
  15. #include "spdlog/sinks/base_sink.h"
  16. #include <array>
  17. #include <QPlainTextEdit>
  18. #include <QTextEdit>
  19. //
  20. // qt_sink class
  21. //
  22. namespace spdlog {
  23. namespace sinks {
  24. template <typename Mutex>
  25. class qt_sink : public base_sink<Mutex> {
  26. public:
  27. qt_sink(QObject *qt_object, std::string meta_method)
  28. : qt_object_(qt_object),
  29. meta_method_(std::move(meta_method)) {
  30. if (!qt_object_) {
  31. throw_spdlog_ex("qt_sink: qt_object is null");
  32. }
  33. }
  34. ~qt_sink() { flush_(); }
  35. protected:
  36. void sink_it_(const details::log_msg &msg) override {
  37. memory_buf_t formatted;
  38. base_sink<Mutex>::formatter_->format(msg, formatted);
  39. const string_view_t str = string_view_t(formatted.data(), formatted.size());
  40. QMetaObject::invokeMethod(
  41. qt_object_, meta_method_.c_str(), Qt::AutoConnection,
  42. Q_ARG(QString, QString::fromUtf8(str.data(), static_cast<int>(str.size())).trimmed()));
  43. }
  44. void flush_() override {}
  45. private:
  46. QObject *qt_object_ = nullptr;
  47. std::string meta_method_;
  48. };
  49. // Qt color sink to QTextEdit.
  50. // Color location is determined by the sink log pattern like in the rest of spdlog sinks.
  51. // Colors can be modified if needed using sink->set_color(level, qtTextCharFormat).
  52. // max_lines is the maximum number of lines that the sink will hold before removing the oldest
  53. // lines. By default, only ascii (latin1) is supported by this sink. Set is_utf8 to true if utf8
  54. // support is needed.
  55. template <typename Mutex>
  56. class qt_color_sink : public base_sink<Mutex> {
  57. public:
  58. qt_color_sink(QTextEdit *qt_text_edit,
  59. int max_lines,
  60. bool dark_colors = false,
  61. bool is_utf8 = false)
  62. : qt_text_edit_(qt_text_edit),
  63. max_lines_(max_lines),
  64. is_utf8_(is_utf8) {
  65. if (!qt_text_edit_) {
  66. throw_spdlog_ex("qt_color_text_sink: text_edit is null");
  67. }
  68. default_color_ = qt_text_edit_->currentCharFormat();
  69. // set colors
  70. QTextCharFormat format;
  71. // trace
  72. format.setForeground(dark_colors ? Qt::darkGray : Qt::gray);
  73. colors_.at(level::trace) = format;
  74. // debug
  75. format.setForeground(dark_colors ? Qt::darkCyan : Qt::cyan);
  76. colors_.at(level::debug) = format;
  77. // info
  78. format.setForeground(dark_colors ? Qt::darkGreen : Qt::green);
  79. colors_.at(level::info) = format;
  80. // warn
  81. format.setForeground(dark_colors ? Qt::darkYellow : Qt::yellow);
  82. colors_.at(level::warn) = format;
  83. // err
  84. format.setForeground(Qt::red);
  85. colors_.at(level::err) = format;
  86. // critical
  87. format.setForeground(Qt::white);
  88. format.setBackground(Qt::red);
  89. colors_.at(level::critical) = format;
  90. }
  91. ~qt_color_sink() { flush_(); }
  92. void set_default_color(QTextCharFormat format) {
  93. // std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  94. default_color_ = format;
  95. }
  96. void set_level_color(level::level_enum color_level, QTextCharFormat format) {
  97. // std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  98. colors_.at(static_cast<size_t>(color_level)) = format;
  99. }
  100. QTextCharFormat &get_level_color(level::level_enum color_level) {
  101. std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  102. return colors_.at(static_cast<size_t>(color_level));
  103. }
  104. QTextCharFormat &get_default_color() {
  105. std::lock_guard<Mutex> lock(base_sink<Mutex>::mutex_);
  106. return default_color_;
  107. }
  108. protected:
  109. struct invoke_params {
  110. invoke_params(int max_lines,
  111. QTextEdit *q_text_edit,
  112. QString payload,
  113. QTextCharFormat default_color,
  114. QTextCharFormat level_color,
  115. int color_range_start,
  116. int color_range_end)
  117. : max_lines(max_lines),
  118. q_text_edit(q_text_edit),
  119. payload(std::move(payload)),
  120. default_color(default_color),
  121. level_color(level_color),
  122. color_range_start(color_range_start),
  123. color_range_end(color_range_end) {}
  124. int max_lines;
  125. QTextEdit *q_text_edit;
  126. QString payload;
  127. QTextCharFormat default_color;
  128. QTextCharFormat level_color;
  129. int color_range_start;
  130. int color_range_end;
  131. };
  132. void sink_it_(const details::log_msg &msg) override {
  133. memory_buf_t formatted;
  134. base_sink<Mutex>::formatter_->format(msg, formatted);
  135. const string_view_t str = string_view_t(formatted.data(), formatted.size());
  136. // apply the color to the color range in the formatted message.
  137. QString payload;
  138. int color_range_start = static_cast<int>(msg.color_range_start);
  139. int color_range_end = static_cast<int>(msg.color_range_end);
  140. if (is_utf8_) {
  141. payload = QString::fromUtf8(str.data(), static_cast<int>(str.size()));
  142. // convert color ranges from byte index to character index.
  143. if (msg.color_range_start < msg.color_range_end) {
  144. color_range_start = QString::fromUtf8(str.data(), msg.color_range_start).size();
  145. color_range_end = QString::fromUtf8(str.data(), msg.color_range_end).size();
  146. }
  147. } else {
  148. payload = QString::fromLatin1(str.data(), static_cast<int>(str.size()));
  149. }
  150. invoke_params params{max_lines_, // max lines
  151. qt_text_edit_, // text edit to append to
  152. std::move(payload), // text to append
  153. default_color_, // default color
  154. colors_.at(msg.level), // color to apply
  155. color_range_start, // color range start
  156. color_range_end}; // color range end
  157. QMetaObject::invokeMethod(
  158. qt_text_edit_, [params]() { invoke_method_(params); }, Qt::AutoConnection);
  159. }
  160. void flush_() override {}
  161. // Add colored text to the text edit widget. This method is invoked in the GUI thread.
  162. // It is a static method to ensure that it is handled correctly even if the sink is destroyed
  163. // prematurely before it is invoked.
  164. static void invoke_method_(invoke_params params) {
  165. auto *document = params.q_text_edit->document();
  166. QTextCursor cursor(document);
  167. // remove first blocks if number of blocks exceeds max_lines
  168. while (document->blockCount() > params.max_lines) {
  169. cursor.select(QTextCursor::BlockUnderCursor);
  170. cursor.removeSelectedText();
  171. cursor.deleteChar(); // delete the newline after the block
  172. }
  173. cursor.movePosition(QTextCursor::End);
  174. cursor.setCharFormat(params.default_color);
  175. // if color range not specified or not not valid, just append the text with default color
  176. if (params.color_range_end <= params.color_range_start) {
  177. cursor.insertText(params.payload);
  178. return;
  179. }
  180. // insert the text before the color range
  181. cursor.insertText(params.payload.left(params.color_range_start));
  182. // insert the colorized text
  183. cursor.setCharFormat(params.level_color);
  184. cursor.insertText(params.payload.mid(params.color_range_start,
  185. params.color_range_end - params.color_range_start));
  186. // insert the text after the color range with default format
  187. cursor.setCharFormat(params.default_color);
  188. cursor.insertText(params.payload.mid(params.color_range_end));
  189. }
  190. QTextEdit *qt_text_edit_;
  191. int max_lines_;
  192. bool is_utf8_;
  193. QTextCharFormat default_color_;
  194. std::array<QTextCharFormat, level::n_levels> colors_;
  195. };
  196. #include "spdlog/details/null_mutex.h"
  197. #include <mutex>
  198. using qt_sink_mt = qt_sink<std::mutex>;
  199. using qt_sink_st = qt_sink<details::null_mutex>;
  200. using qt_color_sink_mt = qt_color_sink<std::mutex>;
  201. using qt_color_sink_st = qt_color_sink<details::null_mutex>;
  202. } // namespace sinks
  203. //
  204. // Factory functions
  205. //
  206. // log to QTextEdit
  207. template <typename Factory = spdlog::synchronous_factory>
  208. inline std::shared_ptr<logger> qt_logger_mt(const std::string &logger_name,
  209. QTextEdit *qt_object,
  210. const std::string &meta_method = "append") {
  211. return Factory::template create<sinks::qt_sink_mt>(logger_name, qt_object, meta_method);
  212. }
  213. template <typename Factory = spdlog::synchronous_factory>
  214. inline std::shared_ptr<logger> qt_logger_st(const std::string &logger_name,
  215. QTextEdit *qt_object,
  216. const std::string &meta_method = "append") {
  217. return Factory::template create<sinks::qt_sink_st>(logger_name, qt_object, meta_method);
  218. }
  219. // log to QPlainTextEdit
  220. template <typename Factory = spdlog::synchronous_factory>
  221. inline std::shared_ptr<logger> qt_logger_mt(const std::string &logger_name,
  222. QPlainTextEdit *qt_object,
  223. const std::string &meta_method = "appendPlainText") {
  224. return Factory::template create<sinks::qt_sink_mt>(logger_name, qt_object, meta_method);
  225. }
  226. template <typename Factory = spdlog::synchronous_factory>
  227. inline std::shared_ptr<logger> qt_logger_st(const std::string &logger_name,
  228. QPlainTextEdit *qt_object,
  229. const std::string &meta_method = "appendPlainText") {
  230. return Factory::template create<sinks::qt_sink_st>(logger_name, qt_object, meta_method);
  231. }
  232. // log to QObject
  233. template <typename Factory = spdlog::synchronous_factory>
  234. inline std::shared_ptr<logger> qt_logger_mt(const std::string &logger_name,
  235. QObject *qt_object,
  236. const std::string &meta_method) {
  237. return Factory::template create<sinks::qt_sink_mt>(logger_name, qt_object, meta_method);
  238. }
  239. template <typename Factory = spdlog::synchronous_factory>
  240. inline std::shared_ptr<logger> qt_logger_st(const std::string &logger_name,
  241. QObject *qt_object,
  242. const std::string &meta_method) {
  243. return Factory::template create<sinks::qt_sink_st>(logger_name, qt_object, meta_method);
  244. }
  245. // log to QTextEdit with colorized output
  246. template <typename Factory = spdlog::synchronous_factory>
  247. inline std::shared_ptr<logger> qt_color_logger_mt(const std::string &logger_name,
  248. QTextEdit *qt_text_edit,
  249. int max_lines,
  250. bool is_utf8 = false) {
  251. return Factory::template create<sinks::qt_color_sink_mt>(logger_name, qt_text_edit, max_lines,
  252. false, is_utf8);
  253. }
  254. template <typename Factory = spdlog::synchronous_factory>
  255. inline std::shared_ptr<logger> qt_color_logger_st(const std::string &logger_name,
  256. QTextEdit *qt_text_edit,
  257. int max_lines,
  258. bool is_utf8 = false) {
  259. return Factory::template create<sinks::qt_color_sink_st>(logger_name, qt_text_edit, max_lines,
  260. false, is_utf8);
  261. }
  262. } // namespace spdlog