src/util/glareval.c

/* Copyright (c) 1991 Regents of the University of California */

#ifndef lint
static char SCCSid[] = "$SunId$ LBL";
#endif

/*
 * Compute pixels for glare calculation
 */

#include "glare.h"
#include <sys/param.h>
                                        /* compute rtrace buffer size */
#ifndef PIPE_BUF
#define PIPE_BUF        512             /* hyperconservative */
#endif
#define MAXPIX          (PIPE_BUF/(6*sizeof(float)) - 1)

#ifndef BSD
#define vfork           fork
#endif

#define MAXSBUF         524268  /* maximum total size of scanline buffer */
#define HSIZE           317     /* size of scanline hash table */
#define NRETIRE         16      /* number of scanlines to retire at once */

int     rt_pid = -1;            /* process id for rtrace */
int     fd_tort, fd_fromrt;     /* pipe descriptors */

FILE    *pictfp = NULL;         /* picture file pointer */
double  exposure;               /* picture exposure */
int     pxsiz, pysiz;           /* picture dimensions */

static int      curpos;         /* current scanline */
static long     *scanpos;       /* scanline positions */

typedef struct scan {
        int     y;              /* scanline position */
        long    lused;          /* for LRU replacement */
        struct scan     *next;  /* next in this hash or free list */
        /* followed by the scanline data */
} SCAN;                 /* buffered scanline */

#define scandata(sl)    ((COLR *)((sl)+1))
#define shash(y)        ((y)%HSIZE)

static SCAN     *freelist;              /* scanline free list */
static SCAN     *hashtab[HSIZE];        /* scanline hash table */

static long     ncall = 0L;     /* number of calls to getpictscan */
static long     nread = 0L;     /* number of scanlines read */

static int      wrongformat = 0;

SCAN    *scanretire();

extern long     ftell();


SCAN *
claimscan(y)                    /* claim scanline from buffers */
int     y;
{
        int     hi = shash(y);
        SCAN    *slast;
        register SCAN   *sl;

        for (sl = hashtab[hi]; sl != NULL; sl = sl->next)
                if (sl->y == y)                         /* active scanline */
                        return(sl);
        for (slast = NULL, sl = freelist; sl != NULL; slast = sl, sl = sl->next)
                if (sl->y == -1 || sl->y == y || sl->next == NULL) {
                        if (slast == NULL)              /* remove from free */
                                freelist = sl->next;
                        else
                                slast->next = sl->next;
                        if (sl->y == y) {               /* reclaim */
                                sl->next = hashtab[hi];
                                hashtab[hi] = sl;
#ifdef DEBUG
                                if (verbose)
                                        fprintf(stderr,
                                                "%s: scanline %d reclaimed\n",
                                                        progname, y);
#endif
                        }
                        return(sl);
                }
        return(scanretire());           /* need more free scanlines */
}


COLR *
getpictscan(y)                  /* get picture scanline */
int     y;
{
        register SCAN   *sl;
        register int    i;
                                        /* first check our buffers */
        sl = claimscan(y);
        if (sl == NULL)
                memerr("claimscan()");
        sl->lused = ncall++;
        if (sl->y == y)                 /* scan hit */
                return(scandata(sl));
                                        /* else read in replacement */
        if (scanpos[y] < 0) {                   /* need to search */
                for (i = y+1; i < curpos; i++)
                        if (scanpos[i] >= 0) {
                                if (fseek(pictfp, scanpos[i], 0) < 0)
                                        goto seekerr;
                                curpos = i;
                                break;
                        }
                while (curpos >= y) {
                        scanpos[curpos] = ftell(pictfp);
                        if (freadcolrs(scandata(sl), pxsiz, pictfp) < 0)
                                goto readerr;
                        nread++;
                        curpos--;
                }
        } else {
                if (curpos != y && fseek(pictfp, scanpos[y], 0) < 0)
                        goto seekerr;
                if (freadcolrs(scandata(sl), pxsiz, pictfp) < 0)
                        goto readerr;
                nread++;
                curpos = y-1;
        }
        sl->y = y;
        i = shash(y);                   /* add to hash list */
        sl->next = hashtab[i];
        hashtab[i] = sl;
        return(scandata(sl));
readerr:
        fprintf(stderr, "%s: picture read error\n", progname);
        exit(1);
seekerr:
        fprintf(stderr, "%s: picture seek error\n", progname);
        exit(1);
}


