/***************************************
  マルチプロセッサ版 サンプルプログラム
    main.c
***************************************/
#include <mpi.h>
#include <stdio.h>
#include "../gmres.h"

void GetInfoProblem(int *,int *,int *);
void Problem(int,int *,double *,double *,int *);

int main(int argc,char **args)
{
  Mat A;
  Vec x,b;
 
  int i,j,flag;
  int ipe,npe;
  int MyStart,MyEnd;
  int ret,index;
  int TotalSize, nsize, max_nsize, tsize;
  int max_iter;
  int ierr;
  int ProblemSize,NonZeroSize,MaxNonZeroSize;
  double tol;
  double start_time;

  PetscInitialize(&argc,&args,NULL,NULL);
  MPI_Comm_size(MPI_COMM_WORLD,&npe);
  MPI_Comm_rank(MPI_COMM_WORLD,&ipe);

  GetInfoProblem(&ProblemSize, &NonZeroSize, &MaxNonZeroSize);

  ierr = MatCreate(MPI_COMM_WORLD,ProblemSize,ProblemSize,&A);
  ierr = MatGetOwnershipRange(&A,&MyStart,&MyEnd);

  ierr = VecCreate(MPI_COMM_WORLD,NULL,ProblemSize,&x);
  ierr = VecCreate(MPI_COMM_WORLD,NULL,ProblemSize,&b);

   for ( i=MyStart; i<MyEnd; i++ ) {
    int ja[16],tsize;
    double a[16],b_value;

    Problem(i,ja,a,&b_value,&tsize);

    for(j=0;j<tsize;j++) {
       MatSetValues(&A,1,&i,1,&ja[j],&a[j],INSERT_VALUES);
    }
    VecSetValues(&b,1,&i,&b_value,INSERT_VALUES);
  }
  ierr=MatAssemblyBegin(&A,MAT_FINAL_ASSEMBLY);
  ierr=MatAssemblyEnd(&A,MAT_FINAL_ASSEMBLY);

  start_time=MPI_Wtime();

  max_iter=1000000;
  tol=1e-12;

  ret=GMRES(&A,&x,&b,max_iter,tol);

  if( ipe==0 ){
     printf("Iteration=%d PE=%d\n",ret,npe);
     printf("Execution Time=%f\n",MPI_Wtime()-start_time);
  }
  
  VecDestroy(&b);
  VecDestroy(&x);
  MatDestroy(&A);
  PetscFinalize();
  return 0;
}