--- ray/src/util/rpiece.c 1993/08/06 12:58:43 2.24 +++ ray/src/util/rpiece.c 2003/10/20 16:01:55 2.42 @@ -1,15 +1,12 @@ -/* Copyright (c) 1993 Regents of the University of California */ - #ifndef lint -static char SCCSid[] = "$SunId$ LBL"; +static const char RCSid[] = "$Id: rpiece.c,v 2.42 2003/10/20 16:01:55 greg Exp $"; #endif - /* * Generate sections of a picture. */ +#include "platform.h" #include "standard.h" -#include #ifndef F_SETLKW @@ -24,9 +21,10 @@ char *argv[]; #else #include + #include "color.h" #include "view.h" -#include "resolu.h" +#include "rtprocess.h" #ifndef NFS #define NFS 1 @@ -40,7 +38,7 @@ char *argv[]; #endif #endif /* protection from SYSV signals(!) */ -#if defined(sgi) || defined(hpux) +#if defined(sgi) #define guard_io() sighold(SIGALRM) #define unguard() sigrelse(SIGALRM) #endif @@ -48,10 +46,12 @@ char *argv[]; #define guard_io() 0 #define unguard() 0 #endif + +extern char *strerror(); + /* rpict command */ char *rpargv[128] = {"rpict", "-S", "1"}; int rpargc = 3; -int rpd[3]; FILE *torp, *fromrp; COLR *pbuf; /* our view parameters */ @@ -62,16 +62,19 @@ int hres = 1024, vres = 1024, hmult = 4, vmult = 4; char *outfile = NULL; int outfd; long scanorig; -int syncfd = -1; /* lock file descriptor */ +FILE *syncfp = NULL; /* synchronization file pointer */ +int synclst = F_UNLCK; /* synchronization file lock status */ int nforked = 0; +#define sflock(t) if ((t)!=synclst) dolock(fileno(syncfp),synclst=t) + char *progname; int verbose = 0; +unsigned timelim = 0; +int rvrlim = -1; -extern long lseek(), ftell(); - int gotalrm = 0; -int onalrm() { gotalrm++; } +void onalrm(int i) { gotalrm++; } main(argc, argv) @@ -83,12 +86,13 @@ char *argv[]; progname = argv[0]; for (i = 1; i < argc; i++) { /* expand arguments */ - while (rval = expandarg(&argc, &argv, i)) - if (rval < 0) { - fprintf(stderr, "%s: cannot expand '%s'", - argv[0], argv[i]); - exit(1); - } + while ((rval = expandarg(&argc, &argv, i)) > 0) + ; + if (rval < 0) { + fprintf(stderr, "%s: cannot expand '%s'", + argv[0], argv[i]); + exit(1); + } if (argv[i][0] == '-') switch (argv[i][1]) { case 'v': @@ -115,8 +119,13 @@ char *argv[]; case 'p': /* pixel aspect ratio? */ if (argv[i][2] != 'a' || argv[i][3]) break; - pixaspect = atof(argv[i+1]); + pixaspect = atof(argv[++i]); continue; + case 'T': /* time limit (hours) */ + if (argv[i][2]) + break; + timelim = atof(argv[++i])*3600. + .5; + break; case 'x': /* overall x resolution */ if (argv[i][2]) break; @@ -137,11 +146,16 @@ char *argv[]; break; vmult = atoi(argv[++i]); continue; + case 'R': /* recover */ + if (argv[i][2]) + break; + rvrlim = 0; + /* fall through */ case 'F': /* syncronization file */ if (argv[i][2]) break; - if ((syncfd = open(argv[++i], - O_RDWR|O_CREAT, 0666)) < 0) { + if ((syncfp = + fdopen(open(argv[++i],O_RDWR|O_CREAT,0666),"r+")) == NULL) { fprintf(stderr, "%s: cannot open\n", argv[i]); exit(1); @@ -174,6 +188,21 @@ char *argv[]; } +dolock(fd, ltyp) /* lock or unlock a file */ +int fd; +int ltyp; +{ + static struct flock fls; /* static so initialized to zeroes */ + + fls.l_type = ltyp; + if (fcntl(fd, F_SETLKW, &fls) < 0) { + fprintf(stderr, "%s: cannot lock/unlock file: %s\n", + progname, strerror(errno)); + exit(1); + } +} + + init(ac, av) /* set up output file and start rpict */ int ac; char **av; @@ -183,15 +212,16 @@ char **av; char *err; FILE *fp; int hr, vr; + SUBPROC rpd; /* since we don't close_process(), this can be local */ /* set up view */ if ((err = setview(&ourview)) != NULL) { fprintf(stderr, "%s: %s\n", progname, err); exit(1); } - if (syncfd != -1) { - char buf[32]; - buf[read(syncfd, buf, sizeof(buf)-1)] = '\0'; - sscanf(buf, "%d %d", &hmult, &vmult); + if (syncfp != NULL) { + sflock(F_RDLCK); + fscanf(syncfp, "%d %d", &hmult, &vmult); + sflock(F_UNLCK); } /* compute piece size */ hres /= hmult; @@ -206,9 +236,11 @@ char **av; rpargv[rpargc] = NULL; /* open output file */ if ((outfd = open(outfile, O_WRONLY|O_CREAT|O_EXCL, 0666)) >= 0) { + dolock(outfd, F_WRLCK); if ((fp = fdopen(dup(outfd), "w")) == NULL) goto filerr; - printargs(ac, av, fp); /* write header */ + newheader("RADIANCE", fp); /* create header */ + printargs(ac, av, fp); fprintf(fp, "SOFTWARE= %s\n", VersionID); fputs(VIEWSTR, fp); fprintview(&ourview, fp); @@ -219,6 +251,7 @@ char **av; putc('\n', fp); fprtresolu(hres*hmult, vres*vmult, fp); } else if ((outfd = open(outfile, O_RDWR)) >= 0) { + dolock(outfd, F_RDLCK); if ((fp = fdopen(dup(outfd), "r+")) == NULL) goto filerr; getheader(fp, NULL, NULL); /* skip header */ @@ -236,16 +269,14 @@ char **av; scanorig = ftell(fp); /* record position of first scanline */ if (fclose(fp) == -1) /* done with stream i/o */ goto filerr; -#if NFS - sync(); /* flush NFS buffers */ -#endif + dolock(outfd, F_UNLCK); /* start rpict process */ - if (open_process(rpd, rpargv) <= 0) { + if (open_process(&rpd, rpargv) <= 0) { fprintf(stderr, "%s: cannot start %s\n", progname, rpargv[0]); exit(1); } - if ((fromrp = fdopen(rpd[0], "r")) == NULL || - (torp = fdopen(rpd[1], "w")) == NULL) { + if ((fromrp = fdopen(rpd.r, "r")) == NULL || + (torp = fdopen(rpd.w, "w")) == NULL) { fprintf(stderr, "%s: cannot open stream to %s\n", progname, rpargv[0]); exit(1); @@ -255,6 +286,8 @@ char **av; exit(1); } signal(SIGALRM, onalrm); + if (timelim) + alarm(timelim); return; filerr: fprintf(stderr, "%s: i/o error on file \"%s\"\n", progname, outfile); @@ -266,47 +299,89 @@ int nextpiece(xp, yp) /* get next piece assignment */ int *xp, *yp; { - struct flock fls; - char buf[64]; - if (gotalrm) /* someone wants us to quit */ return(0); - if (syncfd != -1) { /* use sync file */ - fls.l_type = F_WRLCK; /* gain exclusive access */ - fls.l_whence = 0; - fls.l_start = 0L; - fls.l_len = 0L; - fcntl(syncfd, F_SETLKW, &fls); - lseek(syncfd, 0L, 0); - buf[read(syncfd, buf, sizeof(buf)-1)] = '\0'; - if (sscanf(buf, "%*d %*d %d %d", xp, yp) < 2) { + if (syncfp != NULL) { /* use sync file */ + /* + * So we don't necessarily have to lock and unlock the file + * multiple times (very slow), we establish an exclusive + * lock at the beginning on our synchronization file and + * maintain it in the subroutine rvrpiece(). + */ + sflock(F_WRLCK); + fseek(syncfp, 0L, 0); /* read position */ + if (fscanf(syncfp, "%*d %*d %d %d", xp, yp) < 2) { *xp = hmult-1; *yp = vmult; } + if (rvrlim == 0) /* initialize recovery limit */ + rvrlim = *xp*vmult + *yp; + if (rvrpiece(xp, yp)) { /* do stragglers first */ + sflock(F_UNLCK); + return(1); + } if (--(*yp) < 0) { /* decrement position */ *yp = vmult-1; - if (--(*xp) < 0) { /* all done! */ - close(syncfd); + if (--(*xp) < 0) { /* all done */ + sflock(F_UNLCK); return(0); } } - sprintf(buf, "%4d %4d\n%4d %4d\n", hmult, vmult, *xp, *yp); - lseek(syncfd, 0L, 0); /* write new position */ - write(syncfd, buf, strlen(buf)); - fls.l_type = F_UNLCK; /* release sync file */ - fcntl(syncfd, F_SETLKW, &fls); + fseek(syncfp, 0L, 0); /* write new position */ + fprintf(syncfp, "%4d %4d\n%4d %4d\n\n", hmult, vmult, *xp, *yp); + fflush(syncfp); + sflock(F_UNLCK); /* release sync file */ return(1); } - if (fgets(buf, sizeof(buf), stdin) == NULL) /* use stdin */ - return(0); - if (sscanf(buf, "%d %d", xp, yp) == 2) - return(1); - fprintf(stderr, "%s: input format error\n", progname); - exit(cleanup(1)); + return(scanf("%d %d", xp, yp) == 2); /* use stdin */ } int +rvrpiece(xp, yp) /* check for recoverable pieces */ +register int *xp, *yp; +{ + static char *pdone = NULL; /* which pieces are done */ + static long readpos = -1; /* how far we've read */ + register int i; + /* + * This routine is called by nextpiece() with an + * exclusive lock on syncfp and the file pointer at the + * appropriate position to read in the finished pieces. + */ + if (rvrlim < 0) + return(0); /* only check if asked */ + if (pdone == NULL) /* first call */ + pdone = calloc(hmult*vmult, sizeof(char)); + if (pdone == NULL) { + fprintf(stderr, "%s: out of memory\n", progname); + exit(1); + } + if (readpos != -1) /* mark what's been done */ + fseek(syncfp, readpos, 0); + while (fscanf(syncfp, "%d %d", xp, yp) == 2) + pdone[*xp*vmult+*yp] = 1; + if (!feof(syncfp)) { + fprintf(stderr, "%s: format error in sync file\n", progname); + exit(1); + } + readpos = ftell(syncfp); + i = hmult*vmult; /* find an unaccounted for piece */ + while (i-- > rvrlim) + if (!pdone[i]) { + *xp = i / vmult; + *yp = i % vmult; + pdone[i] = 1; /* consider it done */ + return(1); + } + rvrlim = -1; /* nothing left to recover */ + free(pdone); + pdone = NULL; + return(0); +} + + +int cleanup(rstat) /* close rpict process and clean up */ int rstat; { @@ -327,7 +402,7 @@ rpiece() /* render picture piece by piece */ VIEW pview; int xorg, yorg; /* compute view parameters */ - copystruct(&pview, &ourview); + pview = ourview; switch (ourview.type) { case VT_PER: pview.horiz = 2.*180./PI*atan( @@ -340,6 +415,11 @@ rpiece() /* render picture piece by piece */ pview.horiz = ourview.horiz / hmult; pview.vert = ourview.vert / vmult; break; + case VT_CYL: + pview.horiz = ourview.horiz / hmult; + pview.vert = 2.*180./PI*atan( + tan(PI/180./2.*ourview.vert)/vmult ); + break; case VT_HEM: pview.horiz = 2.*180./PI*asin( sin(PI/180./2.*ourview.horiz)/hmult ); @@ -353,8 +433,8 @@ rpiece() /* render picture piece by piece */ } /* render each piece */ while (nextpiece(&xorg, &yorg)) { - pview.hoff = ourview.hoff + xorg - 0.5*(hmult-1); - pview.voff = ourview.voff + yorg - 0.5*(vmult-1); + pview.hoff = ourview.hoff*hmult + xorg - 0.5*(hmult-1); + pview.voff = ourview.voff*vmult + yorg - 0.5*(vmult-1); fputs(VIEWSTR, torp); fprintview(&pview, torp); putc('\n', torp); @@ -373,7 +453,7 @@ int xpos, ypos; int hr, vr; register int y; /* check bounds */ - if (xpos < 0 | ypos < 0 | xpos >= hmult | ypos >= vmult) { + if ((xpos < 0) | (ypos < 0) | (xpos >= hmult) | (ypos >= vmult)) { fprintf(stderr, "%s: requested piece (%d,%d) out of range\n", progname, xpos, ypos); exit(cleanup(1)); @@ -381,7 +461,7 @@ int xpos, ypos; /* check header from rpict */ guard_io(); getheader(fromrp, NULL, NULL); - if (!fscnresolu(&hr, &vr, fromrp) || hr != hres | vr != vres) { + if (!fscnresolu(&hr, &vr, fromrp) || (hr != hres) | (vr != vres)) { fprintf(stderr, "%s: resolution mismatch from %s\n", progname, rpargv[0]); exit(cleanup(1)); @@ -422,42 +502,54 @@ int xpos, ypos; /* lock file section so NFS doesn't mess up */ fls.l_whence = 0; fls.l_type = F_WRLCK; - fcntl(outfd, F_SETLKW, &fls); + if (fcntl(outfd, F_SETLKW, &fls) < 0) + filerr("lock"); #endif /* write new piece to file */ - if (lseek(outfd, fls.l_start, 0) == -1) - goto seekerr; + if (lseek(outfd, (off_t)fls.l_start, SEEK_SET) < 0) + filerr("seek"); if (hmult == 1) { if (writebuf(outfd, (char *)pbuf, vr*hr*sizeof(COLR)) != vr*hr*sizeof(COLR)) - goto writerr; + filerr("write"); } else for (y = 0; y < vr; y++) { if (writebuf(outfd, (char *)(pbuf+y*hr), hr*sizeof(COLR)) != hr*sizeof(COLR)) - goto writerr; + filerr("write"); if (y < vr-1 && lseek(outfd, - (long)(hmult-1)*hr*sizeof(COLR), - 1) == -1) - goto seekerr; + (off_t)(hmult-1)*hr*sizeof(COLR), + SEEK_CUR) < 0) + filerr("seek"); } +#if NFS + fls.l_type = F_UNLCK; /* release lock */ + if (fcntl(outfd, F_SETLKW, &fls) < 0) + filerr("lock"); +#endif if (verbose) { /* notify caller */ printf("%d %d done\n", xpos, ypos); fflush(stdout); } - if (pid == -1) { /* didn't fork or fork failed */ -#if NFS - fls.l_type = F_UNLCK; /* release lock */ - fcntl(outfd, F_SETLKW, &fls); -#endif - return(0); + if (syncfp != NULL) { /* record what's been done */ + sflock(F_WRLCK); + fseek(syncfp, 0L, 2); /* append index */ + fprintf(syncfp, "%4d %4d\n", xpos, ypos); + fflush(syncfp); + /*** Unlock not necessary, since + sflock(F_UNLCK); _exit() or nextpiece() is next ***/ } - _exit(0); /* else exit child process (releasing lock) */ -seekerr: - fprintf(stderr, "%s: seek error on file \"%s\"\n", progname, outfile); - _exit(1); -writerr: - fprintf(stderr, "%s: write error on file \"%s\"\n", progname, outfile); + if (pid == -1) /* didn't fork or fork failed */ + return(0); + _exit(0); /* else exit child process (releasing locks) */ +} + + +filerr(t) /* report file error and exit */ +char *t; +{ + fprintf(stderr, "%s: %s error on file \"%s\": %s\n", + progname, t, outfile, strerror(errno)); _exit(1); }