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.78 by greg, Thu Feb 29 03:11:13 2024 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 638 | 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 663 | 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 687 | 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->pflags & RMF_OURMEM) {
695   #ifdef MAP_FILE                 /* just matrix data -- leave metadata */
696 <        if (rdst->mapped)
697 <                munmap(rdst->mapped, mapped_size(rdst));
698 <        else
696 >                if (rdst->mapped)
697 >                        munmap(rdst->mapped, rmx_mapped_size(rdst));
698 >                else
699   #endif
700 <        if (rdst->mtx)
701 <                free(rdst->mtx);
700 >                if (rdst->mtx)
701 >                        free(rdst->mtx);
702 >        }
703          rdst->mapped = rsrc->mapped;
704          rdst->mtx = rsrc->mtx;
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   }

Diff Legend

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