ASCII格式的MPI并行IO(我该怎么做?)

2022-03-09 00:00:00 ascii parallel-processing io mpi c++

我知道MPI IO以二进制格式写入文件,但这一次我需要它采用ASCII格式。我需要这个,因为我对图像使用Netpbm,并且标题必须是ASCII(这样程序就可以读取P1、P2、P3、P4、P5或P6,并确定图像的内容是ASCII还是二进制等等)。

我使用的代码是(simple example) MPI parallel io writing garbage中提供的代码,我想知道代码中是否有一些小改动,允许我以ASCII格式而不是二进制格式输出整数缓冲区的内容。

此外,有一种以ASCII格式输出的方式也将使调试变得更容易。


解决方案

可以使用strlen作为缓冲区大小

#include <mpi.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

using namespace std;

int main(int argc,char *argv[])
{

    int  namelen, numprocs, rank = 7;
    char processor_name[MPI_MAX_PROCESSOR_NAME];
    MPI_Init(&argc,&argv);
    MPI_Comm_rank(MPI_COMM_WORLD,&rank);
    MPI_Comm_size(MPI_COMM_WORLD,&numprocs);    
    MPI_Get_processor_name(processor_name,&namelen);
    MPI_Status status;
    MPI_File fh;
    string test;
    const char* buf[100]; 
    string g = "test" + to_string(rank) + ".txt";
    MPI_File_open(MPI_COMM_SELF, g.c_str(), MPI_MODE_CREATE | MPI_MODE_WRONLY,MPI_INFO_NULL,&fh);
    for (int i=0; i < 100; i++){
        test = to_string(i) + "
";
        buf[i] = test.c_str();
        MPI_File_write(fh, buf[i], strlen(buf[i]), MPI_CHAR, &status);
    }
    MPI_File_close(&fh);
    MPI_Finalize();
    return 0;
}

请注意MPI_File_write(fh, buf[i], strlen(buf[i]), MPI_CHAR, &status);

希望这会有帮助。

相关文章