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

#define MAX 100

int main(int argc, char* argv[])
{
  int rank, size;
  MPI_Status status;
  MPI_Request req;

  //inicializácia MPI
  MPI_Init(&argc, &argv);

  //zistenie poradového čísla procesu - rank
  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
  //zistenie celkového počtu procesov - size
  MPI_Comm_size(MPI_COMM_WORLD, &size);

  double A[MAX], B[MAX], result;

  //údaje uložené vo vektoroch A, B bude mať len proces 0
  if(!rank) { 		//túto časť vykoná iba proces 0
    for(int i=0; i<MAX; i++) {
      A[i] = i;		//inicializácia vektora A
      B[i] = i*i;	//inicializácia vektora B
    }
  }
  
  //počet položiek pridelených každému procesu
  const int count = MAX / size;

  double workA[count], workB[count], workResult=0;

  //rozdelenie vektorov na časti jednotlivým procesom
  MPI_Scatter(A, count, MPI_DOUBLE, workA, count, MPI_DOUBLE, 0, MPI_COMM_WORLD);
  MPI_Scatter(B, count, MPI_DOUBLE, workB, count, MPI_DOUBLE, 0, MPI_COMM_WORLD);

  //výpočet čiastkového výsledku
  for(int i=0; i<count; i++) {
    workResult += workA[i] * workB[i];
  }
  
  //kombinovanie poslaných čiastkových výsledkov procesom 0
  MPI_Reduce(&workResult, &result, 1, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD);
  
  if(!rank) {
    printf("Vysledok je: %f\n", result);
  }

  //ukončenie MPI
  MPI_Finalize();

  return(0);
}