#ifdef DEBUG
pict_stats()                    /* print out picture read statistics */
{
        static long     lastcall = 0L;  /* ncall at last report */
        static long     lastread = 0L;  /* nread at last report */

        if (ncall == lastcall)
                return;
        fprintf(stderr, "%s: %ld scanlines read in %ld calls\n",
                        progname, nread-lastread, ncall-lastcall);
        lastcall = ncall;
        lastread = nread;
}
#endif


double
pict_val(vd)                    /* find picture value for view direction */
FVECT   vd;
{
        FVECT   pp;
        double  vpx, vpy, vpz;
        COLOR   res;

        if (pictfp == NULL)
                return(-1.0);
        pp[0] = pictview.vp[0] + vd[0];
        pp[1] = pictview.vp[1] + vd[1];
        pp[2] = pictview.vp[2] + vd[2];
        viewpixel(&vpx, &vpy, &vpz, &pictview, pp);
        if (vpz <= FTINY || vpx < 0. || vpx >= 1. || vpy < 0. || vpy >= 1.)
                return(-1.0);
        colr_color(res, getpictscan((int)(vpy*pysiz))[(int)(vpx*pxsiz)]);
        return(luminance(res)/exposure);
}


double
getviewpix(vh, vv)              /* compute single view pixel */
int     vh, vv;
{
        FVECT   dir;
        float   rt_buf[6];
        double  res;

        if (compdir(dir, vh, vv) < 0)
                return(-1.0);
        npixinvw++;
        if ((res = pict_val(dir)) >= 0.0)
                return(res);
        if (rt_pid == -1) {
                npixmiss++;
                return(-1.0);
        }
        rt_buf[0] = ourview.vp[0];
        rt_buf[1] = ourview.vp[1];
        rt_buf[2] = ourview.vp[2];
        rt_buf[3] = dir[0];
        rt_buf[4] = dir[1];
        rt_buf[5] = dir[2];
        rt_compute(rt_buf, 1);
        return(luminance(rt_buf));
}


getviewspan(vv, vb)             /* compute a span of view pixels */
int     vv;
float   *vb;
{
        float   rt_buf[6*MAXPIX];       /* rtrace send/receive buffer */
        register int    n;              /* number of pixels in buffer */
        short   buf_vh[MAXPIX];         /* pixel positions */
        FVECT   dir;
        register int    vh;

#ifdef DEBUG
        if (verbose)
                fprintf(stderr, "%s: computing view span at %d...\n",
                                progname, vv);
#endif
        n = 0;
        for (vh = -hsize; vh <= hsize; vh++) {
                if (compdir(dir, vh, vv) < 0) {         /* not in view */
                        vb[vh+hsize] = -1.0;
                        continue;
                }
                npixinvw++;
                if ((vb[vh+hsize] = pict_val(dir)) >= 0.0)
                        continue;
                if (rt_pid == -1) {             /* missing information */
                        npixmiss++;
                        continue;
                }
                                                /* send to rtrace */
                if (n >= MAXPIX) {                      /* flush */
                        rt_compute(rt_buf, n);
                        while (n-- > 0)
                                vb[buf_vh[n]+hsize] = luminance(rt_buf+3*n);
                }
                rt_buf[6*n] = ourview.vp[0];
                rt_buf[6*n+1] = ourview.vp[1];
                rt_buf[6*n+2] = ourview.vp[2];
                rt_buf[6*n+3] = dir[0];
                rt_buf[6*n+4] = dir[1];
                rt_buf[6*n+5] = dir[2];
                buf_vh[n++] = vh;
        }
#ifdef DEBUG
        if (verbose)
                pict_stats();
#endif
        if (n > 0) {                            /* process pending buffer */
                rt_compute(rt_buf, n);
                while (n-- > 0)
                        vb[buf_vh[n]+hsize] = luminance(rt_buf+3*n);
        }
}


