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

Comparing ray/src/util/rtcontrib.c (file contents):
Revision 1.55 by greg, Wed Dec 9 21:43:35 2009 UTC vs.
Revision 1.64 by greg, Sat Apr 16 00:39:07 2011 UTC

# Line 20 | Line 20 | static const char RCSid[] = "$Id$";
20   #define MAXMODLIST      1024            /* maximum modifiers we'll track */
21   #endif
22  
23 < int     treebufsiz = BUFSIZ;            /* current tree buffer size */
23 > ssize_t treebufsiz = BUFSIZ;            /* current tree buffer size */
24  
25   typedef double  DCOLOR[3];              /* double-precision color */
26  
# Line 92 | Line 92 | struct rtproc {
92          struct rtproc   *next;          /* next in list of processes */
93          SUBPROC         pd;             /* rtrace pipe descriptors */
94          unsigned long   raynum;         /* ray number for this tree */
95 <        int             bsiz;           /* ray tree buffer length */
95 >        size_t          bsiz;           /* ray tree buffer length */
96          char            *buf;           /* ray tree buffer */
97 <        int             nbr;            /* number of bytes from rtrace */
97 >        size_t          nbr;            /* number of bytes from rtrace */
98   };                              /* rtrace process buffer */
99  
100                                          /* rtrace command and defaults */
# Line 107 | Line 107 | int  rtargc = 9;
107   char            *myrtopts[] = { "-h-", "-x", "1", "-y", "0",
108                                  "-dt", "0", "-as", "0", "-aa", "0", NULL };
109  
110 < #define RTCOEFF         "-o~~TmWdp"     /* compute coefficients only */
111 < #define RTCONTRIB       "-o~~TmVdp"     /* compute ray contributions */
110 > #define RTCOEFF         "-o~~~TmWdp"    /* compute coefficients only */
111 > #define RTCONTRIB       "-o~~~TmVdp"    /* compute ray contributions */
112  
113   struct rtproc   rt0;                    /* head of rtrace process list */
114  
# Line 378 | Line 378 | main(int argc, char *argv[])
378          rtargv[rtargc++] = contrib ? RTCONTRIB : RTCOEFF;
379                                  /* just asking for defaults? */
380          if (!strcmp(argv[i], "-defaults")) {
381 <                char    sxres[16], syres[16];
381 >                char    nps[8], sxres[16], syres[16];
382                  char    *rtpath;
383 –                printf("-n %-2d\t\t\t\t# number of processes\n", nprocs);
383                  printf("-c %-5d\t\t\t# accumulated rays per record\n",
384                                  accumulate);
385                  printf("-V%c\t\t\t\t# output %s\n", contrib ? '+' : '-',
386                                  contrib ? "contributions" : "coefficients");
387                  fflush(stdout);                 /* report OUR options */
388 +                rtargv[rtargc++] = "-n";
389 +                sprintf(nps, "%d", nprocs);
390 +                rtargv[rtargc++] = nps;
391                  rtargv[rtargc++] = header ? "-h+" : "-h-";
392                  sprintf(fmt, "-f%c%c", inpfmt, outfmt);
393                  rtargv[rtargc++] = fmt;
# Line 543 | Line 545 | init(int np)
545                  raysleft = 0;
546          if ((account = accumulate) > 0)
547                  raysleft *= accumulate;
548 <        waitflush = xres;
548 >        waitflush = (yres > 0) & (xres > 1) ? 0 : xres;
549          if (!recover)
550                  return;
551                                          /* recover previous values */
# Line 700 | Line 702 | void
702   printheader(FILE *fout, const char *info)
703   {
704          extern char     VersionID[];
705 <        FILE            *fin = fopen(octree, "r");
706 <        
707 <        if (fin == NULL)
708 <                quit(1);
709 <        checkheader(fin, "ignore", fout);       /* copy octree header */
710 <        fclose(fin);
705 >                                                /* copy octree header */
706 >        if (octree[0] == '!') {
707 >                newheader("RADIANCE", fout);
708 >                fputs(octree+1, fout);
709 >                if (octree[strlen(octree)-1] != '\n')
710 >                        fputc('\n', fout);
711 >        } else {
712 >                FILE    *fin = fopen(octree, "r");
713 >                if (fin == NULL)
714 >                        quit(1);
715 >                checkheader(fin, "ignore", fout);
716 >                fclose(fin);
717 >        }
718          printargs(gargc-1, gargv, fout);        /* add our command */
719          fprintf(fout, "SOFTWARE= %s\n", VersionID);
720          fputnow(fout);
# Line 734 | Line 743 | printresolu(FILE *fout, int xr, int yr)
743   {
744          if ((xr > 0) & (yr > 0))        /* resolution string */
745                  fprtresolu(xr, yr, fout);
737 –        if (xres > 0)                   /* global flush flag */
738 –                fflush(fout);
746   }
747  
748   /* Get output stream pointer (open and write header if new and noopen==0) */
# Line 756 | Line 763 | getostream(const char *ospec, const char *mname, int b
763                          if (header)
764                                  printheader(stdout, NULL);
765                          printresolu(stdout, xres, yres);
766 +                        if (waitflush > 0)
767 +                                fflush(stdout);
768                          stdos.xr = xres; stdos.yr = yres;
769                          using_stdout = 1;
770                  }
# Line 824 | Line 833 | getostream(const char *ospec, const char *mname, int b
833                          if (outfmt == 'a')
834                                  putc('\n', sop->ofp);
835                  }
836 <                if (xres > 0)
836 >                if (waitflush > 0)
837                          fflush(sop->ofp);
838          }
839          sop->reclen += noopen;                  /* add to length if noopen */
# Line 991 | Line 1000 | done_contrib(int navg)
1000          if (using_stdout & (outfmt == 'a'))
1001                  putc('\n', stdout);
1002          if (!waitflush) {
1003 <                waitflush = xres;
1003 >                waitflush = (yres > 0) & (xres > 1) ? 0 : xres;
1004                  if (using_stdout)
1005                          fflush(stdout);
1006          }
# Line 1073 | Line 1082 | wait_rproc(void)
1082   {
1083          struct rtproc   *rtfree = NULL;
1084          fd_set          readset, errset;
1085 <        int             nr;
1085 >        ssize_t         nr;
1086          struct rtproc   *rt;
1087          int             n;
1088          
# Line 1104 | Line 1113 | wait_rproc(void)
1113                                  continue;
1114                          if (rt->buf == NULL) {
1115                                  rt->bsiz = treebufsiz;
1116 <                                rt->buf = (char *)malloc(treebufsiz);
1116 >                                rt->buf = (char *)malloc(rt->bsiz);
1117                          } else if (rt->nbr + BUFSIZ > rt->bsiz) {
1118                                  if (rt->bsiz + BUFSIZ <= treebufsiz)
1119                                          rt->bsiz = treebufsiz;
1120 <                                else
1121 <                                        treebufsiz = rt->bsiz += BUFSIZ;
1120 >                                else if ((treebufsiz = rt->bsiz += BUFSIZ) < 0)
1121 >                                        error(INTERNAL,
1122 >                                            "ray buffer does not fit memory");
1123                                  rt->buf = (char *)realloc(rt->buf, rt->bsiz);
1124                          }
1125                          if (rt->buf == NULL)
1126                                  error(SYSTEM, "out of memory in wait_rproc");
1127 <                        nr = read(rt->pd.r, rt->buf+rt->nbr, rt->bsiz-rt->nbr);
1128 <                        if (nr <= 0)
1127 >                        nr = rt->bsiz - rt->nbr;
1128 >                        if (nr & ~0x7fffffff)   /* avoid 32-bit OS issues */
1129 >                                nr = 0x7fffffff;
1130 >                        nr = read(rt->pd.r, rt->buf+rt->nbr, nr);
1131 >                        if (nr < 0)
1132 >                                error(SYSTEM, "read error from rtrace");
1133 >                        if (!nr)
1134                                  error(USER, "rtrace process died");
1135                          rt->nbr += nr;          /* advance & check */
1136 <                        if (rt->nbr >= 4 && !memcmp(rt->buf+rt->nbr-4,
1137 <                                                        "~\t~\t", 4)) {
1138 <                                rt->nbr -= 4;   /* elide terminator */
1136 >                        if (rt->nbr >= 6 && !memcmp(rt->buf+rt->nbr-6,
1137 >                                                        "~\t~\t~\t", 6)) {
1138 >                                rt->nbr -= 6;   /* elide terminator */
1139                                  queue_raytree(rt);
1140                                  rtfree = rt;    /* ready for next ray */
1141                          }
# Line 1162 | Line 1177 | trace_contribs(FILE *fin)
1177                                  lastray+1 < lastray) {
1178                          while (wait_rproc() != NULL)
1179                                  process_queue();
1180 <                        if (lastray+1 < lastray)
1166 <                                lastdone = lastray = 0;
1180 >                        lastdone = lastray = 0;
1181                  }
1182                  rtp = get_rproc();              /* get avail. rtrace process */
1183                  rtp->raynum = ++lastray;        /* assign ray */

Diff Legend

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