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

#include "mpi.h"

#include "mpi_bnc.h"

#define DIM 5

int main(int argc, char *argv[])
{
	long int i, j, local_dim;
	DMatrix x;
	DMatrix local_x[10];
	DVector vc, vb;
	DVector local_vc, local_vb, local_vtmp;

	long int d_dim[10];
	int myrank, num_procs;

	MPI_Init(&argc, &argv);
	MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
	MPI_Comm_size(MPI_COMM_WORLD, &num_procs);

	local_dim = _mpi_divide_dim(d_dim, DIM, num_procs);
	if(myrank == 0)
	{
		x = init_dmatrix(num_procs * local_dim, num_procs * local_dim);
	
		vb = init_dvector(num_procs * local_dim);
		vc = init_dvector(num_procs * local_dim);
		for(i = 0; i < DIM; i++)
		{
			set_dvector_i(vb, i, (double)(DIM - i));
			for(j = 0; j < DIM; j++)
			{
				set_dmatrix_ij(x, i, j, (double)(i * DIM + j + 1));
			}
		}
	}

	local_vtmp = init_dvector(num_procs * local_dim);

	_mpi_init_dmatrix(local_x, d_dim, DIM, MPI_COMM_WORLD);

	local_vb = _mpi_init_dvector(d_dim, DIM, MPI_COMM_WORLD);
	local_vc = _mpi_init_dvector(d_dim, DIM, MPI_COMM_WORLD);

	_mpi_divide_dmatrix(local_x, d_dim, x, MPI_COMM_WORLD);

	_mpi_divide_dvector(local_vc, d_dim, vc, MPI_COMM_WORLD);
	_mpi_divide_dvector(local_vb, d_dim, vb, MPI_COMM_WORLD);

	_mpi_mul_dmatrix_dvec(local_vc, local_x, local_vb, local_vtmp, MPI_COMM_WORLD);

	_mpi_collect_dvector(vc, d_dim, local_vc, MPI_COMM_WORLD);

	_mpi_free_dmatrix(local_x, MPI_COMM_WORLD);

	_mpi_free_dvector(local_vc);
	_mpi_free_dvector(local_vb);

	free_dvector(local_vtmp);
	
	if(myrank == 0)
	{
		print_dvector(vc);

		free_dmatrix(x);
		free_dvector(vb);
		free_dvector(vc);
	}

	MPI_Finalize();

	return EXIT_SUCCESS;
}