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

Comparing ray/src/rt/rtmain.c (file contents):
Revision 2.9 by schorsch, Tue Mar 30 16:13:01 2004 UTC vs.
Revision 2.58 by greg, Tue Apr 22 17:12:25 2025 UTC

# Line 7 | Line 7 | static const char      RCSid[] = "$Id$";
7  
8   #include "copyright.h"
9  
10 – #include  <sys/types.h>
10   #include  <signal.h>
11  
12   #include  "platform.h"
13   #include  "rtprocess.h" /* getpid() */
14   #include  "resolu.h"
15   #include  "ray.h"
16 + #include  "func.h"
17   #include  "source.h"
18   #include  "ambient.h"
19   #include  "random.h"
20   #include  "paths.h"
21 + #include  "pmapray.h"
22  
23 + extern char     *progname;              /* global argv[0] */
24 +
25                                          /* persistent processes define */
26   #ifdef  F_SETLKW
27   #define  PERSIST        1               /* normal persist */
# Line 26 | Line 29 | static const char      RCSid[] = "$Id$";
29   #define  PCHILD         3               /* child of normal persist */
30   #endif
31  
29 – char  *progname;                        /* argv[0] */
30 – char  *octname;                         /* octree name */
32   char  *sigerr[NSIG];                    /* signal error messages */
32 – char  *shm_boundary = NULL;             /* boundary of shared memory */
33   char  *errfile = NULL;                  /* error output file */
34  
35 < extern char  *formstr();                /* string from format */
36 < extern int  inform;                     /* input format */
37 < extern int  outform;                    /* output format */
38 < extern char  *outvals;                  /* output values */
35 > int  nproc = 1;                         /* number of processes */
36  
37 < extern int  hresolu;                    /* horizontal resolution */
41 < extern int  vresolu;                    /* vertical resolution */
37 > extern int  setrtoutput(void);          /* set output values */
38  
39 < extern int  imm_irrad;                  /* compute immediate irradiance? */
40 < extern int  lim_dist;                   /* limit distance? */
39 > int  inform = 'a';                      /* input format */
40 > int  outform = 'a';                     /* output format */
41 > char  *outvals = "v";                   /* output specification */
42  
43 < extern char  *tralist[];                /* list of modifers to trace (or no) */
44 < extern int  traincl;                    /* include == 1, exclude == 0 */
43 > int  hresolu = 0;                       /* horizontal (scan) size */
44 > int  vresolu = 0;                       /* vertical resolution */
45  
46 + extern int  castonly;                   /* only doing ray-casting? */
47 +
48 + int  imm_irrad = 0;                     /* compute immediate irradiance? */
49 + int  lim_dist = 0;                      /* limit distance? */
50 +
51 + #ifndef MAXMODLIST
52 + #define MAXMODLIST      1024            /* maximum modifiers we'll track */
53 + #endif
54 +
55 + extern void  (*addobjnotify[])();       /* object notification calls */
56 + extern void  tranotify(OBJECT obj);
57 +
58 + char  *tralist[MAXMODLIST];             /* list of modifers to trace (or no) */
59 + int  traincl = -1;                      /* include == 1, exclude == 0 */
60 +
61 + double  (*sens_curve)(const SCOLOR scol) = NULL;        /* spectral conversion for 1-channel */
62 + double  out_scalefactor = 1;            /* output calibration scale factor */
63 + RGBPRIMP  out_prims = stdprims;         /* output color primitives (NULL if spectral) */
64 + static RGBPRIMS  our_prims;             /* private output color primitives */
65 +
66 + static int  loadflags = ~IO_FILES;      /* what to load from octree */
67 +
68   static void onsig(int  signo);
69   static void sigdie(int  signo, char  *msg);
70   static void printdefaults(void);
71  
72 + #ifdef PERSIST
73 + #define RTRACE_FEATURES "Persist\nParallelPersist\nMultiprocessing\n" \
74 +                        "IrradianceCalc\nImmediateIrradiance\nDistanceLimiting\n" \
75 +                        "ParticipatingMedia=Mist\n" \
76 +                        "HessianAmbientCache\nAmbientAveraging\n" \
77 +                        "AmbientValueSharing\nAdaptiveShadowTesting\n" \
78 +                        "InputFormats=a,f,d\nOutputFormats=a,f,d,c\n" \
79 +                        "Outputs=o,d,v,V,w,W,l,L,c,p,n,N,s,m,M,r,x,R,X,~\n" \
80 +                        "OutputCS=RGB,XYZ,Y,S,M,prims,spec\n"
81 + #else
82 + #define RTRACE_FEATURES "IrradianceCalc\nIrradianceCalc\nDistanceLimiting\n" \
83 +                        "ParticipatingMedia=Mist\n" \
84 +                        "HessianAmbientCache\nAmbientAveraging\n" \
85 +                        "AmbientValueSharing\nAdaptiveShadowTesting\n" \
86 +                        "InputFormats=a,f,d\nOutputFormats=a,f,d,c\n" \
87 +                        "Outputs=o,d,v,V,w,W,l,L,c,p,n,N,s,m,M,r,x,R,X,~\n" \
88 +                        "OutputCS=RGB,XYZ,Y,S,M,prims,spec\n"
89 + #endif
90  
91 +
92   int
93   main(int  argc, char  *argv[])
94   {
95   #define  check(ol,al)           if (argv[i][ol] || \
96                                  badarg(argc-i-1,argv+i+1,al)) \
97                                  goto badopt
98 < #define  bool(olen,var)         switch (argv[i][olen]) { \
98 > #define  check_bool(olen,var)           switch (argv[i][olen]) { \
99                                  case '\0': var = !var; break; \
100                                  case 'y': case 'Y': case 't': case 'T': \
101                                  case '+': case '1': var = 1; break; \
102                                  case 'n': case 'N': case 'f': case 'F': \
103                                  case '-': case '0': var = 0; break; \
104                                  default: goto badopt; }
105 <        int  loadflags = ~IO_FILES;
105 >        extern char  *octname;
106          int  persist = 0;
107 <        char  **tralp;
108 <        int  duped1;
107 >        char  *octnm = NULL;
108 >        char  **tralp = NULL;
109 >        int  duped1 = -1;
110          int  rval;
111          int  i;
112                                          /* global program name */
113          progname = argv[0] = fixargv0(argv[0]);
114 +                                        /* feature check only? */
115 +        strcat(RFeatureList, RTRACE_FEATURES);
116 +        if (argc > 1 && !strcmp(argv[1], "-features"))
117 +                return feature_status(argc-2, argv+2);
118 +                                        /* initialize calcomp routines */
119 +        initfunc();
120 +                                        /* add trace notify function */
121 +        for (i = 0; addobjnotify[i] != NULL; i++)
122 +                ;
123 +        addobjnotify[i] = tranotify;
124                                          /* option city */
125          for (i = 1; i < argc; i++) {
126                                                  /* expand arguments */
# Line 98 | Line 147 | main(int  argc, char  *argv[])
147                          continue;
148                  }
149                  switch (argv[i][1]) {
150 +                case 'n':                               /* number of cores */
151 +                        check(2,"i");
152 +                        nproc = atoi(argv[++i]);
153 +                        if (nproc <= 0)
154 +                                error(USER, "bad number of processes");
155 +                        break;
156                  case 'x':                               /* x resolution */
157                          check(2,"i");
158                          hresolu = atoi(argv[++i]);
# Line 106 | Line 161 | main(int  argc, char  *argv[])
161                          check(2,"i");
162                          vresolu = atoi(argv[++i]);
163                          break;
164 <                case 'w':                               /* warnings */
164 >                case 'w':                               /* warnings & spectral */
165                          rval = erract[WARNING].pf != NULL;
166 <                        bool(2,rval);
166 >                        check_bool(2,rval);
167                          if (rval) erract[WARNING].pf = wputs;
168                          else erract[WARNING].pf = NULL;
169                          break;
# Line 119 | Line 174 | main(int  argc, char  *argv[])
174                  case 'l':                               /* limit distance */
175                          if (argv[i][2] != 'd')
176                                  goto badopt;
177 <                        bool(3,lim_dist);
177 >                        check_bool(3,lim_dist);
178                          break;
179                  case 'I':                               /* immed. irradiance */
180 <                        bool(2,imm_irrad);
180 >                        check_bool(2,imm_irrad);
181                          break;
182                  case 'f':                               /* format i/o */
183                          switch (argv[i][2]) {
# Line 154 | Line 209 | main(int  argc, char  *argv[])
209                          break;
210                  case 'h':                               /* header output */
211                          rval = loadflags & IO_INFO;
212 <                        bool(2,rval);
212 >                        check_bool(2,rval);
213                          loadflags = rval ? loadflags | IO_INFO :
214                                          loadflags & ~IO_INFO;
215                          break;
# Line 168 | Line 223 | main(int  argc, char  *argv[])
223                                          tralp = tralist;
224                                  }
225                                  if (argv[i][2] == 'I') {        /* file */
226 <                                        rval = wordfile(tralp,
226 >                                        rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
227                                          getpath(argv[++i],getrlibpath(),R_OK));
228                                          if (rval < 0) {
229                                                  sprintf(errmsg,
# Line 190 | Line 245 | main(int  argc, char  *argv[])
245                                          tralp = tralist;
246                                  }
247                                  if (argv[i][2] == 'E') {        /* file */
248 <                                        rval = wordfile(tralp,
248 >                                        rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
249                                          getpath(argv[++i],getrlibpath(),R_OK));
250                                          if (rval < 0) {
251                                                  sprintf(errmsg,
# Line 208 | Line 263 | main(int  argc, char  *argv[])
263                                  goto badopt;
264                          }
265                          break;
266 +                case 'p':                               /* value output */
267 +                        switch (argv[i][2]) {
268 +                        case 'R':                       /* standard RGB output */
269 +                                if (strcmp(argv[i]+2, "RGB"))
270 +                                        goto badopt;
271 +                                out_prims = stdprims;
272 +                                out_scalefactor = 1;
273 +                                sens_curve = NULL;
274 +                                break;
275 +                        case 'X':                       /* XYZ output */
276 +                                if (strcmp(argv[i]+2, "XYZ"))
277 +                                        goto badopt;
278 +                                out_prims = xyzprims;
279 +                                out_scalefactor = WHTEFFICACY;
280 +                                sens_curve = NULL;
281 +                                break;
282 +                        case 'c': {
283 +                                int     j;
284 +                                check(3,"ffffffff");
285 +                                rval = 0;
286 +                                for (j = 0; j < 8; j++) {
287 +                                        our_prims[0][j] = atof(argv[++i]);
288 +                                        rval |= fabs(our_prims[0][j]-stdprims[0][j]) > .001;
289 +                                }
290 +                                if (rval) {
291 +                                        if (!colorprimsOK(our_prims))
292 +                                                error(USER, "illegal primary chromaticities");
293 +                                        out_prims = our_prims;
294 +                                } else
295 +                                        out_prims = stdprims;
296 +                                out_scalefactor = 1;
297 +                                sens_curve = NULL;
298 +                                } break;
299 +                        case 'Y':                       /* photopic response */
300 +                                if (argv[i][3])
301 +                                        goto badopt;
302 +                                sens_curve = scolor_photopic;
303 +                                out_scalefactor = WHTEFFICACY;
304 +                                break;
305 +                        case 'S':                       /* scotopic response */
306 +                                if (argv[i][3])
307 +                                        goto badopt;
308 +                                sens_curve = scolor_scotopic;
309 +                                out_scalefactor = WHTSCOTOPIC;
310 +                                break;
311 +                        case 'M':                       /* melanopic response */
312 +                                if (argv[i][3])
313 +                                        goto badopt;
314 +                                sens_curve = scolor_melanopic;
315 +                                out_scalefactor = WHTMELANOPIC;
316 +                                break;
317 +                        default:
318 +                                goto badopt;
319 +                        }
320 +                        break;
321 + #if MAXCSAMP>3
322 +                case 'c':                               /* output spectral results */
323 +                        if (argv[i][2] != 'o')
324 +                                goto badopt;
325 +                        rval = (out_prims == NULL) & (sens_curve == NULL);
326 +                        check_bool(3,rval);
327 +                        if (rval) {
328 +                                out_prims = NULL;
329 +                                sens_curve = NULL;
330 +                        } else if (out_prims == NULL)
331 +                                out_prims = stdprims;
332 +                        break;
333 + #endif
334   #ifdef  PERSIST
335                  case 'P':                               /* persist file */
336                          if (argv[i][2] == 'P') {
# Line 224 | Line 347 | main(int  argc, char  *argv[])
347                          goto badopt;
348                  }
349          }
350 +                                        /* set/check spectral sampling */
351 +        rval = setspectrsamp(CNDX, WLPART);
352 +        if (rval < 0)
353 +                error(USER, "unsupported spectral sampling");
354 +        if (sens_curve != NULL)
355 +                out_prims = NULL;
356 +        else if (out_prims != NULL) {
357 +                if (!rval)
358 +                        error(WARNING, "spectral range incompatible with color output");
359 +        } else if (NCSAMP == 3)
360 +                out_prims = stdprims;   /* 3 samples do not a spectrum make */
361 +        if (nproc > 1 && persist)
362 +                error(USER, "multiprocessing incompatible with persist file");
363                                          /* initialize object types */
364          initotypes();
365                                          /* initialize urand */
366 <        initurand(2048);
366 >        if (rand_samp) {
367 >                srandom((long)time(0));
368 >                initurand(0);
369 >        } else {
370 >                srandom(0L);
371 >                initurand(2048);
372 >        }
373                                          /* set up signal handling */
374          sigdie(SIGINT, "Interrupt");
375   #ifdef SIGHUP
# Line 259 | Line 401 | main(int  argc, char  *argv[])
401   #endif
402                                          /* get octree */
403          if (i == argc)
404 <                octname = NULL;
404 >                octnm = NULL;
405          else if (i == argc-1)
406 <                octname = argv[i];
406 >                octnm = argv[i];
407          else
408                  goto badopt;
409 <        if (octname == NULL)
409 >        if (octnm == NULL)
410                  error(USER, "missing octree argument");
411                                          /* set up output */
412   #ifdef  PERSIST
# Line 273 | Line 415 | main(int  argc, char  *argv[])
415                  openheader();
416          }
417   #endif
276 – #ifdef  _WIN32
418          if (outform != 'a')
419                  SET_FILE_BINARY(stdout);
420 <        if (octname == NULL)
421 <                SET_FILE_BINARY(stdin);
281 < #endif
420 >        rval = setrtoutput();
421 >        octname = savqstr(octnm);
422          readoct(octname, loadflags, &thescene, NULL);
423          nsceneobjs = nobjects;
424  
# Line 286 | Line 426 | main(int  argc, char  *argv[])
426                  printargs(i, argv, stdout);
427                  printf("SOFTWARE= %s\n", VersionID);
428                  fputnow(stdout);
429 +                if (rval > 0)           /* saved from setrtoutput() call */
430 +                        fputncomp(rval, stdout);
431 +                if (NCSAMP > 3)
432 +                        fputwlsplit(WLPART, stdout);
433 +                if ((out_prims != stdprims) & (out_prims != NULL))
434 +                        fputprims(out_prims, stdout);
435 +                if ((outform == 'f') | (outform == 'd'))
436 +                        fputendian(stdout);
437                  fputformat(formstr(outform), stdout);
438 <                putchar('\n');
438 >                fputc('\n', stdout);    /* end of header */
439          }
440  
441 <        marksources();                  /* find and mark sources */
441 >        if (!castonly) {        /* any actual ray traversal to do? */
442  
443 <        setambient();                   /* initialize ambient calculation */
443 >                ray_init_pmap();        /* PMAP: set up & load photon maps */
444 >                
445 >                marksources();          /* find and mark sources */
446  
447 +                setambient();           /* initialize ambient calculation */
448 +        } else
449 +                distantsources();       /* else mark only distant sources */
450 +
451 +        fflush(stdout);                 /* in case we're duplicating header */
452 +
453   #ifdef  PERSIST
454          if (persist) {
299 –                fflush(stdout);
455                                                  /* reconnect stdout */
456                  dup2(duped1, fileno(stdout));
457                  close(duped1);
458                  if (persist == PARALLEL) {      /* multiprocessing */
459 <                        preload_objs();         /* preload scene */
305 <                        shm_boundary = (char *)malloc(16);
306 <                        strcpy(shm_boundary, "SHM_BOUNDARY");
459 >                        cow_memshare();         /* preloads scene */
460                          while ((rval=fork()) == 0) {    /* keep on forkin' */
461                                  pflock(1);
462                                  pfhold();
463 +                                ambsync();              /* load new values */
464                          }
465                          if (rval < 0)
466                                  error(SYSTEM, "cannot fork child for persist function");
467 <                        pfdetach();             /* parent exits */
467 >                        pfdetach();             /* parent will run then exit */
468                  }
469          }
470   runagain:
# Line 318 | Line 472 | runagain:
472                  dupheader();                    /* send header to stdout */
473   #endif
474                                          /* trace rays */
475 <        rtrace(NULL);
475 >        rtrace(NULL, nproc);
476                                          /* flush ambient file */
477          ambsync();
478   #ifdef  PERSIST
# Line 333 | Line 487 | runagain:
487                  }
488          }
489          if (persist == PCHILD) {        /* wait for a signal then go again */
336 –                close(duped1);                  /* release output handle */
490                  pfhold();
491                  raynum = nrays = 0;             /* reinitialize */
492                  goto runagain;
493          }
494   #endif
495 +
496 +        ray_done_pmap();           /* PMAP: free photon maps */
497 +        
498          quit(0);
499  
500   badopt:
# Line 347 | Line 503 | badopt:
503          return 1; /* pro forma return */
504  
505   #undef  check
506 < #undef  bool
506 > #undef  check_bool
507   }
508  
509  
510   void
511   wputs(                          /* warning output function */
512 <        char    *s
512 >        const char      *s
513   )
514   {
515          int  lasterrno = errno;
516 +        if (erract[WARNING].pf == NULL)
517 +                return;         /* called by calcomp or someone */
518          eputs(s);
519          errno = lasterrno;
520   }
# Line 364 | Line 522 | wputs(                         /* warning output function */
522  
523   void
524   eputs(                          /* put string to stderr */
525 <        register char  *s
525 >        const char  *s
526   )
527   {
528          static int  midline = 0;
# Line 419 | Line 577 | sigdie(                        /* set fatal signal */
577   static void
578   printdefaults(void)                     /* print default values to stdout */
579   {
580 <        register char  *cp;
580 >        char  *cp;
581  
582 +        printf(erract[WARNING].pf != NULL ?
583 +                        "-w+\t\t\t\t# warning messages on\n" :
584 +                        "-w-\t\t\t\t# warning messages off\n");
585          if (imm_irrad)
586                  printf("-I+\t\t\t\t# immediate irradiance on\n");
587 <        printf("-x  %-9d\t\t\t# x resolution\n", hresolu);
588 <        printf("-y  %-9d\t\t\t# y resolution\n", vresolu);
587 >        printf("-n %-2d\t\t\t\t# number of rendering processes\n", nproc);
588 >        printf("-x %-9d\t\t\t# %s\n", hresolu,
589 >                        vresolu && hresolu ? "x resolution" : "flush interval");
590 >        printf("-y %-9d\t\t\t# y resolution\n", vresolu);
591          printf(lim_dist ? "-ld+\t\t\t\t# limit distance on\n" :
592                          "-ld-\t\t\t\t# limit distance off\n");
593 +        printf(loadflags & IO_INFO ? "-h+\t\t\t\t# output header\n" :
594 +                        "-h-\t\t\t\t# no header\n");
595          printf("-f%c%c\t\t\t\t# format input/output = %s/%s\n",
596                          inform, outform, formstr(inform), formstr(outform));
597 <        printf("-o%s\t\t\t\t# output", outvals);
597 >        printf("-o%-9s\t\t\t# output", outvals);
598          for (cp = outvals; *cp; cp++)
599                  switch (*cp) {
600 <                case 't': printf(" trace"); break;
600 >                case 't': case 'T': printf(" trace"); break;
601                  case 'o': printf(" origin"); break;
602                  case 'd': printf(" direction"); break;
603 +                case 'r': printf(" reflect_contrib"); break;
604 +                case 'R': printf(" reflect_length"); break;
605 +                case 'x': printf(" unreflect_contrib"); break;
606 +                case 'X': printf(" unreflect_length"); break;
607                  case 'v': printf(" value"); break;
608 +                case 'V': printf(" contribution"); break;
609                  case 'l': printf(" length"); break;
610                  case 'L': printf(" first_length"); break;
611                  case 'p': printf(" point"); break;
# Line 443 | Line 613 | printdefaults(void)                    /* print default values to stdou
613                  case 'N': printf(" unperturbed_normal"); break;
614                  case 's': printf(" surface"); break;
615                  case 'w': printf(" weight"); break;
616 +                case 'W': printf(" coefficient"); break;
617                  case 'm': printf(" modifier"); break;
618 +                case 'M': printf(" material"); break;
619 +                case '~': printf(" tilde"); break;
620                  }
621 <        putchar('\n');
622 <        printf(erract[WARNING].pf != NULL ?
623 <                        "-w+\t\t\t\t# warning messages on\n" :
624 <                        "-w-\t\t\t\t# warning messages off\n");
621 >        fputc('\n', stdout);
622 >        if (sens_curve == scolor_photopic)
623 >                printf("-pY\t\t\t\t# photopic output\n");
624 >        else if (sens_curve == scolor_scotopic)
625 >                printf("-pS\t\t\t\t# scotopic output\n");
626 >        else if (sens_curve == scolor_melanopic)
627 >                printf("-pM\t\t\t\t# melanopic output\n");
628 >        else if (out_prims == stdprims)
629 >                printf("-pRGB\t\t\t\t# standard RGB color output\n");
630 >        else if (out_prims == xyzprims)
631 >                printf("-pXYZ\t\t\t\t# CIE XYZ color output\n");
632 >        else if (out_prims != NULL)
633 >                printf("-pc %.4f %.4f %.4f %.4f %.4f %.4f %.4f %.4f\t# output color primaries and white point\n",
634 >                                out_prims[RED][0], out_prims[RED][1],
635 >                                out_prims[GRN][0], out_prims[GRN][1],
636 >                                out_prims[BLU][0], out_prims[BLU][1],
637 >                                out_prims[WHT][0], out_prims[WHT][1]);
638 >        if ((sens_curve == NULL) & (NCSAMP > 3))
639 >                printf(out_prims != NULL ? "-co-\t\t\t\t# output tristimulus colors\n" :
640 >                                "-co+\t\t\t\t# output spectral values\n");
641          print_rdefaults();
642   }

Diff Legend

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