dlaset.c

来自「最著名最快的分子模拟软件」· C语言 代码 · 共 40 行

#include <ctype.h>
#include "gmx_lapack.h"


void
F77_FUNC(dlaset,DLASET)(char *uplo,
	int *m,
	int *n,
	double *alpha,
	double *beta,
	double *a,
	int *lda)
{
  int i,j,k;
  char ch=toupper(*uplo);

  if(ch=='U') {
    for(j=1;j<*n;j++) {
      k = (j < *m) ? j : *m;
      for(i=0;i<k;i++)
	a[j*(*lda)+i] = *alpha;
    }
  } else if(ch=='L') {
    k = (*m < *n) ? *m : *n;
    for(j=0;j<k;j++) {
      for(i=j+1;i<*m;i++)
	a[j*(*lda)+i] = *alpha;
    }
  } else {
    for(j=0;j<*n;j++) {
      for(i=0;i<*m;i++)
	a[j*(*lda)+i] = *alpha;
    }    
  }

  k = (*m < *n) ? *m : *n;
  for(i=0;i<k;i++)
    a[i*(*lda)+i] = *beta;
}