symamgrestorediag.c

来自「a software code for computing selected e」· C语言 代码 · 共 30 行

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


#include <lapack.h>
#include <ilupack.h>
#include <ilupackmacros.h>


void SYMAMGRESTOREDIAG(size_t *Fparam, integer *n, integer *ia, 
		       integer *ja, FLOAT *a)
{
  integer i,j,k;
  REALS   val;
  ILUPACKPARAM *IPparam;
  
  memcpy(&IPparam, Fparam, sizeof(size_t));

  for (i=0; i<*n; i++) {
      for (j=ia[i]; j<ia[i+1]; j++) {
	  k=ja[j-1]-1;
	  if (k==i) 
	     a[j-1]=IPparam->diag[i];
      }
  }
  free(IPparam->diag);
}