--- ray/src/util/cmatrix.c 2016/02/17 23:26:06 2.15 +++ ray/src/util/cmatrix.c 2019/08/14 18:20:02 2.22 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: cmatrix.c,v 2.15 2016/02/17 23:26:06 greg Exp $"; +static const char RCSid[] = "$Id: cmatrix.c,v 2.22 2019/08/14 18:20:02 greg Exp $"; #endif /* * Color matrix routines. @@ -8,6 +8,7 @@ static const char RCSid[] = "$Id: cmatrix.c,v 2.15 201 */ #include +#include "platform.h" #include "standard.h" #include "cmatrix.h" #include "platform.h" @@ -20,7 +21,7 @@ const char *cm_fmt_id[] = { }; const int cm_elem_size[] = { - 0, 0, 3*sizeof(float), 3*sizeof(double), 4, 4 + 0, 0, 4, 4, 3*sizeof(float), 3*sizeof(double) }; /* Allocate a color coefficient matrix */ @@ -33,7 +34,7 @@ cm_alloc(int nrows, int ncols) error(USER, "attempt to create empty matrix"); cm = (CMATRIX *)malloc(sizeof(CMATRIX) + sizeof(COLOR)*(nrows*ncols - 1)); - if (cm == NULL) + if (!cm) error(SYSTEM, "out of memory in cm_alloc()"); cm->nrows = nrows; cm->ncols = ncols; @@ -56,6 +57,8 @@ cm_resize(CMATRIX *cm, int nrows) { size_t old_size, new_size, ra_bounds[2]; + if (!cm) + return(NULL); if (nrows == cm->nrows) return(cm); if (nrows <= 0) { @@ -69,7 +72,7 @@ cm_resize(CMATRIX *cm, int nrows) new_size > ra_bounds[1]) { adjacent_ra_sizes(ra_bounds, new_size); cm = (CMATRIX *)realloc(cm, ra_bounds[1]); - if (cm == NULL) + if (!cm) error(SYSTEM, "out of memory in cm_resize()"); } cm->nrows = nrows; @@ -78,6 +81,7 @@ cm_resize(CMATRIX *cm, int nrows) typedef struct { int dtype; /* data type */ + int need2swap; /* need byte swap? */ int nrows, ncols; /* matrix size */ char *err; /* error message */ } CMINFO; /* header info record */ @@ -86,7 +90,7 @@ static int get_cminfo(char *s, void *p) { CMINFO *ip = (CMINFO *)p; - char fmt[32]; + char fmt[MAXFMTLEN]; int i; if (!strncmp(s, "NCOMP=", 6) && atoi(s+6) != 3) { @@ -101,6 +105,10 @@ get_cminfo(char *s, void *p) ip->ncols = atoi(s+6); return(0); } + if ((i = isbigendian(s)) >= 0) { + ip->need2swap = (nativebigendian() != i); + return(0); + } if (!formatval(fmt, s)) return(0); for (i = 1; i < DTend; i++) @@ -111,16 +119,17 @@ get_cminfo(char *s, void *p) /* Load header to obtain/check data type and number of columns */ char * -cm_getheader(int *dt, int *nr, int *nc, FILE *fp) +cm_getheader(int *dt, int *nr, int *nc, int *swp, FILE *fp) { CMINFO cmi; /* read header */ cmi.dtype = DTfromHeader; + cmi.need2swap = 0; cmi.nrows = cmi.ncols = 0; cmi.err = "unexpected EOF in header"; if (getheader(fp, get_cminfo, &cmi) < 0) return(cmi.err); - if (dt != NULL) { /* get/check data type? */ + if (dt) { /* get/check data type? */ if (cmi.dtype == DTfromHeader) { if (*dt == DTfromHeader) return("missing/unknown data format in header"); @@ -129,18 +138,20 @@ cm_getheader(int *dt, int *nr, int *nc, FILE *fp) else if (*dt != cmi.dtype) return("unexpected data format in header"); } - if (nr != NULL) { /* get/check #rows? */ + if (nr) { /* get/check #rows? */ if (*nr <= 0) *nr = cmi.nrows; else if ((cmi.nrows > 0) & (*nr != cmi.nrows)) return("unexpected row count in header"); } - if (nc != NULL) { /* get/check #columns? */ + if (nc) { /* get/check #columns? */ if (*nc <= 0) *nc = cmi.ncols; else if ((cmi.ncols > 0) & (*nc != cmi.ncols)) return("unexpected column count in header"); } + if (swp) /* get/check swap? */ + *swp = cmi.need2swap; return(NULL); } @@ -150,17 +161,18 @@ cm_load(const char *inspec, int nrows, int ncols, int { const int ROWINC = 2048; FILE *fp = stdin; + int swap = 0; CMATRIX *cm; - if (inspec == NULL) + if (!inspec) inspec = ""; else if (inspec[0] == '!') { fp = popen(inspec+1, "r"); - if (fp == NULL) { + if (!fp) { sprintf(errmsg, "cannot start command '%s'", inspec); error(SYSTEM, errmsg); } - } else if ((fp = fopen(inspec, "r")) == NULL) { + } else if (!(fp = fopen(inspec, "r"))) { sprintf(errmsg, "cannot open file '%s'", inspec); error(SYSTEM, errmsg); } @@ -170,8 +182,8 @@ cm_load(const char *inspec, int nrows, int ncols, int if (dtype != DTascii) SET_FILE_BINARY(fp); /* doesn't really work */ if (!dtype | !ncols) { /* expecting header? */ - char *err = cm_getheader(&dtype, &nrows, &ncols, fp); - if (err != NULL) + char *err = cm_getheader(&dtype, &nrows, &ncols, &swap, fp); + if (err) error(USER, err); if (ncols <= 0) error(USER, "unspecified number of columns"); @@ -212,7 +224,7 @@ cm_load(const char *inspec, int nrows, int ncols, int cm = cm_alloc(guessrows, ncols); } else cm = cm_alloc(nrows, ncols); - if (cm == NULL) /* XXX never happens */ + if (!cm) /* XXX never happens */ return(NULL); if (dtype == DTascii) { /* read text file */ int maxrow = (nrows > 0 ? nrows : 32000); @@ -243,7 +255,7 @@ cm_load(const char *inspec, int nrows, int ncols, int if (sizeof(COLOR) == cm_elem_size[dtype]) { int nread = 0; do { /* read all we can */ - nread += fread(cm->cmem + 3*nread, + nread += getbinary(cm->cmem + 3*nread, sizeof(COLOR), cm->nrows*cm->ncols - nread, fp); @@ -268,7 +280,7 @@ cm_load(const char *inspec, int nrows, int ncols, int if (n <= 0) goto not_handled; while (n--) { - if (fread(dc, sizeof(double), 3, fp) != 3) + if (getbinary(dc, sizeof(double), 3, fp) != 3) goto EOFerror; copycolor(cvp, dc); cvp += 3; @@ -281,7 +293,7 @@ cm_load(const char *inspec, int nrows, int ncols, int if (n <= 0) goto not_handled; while (n--) { - if (fread(fc, sizeof(float), 3, fp) != 3) + if (getbinary(fc, sizeof(float), 3, fp) != 3) goto EOFerror; copycolor(cvp, fc); cvp += 3; @@ -294,6 +306,12 @@ cm_load(const char *inspec, int nrows, int ncols, int error(WARNING, errmsg); } } + if (swap) { + if (dtype == DTfloat) + swap32((char *)cm->cmem, 3*cm->nrows*cm->ncols); + else if (dtype == DTdouble) + swap64((char *)cm->cmem, 3*cm->nrows*cm->ncols); + } if (fp != stdin) { if (inspec[0] != '!') fclose(fp); @@ -322,10 +340,12 @@ cm_column(const CMATRIX *cm, int c) CMATRIX *cvr; int dr; + if (!cm) + return(NULL); if ((c < 0) | (c >= cm->ncols)) error(INTERNAL, "column requested outside matrix"); cvr = cm_alloc(cm->nrows, 1); - if (cvr == NULL) + if (!cvr) return(NULL); for (dr = 0; dr < cm->nrows; dr++) { const COLORV *sp = cm_lval(cm,dr,c); @@ -337,27 +357,6 @@ cm_column(const CMATRIX *cm, int c) return(cvr); } -/* Scale a matrix by a single value */ -CMATRIX * -cm_scale(const CMATRIX *cm1, const COLOR sca) -{ - CMATRIX *cmr; - int dr, dc; - - cmr = cm_alloc(cm1->nrows, cm1->ncols); - if (cmr == NULL) - return(NULL); - for (dr = 0; dr < cmr->nrows; dr++) - for (dc = 0; dc < cmr->ncols; dc++) { - const COLORV *sp = cm_lval(cm1,dr,dc); - COLORV *dp = cm_lval(cmr,dr,dc); - dp[0] = sp[0] * sca[0]; - dp[1] = sp[1] * sca[1]; - dp[2] = sp[2] * sca[2]; - } - return(cmr); -} - /* Multiply two matrices (or a matrix and a vector) and allocate the result */ CMATRIX * cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2) @@ -366,10 +365,12 @@ cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2) CMATRIX *cmr; int dr, dc, i; + if (!cm1 | !cm2) + return(NULL); if ((cm1->ncols <= 0) | (cm1->ncols != cm2->nrows)) error(INTERNAL, "matrix dimension mismatch in cm_multiply()"); cmr = cm_alloc(cm1->nrows, cm2->ncols); - if (cmr == NULL) + if (!cmr) return(NULL); /* optimization: check for zero rows & cols */ if (((cm1->nrows > 5) | (cm2->ncols > 5)) & (cm1->ncols > 5)) { @@ -394,22 +395,22 @@ cm_multiply(const CMATRIX *cm1, const CMATRIX *cm2) COLORV *dp = cm_lval(cmr,dr,dc); double res[3]; dp[0] = dp[1] = dp[2] = 0; - if (rowcheck != NULL && !rowcheck[dr]) + if (rowcheck && !rowcheck[dr]) continue; - if (colcheck != NULL && !colcheck[dc]) + if (colcheck && !colcheck[dc]) continue; res[0] = res[1] = res[2] = 0; for (i = 0; i < cm1->ncols; i++) { const COLORV *cp1 = cm_lval(cm1,dr,i); const COLORV *cp2 = cm_lval(cm2,i,dc); - res[0] += cp1[0] * cp2[0]; - res[1] += cp1[1] * cp2[1]; - res[2] += cp1[2] * cp2[2]; + res[0] += (double)cp1[0] * cp2[0]; + res[1] += (double)cp1[1] * cp2[1]; + res[2] += (double)cp1[2] * cp2[2]; } copycolor(dp, res); } - if (rowcheck != NULL) free(rowcheck); - if (colcheck != NULL) free(colcheck); + if (rowcheck) free(rowcheck); + if (colcheck) free(colcheck); return(cmr); } @@ -418,9 +419,12 @@ int cm_write(const CMATRIX *cm, int dtype, FILE *fp) { static const char tabEOL[2] = {'\t','\n'}; - const COLORV *mp = cm->cmem; + const COLORV *mp; int r, c; + if (!cm) + return(0); + mp = cm->cmem; switch (dtype) { case DTascii: for (r = 0; r < cm->nrows; r++) @@ -434,7 +438,7 @@ cm_write(const CMATRIX *cm, int dtype, FILE *fp) if (sizeof(COLOR) == cm_elem_size[dtype]) { r = cm->ncols*cm->nrows; while (r > 0) { - c = fwrite(mp, sizeof(COLOR), r, fp); + c = putbinary(mp, sizeof(COLOR), r, fp); if (c <= 0) return(0); mp += 3*c; @@ -445,7 +449,7 @@ cm_write(const CMATRIX *cm, int dtype, FILE *fp) r = cm->ncols*cm->nrows; while (r--) { copycolor(dc, mp); - if (fwrite(dc, sizeof(double), 3, fp) != 3) + if (putbinary(dc, sizeof(double), 3, fp) != 3) return(0); mp += 3; } @@ -454,7 +458,7 @@ cm_write(const CMATRIX *cm, int dtype, FILE *fp) r = cm->ncols*cm->nrows; while (r--) { copycolor(fc, mp); - if (fwrite(fc, sizeof(float), 3, fp) != 3) + if (putbinary(fc, sizeof(float), 3, fp) != 3) return(0); mp += 3; }