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

Comparing ray/src/gen/mkillum.c (file contents):
Revision 2.15 by greg, Fri Jun 27 22:27:45 2003 UTC vs.
Revision 2.23 by greg, Thu Oct 28 00:50:47 2004 UTC

# Line 8 | Line 8 | static const char RCSid[] = "$Id$";
8   #include  <signal.h>
9   #include  <ctype.h>
10  
11 – #include  "mkillum.h"
11   #include  "platform.h"
12 + #include  "rtprocess.h"
13 + #include  "mkillum.h"
14 + #include  "random.h"
15  
16                                  /* default parameters */
17   #define  SAMPDENS       48              /* points per projected steradian */
# Line 24 | Line 26 | static const char RCSid[] = "$Id$";
26  
27                                  /* rtrace command and defaults */
28   char  *rtargv[64] = { "rtrace", "-dj", ".25", "-dr", "3", "-dv-",
29 <                "-ab", "2", "-ad", "256", "-as", "128", "-aa", ".15", };
29 >                "-ab", "2", "-ad", "1024", "-as", "512", "-aa", ".1", };
30   int  rtargc = 14;
31                                  /* overriding rtrace options */
32   char  *myrtopts[] = { "-I-", "-i-", "-ld-", "-ov", "-h-",
33                          "-fff", "-y", "0", NULL };
34  
35 < struct rtproc   rt;             /* our rtrace process */
35 > struct rtproc   rt0;            /* head of rtrace process list */
36  
37   struct illum_args  thisillum = {        /* our illum and default values */
38                  0,
# Line 48 | Line 50 | int    matselect = S_ALL;      /* selection criterion */
50  
51   FUN     ofun[NUMOTYPE] = INIT_OTYPE;    /* object types */
52  
53 + char    persistfn[] = "pfXXXXXX";       /* persist file name */
54 +
55   int     gargc;                  /* global argc */
56   char    **gargv;                /* global argv */
57   #define  progname       gargv[0]
# Line 57 | Line 61 | int    doneheader = 0;         /* printed header yet? */
61  
62   int     warnings = 1;           /* print warnings? */
63  
64 + int done_rprocs(struct rtproc *rtp);
65 + void init(int np);
66 + void filter(register FILE       *infp, char     *name);
67 + void xoptions(char      *s, char        *nm);
68 + void printopts(void);
69 + void printhead(register int  ac, register char  **av);
70 + void xobject(FILE  *fp, char  *nm);
71  
72 < main(argc, argv)                /* compute illum distributions using rtrace */
73 < int     argc;
74 < char    *argv[];
72 >
73 > int
74 > main(           /* compute illum distributions using rtrace */
75 >        int     argc,
76 >        char    *argv[]
77 > )
78   {
79 +        int     nprocs = 1;
80          char    *rtpath;
81          FILE    *fp;
82          register int    i;
83                                  /* set global arguments */
84          gargv = argv;
85 +                                /* check for -n option */
86 +        if (!strcmp(argv[1], "-n")) {
87 +                nprocs = atoi(argv[2]);
88 +                if (nprocs <= 0)
89 +                        error(USER, "illegal number of processes");
90 +                i = 3;
91 +        } else
92 +                i = 1;
93                                  /* set up rtrace command */
94 <        for (i = 1; i < argc; i++) {
94 >        for ( ; i < argc; i++) {
95                  if (argv[i][0] == '<' && argv[i][1] == '\0')
96                          break;
97                  rtargv[rtargc++] = argv[i];
# Line 92 | Line 115 | char   *argv[];
115                          }
116          }
117          gargc = i;
118 <        rtargc--;
118 >        if (!strcmp(rtargv[--rtargc], "-defaults"))
119 >                nprocs = 0;
120 >        if (nprocs > 1) {       /* add persist file if parallel invocation */
121 >                rtargv[rtargc++] = "-PP";
122 >                rtargv[rtargc++] = mktemp(persistfn);
123 >        }
124 >                                /* add "mandatory" rtrace options */
125          for (i = 0; myrtopts[i] != NULL; i++)
126                  rtargv[rtargc++] = myrtopts[i];
127 +                                /* finally, put back final argument */
128          rtargv[rtargc++] = argv[gargc-1];
129          rtargv[rtargc] = NULL;
130 <                                /* just asking for defaults? */
101 <        if (!strcmp(argv[gargc-1], "-defaults")) {
130 >        if (!nprocs) {          /* just asking for defaults? */
131                  printopts(); fflush(stdout);
132                  rtpath = getpath(rtargv[0], getenv("PATH"), X_OK);
133                  if (rtpath == NULL) {
# Line 107 | Line 136 | char   *argv[];
136                          exit(1);
137                  }
138                  execv(rtpath, rtargv);
139 <                perror(rtpath);
139 >                perror(rtpath); /* execv() should not return */
140                  exit(1);
141          }
142          if (gargc < 2 || argv[gargc-1][0] == '-')
143                  error(USER, "missing octree argument");
144                                  /* else initialize and run our calculation */
145 <        init();
145 >        init(nprocs);
146          if (gargc+1 < argc)
147                  for (i = gargc+1; i < argc; i++) {
148                          if ((fp = fopen(argv[i], "r")) == NULL) {
# Line 129 | Line 158 | char   *argv[];
158          quit(0);
159   }
160  
161 + static void
162 + killpersist(void)                       /* kill persistent rtrace process */
163 + {
164 +        FILE    *fp = fopen(persistfn, "r");
165 +        int     pid;
166  
167 +        if (fp == NULL)
168 +                return;
169 +        if (fscanf(fp, "%*s %d", &pid) != 1 || kill(pid, SIGALRM) < 0)
170 +                unlink(persistfn);
171 +        fclose(fp);
172 + }
173 +
174 + int
175 + done_rprocs(struct rtproc *rtp)
176 + {
177 +        int     st0, st1 = 0;
178 +
179 +        if (rtp->next != NULL) {        /* close last opened first! */
180 +                st1 = done_rprocs(rtp->next);
181 +                free((void *)rtp->next);
182 +                rtp->next = NULL;
183 +        }
184 +        st0 = close_process(&rtp->pd);
185 +        if (st0 < 0)
186 +                error(WARNING, "unknown return status from rtrace process");
187 +        else if (st0 > 0)
188 +                return(st0);
189 +        return(st1);
190 + }
191 +
192   void
193 < quit(status)                    /* exit with status */
135 < int  status;
193 > quit(int status)                        /* exit with status */
194   {
195          int     rtstat;
196  
197 <        rtstat = close_process(&(rt.pd));
197 >        if (rt0.next != NULL)           /* terminate persistent rtrace */
198 >                killpersist();
199 >                                        /* clean up rtrace process(es) */
200 >        rtstat = done_rprocs(&rt0);
201          if (status == 0)
202 <                if (rtstat < 0)
142 <                        error(WARNING,
143 <                        "unknown return status from rtrace process");
144 <                else
145 <                        status = rtstat;
202 >                status = rtstat;
203          exit(status);
204   }
205  
206 <
207 < init()                          /* start rtrace and set up buffers */
206 > void
207 > init(int np)                            /* start rtrace and set up buffers */
208   {
209 <        extern int  o_face(), o_sphere(), o_ring();
209 >        struct rtproc   *rtp;
210 >        int     i;
211          int     maxbytes;
212                                          /* set up object functions */
213          ofun[OBJ_FACE].funp = o_face;
# Line 159 | Line 217 | init()                         /* start rtrace and set up buffers */
217   #ifdef SIGPIPE /* not present on Windows */
218          signal(SIGPIPE, quit);
219   #endif
220 <                                        /* start rtrace process */
221 <        errno = 0;
222 <        maxbytes = open_process(&(rt.pd), rtargv);
223 <        if (maxbytes == 0) {
224 <                eputs(rtargv[0]);
225 <                eputs(": command not found\n");
226 <                exit(1);
220 >        rtp = &rt0;                     /* start rtrace process(es) */
221 >        for (i = 0; i++ < np; ) {
222 >                errno = 0;
223 >                maxbytes = open_process(&rtp->pd, rtargv);
224 >                if (maxbytes == 0) {
225 >                        eputs(rtargv[0]);
226 >                        eputs(": command not found\n");
227 >                        exit(1);
228 >                }
229 >                if (maxbytes < 0)
230 >                        error(SYSTEM, "cannot start rtrace process");
231 >                if (!i && np > 1)
232 >                        sleep(2);       /* wait for persist file */
233 >                rtp->bsiz = maxbytes/(6*sizeof(float));
234 >                rtp->buf = (float *)malloc(6*sizeof(float)*rtp->bsiz--);
235 >                rtp->dest = (float **)calloc(rtp->bsiz, sizeof(float *));
236 >                if (rtp->buf == NULL || rtp->dest == NULL)
237 >                        error(SYSTEM, "out of memory in init");
238 >                rtp->nrays = 0;
239 >                if (i == np)            /* last process? */
240 >                        break;
241 >                rtp->next = (struct rtproc *)malloc(sizeof(struct rtproc));
242 >                if (rtp->next == NULL)
243 >                        error(SYSTEM, "out of memory in init");
244 >                rtp = rtp->next;
245          }
246 <        if (maxbytes < 0)
171 <                error(SYSTEM, "cannot start rtrace process");
172 <        rt.bsiz = maxbytes/(6*sizeof(float));
173 <        rt.buf = (float *)malloc(6*sizeof(float)*rt.bsiz--);
174 <        rt.dest = (float **)malloc(sizeof(float *)*rt.bsiz);
175 <        if (rt.buf == NULL || rt.dest == NULL)
176 <                error(SYSTEM, "out of memory in init");
177 <        rt.nrays = 0;
246 >        rtp->next = NULL;
247                                          /* set up urand */
248          initurand(16384);
249   }
250  
251  
252   void
253 < eputs(s)                                /* put string to stderr */
254 < register char  *s;
253 > eputs(                          /* put string to stderr */
254 >        register char  *s
255 > )
256   {
257          static int  midline = 0;
258  
# Line 205 | Line 275 | char  *s;
275   }
276  
277  
278 < filter(infp, name)              /* process stream */
279 < register FILE   *infp;
280 < char    *name;
278 > void
279 > filter(         /* process stream */
280 >        register FILE   *infp,
281 >        char    *name
282 > )
283   {
284          char    buf[512];
285          FILE    *pfp;
# Line 237 | Line 309 | char   *name;
309   }
310  
311  
312 < xoptions(s, nm)                 /* process options in string s */
313 < char    *s;
314 < char    *nm;
312 > void
313 > xoptions(                       /* process options in string s */
314 >        char    *s,
315 >        char    *nm
316 > )
317   {
318          extern FILE     *freopen();
319          char    buf[64];
# Line 398 | Line 472 | char   *nm;
472          printf("# %s", s+2);
473   }
474  
475 <
476 < printopts()                     /* print out option default values */
475 > void
476 > printopts(void)                 /* print out option default values */
477   {
478          printf("m=%-15s\t\t# material name\n", thisillum.matname);
479          printf("f=%-15s\t\t# data file name\n", thisillum.datafile);
# Line 420 | Line 494 | printopts()                    /* print out option default values */
494   }
495  
496  
497 < printhead(ac, av)                       /* print out header */
498 < register int  ac;
499 < register char  **av;
497 > void
498 > printhead(                      /* print out header */
499 >        register int  ac,
500 >        register char  **av
501 > )
502   {
503          putchar('#');
504          while (ac-- > 0) {
# Line 433 | Line 509 | register char  **av;
509   }
510  
511  
512 < xobject(fp, nm)                         /* translate an object from fp */
513 < FILE  *fp;
514 < char  *nm;
512 > void
513 > xobject(                                /* translate an object from fp */
514 >        FILE  *fp,
515 >        char  *nm
516 > )
517   {
518          OBJREC  thisobj;
519          char  str[MAXSTR];
# Line 486 | Line 564 | char  *nm;
564          checkhead();
565                                                  /* process object */
566          if (doit)
567 <                (*ofun[thisobj.otype].funp)(&thisobj, &thisillum, &rt, nm);
567 >                (*ofun[thisobj.otype].funp)(&thisobj, &thisillum, &rt0, nm);
568          else
569                  printobj(thisillum.altmat, &thisobj);
570                                                  /* free arguments */

Diff Legend

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