如何正确使用VTK并行XML Writer输出并行文件？

你目前的问题核心在于仅用根进程的子域网格作为并行Writer输入，导致全局范围错误，且默认逻辑会触发额外分片文件生成。下面分步解决：

一、错误原因拆解 #

• 全局范围错误：你给pwriter->SetInputData(grid)传入的是rank0的子域网格，VTK会默认用该网格范围作为整个并行域的范围，而非全局真实范围。
• 额外.vts文件生成：vtkXMLPStructuredGridWriter默认逻辑会尝试写入分片数据，除非明确阻止，因为它原本设计是同时负责写分片和并行元文件。

二、修正后的代码实现 #

要实现「各rank的vts在单独文件夹，pvts在父目录」的需求，需完成以下操作：

1. 所有进程同步各自子域范围，由根进程计算全局网格范围
2. 自定义并行Writer，仅生成.pvts元文件，不写入分片数据
3. 正确设置分片的Source路径

#include <vtkNew.h>
#include <vtkStructuredGrid.h>
#include <vtkXMLStructuredGridWriter.h>
#include <vtkXMLPStructuredGridWriter.h>
#include <mpi.h>
#include <algorithm>

// 自定义并行Writer：仅生成pvts元文件，不写分片数据
class MyPwriter : public vtkXMLPStructuredGridWriter {
public:
    static MyPwriter* New();
    vtkTypeMacro(MyPwriter, vtkXMLPStructuredGridWriter);

protected:
    // 重写WritePiece，空实现避免生成额外vts
    void WritePiece(int vtkNotUsed(index)) override {}

    // 自定义每个分片的Source路径
    void WritePPieceAttributes(int index) override {
        std::string filename = std::to_string(index) + "/test_1.vts";
        this->WriteStringAttribute("Source", filename.c_str());
    }
};

vtkStandardNewMacro(MyPwriter);

int main(int argc, char* argv[]) {
    MPI_Init(&argc, &argv);
    int myrank, nproc;
    MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
    MPI_Comm_size(MPI_COMM_WORLD, &nproc);

    // 1. 生成并写入当前rank的子域vts（原代码逻辑保留）
    vtkNew<vtkStructuredGrid> grid;
    // ... 填充网格数据（此处省略具体逻辑）
    int subExtent[6];
    grid->GetExtent(subExtent); // 获取当前子域的范围

    vtkNew<vtkXMLStructuredGridWriter> writer;
    writer->SetInputData(grid);
    std::string fname = std::to_string(myrank) + "/test_1.vts";
    writer->SetFileName(fname.c_str());
    writer->Write();

    // 2. 同步所有子域范围，计算全局范围（仅根进程处理）
    int globalExtent[6] = {0,0,0,0,0,0};
    if (myrank == 0) {
        // 收集所有进程的子域范围
        int* allExtents = new int[6 * nproc];
        MPI_Gather(subExtent, 6, MPI_INT, allExtents, 6, MPI_INT, 0, MPI_COMM_WORLD);
        
        // 计算全局范围：取所有子域的最小/最大边界
        globalExtent[0] = allExtents[0];
        globalExtent[1] = allExtents[1];
        globalExtent[2] = allExtents[2];
        globalExtent[3] = allExtents[3];
        globalExtent[4] = allExtents[4];
        globalExtent[5] = allExtents[5];
        for (int i = 1; i < nproc; ++i) {
            int* ext = allExtents + 6*i;
            globalExtent[0] = std::min(globalExtent[0], ext[0]);
            globalExtent[1] = std::max(globalExtent[1], ext[1]);
            globalExtent[2] = std::min(globalExtent[2], ext[2]);
            globalExtent[3] = std::max(globalExtent[3], ext[3]);
            globalExtent[4] = std::min(globalExtent[4], ext[4]);
            globalExtent[5] = std::max(globalExtent[5], ext[5]);
        }
        delete[] allExtents;
    } else {
        // 非根进程发送自己的子域范围到根进程
        MPI_Gather(subExtent, 6, MPI_INT, nullptr, 0, MPI_INT, 0, MPI_COMM_WORLD);
    }

    // 3. 根进程生成pvts文件
    if (myrank == 0) {
        vtkNew<MyPwriter> pwriter;
        pwriter->SetFileName("test_1.pvts");
        pwriter->SetNumberOfPieces(nproc);
        // 设置全局网格范围，这是核心修正点
        pwriter->SetInputExtent(globalExtent);
        // 禁用自动写入分片数据（已手动完成）
        pwriter->SetWritePiece(0);
        pwriter->Write();
    }

    MPI_Finalize();
    return 0;
}

三、并行Writer工作逻辑答疑 #

1. 仅根进程调用时如何获取全局范围？
   VTK并行Writer不会自动推导全局范围，必须手动设置（如代码中通过MPI收集所有子域范围后计算）。若不设置，会默认使用输入数据的范围，也就是你之前传入的rank0子域范围。

2. 多进程并行调用时如何避免文件冲突？
   若选择所有进程调用并行Writer，VTK内部会通过MPI同步：每个进程仅负责写入自己的分片数据，根进程负责生成.pvts元文件，内部会处理文件锁和同步逻辑，不会出现并发写冲突。但更高效的方式是像你当前的设计：各进程单独写.vts，根进程仅生成.pvts元文件，减少MPI通信开销。

四、关键注意事项 #

• 必须确保全局范围的正确性，否则ParaView加载时会出现分片拼接错误。
• 自定义Writer时重写WritePiece为空实现，或设置SetWritePiece(0)，避免并行Writer生成额外分片文件。
• 分片的Source路径需保证相对.pvts文件的路径正确（如rank/test_1.vts是相对父目录的路径，ParaView可正确解析）。

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