| 6 |
|
* Controlling process for multiple children |
| 7 |
|
*/ |
| 8 |
|
|
| 9 |
+ |
#include <signal.h> |
| 10 |
|
#include "rcontrib.h" |
| 10 |
– |
#include "platform.h" |
| 11 |
– |
#include "rtprocess.h" |
| 11 |
|
#include "selcall.h" |
| 12 |
|
|
| 13 |
+ |
#define MAXIQ (int)(PIPE_BUF/(sizeof(FVECT)*2)) |
| 14 |
+ |
|
| 15 |
|
/* Modifier contribution queue (results waiting to be output) */ |
| 16 |
|
typedef struct s_binq { |
| 17 |
|
RNUMBER ndx; /* index for this entry */ |
| 27 |
|
RNUMBER r1; /* assigned ray starting index */ |
| 28 |
|
SUBPROC pr; /* PID, i/o descriptors */ |
| 29 |
|
FILE *infp; /* file pointer to read from process */ |
| 30 |
< |
int nr; /* number rays to sum (0 if free) */ |
| 30 |
> |
int nr; /* number of rays to sum (0 if free) */ |
| 31 |
|
} kida[MAXPROCESS]; /* our child processes */ |
| 32 |
|
|
| 33 |
|
|
| 95 |
|
|
| 96 |
|
|
| 97 |
|
/* Add modifier values to accumulation record in queue and clear */ |
| 98 |
< |
void |
| 98 |
> |
static void |
| 99 |
|
queue_modifiers() |
| 100 |
|
{ |
| 101 |
|
MODCONT *mpin, *mpout; |
| 147 |
|
} |
| 148 |
|
return; |
| 149 |
|
} |
| 150 |
< |
b_last = NULL; /* else insert in output queue */ |
| 150 |
> |
b_last = NULL; /* insert in output queue */ |
| 151 |
|
for (b_cur = out_bq; b_cur != NULL && b_cur->ndx < bp->ndx; |
| 152 |
|
b_cur = b_cur->next) |
| 153 |
|
b_last = b_cur; |
| 185 |
|
int nready = 0; |
| 186 |
|
BINQ *bp; |
| 187 |
|
|
| 187 |
– |
if (accumulate <= 0) /* just accumulating? */ |
| 188 |
– |
return(0); |
| 189 |
– |
|
| 188 |
|
for (bp = out_bq; bp != NULL && bp->nadded >= accumulate && |
| 189 |
|
bp->ndx == lastdone+nready*accumulate+1; |
| 190 |
|
bp = bp->next) |
| 201 |
|
int nout = 0; |
| 202 |
|
BINQ *bp; |
| 203 |
|
int i; |
| 204 |
< |
|
| 207 |
< |
if (accumulate <= 0) /* just accumulating? */ |
| 208 |
< |
return(0); |
| 209 |
< |
/* else output ready results */ |
| 204 |
> |
/* output ready results */ |
| 205 |
|
while (out_bq != NULL && out_bq->nadded >= accumulate |
| 206 |
|
&& out_bq->ndx == lastdone+1) { |
| 207 |
|
if ((nmax > 0) & (nout >= nmax)) |
| 269 |
|
int |
| 270 |
|
in_rchild() |
| 271 |
|
{ |
| 272 |
< |
#ifdef _WIN32 |
| 273 |
< |
error(WARNING, "multiprocessing unsupported -- running solo"); |
| 274 |
< |
nproc = 1; |
| 280 |
< |
return(1); |
| 281 |
< |
#else |
| 282 |
< |
/* try to fork ourselves */ |
| 283 |
< |
while (nchild < nproc) { |
| 284 |
< |
int p0[2], p1[2]; |
| 285 |
< |
int pid; |
| 286 |
< |
/* prepare i/o pipes */ |
| 272 |
> |
int rval; |
| 273 |
> |
|
| 274 |
> |
while (nchild < nproc) { /* fork until target reached */ |
| 275 |
|
errno = 0; |
| 276 |
< |
if (pipe(p0) < 0 || pipe(p1) < 0) |
| 277 |
< |
error(SYSTEM, "pipe() call failed!"); |
| 278 |
< |
pid = fork(); /* fork parent process */ |
| 279 |
< |
if (pid == 0) { /* if in child, set up & return true */ |
| 292 |
< |
close(p0[1]); close(p1[0]); |
| 276 |
> |
rval = open_process(&kida[nchild].pr, NULL); |
| 277 |
> |
if (rval < 0) |
| 278 |
> |
error(SYSTEM, "open_process() call failed"); |
| 279 |
> |
if (rval == 0) { /* if in child, set up & return true */ |
| 280 |
|
lu_doall(&modconttab, set_stdout, NULL); |
| 281 |
|
lu_done(&ofiletab); |
| 282 |
|
while (nchild--) { /* don't share other pipes */ |
| 283 |
|
close(kida[nchild].pr.w); |
| 284 |
|
fclose(kida[nchild].infp); |
| 285 |
|
} |
| 299 |
– |
dup2(p0[0], 0); close(p0[0]); |
| 300 |
– |
dup2(p1[1], 1); close(p1[1]); |
| 286 |
|
inpfmt = (sizeof(RREAL)==sizeof(double)) ? 'd' : 'f'; |
| 287 |
|
outfmt = 'd'; |
| 288 |
|
header = 0; |
| 295 |
|
waitflush = xres = 0; |
| 296 |
|
account = accumulate = 0; |
| 297 |
|
} |
| 298 |
< |
return(1); /* child return value */ |
| 298 |
> |
return(1); /* return "true" in child */ |
| 299 |
|
} |
| 300 |
< |
if (pid < 0) |
| 301 |
< |
error(SYSTEM, "fork() call failed!"); |
| 302 |
< |
/* connect parent's pipes */ |
| 303 |
< |
close(p0[0]); close(p1[1]); |
| 319 |
< |
kida[nchild].pr.r = p1[0]; |
| 320 |
< |
kida[nchild].pr.w = p0[1]; |
| 321 |
< |
kida[nchild].pr.pid = pid; |
| 322 |
< |
kida[nchild].pr.running = 1; |
| 323 |
< |
kida[nchild].infp = fdopen(p1[0], "rb"); |
| 300 |
> |
if (rval != PIPE_BUF) |
| 301 |
> |
error(CONSISTENCY, "bad value from open_process()"); |
| 302 |
> |
/* connect to child's output */ |
| 303 |
> |
kida[nchild].infp = fdopen(kida[nchild].pr.r, "rb"); |
| 304 |
|
if (kida[nchild].infp == NULL) |
| 305 |
|
error(SYSTEM, "out of memory in in_rchild()"); |
| 306 |
|
#ifdef getc_unlocked |
| 308 |
|
#endif |
| 309 |
|
kida[nchild++].nr = 0; /* mark as available */ |
| 310 |
|
} |
| 311 |
< |
return(0); /* parent return value */ |
| 332 |
< |
#endif |
| 311 |
> |
return(0); /* return "false" in parent */ |
| 312 |
|
} |
| 313 |
|
|
| 314 |
|
|
| 315 |
|
/* Close child processes */ |
| 316 |
|
void |
| 317 |
< |
end_children() |
| 317 |
> |
end_children(int immed) |
| 318 |
|
{ |
| 319 |
|
int status; |
| 320 |
|
|
| 321 |
|
while (nchild > 0) { |
| 322 |
|
nchild--; |
| 323 |
< |
fclose(kida[nchild].infp); |
| 324 |
< |
kida[nchild].pr.r = -1; /* close(-1) error is ignored */ |
| 325 |
< |
if ((status = close_process(&kida[nchild].pr)) > 0) { |
| 323 |
> |
#ifdef SIGKILL |
| 324 |
> |
if (immed) /* error mode -- quick exit */ |
| 325 |
> |
kill(kida[nchild].pr.pid, SIGKILL); |
| 326 |
> |
#endif |
| 327 |
> |
if ((status = close_process(&kida[nchild].pr)) > 0 && !immed) { |
| 328 |
|
sprintf(errmsg, |
| 329 |
|
"rendering process returned bad status (%d)", |
| 330 |
|
status); |
| 331 |
|
error(WARNING, errmsg); |
| 332 |
|
} |
| 333 |
+ |
fclose(kida[nchild].infp); |
| 334 |
|
} |
| 335 |
|
} |
| 336 |
|
|
| 400 |
|
void |
| 401 |
|
parental_loop() |
| 402 |
|
{ |
| 421 |
– |
#define MAXIQ (int)(PIPE_BUF/(sizeof(FVECT)*2)) |
| 403 |
|
static int ignore_warning_given = 0; |
| 404 |
|
int qlimit = (accumulate == 1) ? 1 : MAXIQ-1; |
| 405 |
|
int ninq = 0; |
| 425 |
|
if ((yres <= 0) | (xres <= 0)) |
| 426 |
|
waitflush = 1; /* flush next */ |
| 427 |
|
put_zero_record(++lastray); |
| 428 |
< |
} else if (++ninq >= qlimit || accumulate > 1 && |
| 428 |
> |
} else if (++ninq >= qlimit || |
| 429 |
|
lastray/accumulate != (lastray+ninq)/accumulate) { |
| 430 |
|
i = next_child_nq(0); /* manages output */ |
| 431 |
|
n = ninq; |
| 448 |
|
} |
| 449 |
|
while (next_child_nq(1) >= 0) /* empty results queue */ |
| 450 |
|
; |
| 451 |
< |
/* output accumulated record */ |
| 452 |
< |
if (accumulate <= 0 || account < accumulate) { |
| 453 |
< |
end_children(); /* frees up file descriptors */ |
| 473 |
< |
if (account < accumulate) { |
| 474 |
< |
error(WARNING, "partial accumulation in final record"); |
| 475 |
< |
accumulate -= account; |
| 476 |
< |
} |
| 477 |
< |
for (i = 0; i < nmods; i++) |
| 478 |
< |
mod_output(out_bq->mca[i]); |
| 479 |
< |
end_record(); |
| 480 |
< |
free_binq(out_bq); |
| 451 |
> |
if (account < accumulate) { |
| 452 |
> |
error(WARNING, "partial accumulation in final record"); |
| 453 |
> |
free_binq(out_bq); /* XXX just ignore it */ |
| 454 |
|
out_bq = NULL; |
| 455 |
|
} |
| 456 |
+ |
free_binq(NULL); /* clean up */ |
| 457 |
+ |
lu_done(&ofiletab); |
| 458 |
|
if (raysleft) |
| 459 |
|
error(USER, "unexpected EOF on input"); |
| 460 |
< |
free_binq(NULL); /* clean up */ |
| 460 |
> |
} |
| 461 |
> |
|
| 462 |
> |
|
| 463 |
> |
/* Wait for the next available child by monitoring "to" pipes */ |
| 464 |
> |
static int |
| 465 |
> |
next_child_ready() |
| 466 |
> |
{ |
| 467 |
> |
fd_set writeset, errset; |
| 468 |
> |
int i, n; |
| 469 |
> |
|
| 470 |
> |
for (i = nchild; i--; ) /* see if there's one free first */ |
| 471 |
> |
if (!kida[i].nr) |
| 472 |
> |
return(i); |
| 473 |
> |
/* prepare select() call */ |
| 474 |
> |
FD_ZERO(&writeset); FD_ZERO(&errset); |
| 475 |
> |
n = 0; |
| 476 |
> |
for (i = nchild; i--; ) { |
| 477 |
> |
FD_SET(kida[i].pr.w, &writeset); |
| 478 |
> |
FD_SET(kida[i].pr.r, &errset); |
| 479 |
> |
if (kida[i].pr.w >= n) |
| 480 |
> |
n = kida[i].pr.w + 1; |
| 481 |
> |
if (kida[i].pr.r >= n) |
| 482 |
> |
n = kida[i].pr.r + 1; |
| 483 |
> |
} |
| 484 |
> |
errno = 0; |
| 485 |
> |
n = select(n, NULL, &writeset, &errset, NULL); |
| 486 |
> |
if (n < 0) |
| 487 |
> |
error(SYSTEM, "select() error in next_child_ready()"); |
| 488 |
> |
n = -1; /* identify waiting child */ |
| 489 |
> |
for (i = nchild; i--; ) { |
| 490 |
> |
if (FD_ISSET(kida[i].pr.r, &errset)) |
| 491 |
> |
error(USER, "rendering process died"); |
| 492 |
> |
if (FD_ISSET(kida[i].pr.w, &writeset)) |
| 493 |
> |
kida[n = i].nr = 0; |
| 494 |
> |
} |
| 495 |
> |
return(n); /* first available child */ |
| 496 |
> |
} |
| 497 |
> |
|
| 498 |
> |
|
| 499 |
> |
/* Modified parental loop for full accumulation mode (-c 0) */ |
| 500 |
> |
void |
| 501 |
> |
feeder_loop() |
| 502 |
> |
{ |
| 503 |
> |
static int ignore_warning_given = 0; |
| 504 |
> |
int ninq = 0; |
| 505 |
> |
FVECT orgdir[2*MAXIQ]; |
| 506 |
> |
int i, n; |
| 507 |
> |
/* load rays from stdin & process */ |
| 508 |
> |
#ifdef getc_unlocked |
| 509 |
> |
flockfile(stdin); /* avoid lock/unlock overhead */ |
| 510 |
> |
#endif |
| 511 |
> |
while (getvec(orgdir[2*ninq]) == 0 && getvec(orgdir[2*ninq+1]) == 0) { |
| 512 |
> |
if (orgdir[2*ninq+1][0] == 0.0 && /* asking for flush? */ |
| 513 |
> |
(orgdir[2*ninq+1][1] == 0.0) & |
| 514 |
> |
(orgdir[2*ninq+1][2] == 0.0)) { |
| 515 |
> |
if (!ignore_warning_given++) |
| 516 |
> |
error(WARNING, |
| 517 |
> |
"dummy ray(s) ignored during accumulation\n"); |
| 518 |
> |
continue; |
| 519 |
> |
} |
| 520 |
> |
if (++ninq >= MAXIQ) { |
| 521 |
> |
i = next_child_ready(); /* get eager child */ |
| 522 |
> |
n = sizeof(FVECT)*2 * ninq; /* give assignment */ |
| 523 |
> |
if (writebuf(kida[i].pr.w, (char *)orgdir, n) != n) |
| 524 |
> |
error(SYSTEM, "pipe write error"); |
| 525 |
> |
kida[i].r1 = lastray+1; |
| 526 |
> |
lastray += kida[i].nr = ninq; |
| 527 |
> |
ninq = 0; |
| 528 |
> |
if (lastray < lastdone) /* RNUMBER wrapped? */ |
| 529 |
> |
lastdone = lastray = 0; |
| 530 |
> |
} |
| 531 |
> |
if (raysleft && !--raysleft) |
| 532 |
> |
break; /* preemptive EOI */ |
| 533 |
> |
} |
| 534 |
> |
if (ninq) { /* polish off input */ |
| 535 |
> |
i = next_child_ready(); |
| 536 |
> |
n = sizeof(FVECT)*2 * ninq; |
| 537 |
> |
if (writebuf(kida[i].pr.w, (char *)orgdir, n) != n) |
| 538 |
> |
error(SYSTEM, "pipe write error"); |
| 539 |
> |
kida[i].r1 = lastray+1; |
| 540 |
> |
lastray += kida[i].nr = ninq; |
| 541 |
> |
ninq = 0; |
| 542 |
> |
} |
| 543 |
> |
memset(orgdir, 0, sizeof(FVECT)*2); /* get results */ |
| 544 |
> |
for (i = nchild; i--; ) { |
| 545 |
> |
writebuf(kida[i].pr.w, (char *)orgdir, sizeof(FVECT)*2); |
| 546 |
> |
queue_results(i); |
| 547 |
> |
} |
| 548 |
> |
if (recover) /* and from before? */ |
| 549 |
> |
queue_modifiers(); |
| 550 |
> |
end_children(0); /* free up file descriptors */ |
| 551 |
> |
for (i = 0; i < nmods; i++) |
| 552 |
> |
mod_output(out_bq->mca[i]); /* output accumulated record */ |
| 553 |
> |
end_record(); |
| 554 |
> |
free_binq(out_bq); /* clean up */ |
| 555 |
> |
out_bq = NULL; |
| 556 |
> |
free_binq(NULL); |
| 557 |
|
lu_done(&ofiletab); |
| 558 |
< |
#undef MAXIQ |
| 558 |
> |
if (raysleft) |
| 559 |
> |
error(USER, "unexpected EOF on input"); |
| 560 |
|
} |
