apr1-gatherv.c

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

int main(int argc, char** argv) {
    int rc;
    rc = MPI_Init(&argc, &argv);
    L17_CHECK_ERROR(MPI_Init, rc);

    int size;
    rc = MPI_Comm_size(MPI_COMM_WORLD, &size);
    L17_CHECK_ERROR(MPI_Comm_size, rc);

    int rank;
    rc = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    L17_CHECK_ERROR(MPI_Comm_rank, rc);

    int* sendbuf;
    int sendcount = rank + 1;
    sendbuf = (int*) malloc(sendcount * sizeof(int));
    if (sendbuf == NULL) {
        printf("Error. Couldn't allocate memory.");
        MPI_Abort(MPI_COMM_WORLD, 1);
    }

    int i;
    for (i = 0; i < sendcount; ++i)
        sendbuf[i] = rank + 1;

    int* recvcounts;
    if (rank == 0) {
        recvcounts = (int*) malloc(size * sizeof(int));
        if (recvcounts == NULL) {
            printf("Error. Couldn't allocate memory.");
            MPI_Abort(MPI_COMM_WORLD, 3);
        }

        for (i = 0; i < size; ++i)
            recvcounts[i] = i + 1;
    }

    int* displs;
    if (rank == 0) {
        displs = (int*) malloc(size * sizeof(int));
        if (displs == NULL) {
            printf("Error. Couldn't allocate memory.");
            MPI_Abort(MPI_COMM_WORLD, 4);
        }

        displs[0] = 0;
        for (i = 1; i < size; ++i)
            displs[i] = displs[i - 1] + recvcounts[i - 1] + i;
    }

    int* recvbuf;
    int recvbufsize;

    if (rank == 0) {
        recvbufsize = displs[size - 1] + size;
        recvbuf = (int*) malloc(recvbufsize * sizeof(int));
        if (recvbuf == NULL) {
            printf("Error. Couldn't allocate memory.");
            MPI_Abort(MPI_COMM_WORLD, 2);
        }

        for (i = 0; i < recvbufsize; ++i)
            recvbuf[i] = 0;
    }

    rc = MPI_Gatherv(sendbuf, sendcount, MPI_INT, recvbuf, recvcounts, displs, MPI_INT, 0, MPI_COMM_WORLD);
    L17_CHECK_ERROR(MIP_Gatherv, rc);

    free(sendbuf);

    if (rank == 0) {
        free(recvcounts);
        free(displs);

        FILE* outputp = fopen("apr1-gatherv.dat", "w");
        if (outputp == NULL) {
            printf("Error. Couldn't open the file for output values.");
            MPI_Abort(MPI_COMM_WORLD, 5);
        }

        for (i = 0; i < recvbufsize; ++i)
            fprintf(outputp, "%d ", recvbuf[i]);
        fprintf(outputp, "\n");

        fclose(outputp);
        free(recvbuf);
    }

    MPI_Finalize();

    return 0;
}