ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rmatrix.c
Revision: 2.40
Committed: Wed Mar 25 01:51:09 2020 UTC (4 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.39: +21 -2 lines
Log Message:
Changed rmtxop and dctimestep to undo any exposure on Radiance input pictures

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.40 static const char RCSid[] = "$Id: rmatrix.c,v 2.39 2019/12/28 18:05:14 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * General matrix operations.
6     */
7    
8     #include <stdlib.h>
9 greg 2.25 #include <errno.h>
10 greg 2.21 #include "rtio.h"
11 schorsch 2.20 #include "platform.h"
12 greg 2.1 #include "resolu.h"
13 greg 2.19 #include "paths.h"
14 greg 2.1 #include "rmatrix.h"
15    
16 greg 2.6 static char rmx_mismatch_warn[] = "WARNING: data type mismatch\n";
17 greg 2.1
18     /* Allocate a nr x nc matrix with n components */
19     RMATRIX *
20     rmx_alloc(int nr, int nc, int n)
21     {
22     RMATRIX *dnew;
23    
24     if ((nr <= 0) | (nc <= 0) | (n <= 0))
25     return(NULL);
26     dnew = (RMATRIX *)malloc(sizeof(RMATRIX)-sizeof(dnew->mtx) +
27     sizeof(dnew->mtx[0])*(n*nr*nc));
28 greg 2.33 if (!dnew)
29 greg 2.1 return(NULL);
30     dnew->nrows = nr; dnew->ncols = nc; dnew->ncomp = n;
31 greg 2.6 dnew->dtype = DTdouble;
32 greg 2.36 dnew->swapin = 0;
33 greg 2.5 dnew->info = NULL;
34 greg 2.1 return(dnew);
35     }
36    
37 greg 2.6 /* Free a RMATRIX array */
38     void
39     rmx_free(RMATRIX *rm)
40     {
41     if (!rm) return;
42     if (rm->info)
43     free(rm->info);
44     free(rm);
45     }
46    
47     /* Resolve data type based on two input types (returns 0 for mismatch) */
48     int
49     rmx_newtype(int dtyp1, int dtyp2)
50     {
51 greg 2.14 if ((dtyp1==DTxyze) | (dtyp1==DTrgbe) |
52     (dtyp2==DTxyze) | (dtyp2==DTrgbe)
53     && dtyp1 != dtyp2)
54 greg 2.6 return(0);
55     if (dtyp1 < dtyp2)
56     return(dtyp1);
57     return(dtyp2);
58     }
59    
60 greg 2.5 /* Append header information associated with matrix data */
61     int
62     rmx_addinfo(RMATRIX *rm, const char *info)
63     {
64 greg 2.33 if (!rm || !info || !*info)
65 greg 2.5 return(0);
66 greg 2.8 if (!rm->info) {
67 greg 2.5 rm->info = (char *)malloc(strlen(info)+1);
68 greg 2.8 if (rm->info) rm->info[0] = '\0';
69     } else
70 greg 2.5 rm->info = (char *)realloc(rm->info,
71     strlen(rm->info)+strlen(info)+1);
72     if (!rm->info)
73     return(0);
74     strcat(rm->info, info);
75     return(1);
76     }
77    
78 greg 2.1 static int
79     get_dminfo(char *s, void *p)
80     {
81 greg 2.6 RMATRIX *ip = (RMATRIX *)p;
82 greg 2.29 char fmt[MAXFMTLEN];
83 greg 2.1 int i;
84    
85 greg 2.6 if (headidval(fmt, s))
86     return(0);
87 greg 2.1 if (!strncmp(s, "NCOMP=", 6)) {
88     ip->ncomp = atoi(s+6);
89     return(0);
90     }
91     if (!strncmp(s, "NROWS=", 6)) {
92     ip->nrows = atoi(s+6);
93     return(0);
94     }
95     if (!strncmp(s, "NCOLS=", 6)) {
96     ip->ncols = atoi(s+6);
97     return(0);
98     }
99 greg 2.35 if ((i = isbigendian(s)) >= 0) {
100     ip->swapin = (nativebigendian() != i);
101     return(0);
102     }
103 greg 2.40 if (isexpos(s)) {
104     double d = exposval(s);
105     scalecolor(ip->mtx, d);
106     return(0);
107     }
108     if (iscolcor(s)) {
109     COLOR ctmp;
110     colcorval(ctmp, s);
111     multcolor(ip->mtx, ctmp);
112     return(0);
113     }
114 greg 2.5 if (!formatval(fmt, s)) {
115 greg 2.6 rmx_addinfo(ip, s);
116 greg 2.1 return(0);
117 greg 2.5 }
118 greg 2.1 for (i = 1; i < DTend; i++)
119     if (!strcmp(fmt, cm_fmt_id[i])) {
120     ip->dtype = i;
121     return(0);
122     }
123     return(-1);
124     }
125    
126     static int
127     rmx_load_ascii(RMATRIX *rm, FILE *fp)
128     {
129     int i, j, k;
130 greg 2.17
131 greg 2.1 for (i = 0; i < rm->nrows; i++)
132     for (j = 0; j < rm->ncols; j++)
133     for (k = 0; k < rm->ncomp; k++)
134     if (fscanf(fp, "%lf", &rmx_lval(rm,i,j,k)) != 1)
135     return(0);
136     return(1);
137     }
138    
139     static int
140     rmx_load_float(RMATRIX *rm, FILE *fp)
141     {
142     int i, j, k;
143     float val[100];
144    
145     if (rm->ncomp > 100) {
146     fputs("Unsupported # components in rmx_load_float()\n", stderr);
147     exit(1);
148     }
149     for (i = 0; i < rm->nrows; i++)
150     for (j = 0; j < rm->ncols; j++) {
151 greg 2.21 if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp)
152 greg 2.1 return(0);
153 greg 2.35 if (rm->swapin)
154     swap32((char *)val, rm->ncomp);
155 greg 2.1 for (k = rm->ncomp; k--; )
156     rmx_lval(rm,i,j,k) = val[k];
157     }
158     return(1);
159     }
160    
161     static int
162     rmx_load_double(RMATRIX *rm, FILE *fp)
163     {
164 greg 2.26 int i, j;
165 greg 2.1
166     for (i = 0; i < rm->nrows; i++)
167 greg 2.35 for (j = 0; j < rm->ncols; j++) {
168 greg 2.26 if (getbinary(&rmx_lval(rm,i,j,0), sizeof(double), rm->ncomp, fp) != rm->ncomp)
169 greg 2.1 return(0);
170 greg 2.35 if (rm->swapin)
171     swap64((char *)&rmx_lval(rm,i,j,0), rm->ncomp);
172     }
173 greg 2.1 return(1);
174     }
175    
176     static int
177     rmx_load_rgbe(RMATRIX *rm, FILE *fp)
178     {
179     COLOR *scan = (COLOR *)malloc(sizeof(COLOR)*rm->ncols);
180     int i, j;
181    
182 greg 2.33 if (!scan)
183 greg 2.1 return(0);
184     for (i = 0; i < rm->nrows; i++) {
185     if (freadscan(scan, rm->ncols, fp) < 0) {
186     free(scan);
187     return(0);
188     }
189     for (j = rm->ncols; j--; ) {
190     rmx_lval(rm,i,j,0) = colval(scan[j],RED);
191     rmx_lval(rm,i,j,1) = colval(scan[j],GRN);
192     rmx_lval(rm,i,j,2) = colval(scan[j],BLU);
193     }
194     }
195     free(scan);
196     return(1);
197     }
198    
199     /* Load matrix from supported file type */
200     RMATRIX *
201 greg 2.13 rmx_load(const char *inspec)
202 greg 2.1 {
203     FILE *fp = stdin;
204 greg 2.6 RMATRIX dinfo;
205 greg 2.1 RMATRIX *dnew;
206    
207 greg 2.33 if (!inspec) { /* reading from stdin? */
208 greg 2.13 inspec = "<stdin>";
209 greg 2.22 SET_FILE_BINARY(stdin);
210 greg 2.13 } else if (inspec[0] == '!') {
211 greg 2.33 if (!(fp = popen(inspec+1, "r")))
212 greg 2.13 return(NULL);
213 greg 2.37 SET_FILE_BINARY(fp);
214 greg 2.1 } else {
215 greg 2.13 const char *sp = inspec; /* check suffix */
216 greg 2.1 while (*sp)
217     ++sp;
218 greg 2.13 while (sp > inspec && sp[-1] != '.')
219 greg 2.1 --sp;
220     if (!strcasecmp(sp, "XML")) { /* assume it's a BSDF */
221 greg 2.13 CMATRIX *cm = cm_loadBTDF((char *)inspec);
222 greg 2.33 if (!cm)
223 greg 2.1 return(NULL);
224     dnew = rmx_from_cmatrix(cm);
225     cm_free(cm);
226 greg 2.18 dnew->dtype = DTascii;
227 greg 2.1 return(dnew);
228     }
229     /* else open it ourselves */
230 greg 2.33 if (!(fp = fopen(inspec, "rb")))
231 greg 2.1 return(NULL);
232     }
233     #ifdef getc_unlocked
234     flockfile(fp);
235     #endif
236     dinfo.nrows = dinfo.ncols = dinfo.ncomp = 0;
237 greg 2.6 dinfo.dtype = DTascii; /* assumed w/o FORMAT */
238 greg 2.35 dinfo.swapin = 0;
239 greg 2.6 dinfo.info = NULL;
240 greg 2.40 dinfo.mtx[0] = dinfo.mtx[1] = dinfo.mtx[2] = 1.;
241 greg 2.3 if (getheader(fp, get_dminfo, &dinfo) < 0) {
242 greg 2.1 fclose(fp);
243     return(NULL);
244     }
245 greg 2.4 if ((dinfo.nrows <= 0) | (dinfo.ncols <= 0)) {
246 greg 2.1 if (!fscnresolu(&dinfo.ncols, &dinfo.nrows, fp)) {
247     fclose(fp);
248     return(NULL);
249     }
250 greg 2.4 if (dinfo.ncomp <= 0)
251     dinfo.ncomp = 3;
252     else if ((dinfo.dtype == DTrgbe) | (dinfo.dtype == DTxyze) &&
253     dinfo.ncomp != 3) {
254     fclose(fp);
255     return(NULL);
256     }
257 greg 2.1 }
258     dnew = rmx_alloc(dinfo.nrows, dinfo.ncols, dinfo.ncomp);
259 greg 2.33 if (!dnew) {
260 greg 2.1 fclose(fp);
261     return(NULL);
262     }
263 greg 2.6 dnew->info = dinfo.info;
264 greg 2.1 switch (dinfo.dtype) {
265     case DTascii:
266 greg 2.37 SET_FILE_TEXT(fp);
267 greg 2.1 if (!rmx_load_ascii(dnew, fp))
268     goto loaderr;
269 greg 2.11 dnew->dtype = DTascii; /* should leave double? */
270 greg 2.1 break;
271     case DTfloat:
272 greg 2.35 dnew->swapin = dinfo.swapin;
273 greg 2.1 if (!rmx_load_float(dnew, fp))
274     goto loaderr;
275 greg 2.6 dnew->dtype = DTfloat;
276 greg 2.1 break;
277     case DTdouble:
278 greg 2.35 dnew->swapin = dinfo.swapin;
279 greg 2.1 if (!rmx_load_double(dnew, fp))
280     goto loaderr;
281 greg 2.6 dnew->dtype = DTdouble;
282 greg 2.1 break;
283     case DTrgbe:
284     case DTxyze:
285     if (!rmx_load_rgbe(dnew, fp))
286     goto loaderr;
287 greg 2.40 dnew->dtype = dinfo.dtype; /* undo exposure? */
288     if ((dinfo.mtx[0] != 1.) | (dinfo.mtx[1] != 1.) |
289     (dinfo.mtx[2] != 1.)) {
290     dinfo.mtx[0] = 1./dinfo.mtx[0];
291     dinfo.mtx[1] = 1./dinfo.mtx[1];
292     dinfo.mtx[2] = 1./dinfo.mtx[2];
293     rmx_scale(dnew, dinfo.mtx);
294     }
295 greg 2.1 break;
296     default:
297     goto loaderr;
298     }
299 greg 2.13 if (fp != stdin) {
300     if (inspec[0] == '!')
301     pclose(fp);
302     else
303     fclose(fp);
304     }
305     #ifdef getc_unlocked
306     else
307     funlockfile(fp);
308     #endif
309 greg 2.1 return(dnew);
310     loaderr: /* should report error? */
311 greg 2.13 if (inspec[0] == '!')
312     pclose(fp);
313     else
314     fclose(fp);
315 greg 2.1 rmx_free(dnew);
316     return(NULL);
317     }
318    
319     static int
320     rmx_write_ascii(const RMATRIX *rm, FILE *fp)
321     {
322 greg 2.30 const char *fmt = (rm->dtype == DTfloat) ? " %.7e" :
323     (rm->dtype == DTrgbe) | (rm->dtype == DTxyze) ? " %.3e" :
324     " %.15e" ;
325 greg 2.1 int i, j, k;
326 greg 2.17
327 greg 2.1 for (i = 0; i < rm->nrows; i++) {
328     for (j = 0; j < rm->ncols; j++) {
329     for (k = 0; k < rm->ncomp; k++)
330 greg 2.30 fprintf(fp, fmt, rmx_lval(rm,i,j,k));
331 greg 2.1 fputc('\t', fp);
332     }
333     fputc('\n', fp);
334     }
335     return(1);
336     }
337    
338     static int
339     rmx_write_float(const RMATRIX *rm, FILE *fp)
340     {
341     int i, j, k;
342     float val[100];
343    
344     if (rm->ncomp > 100) {
345     fputs("Unsupported # components in rmx_write_float()\n", stderr);
346     exit(1);
347     }
348     for (i = 0; i < rm->nrows; i++)
349     for (j = 0; j < rm->ncols; j++) {
350     for (k = rm->ncomp; k--; )
351     val[k] = (float)rmx_lval(rm,i,j,k);
352 greg 2.21 if (putbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp)
353 greg 2.1 return(0);
354     }
355     return(1);
356     }
357    
358     static int
359     rmx_write_double(const RMATRIX *rm, FILE *fp)
360     {
361 greg 2.26 int i, j;
362 greg 2.1
363     for (i = 0; i < rm->nrows; i++)
364 greg 2.26 for (j = 0; j < rm->ncols; j++)
365     if (putbinary(&rmx_lval(rm,i,j,0), sizeof(double), rm->ncomp, fp) != rm->ncomp)
366 greg 2.1 return(0);
367     return(1);
368     }
369    
370     static int
371     rmx_write_rgbe(const RMATRIX *rm, FILE *fp)
372     {
373 greg 2.27 COLR *scan = (COLR *)malloc(sizeof(COLR)*rm->ncols);
374 greg 2.1 int i, j;
375    
376 greg 2.33 if (!scan)
377 greg 2.1 return(0);
378     for (i = 0; i < rm->nrows; i++) {
379     for (j = rm->ncols; j--; )
380 greg 2.27 setcolr(scan[j], rmx_lval(rm,i,j,0),
381 greg 2.1 rmx_lval(rm,i,j,1),
382     rmx_lval(rm,i,j,2) );
383 greg 2.27 if (fwritecolrs(scan, rm->ncols, fp) < 0) {
384 greg 2.1 free(scan);
385     return(0);
386     }
387     }
388     free(scan);
389     return(1);
390     }
391    
392 greg 2.6 /* Write matrix to file type indicated by dtype */
393 greg 2.10 int
394 greg 2.1 rmx_write(const RMATRIX *rm, int dtype, FILE *fp)
395     {
396     RMATRIX *mydm = NULL;
397     int ok = 1;
398    
399 greg 2.33 if (!rm | !fp)
400 greg 2.1 return(0);
401 greg 2.17 #ifdef getc_unlocked
402     flockfile(fp);
403     #endif
404 greg 2.1 /* complete header */
405 greg 2.5 if (rm->info)
406     fputs(rm->info, fp);
407 greg 2.6 if (dtype == DTfromHeader)
408     dtype = rm->dtype;
409     else if ((dtype == DTrgbe) & (rm->dtype == DTxyze))
410     dtype = DTxyze;
411 greg 2.9 else if ((dtype == DTxyze) & (rm->dtype == DTrgbe))
412 greg 2.6 dtype = DTrgbe;
413 greg 2.1 if ((dtype != DTrgbe) & (dtype != DTxyze)) {
414     fprintf(fp, "NROWS=%d\n", rm->nrows);
415     fprintf(fp, "NCOLS=%d\n", rm->ncols);
416     fprintf(fp, "NCOMP=%d\n", rm->ncomp);
417     } else if (rm->ncomp != 3) { /* wrong # components? */
418     double cmtx[3];
419     if (rm->ncomp != 1) /* only convert grayscale */
420     return(0);
421     cmtx[0] = cmtx[1] = cmtx[2] = 1;
422     mydm = rmx_transform(rm, 3, cmtx);
423 greg 2.33 if (!mydm)
424 greg 2.1 return(0);
425     rm = mydm;
426     }
427 greg 2.35 if ((dtype == DTfloat) | (dtype == DTdouble))
428     fputendian(fp); /* important to record */
429 greg 2.1 fputformat((char *)cm_fmt_id[dtype], fp);
430     fputc('\n', fp);
431     switch (dtype) { /* write data */
432     case DTascii:
433     ok = rmx_write_ascii(rm, fp);
434     break;
435     case DTfloat:
436     ok = rmx_write_float(rm, fp);
437     break;
438     case DTdouble:
439     ok = rmx_write_double(rm, fp);
440     break;
441     case DTrgbe:
442     case DTxyze:
443     fprtresolu(rm->ncols, rm->nrows, fp);
444     ok = rmx_write_rgbe(rm, fp);
445     break;
446     default:
447     return(0);
448     }
449     ok &= (fflush(fp) == 0);
450 greg 2.17 #ifdef getc_unlocked
451     funlockfile(fp);
452     #endif
453 greg 2.33 if (mydm)
454     rmx_free(mydm);
455 greg 2.10 return(ok);
456 greg 2.1 }
457    
458     /* Allocate and assign square identity matrix with n components */
459     RMATRIX *
460     rmx_identity(const int dim, const int n)
461     {
462     RMATRIX *rid = rmx_alloc(dim, dim, n);
463 greg 2.12 int i, k;
464 greg 2.1
465 greg 2.33 if (!rid)
466 greg 2.1 return(NULL);
467 greg 2.12 memset(rid->mtx, 0, sizeof(rid->mtx[0])*n*dim*dim);
468 greg 2.1 for (i = dim; i--; )
469 greg 2.12 for (k = n; k--; )
470     rmx_lval(rid,i,i,k) = 1;
471 greg 2.1 return(rid);
472     }
473    
474     /* Duplicate the given matrix */
475     RMATRIX *
476     rmx_copy(const RMATRIX *rm)
477     {
478     RMATRIX *dnew;
479    
480 greg 2.33 if (!rm)
481 greg 2.1 return(NULL);
482     dnew = rmx_alloc(rm->nrows, rm->ncols, rm->ncomp);
483 greg 2.33 if (!dnew)
484 greg 2.1 return(NULL);
485 greg 2.5 rmx_addinfo(dnew, rm->info);
486 greg 2.6 dnew->dtype = rm->dtype;
487 greg 2.1 memcpy(dnew->mtx, rm->mtx,
488     sizeof(rm->mtx[0])*rm->ncomp*rm->nrows*rm->ncols);
489     return(dnew);
490     }
491    
492 greg 2.2 /* Allocate and assign transposed matrix */
493     RMATRIX *
494     rmx_transpose(const RMATRIX *rm)
495 greg 2.1 {
496 greg 2.2 RMATRIX *dnew;
497 greg 2.1 int i, j, k;
498    
499 greg 2.33 if (!rm)
500 greg 2.1 return(0);
501 greg 2.31 if ((rm->nrows == 1) | (rm->ncols == 1)) {
502     dnew = rmx_copy(rm);
503 greg 2.33 if (!dnew)
504 greg 2.32 return(NULL);
505 greg 2.31 dnew->nrows = rm->ncols;
506     dnew->ncols = rm->nrows;
507     return(dnew);
508     }
509 greg 2.2 dnew = rmx_alloc(rm->ncols, rm->nrows, rm->ncomp);
510 greg 2.33 if (!dnew)
511 greg 2.2 return(NULL);
512 greg 2.5 if (rm->info) {
513     rmx_addinfo(dnew, rm->info);
514     rmx_addinfo(dnew, "Transposed rows and columns\n");
515     }
516 greg 2.6 dnew->dtype = rm->dtype;
517 greg 2.2 for (i = dnew->nrows; i--; )
518     for (j = dnew->ncols; j--; )
519     for (k = dnew->ncomp; k--; )
520     rmx_lval(dnew,i,j,k) = rmx_lval(rm,j,i,k);
521     return(dnew);
522 greg 2.1 }
523    
524     /* Multiply (concatenate) two matrices and allocate the result */
525     RMATRIX *
526     rmx_multiply(const RMATRIX *m1, const RMATRIX *m2)
527     {
528     RMATRIX *mres;
529     int i, j, k, h;
530    
531 greg 2.33 if (!m1 | !m2 || (m1->ncomp != m2->ncomp) | (m1->ncols != m2->nrows))
532 greg 2.1 return(NULL);
533     mres = rmx_alloc(m1->nrows, m2->ncols, m1->ncomp);
534 greg 2.33 if (!mres)
535 greg 2.1 return(NULL);
536 greg 2.6 i = rmx_newtype(m1->dtype, m2->dtype);
537     if (i)
538     mres->dtype = i;
539     else
540     rmx_addinfo(mres, rmx_mismatch_warn);
541 greg 2.1 for (i = mres->nrows; i--; )
542     for (j = mres->ncols; j--; )
543 greg 2.8 for (k = mres->ncomp; k--; ) {
544 greg 2.1 long double d = 0;
545 greg 2.8 for (h = m1->ncols; h--; )
546 greg 2.24 d += rmx_lval(m1,i,h,k) * rmx_lval(m2,h,j,k);
547 greg 2.1 rmx_lval(mres,i,j,k) = (double)d;
548     }
549     return(mres);
550     }
551    
552 greg 2.25 /* Element-wise multiplication (or division) of m2 into m1 */
553     int
554     rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide)
555     {
556     int zeroDivides = 0;
557     int i, j, k;
558    
559 greg 2.33 if (!m1 | !m2 || (m1->ncols != m2->ncols) | (m1->nrows != m2->nrows))
560 greg 2.25 return(0);
561     if ((m2->ncomp > 1) & (m2->ncomp != m1->ncomp))
562     return(0);
563     i = rmx_newtype(m1->dtype, m2->dtype);
564     if (i)
565     m1->dtype = i;
566     else
567     rmx_addinfo(m1, rmx_mismatch_warn);
568     for (i = m1->nrows; i--; )
569     for (j = m1->ncols; j--; )
570     if (divide) {
571     double d;
572     if (m2->ncomp == 1) {
573     d = rmx_lval(m2,i,j,0);
574     if (d == 0) {
575     ++zeroDivides;
576     for (k = m1->ncomp; k--; )
577     rmx_lval(m1,i,j,k) = 0;
578     } else {
579     d = 1./d;
580     for (k = m1->ncomp; k--; )
581     rmx_lval(m1,i,j,k) *= d;
582     }
583     } else
584     for (k = m1->ncomp; k--; ) {
585     d = rmx_lval(m2,i,j,k);
586     if (d == 0) {
587     ++zeroDivides;
588     rmx_lval(m1,i,j,k) = 0;
589     } else
590     rmx_lval(m1,i,j,k) /= d;
591     }
592     } else {
593     if (m2->ncomp == 1) {
594     const double d = rmx_lval(m2,i,j,0);
595     for (k = m1->ncomp; k--; )
596     rmx_lval(m1,i,j,k) *= d;
597     } else
598     for (k = m1->ncomp; k--; )
599     rmx_lval(m1,i,j,k) *= rmx_lval(m2,i,j,k);
600     }
601     if (zeroDivides) {
602 greg 2.34 rmx_addinfo(m1, "WARNING: zero divide(s) corrupted results\n");
603 greg 2.25 errno = ERANGE;
604     }
605     return(1);
606     }
607    
608 greg 2.1 /* Sum second matrix into first, applying scale factor beforehand */
609     int
610     rmx_sum(RMATRIX *msum, const RMATRIX *madd, const double sf[])
611     {
612     double *mysf = NULL;
613     int i, j, k;
614    
615 greg 2.33 if (!msum | !madd ||
616 greg 2.1 (msum->nrows != madd->nrows) |
617     (msum->ncols != madd->ncols) |
618     (msum->ncomp != madd->ncomp))
619     return(0);
620 greg 2.33 if (!sf) {
621 greg 2.1 mysf = (double *)malloc(sizeof(double)*msum->ncomp);
622 greg 2.33 if (!mysf)
623 greg 2.1 return(0);
624     for (k = msum->ncomp; k--; )
625     mysf[k] = 1;
626     sf = mysf;
627     }
628 greg 2.6 i = rmx_newtype(msum->dtype, madd->dtype);
629     if (i)
630     msum->dtype = i;
631     else
632     rmx_addinfo(msum, rmx_mismatch_warn);
633 greg 2.1 for (i = msum->nrows; i--; )
634     for (j = msum->ncols; j--; )
635     for (k = msum->ncomp; k--; )
636     rmx_lval(msum,i,j,k) += sf[k] * rmx_lval(madd,i,j,k);
637 greg 2.33 if (mysf)
638     free(mysf);
639 greg 2.1 return(1);
640     }
641    
642     /* Scale the given matrix by the indicated scalar component vector */
643     int
644     rmx_scale(RMATRIX *rm, const double sf[])
645     {
646     int i, j, k;
647    
648 greg 2.33 if (!rm | !sf)
649 greg 2.1 return(0);
650     for (i = rm->nrows; i--; )
651     for (j = rm->ncols; j--; )
652     for (k = rm->ncomp; k--; )
653     rmx_lval(rm,i,j,k) *= sf[k];
654    
655 greg 2.28 if (rm->info)
656     rmx_addinfo(rm, "Applied scalar\n");
657 greg 2.1 return(1);
658     }
659    
660     /* Allocate new matrix and apply component transformation */
661     RMATRIX *
662     rmx_transform(const RMATRIX *msrc, int n, const double cmat[])
663     {
664     int i, j, ks, kd;
665     RMATRIX *dnew;
666    
667 greg 2.33 if (!msrc | (n <= 0) | !cmat)
668 greg 2.1 return(NULL);
669     dnew = rmx_alloc(msrc->nrows, msrc->ncols, n);
670 greg 2.33 if (!dnew)
671 greg 2.1 return(NULL);
672 greg 2.28 if (msrc->info) {
673     char buf[128];
674 greg 2.38 sprintf(buf, "Applied %dx%d component transform\n",
675 greg 2.28 dnew->ncomp, msrc->ncomp);
676     rmx_addinfo(dnew, msrc->info);
677     rmx_addinfo(dnew, buf);
678     }
679 greg 2.6 dnew->dtype = msrc->dtype;
680 greg 2.1 for (i = dnew->nrows; i--; )
681     for (j = dnew->ncols; j--; )
682     for (kd = dnew->ncomp; kd--; ) {
683     double d = 0;
684     for (ks = msrc->ncomp; ks--; )
685     d += cmat[kd*msrc->ncomp + ks] * rmx_lval(msrc,i,j,ks);
686     rmx_lval(dnew,i,j,kd) = d;
687     }
688     return(dnew);
689     }
690    
691     /* Convert a color matrix to newly allocated RMATRIX buffer */
692     RMATRIX *
693     rmx_from_cmatrix(const CMATRIX *cm)
694     {
695     int i, j;
696     RMATRIX *dnew;
697    
698 greg 2.33 if (!cm)
699 greg 2.1 return(NULL);
700     dnew = rmx_alloc(cm->nrows, cm->ncols, 3);
701 greg 2.33 if (!dnew)
702 greg 2.1 return(NULL);
703 greg 2.6 dnew->dtype = DTfloat;
704 greg 2.1 for (i = dnew->nrows; i--; )
705     for (j = dnew->ncols; j--; ) {
706     const COLORV *cv = cm_lval(cm,i,j);
707     rmx_lval(dnew,i,j,0) = cv[0];
708     rmx_lval(dnew,i,j,1) = cv[1];
709     rmx_lval(dnew,i,j,2) = cv[2];
710     }
711     return(dnew);
712     }
713    
714     /* Convert general matrix to newly allocated CMATRIX buffer */
715     CMATRIX *
716     cm_from_rmatrix(const RMATRIX *rm)
717     {
718     int i, j;
719     CMATRIX *cnew;
720    
721 greg 2.33 if (!rm || rm->ncomp != 3)
722 greg 2.1 return(NULL);
723     cnew = cm_alloc(rm->nrows, rm->ncols);
724 greg 2.33 if (!cnew)
725 greg 2.1 return(NULL);
726     for (i = cnew->nrows; i--; )
727     for (j = cnew->ncols; j--; ) {
728     COLORV *cv = cm_lval(cnew,i,j);
729     cv[0] = (COLORV)rmx_lval(rm,i,j,0);
730     cv[1] = (COLORV)rmx_lval(rm,i,j,1);
731     cv[2] = (COLORV)rmx_lval(rm,i,j,2);
732     }
733     return(cnew);
734     }