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

/**
 * @author cristian.chilipirea
 * Run: mpirun -np 4 ./a.out
 */

int mat[15][15] =  {{0,1,1,0,0,0,0,0,0,0,0,0,0,0,0},
					{1,0,0,1,1,1,0,0,0,0,0,0,0,0,0},
					{1,0,0,0,0,0,1,1,0,0,0,0,0,0,0},
					{0,1,0,0,0,0,0,0,1,0,0,0,0,0,0},
					{0,1,0,0,0,0,0,0,0,0,0,0,0,0,0},
					{0,1,0,0,0,0,0,0,0,1,1,0,0,0,0},
					{0,0,1,0,0,0,0,0,0,0,0,0,0,0,0},
					{0,0,1,0,0,0,0,0,0,0,0,1,1,0,0},
					{0,0,0,1,0,0,0,0,0,0,0,0,0,0,0},
					{0,0,0,0,0,1,0,0,0,0,0,0,0,1,0},
					{0,0,0,0,0,1,0,0,0,0,0,0,0,0,0},
					{0,0,0,0,0,0,0,1,0,0,0,0,0,0,1},
					{0,0,0,0,0,0,0,1,0,0,0,0,0,0,0},
					{0,0,0,0,0,0,0,0,0,1,0,0,0,0,0},
					{0,0,0,0,0,0,0,0,0,0,0,1,0,0,0}
					};

int main(int argc, char * argv[]) {
	int rank;
	int nProcesses;
	int level, tag = 1, i, j, k, source;
	int recvmat[15][15];

	MPI_Init(&argc, &argv);
	MPI_Status status;
	MPI_Request request;

	MPI_Comm_rank(MPI_COMM_WORLD, &rank);
	MPI_Comm_size(MPI_COMM_WORLD, &nProcesses);

	for (i = 0 ; i < 15; i++) {
		if (i != rank) {
			for (j = 0; j < 15; j++) {
				mat[i][j] = 0;
			}
		}
	}

	if (rank == 0) {
		level = 0;

		MPI_Send(&level, 1, MPI_INT, 1, tag, MPI_COMM_WORLD);
		MPI_Send(&level, 1, MPI_INT, 2, tag, MPI_COMM_WORLD);

		// tre' sa primeasca

		MPI_Recv(&recvmat, 225, MPI_INT, 1, tag, MPI_COMM_WORLD, &status);
		for (j = 0; j < 15; j++) {
			for (k =0 ; k < 15; k++) {
				if (recvmat[j][k] == 1) {
					mat[j][k] = 1;
				}
			}
		}

		MPI_Recv(&recvmat, 225, MPI_INT, 2, tag, MPI_COMM_WORLD, &status);
		for (j = 0; j < 15; j++) {
			for (k =0 ; k < 15; k++) {
				if (recvmat[j][k] == 1) {
					mat[j][k] = 1;
				}
			}
		}

		for (j = 0; j < 15; j++) {
			for (k =0 ; k < 15; k++) {
				printf("%d ", mat[j][k]);
			}
			printf("\n");
		}
	}
	else {
		MPI_Recv(&level, 1, MPI_INT, MPI_ANY_SOURCE, tag, MPI_COMM_WORLD, &status);
		source = status.MPI_SOURCE;
		printf("%d a primit nivelul %d de la %d\n", rank, level, source);

		level++;

		for (i = 0 ; i < 15; i++) {
			if (mat[rank][i] != 0 && i != source) {
				MPI_Send(&level, 1, MPI_INT, i, tag, MPI_COMM_WORLD);
			}
		}

		for (i = 0 ; i < 15; i++) {
			if (mat[rank][i] != 0 && i != source) {
				MPI_Recv(&recvmat, 225, MPI_INT, i, tag, MPI_COMM_WORLD, &status);

				for (j = 0; j < 15; j++) {
					for (k =0 ; k < 15; k++) {
						if (recvmat[j][k] == 1) {
							mat[j][k] = 1;
						}
					}

				}
			}
		}
		MPI_Send(&mat, 225, MPI_INT, source, tag, MPI_COMM_WORLD);
	}

	MPI_Finalize();
	return 0;
}