11 |
|
#include "standard.h" |
12 |
|
#include "cmatrix.h" |
13 |
|
#include "platform.h" |
14 |
+ |
#include "resolu.h" |
15 |
|
|
16 |
+ |
const char *cm_fmt_id[] = { |
17 |
+ |
"unknown", "ascii", "float", "double", |
18 |
+ |
COLRFMT, CIEFMT |
19 |
+ |
}; |
20 |
+ |
|
21 |
+ |
const int cm_elem_size[] = { |
22 |
+ |
0, 0, 3*sizeof(float), 3*sizeof(double), 4, 4 |
23 |
+ |
}; |
24 |
+ |
|
25 |
|
/* Allocate a color coefficient matrix */ |
26 |
|
CMATRIX * |
27 |
|
cm_alloc(int nrows, int ncols) |
31 |
|
if ((nrows <= 0) | (ncols <= 0)) |
32 |
|
error(USER, "attempt to create empty matrix"); |
33 |
|
cm = (CMATRIX *)malloc(sizeof(CMATRIX) + |
34 |
< |
3*sizeof(COLORV)*(nrows*ncols - 1)); |
34 |
> |
sizeof(COLOR)*(nrows*ncols - 1)); |
35 |
|
if (cm == NULL) |
36 |
|
error(SYSTEM, "out of memory in cm_alloc()"); |
37 |
|
cm->nrows = nrows; |
50 |
|
return(NULL); |
51 |
|
} |
52 |
|
cm = (CMATRIX *)realloc(cm, sizeof(CMATRIX) + |
53 |
< |
3*sizeof(COLORV)*(nrows*cm->ncols - 1)); |
53 |
> |
sizeof(COLOR)*(nrows*cm->ncols - 1)); |
54 |
|
if (cm == NULL) |
55 |
|
error(SYSTEM, "out of memory in cm_resize()"); |
56 |
|
cm->nrows = nrows; |
61 |
|
getDT(char *s, void *p) |
62 |
|
{ |
63 |
|
char fmt[32]; |
64 |
+ |
int i; |
65 |
|
|
66 |
< |
if (formatval(fmt, s)) { |
67 |
< |
if (!strcmp(fmt, "ascii")) |
68 |
< |
*((int *)p) = DTascii; |
69 |
< |
else if (!strcmp(fmt, "float")) |
70 |
< |
*((int *)p) = DTfloat; |
60 |
< |
else if (!strcmp(fmt, "double")) |
61 |
< |
*((int *)p) = DTdouble; |
62 |
< |
else if (!strcmp(fmt, COLRFMT)) |
63 |
< |
*((int *)p) = DTrgbe; |
64 |
< |
else if (!strcmp(fmt, CIEFMT)) |
65 |
< |
*((int *)p) = DTxyze; |
66 |
< |
} |
66 |
> |
if (!formatval(fmt, s)) |
67 |
> |
return(0); |
68 |
> |
for (i = 1; i < DTend; i++) |
69 |
> |
if (!strcmp(fmt, cm_fmt_id[i])) |
70 |
> |
*((int *)p) = i; |
71 |
|
return(0); |
72 |
|
} |
73 |
|
|
169 |
|
break; |
170 |
|
} |
171 |
|
} else { /* read binary file */ |
172 |
< |
if (sizeof(COLORV) == (dtype==DTfloat ? sizeof(float) : |
169 |
< |
sizeof(double))) { |
172 |
> |
if (sizeof(COLOR) == cm_elem_size[dtype]) { |
173 |
|
int nread = 0; |
174 |
|
do { /* read all we can */ |
175 |
|
nread += fread(cm->cmem + 3*nread, |
176 |
< |
3*sizeof(COLORV), |
176 |
> |
sizeof(COLOR), |
177 |
|
cm->nrows*cm->ncols - nread, |
178 |
|
fp); |
179 |
|
if (nrows <= 0) { /* unknown length */ |
315 |
|
for (dr = 0; dr < cmr->nrows; dr++) |
316 |
|
for (dc = 0; dc < cmr->ncols; dc++) { |
317 |
|
COLORV *dp = cm_lval(cmr,dr,dc); |
318 |
+ |
double res[3]; |
319 |
|
dp[0] = dp[1] = dp[2] = 0; |
320 |
|
if (rowcheck != NULL && !rowcheck[dr]) |
321 |
|
continue; |
322 |
|
if (colcheck != NULL && !colcheck[dc]) |
323 |
|
continue; |
324 |
+ |
res[0] = res[1] = res[2] = 0; |
325 |
|
for (i = 0; i < cm1->ncols; i++) { |
326 |
|
const COLORV *cp1 = cm_lval(cm1,dr,i); |
327 |
|
const COLORV *cp2 = cm_lval(cm2,i,dc); |
328 |
< |
dp[0] += cp1[0] * cp2[0]; |
329 |
< |
dp[1] += cp1[1] * cp2[1]; |
330 |
< |
dp[2] += cp1[2] * cp2[2]; |
328 |
> |
res[0] += cp1[0] * cp2[0]; |
329 |
> |
res[1] += cp1[1] * cp2[1]; |
330 |
> |
res[2] += cp1[2] * cp2[2]; |
331 |
|
} |
332 |
+ |
copycolor(dp, res); |
333 |
|
} |
334 |
|
if (rowcheck != NULL) free(rowcheck); |
335 |
|
if (colcheck != NULL) free(colcheck); |
336 |
|
return(cmr); |
337 |
|
} |
338 |
|
|
339 |
< |
/* print out matrix as ASCII text -- no header */ |
340 |
< |
void |
341 |
< |
cm_print(const CMATRIX *cm, FILE *fp) |
339 |
> |
/* write out matrix to file (precede by resolution string if picture) */ |
340 |
> |
int |
341 |
> |
cm_write(const CMATRIX *cm, int dtype, FILE *fp) |
342 |
|
{ |
343 |
< |
int r, c; |
344 |
< |
const COLORV *mp = cm->cmem; |
345 |
< |
|
346 |
< |
for (r = 0; r < cm->nrows; r++) { |
347 |
< |
for (c = 0; c < cm->ncols; c++, mp += 3) |
348 |
< |
fprintf(fp, "\t%.6e %.6e %.6e", mp[0], mp[1], mp[2]); |
349 |
< |
fputc('\n', fp); |
343 |
> |
static const char tabEOL[2] = {'\t','\n'}; |
344 |
> |
const COLORV *mp = cm->cmem; |
345 |
> |
int r, c; |
346 |
> |
|
347 |
> |
switch (dtype) { |
348 |
> |
case DTascii: |
349 |
> |
for (r = 0; r < cm->nrows; r++) |
350 |
> |
for (c = 0; c < cm->ncols; c++, mp += 3) |
351 |
> |
fprintf(fp, "%.6e %.6e %.6e%c", |
352 |
> |
mp[0], mp[1], mp[2], |
353 |
> |
tabEOL[c >= cm->ncols-1]); |
354 |
> |
break; |
355 |
> |
case DTfloat: |
356 |
> |
case DTdouble: |
357 |
> |
if (sizeof(COLOR) == cm_elem_size[dtype]) { |
358 |
> |
r = cm->ncols*cm->nrows; |
359 |
> |
while (r > 0) { |
360 |
> |
c = fwrite(mp, sizeof(COLOR), r, fp); |
361 |
> |
if (c <= 0) |
362 |
> |
return(0); |
363 |
> |
mp += 3*c; |
364 |
> |
r -= c; |
365 |
> |
} |
366 |
> |
} else if (dtype == DTdouble) { |
367 |
> |
double dc[3]; |
368 |
> |
r = cm->ncols*cm->nrows; |
369 |
> |
while (r--) { |
370 |
> |
copycolor(dc, mp); |
371 |
> |
if (fwrite(dc, sizeof(double), 3, fp) != 3) |
372 |
> |
return(0); |
373 |
> |
mp += 3; |
374 |
> |
} |
375 |
> |
} else /* dtype == DTfloat */ { |
376 |
> |
float fc[3]; |
377 |
> |
r = cm->ncols*cm->nrows; |
378 |
> |
while (r--) { |
379 |
> |
copycolor(fc, mp); |
380 |
> |
if (fwrite(fc, sizeof(float), 3, fp) != 3) |
381 |
> |
return(0); |
382 |
> |
mp += 3; |
383 |
> |
} |
384 |
> |
} |
385 |
> |
break; |
386 |
> |
case DTrgbe: |
387 |
> |
case DTxyze: |
388 |
> |
fprtresolu(cm->ncols, cm->nrows, fp); |
389 |
> |
for (r = 0; r < cm->nrows; r++, mp += 3*cm->ncols) |
390 |
> |
if (fwritescan((COLOR *)mp, cm->ncols, fp) < 0) |
391 |
> |
return(0); |
392 |
> |
break; |
393 |
> |
default: |
394 |
> |
fputs("Unsupported data type in cm_write()!\n", stderr); |
395 |
> |
return(0); |
396 |
|
} |
397 |
+ |
return(fflush(fp) == 0); |
398 |
|
} |