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

Comparing ray/src/util/cmatrix.c (file contents):
Revision 2.21 by greg, Mon Aug 12 18:28:37 2019 UTC vs.
Revision 2.23 by greg, Wed Oct 23 17:00:14 2019 UTC

# Line 81 | Line 81 | cm_resize(CMATRIX *cm, int nrows)
81  
82   typedef struct {
83          int     dtype;          /* data type */
84 +        int     need2swap;      /* need byte swap? */
85          int     nrows, ncols;   /* matrix size */
86          char    *err;           /* error message */
87   } CMINFO;               /* header info record */
# Line 104 | Line 105 | get_cminfo(char *s, void *p)
105                  ip->ncols = atoi(s+6);
106                  return(0);
107          }
108 +        if ((i = isbigendian(s)) >= 0) {
109 +                ip->need2swap = (nativebigendian() != i);
110 +                return(0);
111 +        }
112          if (!formatval(fmt, s))
113                  return(0);
114          for (i = 1; i < DTend; i++)
# Line 114 | Line 119 | get_cminfo(char *s, void *p)
119  
120   /* Load header to obtain/check data type and number of columns */
121   char *
122 < cm_getheader(int *dt, int *nr, int *nc, FILE *fp)
122 > cm_getheader(int *dt, int *nr, int *nc, int *swp, FILE *fp)
123   {
124          CMINFO  cmi;
125                                                  /* read header */
126          cmi.dtype = DTfromHeader;
127 +        cmi.need2swap = 0;
128          cmi.nrows = cmi.ncols = 0;
129          cmi.err = "unexpected EOF in header";
130          if (getheader(fp, get_cminfo, &cmi) < 0)
# Line 144 | Line 150 | cm_getheader(int *dt, int *nr, int *nc, FILE *fp)
150                  else if ((cmi.ncols > 0) & (*nc != cmi.ncols))
151                          return("unexpected column count in header");
152          }
153 +        if (swp)                                /* get/check swap? */
154 +                *swp = cmi.need2swap;
155          return(NULL);
156   }
157  
158 + /* Allocate and load image data into matrix */
159 + static CMATRIX *
160 + cm_load_rgbe(FILE *fp, int nrows, int ncols)
161 + {
162 +        CMATRIX *cm;
163 +        COLORV  *mp;
164 +                                                /* header already loaded */
165 +        if ((nrows <= 0) | (ncols <= 0) && !fscnresolu(&ncols, &nrows, fp)) {
166 +                error(USER, "bad picture resolution string");
167 +                return(NULL);
168 +        }
169 +        cm = cm_alloc(nrows, ncols);
170 +        if (!cm)
171 +                return(NULL);
172 +        mp = cm->cmem;
173 +        while (nrows--) {
174 +                if (freadscan((COLOR *)mp, ncols, fp) < 0) {
175 +                        error(USER, "error reading color picture as matrix");
176 +                        cm_free(cm);
177 +                        return(NULL);
178 +                }
179 +                mp += 3*ncols;
180 +        }                                       /* caller closes stream */
181 +        return(cm);
182 + }
183 +
184   /* Allocate and load a matrix from the given input (or stdin if NULL) */
185   CMATRIX *
186   cm_load(const char *inspec, int nrows, int ncols, int dtype)
187   {
188          const int       ROWINC = 2048;
189          FILE            *fp = stdin;
190 +        int             swap = 0;
191          CMATRIX         *cm;
192  
193          if (!inspec)
# Line 173 | Line 208 | cm_load(const char *inspec, int nrows, int ncols, int
208          if (dtype != DTascii)
209                  SET_FILE_BINARY(fp);            /* doesn't really work */
210          if (!dtype | !ncols) {                  /* expecting header? */
211 <                char    *err = cm_getheader(&dtype, &nrows, &ncols, fp);
211 >                char    *err = cm_getheader(&dtype, &nrows, &ncols, &swap, fp);
212                  if (err)
213                          error(USER, err);
214                  if (ncols <= 0)
# Line 184 | Line 219 | cm_load(const char *inspec, int nrows, int ncols, int
219          case DTfloat:
220          case DTdouble:
221                  break;
222 +        case DTrgbe:
223 +        case DTxyze:
224 +                cm = cm_load_rgbe(fp, nrows, ncols);
225 +                goto cleanup;
226          default:
227                  error(USER, "unexpected data type in cm_load()");
228          }
# Line 297 | Line 336 | cm_load(const char *inspec, int nrows, int ncols, int
336                                  error(WARNING, errmsg);
337                  }
338          }
339 +        if (swap) {
340 +                if (dtype == DTfloat)
341 +                        swap32((char *)cm->cmem, 3*cm->nrows*cm->ncols);
342 +                else if (dtype == DTdouble)
343 +                        swap64((char *)cm->cmem, 3*cm->nrows*cm->ncols);
344 +        }
345 + cleanup:
346          if (fp != stdin) {
347                  if (inspec[0] != '!')
348                          fclose(fp);
# Line 313 | Line 359 | cm_load(const char *inspec, int nrows, int ncols, int
359   EOFerror:
360          sprintf(errmsg, "unexpected EOF reading %s", inspec);
361          error(USER, errmsg);
362 +        return(NULL);
363   not_handled:
364          error(INTERNAL, "unhandled data size or length in cm_load()");
365          return(NULL);   /* gratis return */

Diff Legend

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