--- ray/src/rt/rc2.c 2014/07/24 16:28:17 2.13 +++ ray/src/rt/rc2.c 2024/05/22 21:44:03 2.27 @@ -1,14 +1,15 @@ #ifndef lint -static const char RCSid[] = "$Id: rc2.c,v 2.13 2014/07/24 16:28:17 greg Exp $"; +static const char RCSid[] = "$Id: rc2.c,v 2.27 2024/05/22 21:44:03 greg Exp $"; #endif /* * Accumulate ray contributions for a set of materials * File i/o and recovery */ +#include +#include "platform.h" #include "rcontrib.h" #include "resolu.h" -#include /* Close output stream and free record */ static void @@ -112,9 +113,13 @@ printheader(FILE *fout, const char *info) printargs(gargc-1, gargv, fout); /* add our command */ fprintf(fout, "SOFTWARE= %s\n", VersionID); fputnow(fout); - fputs("NCOMP=3\n", fout); /* always RGB */ + fputncomp(NCSAMP, fout); + if (NCSAMP > 3) + fputwlsplit(WLPART, fout); if (info != NULL) /* add extra info if given */ fputs(info, fout); + if ((outfmt == 'f') | (outfmt == 'd')) + fputendian(fout); fputformat(formstr(outfmt), fout); fputc('\n', fout); /* empty line ends header */ } @@ -140,11 +145,15 @@ getostream(const char *ospec, const char *mname, int b LUENT *lep; STREAMOUT *sop; char *cp; - + + info[0] = '\0'; if (ospec == NULL) { /* use stdout? */ if (!noopen & !using_stdout) { if (outfmt != 'a') SET_FILE_BINARY(stdout); +#ifdef getc_unlocked + flockfile(stdout); /* avoid lock/unlock overhead */ +#endif if (header) { cp = info; if (yres > 0) { @@ -161,9 +170,6 @@ getostream(const char *ospec, const char *mname, int b if (waitflush > 0) fflush(stdout); stdos.xr = xres; stdos.yr = yres; -#ifdef getc_unlocked - flockfile(stdout); /* avoid lock/unlock overhead */ -#endif using_stdout = 1; } stdos.ofp = stdout; @@ -183,7 +189,7 @@ getostream(const char *ospec, const char *mname, int b sop = (STREAMOUT *)malloc(sizeof(STREAMOUT)); if (sop == NULL) error(SYSTEM, "out of memory in getostream"); - sop->outpipe = oname[0] == '!'; + sop->outpipe = (oname[0] == '!'); sop->reclen = 0; sop->ofp = NULL; /* open iff noopen==0 */ sop->xr = xres; sop->yr = yres; @@ -193,6 +199,12 @@ getostream(const char *ospec, const char *mname, int b errno = EEXIST; /* file exists */ goto openerr; } + } else if (noopen && outfmt == 'c' && /* stream exists to picture? */ + (sop->xr > 0) & (sop->yr > 0)) { + if (ofl & OF_BIN) + return(NULL); /* let caller offset bins */ + sprintf(errmsg, "output '%s' not a valid picture", oname); + error(WARNING, errmsg); } if (!noopen && sop->ofp == NULL) { /* open output stream */ if (oname[0] == '!') /* output to command */ @@ -261,12 +273,12 @@ getvec(FVECT vec) } break; case 'f': /* binary float */ - if (fread((char *)vf, sizeof(float), 3, stdin) != 3) + if (getbinary(vf, sizeof(float), 3, stdin) != 3) return(-1); VCOPY(vec, vf); break; case 'd': /* binary double */ - if (fread((char *)vd, sizeof(double), 3, stdin) != 3) + if (getbinary(vd, sizeof(double), 3, stdin) != 3) return(-1); VCOPY(vec, vd); break; @@ -279,46 +291,43 @@ getvec(FVECT vec) /* Put out ray contribution to file */ static void -put_contrib(const DCOLOR cnt, FILE *fout) +put_contrib(const DCOLORV *cnt, FILE *fout) { double sf = 1; - COLOR fv; - COLR cv; + SCOLOR fv; + SCOLR cv; + int i; if (accumulate > 1) sf = 1./(double)accumulate; switch (outfmt) { case 'a': - if (accumulate > 1) - fprintf(fout, "%.6e\t%.6e\t%.6e\t", - sf*cnt[0], sf*cnt[1], sf*cnt[2]); - else - fprintf(fout, "%.6e\t%.6e\t%.6e\t", - cnt[0], cnt[1], cnt[2]); + for (i = 0; i < NCSAMP; i++) + fprintf(fout, "%.6e\t", sf*cnt[i]); break; case 'f': - if (accumulate > 1) { - copycolor(fv, cnt); - scalecolor(fv, sf); - } else - copycolor(fv, cnt); - fwrite(fv, sizeof(float), 3, fout); + for (i = NCSAMP; i-- > 0; ) + fv[i] = cnt[i]; + if (accumulate > 1) + scalescolor(fv, sf); + putbinary(fv, sizeof(COLORV), NCSAMP, fout); break; case 'd': if (accumulate > 1) { - DCOLOR dv; - copycolor(dv, cnt); - scalecolor(dv, sf); - fwrite(dv, sizeof(double), 3, fout); + DCOLORV dv[MAXCSAMP]; + for (i = NCSAMP; i-- > 0; ) + dv[i] = sf*cnt[i]; + putbinary(dv, sizeof(DCOLORV), NCSAMP, fout); } else - fwrite(cnt, sizeof(double), 3, fout); + putbinary(cnt, sizeof(DCOLORV), NCSAMP, fout); break; case 'c': + for (i = NCSAMP; i-- > 0; ) + fv[i] = cnt[i]; if (accumulate > 1) - setcolr(cv, sf*cnt[0], sf*cnt[1], sf*cnt[2]); - else - setcolr(cv, cnt[0], cnt[1], cnt[2]); - fwrite(cv, sizeof(cv), 1, fout); + scalescolor(fv, sf); + scolor_scolr(cv, fv); + putbinary(cv, 1, LSCOLR, fout); break; default: error(INTERNAL, "botched output format"); @@ -330,18 +339,18 @@ put_contrib(const DCOLOR cnt, FILE *fout) void mod_output(MODCONT *mp) { - STREAMOUT *sop = getostream(mp->outspec, mp->modname, 0, 0); + STREAMOUT *sop = getostream(mp->outspec, mp->modname, mp->bin0, 0); int j; - put_contrib(mp->cbin[0], sop->ofp); + put_contrib(mp->cbin, sop->ofp); if (mp->nbins > 3 && /* minor optimization */ - sop == getostream(mp->outspec, mp->modname, 1, 0)) { + sop == getostream(mp->outspec, mp->modname, mp->bin0+1, 0)) { for (j = 1; j < mp->nbins; j++) - put_contrib(mp->cbin[j], sop->ofp); + put_contrib(mcbin(mp,j), sop->ofp); } else { for (j = 1; j < mp->nbins; j++) { - sop = getostream(mp->outspec, mp->modname, j, 0); - put_contrib(mp->cbin[j], sop->ofp); + sop = getostream(mp->outspec, mp->modname, mp->bin0+j, 0); + put_contrib(mcbin(mp,j), sop->ofp); } } } @@ -384,31 +393,36 @@ end_record() /* Get ray contribution from previous file */ static int -get_contrib(DCOLOR cnt, FILE *finp) +get_contrib(DCOLORV *cnt, FILE *finp) { - COLOR fv; - COLR cv; + SCOLOR fv; + SCOLR cv; + int i; switch (outfmt) { case 'a': - return(fscanf(finp,"%lf %lf %lf",&cnt[0],&cnt[1],&cnt[2]) == 3); - case 'f': - if (fread(fv, sizeof(fv[0]), 3, finp) != 3) - return(0); - copycolor(cnt, fv); + for (i = 0; i < NCSAMP; i++) + if (fscanf(finp, "%lf", &cnt[i]) != 1) + return(0); return(1); case 'd': - return(fread(cnt, sizeof(cnt[0]), 3, finp) == 3); + return(getbinary(cnt, sizeof(DCOLORV), NCSAMP, finp) == NCSAMP); + case 'f': + if (getbinary(fv, sizeof(COLORV), NCSAMP, finp) != NCSAMP) + return(0); + break; case 'c': - if (fread(cv, sizeof(cv), 1, finp) != 1) + if (getbinary(cv, 1, LSCOLR, finp) != LSCOLR) return(0); - colr_color(fv, cv); - copycolor(cnt, fv); - return(1); + scolr_scolor(fv, cv); + break; default: error(INTERNAL, "botched output format"); } - return(0); /* pro forma return */ + /* copy result from SCOLOR */ + for (i = NCSAMP; i-- > 0; ) + cnt[i] = fv[i]; + return(1); } @@ -438,12 +452,12 @@ reload_output() char *outvfmt; LUENT *oent; int xr, yr; - STREAMOUT sout; - DCOLOR rgbv; + STREAMOUT *sop; + DCOLORV contr[MAXCSAMP]; if (outfmt == 'a') fmode = "r"; - outvfmt = formstr(outfmt); + outvfmt = (char *)formstr(outfmt); /* reload modifier values */ for (i = 0; i < nmods; i++) { mp = (MODCONT *)lu_find(&modconttab,modname[i])->data; @@ -451,50 +465,43 @@ reload_output() error(USER, "cannot reload from stdout"); if (mp->outspec[0] == '!') error(USER, "cannot reload from command"); - for (j = 0; ; j++) { /* load each modifier bin */ - ofl = ofname(oname, mp->outspec, mp->modname, j); + for (j = 0; j < mp->nbins; j++) { /* load each modifier bin */ + ofl = ofname(oname, mp->outspec, mp->modname, mp->bin0+j); if (ofl < 0) error(USER, "bad output file specification"); oent = lu_find(&ofiletab, oname); - if (oent->data != NULL) { - sout = *(STREAMOUT *)oent->data; - } else { - sout.reclen = 0; - sout.outpipe = 0; - sout.xr = xres; sout.yr = yres; - sout.ofp = NULL; - } - if (sout.ofp == NULL) { /* open output as input */ - sout.ofp = fopen(oname, fmode); - if (sout.ofp == NULL) { - if (j == mp->nbins) - break; /* assume end of modifier */ + if (oent->data == NULL) + error(INTERNAL, "unallocated stream in reload_output()"); + sop = (STREAMOUT *)oent->data; + if (sop->ofp == NULL) { /* open output as input */ + sop->ofp = fopen(oname, fmode); + if (sop->ofp == NULL) { sprintf(errmsg, "missing reload file '%s'", oname); error(WARNING, errmsg); break; } #ifdef getc_unlocked - flockfile(sout.ofp); + flockfile(sop->ofp); #endif - if (header && checkheader(sout.ofp, outvfmt, NULL) != 1) { + if (header && checkheader(sop->ofp, outvfmt, NULL) != 1) { sprintf(errmsg, "format mismatch for '%s'", oname); error(USER, errmsg); } - if ((sout.xr > 0) & (sout.yr > 0) && - (!fscnresolu(&xr, &yr, sout.ofp) || - (xr != sout.xr) | - (yr != sout.yr))) { + if ((sop->reclen == 1) & (sop->xr > 0) & (sop->yr > 0) && + (!fscnresolu(&xr, &yr, sop->ofp) || + (xr != sop->xr) | + (yr != sop->yr))) { sprintf(errmsg, "resolution mismatch for '%s'", oname); error(USER, errmsg); } } - /* read in RGB value */ - if (!get_contrib(rgbv, sout.ofp)) { + /* read in spectral value */ + if (!get_contrib(contr, sop->ofp)) { if (!j) { - fclose(sout.ofp); + fclose(sop->ofp); break; /* ignore empty file */ } if (j < mp->nbins) { @@ -504,20 +511,7 @@ reload_output() } break; } - if (j >= mp->nbins) { /* check modifier size */ - sprintf(errmsg, - "mismatched -bn setting for reloading '%s'", - modname[i]); - error(USER, errmsg); - } - - copycolor(mp->cbin[j], rgbv); - if (oent->key == NULL) /* new file entry */ - oent->key = strcpy((char *) - malloc(strlen(oname)+1), oname); - if (oent->data == NULL) - oent->data = (char *)malloc(sizeof(STREAMOUT)); - *(STREAMOUT *)oent->data = sout; + memcpy(mcbin(mp,j), contr, DCOLORSIZ); } } lu_doall(&ofiletab, &myclose, NULL); /* close all files */ @@ -532,7 +526,7 @@ myseeko(const LUENT *e, void *p) off_t nbytes = *(off_t *)p; if (sop->reclen > 1) - nbytes = nbytes * sop->reclen; + nbytes *= (off_t)sop->reclen; if (fseeko(sop->ofp, nbytes, SEEK_CUR) < 0) { sprintf(errmsg, "seek error on file '%s'", e->key); error(SYSTEM, errmsg); @@ -553,7 +547,7 @@ recover_output() int ofl; char oname[1024]; LUENT *oent; - STREAMOUT sout; + STREAMOUT *sop; off_t nvals; int xr, yr; @@ -562,19 +556,19 @@ recover_output() error(USER, "cannot recover ASCII output"); return; case 'f': - outvsiz = sizeof(float)*3; + outvsiz = sizeof(float)*NCSAMP; break; case 'd': - outvsiz = sizeof(double)*3; + outvsiz = sizeof(double)*NCSAMP; break; case 'c': - outvsiz = sizeof(COLR); + outvsiz = LSCOLR; break; default: error(INTERNAL, "botched output format"); return; } - outvfmt = formstr(outfmt); + outvfmt = (char *)formstr(outfmt); /* check modifier outputs */ for (i = 0; i < nmods; i++) { mp = (MODCONT *)lu_find(&modconttab,modname[i])->data; @@ -582,87 +576,61 @@ recover_output() error(USER, "cannot recover from stdout"); if (mp->outspec[0] == '!') error(USER, "cannot recover from command"); - for (j = 0; ; j++) { /* check each bin's file */ - ofl = ofname(oname, mp->outspec, mp->modname, j); + for (j = 0; j < mp->nbins; j++) { /* check each bin's file */ + ofl = ofname(oname, mp->outspec, mp->modname, mp->bin0+j); if (ofl < 0) error(USER, "bad output file specification"); oent = lu_find(&ofiletab, oname); - if (oent->data != NULL) { - sout = *(STREAMOUT *)oent->data; - } else { - sout.reclen = 0; - sout.outpipe = 0; - sout.ofp = NULL; - } - if (sout.ofp != NULL) { /* already open? */ + if (oent->data == NULL) + error(INTERNAL, "unallocated stream in recover_output()"); + sop = (STREAMOUT *)oent->data; + if (sop->ofp != NULL) { /* already open? */ if (ofl & OF_BIN) continue; break; } /* open output */ - sout.ofp = fopen(oname, "rb+"); - if (sout.ofp == NULL) { - if (j == mp->nbins) - break; /* assume end of modifier */ + sop->ofp = fopen(oname, "rb+"); + if (sop->ofp == NULL) { sprintf(errmsg, "missing recover file '%s'", oname); error(WARNING, errmsg); + lastout = 0; break; } - nvals = lseek(fileno(sout.ofp), 0, SEEK_END); + nvals = lseek(fileno(sop->ofp), 0, SEEK_END); if (nvals <= 0) { lastout = 0; /* empty output, quit here */ - fclose(sout.ofp); + fclose(sop->ofp); break; } - if (!sout.reclen) { - if (!(ofl & OF_BIN)) { - sprintf(errmsg, - "need -bn to recover file '%s'", - oname); - error(USER, errmsg); - } - recsiz = outvsiz; - } else - recsiz = outvsiz * sout.reclen; + recsiz = outvsiz * sop->reclen; - lseek(fileno(sout.ofp), 0, SEEK_SET); - if (header && checkheader(sout.ofp, outvfmt, NULL) != 1) { + lseek(fileno(sop->ofp), 0, SEEK_SET); + if (header && checkheader(sop->ofp, outvfmt, NULL) != 1) { sprintf(errmsg, "format mismatch for '%s'", oname); error(USER, errmsg); } - sout.xr = xres; sout.yr = yres; - if ((sout.xr > 0) & (sout.yr > 0) && - (!fscnresolu(&xr, &yr, sout.ofp) || - (xr != sout.xr) | - (yr != sout.yr))) { + if ((sop->reclen == 1) & (sop->xr > 0) & (sop->yr > 0) && + (!fscnresolu(&xr, &yr, sop->ofp) || + (xr != sop->xr) | + (yr != sop->yr))) { sprintf(errmsg, "resolution mismatch for '%s'", oname); error(USER, errmsg); } - nvals = (nvals - (off_t)ftell(sout.ofp)) / recsiz; + nvals = (nvals - (off_t)ftell(sop->ofp)) / recsiz; if ((lastout < 0) | (nvals < lastout)) lastout = nvals; - if (oent->key == NULL) /* new entry */ - oent->key = strcpy((char *) - malloc(strlen(oname)+1), oname); - if (oent->data == NULL) - oent->data = (char *)malloc(sizeof(STREAMOUT)); - *(STREAMOUT *)oent->data = sout; if (!(ofl & OF_BIN)) break; /* no bin separation */ } if (!lastout) { /* empty output */ error(WARNING, "no previous data to recover"); - lu_done(&ofiletab); /* reclose all outputs */ + /* reclose all outputs */ + lu_doall(&ofiletab, &myclose, NULL); return; - } - if (j > mp->nbins) { /* check modifier size */ - sprintf(errmsg, - "mismatched -bn setting for recovering '%s'", - modname[i]); - error(USER, errmsg); } } if (lastout < 0) {