ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rmatrix.c
Revision: 2.39
Committed: Sat Dec 28 18:05:14 2019 UTC (4 years, 3 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.38: +1 -4 lines
Log Message:
Removed redundant include files

File Contents

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