| 16 |
|
#include "lookup.h" |
| 17 |
|
#include "calcomp.h" |
| 18 |
|
|
| 19 |
+ |
#ifndef MAXMODLIST |
| 20 |
|
#define MAXMODLIST 1024 /* maximum modifiers we'll track */ |
| 21 |
+ |
#endif |
| 22 |
|
|
| 23 |
|
int treebufsiz = BUFSIZ; /* current tree buffer size */ |
| 24 |
|
|
| 74 |
|
}; /* rtrace process buffer */ |
| 75 |
|
|
| 76 |
|
/* rtrace command and defaults */ |
| 77 |
< |
char *rtargv[256] = { "rtrace", "-dj", ".5", "-dr", "3", |
| 77 |
> |
char *rtargv[256+2*MAXMODLIST] = { "rtrace", |
| 78 |
> |
"-dj", ".5", "-dr", "3", |
| 79 |
|
"-ab", "1", "-ad", "128", }; |
| 80 |
|
int rtargc = 9; |
| 81 |
|
/* overriding rtrace options */ |
| 113 |
|
int nmods = 0; /* number of modifiers */ |
| 114 |
|
|
| 115 |
|
MODCONT *addmodifier(char *modn, char *outf, char *binv); |
| 116 |
+ |
void addmodfile(char *fname, char *outf, char *binv); |
| 117 |
|
|
| 118 |
|
void init(int np); |
| 119 |
|
int done_rprocs(struct rtproc *rtp); |
| 173 |
|
char *binval = NULL; |
| 174 |
|
char fmt[8]; |
| 175 |
|
int i, j; |
| 176 |
+ |
/* need at least one argument */ |
| 177 |
+ |
if (argc < 2) { |
| 178 |
+ |
fprintf(stderr, |
| 179 |
+ |
"Usage: %s [-n nprocs][-r][-e expr][-f source][-o ospec][-b binv] {-m mod | -M file} [rtrace options] octree\n", |
| 180 |
+ |
argv[0]); |
| 181 |
+ |
exit(1); |
| 182 |
+ |
} |
| 183 |
|
/* global program name */ |
| 184 |
|
gargv = argv; |
| 185 |
|
/* initialize calcomp routines */ |
| 203 |
|
recover++; |
| 204 |
|
continue; |
| 205 |
|
case 'n': /* number of processes */ |
| 206 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 206 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 207 |
|
nprocs = atoi(argv[++i]); |
| 208 |
|
if (nprocs <= 0) |
| 209 |
|
error(USER, "illegal number of processes"); |
| 228 |
|
case 'f': /* file or i/o format */ |
| 229 |
|
if (!argv[i][2]) { |
| 230 |
|
char *fpath; |
| 231 |
< |
if (i >= argc-1) break; |
| 231 |
> |
if (i >= argc-2) break; |
| 232 |
|
fpath = getpath(argv[++i], |
| 233 |
|
getrlibpath(), R_OK); |
| 234 |
|
if (fpath == NULL) { |
| 243 |
|
setformat(argv[i]+2); |
| 244 |
|
continue; |
| 245 |
|
case 'e': /* expression */ |
| 246 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 246 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 247 |
|
scompile(argv[++i], NULL, 0); |
| 248 |
|
continue; |
| 249 |
|
case 'o': /* output file spec. */ |
| 250 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 250 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 251 |
|
curout = argv[++i]; |
| 252 |
|
continue; |
| 253 |
|
case 'x': /* horiz. output resolution */ |
| 254 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 254 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 255 |
|
xres = atoi(argv[++i]); |
| 256 |
|
continue; |
| 257 |
|
case 'y': /* vert. output resolution */ |
| 258 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 258 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 259 |
|
yres = atoi(argv[++i]); |
| 260 |
|
continue; |
| 261 |
|
case 'b': /* bin expression */ |
| 262 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 262 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 263 |
|
binval = argv[++i]; |
| 264 |
|
continue; |
| 265 |
|
case 'm': /* modifier name */ |
| 266 |
< |
if (argv[i][2] || i >= argc-1) break; |
| 266 |
> |
if (argv[i][2] || i >= argc-2) break; |
| 267 |
|
rtargv[rtargc++] = "-ti"; |
| 268 |
|
rtargv[rtargc++] = argv[++i]; |
| 269 |
|
addmodifier(argv[i], curout, binval); |
| 270 |
|
continue; |
| 271 |
+ |
case 'M': /* modifier file */ |
| 272 |
+ |
if (argv[i][2] || i >= argc-2) break; |
| 273 |
+ |
rtargv[rtargc++] = "-tI"; |
| 274 |
+ |
rtargv[rtargc++] = argv[++i]; |
| 275 |
+ |
addmodfile(argv[i], curout, binval); |
| 276 |
+ |
continue; |
| 277 |
|
} |
| 278 |
|
rtargv[rtargc++] = argv[i]; /* assume rtrace option */ |
| 279 |
|
} |
| 322 |
|
error(USER, "missing octree argument"); |
| 323 |
|
rtargv[rtargc++] = octree = argv[i]; |
| 324 |
|
rtargv[rtargc] = NULL; |
| 325 |
< |
/* start rtrace & compute contributions */ |
| 325 |
> |
/* start rtrace */ |
| 326 |
|
init(nprocs); |
| 327 |
|
if (recover) /* perform recovery if requested */ |
| 328 |
|
recover_output(stdin); |
| 329 |
< |
trace_contribs(stdin); |
| 329 |
> |
trace_contribs(stdin); /* compute contributions */ |
| 330 |
|
quit(0); |
| 331 |
|
} |
| 332 |
|
|
| 333 |
+ |
#ifndef SIGALRM |
| 334 |
+ |
#define SIGALRM SIGTERM |
| 335 |
+ |
#endif |
| 336 |
|
/* kill persistent rtrace process */ |
| 337 |
|
static void |
| 338 |
|
killpersist(void) |
| 339 |
|
{ |
| 340 |
|
FILE *fp = fopen(persistfn, "r"); |
| 341 |
< |
int pid; |
| 341 |
> |
RT_PID pid; |
| 342 |
|
|
| 343 |
|
if (fp == NULL) |
| 344 |
|
return; |
| 472 |
|
return mp; |
| 473 |
|
} |
| 474 |
|
|
| 475 |
+ |
/* add modifiers from a file list */ |
| 476 |
+ |
void |
| 477 |
+ |
addmodfile(char *fname, char *outf, char *binv) |
| 478 |
+ |
{ |
| 479 |
+ |
char *mname[MAXMODLIST]; |
| 480 |
+ |
int i; |
| 481 |
+ |
/* find the file & store strings */ |
| 482 |
+ |
if (wordfile(mname, getpath(fname, getrlibpath(), R_OK)) < 0) { |
| 483 |
+ |
sprintf(errmsg, "cannot find modifier file '%s'", fname); |
| 484 |
+ |
error(SYSTEM, errmsg); |
| 485 |
+ |
} |
| 486 |
+ |
for (i = 0; mname[i]; i++) /* add each one */ |
| 487 |
+ |
addmodifier(mname[i], outf, binv); |
| 488 |
+ |
} |
| 489 |
+ |
|
| 490 |
|
/* put string to stderr */ |
| 491 |
|
void |
| 492 |
|
eputs(char *s) |
| 633 |
|
if (header) { |
| 634 |
|
char info[512]; |
| 635 |
|
char *cp = info; |
| 636 |
< |
if (ofl & OF_MODIFIER) { |
| 637 |
< |
sprintf(cp, "MODIFIER=%s\n", mname); |
| 603 |
< |
while (*cp) ++cp; |
| 604 |
< |
} |
| 636 |
> |
sprintf(cp, "MODIFIER=%s\n", mname); |
| 637 |
> |
while (*cp) ++cp; |
| 638 |
|
if (ofl & OF_BIN) { |
| 639 |
|
sprintf(cp, "BIN=%d\n", bn); |
| 640 |
|
while (*cp) ++cp; |
| 1040 |
|
error(USER, "cannot recover from command"); |
| 1041 |
|
/* open output */ |
| 1042 |
|
fp = fopen(oname, "rb+"); |
| 1043 |
< |
if (fp == NULL) |
| 1044 |
< |
break; /* must be end of modifier */ |
| 1043 |
> |
if (fp == NULL) { |
| 1044 |
> |
if (j) |
| 1045 |
> |
break; /* assume end of modifier */ |
| 1046 |
> |
sprintf(errmsg, "missing recover file '%s'", |
| 1047 |
> |
oname); |
| 1048 |
> |
error(USER, errmsg); |
| 1049 |
> |
} |
| 1050 |
|
nvals = lseek(fileno(fp), 0, SEEK_END); |
| 1051 |
|
if (nvals <= 0) { |
| 1052 |
|
lastout = 0; /* empty output, quit here */ |
| 1083 |
|
break; /* no bin separation */ |
| 1084 |
|
} |
| 1085 |
|
if (!lastout) { /* empty output */ |
| 1086 |
< |
error(WARNING, "no data to recover"); |
| 1086 |
> |
error(WARNING, "no previous data to recover"); |
| 1087 |
|
lu_done(&ofiletab); /* reclose all outputs */ |
| 1088 |
|
return; |
| 1089 |
|
} |
| 1098 |
|
} |
| 1099 |
|
} |
| 1100 |
|
if (lastout < 0) { |
| 1101 |
< |
error(WARNING, "no existing output to recover"); |
| 1101 |
> |
error(WARNING, "no output files to recover"); |
| 1102 |
|
return; |
| 1103 |
+ |
} |
| 1104 |
+ |
if (raysleft && lastout >= raysleft) { |
| 1105 |
+ |
error(WARNING, "output appears to be complete"); |
| 1106 |
+ |
/* XXX should read & discard input? */ |
| 1107 |
+ |
quit(0); |
| 1108 |
|
} |
| 1109 |
|
/* seek on all files */ |
| 1110 |
|
nvals = lastout * outvsiz; |