rt_compute(pb, np)              /* process buffer through rtrace */
float   *pb;
int     np;
{
        static float    nbuf[6] = {0.,0.,0.,0.,0.,0.};

#ifdef DEBUG
        if (verbose && np > 1)
                fprintf(stderr, "%s: sending %d samples to rtrace...\n",
                                progname, np);
#endif
        if (writebuf(fd_tort,(char *)pb,6*sizeof(float)*np) < 6*sizeof(float)*np
                || writebuf(fd_tort,(char *)nbuf,sizeof(nbuf)) < sizeof(nbuf)) {
                fprintf(stderr, "%s: error writing to rtrace process\n",
                                progname);
                exit(1);
        }
        if (readbuf(fd_fromrt, (char *)pb, 3*sizeof(float)*np)
                        < 3*sizeof(float)*np) {
                fprintf(stderr, "%s: error reading from rtrace process\n",
                                progname);
                exit(1);
        }
}


getexpos(s)                     /* get exposure from header line */
char    *s;
{
        char    fmt[32];

        if (isexpos(s))
                exposure *= exposval(s);
        else if (isformat(s)) {
                formatval(fmt, s);
                wrongformat = strcmp(fmt, COLRFMT);
        }
}


open_pict(fn)                   /* open picture file */
char    *fn;
{
        if ((pictfp = fopen(fn, "r")) == NULL) {
                fprintf("%s: cannot open\n", fn);
                exit(1);
        }
        exposure = 1.0;
        getheader(pictfp, getexpos, NULL);
        if (wrongformat ||
                        fgetresolu(&pxsiz, &pysiz, pictfp) != (YMAJOR|YDECR)) {
                fprintf("%s: bad picture format\n", fn);
                exit(1);
        }
        initscans();
}


close_pict()                    /* done with picture */
{
        if (pictfp == NULL)
                return;
        fclose(pictfp);
        donescans();
        pictfp = NULL;
}


fork_rtrace(av)                 /* open pipe and start rtrace */
char    *av[];
{
        int     p0[2], p1[2];

        if (pipe(p0) < 0 || pipe(p1) < 0) {
                perror(progname);
                exit(1);
        }
        if ((rt_pid = vfork()) == 0) {          /* if child */
                close(p0[1]);
                close(p1[0]);
                if (p0[0] != 0) {       /* connect p0 to stdin */
                        dup2(p0[0], 0);
                        close(p0[0]);
                }
                if (p1[1] != 0) {       /* connect p1 to stdout */
                        dup2(p1[1], 1);
                        close(p1[1]);
                }
                execvp(av[0], av);
                perror(av[0]);
                _exit(127);
        }
        if (rt_pid == -1) {
                perror(progname);
                exit(1);
        }
        close(p0[0]);
        close(p1[1]);
        fd_tort = p0[1];
        fd_fromrt = p1[0];
}


done_rtrace()                   /* wait for rtrace to finish */
{
        int     pid, status;

        if (rt_pid == -1)
                return;
        close(fd_tort);
        close(fd_fromrt);
        while ((pid = wait(&status)) != -1 && pid != rt_pid)
                ;
        if (pid == rt_pid && status != 0) {
                fprintf(stderr, "%s: bad status (%d) from rtrace\n",
                                progname, status);
                exit(1);
        }
        rt_pid = -1;
}


int
readbuf(fd, bpos, siz)
int     fd;
char    *bpos;
int     siz;
{
        register int    cc, nrem = siz;

        while (nrem > 0 && (cc = read(fd, bpos, nrem)) > 0) {
                bpos += cc;
                nrem -= cc;
        }
        if (cc < 0)
                return(cc);
        return(siz-nrem);
}


int
writebuf(fd, bpos, siz)
char    *bpos;
int     siz;
{
        register int    cc, nrem = siz;

        while (nrem > 0 && (cc = write(fd, bpos, nrem)) > 0) {
                bpos += cc;
                nrem -= cc;
        }
        if (cc < 0)
                return(cc);
        return(siz-nrem);
}


