ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/util/rcollate.c
Revision: 2.39
Committed: Thu Mar 3 03:55:13 2022 UTC (2 years, 1 month ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.38: +3 -2 lines
Log Message:
feat: Added float and double alignment to i/o routines for future mmap() use

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.39 static const char RCSid[] = "$Id: rcollate.c,v 2.38 2022/01/20 17:35:03 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Utility to re-order records in a binary or ASCII data file (matrix)
6     */
7    
8     #include <stdlib.h>
9     #include <ctype.h>
10     #include "platform.h"
11     #include "rtio.h"
12     #include "resolu.h"
13 schorsch 2.23 #if defined(_WIN32) || defined(_WIN64)
14     #undef ftello
15     #define ftello ftell
16     #undef ssize_t
17     #define ssize_t size_t
18 greg 2.7 #else
19 schorsch 2.23 #include <sys/mman.h>
20 greg 2.1 #endif
21    
22 greg 2.31 #define MAXLEVELS 16 /* max RxC.. block pairs */
23    
24 greg 2.1 typedef struct {
25 greg 2.29 void *mapped; /* memory-mapped pointer */
26 greg 2.1 void *base; /* pointer to base memory */
27     size_t len; /* allocated memory length */
28     } MEMLOAD; /* file loaded/mapped into memory */
29    
30     typedef struct {
31     int nw_rec; /* number of words per record */
32 greg 2.37 ssize_t nrecs; /* number of records we found */
33 greg 2.1 char *rec[1]; /* record array (extends struct) */
34     } RECINDEX;
35    
36 greg 2.4 int warnings = 1; /* report warnings? */
37    
38 greg 2.1 /* free loaded file */
39     static void
40     free_load(MEMLOAD *mp)
41     {
42     if (mp == NULL || (mp->base == NULL) | (mp->len <= 0))
43     return;
44     #ifdef MAP_FILE
45     if (mp->mapped)
46 greg 2.29 munmap(mp->mapped, mp->len);
47 greg 2.1 else
48     #endif
49     free(mp->base);
50 greg 2.29 mp->mapped = NULL;
51 greg 2.1 mp->base = NULL;
52     mp->len = 0;
53     }
54    
55 greg 2.20 /* load memory from an input stream, starting from current position */
56     static int
57     load_stream(MEMLOAD *mp, FILE *fp)
58     {
59     size_t alloced = 0;
60     char buf[8192];
61     size_t nr;
62    
63     if (mp == NULL)
64     return(-1);
65 greg 2.29 mp->mapped = NULL;
66 greg 2.20 mp->base = NULL;
67     mp->len = 0;
68     if (fp == NULL)
69     return(-1);
70     while ((nr = fread(buf, 1, sizeof(buf), fp)) > 0) {
71     if (!alloced)
72 greg 2.21 mp->base = malloc(alloced = nr);
73 greg 2.20 else if (mp->len+nr > alloced)
74     mp->base = realloc(mp->base,
75     alloced = alloced*(2+(nr==sizeof(buf)))/2+nr);
76     if (mp->base == NULL)
77     return(-1);
78     memcpy((char *)mp->base + mp->len, buf, nr);
79     mp->len += nr;
80     }
81     if (ferror(fp)) {
82     free_load(mp);
83     return(-1);
84     }
85     if (alloced > mp->len*5/4) /* don't waste too much space */
86     mp->base = realloc(mp->base, mp->len);
87     return(mp->len > 0);
88     }
89    
90 greg 2.38 #if defined(_WIN32) || defined(_WIN64)
91     /* too difficult to fix this */
92     #define load_file load_stream
93     #else
94 greg 2.1 /* load a file into memory */
95     static int
96     load_file(MEMLOAD *mp, FILE *fp)
97     {
98     int fd;
99 greg 2.28 off_t skip, flen, fpos;
100 greg 2.1 if (mp == NULL)
101     return(-1);
102 greg 2.29 mp->mapped = NULL;
103 greg 2.1 mp->base = NULL;
104     mp->len = 0;
105     if (fp == NULL)
106     return(-1);
107     fd = fileno(fp);
108     skip = ftello(fp);
109     flen = lseek(fd, 0, SEEK_END);
110     if (flen <= skip)
111     return((int)(flen - skip));
112     mp->len = (size_t)(flen - skip);
113     #ifdef MAP_FILE
114 greg 2.38 if (mp->len >= 1L<<20) { /* map file if >= 1 MByte */
115 greg 2.29 mp->mapped = mmap(NULL, flen, PROT_READ, MAP_PRIVATE, fd, 0);
116     if (mp->mapped != MAP_FAILED) {
117     mp->base = (char *)mp->mapped + skip;
118 greg 2.1 return(1); /* mmap() success */
119     }
120 greg 2.29 mp->mapped = NULL; /* else fall back to reading it in... */
121 greg 2.1 }
122     #endif
123     if (lseek(fd, skip, SEEK_SET) != skip ||
124     (mp->base = malloc(mp->len)) == NULL) {
125     mp->len = 0;
126     return(-1);
127     }
128 greg 2.28 fpos = skip;
129     while (fpos < flen) { /* read() fails if n > 2 GBytes */
130     ssize_t nread = read(fd, (char *)mp->base+(fpos-skip),
131     (flen-fpos < 1L<<24) ? flen-fpos : 1L<<24);
132 greg 2.27 if (nread <= 0) {
133     free_load(mp);
134     return(-1);
135     }
136 greg 2.28 fpos += nread;
137 greg 2.1 }
138     return(1);
139     }
140 greg 2.38 #endif
141 greg 2.1
142     /* free a record index */
143     #define free_records(rp) free(rp)
144    
145     /* compute record index */
146     static RECINDEX *
147     index_records(const MEMLOAD *mp, int nw_rec)
148     {
149 greg 2.24 int nall = 0;
150 greg 2.1 RECINDEX *rp;
151     char *cp, *mend;
152     int n;
153    
154     if (mp == NULL || (mp->base == NULL) | (mp->len <= 0))
155     return(NULL);
156     if (nw_rec <= 0)
157     return(NULL);
158 greg 2.24 nall = 1000;
159     rp = (RECINDEX *)malloc(sizeof(RECINDEX) + nall*sizeof(char *));
160 greg 2.1 if (rp == NULL)
161     return(NULL);
162     rp->nw_rec = nw_rec;
163     rp->nrecs = 0;
164     cp = (char *)mp->base;
165     mend = cp + mp->len;
166     for ( ; ; ) { /* whitespace-separated words */
167     while (cp < mend && !*cp | isspace(*cp))
168     ++cp;
169     if (cp >= mend)
170     break;
171 greg 2.24 if (rp->nrecs >= nall) {
172     nall += nall>>1; /* get more record space */
173     rp = (RECINDEX *)realloc(rp,
174     sizeof(RECINDEX) + nall*sizeof(char *));
175     if (rp == NULL)
176     return(NULL);
177     }
178 greg 2.1 rp->rec[rp->nrecs++] = cp; /* point to first non-white */
179     n = rp->nw_rec;
180     while (++cp < mend) /* find end of record */
181     if (!*cp | isspace(*cp)) {
182     if (--n <= 0)
183     break; /* got requisite # words */
184     do { /* else find next word */
185     if (*cp == '\n') {
186     fprintf(stderr,
187     "Unexpected EOL in record!\n");
188     free_records(rp);
189     return(NULL);
190     }
191     if (++cp >= mend)
192     break;
193     } while (!*cp | isspace(*cp));
194     }
195     }
196     rp->rec[rp->nrecs] = mend; /* reallocate to save space */
197     rp = (RECINDEX *)realloc(rp,
198     sizeof(RECINDEX) + rp->nrecs*sizeof(char *));
199     return(rp);
200     }
201    
202     /* count number of columns based on first EOL */
203     static int
204     count_columns(const RECINDEX *rp)
205     {
206     char *cp = rp->rec[0];
207     char *mend = rp->rec[rp->nrecs];
208     int i;
209    
210     while (*cp != '\n')
211     if (++cp >= mend)
212     return(0);
213     for (i = 0; i < rp->nrecs; i++)
214     if (rp->rec[i] >= cp)
215     break;
216     return(i);
217     }
218    
219     /* copy nth record from index to stdout */
220     static int
221 greg 2.37 print_record(const RECINDEX *rp, ssize_t n)
222 greg 2.1 {
223     int words2go = rp->nw_rec;
224     char *scp;
225    
226     if ((n < 0) | (n >= rp->nrecs))
227     return(0);
228     scp = rp->rec[n];
229     do {
230     putc(*scp++, stdout);
231     if (!*scp | isspace(*scp)) {
232     if (--words2go <= 0)
233     break;
234     putc(' ', stdout); /* single space btwn. words */
235     do
236     if (++scp >= rp->rec[n+1])
237     break;
238     while (!*scp | isspace(*scp));
239     }
240     } while (scp < rp->rec[n+1]);
241     /* caller adds record sep. */
242     return(1);
243     }
244    
245     /* copy a stream to stdout */
246     static int
247     output_stream(FILE *fp)
248     {
249     char buf[8192];
250     ssize_t n;
251    
252     if (fp == NULL)
253     return(0);
254 greg 2.13 fflush(stdout);
255     while ((n = fread(buf, 1, sizeof(buf), fp)) > 0)
256 greg 2.1 if (write(fileno(stdout), buf, n) != n)
257     return(0);
258 greg 2.13 return(!ferror(fp));
259 greg 2.1 }
260    
261     /* get next word from stream, leaving stream on EOL or start of next word */
262     static char *
263     fget_word(char buf[256], FILE *fp)
264     {
265     int c;
266     char *cp;
267     /* skip nul's and white space */
268     while (!(c = getc(fp)) || isspace(c))
269     ;
270     if (c == EOF)
271     return(NULL);
272     cp = buf;
273     do
274     *cp++ = c;
275     while ((c = getc(fp)) != EOF && !isspace(c) && cp < buf+255);
276     *cp = '\0';
277     while (isspace(c) & (c != '\n'))
278     c = getc(fp);
279     if (c != EOF)
280     ungetc(c, fp);
281     return(buf);
282     }
283    
284 greg 2.9 char *fmtid = NULL; /* format id */
285     int comp_size = 0; /* binary bytes/channel */
286     int n_comp = 0; /* components/record */
287 greg 2.1 int ni_columns = 0; /* number of input columns */
288     int ni_rows = 0; /* number of input rows */
289     int no_columns = 0; /* number of output columns */
290     int no_rows = 0; /* number of output rows */
291 greg 2.16 int transpose = 0; /* transpose rows & cols? */
292     int i_header = 1; /* input header? */
293     int o_header = 1; /* output header? */
294 greg 2.31 int outArray[MAXLEVELS][2]; /* output block nesting */
295     int outLevels = 0; /* number of blocking levels */
296 greg 2.36 int check = 0; /* force data check? */
297 greg 2.31
298     /* parse RxCx... string */
299     static int
300     get_array(const char *spec, int blklvl[][2], int nlvls)
301     {
302     int n;
303    
304     if (nlvls <= 0) {
305     fputs("Too many block levels!\n", stderr);
306     exit(1);
307     }
308     if (sscanf(spec, "%dx%d", &blklvl[0][0], &blklvl[0][1]) != 2) {
309     fputs("Bad block specification!\n", stderr);
310     exit(1);
311     }
312     while (isdigit(*spec))
313     spec++;
314     spec++; /* 'x' */
315     while (isdigit(*spec))
316     spec++;
317     if ((*spec != 'x') & (*spec != 'X')) {
318     if (*spec) {
319     fputs("Blocks must be separated by 'x' or 'X'\n", stderr);
320     exit(1);
321     }
322     return(1);
323     }
324     spec++;
325     n = get_array(spec, blklvl+1, nlvls-1);
326     if (!n)
327     return(0);
328     blklvl[0][0] *= blklvl[1][0];
329     blklvl[0][1] *= blklvl[1][1];
330     return(n+1);
331     }
332 greg 2.1
333 greg 2.9 /* check settings and assign defaults */
334 greg 2.11 static int
335 greg 2.9 check_sizes()
336     {
337     if (fmtid == NULL) {
338     fmtid = "ascii";
339     } else if (!comp_size) {
340     if (!strcmp(fmtid, "float"))
341     comp_size = sizeof(float);
342     else if (!strcmp(fmtid, "double"))
343     comp_size = sizeof(double);
344     else if (!strcmp(fmtid, "byte"))
345     comp_size = 1;
346 greg 2.14 else if (strcmp(fmtid, "ascii")) {
347 greg 2.11 fprintf(stderr, "Unsupported format: %s\n", fmtid);
348     return(0);
349     }
350 greg 2.9 }
351 greg 2.16 if (transpose && (no_rows <= 0) & (no_columns <= 0)) {
352     if (ni_rows > 0) no_columns = ni_rows;
353     if (ni_columns > 0) no_rows = ni_columns;
354     } else if ((no_rows <= 0) & (no_columns > 0) &&
355     !((ni_rows*ni_columns) % no_columns))
356     no_rows = ni_rows*ni_columns/no_columns;
357 greg 2.9 if (n_comp <= 0)
358     n_comp = 3;
359 greg 2.11 return(1);
360 greg 2.9 }
361    
362 greg 2.31 /* call to compute block input position */
363 greg 2.37 static ssize_t
364 greg 2.31 get_block_pos(int r, int c, int blklvl[][2], int nlvls)
365     {
366 greg 2.37 ssize_t n = 0;
367 greg 2.31
368     while (nlvls > 1) {
369     int sr = r/blklvl[1][0];
370     int sc = c/blklvl[1][1];
371     r -= sr*blklvl[1][0];
372     c -= sc*blklvl[1][1];
373     n += sr*blklvl[1][0]*blklvl[0][1] + sc*blklvl[1][0]*blklvl[1][1];
374     blklvl++;
375     nlvls--;
376     }
377     n += r*blklvl[0][1] + c;
378     return(n);
379     }
380    
381     /* return input offset based on array ordering and transpose option */
382 greg 2.37 static ssize_t
383 greg 2.31 get_input_pos(int r, int c)
384     {
385 greg 2.37 ssize_t n;
386 greg 2.31
387     if (outLevels > 1) { /* block reordering */
388     n = get_block_pos(r, c, outArray, outLevels);
389     if (transpose) {
390 greg 2.34 r = n/ni_rows;
391     c = n - r*ni_rows;
392 greg 2.37 n = (ssize_t)c*ni_columns + r;
393 greg 2.31 }
394     } else if (transpose) /* transpose only */
395 greg 2.37 n = (ssize_t)c*ni_columns + r;
396 greg 2.31 else /* XXX should never happen! */
397 greg 2.37 n = (ssize_t)r*no_columns + c;
398 greg 2.31 return(n);
399     }
400    
401     /* output reordered ASCII or binary data from memory */
402 greg 2.1 static int
403 greg 2.31 do_reorder(const MEMLOAD *mp)
404 greg 2.1 {
405     static const char tabEOL[2] = {'\t','\n'};
406     RECINDEX *rp = NULL;
407 greg 2.37 ssize_t nrecords;
408 greg 2.1 int i, j;
409 greg 2.5 /* propogate sizes */
410     if (ni_rows <= 0)
411 greg 2.36 ni_rows = transpose ? no_columns : no_rows;
412 greg 2.5 if (ni_columns <= 0)
413 greg 2.36 ni_columns = transpose ? no_rows : no_columns;
414 greg 2.1 /* get # records (& index) */
415 greg 2.9 if (!comp_size) {
416     if ((rp = index_records(mp, n_comp)) == NULL)
417 greg 2.1 return(0);
418     if (ni_columns <= 0)
419     ni_columns = count_columns(rp);
420     nrecords = rp->nrecs;
421 greg 2.3 } else if ((ni_rows > 0) & (ni_columns > 0)) {
422 greg 2.37 nrecords = (ssize_t)ni_rows*ni_columns;
423 greg 2.9 if (nrecords > mp->len/(n_comp*comp_size)) {
424 greg 2.33 fputs("Input too small for specified size and type\n",
425     stderr);
426 greg 2.3 return(0);
427     }
428     } else
429 greg 2.9 nrecords = mp->len/(n_comp*comp_size);
430 greg 2.1 /* check sizes */
431     if ((ni_rows <= 0) & (ni_columns > 0))
432     ni_rows = nrecords/ni_columns;
433 greg 2.36 else if ((ni_columns <= 0) & (ni_rows > 0))
434 greg 2.1 ni_columns = nrecords/ni_rows;
435 greg 2.37 if (nrecords != (ssize_t)ni_rows*ni_columns)
436 greg 2.1 goto badspec;
437 greg 2.31 if (transpose) {
438     if (no_columns <= 0)
439     no_columns = ni_rows;
440     if (no_rows <= 0)
441     no_rows = ni_columns;
442 greg 2.34 if (outLevels <= 1 &&
443     (no_rows != ni_columns) | (no_columns != ni_rows))
444 greg 2.31 goto badspec;
445 greg 2.33 } else {
446     if (no_columns <= 0)
447     no_columns = ni_columns;
448     if (no_rows <= 0)
449     no_rows = ni_rows;
450     }
451     if (ni_rows*ni_columns != no_rows*no_columns) {
452     fputs("Number of input and output records do not match\n",
453     stderr);
454     return(0);
455 greg 2.31 }
456 greg 2.36 if (o_header) { /* finish header? */
457     printf("NROWS=%d\n", no_rows);
458     printf("NCOLS=%d\n", no_columns);
459 greg 2.39 fputformat(fmtid, stdout);
460 greg 2.36 fputc('\n', stdout);
461     }
462 greg 2.31 /* reorder records */
463 greg 2.1 for (i = 0; i < no_rows; i++) {
464 greg 2.31 for (j = 0; j < no_columns; j++) {
465 greg 2.37 ssize_t n = get_input_pos(i, j);
466 greg 2.33 if (n >= nrecords) {
467     fputs("Index past end-of-file\n", stderr);
468     return(0);
469     }
470 greg 2.1 if (rp != NULL) { /* ASCII output */
471 greg 2.31 print_record(rp, n);
472 greg 2.1 putc(tabEOL[j >= no_columns-1], stdout);
473     } else { /* binary output */
474 greg 2.31 putbinary((char *)mp->base + (n_comp*comp_size)*n,
475 greg 2.25 comp_size, n_comp, stdout);
476 greg 2.1 }
477 greg 2.31 }
478 greg 2.1 if (ferror(stdout)) {
479     fprintf(stderr, "Error writing to stdout\n");
480     return(0);
481     }
482     }
483     if (rp != NULL)
484     free_records(rp);
485     return(1);
486     badspec:
487 greg 2.31 fprintf(stderr, "Bad dimension(s)\n");
488 greg 2.1 return(0);
489     }
490    
491     /* resize ASCII stream input by ignoring EOLs between records */
492     static int
493     do_resize(FILE *fp)
494     {
495 greg 2.37 ssize_t records2go = ni_rows*ni_columns;
496 greg 2.1 int columns2go = no_columns;
497     char word[256];
498 greg 2.36
499     if (o_header) { /* finish header? */
500     if (no_rows > 0)
501     printf("NROWS=%d\n", no_rows);
502     if (no_columns > 0)
503     printf("NCOLS=%d\n", no_columns);
504 greg 2.39 fputformat(fmtid, stdout);
505 greg 2.36 fputc('\n', stdout);
506     }
507 greg 2.1 /* sanity checks */
508 greg 2.36 if (comp_size || !check &
509     (no_columns == ni_columns) & (no_rows == ni_rows))
510 greg 2.19 return(output_stream(fp)); /* no-op -- just copy */
511 greg 2.1 if (no_columns <= 0) {
512     fprintf(stderr, "Missing -oc specification\n");
513     return(0);
514     }
515     if ((records2go <= 0) & (no_rows > 0))
516     records2go = no_rows*no_columns;
517     else if (no_rows*no_columns != records2go) {
518     fprintf(stderr,
519 greg 2.36 "Number of input and output records disagree (%dx%d != %dx%d)\n",
520 greg 2.1 ni_rows, ni_columns, no_rows, no_columns);
521     return(0);
522     }
523     do { /* reshape records */
524     int n;
525    
526 greg 2.9 for (n = n_comp; n--; ) {
527 greg 2.1 if (fget_word(word, fp) == NULL) {
528 greg 2.9 if (records2go > 0 || n < n_comp-1)
529 greg 2.1 break;
530     goto done; /* normal EOD */
531     }
532     fputs(word, stdout);
533     if (n) { /* mid-record? */
534     int c = getc(fp);
535     if ((c == '\n') | (c == EOF))
536     break;
537     ungetc(c, fp);
538     putc(' ', stdout);
539     }
540     }
541     if (n >= 0) {
542     fprintf(stderr, "Incomplete record / unexpected EOF\n");
543     return(0);
544     }
545     if (--columns2go <= 0) { /* time to end output row? */
546     putc('\n', stdout);
547     columns2go = no_columns;
548     } else /* else separate records */
549     putc('\t', stdout);
550     } while (--records2go); /* expected EOD? */
551     done:
552 greg 2.4 if (warnings && columns2go != no_columns)
553 greg 2.1 fprintf(stderr, "Warning -- incomplete final row\n");
554 greg 2.4 if (warnings && fget_word(word, fp) != NULL)
555     fprintf(stderr, "Warning -- characters beyond expected EOD\n");
556 greg 2.1 return(1);
557     }
558    
559     /* process a header line and copy to stdout */
560     static int
561     headline(char *s, void *p)
562     {
563 greg 2.26 static char fmt[MAXFMTLEN];
564 greg 2.9 int n;
565 greg 2.1
566     if (formatval(fmt, s)) {
567 greg 2.9 if (fmtid == NULL) {
568     fmtid = fmt;
569     return(0);
570     }
571 greg 2.1 if (!strcmp(fmt, fmtid))
572     return(0);
573     fprintf(stderr, "Input format '%s' != '%s'\n", fmt, fmtid);
574     return(-1);
575     }
576 greg 2.9 if (!strncmp(s, "NROWS=", 6)) {
577     n = atoi(s+6);
578     if ((ni_rows > 0) & (n != ni_rows)) {
579     fputs("Incorrect input row count\n", stderr);
580     return(-1);
581     }
582     ni_rows = n;
583     return(0);
584     }
585     if (!strncmp(s, "NCOLS=", 6)) {
586     n = atoi(s+6);
587     if ((ni_columns > 0) & (n != ni_columns)) {
588     fputs("Incorrect input column count\n", stderr);
589     return(-1);
590     }
591     ni_columns = n;
592     return(0);
593     }
594     if (!strncmp(s, "NCOMP=", 6)) {
595     n = atoi(s+6);
596     if ((n_comp > 0) & (n != n_comp)) {
597 greg 2.14 fputs("Incorrect number of components\n", stderr);
598 greg 2.9 return(-1);
599     }
600     n_comp = n;
601     return(0);
602     }
603 greg 2.16 if (o_header)
604     fputs(s, stdout); /* copy header info. */
605 greg 2.1 return(0);
606     }
607    
608     /* main routine for converting rows/columns in data file */
609     int
610     main(int argc, char *argv[])
611     {
612 greg 2.14 int a;
613 greg 2.1
614 greg 2.14 for (a = 1; a < argc && argv[a][0] == '-'; a++)
615     switch (argv[a][1]) {
616 greg 2.1 case 'i': /* input */
617 greg 2.14 if (argv[a][2] == 'c') /* columns */
618     ni_columns = atoi(argv[++a]);
619     else if (argv[a][2] == 'r')
620     ni_rows = atoi(argv[++a]);
621 greg 2.1 else
622     goto userr;
623     break;
624     case 'o': /* output */
625 greg 2.14 if (argv[a][2] == 'c') /* columns */
626     no_columns = atoi(argv[++a]);
627     else if (argv[a][2] == 'r')
628     no_rows = atoi(argv[++a]);
629 greg 2.31 else if (argv[a][2] ||
630     !(outLevels=get_array(argv[++a], outArray, MAXLEVELS)))
631 greg 2.1 goto userr;
632     break;
633 greg 2.10 case 'h': /* turn off header */
634 greg 2.14 switch (argv[a][2]) {
635 greg 2.10 case 'i':
636     i_header = 0;
637     break;
638     case 'o':
639     o_header = 0;
640     break;
641     case '\0':
642     i_header = o_header = 0;
643     break;
644     default:
645     goto userr;
646     }
647 greg 2.1 break;
648     case 't': /* transpose on/off */
649     transpose = !transpose;
650     break;
651     case 'f': /* format */
652 greg 2.14 switch (argv[a][2]) {
653 greg 2.1 case 'a': /* ASCII */
654     case 'A':
655     fmtid = "ascii";
656 greg 2.9 comp_size = 0;
657 greg 2.1 break;
658     case 'f': /* float */
659     case 'F':
660     fmtid = "float";
661 greg 2.9 comp_size = sizeof(float);
662 greg 2.1 break;
663     case 'd': /* double */
664     case 'D':
665     fmtid = "double";
666 greg 2.9 comp_size = sizeof(double);
667 greg 2.1 break;
668     case 'b': /* binary (bytes) */
669     case 'B':
670     fmtid = "byte";
671 greg 2.9 comp_size = 1;
672 greg 2.1 break;
673     default:
674     goto userr;
675     }
676 greg 2.14 if (argv[a][3]) {
677     if (!isdigit(argv[a][3]))
678 greg 2.1 goto userr;
679 greg 2.14 n_comp = atoi(argv[a]+3);
680     } else
681     n_comp = 1;
682 greg 2.1 break;
683 greg 2.4 case 'w': /* warnings on/off */
684     warnings = !warnings;
685     break;
686 greg 2.36 case 'c': /* force check operation */
687     check = 1;
688     break;
689 greg 2.1 default:
690     goto userr;
691     }
692 greg 2.14 if (a < argc-1) /* arg count OK? */
693 greg 2.1 goto userr;
694 greg 2.31 if (outLevels) { /* should check consistency? */
695     no_rows = outArray[0][0];
696     no_columns = outArray[0][1];
697     }
698 greg 2.1 /* open input file? */
699 greg 2.14 if (a == argc-1 && freopen(argv[a], "r", stdin) == NULL) {
700     fprintf(stderr, "%s: cannot open for reading\n", argv[a]);
701 greg 2.1 return(1);
702     }
703 greg 2.9 if (comp_size) {
704 greg 2.1 SET_FILE_BINARY(stdin);
705     SET_FILE_BINARY(stdout);
706     }
707     /* check for no-op */
708 greg 2.36 if (!transpose & !check & (outLevels <= 1) & (i_header == o_header) &&
709 greg 2.19 (no_columns == ni_columns) & (no_rows == ni_rows)) {
710 greg 2.4 if (warnings)
711     fprintf(stderr, "%s: no-op -- copying input verbatim\n",
712 greg 2.1 argv[0]);
713     if (!output_stream(stdin))
714     return(1);
715     return(0);
716     }
717 greg 2.10 if (i_header) { /* read header */
718 greg 2.15 if (getheader(stdin, headline, NULL) < 0)
719 greg 2.1 return(1);
720 greg 2.11 if (!check_sizes())
721     return(1);
722 greg 2.9 if (comp_size) { /* a little late... */
723     SET_FILE_BINARY(stdin);
724     SET_FILE_BINARY(stdout);
725     }
726 greg 2.11 } else if (!check_sizes())
727     return(1);
728 greg 2.30 if (o_header) { /* write/add to header */
729     if (!i_header)
730     newheader("RADIANCE", stdout);
731 greg 2.14 printargs(a, argv, stdout);
732 greg 2.9 printf("NCOMP=%d\n", n_comp);
733 greg 2.10 }
734 greg 2.36 if (transpose | check | (outLevels > 1) || (o_header && no_rows <= 0)) {
735 greg 2.19 MEMLOAD myMem; /* need to map into memory */
736 greg 2.14 if (a == argc-1) {
737 greg 2.1 if (load_file(&myMem, stdin) <= 0) {
738     fprintf(stderr, "%s: error loading file into memory\n",
739 greg 2.14 argv[a]);
740 greg 2.1 return(1);
741     }
742     } else if (load_stream(&myMem, stdin) <= 0) {
743     fprintf(stderr, "%s: error loading stdin into memory\n",
744     argv[0]);
745     return(1);
746     }
747 greg 2.31 if (!do_reorder(&myMem))
748 greg 2.1 return(1);
749 greg 2.17 /* free_load(&myMem); about to exit, so don't bother */
750 greg 2.36 } else if (!do_resize(stdin)) /* just reshaping input */
751 greg 2.1 return(1);
752     return(0);
753     userr:
754     fprintf(stderr,
755 greg 2.36 "Usage: %s [-h[io]][-w][-c][-f[afdb][N]][-t][-ic in_col][-ir in_row][-oc out_col][-or out_row][-o RxC[xR1xC1..]] [input.dat]\n",
756 greg 2.1 argv[0]);
757     return(1);
758     }