ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rc3.c
(Generate patch)

Comparing ray/src/rt/rc3.c (file contents):
Revision 2.9 by greg, Thu Jun 14 05:13:25 2012 UTC vs.
Revision 2.20 by greg, Sun Aug 23 00:17:12 2015 UTC

# Line 6 | Line 6 | static const char RCSid[] = "$Id$";
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 */
# Line 26 | Line 27 | static struct {
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  
# Line 94 | Line 95 | free_binq(BINQ *bp)
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;
# Line 146 | Line 147 | queue_output(BINQ *bp)
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;
# Line 184 | Line 185 | queue_ready()
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)
# Line 203 | Line 201 | output_catchup(int nmax)
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))
# Line 270 | Line 265 | set_stdout(const LUENT *le, void *p)
265   }
266  
267  
268 < /* Start child processes if we can */
268 > /* Start child processes if we can (call only once in parent!) */
269   int
270   in_rchild()
271   {
# Line 282 | Line 277 | in_rchild()
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);
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);
# Line 308 | Line 303 | in_rchild()
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()");
311 #ifdef getc_unlocked
312                flockfile(kida[nchild].infp);   /* avoid mutex overhead */
313 #endif
306                  kida[nchild++].nr = 0;  /* mark as available */
307          }
308 + #ifdef getc_unlocked
309 +        for (rval = nchild; rval--; )   /* avoid mutex overhead */
310 +                flockfile(kida[rval].infp);
311 + #endif
312          return(0);                      /* return "false" in parent */
313   }
314  
315  
316   /* Close child processes */
317   void
318 < end_children()
318 > end_children(int immed)
319   {
320          int     status;
321          
322          while (nchild > 0) {
323                  nchild--;
324 <                if ((status = close_process(&kida[nchild].pr)) > 0) {
324 > #ifdef SIGKILL
325 >                if (immed)              /* error mode -- quick exit */
326 >                        kill(kida[nchild].pr.pid, SIGKILL);
327 > #endif
328 >                if ((status = close_process(&kida[nchild].pr)) > 0 && !immed) {
329                          sprintf(errmsg,
330                                  "rendering process returned bad status (%d)",
331                                          status);
# Line 401 | Line 401 | tryagain:                              /* catch up with output? */
401   void
402   parental_loop()
403   {
404 < #define MAXIQ           (int)(PIPE_BUF/(sizeof(FVECT)*2))
405 <        static int      ignore_warning_given = 0;
406 <        int             qlimit = (accumulate == 1) ? 1 : MAXIQ-1;
404 >        const int       qlimit = (accumulate == 1) ? 1 : MAXIQ-1;
405          int             ninq = 0;
406          FVECT           orgdir[2*MAXIQ];
407          int             i, n;
# Line 412 | Line 410 | parental_loop()
410          flockfile(stdin);               /* avoid lock/unlock overhead */
411   #endif
412          while (getvec(orgdir[2*ninq]) == 0 && getvec(orgdir[2*ninq+1]) == 0) {
413 <                if (orgdir[2*ninq+1][0] == 0.0 &&       /* asking for flush? */
414 <                                (orgdir[2*ninq+1][1] == 0.0) &
415 <                                (orgdir[2*ninq+1][2] == 0.0)) {
416 <                        if (accumulate != 1) {
417 <                                if (!ignore_warning_given++)
418 <                                        error(WARNING,
421 <                                "dummy ray(s) ignored during accumulation\n");
422 <                                continue;
423 <                        }
424 <                        while (next_child_nq(1) >= 0)
425 <                                ;                       /* clear the queue */
426 <                        lastdone = lastray = 0;
427 <                        if ((yres <= 0) | (xres <= 0))
428 <                                waitflush = 1;          /* flush next */
429 <                        put_zero_record(++lastray);
430 <                } else if (++ninq >= qlimit || accumulate > 1 &&
413 >                const int       zero_ray = orgdir[2*ninq+1][0] == 0.0 &&
414 >                                                (orgdir[2*ninq+1][1] == 0.0) &
415 >                                                (orgdir[2*ninq+1][2] == 0.0);
416 >                ninq += !zero_ray;
417 >                                /* Zero ray cannot go in input queue */
418 >                if (zero_ray ? ninq : ninq >= qlimit ||
419                              lastray/accumulate != (lastray+ninq)/accumulate) {
420                          i = next_child_nq(0);           /* manages output */
421                          n = ninq;
422 <                        if (accumulate != 1)            /* request flush? */
422 >                        if (accumulate > 1)             /* need terminator? */
423                                  memset(orgdir[2*n++], 0, sizeof(FVECT)*2);
424                          n *= sizeof(FVECT)*2;           /* send assignment */
425                          if (writebuf(kida[i].pr.w, (char *)orgdir, n) != n)
426                                  error(SYSTEM, "pipe write error");
427                          kida[i].r1 = lastray+1;
428                          lastray += kida[i].nr = ninq;   /* mark as busy */
441                        ninq = 0;
429                          if (lastray < lastdone) {       /* RNUMBER wrapped? */
430                                  while (next_child_nq(1) >= 0)
431                                          ;
432 <                                lastdone = lastray = 0;
432 >                                lastray -= ninq;
433 >                                lastdone = lastray %= accumulate;
434                          }
435 +                        ninq = 0;
436                  }
437 +                if (zero_ray) {                         /* put bogus record? */
438 +                        if ((yres <= 0) | (xres <= 1) &&
439 +                                        (lastray+1) % accumulate == 0) {
440 +                                while (next_child_nq(1) >= 0)
441 +                                        ;               /* clear the queue */
442 +                                lastdone = lastray = accumulate-1;
443 +                                waitflush = 1;          /* flush next */
444 +                        }
445 +                        put_zero_record(++lastray);
446 +                }
447                  if (raysleft && !--raysleft)
448                          break;                          /* preemptive EOI */
449          }
450          while (next_child_nq(1) >= 0)           /* empty results queue */
451                  ;
452 <                                                /* output accumulated record */
453 <        if (accumulate <= 0 || account < accumulate) {
454 <                end_children();                 /* frees up file descriptors */
456 <                if (account < accumulate) {
457 <                        error(WARNING, "partial accumulation in final record");
458 <                        accumulate -= account;
459 <                }
460 <                for (i = 0; i < nmods; i++)
461 <                        mod_output(out_bq->mca[i]);
462 <                end_record();
463 <                free_binq(out_bq);
452 >        if (account < accumulate) {
453 >                error(WARNING, "partial accumulation in final record");
454 >                free_binq(out_bq);              /* XXX just ignore it */
455                  out_bq = NULL;
456          }
457 +        free_binq(NULL);                        /* clean up */
458 +        lu_done(&ofiletab);
459          if (raysleft)
460                  error(USER, "unexpected EOF on input");
461 <        free_binq(NULL);                        /* clean up */
461 > }
462 >
463 >
464 > /* Wait for the next available child by monitoring "to" pipes */
465 > static int
466 > next_child_ready()
467 > {
468 >        fd_set                  writeset, errset;
469 >        int                     i, n;
470 >
471 >        for (i = nchild; i--; )         /* see if there's one free first */
472 >                if (!kida[i].nr)
473 >                        return(i);
474 >                                        /* prepare select() call */
475 >        FD_ZERO(&writeset); FD_ZERO(&errset);
476 >        n = 0;
477 >        for (i = nchild; i--; ) {
478 >                FD_SET(kida[i].pr.w, &writeset);
479 >                FD_SET(kida[i].pr.r, &errset);
480 >                if (kida[i].pr.w >= n)
481 >                        n = kida[i].pr.w + 1;
482 >                if (kida[i].pr.r >= n)
483 >                        n = kida[i].pr.r + 1;
484 >        }
485 >        errno = 0;
486 >        n = select(n, NULL, &writeset, &errset, NULL);
487 >        if (n < 0)
488 >                error(SYSTEM, "select() error in next_child_ready()");
489 >        n = -1;                         /* identify waiting child */
490 >        for (i = nchild; i--; ) {
491 >                if (FD_ISSET(kida[i].pr.r, &errset))
492 >                        error(USER, "rendering process died");
493 >                if (FD_ISSET(kida[i].pr.w, &writeset))
494 >                        kida[n = i].nr = 0;
495 >        }
496 >        return(n);                      /* first available child */
497 > }
498 >
499 >
500 > /* Modified parental loop for full accumulation mode (-c 0) */
501 > void
502 > feeder_loop()
503 > {
504 >        static int      ignore_warning_given = 0;
505 >        int             ninq = 0;
506 >        FVECT           orgdir[2*MAXIQ];
507 >        int             i, n;
508 >                                        /* load rays from stdin & process */
509 > #ifdef getc_unlocked
510 >        flockfile(stdin);               /* avoid lock/unlock overhead */
511 > #endif
512 >        while (getvec(orgdir[2*ninq]) == 0 && getvec(orgdir[2*ninq+1]) == 0) {
513 >                if (orgdir[2*ninq+1][0] == 0.0 &&       /* asking for flush? */
514 >                                (orgdir[2*ninq+1][1] == 0.0) &
515 >                                (orgdir[2*ninq+1][2] == 0.0)) {
516 >                        if (!ignore_warning_given++)
517 >                                error(WARNING,
518 >                                "dummy ray(s) ignored during accumulation\n");
519 >                        continue;
520 >                }
521 >                if (++ninq >= MAXIQ) {
522 >                        i = next_child_ready();         /* get eager child */
523 >                        n = sizeof(FVECT)*2 * ninq;     /* give assignment */
524 >                        if (writebuf(kida[i].pr.w, (char *)orgdir, n) != n)
525 >                                error(SYSTEM, "pipe write error");
526 >                        kida[i].r1 = lastray+1;
527 >                        lastray += kida[i].nr = ninq;
528 >                        if (lastray < lastdone)         /* RNUMBER wrapped? */
529 >                                lastdone = lastray = 0;
530 >                        ninq = 0;
531 >                }
532 >                if (raysleft && !--raysleft)
533 >                        break;                          /* preemptive EOI */
534 >        }
535 >        if (ninq) {                             /* polish off input */
536 >                i = next_child_ready();
537 >                n = sizeof(FVECT)*2 * ninq;
538 >                if (writebuf(kida[i].pr.w, (char *)orgdir, n) != n)
539 >                        error(SYSTEM, "pipe write error");
540 >                kida[i].r1 = lastray+1;
541 >                lastray += kida[i].nr = ninq;
542 >                ninq = 0;
543 >        }
544 >        memset(orgdir, 0, sizeof(FVECT)*2);     /* get results */
545 >        for (i = nchild; i--; ) {
546 >                writebuf(kida[i].pr.w, (char *)orgdir, sizeof(FVECT)*2);
547 >                queue_results(i);
548 >        }
549 >        if (recover)                            /* and from before? */
550 >                queue_modifiers();
551 >        end_children(0);                        /* free up file descriptors */
552 >        for (i = 0; i < nmods; i++)
553 >                mod_output(out_bq->mca[i]);     /* output accumulated record */
554 >        end_record();
555 >        free_binq(out_bq);                      /* clean up */
556 >        out_bq = NULL;
557 >        free_binq(NULL);
558          lu_done(&ofiletab);
559 < #undef MAXIQ
559 >        if (raysleft)
560 >                error(USER, "unexpected EOF on input");
561   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines