--- ray/src/util/rmatrix.c 2025/03/22 01:27:22 2.84 +++ ray/src/util/rmatrix.c 2025/04/16 20:20:14 2.91 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: rmatrix.c,v 2.84 2025/03/22 01:27:22 greg Exp $"; +static const char RCSid[] = "$Id: rmatrix.c,v 2.91 2025/04/16 20:20:14 greg Exp $"; #endif /* * General matrix operations. @@ -194,7 +194,7 @@ rmx_load_ascii(rmx_dtype *drp, const RMATRIX *rm, FILE for (j = 0; j < rm->ncols; j++) for (k = rm->ncomp; k-- > 0; ) - if (fscanf(fp, "%lf", drp++) != 1) + if (fscanf(fp, rmx_scanfmt, drp++) != 1) return(0); return(1); } @@ -202,10 +202,16 @@ rmx_load_ascii(rmx_dtype *drp, const RMATRIX *rm, FILE static int rmx_load_float(rmx_dtype *drp, const RMATRIX *rm, FILE *fp) { +#if DTrmx_native==DTfloat + if (getbinary(drp, sizeof(*drp)*rm->ncomp, rm->ncols, fp) != rm->ncols) + return(0); + if (rm->pflags & RMF_SWAPIN) + swap32((char *)drp, rm->ncols*rm->ncomp); +#else int j, k; - float val[100]; + float val[MAXCOMP]; - if (rm->ncomp > 100) { + if (rm->ncomp > MAXCOMP) { fputs("Unsupported # components in rmx_load_float()\n", stderr); exit(1); } @@ -217,16 +223,35 @@ rmx_load_float(rmx_dtype *drp, const RMATRIX *rm, FILE for (k = 0; k < rm->ncomp; k++) *drp++ = val[k]; } +#endif return(1); } static int rmx_load_double(rmx_dtype *drp, const RMATRIX *rm, FILE *fp) { +#if DTrmx_native==DTdouble if (getbinary(drp, sizeof(*drp)*rm->ncomp, rm->ncols, fp) != rm->ncols) return(0); if (rm->pflags & RMF_SWAPIN) swap64((char *)drp, rm->ncols*rm->ncomp); +#else + int j, k; + double val[MAXCOMP]; + + if (rm->ncomp > MAXCOMP) { + fputs("Unsupported # components in rmx_load_double()\n", stderr); + exit(1); + } + for (j = 0; j < rm->ncols; j++) { + if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp) + return(0); + if (rm->pflags & RMF_SWAPIN) + swap64((char *)val, rm->ncomp); + for (k = 0; k < rm->ncomp; k++) + *drp++ = (float)val[k]; + } +#endif return(1); } @@ -256,13 +281,13 @@ rmx_load_rgbe(rmx_dtype *drp, const RMATRIX *rm, FILE static int rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE *fp) { - uby8 *scan; - SCOLOR scol; + COLRV *scan; + COLORV scol[MAXCOMP]; int j, k; - if ((rm->ncomp < 3) | (rm->ncomp > MAXCSAMP)) + if ((rm->ncomp < 3) | (rm->ncomp > MAXCOMP)) return(0); - scan = (uby8 *)tempbuffer((rm->ncomp+1)*rm->ncols); + scan = (COLRV *)tempbuffer((rm->ncomp+1)*rm->ncols); if (!scan) return(0); if (freadscolrs(scan, rm->ncomp, rm->ncols, fp) < 0) @@ -428,9 +453,9 @@ rmx_load(const char *inspec, RMPref rmp) /* undo exposure? */ if ((dnew->cexp[0] != 1.f) | (dnew->cexp[1] != 1.f) | (dnew->cexp[2] != 1.f)) { - double cmlt[MAXCSAMP]; + double cmlt[MAXCOMP]; int i; - if (dnew->ncomp > MAXCSAMP) { + if (dnew->ncomp > MAXCOMP) { fprintf(stderr, "Excess spectral components in: %s\n", inspec); rmx_free(dnew); @@ -447,32 +472,47 @@ rmx_load(const char *inspec, RMPref rmp) return(dnew); } +#if DTrmx_native==DTdouble static int -rmx_write_ascii(const rmx_dtype *dp, int nc, int len, FILE *fp) +rmx_write_float(const rmx_dtype *dp, int len, FILE *fp) { - while (len-- > 0) { - int k = nc; - while (k-- > 0) - fprintf(fp, " %.7e", *dp++); - fputc('\t', fp); + float val; + + while (len--) { + val = (float)*dp++; + if (putbinary(&val, sizeof(val), 1, fp) != 1) + return(0); } - return(fputc('\n', fp) != EOF); + return(1); } - +#else static int -rmx_write_float(const rmx_dtype *dp, int len, FILE *fp) +rmx_write_double(const rmx_dtype *dp, int len, FILE *fp) { - float val; + double val; while (len--) { val = *dp++; - if (putbinary(&val, sizeof(float), 1, fp) != 1) + if (putbinary(&val, sizeof(val), 1, fp) != 1) return(0); } return(1); } +#endif static int +rmx_write_ascii(const rmx_dtype *dp, int nc, int len, FILE *fp) +{ + while (len-- > 0) { + int k = nc; + while (k-- > 0) + fprintf(fp, " %.7e", *dp++); + fputc('\t', fp); + } + return(fputc('\n', fp) != EOF); +} + +static int rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, FILE *fp) { COLR *scan; @@ -494,12 +534,12 @@ rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, F static int rmx_write_spec(const rmx_dtype *dp, int nc, int len, FILE *fp) { - uby8 *scan; - SCOLOR scol; + COLRV *scan; + COLORV scol[MAXCOMP]; int j, k; - if (nc < 3) return(0); - scan = (uby8 *)tempbuffer((nc+1)*len); + if ((nc < 3) | (nc > MAXCOMP)) return(0); + scan = (COLRV *)tempbuffer((nc+1)*len); if (!scan) return(0); for (j = 0; j < len; j++, dp += nc) { for (k = nc; k--; ) @@ -534,9 +574,13 @@ rmx_write_header(const RMATRIX *rm, int dtype, FILE *f return(0); if (rm->info) fputs(rm->info, fp); - if (dtype == DTfromHeader) + if (dtype == DTfromHeader) { dtype = rm->dtype; - else if (dtype == DTrgbe && (rm->dtype == DTxyze || +#if DTrmx_native==DTfloat + if (dtype == DTdouble) /* but stored as float? */ + dtype = DTfloat; +#endif + } else if (dtype == DTrgbe && (rm->dtype == DTxyze || findCIEprims(rm->info))) dtype = DTxyze; else if ((dtype == DTxyze) & (rm->dtype == DTrgbe)) @@ -583,12 +627,17 @@ int rmx_write_data(const rmx_dtype *dp, int nc, int len, int dtype, FILE *fp) { switch (dtype) { - case DTascii: - return(rmx_write_ascii(dp, nc, len, fp)); +#if DTrmx_native==DTdouble case DTfloat: return(rmx_write_float(dp, nc*len, fp)); +#else + case DTdouble: + return(rmx_write_double(dp, nc*len, fp)); +#endif case DTrmx_native: return(putbinary(dp, sizeof(*dp)*nc, len, fp) == len); + case DTascii: + return(rmx_write_ascii(dp, nc, len, fp)); case DTrgbe: case DTxyze: return(rmx_write_rgbe(dp, nc, len, fp)); @@ -676,18 +725,20 @@ rmx_copy(const RMATRIX *rm) int rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int dometa) { - if (!rdst | !rsrc || (rdst->nrows != rsrc->nrows) | - (rdst->ncols != rsrc->ncols) | - (rdst->ncomp != rsrc->ncomp)) + if (!rdst | !rsrc) return(0); - if (dometa) { /* transfer everything? */ rmx_reset(rdst); *rdst = *rsrc; rsrc->info = NULL; rsrc->mapped = NULL; rsrc->mtx = NULL; return(1); } -#ifdef MAP_FILE /* just matrix data -- leave metadata */ + /* just matrix data -- leave metadata */ + if ((rdst->nrows != rsrc->nrows) | + (rdst->ncols != rsrc->ncols) | + (rdst->ncomp != rsrc->ncomp)) + return(0); +#ifdef MAP_FILE if (rdst->mapped) munmap(rdst->mapped, rmx_mapped_size(rdst)); else @@ -703,38 +754,52 @@ rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int do return(1); } -/* Allocate and assign transposed matrix */ -RMATRIX * -rmx_transpose(const RMATRIX *rm) +/* Transpose the given matrix */ +int +rmx_transpose(RMATRIX *rm) { - RMATRIX *dnew; + RMATRIX dnew; int i, j; - if (!rm || !rm->mtx) + if (!rm || !rm->mtx | (rm->ncomp > MAXCOMP)) return(0); - if ((rm->nrows == 1) | (rm->ncols == 1)) { - dnew = rmx_copy(rm); - if (!dnew) - return(NULL); - dnew->nrows = rm->ncols; - dnew->ncols = rm->nrows; - return(dnew); + if (rm->info) + rmx_addinfo(rm, "Transposed rows and columns\n"); + if ((rm->nrows == 1) | (rm->ncols == 1)) { /* vector? */ + j = rm->ncols; + rm->ncols = rm->nrows; + rm->nrows = j; + return(1); } - dnew = rmx_alloc(rm->ncols, rm->nrows, rm->ncomp); - if (!dnew) - return(NULL); - if (rm->info) { - rmx_addinfo(dnew, rm->info); - rmx_addinfo(dnew, "Transposed rows and columns\n"); + if (rm->nrows == rm->ncols) { /* square matrix case */ + rmx_dtype val[MAXCOMP]; + for (j = rm->ncols; j--; ) + for (i = rm->nrows; i--; ) { + if (i == j) continue; + memcpy(val, rmx_val(rm,i,j), + sizeof(rmx_dtype)*rm->ncomp); + memcpy(rmx_lval(rm,i,j), rmx_val(rm,j,i), + sizeof(rmx_dtype)*rm->ncomp); + memcpy(rmx_val(rm,j,i), val, + sizeof(rmx_dtype)*rm->ncomp); + } + return(1); } - dnew->dtype = rm->dtype; - copycolor(dnew->cexp, rm->cexp); - memcpy(dnew->wlpart, rm->wlpart, sizeof(dnew->wlpart)); - for (j = dnew->ncols; j--; ) - for (i = dnew->nrows; i--; ) - memcpy(rmx_lval(dnew,i,j), rmx_val(rm,j,i), - sizeof(rmx_dtype)*dnew->ncomp); - return(dnew); + memset(&dnew, 0, sizeof(dnew)); + dnew.ncols = rm->nrows; dnew.nrows = rm->ncols; + dnew.ncomp = rm->ncomp; + if (!rmx_prepare(&dnew)) + return(0); + dnew.info = rm->info; rm->info = NULL; + dnew.dtype = rm->dtype; + copycolor(dnew.cexp, rm->cexp); + memcpy(dnew.wlpart, rm->wlpart, sizeof(dnew.wlpart)); + for (j = dnew.ncols; j--; ) + for (i = dnew.nrows; i--; ) + memcpy(rmx_lval(&dnew,i,j), rmx_val(rm,j,i), + sizeof(rmx_dtype)*dnew.ncomp); + /* and reassign result */ + return(rmx_transfer_data(rm, &dnew, 1)); } /* Multiply (concatenate) two matrices and allocate the result */ @@ -758,10 +823,11 @@ rmx_multiply(const RMATRIX *m1, const RMATRIX *m2) for (i = mres->nrows; i--; ) for (j = mres->ncols; j--; ) for (k = mres->ncomp; k--; ) { - rmx_dtype d = 0; + double d = 0; for (h = m1->ncols; h--; ) - d += rmx_val(m1,i,h)[k] * rmx_val(m2,h,j)[k]; - rmx_lval(mres,i,j)[k] = d; + d += (double)rmx_val(m1,i,h)[k] * + rmx_val(m2,h,j)[k]; + rmx_lval(mres,i,j)[k] = (rmx_dtype)d; } return(mres); } @@ -853,7 +919,7 @@ rmx_sum(RMATRIX *msum, const RMATRIX *madd, const doub const rmx_dtype *da = rmx_val(madd,i,j); rmx_dtype *ds = rmx_lval(msum,i,j); for (k = msum->ncomp; k--; ) - ds[k] += sf[k] * da[k]; + ds[k] += (rmx_dtype)sf[k] * da[k]; } if (mysf) free(mysf); @@ -872,7 +938,7 @@ rmx_scale(RMATRIX *rm, const double sf[]) for (j = rm->ncols; j--; ) { rmx_dtype *dp = rmx_lval(rm,i,j); for (k = rm->ncomp; k--; ) - dp[k] *= sf[k]; + dp[k] *= (rmx_dtype)sf[k]; } if (rm->info) rmx_addinfo(rm, "Applied scalar\n"); @@ -904,10 +970,10 @@ rmx_transform(const RMATRIX *msrc, int n, const double for (j = dnew->ncols; j--; ) { const rmx_dtype *ds = rmx_val(msrc,i,j); for (kd = dnew->ncomp; kd--; ) { - rmx_dtype d = 0; + double d = 0; for (ks = msrc->ncomp; ks--; ) d += cmat[kd*msrc->ncomp + ks] * ds[ks]; - rmx_lval(dnew,i,j)[kd] = d; + rmx_lval(dnew,i,j)[kd] = (rmx_dtype)d; } } return(dnew); @@ -917,7 +983,6 @@ rmx_transform(const RMATRIX *msrc, int n, const double RMATRIX * rmx_from_cmatrix(const CMATRIX *cm) { - int i, j; RMATRIX *dnew; if (!cm) @@ -925,15 +990,23 @@ rmx_from_cmatrix(const CMATRIX *cm) dnew = rmx_alloc(cm->nrows, cm->ncols, 3); if (!dnew) return(NULL); - dnew->dtype = DTfloat; - for (i = dnew->nrows; i--; ) - for (j = dnew->ncols; j--; ) { - const COLORV *cv = cm_lval(cm,i,j); - rmx_dtype *dp = rmx_lval(dnew,i,j); - dp[0] = cv[0]; - dp[1] = cv[1]; - dp[2] = cv[2]; - } + + dnew->dtype = sizeof(COLORV)==sizeof(float) ? + DTfloat : DTdouble; + + if (sizeof(COLORV) == sizeof(rmx_dtype)) { + memcpy(dnew->mtx, cm->cmem, rmx_array_size(dnew)); + } else { + int i, j; + for (i = dnew->nrows; i--; ) + for (j = dnew->ncols; j--; ) { + const COLORV *cv = cm_lval(cm,i,j); + rmx_dtype *dp = rmx_lval(dnew,i,j); + dp[0] = cv[0]; + dp[1] = cv[1]; + dp[2] = cv[2]; + } + } return(dnew); } @@ -941,33 +1014,41 @@ rmx_from_cmatrix(const CMATRIX *cm) CMATRIX * cm_from_rmatrix(const RMATRIX *rm) { - int i, j; CMATRIX *cnew; - if (!rm || !rm->mtx | (rm->ncomp == 2)) + if (!rm || !rm->mtx | (rm->ncomp == 2) | (rm->ncomp > MAXCOMP)) return(NULL); cnew = cm_alloc(rm->nrows, rm->ncols); if (!cnew) return(NULL); - for (i = cnew->nrows; i--; ) - for (j = cnew->ncols; j--; ) { - const rmx_dtype *dp = rmx_val(rm,i,j); - COLORV *cv = cm_lval(cnew,i,j); - switch (rm->ncomp) { - case 3: - setcolor(cv, dp[0], dp[1], dp[2]); - break; - case 1: - setcolor(cv, dp[0], dp[0], dp[0]); - break; - default: { - SCOLOR scol; - int k; - for (k = rm->ncomp; k--; ) - scol[k] = dp[k]; - scolor2color(cv, scol, rm->ncomp, rm->wlpart); - } break; - } - } + if ((sizeof(COLORV) == sizeof(rmx_dtype)) & (rm->ncomp == 3)) { + memcpy(cnew->cmem, rm->mtx, rmx_array_size(rm)); + } else { + int i, j; + for (i = cnew->nrows; i--; ) + for (j = cnew->ncols; j--; ) { + const rmx_dtype *dp = rmx_val(rm,i,j); + COLORV *cv = cm_lval(cnew,i,j); + switch (rm->ncomp) { + case 3: + setcolor(cv, dp[0], dp[1], dp[2]); + break; + case 1: + setcolor(cv, dp[0], dp[0], dp[0]); + break; + default: + if (sizeof(COLORV) == sizeof(rmx_dtype)) { + scolor2color(cv, (const COLORV *)dp, + rm->ncomp, rm->wlpart); + } else { + COLORV scol[MAXCOMP]; + int k = rm->ncomp; + while (k--) scol[k] = dp[k]; + scolor2color(cv, scol, rm->ncomp, rm->wlpart); + } + break; + } + } + } return(cnew); }