#define   TOTALELEMS   147
#define   MAXPROC      32

#include <stdio.h>
#include <math.h>
#include "ga.h"
#include <mpi.h>

int main(int argc, char **argv)
{
    int me, nprocs;

    int dims[1], chunk[1], ld[1], lo, hi;
    int lo1, hi1, lo2, hi2;
    int g_a, g_b, a[MAXPROC*TOTALELEMS],b[MAXPROC*TOTALELEMS];
    int nelem, i;

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

    // Inicializácia GA
    GA_Initialize();

    // Priradenie ID lokálneho ranku a celkového počtu rankov
    me = GA_Nodeid();
    nprocs = GA_Nnodes();

    // Konfigurácia rozmerov polí
    dims[0]  = nprocs*TOTALELEMS + nprocs/2;
    ld[0]    = dims[0];
    chunk[0] = TOTALELEMS;

    // Vytvor Global Array g_a a duplikuj do g_b
    g_a = NGA_Create(C_INT, 1, dims, "array A", chunk);
    g_b = GA_Duplicate(g_a, "array B");
    // Inicializuj údaje v g_a
    if (me==0) {
       for(i=0; i<dims[0]; i++) a[i] = i;
       lo = 0;
       hi = dims[0]-1;
       NGA_Put(g_a, lo, hi, a, ld);
    }

    // Synchronizuj všetky procesy pred ďalším pokračovaním
    GA_Sync();

    // Zisti, ktoré údaje sú na lokálnom ranku
    NGA_Distribution(g_a, me, lo1, hi1);

    // Kopíruj lokálne údaje z g_a do lokálneho zásobníka a
    NGA_Get(g_a, lo1, hi1, a, ld);

    // Invertuj údaje lokálne
    nelem = hi1 - lo1 + 1;
    for (i=0; i<nelem; i++) b[i] = a[nelem-1-i];

    // Invertuj údaje globálnym uložením invertovaných blokov na správne miesto v global array g_b
    lo2 = dims[0] - hi1 -1;
    hi2 = dims[0] - lo1 -1;
    NGA_Put(g_b, lo2, hi2, b, ld);

    // Synchronizuj všetky procesy
    GA_Sync();

    // Dealokuj polia
    GA_Destroy(g_a);
    GA_Destroy(g_b);

    GA_Terminate();
    MPI_Finalize();
    return(0);
}