使用MPI IO将数组追加到现有文件

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

#define ARRAY_SIZE 10

int main(int argc, char** argv) {
    int rank, size;
    int* array;
    MPI_File file;
    MPI_Datatype array_type;

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    // 打开现有文件并创建MPI文件句柄
    MPI_File_open(MPI_COMM_WORLD, "data.txt", MPI_MODE_WRONLY | MPI_MODE_APPEND, MPI_INFO_NULL, &file);

    // 设置文件偏移量
    MPI_File_set_view(file, 0, MPI_INT, MPI_INT, "native", MPI_INFO_NULL);

    // 定义数组的数据类型
    MPI_Type_contiguous(ARRAY_SIZE, MPI_INT, &array_type);
    MPI_Type_commit(&array_type);

    // 分配内存
    array = (int*)malloc(sizeof(int) * ARRAY_SIZE);

    // 写入数据
    for (int i = 0; i < ARRAY_SIZE; i++) {
        array[i] = rank * ARRAY_SIZE + i;
    }
    MPI_File_write_all(file, array, 1, array_type, MPI_STATUS_IGNORE);

    // 释放内存
    free(array);

    // 关闭MPI文件句柄
    MPI_File_close(&file);

    MPI_Finalize();
    return 0;
}