Mercurial > forge
view main/sparse/SuperLU/SRC/zreadhb.c @ 0:6b33357c7561 octave-forge
Initial revision
author | pkienzle |
---|---|
date | Wed, 10 Oct 2001 19:54:49 +0000 |
parents | |
children | 7dad48fc229c |
line wrap: on
line source
/* * -- SuperLU routine (version 2.0) -- * Univ. of California Berkeley, Xerox Palo Alto Research Center, * and Lawrence Berkeley National Lab. * November 15, 1997 * */ #include <stdio.h> #include <stdlib.h> #include "zsp_defs.h" /* Eat up the rest of the current line */ int zDumpLine(FILE *fp) { register int c; while ((c = fgetc(fp)) != '\n') ; return 0; } int zParseIntFormat(char *buf, int *num, int *size) { char *tmp; tmp = buf; while (*tmp++ != '(') ; sscanf(tmp, "%d", num); while (*tmp != 'I' && *tmp != 'i') ++tmp; ++tmp; sscanf(tmp, "%d", size); return 0; } int zParseFloatFormat(char *buf, int *num, int *size) { char *tmp, *period; tmp = buf; while (*tmp++ != '(') ; sscanf(tmp, "%d", num); while (*tmp != 'E' && *tmp != 'e' && *tmp != 'D' && *tmp != 'd' && *tmp != 'F' && *tmp != 'f') ++tmp; ++tmp; period = tmp; while (*period != '.' && *period != ')') ++period ; *period = '\0'; sscanf(tmp, "%2d", size); return 0; } int zReadVector(FILE *fp, int n, int *where, int perline, int persize) { register int i, j, item; char tmp, buf[100]; i = 0; while (i < n) { fgets(buf, 100, fp); /* read a line at a time */ for (j=0; j<perline && i<n; j++) { tmp = buf[(j+1)*persize]; /* save the char at that place */ buf[(j+1)*persize] = 0; /* null terminate */ item = atoi(&buf[j*persize]); buf[(j+1)*persize] = tmp; /* recover the char at that place */ where[i++] = item - 1; } } return 0; } /* Read complex numbers as pairs of (real, imaginary) */ int zReadValues(FILE *fp, int n, doublecomplex *destination, int perline, int persize) { register int i, j, k, s, pair; register double realpart; char tmp, buf[100]; i = pair = 0; while (i < n) { fgets(buf, 100, fp); /* read a line at a time */ for (j=0; j<perline && i<n; j++) { tmp = buf[(j+1)*persize]; /* save the char at that place */ buf[(j+1)*persize] = 0; /* null terminate */ s = j*persize; for (k = 0; k < persize; ++k) /* No D_ format in C */ if ( buf[s+k] == 'D' || buf[s+k] == 'd' ) buf[s+k] = 'E'; if ( pair == 0 ) { /* The value is real part */ realpart = atof(&buf[s]); pair = 1; } else { /* The value is imaginary part */ destination[i].r = realpart; destination[i++].i = atof(&buf[s]); pair = 0; } buf[(j+1)*persize] = tmp; /* recover the char at that place */ } } return 0; } void zreadhb(int *nrow, int *ncol, int *nonz, doublecomplex **nzval, int **rowind, int **colptr) { /* * Purpose * ======= * * Read a DOUBLE COMPLEX PRECISION matrix stored in Harwell-Boeing format * as described below. * * Line 1 (A72,A8) * Col. 1 - 72 Title (TITLE) * Col. 73 - 80 Key (KEY) * * Line 2 (5I14) * Col. 1 - 14 Total number of lines excluding header (TOTCRD) * Col. 15 - 28 Number of lines for pointers (PTRCRD) * Col. 29 - 42 Number of lines for row (or variable) indices (INDCRD) * Col. 43 - 56 Number of lines for numerical values (VALCRD) * Col. 57 - 70 Number of lines for right-hand sides (RHSCRD) * (including starting guesses and solution vectors * if present) * (zero indicates no right-hand side data is present) * * Line 3 (A3, 11X, 4I14) * Col. 1 - 3 Matrix type (see below) (MXTYPE) * Col. 15 - 28 Number of rows (or variables) (NROW) * Col. 29 - 42 Number of columns (or elements) (NCOL) * Col. 43 - 56 Number of row (or variable) indices (NNZERO) * (equal to number of entries for assembled matrices) * Col. 57 - 70 Number of elemental matrix entries (NELTVL) * (zero in the case of assembled matrices) * Line 4 (2A16, 2A20) * Col. 1 - 16 Format for pointers (PTRFMT) * Col. 17 - 32 Format for row (or variable) indices (INDFMT) * Col. 33 - 52 Format for numerical values of coefficient matrix (VALFMT) * Col. 53 - 72 Format for numerical values of right-hand sides (RHSFMT) * * Line 5 (A3, 11X, 2I14) Only present if there are right-hand sides present * Col. 1 Right-hand side type: * F for full storage or M for same format as matrix * Col. 2 G if a starting vector(s) (Guess) is supplied. (RHSTYP) * Col. 3 X if an exact solution vector(s) is supplied. * Col. 15 - 28 Number of right-hand sides (NRHS) * Col. 29 - 42 Number of row indices (NRHSIX) * (ignored in case of unassembled matrices) * * The three character type field on line 3 describes the matrix type. * The following table lists the permitted values for each of the three * characters. As an example of the type field, RSA denotes that the matrix * is real, symmetric, and assembled. * * First Character: * R Real matrix * C Complex matrix * P Pattern only (no numerical values supplied) * * Second Character: * S Symmetric * U Unsymmetric * H Hermitian * Z Skew symmetric * R Rectangular * * Third Character: * A Assembled * E Elemental matrices (unassembled) * */ register int i, numer_lines = 0, rhscrd = 0; int tmp, colnum, colsize, rownum, rowsize, valnum, valsize; char buf[100], type[4], key[10]; FILE *fp; fp = stdin; /* Line 1 */ fgets(buf, 100, fp); fputs(buf, stdout); #if 0 fscanf(fp, "%72c", buf); buf[72] = 0; printf("Title: %s", buf); fscanf(fp, "%8c", key); key[8] = 0; printf("Key: %s\n", key); zDumpLine(fp); #endif /* Line 2 */ for (i=0; i<5; i++) { fscanf(fp, "%14c", buf); buf[14] = 0; sscanf(buf, "%d", &tmp); if (i == 3) numer_lines = tmp; if (i == 4 && tmp) rhscrd = tmp; } zDumpLine(fp); /* Line 3 */ fscanf(fp, "%3c", type); fscanf(fp, "%11c", buf); /* pad */ type[3] = 0; #ifdef DEBUG printf("Matrix type %s\n", type); #endif fscanf(fp, "%14c", buf); sscanf(buf, "%d", nrow); fscanf(fp, "%14c", buf); sscanf(buf, "%d", ncol); fscanf(fp, "%14c", buf); sscanf(buf, "%d", nonz); fscanf(fp, "%14c", buf); sscanf(buf, "%d", &tmp); if (tmp != 0) printf("This is not an assembled matrix!\n"); if (*nrow != *ncol) printf("Matrix is not square.\n"); zDumpLine(fp); /* Allocate storage for the three arrays ( nzval, rowind, colptr ) */ zallocateA(*ncol, *nonz, nzval, rowind, colptr); /* Line 4: format statement */ fscanf(fp, "%16c", buf); zParseIntFormat(buf, &colnum, &colsize); fscanf(fp, "%16c", buf); zParseIntFormat(buf, &rownum, &rowsize); fscanf(fp, "%20c", buf); zParseFloatFormat(buf, &valnum, &valsize); fscanf(fp, "%20c", buf); zDumpLine(fp); /* Line 5: right-hand side */ if ( rhscrd ) zDumpLine(fp); /* skip RHSFMT */ #ifdef DEBUG printf("%d rows, %d nonzeros\n", *nrow, *nonz); printf("colnum %d, colsize %d\n", colnum, colsize); printf("rownum %d, rowsize %d\n", rownum, rowsize); printf("valnum %d, valsize %d\n", valnum, valsize); #endif zReadVector(fp, *ncol+1, *colptr, colnum, colsize); zReadVector(fp, *nonz, *rowind, rownum, rowsize); if ( numer_lines ) { zReadValues(fp, *nonz, *nzval, valnum, valsize); } fclose(fp); }