src/px/ra_rgbe.c

#ifndef lint
static const char       RCSid[] = "$Id$";
#endif
/*
 *  program to convert from RADIANCE RLE to flat format
 */

#include  <stdio.h>
#include  <math.h>
#include  <time.h>
#include  "color.h"
#include  "resolu.h"

#ifdef MSDOS
#include  <fcntl.h>
#endif

extern int      addhline();

#define dumpheader(fp)  fwrite(headlines, 1, headlen, fp)

int  bradj = 0;                         /* brightness adjustment */

int  doflat = 1;                        /* produce flat file */

int  force = 0;                         /* force file overwrite? */

int  findframe = 0;                     /* find a specific frame? */

int  frameno = 0;                       /* current frame number */
int  fmterr = 0;                        /* got input format error */
char  *headlines;                       /* current header info. */
int  headlen;                           /* current header length */

char  *progname;


main(argc, argv)
int  argc;
char  *argv[];
{
        char    *ospec;
        int  i;

        progname = argv[0];

        for (i = 1; i < argc; i++)
                if (argv[i][0] == '-')
                        switch (argv[i][1]) {
                        case 'r':
                                doflat = !doflat;
                                break;
                        case 'e':
                                if (argv[i+1][0] != '+' && argv[i+1][0] != '-')
                                        goto userr;
                                bradj = atoi(argv[++i]);
                                break;
                        case 'n':
                                findframe = atoi(argv[++i]);
                                break;
                        case 'f':
                                force++;
                                break;
                        case '\0':
                                goto gotfile;
                        default:
                                goto userr;
                        }
                else
                        break;
gotfile:
        if (i < argc-2)
                goto userr;
        if (i <= argc-1 && strcmp(argv[i], "-") &&
                        freopen(argv[i], "r", stdin) == NULL) {
                fprintf(stderr, "%s: can't open input \"%s\"\n",
                                progname, argv[i]);
                exit(1);
        }
#ifdef MSDOS
        setmode(fileno(stdin), O_BINARY);
#endif
        ospec = i==argc-2 ? argv[i+1] : (char *)NULL;
        while (transfer(ospec))
                ;
        exit(0);
userr:
        fprintf(stderr,
        "Usage: %s [-r][-e +/-stops][-f][-n frame] [input [outspec]]\n",
                        progname);
        exit(1);
}


transfer(ospec)                 /* transfer a Radiance picture */
char    *ospec;
{
        char    oname[128];
        FILE    *fp;
        int     order;
        int     xmax, ymax;
        COLR    *scanin;
        int     y;
                                        /* get header info. */
        if (!(y = loadheader(stdin)))
                return(0);
        if (y < 0 || (order = fgetresolu(&xmax, &ymax, stdin)) < 0) {
                fprintf(stderr, "%s: bad input format\n", progname);
                exit(1);
        }
                                        /* did we pass the target frame? */
        if (findframe && findframe < frameno)
                return(0);
                                        /* allocate scanline */
        scanin = (COLR *)tempbuffer(xmax*sizeof(COLR));
        if (scanin == NULL) {
                perror(progname);
                exit(1);
        }
                                        /* skip frame? */
        if (findframe > frameno) {
                for (y = ymax; y--; )
                        if (freadcolrs(scanin, xmax, stdin) < 0) {
                                fprintf(stderr,
                                        "%s: error reading input picture\n",
                                                progname);
                                exit(1);
                        }
                return(1);
        }
                                        /* open output file/command */
        if (ospec == NULL) {
                strcpy(oname, "<stdout>");
                fp = stdout;
        } else {
                sprintf(oname, ospec, frameno);
                if (oname[0] == '!') {
                        if ((fp = popen(oname+1, "w")) == NULL) {
                                fprintf(stderr, "%s: cannot start \"%s\"\n",
                                                progname, oname);
                                exit(1);
                        }
                } else {
                        if (!force && access(oname, 0) >= 0) {
                                fprintf(stderr,
                                        "%s: output file \"%s\" exists\n",
                                                progname, oname);
                                exit(1);
                        }
                        if ((fp = fopen(oname, "w")) == NULL) {
                                fprintf(stderr, "%s: ", progname);
                                perror(oname);
                                exit(1);
                        }
                }
        }
#ifdef MSDOS
        setmode(fileno(fp), O_BINARY);
#endif
        dumpheader(fp);                 /* put out header */
        fputs(progname, fp);
        if (bradj)
                fprintf(fp, " -e %+d", bradj);
        if (!doflat)
                fputs(" -r", fp);
        fputc('\n', fp);
        if (bradj)
                fputexpos(pow(2.0, (double)bradj), fp);
        fputc('\n', fp);
        fputresolu(order, xmax, ymax, fp);
                                        /* transfer picture */
        for (y = ymax; y--; ) {
                if (freadcolrs(scanin, xmax, stdin) < 0) {
                        fprintf(stderr, "%s: error reading input picture\n",
                                        progname);
                        exit(1);
                }
                if (bradj)
                        shiftcolrs(scanin, xmax, bradj);
                if (doflat)
                        fwrite((char *)scanin, sizeof(COLR), xmax, fp);
                else
                        fwritecolrs(scanin, xmax, fp);
                if (ferror(fp)) {
                        fprintf(stderr, "%s: error writing output to \"%s\"\n",
                                        progname, oname);
                        exit(1);
                }
        }
                                        /* clean up */
        if (oname[0] == '!')
                pclose(fp);
        else if (ospec != NULL)
                fclose(fp);
        return(1);
}


