1 |
/* Copyright (c) 1996 Regents of the University of California */ |
2 |
|
3 |
#ifndef lint |
4 |
static char SCCSid[] = "$SunId$ LBL"; |
5 |
#endif |
6 |
|
7 |
/* |
8 |
* Radiance animation control program |
9 |
*/ |
10 |
|
11 |
#include "standard.h" |
12 |
#include <ctype.h> |
13 |
#include <sys/types.h> |
14 |
#include <sys/stat.h> |
15 |
#include "view.h" |
16 |
#include "vars.h" |
17 |
#include "netproc.h" |
18 |
/* default remote shell */ |
19 |
#ifdef _AUX_SOURCE |
20 |
#define REMSH "remsh" |
21 |
#else |
22 |
#define REMSH "rsh" |
23 |
#endif |
24 |
/* input variables */ |
25 |
#define HOST 0 /* rendering host machine */ |
26 |
#define RENDER 1 /* rendering options */ |
27 |
#define PFILT 2 /* pfilt options */ |
28 |
#define PINTERP 3 /* pinterp options */ |
29 |
#define OCTREE 4 /* octree file name */ |
30 |
#define DIRECTORY 5 /* working (sub)directory */ |
31 |
#define BASENAME 6 /* output image base name */ |
32 |
#define VIEWFILE 7 /* animation frame views */ |
33 |
#define START 8 /* starting frame number */ |
34 |
#define END 9 /* ending frame number */ |
35 |
#define RIF 10 /* rad input file */ |
36 |
#define NEXTANIM 11 /* next animation file */ |
37 |
#define ANIMATE 12 /* animation command */ |
38 |
#define TRANSFER 13 /* frame transfer command */ |
39 |
#define ARCHIVE 14 /* archiving command */ |
40 |
#define INTERP 15 /* # frames to interpolate */ |
41 |
#define OVERSAMP 16 /* # times to oversample image */ |
42 |
#define MBLUR 17 /* samples for motion blur */ |
43 |
#define RTRACE 18 /* use rtrace with pinterp? */ |
44 |
#define DISKSPACE 19 /* how much disk space to use */ |
45 |
#define RESOLUTION 20 /* desired final resolution */ |
46 |
#define EXPOSURE 21 /* how to compute exposure */ |
47 |
#define RSH 22 /* remote shell script or program */ |
48 |
|
49 |
int NVARS = 23; /* total number of variables */ |
50 |
|
51 |
VARIABLE vv[] = { /* variable-value pairs */ |
52 |
{"host", 4, 0, NULL, NULL}, |
53 |
{"render", 3, 0, NULL, catvalues}, |
54 |
{"pfilt", 2, 0, NULL, catvalues}, |
55 |
{"pinterp", 2, 0, NULL, catvalues}, |
56 |
{"OCTREE", 3, 0, NULL, onevalue}, |
57 |
{"DIRECTORY", 3, 0, NULL, onevalue}, |
58 |
{"BASENAME", 3, 0, NULL, onevalue}, |
59 |
{"VIEWFILE", 2, 0, NULL, onevalue}, |
60 |
{"START", 2, 0, NULL, intvalue}, |
61 |
{"END", 2, 0, NULL, intvalue}, |
62 |
{"RIF", 3, 0, NULL, onevalue}, |
63 |
{"NEXTANIM", 3, 0, NULL, onevalue}, |
64 |
{"ANIMATE", 2, 0, NULL, onevalue}, |
65 |
{"TRANSFER", 2, 0, NULL, onevalue}, |
66 |
{"ARCHIVE", 2, 0, NULL, onevalue}, |
67 |
{"INTERPOLATE", 3, 0, NULL, intvalue}, |
68 |
{"OVERSAMPLE", 2, 0, NULL, fltvalue}, |
69 |
{"MBLUR", 2, 0, NULL, onevalue}, |
70 |
{"RTRACE", 2, 0, NULL, boolvalue}, |
71 |
{"DISKSPACE", 3, 0, NULL, fltvalue}, |
72 |
{"RESOLUTION", 3, 0, NULL, onevalue}, |
73 |
{"EXPOSURE", 3, 0, NULL, onevalue}, |
74 |
{"RSH", 3, 0, NULL, onevalue}, |
75 |
}; |
76 |
|
77 |
#define SFNAME "STATUS" /* status file name */ |
78 |
|
79 |
struct { |
80 |
char host[64]; /* control host name */ |
81 |
int pid; /* control process id */ |
82 |
char cfname[128]; /* control file name */ |
83 |
int rnext; /* next frame to render */ |
84 |
int fnext; /* next frame to filter */ |
85 |
int tnext; /* next frame to transfer */ |
86 |
} astat; /* animation status */ |
87 |
|
88 |
char *progname; /* our program name */ |
89 |
char *cfname; /* our control file name */ |
90 |
|
91 |
int nowarn = 0; /* turn warnings off? */ |
92 |
int silent = 0; /* silent mode? */ |
93 |
int noaction = 0; /* take no action? */ |
94 |
|
95 |
char *remsh; /* remote shell program/script */ |
96 |
char rendopt[2048]; /* rendering options */ |
97 |
char rresopt[32]; /* rendering resolution options */ |
98 |
char fresopt[32]; /* filter resolution options */ |
99 |
int pfiltalways; /* always use pfilt? */ |
100 |
|
101 |
char arcargs[10240]; /* files to archive */ |
102 |
char *arcfirst, *arcnext; /* pointers to first and next argument */ |
103 |
|
104 |
struct pslot { |
105 |
int pid; /* process ID (0 if empty) */ |
106 |
int fout; /* output frame number */ |
107 |
int (*rcvf)(); /* recover function */ |
108 |
} *pslot; /* process slots */ |
109 |
int npslots; /* number of process slots */ |
110 |
|
111 |
#define phostname(ps) ((ps)->hostname[0] ? (ps)->hostname : astat.host) |
112 |
|
113 |
struct pslot *findpslot(); |
114 |
|
115 |
PSERVER *lastpserver; /* last process server with error */ |
116 |
|
117 |
VIEW *getview(); |
118 |
char *getexp(); |
119 |
|
120 |
extern time_t fdate(), time(); |
121 |
|
122 |
|
123 |
main(argc, argv) |
124 |
int argc; |
125 |
char *argv[]; |
126 |
{ |
127 |
int explicate = 0; |
128 |
int i; |
129 |
|
130 |
progname = argv[0]; /* get arguments */ |
131 |
for (i = 1; i < argc && argv[i][0] == '-'; i++) |
132 |
switch (argv[i][1]) { |
133 |
case 'e': /* print variables */ |
134 |
explicate++; |
135 |
break; |
136 |
case 'w': /* turn off warnings */ |
137 |
nowarn++; |
138 |
break; |
139 |
case 's': /* silent mode */ |
140 |
silent++; |
141 |
break; |
142 |
case 'n': /* take no action */ |
143 |
noaction++; |
144 |
break; |
145 |
default: |
146 |
goto userr; |
147 |
} |
148 |
if (i != argc-1) |
149 |
goto userr; |
150 |
cfname = argv[i]; |
151 |
/* load variables */ |
152 |
loadvars(cfname); |
153 |
/* check variables */ |
154 |
checkvalues(); |
155 |
/* did we get DIRECTORY? */ |
156 |
checkdir(); |
157 |
/* check status */ |
158 |
if (getastat() < 0) { |
159 |
fprintf(stderr, "%s: exiting\n", progname); |
160 |
quit(1); |
161 |
} |
162 |
/* pfilt always if options given */ |
163 |
pfiltalways = vdef(PFILT); |
164 |
/* load RIF if any */ |
165 |
if (vdef(RIF)) |
166 |
getradfile(vval(RIF)); |
167 |
/* set defaults */ |
168 |
setdefaults(); |
169 |
/* print variables */ |
170 |
if (explicate) |
171 |
printvars(stdout); |
172 |
/* set up process servers */ |
173 |
sethosts(); |
174 |
/* run animation */ |
175 |
animate(); |
176 |
/* all done */ |
177 |
if (vdef(NEXTANIM)) { |
178 |
argv[i] = vval(NEXTANIM); /* just change input file */ |
179 |
if (!silent) |
180 |
printargs(argc, argv, stdout); |
181 |
if ((argv[0] = getpath(progname,getenv("PATH"),X_OK)) == NULL) |
182 |
fprintf(stderr, "%s: command not found\n", progname); |
183 |
else |
184 |
execv(progname, argv); |
185 |
quit(1); |
186 |
} |
187 |
quit(0); |
188 |
userr: |
189 |
fprintf(stderr, "Usage: %s [-s][-n][-w][-e] anim_file\n", progname); |
190 |
quit(1); |
191 |
} |
192 |
|
193 |
|
194 |
getastat() /* check/set animation status */ |
195 |
{ |
196 |
char sfname[256]; |
197 |
FILE *fp; |
198 |
|
199 |
sprintf(sfname, "%s/%s", vval(DIRECTORY), SFNAME); |
200 |
if ((fp = fopen(sfname, "r")) == NULL) { |
201 |
if (errno != ENOENT) { |
202 |
perror(sfname); |
203 |
return(-1); |
204 |
} |
205 |
astat.rnext = astat.fnext = astat.tnext = 0; |
206 |
goto setours; |
207 |
} |
208 |
if (fscanf(fp, "Control host: %s\n", astat.host) != 1) |
209 |
goto fmterr; |
210 |
if (fscanf(fp, "Control PID: %d\n", &astat.pid) != 1) |
211 |
goto fmterr; |
212 |
if (fscanf(fp, "Control file: %s\n", astat.cfname) != 1) |
213 |
goto fmterr; |
214 |
if (fscanf(fp, "Next render: %d\n", &astat.rnext) != 1) |
215 |
goto fmterr; |
216 |
if (fscanf(fp, "Next filter: %d\n", &astat.fnext) != 1) |
217 |
goto fmterr; |
218 |
if (fscanf(fp, "Next transfer: %d\n", &astat.tnext) != 1) |
219 |
goto fmterr; |
220 |
fclose(fp); |
221 |
if (astat.pid != 0) { /* thinks it's still running */ |
222 |
if (strcmp(myhostname(), astat.host)) { |
223 |
fprintf(stderr, |
224 |
"%s: process %d may still be running on host %s\n", |
225 |
progname, astat.pid, astat.host); |
226 |
return(-1); |
227 |
} |
228 |
if (kill(astat.pid, 0) != -1 || errno != ESRCH) { |
229 |
fprintf(stderr, "%s: process %d is still running\n", |
230 |
progname, astat.pid); |
231 |
return(-1); |
232 |
} |
233 |
/* assume it is dead */ |
234 |
} |
235 |
if (strcmp(cfname, astat.cfname) && astat.pid != 0) { /* other's */ |
236 |
fprintf(stderr, "%s: unfinished job \"%s\"\n", |
237 |
progname, astat.cfname); |
238 |
return(-1); |
239 |
} |
240 |
/* check control file mods. */ |
241 |
if (!nowarn && fdate(cfname) > fdate(sfname)) |
242 |
fprintf(stderr, |
243 |
"%s: warning - control file modified since last run\n", |
244 |
progname); |
245 |
setours: /* set our values */ |
246 |
strcpy(astat.host, myhostname()); |
247 |
astat.pid = getpid(); |
248 |
strcpy(astat.cfname, cfname); |
249 |
return(0); |
250 |
fmterr: |
251 |
fprintf(stderr, "%s: format error in status file \"%s\"\n", |
252 |
progname, sfname); |
253 |
fclose(fp); |
254 |
return(-1); |
255 |
} |
256 |
|
257 |
|
258 |
putastat() /* put out current status */ |
259 |
{ |
260 |
char buf[256]; |
261 |
FILE *fp; |
262 |
|
263 |
if (noaction) |
264 |
return; |
265 |
sprintf(buf, "%s/%s", vval(DIRECTORY), SFNAME); |
266 |
if ((fp = fopen(buf, "w")) == NULL) { |
267 |
perror(buf); |
268 |
quit(1); |
269 |
} |
270 |
fprintf(fp, "Control host: %s\n", astat.host); |
271 |
fprintf(fp, "Control PID: %d\n", astat.pid); |
272 |
fprintf(fp, "Control file: %s\n", astat.cfname); |
273 |
fprintf(fp, "Next render: %d\n", astat.rnext); |
274 |
fprintf(fp, "Next filter: %d\n", astat.fnext); |
275 |
fprintf(fp, "Next transfer: %d\n", astat.tnext); |
276 |
fclose(fp); |
277 |
} |
278 |
|
279 |
|
280 |
checkdir() /* make sure we have our directory */ |
281 |
{ |
282 |
struct stat stb; |
283 |
|
284 |
if (!vdef(DIRECTORY)) { |
285 |
fprintf(stderr, "%s: %s undefined\n", |
286 |
progname, vnam(DIRECTORY)); |
287 |
quit(1); |
288 |
} |
289 |
if (stat(vval(DIRECTORY), &stb) == -1) { |
290 |
if (errno == ENOENT && mkdir(vval(DIRECTORY), 0777) == 0) |
291 |
return; |
292 |
perror(vval(DIRECTORY)); |
293 |
quit(1); |
294 |
} |
295 |
if (!(stb.st_mode & S_IFDIR)) { |
296 |
fprintf(stderr, "%s: not a directory\n", vval(DIRECTORY)); |
297 |
quit(1); |
298 |
} |
299 |
} |
300 |
|
301 |
|
302 |
setdefaults() /* set default values */ |
303 |
{ |
304 |
extern char *atos(); |
305 |
int decades; |
306 |
char buf[256]; |
307 |
|
308 |
if (vdef(ANIMATE)) { |
309 |
vval(OCTREE) = NULL; |
310 |
vdef(OCTREE) = 0; |
311 |
} else if (!vdef(OCTREE)) { |
312 |
fprintf(stderr, "%s: either %s or %s must be defined\n", |
313 |
progname, vnam(OCTREE), vnam(ANIMATE)); |
314 |
quit(1); |
315 |
} |
316 |
if (!vdef(VIEWFILE)) { |
317 |
fprintf(stderr, "%s: %s undefined\n", progname, vnam(VIEWFILE)); |
318 |
quit(1); |
319 |
} |
320 |
if (!vdef(HOST)) { |
321 |
vval(HOST) = LHOSTNAME; |
322 |
vdef(HOST)++; |
323 |
} |
324 |
if (!vdef(START)) { |
325 |
vval(START) = "1"; |
326 |
vdef(START)++; |
327 |
} |
328 |
if (!vdef(END)) { |
329 |
sprintf(buf, "%d", countviews()+vint(START)-1); |
330 |
vval(END) = savqstr(buf); |
331 |
vdef(END)++; |
332 |
} |
333 |
if (vint(END) < vint(START)) { |
334 |
fprintf(stderr, "%s: ending frame less than starting frame\n", |
335 |
progname); |
336 |
quit(1); |
337 |
} |
338 |
if (!vdef(BASENAME)) { |
339 |
decades = (int)log10((double)vint(END)) + 1; |
340 |
if (decades < 3) decades = 3; |
341 |
sprintf(buf, "%s/frame%%0%dd", vval(DIRECTORY), decades); |
342 |
vval(BASENAME) = savqstr(buf); |
343 |
vdef(BASENAME)++; |
344 |
} |
345 |
if (!vdef(RESOLUTION)) { |
346 |
vval(RESOLUTION) = "640"; |
347 |
vdef(RESOLUTION)++; |
348 |
} |
349 |
if (!vdef(OVERSAMP)) { |
350 |
vval(OVERSAMP) = "2"; |
351 |
vdef(OVERSAMP)++; |
352 |
} |
353 |
if (!vdef(INTERP)) { |
354 |
vval(INTERP) = "0"; |
355 |
vdef(INTERP)++; |
356 |
} |
357 |
if (!vdef(MBLUR)) { |
358 |
vval(MBLUR) = "0"; |
359 |
vdef(MBLUR)++; |
360 |
} |
361 |
if (!vdef(RTRACE)) { |
362 |
vval(RTRACE) = "F"; |
363 |
vdef(RTRACE)++; |
364 |
} |
365 |
if (!vdef(DISKSPACE)) { |
366 |
if (!nowarn) |
367 |
fprintf(stderr, |
368 |
"%s: warning - no %s setting, assuming 100 Mbytes available\n", |
369 |
progname, vnam(DISKSPACE)); |
370 |
vval(DISKSPACE) = "100"; |
371 |
vdef(DISKSPACE)++; |
372 |
} |
373 |
if (!vdef(RSH)) { |
374 |
vval(RSH) = REMSH; |
375 |
vdef(RSH)++; |
376 |
} |
377 |
/* locate remote shell program */ |
378 |
atos(buf, sizeof(buf), vval(RSH)); |
379 |
if ((remsh = getpath(buf, getenv("PATH"), X_OK)) != NULL) |
380 |
remsh = savqstr(remsh); |
381 |
else |
382 |
remsh = vval(RSH); /* will generate error if used */ |
383 |
|
384 |
/* append rendering options */ |
385 |
if (vdef(RENDER)) |
386 |
sprintf(rendopt+strlen(rendopt), " %s", vval(RENDER)); |
387 |
} |
388 |
|
389 |
|
390 |
sethosts() /* set up process servers */ |
391 |
{ |
392 |
extern char *iskip(); |
393 |
char buf[256], *dir, *uname; |
394 |
int np; |
395 |
register char *cp; |
396 |
int i; |
397 |
|
398 |
npslots = 0; |
399 |
if (noaction) |
400 |
return; |
401 |
for (i = 0; i < vdef(HOST); i++) { /* add each host */ |
402 |
dir = uname = NULL; |
403 |
np = 1; |
404 |
strcpy(cp=buf, nvalue(HOST, i)); /* copy to buffer */ |
405 |
cp = sskip(cp); /* skip host name */ |
406 |
while (isspace(*cp)) |
407 |
*cp++ = '\0'; |
408 |
if (*cp) { /* has # processes? */ |
409 |
np = atoi(cp); |
410 |
if ((cp = iskip(cp)) == NULL || (*cp && !isspace(*cp))) |
411 |
badvalue(HOST); |
412 |
while (isspace(*cp)) |
413 |
cp++; |
414 |
if (*cp) { /* has directory? */ |
415 |
dir = cp; |
416 |
cp = sskip(cp); /* skip dir. */ |
417 |
while (isspace(*cp)) |
418 |
*cp++ = '\0'; |
419 |
if (*cp) { /* has user? */ |
420 |
uname = cp; |
421 |
if (*sskip(cp)) |
422 |
badvalue(HOST); |
423 |
} |
424 |
} |
425 |
} |
426 |
if (addpserver(buf, dir, uname, np) == NULL) { |
427 |
if (!nowarn) |
428 |
fprintf(stderr, |
429 |
"%s: cannot execute on host \"%s\"\n", |
430 |
progname, buf); |
431 |
} else |
432 |
npslots += np; |
433 |
} |
434 |
if (npslots == 0) { |
435 |
fprintf(stderr, "%s: no working process servers\n", progname); |
436 |
quit(1); |
437 |
} |
438 |
pslot = (struct pslot *)calloc(npslots, sizeof(struct pslot)); |
439 |
if (pslot == NULL) { |
440 |
perror("malloc"); |
441 |
quit(1); |
442 |
} |
443 |
} |
444 |
|
445 |
|
446 |
getradfile(rfargs) /* run rad and get needed variables */ |
447 |
char *rfargs; |
448 |
{ |
449 |
static short mvar[] = {OCTREE,PFILT,RESOLUTION,EXPOSURE,-1}; |
450 |
char combuf[256]; |
451 |
register int i; |
452 |
register char *cp; |
453 |
/* create rad command */ |
454 |
sprintf(rendopt, " @%s/render.opt", vval(DIRECTORY)); |
455 |
sprintf(combuf, |
456 |
"rad -v 0 -s -e -w %s OPTFILE=%s | egrep '^[ \t]*(NOMATCH", |
457 |
rfargs, rendopt+2); |
458 |
cp = combuf; |
459 |
while (*cp) cp++; /* match unset variables */ |
460 |
for (i = 0; mvar[i] >= 0; i++) |
461 |
if (!vdef(mvar[i])) { |
462 |
*cp++ = '|'; |
463 |
strcpy(cp, vnam(mvar[i])); |
464 |
while (*cp) cp++; |
465 |
} |
466 |
sprintf(cp, ")[ \t]*=' > %s/radset.var", vval(DIRECTORY)); |
467 |
cp += 11; /* point to file name */ |
468 |
if (system(combuf)) { |
469 |
fprintf(stderr, "%s: error executing rad command:\n\t%s\n", |
470 |
progname, combuf); |
471 |
quit(1); |
472 |
} |
473 |
loadvars(cp); /* load variables and remove file */ |
474 |
unlink(cp); |
475 |
} |
476 |
|
477 |
|
478 |
animate() /* run animation */ |
479 |
{ |
480 |
int xres, yres; |
481 |
float pa, mult; |
482 |
int frames_batch; |
483 |
register int i; |
484 |
double d1, d2; |
485 |
/* compute rpict resolution */ |
486 |
i = sscanf(vval(RESOLUTION), "%d %d %f", &xres, &yres, &pa); |
487 |
mult = vflt(OVERSAMP); |
488 |
if (i == 3) { |
489 |
sprintf(rresopt, "-x %d -y %d -pa %.3f", (int)(mult*xres), |
490 |
(int)(mult*yres), pa); |
491 |
sprintf(fresopt, "-x %d -y %d -pa %.3f", xres, yres, pa); |
492 |
} else if (i) { |
493 |
if (i == 1) yres = xres; |
494 |
sprintf(rresopt, "-x %d -y %d", (int)(mult*xres), |
495 |
(int)(mult*yres)); |
496 |
sprintf(fresopt, "-x %d -y %d -pa 1", xres, yres); |
497 |
} else |
498 |
badvalue(RESOLUTION); |
499 |
/* consistency checks */ |
500 |
if (vdef(ANIMATE)) { |
501 |
if (vint(INTERP)) { |
502 |
if (!nowarn) |
503 |
fprintf(stderr, |
504 |
"%s: resetting %s=0 for animation\n", |
505 |
progname, vnam(INTERP)); |
506 |
vval(INTERP) = "0"; |
507 |
} |
508 |
if (atoi(vval(MBLUR))) { /* can't handle this yet */ |
509 |
if (!nowarn) |
510 |
fprintf(stderr, |
511 |
"%s: resetting %s=0 for animation\n", |
512 |
progname, vnam(MBLUR)); |
513 |
vval(MBLUR) = "0"; |
514 |
} |
515 |
} |
516 |
/* figure # frames per batch */ |
517 |
d1 = mult*xres*mult*yres*4; /* space for orig. picture */ |
518 |
if ((i=vint(INTERP)) || atoi(vval(MBLUR))) |
519 |
d1 += mult*xres*mult*yres*sizeof(float); /* Z-buffer */ |
520 |
d2 = xres*yres*4; /* space for final picture */ |
521 |
frames_batch = (i+1)*(vflt(DISKSPACE)*1048576.-d1)/(d1+i*d2); |
522 |
if (frames_batch < i+2) { |
523 |
fprintf(stderr, "%s: insufficient disk space allocated\n", |
524 |
progname); |
525 |
quit(1); |
526 |
} |
527 |
/* initialize archive argument list */ |
528 |
i = 16; |
529 |
if (vdef(ARCHIVE) && strlen(vval(ARCHIVE)) > i) |
530 |
i = strlen(vval(ARCHIVE)); |
531 |
arcnext = arcfirst = arcargs + i; |
532 |
/* initialize status file */ |
533 |
if (astat.rnext == 0) |
534 |
astat.rnext = astat.fnext = astat.tnext = vint(START); |
535 |
putastat(); |
536 |
/* render in batches */ |
537 |
while (astat.tnext <= vint(END)) { |
538 |
renderframes(frames_batch); |
539 |
filterframes(); |
540 |
transferframes(); |
541 |
} |
542 |
/* mark status as finished */ |
543 |
astat.pid = 0; |
544 |
putastat(); |
545 |
/* close open files */ |
546 |
getview(0); |
547 |
getexp(0); |
548 |
} |
549 |
|
550 |
|
551 |
renderframes(nframes) /* render next nframes frames */ |
552 |
int nframes; |
553 |
{ |
554 |
static char vendbuf[16]; |
555 |
VIEW *vp; |
556 |
FILE *fp = NULL; |
557 |
char vfname[128]; |
558 |
int lastframe; |
559 |
register int i; |
560 |
|
561 |
if (astat.tnext < astat.rnext) /* other work to do first */ |
562 |
return; |
563 |
/* create batch view file */ |
564 |
if (!vdef(ANIMATE)) { |
565 |
sprintf(vfname, "%s/anim.vf", vval(DIRECTORY)); |
566 |
if ((fp = fopen(vfname, "w")) == NULL) { |
567 |
perror(vfname); |
568 |
quit(1); |
569 |
} |
570 |
} |
571 |
/* bound batch properly */ |
572 |
lastframe = astat.rnext + nframes - 1; |
573 |
if ((lastframe-1) % (vint(INTERP)+1)) /* need even interval */ |
574 |
lastframe += vint(INTERP)+1 - ((lastframe-1)%(vint(INTERP)+1)); |
575 |
if (lastframe > vint(END)) /* check for end */ |
576 |
lastframe = vint(END); |
577 |
/* render each view */ |
578 |
for (i = astat.rnext; i <= lastframe; i++) { |
579 |
if ((vp = getview(i)) == NULL) { |
580 |
if (!nowarn) |
581 |
fprintf(stderr, |
582 |
"%s: ran out of views before last frame\n", |
583 |
progname); |
584 |
sprintf(vval(END)=vendbuf, "%d", i-1); |
585 |
lastframe = i - 1; |
586 |
break; |
587 |
} |
588 |
if (vdef(ANIMATE)) /* animate frame */ |
589 |
animrend(i, vp); |
590 |
else { /* else record it */ |
591 |
fputs(VIEWSTR, fp); |
592 |
fprintview(vp, fp); |
593 |
putc('\n', fp); |
594 |
} |
595 |
} |
596 |
if (vdef(ANIMATE)) /* wait for renderings to finish */ |
597 |
bwait(0); |
598 |
else { /* else if walk-through */ |
599 |
fclose(fp); /* close view file */ |
600 |
walkwait(astat.rnext, lastframe, vfname); /* walk it */ |
601 |
unlink(vfname); /* remove view file */ |
602 |
} |
603 |
astat.rnext = i; /* update status */ |
604 |
putastat(); |
605 |
} |
606 |
|
607 |
|
608 |
filterframes() /* catch up with filtering */ |
609 |
{ |
610 |
VIEW *vp; |
611 |
register int i; |
612 |
|
613 |
if (astat.tnext < astat.fnext) /* other work to do first */ |
614 |
return; |
615 |
/* filter each view */ |
616 |
for (i = astat.fnext; i < astat.rnext; i++) { |
617 |
if ((vp = getview(i)) == NULL) { /* get view i */ |
618 |
fprintf(stderr, |
619 |
"%s: unexpected error reading view for frame %d\n", |
620 |
progname, i); |
621 |
quit(1); |
622 |
} |
623 |
dofilt(i, vp, getexp(i), 0); /* filter frame */ |
624 |
} |
625 |
bwait(0); /* wait for filter processes */ |
626 |
archive(); /* archive originals */ |
627 |
astat.fnext = i; /* update status */ |
628 |
putastat(); |
629 |
} |
630 |
|
631 |
|
632 |
transferframes() /* catch up with picture transfers */ |
633 |
{ |
634 |
char combuf[10240]; |
635 |
register char *cp; |
636 |
register int i; |
637 |
|
638 |
if (astat.tnext >= astat.fnext) /* nothing to do, yet */ |
639 |
return; |
640 |
if (!vdef(TRANSFER)) { /* no transfer function -- leave 'em */ |
641 |
astat.tnext = astat.fnext; |
642 |
putastat(); /* update status */ |
643 |
return; |
644 |
} |
645 |
strcpy(combuf, vval(TRANSFER)); /* start transfer command */ |
646 |
cp = combuf + strlen(combuf); |
647 |
/* make argument list */ |
648 |
for (i = astat.tnext; i < astat.fnext; i++) { |
649 |
*cp++ = ' '; |
650 |
sprintf(cp, vval(BASENAME), i); |
651 |
while (*cp) cp++; |
652 |
strcpy(cp, ".pic"); |
653 |
cp += 4; |
654 |
} |
655 |
if (runcom(combuf)) { /* transfer frames */ |
656 |
fprintf(stderr, "%s: error running transfer command\n", |
657 |
progname); |
658 |
quit(1); |
659 |
} |
660 |
astat.tnext = i; /* update status */ |
661 |
putastat(); |
662 |
} |
663 |
|
664 |
|
665 |
animrend(frame, vp) /* start animation frame */ |
666 |
int frame; |
667 |
VIEW *vp; |
668 |
{ |
669 |
extern int recover(); |
670 |
char combuf[2048]; |
671 |
char fname[128]; |
672 |
|
673 |
sprintf(fname, vval(BASENAME), frame); |
674 |
strcat(fname, ".unf"); |
675 |
if (access(fname, F_OK) == 0) |
676 |
return; |
677 |
sprintf(combuf, "%s %d | rpict%s%s -w0 %s > %s", vval(ANIMATE), frame, |
678 |
rendopt, viewopt(vp), rresopt, fname); |
679 |
bruncom(combuf, frame, recover); /* run in background */ |
680 |
} |
681 |
|
682 |
|
683 |
walkwait(first, last, vfn) /* walk-through frames */ |
684 |
int first, last; |
685 |
char *vfn; |
686 |
{ |
687 |
char combuf[2048]; |
688 |
char *inspoint; |
689 |
register int i; |
690 |
|
691 |
if (!noaction && vint(INTERP)) /* create dummy frames */ |
692 |
for (i = first; i <= last; i++) |
693 |
if (i < vint(END) && (i-1) % (vint(INTERP)+1)) { |
694 |
sprintf(combuf, vval(BASENAME), i); |
695 |
strcat(combuf, ".unf"); |
696 |
close(open(combuf, O_RDONLY|O_CREAT, 0666)); |
697 |
} |
698 |
/* create command */ |
699 |
sprintf(combuf, "rpict%s -w0", rendopt); |
700 |
if (vint(INTERP) || atoi(vval(MBLUR))) |
701 |
sprintf(combuf+strlen(combuf), " -z %s.zbf", vval(BASENAME)); |
702 |
sprintf(combuf+strlen(combuf), " -o %s.unf %s -S %d", |
703 |
vval(BASENAME), rresopt, first); |
704 |
inspoint = combuf + strlen(combuf); |
705 |
sprintf(inspoint, " %s < %s", vval(OCTREE), vfn); |
706 |
/* run in parallel */ |
707 |
i = (last-first+1)/(vint(INTERP)+1); |
708 |
if (i < 1) i = 1; |
709 |
if (pruncom(combuf, inspoint, i)) { |
710 |
fprintf(stderr, "%s: error rendering frames %d through %d\n", |
711 |
progname, first, last); |
712 |
quit(1); |
713 |
} |
714 |
if (!noaction && vint(INTERP)) /* remove dummy frames */ |
715 |
for (i = first; i <= last; i++) |
716 |
if (i < vint(END) && (i-1) % (vint(INTERP)+1)) { |
717 |
sprintf(combuf, vval(BASENAME), i); |
718 |
strcat(combuf, ".unf"); |
719 |
unlink(combuf); |
720 |
} |
721 |
} |
722 |
|
723 |
|
724 |
int |
725 |
recover(frame) /* recover the specified frame */ |
726 |
int frame; |
727 |
{ |
728 |
static int *rfrm; /* list of recovered frames */ |
729 |
static int nrfrms = 0; |
730 |
char combuf[2048]; |
731 |
char fname[128]; |
732 |
register char *cp; |
733 |
register int i; |
734 |
/* check to see if recovered already */ |
735 |
for (i = nrfrms; i--; ) |
736 |
if (rfrm[i] == frame) |
737 |
return(0); |
738 |
/* build command */ |
739 |
sprintf(fname, vval(BASENAME), frame); |
740 |
if (vdef(ANIMATE)) |
741 |
sprintf(combuf, "%s %d | rpict%s -w0", |
742 |
vval(ANIMATE), frame, rendopt); |
743 |
else |
744 |
sprintf(combuf, "rpict%s -w0", rendopt); |
745 |
cp = combuf + strlen(combuf); |
746 |
if (vint(INTERP) || atoi(vval(MBLUR))) { |
747 |
sprintf(cp, " -z %s.zbf", fname); |
748 |
while (*cp) cp++; |
749 |
} |
750 |
sprintf(cp, " -ro %s.unf", fname); |
751 |
while (*cp) cp++; |
752 |
if (!vdef(ANIMATE)) { |
753 |
*cp++ = ' '; |
754 |
strcpy(cp, vval(OCTREE)); |
755 |
} |
756 |
if (runcom(combuf)) /* run command */ |
757 |
return(1); |
758 |
/* add frame to recovered list */ |
759 |
if (nrfrms) |
760 |
rfrm = (int *)realloc((char *)rfrm, (nrfrms+1)*sizeof(int)); |
761 |
else |
762 |
rfrm = (int *)malloc(sizeof(int)); |
763 |
if (rfrm == NULL) { |
764 |
perror("malloc"); |
765 |
quit(1); |
766 |
} |
767 |
rfrm[nrfrms++] = frame; |
768 |
return(0); |
769 |
} |
770 |
|
771 |
|
772 |
int |
773 |
frecover(frame) /* recover filtered frame */ |
774 |
int frame; |
775 |
{ |
776 |
VIEW *vp; |
777 |
char *ex; |
778 |
|
779 |
vp = getview(frame); |
780 |
ex = getexp(frame); |
781 |
if (dofilt(frame, vp, ex, 2) && dofilt(frame, vp, ex, 1)) |
782 |
return(1); |
783 |
return(0); |
784 |
} |
785 |
|
786 |
|
787 |
archive() /* archive and remove renderings */ |
788 |
{ |
789 |
#define RMCOML (sizeof(rmcom)-1) |
790 |
static char rmcom[] = "rm -f"; |
791 |
register int i; |
792 |
|
793 |
if (arcnext == arcfirst) |
794 |
return; /* nothing to do */ |
795 |
if (vdef(ARCHIVE)) { /* run archive command */ |
796 |
i = strlen(vval(ARCHIVE)); |
797 |
strncpy(arcfirst-i, vval(ARCHIVE), i); |
798 |
if (runcom(arcfirst-i)) { |
799 |
fprintf(stderr, "%s: error running archive command\n", |
800 |
progname); |
801 |
quit(1); |
802 |
} |
803 |
} |
804 |
/* run remove command */ |
805 |
strncpy(arcfirst-RMCOML, rmcom, RMCOML); |
806 |
runcom(arcfirst-RMCOML); |
807 |
arcnext = arcfirst; /* reset argument list */ |
808 |
#undef RMCOML |
809 |
} |
810 |
|
811 |
|
812 |
int |
813 |
dofilt(frame, vp, ep, rvr) /* filter frame */ |
814 |
int frame; |
815 |
VIEW *vp; |
816 |
char *ep; |
817 |
int rvr; |
818 |
{ |
819 |
extern int frecover(); |
820 |
static int iter = 0; |
821 |
char fnbefore[128], fnafter[128]; |
822 |
char combuf[1024], fname0[128], fname1[128]; |
823 |
int usepinterp, usepfilt, nora_rgbe; |
824 |
int frseq[2]; |
825 |
/* check what is needed */ |
826 |
usepinterp = atoi(vval(MBLUR)); |
827 |
usepfilt = pfiltalways | ep==NULL; |
828 |
if (ep != NULL && !strcmp(ep, "1")) |
829 |
ep = "+0"; |
830 |
nora_rgbe = strcmp(vval(OVERSAMP),"1") || ep==NULL || |
831 |
*ep != '+' || *ep != '-' || !isint(ep); |
832 |
/* compute rendered views */ |
833 |
frseq[0] = frame - ((frame-1) % (vint(INTERP)+1)); |
834 |
frseq[1] = frseq[0] + vint(INTERP) + 1; |
835 |
if (frseq[1] > vint(END)) |
836 |
frseq[1] = vint(END); |
837 |
if (frseq[1] == frame) { /* pfilt only */ |
838 |
frseq[0] = frseq[1]; |
839 |
usepinterp = 0; /* update what's needed */ |
840 |
usepfilt |= nora_rgbe; |
841 |
} else if (frseq[0] == frame) { /* no interpolation needed */ |
842 |
if (!rvr && frame > 1+vint(INTERP)) { /* archive previous */ |
843 |
*arcnext++ = ' '; |
844 |
sprintf(arcnext, vval(BASENAME), frame-vint(INTERP)-1); |
845 |
while (*arcnext) arcnext++; |
846 |
strcpy(arcnext, ".unf"); |
847 |
arcnext += 4; |
848 |
if (usepinterp || vint(INTERP)) { /* and Z-buf */ |
849 |
*arcnext++ = ' '; |
850 |
sprintf(arcnext, vval(BASENAME), |
851 |
frame-vint(INTERP)-1); |
852 |
while (*arcnext) arcnext++; |
853 |
strcpy(arcnext, ".zbf"); |
854 |
arcnext += 4; |
855 |
} |
856 |
} |
857 |
if (!usepinterp) /* update what's needed */ |
858 |
usepfilt |= nora_rgbe; |
859 |
} else /* interpolation needed */ |
860 |
usepinterp++; |
861 |
if (frseq[1] >= astat.rnext) /* next batch unavailable */ |
862 |
frseq[1] = frseq[0]; |
863 |
sprintf(fnbefore, vval(BASENAME), frseq[0]); |
864 |
sprintf(fnafter, vval(BASENAME), frseq[1]); |
865 |
if (rvr == 1 && recover(frseq[0])) /* recover before frame? */ |
866 |
return(1); |
867 |
/* generate command */ |
868 |
if (usepinterp) { /* using pinterp */ |
869 |
if (rvr == 2 && recover(frseq[1])) /* recover after? */ |
870 |
return(1); |
871 |
if (atoi(vval(MBLUR))) { |
872 |
FILE *fp; /* motion blurring */ |
873 |
sprintf(fname0, "%s/vw0%c", vval(DIRECTORY), |
874 |
'a'+(iter%26)); |
875 |
if ((fp = fopen(fname0, "w")) == NULL) { |
876 |
perror(fname0); quit(1); |
877 |
} |
878 |
fputs(VIEWSTR, fp); |
879 |
fprintview(vp, fp); |
880 |
putc('\n', fp); fclose(fp); |
881 |
if ((vp = getview(frame+1)) == NULL) { |
882 |
fprintf(stderr, |
883 |
"%s: unexpected error reading view for frame %d\n", |
884 |
progname, frame+1); |
885 |
quit(1); |
886 |
} |
887 |
sprintf(fname1, "%s/vw1%c", vval(DIRECTORY), |
888 |
'a'+(iter%26)); |
889 |
if ((fp = fopen(fname1, "w")) == NULL) { |
890 |
perror(fname1); quit(1); |
891 |
} |
892 |
fputs(VIEWSTR, fp); |
893 |
fprintview(vp, fp); |
894 |
putc('\n', fp); fclose(fp); |
895 |
sprintf(combuf, |
896 |
"(pmblur %s %d %s %s; rm -f %s %s) | pinterp -B", |
897 |
*sskip(vval(MBLUR)) ? sskip2(vval(MBLUR),1) : "1", |
898 |
atoi(vval(MBLUR)), |
899 |
fname0, fname1, fname0, fname1); |
900 |
iter++; |
901 |
} else /* no blurring */ |
902 |
strcpy(combuf, "pinterp"); |
903 |
strcat(combuf, viewopt(vp)); |
904 |
if (vbool(RTRACE)) |
905 |
sprintf(combuf+strlen(combuf), " -ff -fr '%s -w0 %s'", |
906 |
rendopt+1, vval(OCTREE)); |
907 |
if (vdef(PINTERP)) |
908 |
sprintf(combuf+strlen(combuf), " %s", vval(PINTERP)); |
909 |
if (usepfilt) |
910 |
sprintf(combuf+strlen(combuf), " %s", rresopt); |
911 |
else |
912 |
sprintf(combuf+strlen(combuf), " %s -e %s", |
913 |
fresopt, ep); |
914 |
sprintf(combuf+strlen(combuf), " %s.unf %s.zbf", |
915 |
fnbefore, fnbefore); |
916 |
if (frseq[1] != frseq[0]) |
917 |
sprintf(combuf+strlen(combuf), " %s.unf %s.zbf", |
918 |
fnafter, fnafter); |
919 |
if (usepfilt) { /* also pfilt */ |
920 |
if (vdef(PFILT)) |
921 |
sprintf(combuf+strlen(combuf), " | pfilt %s", |
922 |
vval(PFILT)); |
923 |
else |
924 |
strcat(combuf, " | pfilt"); |
925 |
if (ep != NULL) |
926 |
sprintf(combuf+strlen(combuf), " -1 -e %s %s", |
927 |
ep, fresopt); |
928 |
else |
929 |
sprintf(combuf+strlen(combuf), " %s", fresopt); |
930 |
} |
931 |
} else if (usepfilt) { /* pfilt only */ |
932 |
if (rvr == 2) |
933 |
return(1); |
934 |
if (vdef(PFILT)) |
935 |
sprintf(combuf, "pfilt %s", vval(PFILT)); |
936 |
else |
937 |
strcpy(combuf, "pfilt"); |
938 |
if (ep != NULL) |
939 |
sprintf(combuf+strlen(combuf), " -1 -e %s %s %s.unf", |
940 |
ep, fresopt, fnbefore); |
941 |
else |
942 |
sprintf(combuf+strlen(combuf), " %s %s.unf", |
943 |
fresopt, fnbefore); |
944 |
} else { /* else just check it */ |
945 |
if (rvr == 2) |
946 |
return(1); |
947 |
sprintf(combuf, "ra_rgbe -e %s -r %s.unf", ep, fnbefore); |
948 |
} |
949 |
/* output file name */ |
950 |
sprintf(fname0, vval(BASENAME), frame); |
951 |
sprintf(combuf+strlen(combuf), " > %s.pic", fname0); |
952 |
if (rvr) /* in recovery */ |
953 |
return(runcom(combuf)); |
954 |
bruncom(combuf, frame, frecover); /* else run in background */ |
955 |
return(0); |
956 |
} |
957 |
|
958 |
|
959 |
VIEW * |
960 |
getview(n) /* get view number n */ |
961 |
int n; |
962 |
{ |
963 |
static FILE *viewfp = NULL; /* view file pointer */ |
964 |
static int viewnum = 0; /* current view number */ |
965 |
static VIEW curview = STDVIEW; /* current view */ |
966 |
char linebuf[256]; |
967 |
|
968 |
if (n == 0) { /* signal to close file and clean up */ |
969 |
if (viewfp != NULL) { |
970 |
fclose(viewfp); |
971 |
viewfp = NULL; |
972 |
viewnum = 0; |
973 |
copystruct(&curview, &stdview); |
974 |
} |
975 |
return(NULL); |
976 |
} |
977 |
if (viewfp == NULL) { /* open file */ |
978 |
if ((viewfp = fopen(vval(VIEWFILE), "r")) == NULL) { |
979 |
perror(vval(VIEWFILE)); |
980 |
quit(1); |
981 |
} |
982 |
} else if (n < viewnum) { /* rewind file */ |
983 |
if (viewnum == 1 && feof(viewfp)) |
984 |
return(&curview); /* just one view */ |
985 |
if (fseek(viewfp, 0L, 0) == EOF) { |
986 |
perror(vval(VIEWFILE)); |
987 |
quit(1); |
988 |
} |
989 |
copystruct(&curview, &stdview); |
990 |
viewnum = 0; |
991 |
} |
992 |
while (n > viewnum) { /* scan to desired view */ |
993 |
if (fgets(linebuf, sizeof(linebuf), viewfp) == NULL) |
994 |
return(viewnum==1 ? &curview : (VIEW *)NULL); |
995 |
if (isview(linebuf) && sscanview(&curview, linebuf) > 0) |
996 |
viewnum++; |
997 |
} |
998 |
return(&curview); /* return it */ |
999 |
} |
1000 |
|
1001 |
|
1002 |
int |
1003 |
countviews() /* count views in view file */ |
1004 |
{ |
1005 |
register int n = 0; |
1006 |
|
1007 |
while (getview(n+1) != NULL) |
1008 |
n++; |
1009 |
return(n); |
1010 |
} |
1011 |
|
1012 |
|
1013 |
char * |
1014 |
getexp(n) /* get exposure for nth frame */ |
1015 |
int n; |
1016 |
{ |
1017 |
extern char *fskip(); |
1018 |
static char expval[32]; |
1019 |
static FILE *expfp = NULL; |
1020 |
static long *exppos; |
1021 |
static int curfrm; |
1022 |
register char *cp; |
1023 |
|
1024 |
if (n == 0) { /* signal to close file */ |
1025 |
if (expfp != NULL) { |
1026 |
fclose(expfp); |
1027 |
free((char *)exppos); |
1028 |
expfp = NULL; |
1029 |
} |
1030 |
return(NULL); |
1031 |
} else if (n > vint(END)) /* request past end (error?) */ |
1032 |
return(NULL); |
1033 |
if (!vdef(EXPOSURE)) /* no setting (auto) */ |
1034 |
return(NULL); |
1035 |
if (isflt(vval(EXPOSURE))) /* always the same */ |
1036 |
return(vval(EXPOSURE)); |
1037 |
if (expfp == NULL) { /* open exposure file */ |
1038 |
if ((expfp = fopen(vval(EXPOSURE), "r")) == NULL) { |
1039 |
fprintf(stderr, |
1040 |
"%s: cannot open exposure file \"%s\"\n", |
1041 |
progname, vval(EXPOSURE)); |
1042 |
quit(1); |
1043 |
} |
1044 |
curfrm = vint(END) + 1; /* init lookup tab. */ |
1045 |
exppos = (long *)malloc(curfrm*sizeof(long *)); |
1046 |
if (exppos == NULL) { |
1047 |
perror(progname); |
1048 |
quit(1); |
1049 |
} |
1050 |
while (curfrm--) |
1051 |
exppos[curfrm] = -1L; |
1052 |
curfrm = 0; |
1053 |
} |
1054 |
/* find position in file */ |
1055 |
if (n-1 != curfrm && n != curfrm && exppos[n-1] >= 0 && |
1056 |
fseek(expfp, exppos[curfrm=n-1], 0) == EOF) { |
1057 |
fprintf(stderr, "%s: seek error on exposure file\n", progname); |
1058 |
quit(1); |
1059 |
} |
1060 |
while (n > curfrm) { /* read exposure */ |
1061 |
if (exppos[curfrm] < 0) |
1062 |
exppos[curfrm] = ftell(expfp); |
1063 |
if (fgets(expval, sizeof(expval), expfp) == NULL) { |
1064 |
fprintf(stderr, "%s: too few exposures\n", |
1065 |
vval(EXPOSURE)); |
1066 |
quit(1); |
1067 |
} |
1068 |
curfrm++; |
1069 |
cp = fskip(expval); /* check format */ |
1070 |
if (cp != NULL) |
1071 |
while (isspace(*cp)) |
1072 |
*cp++ = '\0'; |
1073 |
if (cp == NULL || *cp) { |
1074 |
fprintf(stderr, |
1075 |
"%s: exposure format error on line %d\n", |
1076 |
vval(EXPOSURE), curfrm); |
1077 |
quit(1); |
1078 |
} |
1079 |
} |
1080 |
return(expval); /* return value */ |
1081 |
} |
1082 |
|
1083 |
|
1084 |
struct pslot * |
1085 |
findpslot(pid) /* find or allocate a process slot */ |
1086 |
int pid; |
1087 |
{ |
1088 |
register struct pslot *psempty = NULL; |
1089 |
register int i; |
1090 |
|
1091 |
for (i = 0; i < npslots; i++) { /* look for match */ |
1092 |
if (pslot[i].pid == pid) |
1093 |
return(pslot+i); |
1094 |
if (psempty == NULL && pslot[i].pid == 0) |
1095 |
psempty = pslot+i; |
1096 |
} |
1097 |
return(psempty); /* return emtpy slot (error if NULL) */ |
1098 |
} |
1099 |
|
1100 |
|
1101 |
int |
1102 |
donecom(ps, pn, status) /* clean up after finished process */ |
1103 |
PSERVER *ps; |
1104 |
int pn; |
1105 |
int status; |
1106 |
{ |
1107 |
register PROC *pp; |
1108 |
register struct pslot *psl; |
1109 |
|
1110 |
pp = ps->proc + pn; |
1111 |
if (pp->elen) { /* pass errors */ |
1112 |
if (ps->hostname[0]) |
1113 |
fprintf(stderr, "%s: ", ps->hostname); |
1114 |
fprintf(stderr, "Error output from: %s\n", pp->com); |
1115 |
fputs(pp->errs, stderr); |
1116 |
fflush(stderr); |
1117 |
if (ps->hostname[0]) |
1118 |
status = 1; /* because rsh doesn't return status */ |
1119 |
} |
1120 |
lastpserver = NULL; |
1121 |
psl = findpslot(pp->pid); /* check for bruncom() slot */ |
1122 |
if (psl->pid) { |
1123 |
if (status) { |
1124 |
if (psl->rcvf != NULL) /* attempt recovery */ |
1125 |
status = (*psl->rcvf)(psl->fout); |
1126 |
if (status) { |
1127 |
fprintf(stderr, |
1128 |
"%s: error rendering frame %d\n", |
1129 |
progname, psl->fout); |
1130 |
quit(1); |
1131 |
} |
1132 |
lastpserver = ps; |
1133 |
} |
1134 |
psl->pid = 0; /* free process slot */ |
1135 |
} else if (status) |
1136 |
lastpserver = ps; |
1137 |
freestr(pp->com); /* free command string */ |
1138 |
return(status); |
1139 |
} |
1140 |
|
1141 |
|
1142 |
int |
1143 |
serverdown() /* check status of last process server */ |
1144 |
{ |
1145 |
if (lastpserver == NULL || !lastpserver->hostname[0]) |
1146 |
return(0); |
1147 |
if (pserverOK(lastpserver)) /* server still up? */ |
1148 |
return(0); |
1149 |
delpserver(lastpserver); /* else delete it */ |
1150 |
if (pslist == NULL) { |
1151 |
fprintf(stderr, "%s: all process servers are down\n", |
1152 |
progname); |
1153 |
quit(1); |
1154 |
} |
1155 |
return(1); |
1156 |
} |
1157 |
|
1158 |
|
1159 |
int |
1160 |
bruncom(com, fout, rf) /* run a command in the background */ |
1161 |
char *com; |
1162 |
int fout; |
1163 |
int (*rf)(); |
1164 |
{ |
1165 |
int pid; |
1166 |
register struct pslot *psl; |
1167 |
|
1168 |
if (noaction) { |
1169 |
if (!silent) |
1170 |
printf("\t%s\n", com); /* echo command */ |
1171 |
return(0); |
1172 |
} |
1173 |
com = savestr(com); /* else start it when we can */ |
1174 |
while ((pid = startjob(NULL, com, donecom)) == -1) |
1175 |
bwait(1); |
1176 |
if (!silent) { /* echo command */ |
1177 |
PSERVER *ps; |
1178 |
int psn = pid; |
1179 |
ps = findjob(&psn); |
1180 |
printf("\t%s\n", com); |
1181 |
printf("\tProcess started on %s\n", phostname(ps)); |
1182 |
fflush(stdout); |
1183 |
} |
1184 |
psl = findpslot(pid); /* record info. in appropriate slot */ |
1185 |
psl->pid = pid; |
1186 |
psl->fout = fout; |
1187 |
psl->rcvf = rf; |
1188 |
return(pid); |
1189 |
} |
1190 |
|
1191 |
|
1192 |
bwait(ncoms) /* wait for batch job(s) to finish */ |
1193 |
int ncoms; |
1194 |
{ |
1195 |
int status; |
1196 |
|
1197 |
if (noaction) |
1198 |
return; |
1199 |
while ((status = wait4job(NULL, -1)) != -1) { |
1200 |
serverdown(); /* update server status */ |
1201 |
if (--ncoms == 0) |
1202 |
break; /* done enough */ |
1203 |
} |
1204 |
} |
1205 |
|
1206 |
|
1207 |
int |
1208 |
pruncom(com, ppins, maxcopies) /* run a command in parallel over network */ |
1209 |
char *com, *ppins; |
1210 |
int maxcopies; |
1211 |
{ |
1212 |
int retstatus = 0; |
1213 |
int hostcopies; |
1214 |
char buf[10240], *com1, *s; |
1215 |
int status; |
1216 |
int pfd; |
1217 |
register int n; |
1218 |
register PSERVER *ps; |
1219 |
|
1220 |
if (!silent) |
1221 |
printf("\t%s\n", com); /* echo command */ |
1222 |
if (noaction) |
1223 |
return(0); |
1224 |
fflush(stdout); |
1225 |
/* start jobs on each server */ |
1226 |
for (ps = pslist; ps != NULL; ps = ps->next) { |
1227 |
hostcopies = 0; |
1228 |
if (maxcopies > 1 && ps->nprocs > 1 && ppins != NULL) { |
1229 |
strcpy(com1=buf, com); /* build -PP command */ |
1230 |
sprintf(com1+(ppins-com), " -PP %s/%s.persist", |
1231 |
vval(DIRECTORY), phostname(ps)); |
1232 |
strcat(com1, ppins); |
1233 |
} else |
1234 |
com1 = com; |
1235 |
while (maxcopies > 0) { |
1236 |
s = savestr(com1); |
1237 |
if (startjob(ps, s, donecom) != -1) { |
1238 |
sleep(20); |
1239 |
hostcopies++; |
1240 |
maxcopies--; |
1241 |
} else { |
1242 |
freestr(s); |
1243 |
break; |
1244 |
} |
1245 |
} |
1246 |
if (!silent && hostcopies) { |
1247 |
if (hostcopies > 1) |
1248 |
printf("\t%d duplicate processes", hostcopies); |
1249 |
else |
1250 |
printf("\tProcess"); |
1251 |
printf(" started on %s\n", phostname(ps)); |
1252 |
fflush(stdout); |
1253 |
} |
1254 |
} |
1255 |
/* wait for jobs to finish */ |
1256 |
while ((status = wait4job(NULL, -1)) != -1) |
1257 |
retstatus += status && !serverdown(); |
1258 |
/* terminate parallel rpict's */ |
1259 |
for (ps = pslist; ps != NULL; ps = ps->next) { |
1260 |
sprintf(buf, "%s/%s.persist", vval(DIRECTORY), phostname(ps)); |
1261 |
if ((pfd = open(buf, O_RDONLY)) >= 0) { |
1262 |
n = read(pfd, buf, sizeof(buf)-1); /* get PID */ |
1263 |
buf[n] = '\0'; |
1264 |
close(pfd); |
1265 |
for (n = 0; buf[n] && !isspace(buf[n]); n++) |
1266 |
; |
1267 |
/* terminate */ |
1268 |
sprintf(buf, "kill -ALRM %d", atoi(buf+n)); |
1269 |
wait4job(ps, startjob(ps, buf, NULL)); |
1270 |
} |
1271 |
} |
1272 |
return(retstatus); |
1273 |
} |
1274 |
|
1275 |
|
1276 |
runcom(cs) /* run a command locally and wait for it */ |
1277 |
char *cs; |
1278 |
{ |
1279 |
if (!silent) /* echo it */ |
1280 |
printf("\t%s\n", cs); |
1281 |
if (noaction) |
1282 |
return(0); |
1283 |
fflush(stdout); /* flush output and pass to shell */ |
1284 |
return(system(cs)); |
1285 |
} |
1286 |
|
1287 |
|
1288 |
rmfile(fn) /* remove a file */ |
1289 |
char *fn; |
1290 |
{ |
1291 |
if (!silent) |
1292 |
#ifdef MSDOS |
1293 |
printf("\tdel %s\n", fn); |
1294 |
#else |
1295 |
printf("\trm -f %s\n", fn); |
1296 |
#endif |
1297 |
if (noaction) |
1298 |
return(0); |
1299 |
return(unlink(fn)); |
1300 |
} |
1301 |
|
1302 |
|
1303 |
badvalue(vc) /* report bad variable value and exit */ |
1304 |
int vc; |
1305 |
{ |
1306 |
fprintf(stderr, "%s: bad value for variable '%s'\n", |
1307 |
progname, vnam(vc)); |
1308 |
quit(1); |
1309 |
} |