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.14 by greg, Wed Apr 5 06:22:57 2006 UTC vs.
Revision 2.45 by greg, Tue Aug 15 00:46:56 2023 UTC

# Line 17 | Line 17 | static const char      RCSid[] = "$Id$";
17   #include  "ambient.h"
18   #include  "random.h"
19   #include  "paths.h"
20 + #include  "pmapray.h"
21  
22 + extern char     *progname;              /* global argv[0] */
23 +
24 + extern char     *shm_boundary;          /* boundary of shared memory */
25 +
26                                          /* persistent processes define */
27   #ifdef  F_SETLKW
28   #define  PERSIST        1               /* normal persist */
# Line 25 | Line 30 | static const char      RCSid[] = "$Id$";
30   #define  PCHILD         3               /* child of normal persist */
31   #endif
32  
28 char  *progname;                        /* argv[0] */
29 char  *octname;                         /* octree name */
33   char  *sigerr[NSIG];                    /* signal error messages */
31 char  *shm_boundary = NULL;             /* boundary of shared memory */
34   char  *errfile = NULL;                  /* error output file */
35  
36 < extern char  *formstr();                /* string from format */
35 < extern int  inform;                     /* input format */
36 < extern int  outform;                    /* output format */
37 < extern char  *outvals;                  /* output values */
36 > int  nproc = 1;                         /* number of processes */
37  
38 < extern int  hresolu;                    /* horizontal resolution */
40 < extern int  vresolu;                    /* vertical resolution */
38 > extern int  setrtoutput(void);          /* set output values */
39  
40 < extern int  imm_irrad;                  /* compute immediate irradiance? */
41 < extern int  lim_dist;                   /* limit distance? */
40 > int  inform = 'a';                      /* input format */
41 > int  outform = 'a';                     /* output format */
42 > char  *outvals = "v";                   /* output specification */
43  
44 < extern char  *tralist[];                /* list of modifers to trace (or no) */
45 < extern int  traincl;                    /* include == 1, exclude == 0 */
44 > int  hresolu = 0;                       /* horizontal (scan) size */
45 > int  vresolu = 0;                       /* vertical resolution */
46  
47 + extern int  castonly;                   /* only doing ray-casting? */
48 +
49 + int  imm_irrad = 0;                     /* compute immediate irradiance? */
50 + int  lim_dist = 0;                      /* limit distance? */
51 +
52 + #ifndef MAXMODLIST
53 + #define MAXMODLIST      1024            /* maximum modifiers we'll track */
54 + #endif
55 +
56 + extern void  (*addobjnotify[])();       /* object notification calls */
57 + extern void  tranotify(OBJECT obj);
58 +
59 + char  *tralist[MAXMODLIST];             /* list of modifers to trace (or no) */
60 + int  traincl = -1;                      /* include == 1, exclude == 0 */
61 +
62   static int  loadflags = ~IO_FILES;      /* what to load from octree */
63  
64   static void onsig(int  signo);
65   static void sigdie(int  signo, char  *msg);
66   static void printdefaults(void);
67  
68 + #ifdef PERSIST
69 + #define RTRACE_FEATURES "Persist\nParallelPersist\nMultiprocessing\n" \
70 +                        "IrradianceCalc\nImmediateIrradiance\nDistanceLimiting\n" \
71 +                        "HessianAmbientCache\nAmbientAveraging\n" \
72 +                        "AmbientValueSharing\nAdaptiveShadowTesting\n" \
73 +                        "InputFormats=a,f,d\nOutputFormats=a,f,d,c\n" \
74 +                        "Outputs=o,d,v,V,w,W,l,L,c,p,n,N,s,m,M,r,x,R,X,~\n"
75 + #else
76 + #define RTRACE_FEATURES "IrradianceCalc\nIrradianceCalc\nDistanceLimiting\n" \
77 +                        "HessianAmbientCache\nAmbientAveraging\n" \
78 +                        "AmbientValueSharing\nAdaptiveShadowTesting\n" \
79 +                        "InputFormats=a,f,d\nOutputFormats=a,f,d,c\n" \
80 +                        "Outputs=o,d,v,V,w,W,l,L,c,p,n,N,s,m,M,r,x,R,X,~\n"
81 + #endif
82  
83 +
84   int
85   main(int  argc, char  *argv[])
86   {
87   #define  check(ol,al)           if (argv[i][ol] || \
88                                  badarg(argc-i-1,argv+i+1,al)) \
89                                  goto badopt
90 < #define  bool(olen,var)         switch (argv[i][olen]) { \
90 > #define  check_bool(olen,var)           switch (argv[i][olen]) { \
91                                  case '\0': var = !var; break; \
92                                  case 'y': case 'Y': case 't': case 'T': \
93                                  case '+': case '1': var = 1; break; \
94                                  case 'n': case 'N': case 'f': case 'F': \
95                                  case '-': case '0': var = 0; break; \
96                                  default: goto badopt; }
97 +        extern char  *octname;
98          int  persist = 0;
99 <        char  **tralp;
100 <        int  duped1;
99 >        char  *octnm = NULL;
100 >        char  **tralp = NULL;
101 >        int  duped1 = -1;
102          int  rval;
103          int  i;
104                                          /* global program name */
105          progname = argv[0] = fixargv0(argv[0]);
106 +                                        /* feature check only? */
107 +        strcat(RFeatureList, RTRACE_FEATURES);
108 +        if (argc > 1 && !strcmp(argv[1], "-features"))
109 +                return feature_status(argc-2, argv+2);
110 +                                        /* add trace notify function */
111 +        for (i = 0; addobjnotify[i] != NULL; i++)
112 +                ;
113 +        addobjnotify[i] = tranotify;
114                                          /* option city */
115          for (i = 1; i < argc; i++) {
116                                                  /* expand arguments */
# Line 98 | Line 137 | main(int  argc, char  *argv[])
137                          continue;
138                  }
139                  switch (argv[i][1]) {
140 +                case 'n':                               /* number of cores */
141 +                        check(2,"i");
142 +                        nproc = atoi(argv[++i]);
143 +                        if (nproc <= 0)
144 +                                error(USER, "bad number of processes");
145 +                        break;
146                  case 'x':                               /* x resolution */
147                          check(2,"i");
148                          hresolu = atoi(argv[++i]);
# Line 108 | Line 153 | main(int  argc, char  *argv[])
153                          break;
154                  case 'w':                               /* warnings */
155                          rval = erract[WARNING].pf != NULL;
156 <                        bool(2,rval);
156 >                        check_bool(2,rval);
157                          if (rval) erract[WARNING].pf = wputs;
158                          else erract[WARNING].pf = NULL;
159                          break;
# Line 119 | Line 164 | main(int  argc, char  *argv[])
164                  case 'l':                               /* limit distance */
165                          if (argv[i][2] != 'd')
166                                  goto badopt;
167 <                        bool(3,lim_dist);
167 >                        check_bool(3,lim_dist);
168                          break;
169                  case 'I':                               /* immed. irradiance */
170 <                        bool(2,imm_irrad);
170 >                        check_bool(2,imm_irrad);
171                          break;
172                  case 'f':                               /* format i/o */
173                          switch (argv[i][2]) {
# Line 154 | Line 199 | main(int  argc, char  *argv[])
199                          break;
200                  case 'h':                               /* header output */
201                          rval = loadflags & IO_INFO;
202 <                        bool(2,rval);
202 >                        check_bool(2,rval);
203                          loadflags = rval ? loadflags | IO_INFO :
204                                          loadflags & ~IO_INFO;
205                          break;
# Line 168 | Line 213 | main(int  argc, char  *argv[])
213                                          tralp = tralist;
214                                  }
215                                  if (argv[i][2] == 'I') {        /* file */
216 <                                        rval = wordfile(tralp,
216 >                                        rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
217                                          getpath(argv[++i],getrlibpath(),R_OK));
218                                          if (rval < 0) {
219                                                  sprintf(errmsg,
# Line 190 | Line 235 | main(int  argc, char  *argv[])
235                                          tralp = tralist;
236                                  }
237                                  if (argv[i][2] == 'E') {        /* file */
238 <                                        rval = wordfile(tralp,
238 >                                        rval = wordfile(tralp, MAXMODLIST-(tralp-tralist),
239                                          getpath(argv[++i],getrlibpath(),R_OK));
240                                          if (rval < 0) {
241                                                  sprintf(errmsg,
# Line 224 | Line 269 | main(int  argc, char  *argv[])
269                          goto badopt;
270                  }
271          }
272 +        if (nproc > 1 && persist)
273 +                error(USER, "multiprocessing incompatible with persist file");
274                                          /* initialize object types */
275          initotypes();
276                                          /* initialize urand */
# Line 265 | Line 312 | main(int  argc, char  *argv[])
312   #endif
313                                          /* get octree */
314          if (i == argc)
315 <                octname = NULL;
315 >                octnm = NULL;
316          else if (i == argc-1)
317 <                octname = argv[i];
317 >                octnm = argv[i];
318          else
319                  goto badopt;
320 <        if (octname == NULL)
320 >        if (octnm == NULL)
321                  error(USER, "missing octree argument");
322                                          /* set up output */
323   #ifdef  PERSIST
# Line 281 | Line 328 | main(int  argc, char  *argv[])
328   #endif
329          if (outform != 'a')
330                  SET_FILE_BINARY(stdout);
331 +        rval = setrtoutput();
332 +        octname = savqstr(octnm);
333          readoct(octname, loadflags, &thescene, NULL);
334          nsceneobjs = nobjects;
335  
# Line 288 | Line 337 | main(int  argc, char  *argv[])
337                  printargs(i, argv, stdout);
338                  printf("SOFTWARE= %s\n", VersionID);
339                  fputnow(stdout);
340 +                if (rval > 0)           /* saved from setrtoutput() call */
341 +                        printf("NCOMP=%d\n", rval);
342 +                if ((outform == 'f') | (outform == 'd'))
343 +                        fputendian(stdout);
344                  fputformat(formstr(outform), stdout);
345                  putchar('\n');
346          }
347  
348 <        marksources();                  /* find and mark sources */
348 >        if (!castonly) {        /* any actual ray traversal to do? */
349  
350 <        setambient();                   /* initialize ambient calculation */
350 >                ray_init_pmap();        /* PMAP: set up & load photon maps */
351 >                
352 >                marksources();          /* find and mark sources */
353  
354 +                setambient();           /* initialize ambient calculation */
355 +        } else
356 +                distantsources();       /* else mark only distant sources */
357 +
358 +        fflush(stdout);                 /* in case we're duplicating header */
359 +
360   #ifdef  PERSIST
361          if (persist) {
301                fflush(stdout);
362                                                  /* reconnect stdout */
363                  dup2(duped1, fileno(stdout));
364                  close(duped1);
# Line 309 | Line 369 | main(int  argc, char  *argv[])
369                          while ((rval=fork()) == 0) {    /* keep on forkin' */
370                                  pflock(1);
371                                  pfhold();
372 +                                ambsync();              /* load new values */
373                          }
374                          if (rval < 0)
375                                  error(SYSTEM, "cannot fork child for persist function");
376 <                        pfdetach();             /* parent exits */
376 >                        pfdetach();             /* parent will run then exit */
377                  }
378          }
379   runagain:
# Line 320 | Line 381 | runagain:
381                  dupheader();                    /* send header to stdout */
382   #endif
383                                          /* trace rays */
384 <        rtrace(NULL);
384 >        rtrace(NULL, nproc);
385                                          /* flush ambient file */
386          ambsync();
387   #ifdef  PERSIST
# Line 335 | Line 396 | runagain:
396                  }
397          }
398          if (persist == PCHILD) {        /* wait for a signal then go again */
338                close(duped1);                  /* release output handle */
399                  pfhold();
400                  raynum = nrays = 0;             /* reinitialize */
401                  goto runagain;
402          }
403   #endif
404 +
405 +        ray_done_pmap();           /* PMAP: free photon maps */
406 +        
407          quit(0);
408  
409   badopt:
# Line 349 | Line 412 | badopt:
412          return 1; /* pro forma return */
413  
414   #undef  check
415 < #undef  bool
415 > #undef  check_bool
416   }
417  
418  
419   void
420   wputs(                          /* warning output function */
421 <        char    *s
421 >        const char      *s
422   )
423   {
424          int  lasterrno = errno;
# Line 366 | Line 429 | wputs(                         /* warning output function */
429  
430   void
431   eputs(                          /* put string to stderr */
432 <        register char  *s
432 >        const char  *s
433   )
434   {
435          static int  midline = 0;
# Line 421 | Line 484 | sigdie(                        /* set fatal signal */
484   static void
485   printdefaults(void)                     /* print default values to stdout */
486   {
487 <        register char  *cp;
487 >        char  *cp;
488  
489          if (imm_irrad)
490                  printf("-I+\t\t\t\t# immediate irradiance on\n");
491 <        printf("-x  %-9d\t\t\t# x resolution (flush interval)\n", hresolu);
492 <        printf("-y  %-9d\t\t\t# y resolution\n", vresolu);
491 >        printf("-n %-2d\t\t\t\t# number of rendering processes\n", nproc);
492 >        printf("-x %-9d\t\t\t# %s\n", hresolu,
493 >                        vresolu && hresolu ? "x resolution" : "flush interval");
494 >        printf("-y %-9d\t\t\t# y resolution\n", vresolu);
495          printf(lim_dist ? "-ld+\t\t\t\t# limit distance on\n" :
496                          "-ld-\t\t\t\t# limit distance off\n");
497          printf("-h%c\t\t\t\t# %s header\n", loadflags & IO_INFO ? '+' : '-',
# Line 439 | Line 504 | printdefaults(void)                    /* print default values to stdou
504                  case 't': case 'T': printf(" trace"); break;
505                  case 'o': printf(" origin"); break;
506                  case 'd': printf(" direction"); break;
507 +                case 'r': printf(" reflect_contrib"); break;
508 +                case 'R': printf(" reflect_length"); break;
509 +                case 'x': printf(" unreflect_contrib"); break;
510 +                case 'X': printf(" unreflect_length"); break;
511                  case 'v': printf(" value"); break;
512                  case 'V': printf(" contribution"); break;
513                  case 'l': printf(" length"); break;
# Line 451 | Line 520 | printdefaults(void)                    /* print default values to stdou
520                  case 'W': printf(" coefficient"); break;
521                  case 'm': printf(" modifier"); break;
522                  case 'M': printf(" material"); break;
523 <                case '-': printf(" stroke"); break;
523 >                case '~': printf(" tilde"); break;
524                  }
525          putchar('\n');
526          printf(erract[WARNING].pf != NULL ?

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines