ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/data.c
Revision: 2.19
Committed: Thu Jun 5 19:29:34 2003 UTC (20 years, 10 months ago) by schorsch
Content type: text/plain
Branch: MAIN
Changes since 2.18: +2 -7 lines
Log Message:
Macros for setting binary file mode. Replacing MSDOS by _WIN32.

File Contents

# User Rev Content
1 greg 1.1 #ifndef lint
2 schorsch 2.19 static const char RCSid[] = "$Id: data.c,v 2.18 2003/05/15 05:13:35 greg Exp $";
3 greg 1.1 #endif
4     /*
5     * data.c - routines dealing with interpolated data.
6     */
7    
8 greg 2.16 #include "copyright.h"
9 greg 2.15
10 greg 1.1 #include "standard.h"
11     #include "color.h"
12 greg 1.11 #include "resolu.h"
13 greg 1.1 #include "data.h"
14    
15 greg 2.7 /* picture memory usage before warning */
16     #ifndef PSIZWARN
17 greg 2.18 #ifdef SMLMEM
18     #define PSIZWARN 1500000
19     #else
20 gregl 2.13 #define PSIZWARN 5000000
21 greg 2.7 #endif
22     #endif
23 greg 1.1
24 greg 2.7 #ifndef TABSIZ
25 greg 2.6 #define TABSIZ 97 /* table size (prime) */
26 greg 2.7 #endif
27 greg 2.6
28     #define hash(s) (shash(s)%TABSIZ)
29    
30    
31     static DATARRAY *dtab[TABSIZ]; /* data array list */
32 greg 1.1
33    
34     DATARRAY *
35     getdata(dname) /* get data array dname */
36     char *dname;
37     {
38     char *dfname;
39     FILE *fp;
40     int asize;
41 greg 1.4 register int i, j;
42 greg 1.1 register DATARRAY *dp;
43     /* look for array in list */
44 greg 2.6 for (dp = dtab[hash(dname)]; dp != NULL; dp = dp->next)
45 greg 1.1 if (!strcmp(dname, dp->name))
46     return(dp); /* found! */
47     /*
48     * If we haven't loaded the data already, we will look
49 greg 2.5 * for it in the directories specified by the library path.
50 greg 1.1 *
51     * The file has the following format:
52     *
53 greg 1.4 * N
54 greg 1.1 * beg0 end0 n0
55     * beg1 end1 n1
56     * . . .
57 greg 1.4 * begN endN nN
58 greg 1.1 * data, later dimensions changing faster
59     * . . .
60     *
61 greg 1.4 * For irregularly spaced points, the following can be
62     * substituted for begi endi ni:
63     *
64 greg 1.6 * 0 0 ni p0i p1i .. pni
65 greg 1.1 */
66    
67 greg 2.17 if ((dfname = getpath(dname, getrlibpath(), R_OK)) == NULL) {
68 greg 1.1 sprintf(errmsg, "cannot find data file \"%s\"", dname);
69     error(USER, errmsg);
70     }
71     if ((fp = fopen(dfname, "r")) == NULL) {
72     sprintf(errmsg, "cannot open data file \"%s\"", dfname);
73     error(SYSTEM, errmsg);
74     }
75     /* get dimensions */
76 greg 2.15 if (fgetval(fp, 'i', (char *)&asize) <= 0)
77 greg 1.1 goto scanerr;
78 greg 2.12 if (asize <= 0 | asize > MAXDDIM) {
79 greg 1.1 sprintf(errmsg, "bad number of dimensions for \"%s\"", dname);
80     error(USER, errmsg);
81     }
82 greg 2.12 if ((dp = (DATARRAY *)malloc(sizeof(DATARRAY))) == NULL)
83     goto memerr;
84     dp->name = savestr(dname);
85     dp->type = DATATY;
86     dp->nd = asize;
87 greg 1.1 asize = 1;
88     for (i = 0; i < dp->nd; i++) {
89 greg 2.15 if (fgetval(fp, DATATY, (char *)&dp->dim[i].org) <= 0)
90 greg 1.6 goto scanerr;
91 greg 2.15 if (fgetval(fp, DATATY, (char *)&dp->dim[i].siz) <= 0)
92 greg 1.10 goto scanerr;
93 greg 2.15 if (fgetval(fp, 'i', (char *)&dp->dim[i].ne) <= 0)
94 greg 1.10 goto scanerr;
95 greg 1.6 if (dp->dim[i].ne < 2)
96     goto scanerr;
97     asize *= dp->dim[i].ne;
98     if ((dp->dim[i].siz -= dp->dim[i].org) == 0) {
99 greg 2.12 dp->dim[i].p = (DATATYPE *)
100     malloc(dp->dim[i].ne*sizeof(DATATYPE));
101 greg 1.4 if (dp->dim[i].p == NULL)
102     goto memerr;
103 greg 2.10 for (j = 0; j < dp->dim[i].ne; j++)
104 greg 2.15 if (fgetval(fp, DATATY,
105     (char *)&dp->dim[i].p[j]) <= 0)
106 greg 1.4 goto scanerr;
107     for (j = 1; j < dp->dim[i].ne-1; j++)
108     if ((dp->dim[i].p[j-1] < dp->dim[i].p[j]) !=
109     (dp->dim[i].p[j] < dp->dim[i].p[j+1]))
110     goto scanerr;
111     dp->dim[i].org = dp->dim[i].p[0];
112     dp->dim[i].siz = dp->dim[i].p[dp->dim[i].ne-1]
113     - dp->dim[i].p[0];
114     } else
115 greg 1.6 dp->dim[i].p = NULL;
116 greg 1.1 }
117 greg 2.12 if ((dp->arr.d = (DATATYPE *)malloc(asize*sizeof(DATATYPE))) == NULL)
118 greg 1.1 goto memerr;
119    
120 greg 2.10 for (i = 0; i < asize; i++)
121 greg 2.15 if (fgetval(fp, DATATY, (char *)&dp->arr.d[i]) <= 0)
122 greg 1.1 goto scanerr;
123     fclose(fp);
124 greg 2.6 i = hash(dname);
125     dp->next = dtab[i];
126     return(dtab[i] = dp);
127 greg 1.1
128     memerr:
129     error(SYSTEM, "out of memory in getdata");
130     scanerr:
131     sprintf(errmsg, "%s in data file \"%s\"",
132     feof(fp) ? "unexpected EOF" : "bad format", dfname);
133     error(USER, errmsg);
134     }
135    
136    
137 greg 2.15 static int
138 greg 2.4 headaspect(s, iap) /* check string for aspect ratio */
139 greg 1.5 char *s;
140 greg 2.4 double *iap;
141 greg 1.5 {
142 gwlarson 2.14 char fmt[32];
143    
144 greg 1.5 if (isaspect(s))
145 greg 2.4 *iap *= aspectval(s);
146 greg 2.15 else if (formatval(fmt, s) && !globmatch(PICFMT, fmt))
147 gwlarson 2.14 *iap = 0.0;
148     return(0);
149 greg 1.5 }
150    
151    
152 greg 1.1 DATARRAY *
153     getpict(pname) /* get picture pname */
154     char *pname;
155     {
156 greg 2.4 double inpaspect;
157 greg 1.1 char *pfname;
158     FILE *fp;
159 greg 2.12 COLR *scanin;
160 greg 1.11 int sl, ns;
161 greg 2.3 RESOLU inpres;
162 greg 1.11 FLOAT loc[2];
163     int y;
164     register int x, i;
165 greg 1.1 register DATARRAY *pp;
166     /* look for array in list */
167 greg 2.12 for (pp = dtab[hash(pname)]; pp != NULL; pp = pp->next)
168 greg 1.1 if (!strcmp(pname, pp->name))
169     return(pp); /* found! */
170    
171 greg 2.17 if ((pfname = getpath(pname, getrlibpath(), R_OK)) == NULL) {
172 greg 1.1 sprintf(errmsg, "cannot find picture file \"%s\"", pname);
173     error(USER, errmsg);
174     }
175 greg 2.12 if ((pp = (DATARRAY *)malloc(3*sizeof(DATARRAY))) == NULL)
176 greg 1.1 goto memerr;
177    
178 greg 2.12 pp[0].name = savestr(pname);
179 greg 1.1
180     if ((fp = fopen(pfname, "r")) == NULL) {
181     sprintf(errmsg, "cannot open picture file \"%s\"", pfname);
182     error(SYSTEM, errmsg);
183     }
184 schorsch 2.19 SET_FILE_BINARY(fp);
185 greg 1.1 /* get dimensions */
186 greg 1.5 inpaspect = 1.0;
187 greg 2.15 getheader(fp, headaspect, (char *)&inpaspect);
188 gwlarson 2.14 if (inpaspect <= FTINY || !fgetsresolu(&inpres, fp))
189 greg 1.1 goto readerr;
190 greg 2.12 pp[0].nd = 2;
191     pp[0].dim[0].ne = inpres.yr;
192     pp[0].dim[1].ne = inpres.xr;
193     pp[0].dim[0].org =
194     pp[0].dim[1].org = 0.0;
195     if (inpres.xr <= inpres.yr*inpaspect) {
196     pp[0].dim[0].siz = inpaspect *
197     (double)inpres.yr/inpres.xr;
198     pp[0].dim[1].siz = 1.0;
199     } else {
200     pp[0].dim[0].siz = 1.0;
201     pp[0].dim[1].siz = (double)inpres.xr/inpres.yr /
202     inpaspect;
203     }
204     pp[0].dim[0].p = pp[0].dim[1].p = NULL;
205     sl = scanlen(&inpres); /* allocate array */
206     ns = numscans(&inpres);
207     i = ns*sl*sizeof(COLR);
208 greg 2.7 #if PSIZWARN
209 greg 2.12 if (i > PSIZWARN) { /* memory warning */
210 greg 2.7 sprintf(errmsg, "picture file \"%s\" using %d bytes of memory",
211     pname, i);
212     error(WARNING, errmsg);
213     }
214     #endif
215 greg 2.12 if ((pp[0].arr.c = (COLR *)malloc(i)) == NULL)
216     goto memerr;
217 greg 1.1 /* load picture */
218 greg 2.12 if ((scanin = (COLR *)malloc(sl*sizeof(COLR))) == NULL)
219 greg 1.1 goto memerr;
220 greg 1.11 for (y = 0; y < ns; y++) {
221 greg 2.12 if (freadcolrs(scanin, sl, fp) < 0)
222 greg 1.1 goto readerr;
223 greg 1.11 for (x = 0; x < sl; x++) {
224     pix2loc(loc, &inpres, x, y);
225     i = (int)(loc[1]*inpres.yr)*inpres.xr +
226     (int)(loc[0]*inpres.xr);
227 greg 2.12 copycolr(pp[0].arr.c[i], scanin[x]);
228 greg 1.11 }
229 greg 1.1 }
230 greg 2.15 free((void *)scanin);
231 greg 1.1 fclose(fp);
232 greg 2.6 i = hash(pname);
233 greg 2.12 pp[0].next = dtab[i]; /* link into picture list */
234     copystruct(&pp[1], &pp[0]);
235     copystruct(&pp[2], &pp[0]);
236     pp[0].type = RED; /* differentiate RGB records */
237     pp[1].type = GRN;
238     pp[2].type = BLU;
239     return(dtab[i] = pp);
240 greg 1.1
241     memerr:
242     error(SYSTEM, "out of memory in getpict");
243     readerr:
244     sprintf(errmsg, "bad picture file \"%s\"", pfname);
245     error(USER, errmsg);
246     }
247    
248    
249 greg 2.15 void
250     freedata(dta) /* release data array reference */
251     DATARRAY *dta;
252 greg 1.1 {
253 greg 2.5 DATARRAY head;
254 greg 2.6 int hval, nents;
255 greg 2.15 register DATARRAY *dpl, *dp;
256 greg 1.4 register int i;
257 greg 1.1
258 greg 2.15 if (dta == NULL) { /* free all if NULL */
259 greg 2.6 hval = 0; nents = TABSIZ;
260     } else {
261 greg 2.15 hval = hash(dta->name); nents = 1;
262 greg 2.6 }
263     while (nents--) {
264     head.next = dtab[hval];
265     dpl = &head;
266     while ((dp = dpl->next) != NULL)
267 greg 2.15 if ((dta == NULL | dta == dp)) {
268 greg 2.6 dpl->next = dp->next;
269 greg 2.12 if (dp->type == DATATY)
270 greg 2.15 free((void *)dp->arr.d);
271 greg 2.12 else
272 greg 2.15 free((void *)dp->arr.c);
273 greg 2.6 for (i = 0; i < dp->nd; i++)
274     if (dp->dim[i].p != NULL)
275 greg 2.15 free((void *)dp->dim[i].p);
276 greg 2.6 freestr(dp->name);
277 greg 2.15 free((void *)dp);
278 greg 2.6 } else
279     dpl = dp;
280     dtab[hval++] = head.next;
281     }
282 greg 1.1 }
283    
284    
285     double
286     datavalue(dp, pt) /* interpolate data value at a point */
287     register DATARRAY *dp;
288 greg 2.3 double *pt;
289 greg 1.1 {
290     DATARRAY sd;
291     int asize;
292 greg 1.6 int lower, upper;
293 greg 1.1 register int i;
294 greg 2.9 double x, y0, y1;
295 greg 1.4 /* set up dimensions for recursion */
296 greg 2.12 if (dp->nd > 1) {
297     sd.name = dp->name;
298     sd.type = dp->type;
299     sd.nd = dp->nd - 1;
300     asize = 1;
301     for (i = 0; i < sd.nd; i++) {
302     sd.dim[i].org = dp->dim[i+1].org;
303     sd.dim[i].siz = dp->dim[i+1].siz;
304     sd.dim[i].p = dp->dim[i+1].p;
305     asize *= sd.dim[i].ne = dp->dim[i+1].ne;
306     }
307 greg 1.1 }
308 greg 1.4 /* get independent variable */
309     if (dp->dim[0].p == NULL) { /* evenly spaced points */
310     x = (pt[0] - dp->dim[0].org)/dp->dim[0].siz;
311 greg 2.9 x *= (double)(dp->dim[0].ne - 1);
312 greg 1.4 i = x;
313     if (i < 0)
314     i = 0;
315     else if (i > dp->dim[0].ne - 2)
316     i = dp->dim[0].ne - 2;
317     } else { /* unevenly spaced points */
318 greg 1.6 if (dp->dim[0].siz > 0.0) {
319     lower = 0;
320     upper = dp->dim[0].ne;
321     } else {
322     lower = dp->dim[0].ne;
323     upper = 0;
324     }
325     do {
326     i = (lower + upper) >> 1;
327     if (pt[0] >= dp->dim[0].p[i])
328     lower = i;
329 greg 1.8 else
330 greg 1.6 upper = i;
331     } while (i != (lower + upper) >> 1);
332 greg 1.8 if (i > dp->dim[0].ne - 2)
333 greg 1.4 i = dp->dim[0].ne - 2;
334     x = i + (pt[0] - dp->dim[0].p[i]) /
335     (dp->dim[0].p[i+1] - dp->dim[0].p[i]);
336     }
337     /* get dependent variable */
338 greg 2.12 if (dp->nd > 1) {
339     if (dp->type == DATATY) {
340     sd.arr.d = dp->arr.d + i*asize;
341     y0 = datavalue(&sd, pt+1);
342     sd.arr.d = dp->arr.d + (i+1)*asize;
343     y1 = datavalue(&sd, pt+1);
344     } else {
345     sd.arr.c = dp->arr.c + i*asize;
346     y0 = datavalue(&sd, pt+1);
347     sd.arr.c = dp->arr.c + (i+1)*asize;
348     y1 = datavalue(&sd, pt+1);
349     }
350 greg 1.1 } else {
351 greg 2.12 if (dp->type == DATATY) {
352     y0 = dp->arr.d[i];
353     y1 = dp->arr.d[i+1];
354     } else {
355     y0 = colrval(dp->arr.c[i],dp->type);
356     y1 = colrval(dp->arr.c[i+1],dp->type);
357     }
358 greg 1.1 }
359     /*
360     * Extrapolate as far as one division, then
361     * taper off harmonically to zero.
362     */
363     if (x > i+2)
364 greg 2.9 return( (2*y1-y0)/(x-(i-1)) );
365 greg 1.1
366 greg 2.9 if (x < i-1)
367     return( (2*y0-y1)/(i-x) );
368    
369     return( y0*((i+1)-x) + y1*(x-i) );
370 greg 1.1 }