SCAN *
scanretire()                    /* retire old scanlines to free list */
{
        SCAN    *sold[NRETIRE];
        int     n;
        int     h;
        register SCAN   *sl;
        register int    i;
                                        /* grab the NRETIRE oldest scanlines */
        sold[n = 0] = NULL;
        for (h = 0; h < HSIZE; h++)
                for (sl = hashtab[h]; sl != NULL; sl = sl->next) {
                        for (i = n; i && sold[i-1]->lused > sl->lused; i--)
                                if (i < NRETIRE)
                                        sold[i] = sold[i-1];
                        if (i < NRETIRE) {
                                sold[i] = sl;
                                if (n < NRETIRE)        /* grow list */
                                        n++;
                        }
                }
                                        /* put scanlines into free list */
        for (i = 0; i < n; i++) {
                h = shash(sold[i]->y);
                sl = hashtab[h];
                if (sl == sold[i])
                        hashtab[h] = sl->next;
                else {
                        while (sl->next != sold[i])     /* IS in list */
                                sl = sl->next;
                        sl->next = sold[i]->next;
                }
                if (i > 0) {            /* save oldest as return value */
                        sold[i]->next = freelist;
                        freelist = sold[i];
                }
        }
        return(sold[0]);
}


static char     *scan_buf;


initscans()                             /* initialize scanline buffers */
{
        int     scansize;
        register SCAN   *ptr;
        register int    i;
                                        /* initialize positions */
        scanpos = (long *)malloc(pysiz*sizeof(long));
        if (scanpos == NULL)
                memerr("scanline positions");
        for (i = pysiz-1; i >= 0; i--)
                scanpos[i] = -1L;
        curpos = pysiz-1;
                                        /* clear hash table */
        for (i = 0; i < HSIZE; i++)
                hashtab[i] = NULL;
                                        /* allocate scanline buffers */
        scansize = sizeof(SCAN) + pxsiz*sizeof(COLR);
#ifdef ALIGN
        scansize = scansize+(sizeof(ALIGN)-1)) & ~(sizeof(ALIGN)-1);
#endif
        i = MAXSBUF / scansize;         /* compute number to allocate */
        if (i > HSIZE)
                i = HSIZE;
        scan_buf = malloc(i*scansize);  /* get in one big chunk */
        if (scan_buf == NULL)
                memerr("scanline buffers");
        ptr = (SCAN *)scan_buf;
        freelist = NULL;                /* build our free list */
        while (i-- > 0) {
                ptr->y = -1;
                ptr->lused = -1;
                ptr->next = freelist;
                freelist = ptr;
                ptr = (SCAN *)((char *)ptr + scansize); /* beware of C bugs */
        }
}


