如何为2阶std::mdspan实现编译期矩阵格式化？

问题根源 #

你的代码中"{:" + std::string{spec} + "}"是运行时动态构造的字符串，但std::format_to要求格式字符串必须是编译期常量表达式，这直接导致编译失败。要解决这个问题，必须借助元素类型自身的std::formatter来处理格式规格，而非手动拼接格式字符串。

正确实现方案 #

我们可以在std::mdspan的formatter中内嵌一个元素类型的formatter实例，将格式规格直接转发给它，这样既复用了标准库的格式化逻辑，又满足编译期要求：

#include <print>
#include <mdspan>
#include <format>
#include <iterator>

template<class T, class Extents, class Layout, class Accessor>
requires (Extents::rank() == 2)
struct std::formatter<std::mdspan<T, Extents, Layout, Accessor>> {
    // 内嵌元素类型的formatter，负责处理单个元素的格式化规则
    std::formatter<T> elem_formatter;

    // 转发parse逻辑给元素formatter，确保格式规格在编译期被解析
    constexpr auto parse(format_parse_context& ctx) {
        return elem_formatter.parse(ctx);
    }

    auto format(const std::mdspan<T, Extents, Layout, Accessor>& md, format_context& ctx) const {
        auto out = ctx.out();
        *out++ = '[';

        for (size_t i = 0; i < md.extent(0); ++i) {
            if (i != 0) {
                out = std::format_to(out, "\n ");
            }
            *out++ = '[';

            for (size_t j = 0; j < md.extent(1); ++j) {
                if (j != 0) {
                    *out++ = ' ';
                }
                // 使用元素formatter格式化每个元素，自动应用传入的格式规格
                out = elem_formatter.format(md[i, j], ctx);
            }

            *out++ = ']';
        }

        *out++ = ']';
        return out;
    }
};

int main() {
    double arr[] = {1.1111, 2.2222, 3.3333, 4.4444};
    std::mdspan mat(arr, 2, 2);

    // 测试默认格式
    std::println("默认格式:\n{}", mat);
    // 测试科学计数法格式
    std::println("\n科学计数法格式(.2e):\n{:.2e}", mat);
    // 测试固定精度格式
    std::println("\n固定精度格式(.3f):\n{:.3f}", mat);
}

代码解释 #

1. 内嵌元素formatter：std::formatter<T>已经实现了对应类型的格式化逻辑（比如double的.2e、.3f等规格），直接复用它可以避免重复实现格式解析，同时保证编译期合规。
2. 转发parse函数：将格式上下文直接传给元素formatter的parse方法，让它负责解析格式规格，这样所有标准支持的格式都能直接生效，无需手动处理。
3. 格式化矩阵：遍历矩阵的行和列，用元素formatter格式化每个元素，同时处理矩阵的括号、换行和空格分隔，保证输出是清晰的矩阵形式。

通用格式化技巧 #

• 复用已有formatter：自定义formatter时，优先考虑复用标准库或已有类型的formatter，既减少代码量，又能保证兼容性和编译期正确性。
• 严格遵循format上下文规则：format_context和format_parse_context的使用必须符合标准要求，parse函数必须是constexpr（C++20及以后），format函数要正确返回输出迭代器。
• 避免动态格式字符串：永远不要在format函数中动态构造格式字符串，必须通过formatter的parse阶段提前解析格式规则，确保格式字符串是编译期常量。

内容的提问来源于stack exchange，提问作者olafwx