ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rcomb.c
Revision: 2.19
Committed: Mon Jun 3 18:55:51 2024 UTC (11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.18: +57 -58 lines
Log Message:
refactor: Changed buffer types in process(), readbuf(), writebuf()

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rcomb.c,v 2.18 2024/05/23 19:29:41 greg Exp $";
3 #endif
4 /*
5 * General component matrix combiner, operating on a row at a time.
6 *
7 * Multi-processing mode under Unix creates children that each work
8 * on one input row at a time, fed by the original process. Final conversion
9 * and output to stdout is sorted by first child while its siblings send it
10 * their record calculations.
11 */
12
13 #include <math.h>
14 #include "platform.h"
15 #include "rtprocess.h"
16 #include "rtio.h"
17 #include "rmatrix.h"
18 #include "calcomp.h"
19
20 #ifndef M_PI
21 #define M_PI 3.14159265358979323846
22 #endif
23
24 #define MAXCOMP MAXCSAMP /* #components we support */
25
26 /* Unary matrix operation(s) */
27 typedef struct {
28 double cmat[MAXCOMP*MAXCOMP]; /* component transformation */
29 double sca[MAXCOMP]; /* scalar coefficients */
30 const char *csym; /* symbolic coefficients */
31 short clen; /* number of coefficients */
32 short nsf; /* number of scalars */
33 } RUNARYOP;
34
35 /* Input matrix */
36 typedef struct {
37 const char *inspec; /* input specification */
38 RUNARYOP preop; /* transform operation */
39 RMATRIX imx; /* input matrix header info */
40 RMATRIX *rmp; /* active single-row matrix */
41 FILE *infp; /* open input stream */
42 } ROPMAT;
43
44 ROPMAT *mop = NULL; /* allocated input array */
45 int nall = 0; /* number allocated */
46 int nmats = 0; /* number of actual inputs */
47
48 RMATRIX *mcat = NULL; /* final concatenation */
49 int mcat_last = 0; /* goes after trailing ops? */
50
51 int in_nrows; /* number of input rows (or 0) */
52 #define in_ncols (mop[0].rmp->ncols) /* number of input columns */
53 #define in_ncomp (mop[0].rmp->ncomp) /* input #components */
54
55 extern int nowarn; /* turn off warnings? */
56
57 int cur_row; /* current input/output row */
58 int cur_col; /* current input/output column */
59 int cur_chan; /* if we're looping channels */
60
61 SUBPROC *cproc = NULL; /* child process array */
62 int nchildren = 0; /* # of child processes */
63 int inchild = -1; /* our child ID (-1: parent) */
64
65 static int checksymbolic(ROPMAT *rop);
66
67 static int
68 split_input(ROPMAT *rop)
69 {
70 if (rop->rmp == &rop->imx && !(rop->rmp = rmx_copy(&rop->imx))) {
71 fputs("Out of memory in split_input()\n", stderr);
72 return(0);
73 }
74 rmx_reset(rop->rmp);
75 return(1);
76 }
77
78 /* Check/set transform based on a reference input file */
79 static int
80 checkreffile(ROPMAT *rop)
81 {
82 static const char *curRF = NULL;
83 static RMATRIX refm;
84 const int nc = rop->imx.ncomp;
85 int i;
86
87 if (!curRF || strcmp(rop->preop.csym, curRF)) {
88 FILE *fp = fopen(rop->preop.csym, "rb");
89 if (!rmx_load_header(&refm, fp)) {
90 fprintf(stderr, "%s: cannot read info header\n",
91 rop->preop.csym);
92 curRF = NULL;
93 if (fp) fclose(fp);
94 return(0);
95 }
96 fclose(fp);
97 curRF = rop->preop.csym;
98 }
99 if (refm.ncomp == 3) {
100 rop->preop.csym = (refm.dtype == DTxyze) ? "XYZ" : "RGB";
101 return(checksymbolic(rop));
102 }
103 if (refm.ncomp == 2) {
104 fprintf(stderr, "%s: cannot convert to 2 components\n",
105 curRF);
106 return(0);
107 }
108 if (refm.ncomp == 1) {
109 rop->preop.csym = "Y"; /* XXX big assumption */
110 return(checksymbolic(rop));
111 }
112 if (refm.ncomp == nc &&
113 !memcmp(refm.wlpart, rop->imx.wlpart, sizeof(refm.wlpart)))
114 return(1); /* nothing to do */
115
116 if ((nc <= 3) | (nc > MAXCSAMP) | (refm.ncomp > MAXCSAMP)) {
117 fprintf(stderr, "%s: cannot resample from %d to %d components\n",
118 curRF, nc, refm.ncomp);
119 return(0);
120 }
121 if (!split_input(rop)) /* get our own struct */
122 return(0);
123 rop->preop.clen = refm.ncomp * nc; /* compute spec to ref */
124
125 for (i = 0; i < nc; i++) {
126 SCOLOR scstim, scresp;
127 int j;
128 memset(scstim, 0, sizeof(COLORV)*nc);
129 scstim[i] = 1.f;
130 convertscolor(scresp, refm.ncomp, refm.wlpart[0], refm.wlpart[3],
131 scstim, nc, rop->imx.wlpart[0], rop->imx.wlpart[3]);
132 for (j = refm.ncomp; j-- > 0; )
133 rop->preop.cmat[j*nc + i] = scresp[j];
134 }
135 /* remember new spectral params */
136 memcpy(rop->rmp->wlpart, refm.wlpart, sizeof(rop->rmp->wlpart));
137 rop->rmp->ncomp = refm.ncomp;
138 return(1);
139 }
140
141 /* Compute conversion row from spectrum to one channel of RGB */
142 static void
143 rgbrow(ROPMAT *rop, int r, int p)
144 {
145 const int nc = rop->imx.ncomp;
146 const float * wlp = rop->imx.wlpart;
147 int i;
148
149 for (i = nc; i--; ) {
150 int nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
151 int nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
152 COLOR crgb;
153 spec_rgb(crgb, nmStart, nmEnd);
154 rop->preop.cmat[r*nc+i] = crgb[p];
155 }
156 }
157
158 /* Compute conversion row from spectrum to one channel of XYZ */
159 static void
160 xyzrow(ROPMAT *rop, int r, int p)
161 {
162 const int nc = rop->imx.ncomp;
163 const float * wlp = rop->imx.wlpart;
164 int i;
165
166 for (i = nc; i--; ) {
167 int nmEnd = wlp[0] + (wlp[3] - wlp[0])*i/nc;
168 int nmStart = wlp[0] + (wlp[3] - wlp[0])*(i+1)/nc;
169 COLOR cxyz;
170 spec_cie(cxyz, nmStart, nmEnd);
171 rop->preop.cmat[r*nc+i] = cxyz[p];
172 }
173 }
174
175 /* Use the spectral sensitivity function to compute matrix coefficients */
176 static void
177 sensrow(ROPMAT *rop, int r, double (*sf)(SCOLOR sc, int ncs, const float wlpt[4]))
178 {
179 const int nc = rop->imx.ncomp;
180 int i;
181
182 for (i = nc; i--; ) {
183 SCOLOR sclr;
184 memset(sclr, 0, sizeof(COLORV)*nc);
185 sclr[i] = 1.f;
186 rop->preop.cmat[r*nc+i] = (*sf)(sclr, nc, rop->imx.wlpart);
187 }
188 }
189
190 /* Check/set symbolic transform */
191 static int
192 checksymbolic(ROPMAT *rop)
193 {
194 const int nc = rop->imx.ncomp;
195 const int dt = rop->imx.dtype;
196 double cf = 1;
197 int i, j;
198 /* check suffix => reference file */
199 if (strchr(rop->preop.csym, '.') > rop->preop.csym)
200 return(checkreffile(rop));
201
202 if (nc < 3) {
203 fprintf(stderr, "%s: -c '%s' requires at least 3 components\n",
204 rop->inspec, rop->preop.csym);
205 return(0);
206 }
207 rop->preop.clen = strlen(rop->preop.csym) * nc;
208 if (rop->preop.clen > MAXCOMP*MAXCOMP) {
209 fprintf(stderr, "%s: -c '%s' results in too many components\n",
210 rop->inspec, rop->preop.csym);
211 return(0);
212 }
213 for (j = 0; rop->preop.csym[j]; j++) {
214 int comp = 0;
215 switch (rop->preop.csym[j]) {
216 case 'B':
217 case 'b':
218 ++comp;
219 /* fall through */
220 case 'G':
221 case 'g':
222 ++comp;
223 /* fall through */
224 case 'R':
225 case 'r':
226 if (rop->preop.csym[j] <= 'Z')
227 cf = 1./WHTEFFICACY;
228 if (dt == DTxyze) {
229 for (i = 3; i--; )
230 rop->preop.cmat[j*nc+i] = cf*xyz2rgbmat[comp][i];
231 } else if (nc == 3)
232 rop->preop.cmat[j*nc+comp] = 1.;
233 else
234 rgbrow(rop, j, comp);
235 break;
236 case 'Z':
237 case 'z':
238 ++comp;
239 /* fall through */
240 case 'Y':
241 case 'y':
242 ++comp;
243 /* fall through */
244 case 'X':
245 case 'x':
246 if ((rop->preop.csym[j] <= 'Z') & (dt != DTxyze))
247 cf = WHTEFFICACY;
248 if (dt == DTxyze) {
249 rop->preop.cmat[j*nc+comp] = 1.;
250 } else if (nc == 3) {
251 for (i = 3; i--; )
252 rop->preop.cmat[j*nc+i] =
253 rgb2xyzmat[comp][i];
254 } else if (comp == CIEY)
255 sensrow(rop, j, scolor2photopic);
256 else
257 xyzrow(rop, j, comp);
258
259 for (i = nc*(cf != 1); i--; )
260 rop->preop.cmat[j*nc+i] *= cf;
261 break;
262 case 'S': /* scotopic (il)luminance */
263 cf = WHTSCOTOPIC;
264 /* fall through */
265 case 's':
266 sensrow(rop, j, scolor2scotopic);
267 for (i = nc*(cf != 1); i--; )
268 rop->preop.cmat[j*nc+i] *= cf;
269 break;
270 case 'M': /* melanopic (il)luminance */
271 cf = WHTMELANOPIC;
272 /* fall through */
273 case 'm':
274 sensrow(rop, j, scolor2melanopic);
275 for (i = nc*(cf != 1); i--; )
276 rop->preop.cmat[j*nc+i] *= cf;
277 break;
278 case 'A': /* average component */
279 case 'a':
280 for (i = nc; i--; )
281 rop->preop.cmat[j*nc+i] = 1./(double)nc;
282 break;
283 default:
284 fprintf(stderr, "%s: -c '%c' unsupported\n",
285 rop->inspec, rop->preop.csym[j]);
286 return(0);
287 }
288 }
289 if (!split_input(rop)) /* get our own struct */
290 return(0);
291 memcpy(rop->rmp->wlpart, WLPART, sizeof(rop->rmp->wlpart));
292 rop->rmp->ncomp = rop->preop.clen / nc;
293 /* decide on output type */
294 if (!strcasecmp(rop->preop.csym, "XYZ")) {
295 if (dt <= DTspec)
296 rop->rmp->dtype = DTxyze;
297 } else if (!strcasecmp(rop->preop.csym, "RGB")) {
298 if (dt <= DTspec)
299 rop->rmp->dtype = DTrgbe;
300 } else if (rop->rmp->dtype == DTspec)
301 rop->rmp->dtype = DTfloat;
302 return(1);
303 }
304
305 static int
306 get_component_xfm(ROPMAT *rop)
307 {
308 int i, j;
309
310 if (rop->rmp != &rop->imx) { /* reset destination matrix */
311 rmx_free(rop->rmp);
312 rop->rmp = &rop->imx;
313 }
314 if (rop->preop.csym && /* symbolic transform? */
315 !checksymbolic(rop))
316 return(0);
317 /* undo exposure? */
318 if (fabs(1. - bright(rop->rmp->cexp)) > .025) {
319 if (rop->rmp->ncomp == 1)
320 rop->rmp->cexp[RED] = rop->rmp->cexp[GRN] =
321 rop->rmp->cexp[BLU] = bright(rop->rmp->cexp);
322 if (rop->preop.nsf <= 0) {
323 rop->preop.nsf = i = rop->rmp->ncomp;
324 while (i--)
325 rop->preop.sca[i] = 1.;
326 }
327 if (rop->preop.nsf == 1) {
328 if (rop->rmp->ncomp == 3) {
329 rop->preop.sca[2] = rop->preop.sca[1] =
330 rop->preop.sca[0];
331 rop->preop.nsf = 3;
332 } else
333 rop->preop.sca[0] /= bright(rop->rmp->cexp);
334 }
335 if (rop->preop.nsf == 3) {
336 opcolor(rop->preop.sca, /=, rop->rmp->cexp);
337 } else if (rop->preop.nsf > 3) { /* punt */
338 double mult = 1./bright(rop->rmp->cexp);
339 for (i = rop->preop.nsf; i--; )
340 rop->preop.sca[i] *= mult;
341 }
342 setcolor(rop->rmp->cexp, 1., 1., 1.);
343 }
344 if (rop->preop.clen > 0) { /* use component transform? */
345 if (rop->preop.clen % rop->imx.ncomp) {
346 fprintf(stderr, "%s: -c must have N x %d coefficients\n",
347 rop->inspec, rop->imx.ncomp);
348 return(0);
349 }
350 if (rop->preop.nsf > 0) { /* scale transform, instead */
351 if (rop->preop.nsf == 1) {
352 for (i = rop->preop.clen; i--; )
353 rop->preop.cmat[i] *= rop->preop.sca[0];
354 } else if (rop->preop.nsf*rop->imx.ncomp != rop->preop.clen) {
355 fprintf(stderr, "%s: -s must have one or %d factors\n",
356 rop->inspec,
357 rop->preop.clen/rop->imx.ncomp);
358 return(0);
359 } else {
360 for (i = rop->preop.nsf; i--; )
361 for (j = rop->imx.ncomp; j--; )
362 rop->preop.cmat[i*rop->imx.ncomp+j]
363 *= rop->preop.sca[i];
364 }
365 }
366 rop->preop.nsf = 0; /* now folded in */
367 if (!split_input(rop)) /* get our own struct */
368 return(0);
369 rop->rmp->ncomp = rop->preop.clen / rop->imx.ncomp;
370 if ((rop->rmp->ncomp > 3) & (rop->rmp->dtype <= DTspec)) {
371 rop->rmp->dtype = DTfloat; /* probably not actual spectrum */
372 memcpy(rop->rmp->wlpart, WLPART, sizeof(rop->rmp->wlpart));
373 }
374 } else if (rop->preop.nsf > 0) { /* else use scalar(s)? */
375 if (rop->preop.nsf == 1) {
376 for (i = rop->rmp->ncomp; --i; )
377 rop->preop.sca[i] = rop->preop.sca[0];
378 rop->preop.nsf = rop->rmp->ncomp;
379 } else if (rop->preop.nsf != rop->rmp->ncomp) {
380 fprintf(stderr, "%s: -s must have one or %d factors\n",
381 rop->inspec, rop->rmp->ncomp);
382 return(0);
383 }
384 }
385 return(1);
386 }
387
388 static int
389 apply_op(RMATRIX *dst, const RMATRIX *src, const RUNARYOP *ro)
390 {
391 if (ro->clen > 0) {
392 RMATRIX *res = rmx_transform(src, dst->ncomp, ro->cmat);
393 if (!res) {
394 fputs("Error in call to rmx_transform()\n", stderr);
395 return(0);
396 }
397 if (!rmx_transfer_data(dst, res, 0))
398 return(0);
399 rmx_free(res);
400 } else if (dst != src)
401 memcpy(dst->mtx, src->mtx, rmx_array_size(dst));
402 if (ro->nsf == dst->ncomp)
403 rmx_scale(dst, ro->sca);
404 return(1);
405 }
406
407 static int
408 open_input(ROPMAT *rop)
409 {
410 int outtype;
411
412 if (!rop || !rop->inspec || !rop->inspec[0])
413 return(0);
414 if (rop->inspec == stdin_name)
415 rop->infp = stdin;
416 else if (rop->inspec[0] == '!')
417 rop->infp = popen(rop->inspec+1, "r");
418 else
419 rop->infp = fopen(rop->inspec, "rb");
420
421 if (!rmx_load_header(&rop->imx, rop->infp)) {
422 fprintf(stderr, "Bad header from: %s\n", rop->inspec);
423 return(0);
424 }
425 return(get_component_xfm(rop));
426 }
427
428 /* Return nominal wavelength associated with input component (return nm) */
429 static double
430 l_wavelength(char *nam)
431 {
432 double comp = argument(1);
433
434 if ((comp < -.5) | (comp >= in_ncomp+.5)) {
435 errno = EDOM;
436 return(.0);
437 }
438 if (comp < .5) /* asking for #components? */
439 return(in_ncomp);
440
441 if (in_ncomp == 3) { /* special case for RGB */
442 const int w0 = (int)(comp - .5);
443 return(mop[0].rmp->wlpart[w0] +
444 (comp-.5)*(mop[0].rmp->wlpart[w0+1] -
445 mop[0].rmp->wlpart[w0]));
446 }
447 return(mop[0].rmp->wlpart[0] + /* general case, even div. */
448 (comp-.5)/(double)in_ncomp *
449 (mop[0].rmp->wlpart[3] - mop[0].rmp->wlpart[0]));
450 }
451
452 /* Return ith input with optional channel selector */
453 static double
454 l_chanin(char *nam)
455 {
456 double inp = argument(1);
457 int mi, chan;
458
459 if ((mi = (int)(inp-.5)) < 0 || mi >= nmats) {
460 errno = EDOM;
461 return(.0);
462 }
463 if (inp < .5) /* asking for #inputs? */
464 return(nmats);
465
466 if (nargum() >= 2) {
467 double cval = argument(2);
468 if (cval < .5 || (chan = (int)(cval-.5)) >= in_ncomp) {
469 errno = EDOM;
470 return(.0);
471 }
472 } else
473 chan = cur_chan;
474
475 return(mop[mi].rmp->mtx[cur_col*in_ncomp + chan]);
476 }
477
478 static int
479 initialize(RMATRIX *imp)
480 {
481 int i;
482 /* XXX struct is zeroed coming in */
483 setcolor(imp->cexp, 1.f, 1.f, 1.f);
484 for (i = 0; i < nmats; i++) { /* open each input */
485 int restype;
486 if (!open_input(&mop[i]))
487 return(0);
488 restype = mop[i].rmp->dtype;
489 if (!imp->dtype || (restype = rmx_newtype(restype, imp->dtype)) > 0)
490 imp->dtype = restype;
491 else
492 fprintf(stderr, "%s: warning - data type mismatch\n",
493 mop[i].inspec);
494 if (!i) {
495 imp->ncols = mop[0].rmp->ncols;
496 imp->ncomp = mop[0].rmp->ncomp;
497 memcpy(imp->wlpart, mop[0].rmp->wlpart, sizeof(imp->wlpart));
498 } else if ((mop[i].rmp->ncols != imp->ncols) |
499 (mop[i].rmp->ncomp != imp->ncomp) |
500 ((in_nrows > 0) & (mop[i].rmp->nrows > 0) &
501 (mop[i].rmp->nrows != in_nrows))) {
502 fprintf(stderr, "%s: mismatch in size or #components\n",
503 mop[i].inspec);
504 return(0);
505 } /* XXX should check wlpart? */
506 if (in_nrows <= 0)
507 in_nrows = imp->nrows = mop[i].rmp->nrows;
508 } /* set up .cal environment */
509 esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
510 esupport &= ~(E_OUTCHAN|E_INCHAN);
511 varset("PI", ':', M_PI);
512 varset("nfiles", ':', nmats);
513 varset("nrows", ':', in_nrows);
514 varset("ncols", ':', in_ncols);
515 varset("ncomp", ':', in_ncomp);
516 varset("R", ':', 1.);
517 varset("G", ':', 2.);
518 varset("B", ':', 3.);
519 funset("wl", 1, ':', l_wavelength);
520 funset("ci", 1, '=', l_chanin);
521 scompile("ri(i)=ci(i,R);gi(i)=ci(i,G);bi(i)=ci(i,B)", NULL, 0);
522 return(1);
523 }
524
525 static void
526 output_headinfo(FILE *fp)
527 {
528 int i;
529
530 for (i = 0; i < nmats; i++) {
531 const char *cp = mop[i].imx.info;
532 fputs(mop[i].inspec, fp);
533 fputs(":\n", fp);
534 if (!cp) continue;
535 while (*cp) {
536 if (*cp == '\n') {
537 cp++; /* avoid inadvertant terminus */
538 continue;
539 }
540 fputc('\t', fp); /* indent this input's info */
541 do
542 putc(*cp, fp);
543 while (*cp++ != '\n');
544 }
545 }
546 }
547
548 static void
549 reset_inputs(void)
550 {
551 int i = nmats;
552
553 fpurge(stdin); /* discard previous matrix input */
554 while (i--) {
555 if (mop[i].infp != stdin)
556 fclose(mop[i].infp); /* ! pclose() */
557 mop[i].infp = stdin;
558 mop[i].imx.dtype = DTdouble;
559 mop[i].imx.pflags &= ~RMF_SWAPIN;
560 }
561 }
562
563 static int
564 spawned_children(int np)
565 {
566 size_t recsize = 0;
567 int i, rv;
568
569 #if defined(_WIN32) || defined(_WIN64)
570 if (np > 1) {
571 fputs("Warning: only one process under Windows\n", stderr);
572 np = 1;
573 } else
574 #endif
575 if ((in_nrows > 0) & (np*4 > in_nrows))
576 np = in_nrows/4;
577 /* we'll be doing a row at a time */
578 for (i = 0; i < nmats; i++) {
579 mop[i].imx.nrows = 1;
580 if (!rmx_prepare(&mop[i].imx))
581 goto memerror;
582 recsize += rmx_array_size(&mop[i].imx);
583 if (mop[i].rmp != &mop[i].imx) {
584 mop[i].rmp->nrows = 1;
585 if (!rmx_prepare(mop[i].rmp))
586 goto memerror;
587 }
588 }
589 /* prep output row buffer */
590 if (mcat || mop[nmats].preop.clen > 0) {
591 if (!split_input(&mop[nmats])) /* need separate buffer */
592 return(0);
593 if (mop[nmats].preop.clen > 0)
594 mop[nmats].rmp->ncomp = mop[nmats].preop.clen /
595 mop[nmats].imx.ncomp;
596 mop[nmats].rmp->nrows = 1;
597 if (!mcat | !mcat_last && !rmx_prepare(mop[nmats].rmp))
598 goto memerror;
599 }
600 mop[nmats].imx.nrows = 1;
601 if (!rmx_prepare(&mop[nmats].imx))
602 goto memerror;
603 if (np <= 1) { /* single process return */
604 #ifdef getc_unlocked
605 for (i = 0; i < nmats; i++)
606 flockfile(mop[i].infp);
607 flockfile(stdout);
608 #endif
609 return(0);
610 }
611 fflush(stdout); /* flush header & spawn children */
612 nchildren = np+1;
613 cproc = (SUBPROC *)malloc(sizeof(SUBPROC)*nchildren);
614 if (!cproc)
615 goto memerror;
616 /* first child becomes stdout filter */
617 cproc[0].flags = PF_FILT_OUT;
618 cproc[0].w = fileno(stdout);
619 cproc[0].r = 0;
620 cproc[0].pid = -1;
621 rv = open_process(&cproc[i], NULL);
622 if (rv < 0) {
623 perror("fork");
624 exit(1);
625 }
626 if (rv == 0) { /* loop if first child */
627 inchild = 0;
628 reset_inputs();
629 output_loop();
630 _exit(0);
631 }
632 /* start other children */
633 for (i = 1; (i < nchildren) & (rv > 0); i++) {
634 cproc[i].flags = 0;
635 cproc[i].w = 1;
636 cproc[i].r = 0;
637 cproc[i].pid = -1;
638 rv = open_process(&cproc[i], NULL);
639 }
640 if (rv < 0) {
641 perror("fork");
642 close_processes(cproc, i);
643 exit(1);
644 }
645 if (rv > 0) /* parent return? */
646 return(1);
647 inchild = i; /* our child index */
648 while (--i > 0) /* don't share siblings' pipes */
649 close(cproc[i].w);
650 reset_inputs();
651 #ifdef getc_unlocked
652 flockfile(stdin);
653 #endif
654 return(0); /* child return */
655 memerror:
656 fputs("Out of memory in spawned_children()\n", stderr);
657 exit(1);
658 }
659
660 static int
661 parent_loop(void)
662 {
663 int i;
664
665 rmx_reset(&mop[nmats].imx); /* not doing output side */
666 if (mop[nmats].rmp != &mop[nmats].imx)
667 rmx_reset(mop[nmats].rmp);
668 #ifdef getc_unlocked
669 for (i = 0; i < nmats; i++)
670 flockfile(mop[i].infp);
671 #endif
672 /* load & send rows to kids */
673 for (cur_row = 0; (in_nrows <= 0) | (cur_row < in_nrows); cur_row++) {
674 int wfd = cproc[1 + cur_row%(nchildren-1)].w;
675 for (i = 0; i < nmats; i++)
676 if (!rmx_load_row(mop[i].imx.mtx, &mop[i].imx, mop[i].infp)) {
677 if (cur_row > in_nrows) /* unknown #input rows? */
678 break;
679 fprintf(stderr, "%s: read error at row %d\n",
680 mop[i].inspec, cur_row);
681 return(0);
682 }
683 if (i < nmats)
684 break;
685 for (i = 0; i < nmats; i++)
686 if (writebuf(wfd, mop[i].imx.mtx, rmx_array_size(&mop[i].imx))
687 != rmx_array_size(&mop[i].imx))
688 return(0);
689 }
690 i = close_processes(cproc, nchildren);
691 free(cproc); cproc = NULL;
692 if (i < 0) {
693 fputs("Warning: missing child in parent_loop()\n", stderr);
694 return(1);
695 }
696 if (i > 0) {
697 fprintf(stderr, "Child exited with status %d\n", i);
698 return(0);
699 }
700 return(1);
701 memerror:
702 fputs("Out of memory in parent_loop()\n", stderr);
703 exit(1);
704 }
705
706 static int
707 combine_input(void)
708 {
709 const int row0 = (inchild > 0)*(inchild-1);
710 const int rstep = nchildren + !nchildren;
711 ROPMAT *res = &mop[nmats];
712 int set_r, set_c;
713 RMATRIX *tmp = NULL;
714 int co_set;
715 int i;
716
717 if (mcat && mcat_last &&
718 !(tmp = rmx_alloc(1, res->imx.ncols, res->rmp->ncomp)))
719 goto memerror;
720 /* figure out what the user set */
721 co_set = fundefined("co");
722 if (!co_set)
723 co_set = -vardefined("co");
724 if (!co_set & (in_ncomp == 3) && vardefined("ro") &&
725 vardefined("go") && vardefined("bo")) {
726 scompile("co(p)=select(p,ro,go,bo)", NULL, 0);
727 co_set = 1;
728 }
729 if (co_set) { /* set if user wants, didn't set */
730 set_r = varlookup("r") != NULL && !vardefined("r");
731 set_c = varlookup("c") != NULL && !vardefined("c");
732 } else /* save a little time */
733 set_r = set_c = 0;
734 /* read/process row-by-row */
735 for (cur_row = row0; (in_nrows <= 0) | (cur_row < in_nrows); cur_row += rstep) {
736 RMATRIX *mres = NULL;
737 for (i = 0; i < nmats; i++)
738 if (!rmx_load_row(mop[i].imx.mtx, &mop[i].imx, mop[i].infp)) {
739 if (cur_row > in_nrows) /* unknown #input rows? */
740 break;
741 fprintf(stderr, "%s: read error at row %d\n",
742 mop[i].inspec, cur_row);
743 return(0);
744 }
745 if (i < nmats)
746 break;
747 for (i = 0; i < nmats; i++)
748 if (!apply_op(mop[i].rmp, &mop[i].imx, &mop[i].preop))
749 return(0);
750 if (set_r) varset("r", '=', cur_row);
751 for (cur_col = 0; cur_col < in_ncols; cur_col++) {
752 if (set_c) varset("c", '=', cur_col);
753 for (cur_chan = 0; cur_chan < in_ncomp; cur_chan++) {
754 const int ndx = cur_col*in_ncomp + cur_chan;
755 eclock++;
756 if (!co_set) { /* just summing elements? */
757 res->imx.mtx[ndx] = 0;
758 for (i = nmats; i--; )
759 res->imx.mtx[ndx] += mop[i].rmp->mtx[ndx];
760 } else if (co_set > 0) {
761 double dchan = cur_chan+1;
762 res->imx.mtx[ndx] = funvalue("co", 1, &dchan);
763 } else
764 res->imx.mtx[ndx] = varvalue("co");
765 }
766 } /* final conversions */
767 if (!mcat) {
768 if (!apply_op(res->rmp, &res->imx, &res->preop))
769 return(0);
770 } else if (mcat_last) {
771 if (!apply_op(tmp, &res->imx, &res->preop))
772 return(0);
773 mres = rmx_multiply(tmp, mcat);
774 if (!mres)
775 goto multerror;
776 if (!rmx_transfer_data(res->rmp, mres, 0))
777 return(0);
778 } else /* mcat && !mcat_last */ {
779 mres = rmx_multiply(&res->imx, mcat);
780 if (!mres)
781 goto multerror;
782 if (!apply_op(res->rmp, mres, &res->preop))
783 return(0);
784 }
785 rmx_free(mres); mres = NULL;
786 if (inchild > 0) { /* children share stdout */
787 i = getc(stdin); /* signals it's our turn */
788 if (i != EOF) ungetc(i, stdin);
789 }
790 if (!rmx_write_data(res->rmp->mtx, res->rmp->ncomp,
791 res->rmp->ncols, res->rmp->dtype, stdout))
792 return(0);
793 if (inchild > 0 && fflush(stdout) == EOF)
794 return(0);
795 }
796 return(inchild > 0 || fflush(stdout) != EOF);
797 memerror:
798 fputs("Out of buffer space in combine_input()\n", stderr);
799 return(0);
800 multerror:
801 fputs("Unexpected matrix multiply error in combine_input()\n", stderr);
802 return(0);
803 }
804
805 static int
806 get_factors(double da[], int n, char *av[])
807 {
808 int ac;
809
810 for (ac = 0; ac < n && isflt(av[ac]); ac++)
811 da[ac] = atof(av[ac]);
812 return(ac);
813 }
814
815 static void
816 resize_inparr(int n2alloc)
817 {
818 int i;
819
820 if (n2alloc == nall)
821 return;
822 for (i = nall; i > n2alloc; i--) {
823 rmx_reset(&mop[i].imx);
824 if (mop[i].rmp != &mop[i].imx)
825 rmx_free(mop[i].rmp);
826 }
827 mop = (ROPMAT *)realloc(mop, n2alloc*sizeof(ROPMAT));
828 if (mop == NULL) {
829 fputs("Out of memory in resize_inparr()\n", stderr);
830 exit(1);
831 }
832 if (n2alloc > nall)
833 memset(mop+nall, 0, (n2alloc-nall)*sizeof(ROPMAT));
834 nall = n2alloc;
835 }
836
837 /* Load one or more matrices and operate on them, sending results to stdout */
838 int
839 main(int argc, char *argv[])
840 {
841
842 int outfmt = DTfromHeader;
843 const char *defCsym = NULL;
844 int echoheader = 1;
845 int stdin_used = 0;
846 int nproc = 1;
847 const char *mcat_spec = NULL;
848 int n2comp = 0;
849 uby8 comp_ndx[128];
850 int i;
851 /* get starting input array */
852 mop = (ROPMAT *)calloc(nall=2, sizeof(ROPMAT));
853 /* get options and arguments */
854 for (i = 1; i < argc; i++)
855 if (argv[i][0] != '-' || !argv[i][1]) {
856 if (argv[i][0] == '-') {
857 if (stdin_used++) goto stdin_error;
858 mop[nmats].inspec = stdin_name;
859 } else
860 mop[nmats].inspec = argv[i];
861 if (!mop[nmats].preop.csym)
862 mop[nmats].preop.csym = defCsym;
863 if (++nmats >= nall)
864 resize_inparr(nmats + (nmats>>2) + 2);
865 } else {
866 int n = argc-1 - i;
867 switch (argv[i][1]) { /* get option */
868 case 'w':
869 nowarn = !nowarn;
870 break;
871 case 'h':
872 echoheader = !echoheader;
873 break;
874 case 'n':
875 nproc = atoi(argv[++i]);
876 if (nproc <= 0)
877 goto userr;
878 break;
879 case 'e':
880 if (!n) goto userr;
881 comp_ndx[n2comp++] = i++;
882 break;
883 case 'f':
884 switch (argv[i][2]) {
885 case '\0':
886 if (!n) goto userr;
887 comp_ndx[n2comp++] = i++;
888 break;
889 case 'd':
890 outfmt = DTdouble;
891 break;
892 case 'f':
893 outfmt = DTfloat;
894 break;
895 case 'a':
896 outfmt = DTascii;
897 break;
898 case 'c':
899 outfmt = DTrgbe;
900 break;
901 default:
902 goto userr;
903 }
904 break;
905 case 's':
906 if (n > MAXCOMP) n = MAXCOMP;
907 i += mop[nmats].preop.nsf =
908 get_factors(mop[nmats].preop.sca,
909 n, argv+i+1);
910 if (mop[nmats].preop.nsf <= 0) {
911 fprintf(stderr, "%s: -s missing arguments\n",
912 argv[0]);
913 goto userr;
914 }
915 break;
916 case 'C':
917 if (!n || isflt(argv[i+1]))
918 goto userr;
919 defCsym = mop[nmats].preop.csym = argv[++i];
920 mop[nmats].preop.clen = 0;
921 mcat_last = 0;
922 break;
923 case 'c':
924 if (n && !isflt(argv[i+1])) {
925 mop[nmats].preop.csym = argv[++i];
926 mop[nmats].preop.clen = 0;
927 mcat_last = 0;
928 break;
929 }
930 if (n > MAXCOMP*MAXCOMP) n = MAXCOMP*MAXCOMP;
931 i += mop[nmats].preop.clen =
932 get_factors(mop[nmats].preop.cmat,
933 n, argv+i+1);
934 if (mop[nmats].preop.clen <= 0) {
935 fprintf(stderr, "%s: -c missing arguments\n",
936 argv[0]);
937 goto userr;
938 }
939 mop[nmats].preop.csym = NULL;
940 mcat_last = 0;
941 break;
942 case 'm':
943 if (!n) goto userr;
944 if (argv[++i][0] == '-' && !argv[i][1]) {
945 if (stdin_used++) goto stdin_error;
946 mcat_spec = stdin_name;
947 } else
948 mcat_spec = argv[i];
949 mcat_last = 1;
950 break;
951 default:
952 fprintf(stderr, "%s: unknown option '%s'\n",
953 argv[0], argv[i]);
954 goto userr;
955 }
956 }
957 if (!nmats) {
958 fprintf(stderr, "%s: need at least one input matrix\n", argv[0]);
959 goto userr;
960 }
961 resize_inparr(nmats+1); /* extra matrix at end for result */
962 mop[nmats].inspec = "trailing_ops";
963 /* load final concatenation matrix */
964 if (mcat_spec && !(mcat = rmx_load(mcat_spec, RMPnone))) {
965 fprintf(stderr, "%s: error loading concatenation matrix: %s\n",
966 argv[0], mcat_spec);
967 return(1);
968 }
969 /* get/check inputs, set constants */
970 if (!initialize(&mop[nmats].imx))
971 return(1);
972
973 for (i = 0; i < n2comp; i++) /* user .cal files and expressions */
974 if (argv[comp_ndx[i]][1] == 'f') {
975 char *fpath = getpath(argv[comp_ndx[i]+1],
976 getrlibpath(), 0);
977 if (fpath == NULL) {
978 fprintf(stderr, "%s: cannot find file '%s'\n",
979 argv[0], argv[comp_ndx[i]+1]);
980 return(1);
981 }
982 fcompile(fpath);
983 } else /* (argv[comp_ndx[i]][1] == 'e') */
984 scompile(argv[comp_ndx[i]+1], NULL, 0);
985
986 /* get trailing color transform */
987 if (!get_component_xfm(&mop[nmats]))
988 return(1);
989 /* adjust output dimensions and #components */
990 if (mcat) {
991 if (mop[nmats].imx.ncols != mcat->nrows) {
992 fprintf(stderr,
993 "%s: number of input columns does not match number of rows in '%s'\n",
994 argv[0], mcat_spec);
995 return(1);
996 }
997 if (mcat->ncomp != (mcat_last ? mop[nmats].rmp->ncomp : mop[nmats].imx.ncomp)) {
998 fprintf(stderr,
999 "%s: number of components does not match those in '%s'\n",
1000 argv[0], mcat_spec);
1001 return(1);
1002 }
1003 if (!split_input(&mop[nmats]))
1004 return(1);
1005 mop[nmats].rmp->ncols = mcat->ncols;
1006 }
1007 newheader("RADIANCE", stdout); /* write output header */
1008 if (echoheader)
1009 output_headinfo(stdout);
1010 printargs(argc, argv, stdout);
1011 fputnow(stdout);
1012 mop[nmats].rmp->dtype = rmx_write_header(mop[nmats].rmp, outfmt, stdout);
1013 if (!mop[nmats].rmp->dtype) {
1014 fprintf(stderr, "%s: unsupported output format\n", argv[0]);
1015 return(1);
1016 }
1017 doptimize(1); /* optimize definitions */
1018 if (spawned_children(nproc)) /* running in parent process? */
1019 return(parent_loop() ? 0 : 1);
1020 /* process & write rows */
1021 return(combine_input() ? 0 : 1);
1022 stdin_error:
1023 fprintf(stderr, "%s: %s used for more than one input\n",
1024 argv[0], stdin_name);
1025 return(1);
1026 userr:
1027 fprintf(stderr,
1028 "Usage: %s [-h][-f{adfc}][-n nproc][-e expr][-f file][-s sf .. | -c ce ..] m1 .. -m mcat > mres\n",
1029 argv[0]);
1030 return(1);
1031 }