--- ray/src/util/rpiece.c 1993/08/06 12:58:43 2.24 +++ ray/src/util/rpiece.c 1998/06/18 09:23:49 2.36 @@ -1,4 +1,4 @@ -/* Copyright (c) 1993 Regents of the University of California */ +/* Copyright (c) 1995 Regents of the University of California */ #ifndef lint static char SCCSid[] = "$SunId$ LBL"; @@ -9,7 +9,6 @@ static char SCCSid[] = "$SunId$ LBL"; */ #include "standard.h" -#include #ifndef F_SETLKW @@ -40,7 +39,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,6 +47,11 @@ char *argv[]; #define guard_io() 0 #define unguard() 0 #endif + +#ifndef linux +extern char *sys_errlist[]; +#endif + /* rpict command */ char *rpargv[128] = {"rpict", "-S", "1"}; int rpargc = 3; @@ -62,14 +66,17 @@ 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++; } @@ -115,8 +122,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 +149,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 +191,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, sys_errlist[errno]); + exit(1); + } +} + + init(ac, av) /* set up output file and start rpict */ int ac; char **av; @@ -188,10 +220,10 @@ char **av; 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 +238,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 +253,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,9 +271,7 @@ 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) { fprintf(stderr, "%s: cannot start %s\n", progname, rpargv[0]); @@ -255,6 +288,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 +301,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; { @@ -340,6 +417,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 +435,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); @@ -422,42 +504,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, fls.l_start, 0) < 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; + 1) < 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, sys_errlist[errno]); _exit(1); }