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

Comparing ray/src/rt/raypcalls.c (file contents):
Revision 2.22 by greg, Sat Dec 12 05:20:10 2009 UTC vs.
Revision 2.40 by greg, Wed Aug 21 20:42:20 2024 UTC

# Line 73 | Line 73 | static const char      RCSid[] = "$Id$";
73   *  results aren't ready, but will immediately return 0.
74   *  If the second argument is 0, the call will block
75   *  until a value is available, returning 0 only if the
76 < *  queue is completely empty.  A negative return value
76 > *  queue is completely empty.  Setting the second argument
77 > *  to -1 returns 0 unless a ray is ready in the queue and
78 > *  no system calls are needed.  A negative return value
79   *  indicates that a rendering process died.  If this
80   *  happens, ray_pclose(0) is automatically called to close
81   *  all child processes, and ray_pnprocs is set to zero.
# Line 86 | Line 88 | static const char      RCSid[] = "$Id$";
88   *              ray_psend(&myRay);
89   *      }
90   *
91 < *  Note that it is a fatal error to call ra_psend() when
92 < *  ray_pnidle is zero.  The ray_presult() and/or ray_pqueue()
93 < *  functions may be called subsequently to read back the results.
91 > *  Note that it is a mistake to call ra_psend() when
92 > *  ray_pnidle is zero, and nothing will be sent in
93 > *  this case.  Otherwise, the ray_presult() and/or ray_pqueue()
94 > *  functions may be called subsequently to read back the results
95 > *  of rays queued by ray_psend().
96   *
97   *  When you are done, you may call ray_pdone(1) to close
98   *  all child processes and clean up memory used by Radiance.
# Line 97 | Line 101 | static const char      RCSid[] = "$Id$";
101   *  and fonts that are likely to be used in subsequent renderings.
102   *  Whether you need to clean up memory or not, you should
103   *  at least call ray_pclose(0) to await the child processes.
104 + *  The caller should define a quit() function that calls
105 + *  ray_pclose(0) if ray_pnprocs > 0.
106   *
107   *  Warning:  You cannot affect any of the rendering processes
108   *  by changing global parameter values onece ray_pinit() has
# Line 141 | Line 147 | static const char      RCSid[] = "$Id$";
147   #include  "selcall.h"
148  
149   #ifndef RAYQLEN
150 < #define RAYQLEN         12              /* # rays to send at once */
150 > #define RAYQLEN         24              /* # rays to send at once */
151   #endif
152  
153   #ifndef MAX_RPROCS
# Line 152 | Line 158 | static const char      RCSid[] = "$Id$";
158   #endif
159   #endif
160  
155 – extern char     *shm_boundary;          /* boundary of shared memory */
156 –
161   int             ray_pnprocs = 0;        /* number of child processes */
162   int             ray_pnidle = 0;         /* number of idle children */
163  
164   static struct child_proc {
165 <        int     pid;                            /* child process id */
165 >        RT_PID  pid;                            /* child process id */
166          int     fd_send;                        /* write to child here */
167          int     fd_recv;                        /* read from child here */
168          int     npending;                       /* # rays in process */
# Line 166 | Line 170 | static struct child_proc {
170   } r_proc[MAX_NPROCS];                   /* our child processes */
171  
172   static RAY      r_queue[2*RAYQLEN];     /* ray i/o buffer */
173 < static int      r_send_next;            /* next send ray placement */
174 < static int      r_recv_first;           /* position of first unreported ray */
175 < static int      r_recv_next;            /* next received ray placement */
173 > static int      r_send_next = 0;        /* next send ray placement */
174 > static int      r_recv_first = RAYQLEN; /* position of first unreported ray */
175 > static int      r_recv_next = RAYQLEN;  /* next received ray placement */
176  
177 + static int      samplestep = 1;         /* sample step size */
178 +
179   #define sendq_full()    (r_send_next >= RAYQLEN)
180  
181   static int ray_pflush(void);
# Line 187 | Line 193 | ray_pinit(             /* initialize ray-tracing processes */
193  
194          ray_init(otnm);                 /* load the shared scene */
195  
190 –        r_send_next = 0;                /* set up queue */
191 –        r_recv_first = r_recv_next = RAYQLEN;
192 –
196          ray_popen(nproc);               /* fork children */
197   }
198  
# Line 207 | Line 210 | ray_pflush(void)                       /* send queued rays to idle childre
210          for (i = ray_pnprocs; nc && i--; ) {
211                  if (r_proc[i].npending > 0)
212                          continue;       /* child looks busy */
213 <                n = (r_send_next - sfirst)/nc--;
213 >                n = (r_send_next - sfirst) / nc--;
214                  if (!n)
215                          continue;
216                                          /* smuggle set size in crtype */
217                  r_queue[sfirst].crtype = n;
218 <                nw = writebuf(r_proc[i].fd_send, (char *)&r_queue[sfirst],
218 >                nw = writebuf(r_proc[i].fd_send, &r_queue[sfirst],
219                                  sizeof(RAY)*n);
220                  if (nw != sizeof(RAY)*n)
221                          return(-1);     /* write error */
# Line 223 | Line 226 | ray_pflush(void)                       /* send queued rays to idle childre
226                  ray_pnidle--;           /* now she's busy */
227          }
228          if (sfirst != r_send_next)
229 <                error(CONSISTENCY, "code screwup in ray_pflush");
229 >                error(CONSISTENCY, "code screwup in ray_pflush()");
230          r_send_next = 0;
231          return(sfirst);                 /* return total # sent */
232   }
233  
234  
235 < void
235 > int
236   ray_psend(                      /* add a ray to our send queue */
237          RAY     *r
238   )
239   {
240 <        if (r == NULL)
241 <                return;
240 >        int     rv;
241 >
242 >        if ((r == NULL) | (ray_pnidle <= 0))
243 >                return(0);
244                                          /* flush output if necessary */
245 <        if (sendq_full() && ray_pflush() <= 0)
246 <                error(INTERNAL, "ray_pflush failed in ray_psend");
245 >        if (sendq_full() && (rv = ray_pflush()) <= 0)
246 >                return(rv);
247  
248          r_queue[r_send_next++] = *r;
249 +        return(1);
250   }
251  
252  
# Line 259 | Line 265 | ray_pqueue(                    /* queue a ray for computation */
265                          return(-1);
266                                          /* put new ray in queue */
267                  r_queue[r_send_next++] = mySend;
268 <                                /* XXX r_send_next may now be > RAYQLEN */
268 >
269                  return(1);
270          }
271                                          /* else add ray to send queue */
# Line 282 | Line 288 | ray_presult(           /* check for a completed ray */
288          static struct timeval   tpoll;  /* zero timeval struct */
289          static fd_set   readset, errset;
290          int     n, ok;
291 <        register int    pn;
291 >        int     pn;
292  
293          if (r == NULL)
294                  return(0);
# Line 291 | Line 297 | ray_presult(           /* check for a completed ray */
297                  *r = r_queue[r_recv_first++];
298                  return(1);
299          }
300 +        if (poll < 0)                   /* immediate polling mode? */
301 +                return(0);
302 +
303          n = ray_pnprocs - ray_pnidle;   /* pending before flush? */
304  
305          if (ray_pflush() < 0)           /* send new rays to process */
# Line 325 | Line 334 | getready:                              /* any children waiting for us? */
334                                  poll ? &tpoll : (struct timeval *)NULL)) < 0)
335                          if (errno != EINTR) {
336                                  error(WARNING,
337 <                                        "select call failed in ray_presult");
337 >                                        "select call failed in ray_presult()");
338                                  ray_pclose(0);
339                                  return(-1);
340                          }
# Line 337 | Line 346 | getready:                              /* any children waiting for us? */
346                  error(CONSISTENCY, "buffer shortage in ray_presult()");
347  
348                                          /* read rendered ray data */
349 <        n = readbuf(r_proc[pn].fd_recv, (char *)&r_queue[r_recv_next],
349 >        n = readbuf(r_proc[pn].fd_recv, &r_queue[r_recv_next],
350                          sizeof(RAY)*r_proc[pn].npending);
351          if (n > 0) {
352                  r_recv_next += n/sizeof(RAY);
# Line 357 | Line 366 | getready:                              /* any children waiting for us? */
366          }
367                                          /* preen returned rays */
368          for (n = r_recv_next - r_recv_first; n--; ) {
369 <                register RAY    *rp = &r_queue[r_recv_first + n];
369 >                RAY     *rp = &r_queue[r_recv_first + n];
370                  rp->rno = r_proc[pn].rno[n];
371                  rp->parent = NULL;
372                  rp->newcset = rp->clipset = NULL;
# Line 377 | Line 386 | ray_pdone(             /* reap children and free data */
386   {
387          ray_pclose(0);                  /* close child processes */
388  
389 <        if (shm_boundary != NULL) {     /* clear shared memory boundary */
390 <                free((void *)shm_boundary);
382 <                shm_boundary = NULL;
383 <        }
389 >        cow_doneshare();                /* clear shared memory boundary */
390 >
391          ray_done(freall);               /* free rendering data */
392   }
393  
# Line 392 | Line 399 | ray_pchild(    /* process rays (never returns) */
399   )
400   {
401          int     n;
402 <        register int    i;
402 >        int     i;
403                                          /* flag child process for quit() */
404          ray_pnprocs = -1;
405                                          /* read each ray request set */
# Line 403 | Line 410 | ray_pchild(    /* process rays (never returns) */
410                                          /* get smuggled set length */
411                  n2 = sizeof(RAY)*r_queue[0].crtype - n;
412                  if (n2 < 0)
413 <                        error(INTERNAL, "buffer over-read in ray_pchild");
413 >                        error(INTERNAL, "buffer over-read in ray_pchild()");
414                  if (n2 > 0) {           /* read the rest of the set */
415                          i = readbuf(fd_in, (char *)r_queue + n, n2);
416                          if (i != n2)
# Line 418 | Line 425 | ray_pchild(    /* process rays (never returns) */
425                          r_queue[i].clipset = NULL;
426                          r_queue[i].slights = NULL;
427                          r_queue[i].rlvl = 0;
428 <                        samplendx++;
428 >                        samplendx += samplestep;
429                          rayclear(&r_queue[i]);
430                          rayvalue(&r_queue[i]);
431                  }
432                                          /* write back our results */
433 <                i = writebuf(fd_out, (char *)r_queue, sizeof(RAY)*n);
433 >                i = writebuf(fd_out, r_queue, sizeof(RAY)*n);
434                  if (i != sizeof(RAY)*n)
435 <                        error(SYSTEM, "write error in ray_pchild");
435 >                        error(SYSTEM, "write error in ray_pchild()");
436          }
437          if (n)
438 <                error(SYSTEM, "read error in ray_pchild");
438 >                error(SYSTEM, "read error in ray_pchild()");
439          ambsync();
440          quit(0);                        /* normal exit */
441   }
# Line 444 | Line 451 | ray_popen(                     /* open the specified # processes */
451                  nadd = MAX_NPROCS - ray_pnprocs;
452          if (nadd <= 0)
453                  return;
454 +        if (nobjects <= 0)
455 +                error(CONSISTENCY, "ray_popen() called before scene loaded");
456          ambsync();                      /* load any new ambient values */
457 <        if (shm_boundary == NULL) {     /* first child process? */
449 <                preload_objs();         /* preload auxiliary data */
450 <                                        /* set shared memory boundary */
451 <                shm_boundary = (char *)malloc(16);
452 <                strcpy(shm_boundary, "SHM_BOUNDARY");
453 <        }
457 >        cow_memshare();                 /* copy-on-write shared memory */
458          fflush(NULL);                   /* clear pending output */
459 +        samplestep = ray_pnprocs + nadd;
460          while (nadd--) {                /* fork each new process */
461                  int     p0[2], p1[2];
462                  if (pipe(p0) < 0 || pipe(p1) < 0)
# Line 463 | Line 468 | ray_popen(                     /* open the specified # processes */
468                                  close(r_proc[pn].fd_recv);
469                          }
470                          close(p0[0]); close(p1[1]);
471 +                        close(0);       /* don't share stdin */
472                                          /* following call never returns */
473                          ray_pchild(p1[0], p0[1]);
474                  }
475                  if (r_proc[ray_pnprocs].pid < 0)
476                          error(SYSTEM, "cannot fork child process");
477                  close(p1[0]); close(p0[1]);
478 +                if (rand_samp)          /* decorrelate random sequence */
479 +                        srandom(random());
480 +                else
481 +                        samplendx++;
482                  /*
483                   * Close write stream on exec to avoid multiprocessing deadlock.
484                   * No use in read stream without it, so set flag there as well.
# Line 490 | Line 500 | ray_pclose(            /* close one or more child processes */
500   )
501   {
502          static int      inclose = 0;
503 <        RAY     res;
503 >        RAY             res;
504 >        int             i, status = 0;
505 >                                        /* check no child / in child */
506 >        if (ray_pnprocs <= 0)
507 >                return;
508                                          /* check recursion */
509          if (inclose)
510                  return;
# Line 499 | Line 513 | ray_pclose(            /* close one or more child processes */
513          if ((nsub <= 0) | (nsub > ray_pnprocs))
514                  nsub = ray_pnprocs;
515                                          /* clear our ray queue */
516 +        i = r_send_next;
517 +        r_send_next = 0;
518          while (ray_presult(&res,0) > 0)
519 <                ;
520 <                                        /* clean up children */
521 <        while (nsub--) {
522 <                int     status;
523 <                ray_pnprocs--;
524 <                close(r_proc[ray_pnprocs].fd_recv);
525 <                close(r_proc[ray_pnprocs].fd_send);
526 <                if (waitpid(r_proc[ray_pnprocs].pid, &status, 0) < 0)
519 >                ++i;
520 >        if (i) {
521 >                sprintf(errmsg, "dropped %d rays in ray_pclose()", i);
522 >                error(WARNING, errmsg);
523 >        }
524 >        r_recv_first = r_recv_next = RAYQLEN;
525 >                                        /* close send pipes */
526 >        for (i = ray_pnprocs-nsub; i < ray_pnprocs; i++)
527 >                close(r_proc[i].fd_send);
528 >
529 >        if (nsub == 1) {                /* awaiting single process? */
530 >                if (waitpid(r_proc[ray_pnprocs-1].pid, &status, 0) < 0)
531                          status = 127<<8;
532 <                if (status) {
533 <                        sprintf(errmsg,
534 <                                "rendering process %d exited with code %d",
535 <                                        r_proc[ray_pnprocs].pid, status>>8);
536 <                        error(WARNING, errmsg);
532 >                close(r_proc[ray_pnprocs-1].fd_recv);
533 >        } else                          /* else unordered wait */
534 >                for (i = 0; i < nsub; ) {
535 >                        int     j, mystatus;
536 >                        RT_PID  pid = wait(&mystatus);
537 >                        if (pid < 0) {
538 >                                status = 127<<8;
539 >                                break;
540 >                        }
541 >                        for (j = ray_pnprocs-nsub; j < ray_pnprocs; j++)
542 >                                if (r_proc[j].pid == pid) {
543 >                                        if (mystatus)
544 >                                                status = mystatus;
545 >                                        close(r_proc[j].fd_recv);
546 >                                        ++i;
547 >                                }
548                  }
549 <                ray_pnidle--;
549 >        ray_pnprocs -= nsub;
550 >        ray_pnidle -= nsub;
551 >        if (status) {
552 >                sprintf(errmsg, "rendering process exited with code %d", status>>8);
553 >                error(WARNING, errmsg);
554          }
555          inclose--;
521 – }
522 –
523 –
524 – void
525 – quit(ec)                        /* make sure exit is called */
526 – int     ec;
527 – {
528 –        if (ray_pnprocs > 0)    /* close children if any */
529 –                ray_pclose(0);          
530 –        exit(ec);
556   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines