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.75 by greg, Mon Dec 11 19:00:22 2023 UTC vs.
Revision 2.79 by greg, Sun May 19 15:32:24 2024 UTC

# Line 18 | Line 18 | static const char RCSid[] = "$Id$";
18  
19   static const char       rmx_mismatch_warn[] = "WARNING: data type mismatch\n";
20  
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 –
21   /* Initialize a RMATRIX struct but don't allocate array space */
22   RMATRIX *
23   rmx_new(int nr, int nc, int n)
# Line 53 | Line 50 | rmx_prepare(RMATRIX *rm)
50                  return(1);
51          if ((rm->nrows <= 0) | (rm->ncols <= 0) | (rm->ncomp <= 0))
52                  return(0);
53 <        rm->mtx = (double *)malloc(array_size(rm));
53 >        rm->mtx = (double *)malloc(rmx_array_size(rm));
54 >        rm->pflags |= RMF_OURMEM;
55          return(rm->mtx != NULL);
56   }
57  
# Line 79 | Line 77 | rmx_reset(RMATRIX *rm)
77                  free(rm->info);
78                  rm->info = NULL;
79          }
80 <        if (rm->mtx) {
80 >        if (rm->mtx && rm->pflags & RMF_OURMEM) {
81   #ifdef MAP_FILE
82                  if (rm->mapped) {
83 <                        munmap(rm->mapped, mapped_size(rm));
83 >                        munmap(rm->mapped, rmx_mapped_size(rm));
84                          rm->mapped = NULL;
85                  } else
86   #endif
87                          free(rm->mtx);
88 <                rm->mtx = NULL;
88 >                rm->pflags &= ~RMF_OURMEM;
89          }
90 +        rm->mtx = NULL;
91   }
92  
93   /* Free an RMATRIX struct and data */
# Line 157 | Line 156 | get_dminfo(char *s, void *p)
156                  return(0);
157          }
158          if ((i = isbigendian(s)) >= 0) {
159 <                ip->swapin = (nativebigendian() != i);
159 >                if (nativebigendian() != i)
160 >                        ip->pflags |= RMF_SWAPIN;
161 >                else
162 >                        ip->pflags &= ~RMF_SWAPIN;
163                  return(0);
164          }
165          if (isexpos(s)) {
# Line 212 | Line 214 | rmx_load_float(double *drp, const RMATRIX *rm, FILE *f
214          for (j = 0; j < rm->ncols; j++) {
215                  if (getbinary(val, sizeof(val[0]), rm->ncomp, fp) != rm->ncomp)
216                          return(0);
217 <                if (rm->swapin)
217 >                if (rm->pflags & RMF_SWAPIN)
218                          swap32((char *)val, rm->ncomp);
219                  for (k = 0; k < rm->ncomp; k++)
220                          *drp++ = val[k];
# Line 225 | Line 227 | rmx_load_double(double *drp, const RMATRIX *rm, FILE *
227   {
228          if (getbinary(drp, sizeof(*drp)*rm->ncomp, rm->ncols, fp) != rm->ncols)
229                  return(0);
230 <        if (rm->swapin)
230 >        if (rm->pflags & RMF_SWAPIN)
231                  swap64((char *)drp, rm->ncols*rm->ncomp);
232          return(1);
233   }
# Line 287 | Line 289 | rmx_load_header(RMATRIX *rm, FILE *fp)
289                  rm->ncomp = 3;
290                  setcolor(rm->cexp, 1.f, 1.f, 1.f);
291                  memcpy(rm->wlpart, WLPART, sizeof(rm->wlpart));
292 <                rm->swapin = 0;
292 >                rm->pflags = 0;
293          }
294          rm->dtype = DTascii;                    /* assumed w/o FORMAT */
295          if (getheader(fp, get_dminfo, rm) < 0) {
# Line 336 | Line 338 | rmx_load_data(RMATRIX *rm, FILE *fp)
338          int     i;
339   #ifdef MAP_FILE
340          long    pos;            /* map memory for file > 1MB if possible */
341 <        if ((rm->dtype == DTdouble) & !rm->swapin && array_size(rm) >= 1L<<20 &&
341 >        if ((rm->dtype == DTdouble) & !(rm->pflags & RMF_SWAPIN) &&
342 >                        rmx_array_size(rm) >= 1L<<20 &&
343                          (pos = ftell(fp)) >= 0 && !(pos % sizeof(double))) {
344 <                rm->mapped = mmap(NULL, array_size(rm)+pos, PROT_READ|PROT_WRITE,
344 >                rm->mapped = mmap(NULL, rmx_array_size(rm)+pos, PROT_READ|PROT_WRITE,
345                                          MAP_PRIVATE, fileno(fp), 0);
346                  if (rm->mapped != MAP_FAILED) {
347                          rm->mtx = (double *)rm->mapped + pos/sizeof(double);
348 +                        rm->pflags |= RMF_OURMEM;
349                          return(1);
350                  }               /* else fall back on reading into memory */
351                  rm->mapped = NULL;
# Line 349 | Line 353 | rmx_load_data(RMATRIX *rm, FILE *fp)
353   #endif
354          if (!rmx_prepare(rm)) { /* need in-core matrix array */
355                  fprintf(stderr, "Cannot allocate %g MByte matrix array\n",
356 <                                (1./(1L<<20))*(double)array_size(rm));
356 >                                (1./(1L<<20))*(double)rmx_array_size(rm));
357                  return(0);
358          }
359          for (i = 0; i < rm->nrows; i++)
# Line 393 | Line 397 | rmx_load(const char *inspec, RMPref rmp)
397                  }                               /* else open it ourselves */
398                  fp = fopen(inspec, "r");
399          }
400 <        if (!fp)
400 >        if (!fp) {
401 >                fprintf(stderr, "Cannot open for reading: %s\n", inspec);
402                  return(NULL);
403 +        }
404   #ifdef getc_unlocked
405          flockfile(fp);
406   #endif
# Line 499 | Line 505 | rmx_write_spec(const double *dp, int nc, int len, FILE
505          if (nc < 3) return(0);
506          scan = (uby8 *)tempbuffer((nc+1)*len);
507          if (!scan) return(0);
508 <        for (j = len; j--; dp += nc) {
508 >        for (j = 0; j < len; j++, dp += nc) {
509                  for (k = nc; k--; )
510                          scol[k] = dp[k];
511                  scolor2scolr(scan+j*(nc+1), scol, nc);
# Line 636 | Line 642 | rmx_identity(const int dim, const int n)
642  
643          if (!rid)
644                  return(NULL);
645 <        memset(rid->mtx, 0, array_size(rid));
645 >        memset(rid->mtx, 0, rmx_array_size(rid));
646          for (i = dim; i--; ) {
647              double      *dp = rmx_lval(rid,i,i);
648              for (k = n; k--; )
# Line 661 | Line 667 | rmx_copy(const RMATRIX *rm)
667                          rmx_free(dnew);
668                          return(NULL);
669                  }
670 <                memcpy(dnew->mtx, rm->mtx, array_size(dnew));
670 >                memcpy(dnew->mtx, rm->mtx, rmx_array_size(dnew));
671          }
672          rmx_addinfo(dnew, rm->info);
673          dnew->dtype = rm->dtype;
# Line 685 | Line 691 | rmx_transfer_data(RMATRIX *rdst, RMATRIX *rsrc, int do
691                  rsrc->info = NULL; rsrc->mapped = NULL; rsrc->mtx = NULL;
692                  return(1);
693          }
694 <        if (rdst->mapped)
695 <                return(0);      /* XXX can't handle this case */
696 <                                /* just matrix data -- leave metadata */
697 <        if (rdst->mtx) free(rdst->mtx);
694 >        if (rdst->pflags & RMF_OURMEM) {
695 > #ifdef MAP_FILE                 /* just matrix data -- leave metadata */
696 >                if (rdst->mapped)
697 >                        munmap(rdst->mapped, rmx_mapped_size(rdst));
698 >                else
699 > #endif
700 >                if (rdst->mtx)
701 >                        free(rdst->mtx);
702 >        }
703 >        rdst->mapped = rsrc->mapped;
704          rdst->mtx = rsrc->mtx;
705 <        rsrc->mtx = NULL;
705 >        if (rsrc->pflags & RMF_OURMEM)
706 >                rdst->pflags |= RMF_OURMEM;
707 >        else
708 >                rdst->pflags &= ~RMF_OURMEM;
709 >        rsrc->mapped = NULL; rsrc->mtx = NULL;
710          return(1);
711   }
712  

Diff Legend

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