能否用C++函数或宏将另一函数的代码写入文件？

当然可以实现这个需求！针对你想把函数/测试用例的代码写入文件、自动更新文档示例的场景，我整理了几种实用的方案：

方案1：纯C++预处理器宏实现（无需额外工具） #

这种方案利用C++预处理器的字符串化能力，直接在定义函数时捕获其源码，不需要额外的构建步骤。核心思路是用宏包裹目标函数，同时生成一个能返回该函数代码字符串的辅助函数。

#include <fstream>
#include <string>

// 辅助宏：处理字符串化的嵌套展开问题
#define STRINGIFY(x) #x
#define EXPAND_AND_STRINGIFY(x) STRINGIFY(x)

// 宏：定义目标函数，并自动生成获取其代码的函数
#define DEFINE_WRITABLE_FUNCTION(func_def) \
    func_def \
    std::string get_##func_def##_code() { \
        // 用原始字符串字面量避免转义问题（C++11及以上支持）
        return EXPAND_AND_STRINGIFY(func_def); \
    }

// 用宏定义add函数，同时生成get_add_code()
DEFINE_WRITABLE_FUNCTION(int add(int a, int b) { return a + b; })

// 通用写入文件函数
void writeCodeToFile(const std::string& code, const std::string& fname) {
    std::ofstream outFile(fname);
    if (outFile.is_open()) {
        outFile << code;
        outFile.close();
    }
}

// 你需要的写入add函数的接口
void functionThatWritesAddToAFile(std::string fname) {
    writeCodeToFile(get_add_code(), fname);
}

优缺点 #

• ✅ 纯C++实现，不需要额外工具或构建步骤
• ❌ 对复杂函数（包含多行注释、特殊字符、嵌套宏）的处理可能出现格式问题
• ❌ 必须用指定宏包裹函数定义，侵入性较强

方案2：构建脚本辅助（更适合复杂场景） #

如果你的函数或测试用例结构复杂，推荐用构建脚本（比如Python）提前提取目标代码，生成包含代码字符串的头文件，再在C++程序中引用。这种方式更稳定，也不需要修改原函数的定义方式。

步骤1：编写Python提取脚本 #

import re

# 读取你的源文件
source_path = "your_source.cpp"
output_header = "function_codes.h"

with open(source_path, "r", encoding="utf-8") as f:
    content = f.read()

# 用正则匹配add函数（可根据实际函数结构调整正则表达式）
# 注意：复杂函数需要更精细的正则，比如匹配函数体的大括号范围
func_pattern = re.compile(r'int add\(int a, int b\)\s*\{\s*.*?\s*\}', re.DOTALL)
match_result = func_pattern.search(content)

if match_result:
    func_code = match_result.group(0)
    # 生成头文件，用原始字符串字面量存储代码
    with open(output_header, "w", encoding="utf-8") as f:
        f.write(f"""\
#pragma once
#include <string>

const std::string add_function_code = R"RAW_CODE({func_code})RAW_CODE";
""")
else:
    print(f"未找到add函数定义")

步骤2：在C++中引用并写入文件 #

#include "function_codes.h"
#include <fstream>

void functionThatWritesAddToAFile(std::string fname) {
    std::ofstream out_file(fname);
    if (out_file.is_open()) {
        out_file << add_function_code;
        out_file.close();
    }
}

优缺点 #

• ✅ 支持复杂函数、多行代码、特殊字符，格式还原准确
• ✅ 不侵入原代码结构，原函数可以正常编写
• ❌ 需要在构建流程中添加脚本执行步骤（比如在CMake中添加自定义命令）

方案3：结合GoogleTest的适配 #

如果你的示例是GoogleTest测试用例，同样可以用类似的思路处理：要么用宏包裹测试用例捕获代码，要么用脚本提取测试用例的源码。

宏包裹测试用例的示例 #

#include <gtest/gtest.h>
#include <string>

#define WRITABLE_TEST(test_case, test_name) \
    TEST(test_case, test_name) \
    std::string get_##test_case##_##test_name##_code() { \
        return EXPAND_AND_STRINGIFY( \
            TEST(test_case, test_name) { \
                // 这里是测试代码 \
            } \
        ); \
    }

// 定义可写入的测试用例
WRITABLE_TEST(AdditionTest, TwoPositiveNumbers) {
    EXPECT_EQ(add(2, 3), 5);
    EXPECT_EQ(add(10, 20), 30);
}

// 测试运行后写入文件
void writeTestCodeToFile(std::string fname) {
    std::ofstream out(fname);
    if (out.is_open()) {
        out << get_AdditionTest_TwoPositiveNumbers_code();
        out.close();
    }
}

总结建议 #

• 简单函数或快速原型：用方案1的宏实现
• 复杂函数、测试用例或长期维护的项目：优先方案2的脚本辅助方式，更稳定可靠
• 结合Sphinx的literalinclude时，建议将生成的代码文件放在Sphinx能访问到的目录下，确保文档构建时能正确读取

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