complex_printer.hpp

浏览该文件的文档.

/*
 * @Description: 复杂打印器类 头文件及其实现
 * @Version:
 * @Autor: dreamy-xay
 * @date: 2024-01-23
 * @LastEditors: dreamy-xay
 * @LastEditTime: 2024-03-04
 */
 
#ifndef COMMON_COMPLEX_PRINTER_HPP
#define COMMON_COMPLEX_PRINTER_HPP
 
#include <cstring>
#include <iostream>
 
#include "common/debug/color.hpp"
#include "common/debug/printer.hpp"
#include "common/enum.h"
#include "common/macro.h"
#include "common/type.h"
#include "common/utils/string.hpp"
 
namespace cm {
 
class ComplexPrinter : public Printer {
   private:
    std::string param_name_output_color;        // 默认参数名输出的颜色
    std::string param_name_output_color_cache;  // 默认参数名输出的颜色缓存
 
    std::vector<std::string> param_names;  // 参数名列表
    int current_param_index;               // 当前打印的参数名索引
 
   public:
    ComplexPrinter(const std::string& default_split_string = "\n", const std::string& output_color = Color::BLUE, const std::string& param_name_output_color = Color::GREEN + Color::FONT_BOLD);
    ~ComplexPrinter();
 
    ComplexPrinter& ParseParamNames(std::string param_names_str);
 
    template <typename... ARGS>
    std::ostream& operator()(const std::string& param_names_str, ARGS... args);
 
    ComplexPrinter& SetParamNameOutputColor(const std::string& param_name_output_color);
 
   private:
    std::string GetPrintPrefix(bool is_last_arg) override;
};
 
inline ComplexPrinter::ComplexPrinter(const std::string& split_string, const std::string& output_color, const std::string& param_name_output_color) : Printer(split_string, output_color), param_name_output_color(param_name_output_color), param_names(), current_param_index(-1) {
}
 
inline ComplexPrinter::~ComplexPrinter() {}
 
inline ComplexPrinter& ComplexPrinter::ParseParamNames(std::string param_names_str) {
    param_names.clear();             // 清空参数名字列表
    param_names_str.push_back(',');  // 向最后添加逗号
 
    int is_not_split = 0;                   // 是否支持分隔符
    size_t len = param_names_str.length();  // 参数名子列表字符串长度
    size_t prev_pos = 0;                    // 存储上一个逗号的位置的后一个索引
    std::string token;                      // 存储每个参数名
 
    // 获取参数名列表（通过逗号分割）
    for (size_t i = 0; i < len; ++i)
        if (param_names_str[i] == '(')
            ++is_not_split;
        else if (param_names_str[i] == ')')
            --is_not_split;
        else if (param_names_str[i] == ',' && !is_not_split) {
            token = param_names_str.substr(prev_pos, i - prev_pos);
            prev_pos = i + 1;
 
            size_t first = token.find_first_not_of(" \t\n\r");  // 删除参数名首尾空格和回车
            size_t last = token.find_last_not_of(" \t\n\r");
 
            if (first == std::string::npos)  // 如果字符串由回车或空格组成，则不动源字符串
                param_names.emplace_back("<" + token + ">");
            else
                param_names.emplace_back("<" + token.substr(first, last - first + 1) + ">");  // 删除参数名首尾空格和回车
        }
 
    current_param_index = 0;  // 初始化当前打印的参数索引
 
    return *this;
}
 
template <typename... ARGS>
inline std::ostream& ComplexPrinter::operator()(const std::string& param_names_str, ARGS... args) {
    this->ParseParamNames(param_names_str);  // 解析所有参数名
    this->Print(args...);                    // 打印所有参数
 
    return std::cout;
}
 
inline ComplexPrinter& ComplexPrinter::SetParamNameOutputColor(const std::string& param_name_output_color) {
    this->param_name_output_color = param_name_output_color;
 
    return *this;
}
 
inline std::string ComplexPrinter::GetPrintPrefix(bool is_last_arg) {
    Cm_Assert(current_param_index >= 0 && param_names.size() > current_param_index, "Parameter name list string data error!!!");
 
    // 如果撤销了颜色打印，则特殊处理
    if (is_cancel_color) {
        param_name_output_color_cache = param_name_output_color;
        param_name_output_color = "";
    }
 
    // 新的前缀字符串
    std::string prefix;
 
    prefix += param_name_output_color;
    prefix += param_names[current_param_index++];
    prefix += ": ";
    prefix += Color::RESET;
 
    // 如果是最后一个参数
    if (is_last_arg) {
        current_param_index = -1;
 
        // 如果撤销了颜色打印，则恢复颜色打印
        if (is_cancel_color)
            param_name_output_color = param_name_output_color_cache;
    }
 
    return std::move(prefix);
}
 
}  // namespace cm
 
#endif