163 lines
4.7 KiB
C
163 lines
4.7 KiB
C
/*BHEADER**********************************************************************
|
|
* Copyright (c) 2007, Lawrence Livermore National Security, LLC.
|
|
* Produced at the Lawrence Livermore National Laboratory.
|
|
* Written by the HYPRE team. UCRL-CODE-222953.
|
|
* All rights reserved.
|
|
*
|
|
* This file is part of HYPRE (see http://www.llnl.gov/CASC/hypre/).
|
|
* Please see the COPYRIGHT_and_LICENSE file for the copyright notice,
|
|
* disclaimer, contact information and the GNU Lesser General Public License.
|
|
*
|
|
* HYPRE is free software; you can redistribute it and/or modify it under the
|
|
* terms of the GNU General Public License (as published by the Free Software
|
|
* Foundation) version 2.1 dated February 1999.
|
|
*
|
|
* HYPRE is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the IMPLIED WARRANTY OF MERCHANTABILITY or FITNESS
|
|
* FOR A PARTICULAR PURPOSE. See the terms and conditions of the GNU General
|
|
* Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
* $Revision$
|
|
***********************************************************************EHEADER*/
|
|
|
|
|
|
|
|
|
|
#include "hypre_lapack.h"
|
|
#include "f2c.h"
|
|
|
|
/* Subroutine */ int dlaset_(char *uplo, integer *m, integer *n, doublereal *
|
|
alpha, doublereal *beta, doublereal *a, integer *lda)
|
|
{
|
|
/* -- LAPACK auxiliary routine (version 3.0) --
|
|
Univ. of Tennessee, Univ. of California Berkeley, NAG Ltd.,
|
|
Courant Institute, Argonne National Lab, and Rice University
|
|
October 31, 1992
|
|
|
|
|
|
Purpose
|
|
=======
|
|
|
|
DLASET initializes an m-by-n matrix A to BETA on the diagonal and
|
|
ALPHA on the offdiagonals.
|
|
|
|
Arguments
|
|
=========
|
|
|
|
UPLO (input) CHARACTER*1
|
|
Specifies the part of the matrix A to be set.
|
|
= 'U': Upper triangular part is set; the strictly lower
|
|
triangular part of A is not changed.
|
|
= 'L': Lower triangular part is set; the strictly upper
|
|
triangular part of A is not changed.
|
|
Otherwise: All of the matrix A is set.
|
|
|
|
M (input) INTEGER
|
|
The number of rows of the matrix A. M >= 0.
|
|
|
|
N (input) INTEGER
|
|
The number of columns of the matrix A. N >= 0.
|
|
|
|
ALPHA (input) DOUBLE PRECISION
|
|
The constant to which the offdiagonal elements are to be set.
|
|
|
|
BETA (input) DOUBLE PRECISION
|
|
The constant to which the diagonal elements are to be set.
|
|
|
|
A (input/output) DOUBLE PRECISION array, dimension (LDA,N)
|
|
On exit, the leading m-by-n submatrix of A is set as follows:
|
|
|
|
if UPLO = 'U', A(i,j) = ALPHA, 1<=i<=j-1, 1<=j<=n,
|
|
if UPLO = 'L', A(i,j) = ALPHA, j+1<=i<=m, 1<=j<=n,
|
|
otherwise, A(i,j) = ALPHA, 1<=i<=m, 1<=j<=n, i.ne.j,
|
|
|
|
and, for all UPLO, A(i,i) = BETA, 1<=i<=min(m,n).
|
|
|
|
LDA (input) INTEGER
|
|
The leading dimension of the array A. LDA >= max(1,M).
|
|
|
|
=====================================================================
|
|
|
|
|
|
Parameter adjustments */
|
|
/* System generated locals */
|
|
integer a_dim1, a_offset, i__1, i__2, i__3;
|
|
/* Local variables */
|
|
static integer i__, j;
|
|
extern logical lsame_(char *, char *);
|
|
#define a_ref(a_1,a_2) a[(a_2)*a_dim1 + a_1]
|
|
|
|
a_dim1 = *lda;
|
|
a_offset = 1 + a_dim1 * 1;
|
|
a -= a_offset;
|
|
|
|
/* Function Body */
|
|
if (lsame_(uplo, "U")) {
|
|
|
|
/* Set the strictly upper triangular or trapezoidal part of the
|
|
array to ALPHA. */
|
|
|
|
i__1 = *n;
|
|
for (j = 2; j <= i__1; ++j) {
|
|
/* Computing MIN */
|
|
i__3 = j - 1;
|
|
i__2 = min(i__3,*m);
|
|
for (i__ = 1; i__ <= i__2; ++i__) {
|
|
a_ref(i__, j) = *alpha;
|
|
/* L10: */
|
|
}
|
|
/* L20: */
|
|
}
|
|
|
|
} else if (lsame_(uplo, "L")) {
|
|
|
|
/* Set the strictly lower triangular or trapezoidal part of the
|
|
array to ALPHA. */
|
|
|
|
i__1 = min(*m,*n);
|
|
for (j = 1; j <= i__1; ++j) {
|
|
i__2 = *m;
|
|
for (i__ = j + 1; i__ <= i__2; ++i__) {
|
|
a_ref(i__, j) = *alpha;
|
|
/* L30: */
|
|
}
|
|
/* L40: */
|
|
}
|
|
|
|
} else {
|
|
|
|
/* Set the leading m-by-n submatrix to ALPHA. */
|
|
|
|
i__1 = *n;
|
|
for (j = 1; j <= i__1; ++j) {
|
|
i__2 = *m;
|
|
for (i__ = 1; i__ <= i__2; ++i__) {
|
|
a_ref(i__, j) = *alpha;
|
|
/* L50: */
|
|
}
|
|
/* L60: */
|
|
}
|
|
}
|
|
|
|
/* Set the first min(M,N) diagonal elements to BETA. */
|
|
|
|
i__1 = min(*m,*n);
|
|
for (i__ = 1; i__ <= i__1; ++i__) {
|
|
a_ref(i__, i__) = *beta;
|
|
/* L70: */
|
|
}
|
|
|
|
return 0;
|
|
|
|
/* End of DLASET */
|
|
|
|
} /* dlaset_ */
|
|
|
|
#undef a_ref
|
|
|
|
|