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

Comparing ray/src/util/rmatrix.c (file contents):
Revision 2.81 by greg, Thu Jun 6 16:46:31 2024 UTC vs.
Revision 2.86 by greg, Fri Mar 28 00:06:36 2025 UTC

# Line 16 | Line 16 | static const char RCSid[] = "$Id$";
16   #include <sys/mman.h>
17   #endif
18  
19 + #ifndef MAXCOMP
20 + #define MAXCOMP         MAXCSAMP        /* #components we support */
21 + #endif
22 +
23   static const char       rmx_mismatch_warn[] = "WARNING: data type mismatch\n";
24  
25   /* Initialize a RMATRIX struct but don't allocate array space */
# Line 46 | Line 50 | int
50   rmx_prepare(RMATRIX *rm)
51   {
52          if (!rm) return(0);
53 <        if (rm->mtx)
53 >        if (rm->mtx)                    /* assume it's right size */
54                  return(1);
55          if ((rm->nrows <= 0) | (rm->ncols <= 0) | (rm->ncomp <= 0))
56                  return(0);
57 <        rm->mtx = (double *)malloc(rmx_array_size(rm));
57 >        rm->mtx = (rmx_dtype *)malloc(rmx_array_size(rm));
58          rm->pflags |= RMF_FREEMEM;
59          return(rm->mtx != NULL);
60   }
# Line 61 | Line 65 | rmx_alloc(int nr, int nc, int n)
65   {
66          RMATRIX *dnew = rmx_new(nr, nc, n);
67  
68 <        if (dnew && !rmx_prepare(dnew)) {
68 >        if (!rmx_prepare(dnew)) {
69                  rmx_free(dnew);
70 <                dnew = NULL;
70 >                return(NULL);
71          }
72          return(dnew);
73   }
# Line 116 | Line 120 | rmx_newtype(int dtyp1, int dtyp2)
120   int
121   rmx_addinfo(RMATRIX *rm, const char *info)
122   {
123 <        int     oldlen = 0;
123 >        size_t  oldlen = 0;
124  
125          if (!rm || !info || !*info)
126                  return(0);
127          if (!rm->info) {
128                  rm->info = (char *)malloc(strlen(info)+1);
125 –                if (rm->info) rm->info[0] = '\0';
129          } else {
130                  oldlen = strlen(rm->info);
131                  rm->info = (char *)realloc(rm->info,
# Line 141 | Line 144 | get_dminfo(char *s, void *p)
144          char    fmt[MAXFMTLEN];
145          int     i;
146  
147 <        if (headidval(NULL, s))
147 >        if (isheadid(s))
148                  return(0);
149          if (isncomp(s)) {
150                  ip->ncomp = ncompval(s);
151 <                return(0);
151 >                return(ip->ncomp - 1);
152          }
153          if (!strncmp(s, "NROWS=", 6)) {
154                  ip->nrows = atoi(s+6);
155 <                return(0);
155 >                return(ip->nrows - 1);
156          }
157          if (!strncmp(s, "NCOLS=", 6)) {
158                  ip->ncols = atoi(s+6);
159 <                return(0);
159 >                return(ip->ncols - 1);
160          }
161          if ((i = isbigendian(s)) >= 0) {
162                  if (nativebigendian() != i)
# Line 165 | Line 168 | get_dminfo(char *s, void *p)
168          if (isexpos(s)) {
169                  float   f = exposval(s);
170                  scalecolor(ip->cexp, f);
171 <                return(0);
171 >                return(f > .0 ? 0 : -1);
172          }
173          if (iscolcor(s)) {
174                  COLOR   ctmp;
175 <                colcorval(ctmp, s);
175 >                if (!colcorval(ctmp, s)) return(-1);
176                  multcolor(ip->cexp, ctmp);
177                  return(0);
178          }
179 <        if (iswlsplit(s)) {
180 <                wlsplitval(ip->wlpart, s);
181 <                return(0);
179 <        }
179 >        if (iswlsplit(s))
180 >                return(wlsplitval(ip->wlpart, s) - 1);
181 >
182          if (!formatval(fmt, s)) {
183                  rmx_addinfo(ip, s);
184                  return(0);
# Line 186 | Line 188 | get_dminfo(char *s, void *p)
188                          ip->dtype = i;
189                          return(0);
190                  }
191 <        return(-1);
191 >        return(-1);             /* bad format */
192   }
193  
194   static int
195 < rmx_load_ascii(double *drp, const RMATRIX *rm, FILE *fp)
195 > rmx_load_ascii(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
196   {
197          int     j, k;
198  
# Line 202 | Line 204 | rmx_load_ascii(double *drp, const RMATRIX *rm, FILE *f
204   }
205  
206   static int
207 < rmx_load_float(double *drp, const RMATRIX *rm, FILE *fp)
207 > rmx_load_float(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
208   {
209          int     j, k;
210 <        float   val[100];
210 >        float   val[MAXCOMP];
211  
212 <        if (rm->ncomp > 100) {
212 >        if (rm->ncomp > MAXCOMP) {
213                  fputs("Unsupported # components in rmx_load_float()\n", stderr);
214                  exit(1);
215          }
# Line 223 | Line 225 | rmx_load_float(double *drp, const RMATRIX *rm, FILE *f
225   }
226  
227   static int
228 < rmx_load_double(double *drp, const RMATRIX *rm, FILE *fp)
228 > rmx_load_double(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
229   {
228 –        if (DTrmx_native != DTdouble)
229 –                return(0);
230          if (getbinary(drp, sizeof(*drp)*rm->ncomp, rm->ncols, fp) != rm->ncols)
231                  return(0);
232          if (rm->pflags & RMF_SWAPIN)
# Line 235 | Line 235 | rmx_load_double(double *drp, const RMATRIX *rm, FILE *
235   }
236  
237   static int
238 < rmx_load_rgbe(double *drp, const RMATRIX *rm, FILE *fp)
238 > rmx_load_rgbe(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
239   {
240          COLR    *scan;
241          COLOR   col;
# Line 258 | Line 258 | rmx_load_rgbe(double *drp, const RMATRIX *rm, FILE *fp
258   }
259  
260   static int
261 < rmx_load_spec(double *drp, const RMATRIX *rm, FILE *fp)
261 > rmx_load_spec(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
262   {
263 <        uby8    *scan;
264 <        SCOLOR  scol;
263 >        COLRV   *scan;
264 >        COLORV  scol[MAXCOMP];
265          int     j, k;
266  
267 <        if ((rm->ncomp < 3) | (rm->ncomp > MAXCSAMP))
267 >        if ((rm->ncomp < 3) | (rm->ncomp > MAXCOMP))
268                  return(0);
269 <        scan = (uby8 *)tempbuffer((rm->ncomp+1)*rm->ncols);
269 >        scan = (COLRV *)tempbuffer((rm->ncomp+1)*rm->ncols);
270          if (!scan)
271                  return(0);
272          if (freadscolrs(scan, rm->ncomp, rm->ncols, fp) < 0)
# Line 295 | Line 295 | rmx_load_header(RMATRIX *rm, FILE *fp)
295          }
296          rm->dtype = DTascii;                    /* assumed w/o FORMAT */
297          if (getheader(fp, get_dminfo, rm) < 0) {
298 <                fputs("Unrecognized matrix format\n", stderr);
298 >                fputs("Bad matrix header\n", stderr);
299                  return(0);
300          }
301          if ((rm->dtype == DTrgbe) | (rm->dtype == DTxyze) &&
# Line 311 | Line 311 | rmx_load_header(RMATRIX *rm, FILE *fp)
311          return(1);
312   }
313  
314 < /* Load next row as double (cannot be XML) */
314 > /* Load next row as rmx_dtype (cannot be XML) */
315   int
316 < rmx_load_row(double *drp, const RMATRIX *rm, FILE *fp)
316 > rmx_load_row(rmx_dtype *drp, const RMATRIX *rm, FILE *fp)
317   {
318          switch (rm->dtype) {
319          case DTascii:
# Line 342 | Line 342 | rmx_load_data(RMATRIX *rm, FILE *fp)
342          long    pos;            /* map memory for file > 1MB if possible */
343          if ((rm->dtype == DTrmx_native) & !(rm->pflags & RMF_SWAPIN) &
344                          (rmx_array_size(rm) >= 1L<<20) &&
345 <                        (pos = ftell(fp)) >= 0 && !(pos % sizeof(double))) {
345 >                        (pos = ftell(fp)) >= 0 && !(pos % sizeof(rmx_dtype))) {
346                  rm->mapped = mmap(NULL, rmx_array_size(rm)+pos, PROT_READ|PROT_WRITE,
347                                          MAP_PRIVATE, fileno(fp), 0);
348                  if (rm->mapped != MAP_FAILED) {
349                          if (rm->pflags & RMF_FREEMEM)
350                                  free(rm->mtx);
351 <                        rm->mtx = (double *)rm->mapped + pos/sizeof(double);
351 >                        rm->mtx = (rmx_dtype *)rm->mapped + pos/sizeof(rmx_dtype);
352                          rm->pflags &= ~RMF_FREEMEM;
353                          return(1);
354                  }               /* else fall back on reading into memory */
# Line 382 | Line 382 | rmx_load(const char *inspec, RMPref rmp)
382                  fp = stdin;
383          else if (inspec[0] == '!')
384                  fp = popen(inspec+1, "r");
385 <        else {
386 <                const char      *sp = inspec;   /* check suffix */
387 <                while (*sp)
388 <                        ++sp;
389 <                while (sp > inspec && sp[-1] != '.')
390 <                        --sp;
391 <                if (!strcasecmp(sp, "XML")) {   /* assume it's a BSDF */
385 >        else {                                  /* check suffix */
386 >                const char      *sp = strrchr(inspec, '.');
387 >                if (sp > inspec && !strcasecmp(sp+1, "XML")) {  /* BSDF? */
388                          CMATRIX *cm = rmp==RMPnone ? (CMATRIX *)NULL :
389                                          rmp==RMPtrans ? cm_loadBTDF(inspec) :
390                                          cm_loadBRDF(inspec, rmp==RMPreflB) ;
# Line 436 | Line 432 | rmx_load(const char *inspec, RMPref rmp)
432                                                  /* undo exposure? */
433          if ((dnew->cexp[0] != 1.f) |
434                          (dnew->cexp[1] != 1.f) | (dnew->cexp[2] != 1.f)) {
435 <                double  cmlt[MAXCSAMP];
435 >                double  cmlt[MAXCOMP];
436                  int     i;
437 <                if (dnew->ncomp > MAXCSAMP) {
437 >                if (dnew->ncomp > MAXCOMP) {
438                          fprintf(stderr, "Excess spectral components in: %s\n",
439                                          inspec);
440                          rmx_free(dnew);
# Line 456 | Line 452 | rmx_load(const char *inspec, RMPref rmp)
452   }
453  
454   static int
455 < rmx_write_ascii(const double *dp, int nc, int len, FILE *fp)
455 > rmx_write_ascii(const rmx_dtype *dp, int nc, int len, FILE *fp)
456   {
457          while (len-- > 0) {
458                  int     k = nc;
# Line 468 | Line 464 | rmx_write_ascii(const double *dp, int nc, int len, FIL
464   }
465  
466   static int
467 < rmx_write_float(const double *dp, int len, FILE *fp)
467 > rmx_write_float(const rmx_dtype *dp, int len, FILE *fp)
468   {
469          float   val;
470  
# Line 481 | Line 477 | rmx_write_float(const double *dp, int len, FILE *fp)
477   }
478  
479   static int
480 < rmx_write_rgbe(const double *dp, int nc, int len, FILE *fp)
480 > rmx_write_rgbe(const rmx_dtype *dp, int nc, int len, FILE *fp)
481   {
482          COLR    *scan;
483          int     j;
# Line 500 | Line 496 | rmx_write_rgbe(const double *dp, int nc, int len, FILE
496   }
497  
498   static int
499 < rmx_write_spec(const double *dp, int nc, int len, FILE *fp)
499 > rmx_write_spec(const rmx_dtype *dp, int nc, int len, FILE *fp)
500   {
501 <        uby8    *scan;
502 <        SCOLOR  scol;
501 >        COLRV   *scan;
502 >        COLORV  scol[MAXCOMP];
503          int     j, k;
504  
505 <        if (nc < 3) return(0);
506 <        scan = (uby8 *)tempbuffer((nc+1)*len);
505 >        if ((nc < 3) | (nc > MAXCOMP)) return(0);
506 >        scan = (COLRV *)tempbuffer((nc+1)*len);
507          if (!scan) return(0);
508          for (j = 0; j < len; j++, dp += nc) {
509                  for (k = nc; k--; )
# Line 588 | Line 584 | rmx_write_header(const RMATRIX *rm, int dtype, FILE *f
584  
585   /* Write out matrix data (usually by row) */
586   int
587 < rmx_write_data(const double *dp, int nc, int len, int dtype, FILE *fp)
587 > rmx_write_data(const rmx_dtype *dp, int nc, int len, int dtype, FILE *fp)
588   {
589          switch (dtype) {
590          case DTascii:
# Line 648 | Line 644 | rmx_identity(const int dim, const int n)
644                  return(NULL);
645          memset(rid->mtx, 0, rmx_array_size(rid));
646          for (i = dim; i--; ) {
647 <            double      *dp = rmx_lval(rid,i,i);
647 >            rmx_dtype   *dp = rmx_lval(rid,i,i);
648              for (k = n; k--; )
649                  dp[k] = 1.;
650          }
# Line 700 | Line 696 | rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int do
696                  munmap(rdst->mapped, rmx_mapped_size(rdst));
697          else
698   #endif
699 <        if (rdst->pflags & RMF_FREEMEM)
699 >        if (rdst->pflags & RMF_FREEMEM) {
700                  free(rdst->mtx);
701 +                rdst->pflags &= ~RMF_FREEMEM;
702 +        }
703          rdst->mapped = rsrc->mapped;
704          rdst->mtx = rsrc->mtx;
705 <        if (rsrc->pflags & RMF_FREEMEM)
708 <                rdst->pflags |= RMF_FREEMEM;
709 <        else
710 <                rdst->pflags &= ~RMF_FREEMEM;
705 >        rdst->pflags |= rsrc->pflags & RMF_FREEMEM;
706          rsrc->mapped = NULL; rsrc->mtx = NULL;
707          return(1);
708   }
# Line 742 | Line 737 | rmx_transpose(const RMATRIX *rm)
737          for (j = dnew->ncols; j--; )
738              for (i = dnew->nrows; i--; )
739                  memcpy(rmx_lval(dnew,i,j), rmx_val(rm,j,i),
740 <                                sizeof(double)*dnew->ncomp);
740 >                                sizeof(rmx_dtype)*dnew->ncomp);
741          return(dnew);
742   }
743  
# Line 767 | Line 762 | rmx_multiply(const RMATRIX *m1, const RMATRIX *m2)
762          for (i = mres->nrows; i--; )
763              for (j = mres->ncols; j--; )
764                  for (k = mres->ncomp; k--; ) {
765 <                    double      d = 0;
765 >                    rmx_dtype   d = 0;
766                      for (h = m1->ncols; h--; )
767                          d += rmx_val(m1,i,h)[k] * rmx_val(m2,h,j)[k];
768                      rmx_lval(mres,i,j)[k] = d;
# Line 795 | Line 790 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
790          for (i = m1->nrows; i--; )
791              for (j = m1->ncols; j--; )
792                  if (divide) {
793 <                    double      d;
793 >                    rmx_dtype   d;
794                      if (m2->ncomp == 1) {
795                          d = rmx_val(m2,i,j)[0];
796                          if (d == 0) {
# Line 818 | Line 813 | rmx_elemult(RMATRIX *m1, const RMATRIX *m2, int divide
813                          }
814                  } else {
815                      if (m2->ncomp == 1) {
816 <                        const double    d = rmx_val(m2,i,j)[0];
816 >                        const rmx_dtype d = rmx_val(m2,i,j)[0];
817                          for (k = m1->ncomp; k--; )
818                              rmx_lval(m1,i,j)[k] *= d;
819                      } else
# Line 859 | Line 854 | rmx_sum(RMATRIX *msum, const RMATRIX *madd, const doub
854                  rmx_addinfo(msum, rmx_mismatch_warn);
855          for (i = msum->nrows; i--; )
856              for (j = msum->ncols; j--; ) {
857 <                const double    *da = rmx_val(madd,i,j);
858 <                double          *ds = rmx_lval(msum,i,j);
857 >                const rmx_dtype *da = rmx_val(madd,i,j);
858 >                rmx_dtype       *ds = rmx_lval(msum,i,j);
859                  for (k = msum->ncomp; k--; )
860                       ds[k] += sf[k] * da[k];
861              }
# Line 879 | Line 874 | rmx_scale(RMATRIX *rm, const double sf[])
874                  return(0);
875          for (i = rm->nrows; i--; )
876              for (j = rm->ncols; j--; ) {
877 <                double  *dp = rmx_lval(rm,i,j);
877 >                rmx_dtype       *dp = rmx_lval(rm,i,j);
878                  for (k = rm->ncomp; k--; )
879                      dp[k] *= sf[k];
880              }
# Line 911 | Line 906 | rmx_transform(const RMATRIX *msrc, int n, const double
906          dnew->dtype = msrc->dtype;
907          for (i = dnew->nrows; i--; )
908              for (j = dnew->ncols; j--; ) {
909 <                const double    *ds = rmx_val(msrc,i,j);
909 >                const rmx_dtype *ds = rmx_val(msrc,i,j);
910                  for (kd = dnew->ncomp; kd--; ) {
911 <                    double      d = 0;
911 >                    rmx_dtype   d = 0;
912                      for (ks = msrc->ncomp; ks--; )
913                          d += cmat[kd*msrc->ncomp + ks] * ds[ks];
914                      rmx_lval(dnew,i,j)[kd] = d;
# Line 938 | Line 933 | rmx_from_cmatrix(const CMATRIX *cm)
933          for (i = dnew->nrows; i--; )
934              for (j = dnew->ncols; j--; ) {
935                  const COLORV    *cv = cm_lval(cm,i,j);
936 <                double          *dp = rmx_lval(dnew,i,j);
936 >                rmx_dtype       *dp = rmx_lval(dnew,i,j);
937                  dp[0] = cv[0];
938                  dp[1] = cv[1];
939                  dp[2] = cv[2];
# Line 953 | Line 948 | cm_from_rmatrix(const RMATRIX *rm)
948          int     i, j;
949          CMATRIX *cnew;
950  
951 <        if (!rm || !rm->mtx | (rm->ncomp == 2))
951 >        if (!rm || !rm->mtx | (rm->ncomp == 2) | (rm->ncomp > MAXCOMP))
952                  return(NULL);
953          cnew = cm_alloc(rm->nrows, rm->ncols);
954          if (!cnew)
955                  return(NULL);
956          for (i = cnew->nrows; i--; )
957              for (j = cnew->ncols; j--; ) {
958 <                const double    *dp = rmx_val(rm,i,j);
958 >                const rmx_dtype *dp = rmx_val(rm,i,j);
959                  COLORV          *cv = cm_lval(cnew,i,j);
960                  switch (rm->ncomp) {
961                  case 3:
# Line 970 | Line 965 | cm_from_rmatrix(const RMATRIX *rm)
965                      setcolor(cv, dp[0], dp[0], dp[0]);
966                      break;
967                  default: {
968 <                        SCOLOR  scol;
968 >                        COLORV  scol[MAXCOMP];
969                          int     k;
970                          for (k = rm->ncomp; k--; )
971                                  scol[k] = dp[k];

Diff Legend

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