ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/src/util/rmatrix.c
(Generate patch)

Comparing src/util/rmatrix.c (file contents):
Revision 2.26 by greg, Mon Apr 9 21:30:07 2018 UTC vs.
Revision 2.59 by greg, Sat Jul 1 15:25:26 2023 UTC

# Line 5 | Line 5 | static const char RCSid[] = "$Id$";
5   * General matrix operations.
6   */
7  
8 – #include <stdio.h>
8   #include <stdlib.h>
10 – #include <string.h>
11 – #include <fcntl.h>
9   #include <errno.h>
10   #include "rtio.h"
11   #include "platform.h"
12   #include "resolu.h"
13   #include "paths.h"
14   #include "rmatrix.h"
15 + #if !defined(_WIN32) && !defined(_WIN64)
16 + #include <sys/mman.h>
17 + #endif
18  
19   static char     rmx_mismatch_warn[] = "WARNING: data type mismatch\n";
20  
21 < /* Allocate a nr x nc matrix with n components */
21 > #define array_size(rm)  (sizeof(double)*(rm)->nrows*(rm)->ncols*(rm)->ncomp)
22 > #define mapped_size(rm) ((char *)(rm)->mtx + array_size(rm) - (char *)(rm)->mapped)
23 >
24 > /* Initialize a RMATRIX struct but don't allocate array space */
25   RMATRIX *
26 + rmx_new(int nr, int nc, int n)
27 + {
28 +        RMATRIX *dnew = (RMATRIX *)calloc(1, sizeof(RMATRIX));
29 +
30 +        if (dnew) {
31 +                dnew->dtype = DTdouble;
32 +                dnew->nrows = nr;
33 +                dnew->ncols = nc;
34 +                dnew->ncomp = n;
35 +                setcolor(dnew->cexp, 1.f, 1.f, 1.f);
36 +        }
37 +        return(dnew);
38 + }
39 +
40 + /* Prepare a RMATRIX for writing (allocate array if needed) */
41 + int
42 + rmx_prepare(RMATRIX *rm)
43 + {
44 +        if (!rm) return(0);
45 +        if (rm->mtx)
46 +                return(1);
47 +        rm->mtx = (double *)malloc(array_size(rm));
48 +        return(rm->mtx != NULL);
49 + }
50 +
51 + /* Call rmx_new() and rmx_prepare() */
52 + RMATRIX *
53   rmx_alloc(int nr, int nc, int n)
54   {
55 <        RMATRIX *dnew;
55 >        RMATRIX *dnew = rmx_new(nr, nc, n);
56  
57 <        if ((nr <= 0) | (nc <= 0) | (n <= 0))
58 <                return(NULL);
59 <        dnew = (RMATRIX *)malloc(sizeof(RMATRIX)-sizeof(dnew->mtx) +
60 <                                        sizeof(dnew->mtx[0])*(n*nr*nc));
31 <        if (dnew == NULL)
32 <                return(NULL);
33 <        dnew->nrows = nr; dnew->ncols = nc; dnew->ncomp = n;
34 <        dnew->dtype = DTdouble;
35 <        dnew->info = NULL;
57 >        if (dnew && !rmx_prepare(dnew)) {
58 >                rmx_free(dnew);
59 >                dnew = NULL;
60 >        }
61          return(dnew);
62   }
63  
# Line 43 | Line 68 | rmx_free(RMATRIX *rm)
68          if (!rm) return;
69          if (rm->info)
70                  free(rm->info);
71 + #ifdef MAP_FILE
72 +        if (rm->mapped)
73 +                munmap(rm->mapped, mapped_size(rm));
74 +        else
75 + #endif
76 +                free(rm->mtx);
77          free(rm);
78   }
79  
# Line 63 | Line 94 | rmx_newtype(int dtyp1, int dtyp2)
94   int
95   rmx_addinfo(RMATRIX *rm, const char *info)
96   {
97 <        if (!info || !*info)
97 >        int     oldlen = 0;
98 >
99 >        if (!rm || !info || !*info)
100                  return(0);
101          if (!rm->info) {
102                  rm->info = (char *)malloc(strlen(info)+1);
103                  if (rm->info) rm->info[0] = '\0';
104 <        } else
104 >        } else {
105 >                oldlen = strlen(rm->info);
106                  rm->info = (char *)realloc(rm->info,
107 <                                strlen(rm->info)+strlen(info)+1);
107 >                                oldlen+strlen(info)+1);
108 >        }
109          if (!rm->info)
110                  return(0);
111 <        strcat(rm->info, info);
111 >        strcpy(rm->info+oldlen, info);
112          return(1);
113   }
114  
# Line 81 | Line 116 | static int
116   get_dminfo(char *s, void *p)
117   {
118          RMATRIX *ip = (RMATRIX *)p;
119 <        char    fmt[64];
119 >        char    fmt[MAXFMTLEN];
120          int     i;
121  
122 <        if (headidval(fmt, s))
122 >        if (headidval(NULL, s))
123                  return(0);
124          if (!strncmp(s, "NCOMP=", 6)) {
125                  ip->ncomp = atoi(s+6);
# Line 98 | Line 133 | get_dminfo(char *s, void *p)
133                  ip->ncols = atoi(s+6);
134                  return(0);
135          }
136 +        if ((i = isbigendian(s)) >= 0) {
137 +                ip->swapin = (nativebigendian() != i);
138 +                return(0);
139 +        }
140 +        if (isexpos(s)) {
141 +                float   f = exposval(s);
142 +                scalecolor(ip->cexp, f);
143 +                return(0);
144 +        }
145 +        if (iscolcor(s)) {
146 +                COLOR   ctmp;
147 +                colcorval(ctmp, s);
148 +                multcolor(ip->cexp, ctmp);
149 +                return(0);
150 +        }
151          if (!formatval(fmt, s)) {
152                  rmx_addinfo(ip, s);
153                  return(0);
154 <        }
154 >        }                       /* else check format */
155          for (i = 1; i < DTend; i++)
156                  if (!strcmp(fmt, cm_fmt_id[i])) {
157                          ip->dtype = i;
# Line 115 | Line 165 | rmx_load_ascii(RMATRIX *rm, FILE *fp)
165   {
166          int     i, j, k;
167  
168 +        if (!rmx_prepare(rm))
169 +                return(0);
170          for (i = 0; i < rm->nrows; i++)
171 <            for (j = 0; j < rm->ncols; j++)
171 >            for (j = 0; j < rm->ncols; j++) {
172 >                double  *dp = rmx_lval(rm,i,j);
173                  for (k = 0; k < rm->ncomp; k++)
174 <                    if (fscanf(fp, "%lf", &rmx_lval(rm,i,j,k)) != 1)
174 >                    if (fscanf(fp, "%lf", &dp[k]) != 1)
175                          return(0);
176 +            }
177          return(1);
178   }
179  
# Line 133 | Line 187 | rmx_load_float(RMATRIX *rm, FILE *fp)
187                  fputs("Unsupported # components in rmx_load_float()\n", stderr);
188                  exit(1);
189          }
190 +        if (!rmx_prepare(rm))
191 +                return(0);
192          for (i = 0; i < rm->nrows; i++)
193              for (j = 0; j < rm->ncols; j++) {
194 +                double  *dp = rmx_lval(rm,i,j);
195                  if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp)
196                      return(0);
197 +                if (rm->swapin)
198 +                    swap32((char *)val, rm->ncomp);
199                  for (k = rm->ncomp; k--; )
200 <                     rmx_lval(rm,i,j,k) = val[k];
200 >                     dp[k] = val[k];
201              }
202          return(1);
203   }
# Line 146 | Line 205 | rmx_load_float(RMATRIX *rm, FILE *fp)
205   static int
206   rmx_load_double(RMATRIX *rm, FILE *fp)
207   {
208 <        int     i, j;
209 <
210 <        for (i = 0; i < rm->nrows; i++)
211 <            for (j = 0; j < rm->ncols; j++)
212 <                if (getbinary(&rmx_lval(rm,i,j,0), sizeof(double), rm->ncomp, fp) != rm->ncomp)
213 <                    return(0);
208 >        int     i;
209 > #ifdef MAP_FILE
210 >        long    pos;            /* map memory for file > 1MB if possible */
211 >        if (!rm->swapin && array_size(rm) >= 1L<<20 &&
212 >                        (pos = ftell(fp)) >= 0 && !(pos % sizeof(double))) {
213 >                rm->mapped = mmap(NULL, array_size(rm)+pos, PROT_READ|PROT_WRITE,
214 >                                        MAP_PRIVATE, fileno(fp), 0);
215 >                if (rm->mapped != MAP_FAILED) {
216 >                        rm->mtx = (double *)rm->mapped + pos/sizeof(double);
217 >                        return(1);
218 >                }               /* else fall back on reading into memory */
219 >                rm->mapped = NULL;
220 >        }
221 > #endif
222 >        if (!rmx_prepare(rm))
223 >                return(0);
224 >        for (i = 0; i < rm->nrows; i++) {
225 >                if (getbinary(rmx_lval(rm,i,0), sizeof(double)*rm->ncomp,
226 >                                        rm->ncols, fp) != rm->ncols)
227 >                        return(0);
228 >                if (rm->swapin)
229 >                        swap64((char *)rmx_lval(rm,i,0), rm->ncols*rm->ncomp);
230 >        }
231          return(1);
232   }
233  
# Line 161 | Line 237 | rmx_load_rgbe(RMATRIX *rm, FILE *fp)
237          COLOR   *scan = (COLOR *)malloc(sizeof(COLOR)*rm->ncols);
238          int     i, j;
239  
240 <        if (scan == NULL)
240 >        if (!scan)
241                  return(0);
242 +        if (!rmx_prepare(rm))
243 +                return(0);
244          for (i = 0; i < rm->nrows; i++) {
245 +            double      *dp = rmx_lval(rm,i,0);
246              if (freadscan(scan, rm->ncols, fp) < 0) {
247                  free(scan);
248                  return(0);
249              }
250 <            for (j = rm->ncols; j--; ) {
251 <                rmx_lval(rm,i,j,0) = colval(scan[j],RED);
252 <                rmx_lval(rm,i,j,1) = colval(scan[j],GRN);
253 <                rmx_lval(rm,i,j,2) = colval(scan[j],BLU);
250 >            for (j = 0; j < rm->ncols; j++, dp += 3) {
251 >                dp[0] = colval(scan[j],RED);
252 >                dp[1] = colval(scan[j],GRN);
253 >                dp[2] = colval(scan[j],BLU);
254              }
255          }
256          free(scan);
# Line 180 | Line 259 | rmx_load_rgbe(RMATRIX *rm, FILE *fp)
259  
260   /* Load matrix from supported file type */
261   RMATRIX *
262 < rmx_load(const char *inspec)
262 > rmx_load(const char *inspec, RMPref rmp)
263   {
264 <        FILE            *fp = stdin;
186 <        RMATRIX         dinfo;
264 >        FILE            *fp;
265          RMATRIX         *dnew;
266  
267 <        if (inspec == NULL) {                   /* reading from stdin? */
268 <                inspec = "<stdin>";
269 <                SET_FILE_BINARY(stdin);
267 >        if (!inspec)
268 >                inspec = stdin_name;
269 >        else if (!*inspec)
270 >                return(NULL);
271 >        if (inspec == stdin_name) {             /* reading from stdin? */
272 >                fp = stdin;
273          } else if (inspec[0] == '!') {
274 <                if ((fp = popen(inspec+1, "r")) == NULL)
274 >                if (!(fp = popen(inspec+1, "r")))
275                          return(NULL);
195 –                SET_FILE_BINARY(stdin);
276          } else {
277                  const char      *sp = inspec;   /* check suffix */
278                  while (*sp)
# Line 200 | Line 280 | rmx_load(const char *inspec)
280                  while (sp > inspec && sp[-1] != '.')
281                          --sp;
282                  if (!strcasecmp(sp, "XML")) {   /* assume it's a BSDF */
283 <                        CMATRIX *cm = cm_loadBTDF((char *)inspec);
284 <                        if (cm == NULL)
283 >                        CMATRIX *cm = rmp==RMPtrans ? cm_loadBTDF(inspec) :
284 >                                        cm_loadBRDF(inspec, rmp==RMPreflB) ;
285 >                        if (!cm)
286                                  return(NULL);
287                          dnew = rmx_from_cmatrix(cm);
288                          cm_free(cm);
# Line 209 | Line 290 | rmx_load(const char *inspec)
290                          return(dnew);
291                  }
292                                                  /* else open it ourselves */
293 <                if ((fp = fopen(inspec, "rb")) == NULL)
293 >                if (!(fp = fopen(inspec, "r")))
294                          return(NULL);
295          }
296 +        SET_FILE_BINARY(fp);
297   #ifdef getc_unlocked
298          flockfile(fp);
299   #endif
300 <        dinfo.nrows = dinfo.ncols = dinfo.ncomp = 0;
219 <        dinfo.dtype = DTascii;                  /* assumed w/o FORMAT */
220 <        dinfo.info = NULL;
221 <        if (getheader(fp, get_dminfo, &dinfo) < 0) {
300 >        if (!(dnew = rmx_new(0,0,3))) {
301                  fclose(fp);
302                  return(NULL);
303          }
304 <        if ((dinfo.nrows <= 0) | (dinfo.ncols <= 0)) {
305 <                if (!fscnresolu(&dinfo.ncols, &dinfo.nrows, fp)) {
304 >        dnew->dtype = DTascii;                  /* assumed w/o FORMAT */
305 >        if (getheader(fp, get_dminfo, dnew) < 0) {
306 >                fclose(fp);
307 >                return(NULL);
308 >        }
309 >        if ((dnew->nrows <= 0) | (dnew->ncols <= 0)) {
310 >                if (!fscnresolu(&dnew->ncols, &dnew->nrows, fp)) {
311                          fclose(fp);
312                          return(NULL);
313                  }
314 <                if (dinfo.ncomp <= 0)
315 <                        dinfo.ncomp = 3;
232 <                else if ((dinfo.dtype == DTrgbe) | (dinfo.dtype == DTxyze) &&
233 <                                dinfo.ncomp != 3) {
314 >                if ((dnew->dtype == DTrgbe) | (dnew->dtype == DTxyze) &&
315 >                                dnew->ncomp != 3) {
316                          fclose(fp);
317                          return(NULL);
318                  }
319          }
320 <        dnew = rmx_alloc(dinfo.nrows, dinfo.ncols, dinfo.ncomp);
239 <        if (dnew == NULL) {
240 <                fclose(fp);
241 <                return(NULL);
242 <        }
243 <        dnew->info = dinfo.info;
244 <        switch (dinfo.dtype) {
320 >        switch (dnew->dtype) {
321          case DTascii:
322 <                SET_FILE_TEXT(stdin);
322 >                SET_FILE_TEXT(fp);
323                  if (!rmx_load_ascii(dnew, fp))
324                          goto loaderr;
325                  dnew->dtype = DTascii;          /* should leave double? */
# Line 262 | Line 338 | rmx_load(const char *inspec)
338          case DTxyze:
339                  if (!rmx_load_rgbe(dnew, fp))
340                          goto loaderr;
265 –                dnew->dtype = dinfo.dtype;
341                  break;
342          default:
343                  goto loaderr;
# Line 277 | Line 352 | rmx_load(const char *inspec)
352          else
353                  funlockfile(fp);
354   #endif
355 +                                                /* undo exposure? */
356 +        if (dnew->ncomp == 3 && (dnew->cexp[0] != 1.f) |
357 +                        (dnew->cexp[1] != 1.f) | (dnew->cexp[2] != 1.f)) {
358 +                double  cmlt[3];
359 +                cmlt[0] = 1./dnew->cexp[0];
360 +                cmlt[1] = 1./dnew->cexp[1];
361 +                cmlt[2] = 1./dnew->cexp[2];
362 +                rmx_scale(dnew, cmlt);
363 +                setcolor(dnew->cexp, 1.f, 1.f, 1.f);
364 +        }
365          return(dnew);
366   loaderr:                                        /* should report error? */
367          if (inspec[0] == '!')
# Line 290 | Line 375 | loaderr:                                       /* should report error? */
375   static int
376   rmx_write_ascii(const RMATRIX *rm, FILE *fp)
377   {
378 +        const char      *fmt = (rm->dtype == DTfloat) ? " %.7e" :
379 +                        (rm->dtype == DTrgbe) | (rm->dtype == DTxyze) ? " %.3e" :
380 +                                " %.15e" ;
381          int     i, j, k;
382  
383          for (i = 0; i < rm->nrows; i++) {
384              for (j = 0; j < rm->ncols; j++) {
385 +                const double    *dp = rmx_lval(rm,i,j);
386                  for (k = 0; k < rm->ncomp; k++)
387 <                    fprintf(fp, " %.15e", rmx_lval(rm,i,j,k));
387 >                    fprintf(fp, fmt, dp[k]);
388                  fputc('\t', fp);
389              }
390              fputc('\n', fp);
# Line 315 | Line 404 | rmx_write_float(const RMATRIX *rm, FILE *fp)
404          }
405          for (i = 0; i < rm->nrows; i++)
406              for (j = 0; j < rm->ncols; j++) {
407 +                const double    *dp = rmx_lval(rm,i,j);
408                  for (k = rm->ncomp; k--; )
409 <                    val[k] = (float)rmx_lval(rm,i,j,k);
410 <                if (putbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp)
409 >                    val[k] = (float)dp[k];
410 >                if (putbinary(val, sizeof(float), rm->ncomp, fp) != rm->ncomp)
411                          return(0);
412              }
413          return(1);
# Line 326 | Line 416 | rmx_write_float(const RMATRIX *rm, FILE *fp)
416   static int
417   rmx_write_double(const RMATRIX *rm, FILE *fp)
418   {
419 <        int     i, j;
419 >        int     i;
420  
421          for (i = 0; i < rm->nrows; i++)
422 <            for (j = 0; j < rm->ncols; j++)
423 <                if (putbinary(&rmx_lval(rm,i,j,0), sizeof(double), rm->ncomp, fp) != rm->ncomp)
422 >                if (putbinary(rmx_lval(rm,i,0), sizeof(double)*rm->ncomp,
423 >                                        rm->ncols, fp) != rm->ncols)
424                          return(0);
425          return(1);
426   }
# Line 338 | Line 428 | rmx_write_double(const RMATRIX *rm, FILE *fp)
428   static int
429   rmx_write_rgbe(const RMATRIX *rm, FILE *fp)
430   {
431 <        COLOR   *scan = (COLOR *)malloc(sizeof(COLOR)*rm->ncols);
431 >        COLR    *scan = (COLR *)malloc(sizeof(COLR)*rm->ncols);
432          int     i, j;
433  
434 <        if (scan == NULL)
434 >        if (!scan)
435                  return(0);
436          for (i = 0; i < rm->nrows; i++) {
437 <            for (j = rm->ncols; j--; )
438 <                setcolor(scan[j],       rmx_lval(rm,i,j,0),
439 <                                        rmx_lval(rm,i,j,1),
440 <                                        rmx_lval(rm,i,j,2)      );
441 <            if (fwritescan(scan, rm->ncols, fp) < 0) {
437 >            for (j = rm->ncols; j--; ) {
438 >                const double    *dp = rmx_lval(rm,i,j);
439 >                if (rm->ncomp == 1)
440 >                        setcolr(scan[j], dp[0], dp[0], dp[0]);
441 >                else
442 >                        setcolr(scan[j], dp[0], dp[1], dp[2]);
443 >            }
444 >            if (fwritecolrs(scan, rm->ncols, fp) < 0) {
445                  free(scan);
446                  return(0);
447              }
# Line 357 | Line 450 | rmx_write_rgbe(const RMATRIX *rm, FILE *fp)
450          return(1);
451   }
452  
453 + /* Check if CIE XYZ primaries were specified */
454 + static int
455 + findCIEprims(const char *info)
456 + {
457 +        RGBPRIMS        prims;
458 +
459 +        if (!info)
460 +                return(0);
461 +        info = strstr(info, PRIMARYSTR);
462 +        if (!info || !primsval(prims, info))
463 +                return(0);
464 +
465 +        return((prims[RED][CIEX] > .99) & (prims[RED][CIEY] < .01) &&
466 +                        (prims[GRN][CIEX] < .01) & (prims[GRN][CIEY] > .99) &&
467 +                        (prims[BLU][CIEX] < .01) & (prims[BLU][CIEY] < .01));
468 + }
469 +
470   /* Write matrix to file type indicated by dtype */
471   int
472   rmx_write(const RMATRIX *rm, int dtype, FILE *fp)
473   {
364 –        RMATRIX *mydm = NULL;
474          int     ok = 1;
475  
476 <        if ((rm == NULL) | (fp == NULL))
476 >        if (!rm | !fp || !rm->mtx)
477                  return(0);
478   #ifdef getc_unlocked
479          flockfile(fp);
# Line 374 | Line 483 | rmx_write(const RMATRIX *rm, int dtype, FILE *fp)
483                  fputs(rm->info, fp);
484          if (dtype == DTfromHeader)
485                  dtype = rm->dtype;
486 <        else if ((dtype == DTrgbe) & (rm->dtype == DTxyze))
486 >        else if (dtype == DTrgbe && (rm->dtype == DTxyze ||
487 >                                        findCIEprims(rm->info)))
488                  dtype = DTxyze;
489          else if ((dtype == DTxyze) & (rm->dtype == DTrgbe))
490                  dtype = DTrgbe;
491 +        if (rm->ncomp == 3) {                   /* write exposure? */
492 +                if ((rm->cexp[RED] != rm->cexp[GRN]) |
493 +                                (rm->cexp[GRN] != rm->cexp[BLU]))
494 +                        fputcolcor(rm->cexp, fp);
495 +                else if (rm->cexp[GRN] != 1.f)
496 +                        fputexpos(rm->cexp[GRN], fp);
497 +        }
498          if ((dtype != DTrgbe) & (dtype != DTxyze)) {
499                  fprintf(fp, "NROWS=%d\n", rm->nrows);
500                  fprintf(fp, "NCOLS=%d\n", rm->ncols);
501                  fprintf(fp, "NCOMP=%d\n", rm->ncomp);
502 <        } else if (rm->ncomp != 3) {            /* wrong # components? */
503 <                double  cmtx[3];
504 <                if (rm->ncomp != 1)             /* only convert grayscale */
505 <                        return(0);
506 <                cmtx[0] = cmtx[1] = cmtx[2] = 1;
390 <                mydm = rmx_transform(rm, 3, cmtx);
391 <                if (mydm == NULL)
392 <                        return(0);
393 <                rm = mydm;
394 <        }
395 <        fputformat((char *)cm_fmt_id[dtype], fp);
502 >        } else if ((rm->ncomp != 3) & (rm->ncomp != 1))
503 >                return(0);                      /* wrong # components */
504 >        if ((dtype == DTfloat) | (dtype == DTdouble))
505 >                fputendian(fp);                 /* important to record */
506 >        fputformat(cm_fmt_id[dtype], fp);
507          fputc('\n', fp);
508          switch (dtype) {                        /* write data */
509          case DTascii:
# Line 416 | Line 527 | rmx_write(const RMATRIX *rm, int dtype, FILE *fp)
527   #ifdef getc_unlocked
528          funlockfile(fp);
529   #endif
419 –        rmx_free(mydm);
530          return(ok);
531   }
532  
# Line 427 | Line 537 | rmx_identity(const int dim, const int n)
537          RMATRIX *rid = rmx_alloc(dim, dim, n);
538          int     i, k;
539  
540 <        if (rid == NULL)
540 >        if (!rid)
541                  return(NULL);
542 <        memset(rid->mtx, 0, sizeof(rid->mtx[0])*n*dim*dim);
543 <        for (i = dim; i--; )
542 >        memset(rid->mtx, 0, array_size(rid));
543 >        for (i = dim; i--; ) {
544 >            double      *dp = rmx_lval(rid,i,i);
545              for (k = n; k--; )
546 <                rmx_lval(rid,i,i,k) = 1;
546 >                dp[k] = 1.;
547 >        }
548          return(rid);
549   }
550  
# Line 442 | Line 554 | rmx_copy(const RMATRIX *rm)
554   {
555          RMATRIX *dnew;
556  
557 <        if (rm == NULL)
557 >        if (!rm)
558                  return(NULL);
559          dnew = rmx_alloc(rm->nrows, rm->ncols, rm->ncomp);
560 <        if (dnew == NULL)
560 >        if (!dnew)
561                  return(NULL);
562          rmx_addinfo(dnew, rm->info);
563          dnew->dtype = rm->dtype;
564 <        memcpy(dnew->mtx, rm->mtx,
453 <                sizeof(rm->mtx[0])*rm->ncomp*rm->nrows*rm->ncols);
564 >        memcpy(dnew->mtx, rm->mtx, array_size(dnew));
565          return(dnew);
566   }
567  
# Line 459 | Line 570 | RMATRIX *
570   rmx_transpose(const RMATRIX *rm)
571   {
572          RMATRIX *dnew;
573 <        int     i, j, k;
573 >        int     i, j;
574  
575 <        if (rm == NULL)
575 >        if (!rm)
576                  return(0);
577 +        if ((rm->nrows == 1) | (rm->ncols == 1)) {
578 +                dnew = rmx_copy(rm);
579 +                if (!dnew)
580 +                        return(NULL);
581 +                dnew->nrows = rm->ncols;
582 +                dnew->ncols = rm->nrows;
583 +                return(dnew);
584 +        }
585          dnew = rmx_alloc(rm->ncols, rm->nrows, rm->ncomp);
586 <        if (dnew == NULL)
586 >        if (!dnew)
587                  return(NULL);
588          if (rm->info) {
589                  rmx_addinfo(dnew, rm->info);
590                  rmx_addinfo(dnew, "Transposed rows and columns\n");
591          }
592          dnew->dtype = rm->dtype;
593 <        for (i = dnew->nrows; i--; )
594 <            for (j = dnew->ncols; j--; )
595 <                for (k = dnew->ncomp; k--; )
596 <                        rmx_lval(dnew,i,j,k) = rmx_lval(rm,j,i,k);
593 >        for (j = dnew->ncols; j--; )
594 >            for (i = dnew->nrows; i--; )
595 >                memcpy(rmx_lval(dnew,i,j), rmx_lval(rm,j,i),
596 >                                sizeof(double)*dnew->ncomp);
597          return(dnew);
598   }
599  
# Line 485 | Line 604 | rmx_multiply(const RMATRIX *m1, const RMATRIX *m2)
604          RMATRIX *mres;
605          int     i, j, k, h;
606  
607 <        if ((m1 == NULL) | (m2 == NULL) ||
489 <                        (m1->ncomp != m2->ncomp) | (m1->ncols != m2->nrows))
607 >        if (!m1 | !m2 || (m1->ncomp != m2->ncomp) | (m1->ncols != m2->nrows))
608                  return(NULL);
609          mres = rmx_alloc(m1->nrows, m2->ncols, m1->ncomp);
610 <        if (mres == NULL)
610 >        if (!mres)
611                  return(NULL);
612          i = rmx_newtype(m1->dtype, m2->dtype);
613          if (i)
# Line 499 | Line 617 | rmx_multiply(const RMATRIX *m1, const RMATRIX *m2)
617          for (i = mres->nrows; i--; )
618              for (j = mres->ncols; j--; )
619                  for (k = mres->ncomp; k--; ) {
620 <                    long double d = 0;
620 >                    double      d = 0;
621                      for (h = m1->ncols; h--; )
622 <                        d += rmx_lval(m1,i,h,k) * rmx_lval(m2,h,j,k);
623 <                    rmx_lval(mres,i,j,k) = (double)d;
622 >                        d += rmx_lval(m1,i,h)[k] * rmx_lval(m2,h,j)[k];
623 >                    rmx_lval(mres,i,j)[k] = d;
624                  }
625          return(mres);
626   }
# Line 514 | Line 632 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
632          int     zeroDivides = 0;
633          int     i, j, k;
634  
635 <        if ((m1 == NULL) | (m2 == NULL) ||
518 <                        (m1->ncols != m2->ncols) | (m1->nrows != m2->nrows))
635 >        if (!m1 | !m2 || (m1->ncols != m2->ncols) | (m1->nrows != m2->nrows))
636                  return(0);
637          if ((m2->ncomp > 1) & (m2->ncomp != m1->ncomp))
638                  return(0);
# Line 529 | Line 646 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
646                  if (divide) {
647                      double      d;
648                      if (m2->ncomp == 1) {
649 <                        d = rmx_lval(m2,i,j,0);
649 >                        d = rmx_lval(m2,i,j)[0];
650                          if (d == 0) {
651                              ++zeroDivides;
652                              for (k = m1->ncomp; k--; )
653 <                                rmx_lval(m1,i,j,k) = 0;
653 >                                rmx_lval(m1,i,j)[k] = 0;
654                          } else {
655                              d = 1./d;
656                              for (k = m1->ncomp; k--; )
657 <                                rmx_lval(m1,i,j,k) *= d;
657 >                                rmx_lval(m1,i,j)[k] *= d;
658                          }
659                      } else
660                          for (k = m1->ncomp; k--; ) {
661 <                            d = rmx_lval(m2,i,j,k);
661 >                            d = rmx_lval(m2,i,j)[k];
662                              if (d == 0) {
663                                  ++zeroDivides;
664 <                                rmx_lval(m1,i,j,k) = 0;
664 >                                rmx_lval(m1,i,j)[k] = 0;
665                              } else
666 <                                rmx_lval(m1,i,j,k) /= d;
666 >                                rmx_lval(m1,i,j)[k] /= d;
667                          }
668                  } else {
669                      if (m2->ncomp == 1) {
670 <                        const double    d = rmx_lval(m2,i,j,0);
670 >                        const double    d = rmx_lval(m2,i,j)[0];
671                          for (k = m1->ncomp; k--; )
672 <                            rmx_lval(m1,i,j,k) *= d;
672 >                            rmx_lval(m1,i,j)[k] *= d;
673                      } else
674                          for (k = m1->ncomp; k--; )
675 <                            rmx_lval(m1,i,j,k) *= rmx_lval(m2,i,j,k);
675 >                            rmx_lval(m1,i,j)[k] *= rmx_lval(m2,i,j)[k];
676                  }
677          if (zeroDivides) {
678 <                fputs("Divide by zero in rmx_elemult()\n", stderr);
678 >                rmx_addinfo(m1, "WARNING: zero divide(s) corrupted results\n");
679                  errno = ERANGE;
680          }
681          return(1);
# Line 571 | Line 688 | rmx_sum(RMATRIX *msum, const RMATRIX *madd, const doub
688          double  *mysf = NULL;
689          int     i, j, k;
690  
691 <        if ((msum == NULL) | (madd == NULL) ||
691 >        if (!msum | !madd ||
692                          (msum->nrows != madd->nrows) |
693                          (msum->ncols != madd->ncols) |
694                          (msum->ncomp != madd->ncomp))
695                  return(0);
696 <        if (sf == NULL) {
696 >        if (!sf) {
697                  mysf = (double *)malloc(sizeof(double)*msum->ncomp);
698 <                if (mysf == NULL)
698 >                if (!mysf)
699                          return(0);
700                  for (k = msum->ncomp; k--; )
701                          mysf[k] = 1;
# Line 590 | Line 707 | rmx_sum(RMATRIX *msum, const RMATRIX *madd, const doub
707          else
708                  rmx_addinfo(msum, rmx_mismatch_warn);
709          for (i = msum->nrows; i--; )
710 <            for (j = msum->ncols; j--; )
710 >            for (j = msum->ncols; j--; ) {
711 >                const double    *da = rmx_lval(madd,i,j);
712 >                double          *ds = rmx_lval(msum,i,j);
713                  for (k = msum->ncomp; k--; )
714 <                     rmx_lval(msum,i,j,k) += sf[k] * rmx_lval(madd,i,j,k);
715 <
716 <        free(mysf);
714 >                     ds[k] += sf[k] * da[k];
715 >            }
716 >        if (mysf)
717 >                free(mysf);
718          return(1);
719   }
720  
# Line 604 | Line 724 | rmx_scale(RMATRIX *rm, const double sf[])
724   {
725          int     i, j, k;
726  
727 <        if ((rm == NULL) | (sf == NULL))
727 >        if (!rm | !sf)
728                  return(0);
729          for (i = rm->nrows; i--; )
730 <            for (j = rm->ncols; j--; )
730 >            for (j = rm->ncols; j--; ) {
731 >                double  *dp = rmx_lval(rm,i,j);
732                  for (k = rm->ncomp; k--; )
733 <                    rmx_lval(rm,i,j,k) *= sf[k];
734 <
733 >                    dp[k] *= sf[k];
734 >            }
735 >        if (rm->info)
736 >                rmx_addinfo(rm, "Applied scalar\n");
737          return(1);
738   }
739  
# Line 621 | Line 744 | rmx_transform(const RMATRIX *msrc, int n, const double
744          int     i, j, ks, kd;
745          RMATRIX *dnew;
746  
747 <        if ((msrc == NULL) | (n <= 0) | (cmat == NULL))
747 >        if (!msrc | (n <= 0) | !cmat)
748                  return(NULL);
749          dnew = rmx_alloc(msrc->nrows, msrc->ncols, n);
750 <        if (dnew == NULL)
750 >        if (!dnew)
751                  return(NULL);
752 +        if (msrc->info) {
753 +                char    buf[128];
754 +                sprintf(buf, "Applied %dx%d component transform\n",
755 +                                dnew->ncomp, msrc->ncomp);
756 +                rmx_addinfo(dnew, msrc->info);
757 +                rmx_addinfo(dnew, buf);
758 +        }
759          dnew->dtype = msrc->dtype;
760          for (i = dnew->nrows; i--; )
761 <            for (j = dnew->ncols; j--; )
761 >            for (j = dnew->ncols; j--; ) {
762 >                const double    *ds = rmx_lval(msrc,i,j);
763                  for (kd = dnew->ncomp; kd--; ) {
764                      double      d = 0;
765                      for (ks = msrc->ncomp; ks--; )
766 <                        d += cmat[kd*msrc->ncomp + ks] * rmx_lval(msrc,i,j,ks);
767 <                    rmx_lval(dnew,i,j,kd) = d;
766 >                        d += cmat[kd*msrc->ncomp + ks] * ds[ks];
767 >                    rmx_lval(dnew,i,j)[kd] = d;
768                  }
769 +            }
770          return(dnew);
771   }
772  
# Line 645 | Line 777 | rmx_from_cmatrix(const CMATRIX *cm)
777          int     i, j;
778          RMATRIX *dnew;
779  
780 <        if (cm == NULL)
780 >        if (!cm)
781                  return(NULL);
782          dnew = rmx_alloc(cm->nrows, cm->ncols, 3);
783 <        if (dnew == NULL)
783 >        if (!dnew)
784                  return(NULL);
785          dnew->dtype = DTfloat;
786          for (i = dnew->nrows; i--; )
787              for (j = dnew->ncols; j--; ) {
788                  const COLORV    *cv = cm_lval(cm,i,j);
789 <                rmx_lval(dnew,i,j,0) = cv[0];
790 <                rmx_lval(dnew,i,j,1) = cv[1];
791 <                rmx_lval(dnew,i,j,2) = cv[2];
789 >                double          *dp = rmx_lval(dnew,i,j);
790 >                dp[0] = cv[0];
791 >                dp[1] = cv[1];
792 >                dp[2] = cv[2];
793              }
794          return(dnew);
795   }
# Line 668 | Line 801 | cm_from_rmatrix(const RMATRIX *rm)
801          int     i, j;
802          CMATRIX *cnew;
803  
804 <        if (rm == NULL || rm->ncomp != 3)
804 >        if (!rm || !rm->mtx | ((rm->ncomp != 3) & (rm->ncomp != 1)))
805                  return(NULL);
806          cnew = cm_alloc(rm->nrows, rm->ncols);
807 <        if (cnew == NULL)
807 >        if (!cnew)
808                  return(NULL);
809          for (i = cnew->nrows; i--; )
810              for (j = cnew->ncols; j--; ) {
811 <                COLORV  *cv = cm_lval(cnew,i,j);
812 <                cv[0] = (COLORV)rmx_lval(rm,i,j,0);
813 <                cv[1] = (COLORV)rmx_lval(rm,i,j,1);
814 <                cv[2] = (COLORV)rmx_lval(rm,i,j,2);
811 >                const double    *dp = rmx_lval(rm,i,j);
812 >                COLORV          *cv = cm_lval(cnew,i,j);
813 >                if (rm->ncomp == 1)
814 >                    setcolor(cv, dp[0], dp[0], dp[0]);
815 >                else
816 >                    setcolor(cv, dp[0], dp[1], dp[2]);
817              }
818          return(cnew);
819   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines