src/px/ra_xyze.c

#ifndef lint
static const char       RCSid[] = "$Id: ra_xyze.c,v 2.13 2019/12/28 18:05:14 greg Exp $";
#endif
/*
 *  Program to convert between RADIANCE RGBE and XYZE formats
 *  Added white-balance adjustment 10/01 (GW).
 */

#include  <math.h>

#include  "platform.h"
#include  "color.h"
#include  "resolu.h"
#include  "rtio.h"

enum {InpUNK, InpRGB, InpXYZ, InpSPEC}; /* input format */
int  infmt = InpUNK;
int  rgbout = 0;                        /* output should be RGBE? */
RGBPRIMS  inprims = STDPRIMS;           /* input primaries */
RGBPRIMS  outprims = STDPRIMS;          /* output primaries */
double  expcomp = 1.0;                  /* exposure compensation */
int  doflat = -1;                       /* produce flat file? */
double  origexp = -1.0;                 /* original exposure */
char  *progname;

static gethfunc headline;
static void quiterr(char *err);
static void myreadscan(COLOR *scn, int len);
static void convert(void);


static int
headline(                               /* process header line */
        char    *s,
        void    *p
)
{
        char    fmt[MAXFMTLEN];

        if (formatval(fmt, s)) {        /* check if format string */
                if (!strcmp(fmt,COLRFMT))
                        infmt = InpRGB;
                else if (!strcmp(fmt,CIEFMT))
                        infmt = InpXYZ;
                else if (!strcmp(fmt,SPECFMT))
                        infmt = InpSPEC;
                else
                        infmt = InpUNK;
                return(0);              /* don't echo */
        }
        if (origexp > 0.0 && isexpos(s)) {
                origexp *= exposval(s);
                return(0);              /* don't echo */
        }
        if (isprims(s)) {               /* get input primaries */
                primsval(inprims, s);
                return(0);              /* don't echo */
        }
        if (iswlsplit(s)) {             /* get wavelength limits */
                wlsplitval(WLPART, s);
                return(0);              /* don't echo */
        }
        if (isncomp(s)) {               /* number of color samples */
                NCSAMP = ncompval(s);
                return(0);              /* don't echo */
        }
                                        /* should I grok colcorr also? */
        return(fputs(s, stdout));
}


int
main(int  argc, char  *argv[])
{
        int  i;
        SET_DEFAULT_BINARY();
        SET_FILE_BINARY(stdin);
        SET_FILE_BINARY(stdout);
        progname = argv[0];

        for (i = 1; i < argc; i++)
                if (argv[i][0] == '-')
                        switch (argv[i][1]) {
                        case 'c':               /* rle-compressed output */
                                doflat = 0;
                                break;
                        case 'u':               /* flat output */
                                doflat = 1;
                                break;
                        case 'r':               /* RGBE output */
                                rgbout = 1;
                                break;
                        case 'p':               /* RGB primaries */
                                if (i+8 >= argc)
                                        goto userr;
                                outprims[RED][CIEX] = atof(argv[++i]);
                                outprims[RED][CIEY] = atof(argv[++i]);
                                outprims[GRN][CIEX] = atof(argv[++i]);
                                outprims[GRN][CIEY] = atof(argv[++i]);
                                outprims[BLU][CIEX] = atof(argv[++i]);
                                outprims[BLU][CIEY] = atof(argv[++i]);
                                outprims[WHT][CIEX] = atof(argv[++i]);
                                outprims[WHT][CIEY] = atof(argv[++i]);
                                break;
                        case 'o':               /* original exposure */
                                origexp = 1.0;
                                break;
                        case 'e':               /* exposure compensation */
                                expcomp = atof(argv[++i]);
                                if (argv[i][0] == '+' || argv[i][0] == '-')
                                        expcomp = pow(2., expcomp);
                                break;
                        default:
                                goto userr;
                        }
                else
                        break;

        if (doflat < 0)
                doflat = !rgbout;
        if (i < argc-2)
                goto userr;
        if (i <= argc-1 && freopen(argv[i], "r", stdin) == NULL) {
                fprintf(stderr, "%s: can't open input \"%s\"\n",
                                progname, argv[i]);
                exit(1);
        }
        if (i == argc-2 && freopen(argv[i+1], "w", stdout) == NULL) {
                fprintf(stderr, "%s: can't open output \"%s\"\n",
                                progname, argv[i+1]);
                exit(1);
        }
        getheader(stdin, headline, NULL);
        if (infmt == InpUNK)
                quiterr("unrecognized/missing input file format");
        printargs(argc, argv, stdout);          /* add to header */
        convert();                              /* convert picture */
        exit(0);
userr:
        fprintf(stderr, "Usage: %s [-r][-o][-e exp][-c|-u]", progname);
        fprintf(stderr, "[-p rx ry gx gy bx by wx wy] [input [output]]\n");
        exit(1);
}


