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

#include <mpi.h>

#include "smut.h"
#include "stripesmut.h"

int
main (int argc, char **argv)
{
  FILE *f;
  int err;

  struct spcsr_t A, Acopy;
  struct gcsr_t gA;

  int rank, csize;
  int p;

  assert(sizeof(int) == 4);

  spcsr_init_clear(&A);
  gcsr_init_clear(&gA);

  MPI_Init (&argc, &argv);

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

  if (0 == rank) {
    printf ("MPI world size: %d\n", csize);

    if (argc <= 2) {
      printf("No files\n");
      return -1;
    }

    f = fopen(argv[1], "rb");
    if (!f) {
      perror("Error opening file: ");
      return -1;
    }
    err = spcsr_load_binfile (f, &A);
    if (err) {
      printf ("Error reading: %d\n", err);
      return -1;
    }
    fclose(f);
    spcsr_copy (&A, &Acopy);
  }

  gcsr_take_csr_root (&A, &gA, MPI_COMM_WORLD);

  {
    for (p = 0; p < csize; ++p) {
      MPI_Barrier(MPI_COMM_WORLD);
      if (p == rank) {
	printf("**** %d: gA(%d, %d), lA(%d, %d; %d) rows [%d, %d)\n",
	       rank, gA.nr, gA.nc, gA.local.nr, gA.local.nc, gA.local.nent,
	       gA.rowmap.lindstart, gA.rowmap.lindstart + gA.rowmap.lnind);
	printf("**** %d: local rowoff[1] = %d\n", rank,
	       (gA.local.rowoff? gA.local.rowoff[1] : -1));
      }
      MPI_Barrier(MPI_COMM_WORLD);
    }
  }

  /* fprintf(stderr, "%d: ugh1\n", rank); */
  gcsr_redist_ents (&gA, 0, MPI_COMM_WORLD);
  /* fprintf(stderr, "%d: ugh2\n", rank); */

  {
    for (p = 0; p < csize; ++p) {
      MPI_Barrier(MPI_COMM_WORLD);
      if (p == rank) {
	printf("---- %d: gA(%d, %d), lA(%d, %d; %d) rows [%d, %d)\n",
	       rank, gA.nr, gA.nc, gA.local.nr, gA.local.nc, gA.local.nent,
	       gA.rowmap.lindstart, gA.rowmap.lindstart + gA.rowmap.lnind);

	printf("---- %d: local rowoff[1] = %d\n", rank,
	       (gA.local.rowoff? gA.local.rowoff[1] : -1));
      }
      MPI_Barrier(MPI_COMM_WORLD);
    }
  }

  gcsr_give_csr_root (&gA, &A, MPI_COMM_WORLD);

  if (0 == rank) {
    int i, k;
    f = fopen(argv[2], "wb");
    if (!f) {
      perror("Error opening file: ");
      return -1;
    }
    err = spcsr_write_binfile (&A, f);
    fclose(f);

    assert(A.nr == Acopy.nr);
    assert(A.nc == Acopy.nc);
    assert(A.nent == Acopy.nent);
    for (i = 0; i < A.nr; ++i) {
      if (A.rowoff[i] != Acopy.rowoff[i]) {
	fprintf(stderr, "A and copy differ, rowoff[%d] = %d or %d\n",
		i, A.rowoff[i], Acopy.rowoff[i]);
      }
      for (k = A.rowoff[i]; k != A.rowoff[i+1]; ++k) {
	if (A.colind[k] != Acopy.colind[k])
	  fprintf(stderr, "A and copy differ, colind[%d] = %d or %d\n",
		  k, A.colind[k], Acopy.colind[k]);
	if (A.entry[k] != Acopy.entry[k])
	  fprintf(stderr, "A and copy differ, entry[%d] = %g or %g\n",
		  k, A.entry[k], Acopy.entry[k]);
      }
    }
  }

  spcsr_free (&A);
  gcsr_free (&gA, MPI_COMM_WORLD);
  MPI_Finalize();
  return 0;
}