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.4 by greg, Thu May 29 17:28:09 2014 UTC vs.
Revision 2.15 by greg, Wed Feb 17 23:26:06 2016 UTC

# Line 11 | Line 11 | static const char RCSid[] = "$Id$";
11   #include "standard.h"
12   #include "cmatrix.h"
13   #include "platform.h"
14 + #include "paths.h"
15   #include "resolu.h"
16  
17   const char      *cm_fmt_id[] = {
18 <                        "unknown", "ascii", "float", "double",
19 <                        COLRFMT, CIEFMT
18 >                        "unknown", "ascii", COLRFMT, CIEFMT,
19 >                        "float", "double"
20                  };
21  
22   const int       cm_elem_size[] = {
# Line 39 | Line 40 | cm_alloc(int nrows, int ncols)
40          return(cm);
41   }
42  
43 + static void
44 + adjacent_ra_sizes(size_t bounds[2], size_t target)
45 + {
46 +        bounds[0] = 0; bounds[1] = 2048;
47 +        while (bounds[1] < target) {
48 +                bounds[0] = bounds[1];
49 +                bounds[1] += bounds[1]>>1;
50 +        }
51 + }
52 +
53   /* Resize color coefficient matrix */
54   CMATRIX *
55   cm_resize(CMATRIX *cm, int nrows)
56   {
57 +        size_t  old_size, new_size, ra_bounds[2];
58 +
59          if (nrows == cm->nrows)
60                  return(cm);
61          if (nrows <= 0) {
62                  cm_free(cm);
63                  return(NULL);
64          }
65 <        cm = (CMATRIX *)realloc(cm, sizeof(CMATRIX) +
66 <                        sizeof(COLOR)*(nrows*cm->ncols - 1));
67 <        if (cm == NULL)
68 <                error(SYSTEM, "out of memory in cm_resize()");
65 >        old_size = sizeof(CMATRIX) + sizeof(COLOR)*(cm->nrows*cm->ncols - 1);
66 >        adjacent_ra_sizes(ra_bounds, old_size);
67 >        new_size = sizeof(CMATRIX) + sizeof(COLOR)*(nrows*cm->ncols - 1);
68 >        if (nrows < cm->nrows ? new_size <= ra_bounds[0] :
69 >                                new_size > ra_bounds[1]) {
70 >                adjacent_ra_sizes(ra_bounds, new_size);
71 >                cm = (CMATRIX *)realloc(cm, ra_bounds[1]);
72 >                if (cm == NULL)
73 >                        error(SYSTEM, "out of memory in cm_resize()");
74 >        }
75          cm->nrows = nrows;
76          return(cm);
77   }
78  
79 + typedef struct {
80 +        int     dtype;          /* data type */
81 +        int     nrows, ncols;   /* matrix size */
82 +        char    *err;           /* error message */
83 + } CMINFO;               /* header info record */
84 +
85   static int
86 < getDT(char *s, void *p)
86 > get_cminfo(char *s, void *p)
87   {
88 +        CMINFO  *ip = (CMINFO *)p;
89          char    fmt[32];
90          int     i;
91 <        
91 >
92 >        if (!strncmp(s, "NCOMP=", 6) && atoi(s+6) != 3) {
93 >                ip->err = "unexpected # components (must be 3)";
94 >                return(-1);
95 >        }
96 >        if (!strncmp(s, "NROWS=", 6)) {
97 >                ip->nrows = atoi(s+6);
98 >                return(0);
99 >        }
100 >        if (!strncmp(s, "NCOLS=", 6)) {
101 >                ip->ncols = atoi(s+6);
102 >                return(0);
103 >        }
104          if (!formatval(fmt, s))
105                  return(0);
106          for (i = 1; i < DTend; i++)
107                  if (!strcmp(fmt, cm_fmt_id[i]))
108 <                        *((int *)p) = i;
108 >                        ip->dtype = i;
109          return(0);
110   }
111  
112 < /* Load header to obtain data type */
113 < int
114 < getDTfromHeader(FILE *fp)
112 > /* Load header to obtain/check data type and number of columns */
113 > char *
114 > cm_getheader(int *dt, int *nr, int *nc, FILE *fp)
115   {
116 <        int     dt = DTfromHeader;
117 <        
118 <        if (getheader(fp, getDT, &dt) < 0)
119 <                error(SYSTEM, "header read error");
120 <        if (dt == DTfromHeader)
121 <                error(USER, "missing data format in header");
122 <        return(dt);
116 >        CMINFO  cmi;
117 >                                                /* read header */
118 >        cmi.dtype = DTfromHeader;
119 >        cmi.nrows = cmi.ncols = 0;
120 >        cmi.err = "unexpected EOF in header";
121 >        if (getheader(fp, get_cminfo, &cmi) < 0)
122 >                return(cmi.err);
123 >        if (dt != NULL) {                       /* get/check data type? */
124 >                if (cmi.dtype == DTfromHeader) {
125 >                        if (*dt == DTfromHeader)
126 >                                return("missing/unknown data format in header");
127 >                } else if (*dt == DTfromHeader)
128 >                        *dt = cmi.dtype;
129 >                else if (*dt != cmi.dtype)
130 >                        return("unexpected data format in header");
131 >        }
132 >        if (nr != NULL) {                       /* get/check #rows? */
133 >                if (*nr <= 0)
134 >                        *nr = cmi.nrows;
135 >                else if ((cmi.nrows > 0) & (*nr != cmi.nrows))
136 >                        return("unexpected row count in header");
137 >        }
138 >        if (nc != NULL) {                       /* get/check #columns? */
139 >                if (*nc <= 0)
140 >                        *nc = cmi.ncols;
141 >                else if ((cmi.ncols > 0) & (*nc != cmi.ncols))
142 >                        return("unexpected column count in header");
143 >        }
144 >        return(NULL);
145   }
146  
147 < /* Allocate and load a matrix from the given file (or stdin if NULL) */
147 > /* Allocate and load a matrix from the given input (or stdin if NULL) */
148   CMATRIX *
149 < cm_load(const char *fname, int nrows, int ncols, int dtype)
149 > cm_load(const char *inspec, int nrows, int ncols, int dtype)
150   {
151 <        FILE    *fp = stdin;
152 <        CMATRIX *cm;
151 >        const int       ROWINC = 2048;
152 >        FILE            *fp = stdin;
153 >        CMATRIX         *cm;
154  
155 <        if (ncols <= 0)
156 <                error(USER, "Non-positive number of columns");
157 <        if (fname == NULL)
158 <                fname = "<stdin>";
159 <        else if ((fp = fopen(fname, "r")) == NULL) {
160 <                sprintf(errmsg, "cannot open file '%s'", fname);
155 >        if (inspec == NULL)
156 >                inspec = "<stdin>";
157 >        else if (inspec[0] == '!') {
158 >                fp = popen(inspec+1, "r");
159 >                if (fp == NULL) {
160 >                        sprintf(errmsg, "cannot start command '%s'", inspec);
161 >                        error(SYSTEM, errmsg);
162 >                }
163 >        } else if ((fp = fopen(inspec, "r")) == NULL) {
164 >                sprintf(errmsg, "cannot open file '%s'", inspec);
165                  error(SYSTEM, errmsg);
166          }
167   #ifdef getc_unlocked
# Line 104 | Line 169 | cm_load(const char *fname, int nrows, int ncols, int d
169   #endif
170          if (dtype != DTascii)
171                  SET_FILE_BINARY(fp);            /* doesn't really work */
172 <        if (dtype == DTfromHeader)
173 <                dtype = getDTfromHeader(fp);
172 >        if (!dtype | !ncols) {                  /* expecting header? */
173 >                char    *err = cm_getheader(&dtype, &nrows, &ncols, fp);
174 >                if (err != NULL)
175 >                        error(USER, err);
176 >                if (ncols <= 0)
177 >                        error(USER, "unspecified number of columns");
178 >        }
179          switch (dtype) {
180          case DTascii:
181          case DTfloat:
# Line 116 | Line 186 | cm_load(const char *fname, int nrows, int ncols, int d
186          }
187          if (nrows <= 0) {                       /* don't know length? */
188                  int     guessrows = 147;        /* usually big enough */
189 <                if ((dtype != DTascii) & (fp != stdin)) {
189 >                if ((dtype != DTascii) & (fp != stdin) & (inspec[0] != '!')) {
190                          long    startpos = ftell(fp);
191                          if (fseek(fp, 0L, SEEK_END) == 0) {
192                                  long    endpos = ftell(fp);
# Line 126 | Line 196 | cm_load(const char *fname, int nrows, int ncols, int d
196                                  if ((endpos - startpos) % (ncols*elemsiz)) {
197                                          sprintf(errmsg,
198                                          "improper length for binary file '%s'",
199 <                                                        fname);
199 >                                                        inspec);
200                                          error(USER, errmsg);
201                                  }
202                                  guessrows = (endpos - startpos)/(ncols*elemsiz);
203                                  if (fseek(fp, startpos, SEEK_SET) < 0) {
204                                          sprintf(errmsg,
205                                                  "fseek() error on file '%s'",
206 <                                                        fname);
206 >                                                        inspec);
207                                          error(SYSTEM, errmsg);
208                                  }
209                                  nrows = guessrows;      /* we're confident */
# Line 149 | Line 219 | cm_load(const char *fname, int nrows, int ncols, int d
219                  int     r, c;
220                  for (r = 0; r < maxrow; r++) {
221                      if (r >= cm->nrows)                 /* need more space? */
222 <                        cm = cm_resize(cm, 2*cm->nrows);
222 >                        cm = cm_resize(cm, cm->nrows+ROWINC);
223                      for (c = 0; c < ncols; c++) {
224                          COLORV  *cv = cm_lval(cm,r,c);
225 <                        if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3)
225 >                        if (fscanf(fp, COLSPEC, cv, cv+1, cv+2) != 3) {
226                                  if ((nrows <= 0) & (r > 0) & !c) {
227                                          cm = cm_resize(cm, maxrow=r);
228                                          break;
229                                  } else
230                                          goto EOFerror;
231 +                        }
232                      }
233                  }
234                  while ((c = getc(fp)) != EOF)
235                          if (!isspace(c)) {
236                                  sprintf(errmsg,
237 <                                "unexpected data at end of ascii file %s",
238 <                                                fname);
237 >                                "unexpected data at end of ascii input '%s'",
238 >                                                inspec);
239                                  error(WARNING, errmsg);
240                                  break;
241                          }
# Line 179 | Line 250 | cm_load(const char *fname, int nrows, int ncols, int d
250                                  if (nrows <= 0) {       /* unknown length */
251                                          if (nread == cm->nrows*cm->ncols)
252                                                          /* need more space? */
253 <                                                cm = cm_resize(cm, 2*cm->nrows);
253 >                                                cm = cm_resize(cm, cm->nrows+ROWINC);
254                                          else if (nread && !(nread % cm->ncols))
255                                                          /* seem to be  done */
256                                                  cm = cm_resize(cm, nread/cm->ncols);
# Line 218 | Line 289 | cm_load(const char *fname, int nrows, int ncols, int d
289                  }
290                  if (fgetc(fp) != EOF) {
291                                  sprintf(errmsg,
292 <                                "unexpected data at end of binary file %s",
293 <                                                fname);
292 >                                "unexpected data at end of binary input '%s'",
293 >                                                inspec);
294                                  error(WARNING, errmsg);
295                  }
296          }
297 <        if (fp != stdin)
298 <                fclose(fp);
297 >        if (fp != stdin) {
298 >                if (inspec[0] != '!')
299 >                        fclose(fp);
300 >                else if (pclose(fp)) {
301 >                        sprintf(errmsg, "error running command '%s'", inspec);
302 >                        error(WARNING, errmsg);
303 >                }
304 >        }
305   #ifdef getc_unlocked
306          else
307                  funlockfile(fp);
308   #endif
309          return(cm);
310   EOFerror:
311 <        sprintf(errmsg, "unexpected EOF reading %s", fname);
311 >        sprintf(errmsg, "unexpected EOF reading %s", inspec);
312          error(USER, errmsg);
313   not_handled:
314          error(INTERNAL, "unhandled data size or length in cm_load()");

Diff Legend

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