static void
quiterr(                /* print message and exit */
        char  *err
)
{
        if (err != NULL) {
                fprintf(stderr, "%s: %s\n", progname, err);
                exit(1);
        }
        exit(0);
}


static void
myreadscan(COLOR *scn, int len)
{
        if (infmt == InpSPEC) {         /* read & convert to XYZ */
                SCOLR   sclr;
                SCOLOR  scol;
                while (len-- > 0) {
                        if (getbinary(sclr, LSCOLR, 1, stdin) != 1)
                                goto readerr;
                        scolr_scolor(scol, sclr);
                        scolor_cie(*scn++, scol);
                }
                return;
        }                               /* else read as RGBE/XYZE */
        if (freadscan(scn, len, stdin) >= 0)
                return;
readerr:
        quiterr("error reading input picture");
}


static void
convert(void)                           /* convert to XYZE or RGBE picture */
{
        int     order;
        int     xmax, ymax;
        COLORMAT        xfm;
        COLOR   *scanin;
        COLR    *scanout;
        double  exp2do = expcomp;
        double  exp2report = expcomp;
        int     y;
        int     x;
                                                /* recover original? */
        if (origexp > 0.0)
                exp2do /= origexp;
                                                /* compute transform */
        if (rgbout) {
                if (infmt == InpRGB) {          /* RGBE -> RGBE */
                        comprgb2rgbWBmat(xfm, inprims, outprims);
                } else {                        /* XYZE/Spectral -> RGBE */
                        compxyz2rgbWBmat(xfm, outprims);
                }
                if (infmt == InpXYZ) {
                        if (origexp > 0.0)
                                exp2do /= WHTEFFICACY;
                        else
                                exp2report *= WHTEFFICACY;
                }
        } else {
                if (infmt == InpRGB) {          /* RGBE -> XYZE */
                        comprgb2xyzWBmat(xfm, inprims);
                } else {                        /* XYZE/Spectral -> XYZE */
                        memset(xfm, 0, sizeof(xfm));
                        for (x = 3; x--; )
                                xfm[x][x] = 1.;
                }
                if (infmt != InpXYZ) {
                        if (origexp > 0)
                                exp2do *= WHTEFFICACY;
                        else
                                exp2report /= WHTEFFICACY;
                }
        }
        for (y = 0; y < 3; y++)
                for (x = 0; x < 3; x++)
                        xfm[y][x] *= exp2do;
                                                /* get input resolution */
        if ((order = fgetresolu(&xmax, &ymax, stdin)) < 0)
                quiterr("bad picture format");
                                                /* complete output header */
        if ((exp2report < 0.99) | (exp2report > 1.01))
                fputexpos(exp2report, stdout);
        if (rgbout) {
                fputprims(outprims, stdout);
                fputformat(COLRFMT, stdout);
        } else
                fputformat(CIEFMT, stdout);
        putc('\n', stdout);
        fputresolu(order, xmax, ymax, stdout);
                                                /* allocate scanline */
        scanin = (COLOR *)malloc(xmax*sizeof(COLOR));
        if (scanin == NULL)
                quiterr("out of memory in convert");
        scanout = doflat ? (COLR *)malloc(xmax*sizeof(COLR)) : (COLR *)NULL;
                                                /* convert image */
        for (y = 0; y < ymax; y++) {
                myreadscan(scanin, xmax);
                for (x = 0; x < xmax; x++) {
                        colortrans(scanin[x], xfm, scanin[x]);
                        if (rgbout)
                                clipgamut(scanin[x], bright(scanin[x]),
                                                CGAMUT_LOWER, cblack, cwhite);
                }
                if (scanout != NULL) {
                        for (x = 0; x < xmax; x++)
                                setcolr(scanout[x], colval(scanin[x],RED),
                                                colval(scanin[x],GRN),
                                                colval(scanin[x],BLU));
                        putbinary(scanout, sizeof(COLR), xmax, stdout);
                } else
                        fwritescan(scanin, xmax, stdout);
                if (ferror(stdout))
                        quiterr("error writing output picture");
        }
                                                /* free scanline */
        free(scanin);
        if (scanout != NULL)
                free(scanout);
}
Revision:	2.14
Committed:	Tue Oct 1 01:16:26 2024 UTC (8 months, 4 weeks ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.13:	+60 -26 lines
Log Message:	feat(ra_xyze): added accurate conversion of hyperspectral pictures on input
#	User	Rev	Content
1	greg	2.1	#ifndef lint
2	greg	2.14	static const char RCSid[] = "$Id: ra_xyze.c,v 2.13 2019/12/28 18:05:14 greg Exp $";
3	greg	2.1	#endif
4			/*
5			* Program to convert between RADIANCE RGBE and XYZE formats
6	greg	2.6	* Added white-balance adjustment 10/01 (GW).
7	greg	2.1	*/
8
9			#include <math.h>
10	schorsch	2.7
11			#include "platform.h"
12	greg	2.1	#include "color.h"
13			#include "resolu.h"
14	greg	2.11	#include "rtio.h"
15	greg	2.1
16	greg	2.14	enum {InpUNK, InpRGB, InpXYZ, InpSPEC}; /* input format */
17			int infmt = InpUNK;
18	greg	2.1	int rgbout = 0; /* output should be RGBE? */
19			RGBPRIMS inprims = STDPRIMS; /* input primaries */
20			RGBPRIMS outprims = STDPRIMS; /* output primaries */
21			double expcomp = 1.0; /* exposure compensation */
22			int doflat = -1; /* produce flat file? */
23	greg	2.10	double origexp = -1.0; /* original exposure */
24	greg	2.1	char *progname;
25
26	schorsch	2.8	static gethfunc headline;
27	schorsch	2.9	static void quiterr(char *err);
28	greg	2.14	static void myreadscan(COLOR *scn, int len);
29	schorsch	2.9	static void convert(void);
30
31	greg	2.1
32	schorsch	2.8
33			static int
34			headline( /* process header line */
35			char *s,
36			void *p
37			)
38	greg	2.1	{
39	greg	2.12	char fmt[MAXFMTLEN];
40	greg	2.1
41			if (formatval(fmt, s)) { /* check if format string */
42			if (!strcmp(fmt,COLRFMT))
43	greg	2.14	infmt = InpRGB;
44	greg	2.1	else if (!strcmp(fmt,CIEFMT))
45	greg	2.14	infmt = InpXYZ;
46			else if (!strcmp(fmt,SPECFMT))
47			infmt = InpSPEC;
48	greg	2.1	else
49	greg	2.14	infmt = InpUNK;
50	gwlarson	2.5	return(0); /* don't echo */
51	greg	2.1	}
52	greg	2.10	if (origexp > 0.0 && isexpos(s)) {
53			origexp *= exposval(s);
54			return(0); /* don't echo */
55			}
56	greg	2.1	if (isprims(s)) { /* get input primaries */
57			primsval(inprims, s);
58	gwlarson	2.5	return(0); /* don't echo */
59	greg	2.1	}
60	greg	2.14	if (iswlsplit(s)) { /* get wavelength limits */
61			wlsplitval(WLPART, s);
62			return(0); /* don't echo */
63			}
64			if (isncomp(s)) { /* number of color samples */
65			NCSAMP = ncompval(s);
66			return(0); /* don't echo */
67			}
68	greg	2.1	/* should I grok colcorr also? */
69	gwlarson	2.5	return(fputs(s, stdout));
70	greg	2.1	}
71
72
73	schorsch	2.9	int
74			main(int argc, char *argv[])
75	greg	2.1	{
76			int i;
77	schorsch	2.7	SET_DEFAULT_BINARY();
78			SET_FILE_BINARY(stdin);
79			SET_FILE_BINARY(stdout);
80	greg	2.1	progname = argv[0];
81
82			for (i = 1; i < argc; i++)
83			if (argv[i][0] == '-')
84			switch (argv[i][1]) {
85			case 'c': /* rle-compressed output */
86			doflat = 0;
87			break;
88			case 'u': /* flat output */
89			doflat = 1;
90			break;
91			case 'r': /* RGBE output */
92			rgbout = 1;
93			break;
94			case 'p': /* RGB primaries */
95	greg	2.2	if (i+8 >= argc)
96	greg	2.1	goto userr;
97			outprims[RED][CIEX] = atof(argv[++i]);
98			outprims[RED][CIEY] = atof(argv[++i]);
99			outprims[GRN][CIEX] = atof(argv[++i]);
100			outprims[GRN][CIEY] = atof(argv[++i]);
101			outprims[BLU][CIEX] = atof(argv[++i]);
102			outprims[BLU][CIEY] = atof(argv[++i]);
103			outprims[WHT][CIEX] = atof(argv[++i]);
104			outprims[WHT][CIEY] = atof(argv[++i]);
105			break;
106	greg	2.10	case 'o': /* original exposure */
107			origexp = 1.0;
108			break;
109	greg	2.1	case 'e': /* exposure compensation */
110			expcomp = atof(argv[++i]);
111			if (argv[i][0] == '+' \|\| argv[i][0] == '-')
112			expcomp = pow(2., expcomp);
113			break;
114			default:
115			goto userr;
116			}
117			else
118			break;
119
120			if (doflat < 0)
121			doflat = !rgbout;
122			if (i < argc-2)
123			goto userr;
124			if (i <= argc-1 && freopen(argv[i], "r", stdin) == NULL) {
125			fprintf(stderr, "%s: can't open input \"%s\"\n",
126			progname, argv[i]);
127			exit(1);
128			}
129			if (i == argc-2 && freopen(argv[i+1], "w", stdout) == NULL) {
130			fprintf(stderr, "%s: can't open output \"%s\"\n",
131			progname, argv[i+1]);
132			exit(1);
133			}
134			getheader(stdin, headline, NULL);
135	greg	2.14	if (infmt == InpUNK)
136			quiterr("unrecognized/missing input file format");
137	greg	2.1	printargs(argc, argv, stdout); /* add to header */
138			convert(); /* convert picture */
139			exit(0);
140			userr:
141	greg	2.10	fprintf(stderr, "Usage: %s [-r][-o][-e exp][-c\|-u]", progname);
142	greg	2.1	fprintf(stderr, "[-p rx ry gx gy bx by wx wy] [input [output]]\n");
143			exit(1);
144			}
145
146
147	schorsch	2.9	static void
148			quiterr( /* print message and exit */
149			char *err
150			)
151	greg	2.1	{
152			if (err != NULL) {
153			fprintf(stderr, "%s: %s\n", progname, err);
154			exit(1);
155			}
156			exit(0);
157			}
158
159
160	schorsch	2.9	static void
161	greg	2.14	myreadscan(COLOR *scn, int len)
162			{
163			if (infmt == InpSPEC) { /* read & convert to XYZ */
164			SCOLR sclr;
165			SCOLOR scol;
166			while (len-- > 0) {
167			if (getbinary(sclr, LSCOLR, 1, stdin) != 1)
168			goto readerr;
169			scolr_scolor(scol, sclr);
170			scolor_cie(*scn++, scol);
171			}
172			return;
173			} /* else read as RGBE/XYZE */
174			if (freadscan(scn, len, stdin) >= 0)
175			return;
176			readerr:
177			quiterr("error reading input picture");
178			}
179
180
181			static void
182	schorsch	2.9	convert(void) /* convert to XYZE or RGBE picture */
183	greg	2.1	{
184			int order;
185			int xmax, ymax;
186			COLORMAT xfm;
187	greg	2.14	COLOR *scanin;
188			COLR *scanout;
189	greg	2.10	double exp2do = expcomp;
190			double exp2report = expcomp;
191	greg	2.1	int y;
192	greg	2.14	int x;
193	greg	2.10	/* recover original? */
194			if (origexp > 0.0)
195			exp2do /= origexp;
196	greg	2.1	/* compute transform */
197			if (rgbout) {
198	greg	2.14	if (infmt == InpRGB) { /* RGBE -> RGBE */
199	greg	2.6	comprgb2rgbWBmat(xfm, inprims, outprims);
200	greg	2.14	} else { /* XYZE/Spectral -> RGBE */
201	greg	2.6	compxyz2rgbWBmat(xfm, outprims);
202	greg	2.14	}
203			if (infmt == InpXYZ) {
204	greg	2.10	if (origexp > 0.0)
205			exp2do /= WHTEFFICACY;
206			else
207			exp2report *= WHTEFFICACY;
208	greg	2.1	}
209			} else {
210	greg	2.14	if (infmt == InpRGB) { /* RGBE -> XYZE */
211	greg	2.6	comprgb2xyzWBmat(xfm, inprims);
212	greg	2.14	} else { /* XYZE/Spectral -> XYZE */
213			memset(xfm, 0, sizeof(xfm));
214			for (x = 3; x--; )
215			xfm[x][x] = 1.;
216			}
217			if (infmt != InpXYZ) {
218			if (origexp > 0)
219	greg	2.10	exp2do *= WHTEFFICACY;
220			else
221			exp2report /= WHTEFFICACY;
222	greg	2.1	}
223			}
224	gregl	2.4	for (y = 0; y < 3; y++)
225			for (x = 0; x < 3; x++)
226	greg	2.10	xfm[y][x] *= exp2do;
227	greg	2.1	/* get input resolution */
228			if ((order = fgetresolu(&xmax, &ymax, stdin)) < 0)
229			quiterr("bad picture format");
230			/* complete output header */
231	greg	2.10	if ((exp2report < 0.99) \| (exp2report > 1.01))
232			fputexpos(exp2report, stdout);
233	greg	2.1	if (rgbout) {
234			fputprims(outprims, stdout);
235			fputformat(COLRFMT, stdout);
236			} else
237			fputformat(CIEFMT, stdout);
238			putc('\n', stdout);
239			fputresolu(order, xmax, ymax, stdout);
240			/* allocate scanline */
241			scanin = (COLOR )malloc(xmaxsizeof(COLOR));
242			if (scanin == NULL)
243			quiterr("out of memory in convert");
244	greg	2.14	scanout = doflat ? (COLR )malloc(xmaxsizeof(COLR)) : (COLR *)NULL;
245	greg	2.1	/* convert image */
246			for (y = 0; y < ymax; y++) {
247	greg	2.14	myreadscan(scanin, xmax);
248	greg	2.3	for (x = 0; x < xmax; x++) {
249	greg	2.1	colortrans(scanin[x], xfm, scanin[x]);
250	greg	2.3	if (rgbout)
251			clipgamut(scanin[x], bright(scanin[x]),
252			CGAMUT_LOWER, cblack, cwhite);
253			}
254	greg	2.1	if (scanout != NULL) {
255			for (x = 0; x < xmax; x++)
256			setcolr(scanout[x], colval(scanin[x],RED),
257			colval(scanin[x],GRN),
258			colval(scanin[x],BLU));
259	greg	2.14	putbinary(scanout, sizeof(COLR), xmax, stdout);
260	greg	2.1	} else
261			fwritescan(scanin, xmax, stdout);
262			if (ferror(stdout))
263			quiterr("error writing output picture");
264			}
265			/* free scanline */
266	greg	2.14	free(scanin);
267	greg	2.1	if (scanout != NULL)
268	greg	2.14	free(scanout);
269	greg	2.1	}