int
addhline(s)                     /* add a line to our info. header */
char    *s;
{
        char    fmt[32];
        int     n;

        if (formatval(fmt, s))
                fmterr += !globmatch(PICFMT, fmt);
        else if (!strncmp(s, "FRAME=", 6))
                frameno = atoi(s+6);
        n = strlen(s);
        if (headlen)
                headlines = (char *)realloc((void *)headlines, headlen+n+1);
        else
                headlines = (char *)malloc(n+1);
        if (headlines == NULL) {
                perror(progname);
                exit(1);
        }
        strcpy(headlines+headlen, s);
        headlen += n;
        return(0);
}


loadheader(fp)                  /* load an info. header into memory */
FILE    *fp;
{
        fmterr = 0; frameno = 0;
        if (headlen) {                  /* free old header */
                free(headlines);
                headlen = 0;
        }
        if (getheader(fp, addhline, NULL) < 0)
                return(0);
        if (fmterr)
                return(-1);
        return(1);
}
Revision:	2.11
Committed:	Wed Apr 23 00:52:34 2003 UTC (21 years ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.10:	+1 -1 lines
Log Message:	Added (void *) cast to realloc calls
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.10	static const char RCSid[] = "$Id$";
3	greg	1.1	#endif
4			/*
5			* program to convert from RADIANCE RLE to flat format
6			*/
7
8			#include <stdio.h>
9	greg	2.4	#include <math.h>
10	greg	2.10	#include <time.h>
11	greg	2.4	#include "color.h"
12			#include "resolu.h"
13
14	greg	2.3	#ifdef MSDOS
15			#include <fcntl.h>
16			#endif
17	greg	1.1
18	gregl	2.8	extern int addhline();
19	greg	2.3
20	gregl	2.8	#define dumpheader(fp) fwrite(headlines, 1, headlen, fp)
21
22	greg	1.1	int bradj = 0; /* brightness adjustment */
23
24			int doflat = 1; /* produce flat file */
25
26	gregl	2.8	int force = 0; /* force file overwrite? */
27
28			int findframe = 0; /* find a specific frame? */
29
30			int frameno = 0; /* current frame number */
31			int fmterr = 0; /* got input format error */
32			char headlines; / current header info. */
33			int headlen; /* current header length */
34
35	greg	1.1	char *progname;
36
37
38			main(argc, argv)
39			int argc;
40			char *argv[];
41			{
42	gregl	2.8	char *ospec;
43	greg	1.1	int i;
44	gregl	2.7
45	greg	1.1	progname = argv[0];
46
47			for (i = 1; i < argc; i++)
48			if (argv[i][0] == '-')
49			switch (argv[i][1]) {
50			case 'r':
51			doflat = !doflat;
52			break;
53			case 'e':
54			if (argv[i+1][0] != '+' && argv[i+1][0] != '-')
55			goto userr;
56			bradj = atoi(argv[++i]);
57			break;
58	gregl	2.8	case 'n':
59			findframe = atoi(argv[++i]);
60			break;
61			case 'f':
62			force++;
63			break;
64			case '\0':
65			goto gotfile;
66	greg	1.1	default:
67			goto userr;
68			}
69			else
70			break;
71	gregl	2.8	gotfile:
72	greg	1.1	if (i < argc-2)
73			goto userr;
74	gregl	2.8	if (i <= argc-1 && strcmp(argv[i], "-") &&
75			freopen(argv[i], "r", stdin) == NULL) {
76	greg	1.1	fprintf(stderr, "%s: can't open input \"%s\"\n",
77			progname, argv[i]);
78			exit(1);
79			}
80	gregl	2.7	#ifdef MSDOS
81			setmode(fileno(stdin), O_BINARY);
82			#endif
83	gregl	2.8	ospec = i==argc-2 ? argv[i+1] : (char *)NULL;
84			while (transfer(ospec))
85			;
86	greg	1.1	exit(0);
87			userr:
88	gregl	2.8	fprintf(stderr,
89			"Usage: %s [-r][-e +/-stops][-f][-n frame] [input [outspec]]\n",
90	greg	1.1	progname);
91			exit(1);
92			}
93
94
95	gregl	2.8	transfer(ospec) /* transfer a Radiance picture */
96			char *ospec;
97	greg	1.1	{
98	gregl	2.8	char oname[128];
99			FILE *fp;
100	greg	1.2	int order;
101	greg	2.3	int xmax, ymax;
102	greg	1.1	COLR *scanin;
103			int y;
104	gregl	2.8	/* get header info. */
105			if (!(y = loadheader(stdin)))
106			return(0);
107			if (y < 0 \|\| (order = fgetresolu(&xmax, &ymax, stdin)) < 0) {
108			fprintf(stderr, "%s: bad input format\n", progname);
109			exit(1);
110			}
111			/* did we pass the target frame? */
112			if (findframe && findframe < frameno)
113			return(0);
114			/* allocate scanline */
115			scanin = (COLR )tempbuffer(xmaxsizeof(COLR));
116			if (scanin == NULL) {
117			perror(progname);
118			exit(1);
119			}
120			/* skip frame? */
121			if (findframe > frameno) {
122			for (y = ymax; y--; )
123			if (freadcolrs(scanin, xmax, stdin) < 0) {
124			fprintf(stderr,
125			"%s: error reading input picture\n",
126			progname);
127			exit(1);
128			}
129			return(1);
130			}
131			/* open output file/command */
132			if (ospec == NULL) {
133			strcpy(oname, "<stdout>");
134			fp = stdout;
135			} else {
136			sprintf(oname, ospec, frameno);
137			if (oname[0] == '!') {
138			if ((fp = popen(oname+1, "w")) == NULL) {
139			fprintf(stderr, "%s: cannot start \"%s\"\n",
140			progname, oname);
141			exit(1);
142			}
143			} else {
144			if (!force && access(oname, 0) >= 0) {
145			fprintf(stderr,
146			"%s: output file \"%s\" exists\n",
147			progname, oname);
148			exit(1);
149			}
150			if ((fp = fopen(oname, "w")) == NULL) {
151			fprintf(stderr, "%s: ", progname);
152			perror(oname);
153			exit(1);
154			}
155			}
156			}
157			#ifdef MSDOS
158			setmode(fileno(fp), O_BINARY);
159			#endif
160			dumpheader(fp); /* put out header */
161			fputs(progname, fp);
162	greg	1.1	if (bradj)
163	gregl	2.8	fprintf(fp, " -e %+d", bradj);
164	gregl	2.7	if (!doflat)
165	gregl	2.8	fputs(" -r", fp);
166			fputc('\n', fp);
167	greg	2.2	if (bradj)
168	gregl	2.8	fputexpos(pow(2.0, (double)bradj), fp);
169			fputc('\n', fp);
170			fputresolu(order, xmax, ymax, fp);
171			/* transfer picture */
172	gregl	2.7	for (y = ymax; y--; ) {
173	gregl	2.8	if (freadcolrs(scanin, xmax, stdin) < 0) {
174			fprintf(stderr, "%s: error reading input picture\n",
175			progname);
176			exit(1);
177			}
178	greg	1.1	if (bradj)
179			shiftcolrs(scanin, xmax, bradj);
180			if (doflat)
181	gregl	2.8	fwrite((char *)scanin, sizeof(COLR), xmax, fp);
182	greg	1.1	else
183	gregl	2.8	fwritecolrs(scanin, xmax, fp);
184			if (ferror(fp)) {
185			fprintf(stderr, "%s: error writing output to \"%s\"\n",
186			progname, oname);
187			exit(1);
188			}
189	greg	1.1	}
190	gregl	2.8	/* clean up */
191			if (oname[0] == '!')
192			pclose(fp);
193			else if (ospec != NULL)
194			fclose(fp);
195			return(1);
196			}
197
198
199	gwlarson	2.9	int
200	gregl	2.8	addhline(s) /* add a line to our info. header */
201			char *s;
202			{
203			char fmt[32];
204			int n;
205
206			if (formatval(fmt, s))
207			fmterr += !globmatch(PICFMT, fmt);
208			else if (!strncmp(s, "FRAME=", 6))
209			frameno = atoi(s+6);
210			n = strlen(s);
211			if (headlen)
212	greg	2.11	headlines = (char )realloc((void )headlines, headlen+n+1);
213	gregl	2.8	else
214			headlines = (char *)malloc(n+1);
215			if (headlines == NULL) {
216			perror(progname);
217			exit(1);
218			}
219			strcpy(headlines+headlen, s);
220			headlen += n;
221	gwlarson	2.9	return(0);
222	gregl	2.8	}
223
224
225			loadheader(fp) /* load an info. header into memory */
226			FILE *fp;
227			{
228			fmterr = 0; frameno = 0;
229			if (headlen) { /* free old header */
230			free(headlines);
231			headlen = 0;
232			}
233			if (getheader(fp, addhline, NULL) < 0)
234			return(0);
235			if (fmterr)
236			return(-1);
237			return(1);
238	greg	1.1	}