donescans()                             /* free up scanlines */
{
        free(scan_buf);
        free((char *)scanpos);
}
Revision:	1.14
Committed:	Fri Apr 19 17:41:06 1991 UTC (33 years ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 1.13:	+3 -6 lines
Log Message:	fixed computation of solid angle changed compdir()
#	User	Rev	Content
1	greg	1.1	/* Copyright (c) 1991 Regents of the University of California */
2
3			#ifndef lint
4			static char SCCSid[] = "$SunId$ LBL";
5			#endif
6
7			/*
8			* Compute pixels for glare calculation
9			*/
10
11			#include "glare.h"
12			#include <sys/param.h>
13			/* compute rtrace buffer size */
14			#ifndef PIPE_BUF
15			#define PIPE_BUF 512 /* hyperconservative */
16			#endif
17			#define MAXPIX (PIPE_BUF/(6*sizeof(float)) - 1)
18
19			#ifndef BSD
20			#define vfork fork
21			#endif
22
23	greg	1.11	#define MAXSBUF 524268 /* maximum total size of scanline buffer */
24	greg	1.9	#define HSIZE 317 /* size of scanline hash table */
25	greg	1.8	#define NRETIRE 16 /* number of scanlines to retire at once */
26	greg	1.1
27			int rt_pid = -1; /* process id for rtrace */
28			int fd_tort, fd_fromrt; /* pipe descriptors */
29
30			FILE pictfp = NULL; / picture file pointer */
31			double exposure; /* picture exposure */
32			int pxsiz, pysiz; /* picture dimensions */
33	greg	1.8
34			static int curpos; /* current scanline */
35			static long scanpos; / scanline positions */
36
37			typedef struct scan {
38			int y; /* scanline position */
39	greg	1.1	long lused; /* for LRU replacement */
40	greg	1.8	struct scan next; / next in this hash or free list */
41			/* followed by the scanline data */
42			} SCAN; /* buffered scanline */
43	greg	1.1
44	greg	1.8	#define scandata(sl) ((COLR *)((sl)+1))
45			#define shash(y) ((y)%HSIZE)
46
47			static SCAN freelist; / scanline free list */
48			static SCAN hashtab[HSIZE]; / scanline hash table */
49
50	greg	1.2	static long ncall = 0L; /* number of calls to getpictscan */
51			static long nread = 0L; /* number of scanlines read */
52	greg	1.1
53	greg	1.13	static int wrongformat = 0;
54
55	greg	1.8	SCAN *scanretire();
56
57	greg	1.5	extern long ftell();
58	greg	1.2
59	greg	1.5
60	greg	1.8	SCAN *
61			claimscan(y) /* claim scanline from buffers */
62			int y;
63			{
64			int hi = shash(y);
65			SCAN *slast;
66			register SCAN *sl;
67
68			for (sl = hashtab[hi]; sl != NULL; sl = sl->next)
69			if (sl->y == y) /* active scanline */
70			return(sl);
71			for (slast = NULL, sl = freelist; sl != NULL; slast = sl, sl = sl->next)
72			if (sl->y == -1 \|\| sl->y == y \|\| sl->next == NULL) {
73			if (slast == NULL) /* remove from free */
74			freelist = sl->next;
75			else
76			slast->next = sl->next;
77			if (sl->y == y) { /* reclaim */
78			sl->next = hashtab[hi];
79			hashtab[hi] = sl;
80	greg	1.11	#ifdef DEBUG
81			if (verbose)
82			fprintf(stderr,
83			"%s: scanline %d reclaimed\n",
84			progname, y);
85			#endif
86	greg	1.8	}
87			return(sl);
88			}
89			return(scanretire()); /* need more free scanlines */
90			}
91
92
93	greg	1.1	COLR *
94			getpictscan(y) /* get picture scanline */
95			int y;
96			{
97	greg	1.8	register SCAN *sl;
98	greg	1.1	register int i;
99			/* first check our buffers */
100	greg	1.8	sl = claimscan(y);
101			if (sl == NULL)
102			memerr("claimscan()");
103			sl->lused = ncall++;
104			if (sl->y == y) /* scan hit */
105			return(scandata(sl));
106			/* else read in replacement */
107	greg	1.5	if (scanpos[y] < 0) { /* need to search */
108			for (i = y+1; i < curpos; i++)
109			if (scanpos[i] >= 0) {
110			if (fseek(pictfp, scanpos[i], 0) < 0)
111			goto seekerr;
112			curpos = i;
113			break;
114			}
115			while (curpos >= y) {
116	greg	1.1	scanpos[curpos] = ftell(pictfp);
117	greg	1.8	if (freadcolrs(scandata(sl), pxsiz, pictfp) < 0)
118	greg	1.1	goto readerr;
119	greg	1.5	nread++;
120	greg	1.1	curpos--;
121			}
122	greg	1.5	} else {
123			if (curpos != y && fseek(pictfp, scanpos[y], 0) < 0)
124			goto seekerr;
125	greg	1.8	if (freadcolrs(scandata(sl), pxsiz, pictfp) < 0)
126	greg	1.5	goto readerr;
127			nread++;
128			curpos = y-1;
129	greg	1.1	}
130	greg	1.8	sl->y = y;
131			i = shash(y); /* add to hash list */
132			sl->next = hashtab[i];
133			hashtab[i] = sl;
134			return(scandata(sl));
135	greg	1.1	readerr:
136			fprintf(stderr, "%s: picture read error\n", progname);
137			exit(1);
138	greg	1.5	seekerr:
139			fprintf(stderr, "%s: picture seek error\n", progname);
140			exit(1);
141	greg	1.1	}
142
143
144	greg	1.6	#ifdef DEBUG
145	greg	1.2	pict_stats() /* print out picture read statistics */
146			{
147			static long lastcall = 0L; /* ncall at last report */
148			static long lastread = 0L; /* nread at last report */
149
150			if (ncall == lastcall)
151			return;
152	greg	1.6	fprintf(stderr, "%s: %ld scanlines read in %ld calls\n",
153			progname, nread-lastread, ncall-lastcall);
154	greg	1.2	lastcall = ncall;
155			lastread = nread;
156			}
157	greg	1.6	#endif
158	greg	1.2
159
160	greg	1.1	double
161			pict_val(vd) /* find picture value for view direction */
162			FVECT vd;
163			{
164			FVECT pp;
165			double vpx, vpy, vpz;
166			COLOR res;
167
168			if (pictfp == NULL)
169			return(-1.0);
170			pp[0] = pictview.vp[0] + vd[0];
171			pp[1] = pictview.vp[1] + vd[1];
172			pp[2] = pictview.vp[2] + vd[2];
173			viewpixel(&vpx, &vpy, &vpz, &pictview, pp);
174			if (vpz <= FTINY \|\| vpx < 0. \|\| vpx >= 1. \|\| vpy < 0. \|\| vpy >= 1.)
175			return(-1.0);
176			colr_color(res, getpictscan((int)(vpypysiz))[(int)(vpxpxsiz)]);
177			return(luminance(res)/exposure);
178			}
179
180
181			double
182	greg	1.14	getviewpix(vh, vv) /* compute single view pixel */
183	greg	1.1	int vh, vv;
184			{
185			FVECT dir;
186			float rt_buf[6];
187			double res;
188
189	greg	1.14	if (compdir(dir, vh, vv) < 0)
190	greg	1.1	return(-1.0);
191	greg	1.7	npixinvw++;
192	greg	1.1	if ((res = pict_val(dir)) >= 0.0)
193			return(res);
194	greg	1.7	if (rt_pid == -1) {
195			npixmiss++;
196	greg	1.1	return(-1.0);
197	greg	1.7	}
198	greg	1.1	rt_buf[0] = ourview.vp[0];
199			rt_buf[1] = ourview.vp[1];
200			rt_buf[2] = ourview.vp[2];
201			rt_buf[3] = dir[0];
202			rt_buf[4] = dir[1];
203			rt_buf[5] = dir[2];
204			rt_compute(rt_buf, 1);
205			return(luminance(rt_buf));
206			}
207
208
209			getviewspan(vv, vb) /* compute a span of view pixels */
210			int vv;
211			float *vb;
212			{
213			float rt_buf[6MAXPIX]; / rtrace send/receive buffer */
214	greg	1.3	register int n; /* number of pixels in buffer */
215	greg	1.1	short buf_vh[MAXPIX]; /* pixel positions */
216			FVECT dir;
217			register int vh;
218
219	greg	1.4	#ifdef DEBUG
220	greg	1.1	if (verbose)
221			fprintf(stderr, "%s: computing view span at %d...\n",
222			progname, vv);
223	greg	1.4	#endif
224	greg	1.3	n = 0;
225	greg	1.1	for (vh = -hsize; vh <= hsize; vh++) {
226	greg	1.14	if (compdir(dir, vh, vv) < 0) { /* not in view */
227	greg	1.1	vb[vh+hsize] = -1.0;
228			continue;
229			}
230	greg	1.7	npixinvw++;
231	greg	1.1	if ((vb[vh+hsize] = pict_val(dir)) >= 0.0)
232			continue;
233	greg	1.7	if (rt_pid == -1) { /* missing information */
234			npixmiss++;
235	greg	1.1	continue;
236	greg	1.7	}
237	greg	1.1	/* send to rtrace */
238	greg	1.3	if (n >= MAXPIX) { /* flush */
239			rt_compute(rt_buf, n);
240			while (n-- > 0)
241			vb[buf_vh[n]+hsize] = luminance(rt_buf+3*n);
242	greg	1.1	}
243	greg	1.3	rt_buf[6*n] = ourview.vp[0];
244			rt_buf[6*n+1] = ourview.vp[1];
245			rt_buf[6*n+2] = ourview.vp[2];
246			rt_buf[6*n+3] = dir[0];
247			rt_buf[6*n+4] = dir[1];
248			rt_buf[6*n+5] = dir[2];
249			buf_vh[n++] = vh;
250	greg	1.1	}
251	greg	1.4	#ifdef DEBUG
252	greg	1.2	if (verbose)
253			pict_stats();
254	greg	1.4	#endif
255	greg	1.3	if (n > 0) { /* process pending buffer */
256			rt_compute(rt_buf, n);
257			while (n-- > 0)
258			vb[buf_vh[n]+hsize] = luminance(rt_buf+3*n);
259			}
260	greg	1.1	}
261
262
263			rt_compute(pb, np) /* process buffer through rtrace */
264			float *pb;
265			int np;
266			{
267			static float nbuf[6] = {0.,0.,0.,0.,0.,0.};
268
269	greg	1.4	#ifdef DEBUG
270	greg	1.1	if (verbose && np > 1)
271			fprintf(stderr, "%s: sending %d samples to rtrace...\n",
272			progname, np);
273	greg	1.4	#endif
274	greg	1.1	if (writebuf(fd_tort,(char )pb,6sizeof(float)np) < 6sizeof(float)*np
275			\|\| writebuf(fd_tort,(char *)nbuf,sizeof(nbuf)) < sizeof(nbuf)) {
276			fprintf(stderr, "%s: error writing to rtrace process\n",
277			progname);
278			exit(1);
279			}
280			if (readbuf(fd_fromrt, (char )pb, 3sizeof(float)*np)
281			< 3sizeof(float)np) {
282			fprintf(stderr, "%s: error reading from rtrace process\n",
283			progname);
284			exit(1);
285			}
286			}
287
288
289			getexpos(s) /* get exposure from header line */
290			char *s;
291			{
292	greg	1.13	char fmt[32];
293
294	greg	1.1	if (isexpos(s))
295			exposure *= exposval(s);
296	greg	1.13	else if (isformat(s)) {
297			formatval(fmt, s);
298			wrongformat = strcmp(fmt, COLRFMT);
299			}
300	greg	1.1	}
301
302
303			open_pict(fn) /* open picture file */
304			char *fn;
305			{
306			if ((pictfp = fopen(fn, "r")) == NULL) {
307			fprintf("%s: cannot open\n", fn);
308			exit(1);
309			}
310			exposure = 1.0;
311	greg	1.13	getheader(pictfp, getexpos, NULL);
312			if (wrongformat \|\|
313			fgetresolu(&pxsiz, &pysiz, pictfp) != (YMAJOR\|YDECR)) {
314			fprintf("%s: bad picture format\n", fn);
315	greg	1.1	exit(1);
316			}
317	greg	1.8	initscans();
318	greg	1.1	}
319
320
321			close_pict() /* done with picture */
322			{
323			if (pictfp == NULL)
324			return;
325			fclose(pictfp);
326	greg	1.8	donescans();
327	greg	1.1	pictfp = NULL;
328			}
329
330
331			fork_rtrace(av) /* open pipe and start rtrace */
332			char *av[];
333			{
334			int p0[2], p1[2];
335
336			if (pipe(p0) < 0 \|\| pipe(p1) < 0) {
337			perror(progname);
338			exit(1);
339			}
340			if ((rt_pid = vfork()) == 0) { /* if child */
341			close(p0[1]);
342			close(p1[0]);
343			if (p0[0] != 0) { /* connect p0 to stdin */
344			dup2(p0[0], 0);
345			close(p0[0]);
346			}
347			if (p1[1] != 0) { /* connect p1 to stdout */
348			dup2(p1[1], 1);
349			close(p1[1]);
350			}
351			execvp(av[0], av);
352			perror(av[0]);
353			_exit(127);
354			}
355			if (rt_pid == -1) {
356			perror(progname);
357			exit(1);
358			}
359	greg	1.3	close(p0[0]);
360			close(p1[1]);
361	greg	1.1	fd_tort = p0[1];
362			fd_fromrt = p1[0];
363			}
364
365
366			done_rtrace() /* wait for rtrace to finish */
367			{
368			int pid, status;
369
370			if (rt_pid == -1)
371			return;
372			close(fd_tort);
373			close(fd_fromrt);
374			while ((pid = wait(&status)) != -1 && pid != rt_pid)
375			;
376			if (pid == rt_pid && status != 0) {
377			fprintf(stderr, "%s: bad status (%d) from rtrace\n",
378			progname, status);
379			exit(1);
380			}
381			rt_pid = -1;
382			}
383
384
385			int
386			readbuf(fd, bpos, siz)
387			int fd;
388			char *bpos;
389			int siz;
390			{
391			register int cc, nrem = siz;
392
393			while (nrem > 0 && (cc = read(fd, bpos, nrem)) > 0) {
394			bpos += cc;
395			nrem -= cc;
396			}
397			if (cc < 0)
398			return(cc);
399			return(siz-nrem);
400			}
401
402
403			int
404			writebuf(fd, bpos, siz)
405			char *bpos;
406			int siz;
407			{
408			register int cc, nrem = siz;
409
410			while (nrem > 0 && (cc = write(fd, bpos, nrem)) > 0) {
411			bpos += cc;
412			nrem -= cc;
413			}
414			if (cc < 0)
415			return(cc);
416			return(siz-nrem);
417	greg	1.8	}
418
419
420			SCAN *
421			scanretire() /* retire old scanlines to free list */
422			{
423			SCAN *sold[NRETIRE];
424			int n;
425			int h;
426			register SCAN *sl;
427			register int i;
428			/* grab the NRETIRE oldest scanlines */
429			sold[n = 0] = NULL;
430	greg	1.10	for (h = 0; h < HSIZE; h++)
431			for (sl = hashtab[h]; sl != NULL; sl = sl->next) {
432			for (i = n; i && sold[i-1]->lused > sl->lused; i--)
433			if (i < NRETIRE)
434	greg	1.8	sold[i] = sold[i-1];
435			if (i < NRETIRE) {
436	greg	1.10	sold[i] = sl;
437	greg	1.8	if (n < NRETIRE) /* grow list */
438			n++;
439	greg	1.10	}
440			}
441			/* put scanlines into free list */
442			for (i = 0; i < n; i++) {
443			h = shash(sold[i]->y);
444			sl = hashtab[h];
445			if (sl == sold[i])
446			hashtab[h] = sl->next;
447			else {
448			while (sl->next != sold[i]) /* IS in list */
449	greg	1.8	sl = sl->next;
450	greg	1.10	sl->next = sold[i]->next;
451	greg	1.8	}
452	greg	1.10	if (i > 0) { /* save oldest as return value */
453			sold[i]->next = freelist;
454			freelist = sold[i];
455			}
456	greg	1.8	}
457			return(sold[0]);
458			}
459
460
461			static char *scan_buf;
462
463
464			initscans() /* initialize scanline buffers */
465			{
466			int scansize;
467			register SCAN *ptr;
468			register int i;
469			/* initialize positions */
470			scanpos = (long )malloc(pysizsizeof(long));
471			if (scanpos == NULL)
472			memerr("scanline positions");
473			for (i = pysiz-1; i >= 0; i--)
474			scanpos[i] = -1L;
475			curpos = pysiz-1;
476			/* clear hash table */
477			for (i = 0; i < HSIZE; i++)
478			hashtab[i] = NULL;
479			/* allocate scanline buffers */
480			scansize = sizeof(SCAN) + pxsiz*sizeof(COLR);
481			#ifdef ALIGN
482			scansize = scansize+(sizeof(ALIGN)-1)) & ~(sizeof(ALIGN)-1);
483			#endif
484			i = MAXSBUF / scansize; /* compute number to allocate */
485			if (i > HSIZE)
486			i = HSIZE;
487			scan_buf = malloc(iscansize); / get in one big chunk */
488			if (scan_buf == NULL)
489			memerr("scanline buffers");
490			ptr = (SCAN *)scan_buf;
491			freelist = NULL; /* build our free list */
492			while (i-- > 0) {
493			ptr->y = -1;
494			ptr->lused = -1;
495			ptr->next = freelist;
496			freelist = ptr;
497			ptr = (SCAN )((char )ptr + scansize); /* beware of C bugs */
498			}
499			}
500
501
502			donescans() /* free up scanlines */
503			{
504			free(scan_buf);
505			free((char *)scanpos);
506	greg	1.1	}