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

Comparing ray/src/px/ra_bmp.c (file contents):
Revision 2.1 by greg, Fri Mar 26 03:11:50 2004 UTC vs.
Revision 2.16 by greg, Fri Jun 6 19:11:21 2025 UTC

# Line 5 | Line 5 | static const char RCSid[] = "$Id$";
5   *  program to convert between RADIANCE and Windows BMP file
6   */
7  
8 < #include  <stdio.h>
8 > #include  <math.h>
9 >
10 > #include  "rtio.h"
11 > #include  "paths.h"
12   #include  "platform.h"
13   #include  "color.h"
14 + #include  "tonemap.h"
15   #include  "resolu.h"
16   #include  "bmpfile.h"
17  
18 < int  bradj = 0;                         /* brightness adjustment */
18 > int             bradj = 0;              /* brightness adjustment */
19  
20 < double  gamcor = 2.2;                   /* gamma correction value */
20 > double          gamcor = 2.2;           /* gamma correction value */
21  
22 < char  *progname;
22 > static void quiterr(const char *err);
23 > static void tmap2bmp(char *fnin, char *fnout, char *expec,
24 >                                RGBPRIMP monpri, double gamval);
25 > static void rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, RGBPRIMP monpri);
26 > static void bmp2rad(BMPReader *brd, FILE *rfp, int inv);
27  
28 < void    quiterr(const char *);
29 < void    rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, int gry);
22 < void    bmp2rad(BMPReader *brd, FILE *rfp, int inv);
28 > static RGBPRIMP rgbinp = stdprims;      /* RGB input primitives */
29 > static RGBPRIMS myinprims;              /* custom primitives holder */
30  
31 + static gethfunc headline;
32  
33 +
34   int
35   main(int argc, char *argv[])
36   {
37 <        char    *inpfile=NULL, *outfile=NULL;
38 <        int     gryflag = 0;
39 <        int     reverse = 0;
40 <        RESOLU  rs;
41 <        int     i;
37 >        char            *inpfile=NULL, *outfile=NULL;
38 >        char            *expec = NULL;
39 >        int             reverse = 0;
40 >        RGBPRIMP        rgbp = stdprims;
41 >        RGBPRIMS        myprims;
42 >        RESOLU          rs;
43 >        int             i;
44          
45 <        progname = argv[0];
45 >        fixargv0(argv[0]);              /* assigns progname */
46  
47          for (i = 1; i < argc; i++)
48 <                if (argv[i][0] == '-')
48 >                if (argv[i][0] == '-' && argv[i][1])
49                          switch (argv[i][1]) {
50                          case 'b':
51 <                                gryflag = 1;
51 >                                rgbp = NULL;
52                                  break;
53                          case 'g':
54                                  gamcor = atof(argv[++i]);
55                                  break;
56                          case 'e':
57                                  if (argv[i+1][0] != '+' && argv[i+1][0] != '-')
58 +                                        expec = argv[++i];
59 +                                else
60 +                                        bradj = atoi(argv[++i]);
61 +                                break;
62 +                        case 'p':
63 +                                if (argc-i < 9)
64                                          goto userr;
65 <                                bradj = atoi(argv[++i]);
65 >                                myprims[RED][CIEX] = atof(argv[++i]);
66 >                                myprims[RED][CIEY] = atof(argv[++i]);
67 >                                myprims[GRN][CIEX] = atof(argv[++i]);
68 >                                myprims[GRN][CIEY] = atof(argv[++i]);
69 >                                myprims[BLU][CIEX] = atof(argv[++i]);
70 >                                myprims[BLU][CIEY] = atof(argv[++i]);
71 >                                myprims[WHT][CIEX] = atof(argv[++i]);
72 >                                myprims[WHT][CIEY] = atof(argv[++i]);
73 >                                if (rgbp == stdprims)
74 >                                        rgbp = myprims;
75                                  break;
76                          case 'r':
77                                  reverse = !reverse;
78                                  break;
53 –                        case '\0':
54 –                                break;
79                          default:
80                                  goto userr;
81                          }
# Line 70 | Line 94 | main(int argc, char *argv[])
94  
95          if (i == argc-2 && strcmp(argv[i+1], "-"))
96                  outfile = argv[i+1];
97 +                                        /* check for tone-mapping */
98 +        if (expec != NULL) {
99 +                if (reverse)
100 +                        goto userr;
101 +                tmap2bmp(inpfile, outfile, expec, rgbp, gamcor);
102 +                return(0);
103 +        }
104  
105          setcolrgam(gamcor);             /* set up conversion */
106  
# Line 100 | Line 131 | main(int argc, char *argv[])
131                  rs.xr = rdr->hdr->width;
132                  rs.yr = rdr->hdr->height;
133                  rs.rt = YMAJOR;
134 +                                        /* write scans downward if we can */
135                  if (rdr->hdr->yIsDown || inpfile != NULL)
136                          rs.rt |= YDECR;
137                  fputsresolu(&rs, stdout);
138                                          /* convert file */
139                  bmp2rad(rdr, stdout, !rdr->hdr->yIsDown && inpfile!=NULL);
140                                          /* flush output */
141 <                BMPcloseInput(rdr);     /* COMMENT OUT LATER */
141 >                BMPcloseInput(rdr);
142                  if (fflush(stdout) < 0)
143                          quiterr("error writing Radiance output");
144          } else {
# Line 119 | Line 151 | main(int argc, char *argv[])
151                          exit(1);
152                  }
153                                          /* get header info. */
154 <                if (checkheader(stdin, COLRFMT, NULL) < 0 ||
154 >                if (getheader(stdin, headline, NULL) < 0 ||
155                                  !fgetsresolu(&rs, stdin))
156                          quiterr("bad Radiance picture format");
157                                          /* initialize BMP header */
158 <                if (gryflag)
159 <                        hdr = BMPmappedHeader(numscans(&rs),
160 <                                                scanlen(&rs), 0, 256);
161 <                else
162 <                        hdr = BMPtruecolorHeader(numscans(&rs),
163 <                                                scanlen(&rs), 0);
158 >                if (rgbp == NULL) {
159 >                        hdr = BMPmappedHeader(scanlen(&rs),
160 >                                                numscans(&rs), 0, 256);
161 >                        /*
162 >                        if (outfile != NULL)
163 >                                hdr->compr = BI_RLE8;
164 >                        */
165 >                } else
166 >                        hdr = BMPtruecolorHeader(scanlen(&rs),
167 >                                                numscans(&rs), 0);
168                  if (hdr == NULL)
169                          quiterr("cannot initialize BMP header");
170 <                hdr->yIsDown = (rs.rt & YDECR) && outfile == NULL;
170 >                                        /* set up output direction */
171 >                hdr->yIsDown = ((outfile == NULL) | (hdr->compr == BI_RLE8));
172                                          /* open BMP output */
173                  if (outfile != NULL)
174                          wtr = BMPopenOutputFile(outfile, hdr);
175                  else
176                          wtr = BMPopenOutputStream(stdout, hdr);
177 +                if (wtr == NULL)
178 +                        quiterr("cannot allocate writer structure");
179                                          /* convert file */
180 <                rad2bmp(stdin, wtr, !hdr->yIsDown && (rs.rt&YDECR), gryflag);
180 >                rad2bmp(stdin, wtr, !hdr->yIsDown, rgbp);
181                                          /* flush output */
182                  if (fflush((FILE *)wtr->c_data) < 0)
183                          quiterr("error writing BMP output");
184 <                BMPcloseOutput(wtr);    /* COMMENT OUT LATER */
184 >                BMPcloseOutput(wtr);
185          }
186 <        exit(0);                        /* success */
186 >        return(0);                      /* success */
187   userr:
188          fprintf(stderr,
189 <                "Usage: %s [-r][-g gamma][-e +/-stops] [input [output]]\n",
189 > "Usage: %s [-b][-g gamma][-e spec][-p xr yr xg yg xb yb xw yw] [input|- [output]]\n",
190                          progname);
191 <        exit(1);
192 <        return(1);      /* to keep compiler happy */
191 >        fprintf(stderr,
192 >                "   or: %s -r [-g gamma][-e +/-stops] [input|- [output]]\n",
193 >                        progname);
194 >        return(1);
195   }
196  
197   /* print message and exit */
198 < void
198 > static void
199   quiterr(const char *err)
200   {
201          if (err != NULL) {
# Line 164 | Line 205 | quiterr(const char *err)
205          exit(0);
206   }
207  
208 + /* process header line (don't echo) */
209 + static int
210 + headline(char *s, void *p)
211 + {
212 +        char    fmt[MAXFMTLEN];
213 +
214 +        if (formatval(fmt, s)) {        /* check if format string */
215 +                if (!strcmp(fmt,COLRFMT))
216 +                        return(0);
217 +                if (!strcmp(fmt,CIEFMT)) {
218 +                        rgbinp = TM_XYZPRIM;
219 +                        return(0);
220 +                }
221 +                if (!strcmp(fmt,SPECFMT))
222 +                        return(0);
223 +                return(-1);
224 +        }
225 +        if (isprims(s)) {               /* get input primaries */
226 +                primsval(myinprims, s);
227 +                rgbinp = myinprims;
228 +                return(0);
229 +        }
230 +        if (isncomp(s)) {
231 +                NCSAMP = ncompval(s);
232 +                return(0);
233 +        }
234 +        if (iswlsplit(s)) {
235 +                wlsplitval(WLPART, s);
236 +                return(0);
237 +        }
238 +                                        /* should I grok colcorr also? */
239 +        return(0);
240 + }
241 +
242 +
243   /* convert Radiance picture to BMP */
244 < void
245 < rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, int gry)
244 > static void
245 > rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, RGBPRIMP monpri)
246   {
247 +        int     usexfm = 0;
248 +        COLORMAT        xfm;
249          COLR    *scanin;
250 +        COLOR   cval;
251          int     y, yend, ystp;
252          int     x;
253                                                  /* allocate scanline */
254          scanin = (COLR *)malloc(bwr->hdr->width*sizeof(COLR));
255          if (scanin == NULL)
256                  quiterr("out of memory in rad2bmp");
257 +                                                /* set up color conversion */
258 +        usexfm = (monpri != NULL ? rgbinp != monpri :
259 +                        rgbinp != TM_XYZPRIM && rgbinp != stdprims);
260 +        if (usexfm) {
261 +                RGBPRIMP        destpri = monpri != NULL ? monpri : stdprims;
262 +                double          expcomp = pow(2.0, (double)bradj);
263 +                if (rgbinp == TM_XYZPRIM)
264 +                        compxyz2rgbWBmat(xfm, destpri);
265 +                else
266 +                        comprgb2rgbWBmat(xfm, rgbinp, destpri);
267 +                for (y = 0; y < 3; y++)
268 +                        for (x = 0; x < 3; x++)
269 +                                xfm[y][x] *= expcomp;
270 +        }
271                                                  /* convert image */
272          if (inv) {
273                  y = bwr->hdr->height - 1;
# Line 185 | Line 278 | rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, int gry)
278          }
279                                                  /* convert each scanline */
280          for ( ; y != yend; y += ystp) {
281 <                if (freadcolrs(scanin, bwr->hdr->width, rfp) < 0)
281 >                if (fread2colrs(scanin, bwr->hdr->width, rfp, NCSAMP, WLPART) < 0)
282                          quiterr("error reading Radiance picture");
283 <                if (bradj)
283 >                if (usexfm)
284 >                        for (x = bwr->hdr->width; x--; ) {
285 >                                colr_color(cval, scanin[x]);
286 >                                colortrans(cval, xfm, cval);
287 >                                setcolr(scanin[x], colval(cval,RED),
288 >                                                colval(cval,GRN),
289 >                                                colval(cval,BLU));
290 >                        }
291 >                else if (bradj)
292                          shiftcolrs(scanin, bwr->hdr->width, bradj);
293 <                for (x = gry ? bwr->hdr->width : 0; x--; )
294 <                        scanin[x][GRN] = normbright(scanin[x]);
293 >                if (monpri == NULL && rgbinp != TM_XYZPRIM)
294 >                        for (x = bwr->hdr->width; x--; )
295 >                                scanin[x][GRN] = normbright(scanin[x]);
296                  colrs_gambs(scanin, bwr->hdr->width);
297 <                if (gry)
297 >                if (monpri == NULL)
298                          for (x = bwr->hdr->width; x--; )
299                                  bwr->scanline[x] = scanin[x][GRN];
300                  else
# Line 211 | Line 313 | rad2bmp(FILE *rfp, BMPWriter *bwr, int inv, int gry)
313   }
314  
315   /* convert BMP file to Radiance */
316 < void
316 > static void
317   bmp2rad(BMPReader *brd, FILE *rfp, int inv)
318   {
319          COLR    *scanout;
# Line 248 | Line 350 | bmp2rad(BMPReader *brd, FILE *rfp, int inv)
350          }
351                                                  /* clean up */
352          free((void *)scanout);
353 + }
354 +
355 + /* Tone-map and convert Radiance picture */
356 + static void
357 + tmap2bmp(char *fnin, char *fnout, char *expec, RGBPRIMP monpri, double gamval)
358 + {
359 +        int             tmflags;
360 +        BMPHeader       *hdr;
361 +        BMPWriter       *wtr;
362 +        FILE            *fp;
363 +        int             xr, yr;
364 +        uby8            *pa;
365 +        int             i;
366 +                                        /* check tone-mapping spec */
367 +        i = strlen(expec);
368 +        if (i && !strncmp(expec, "auto", i))
369 +                tmflags = TM_F_CAMERA;
370 +        else if (i && !strncmp(expec, "human", i))
371 +                tmflags = TM_F_HUMAN & ~TM_F_UNIMPL;
372 +        else if (i && !strncmp(expec, "linear", i))
373 +                tmflags = TM_F_LINEAR;
374 +        else
375 +                quiterr("illegal exposure specification (auto|human|linear)");
376 +        if (monpri == NULL) {
377 +                tmflags |= TM_F_BW;
378 +                monpri = stdprims;
379 +        }
380 +                                        /* open Radiance input */
381 +        if (fnin == NULL)
382 +                fp = stdin;
383 +        else if ((fp = fopen(fnin, "r")) == NULL) {
384 +                fprintf(stderr, "%s: cannot open\n", fnin);
385 +                exit(1);
386 +        }
387 +                                        /* tone-map picture */
388 +        if (tmMapPicture(&pa, &xr, &yr, tmflags, monpri, gamval,
389 +                        0., 0., fnin, fp) != TM_E_OK)
390 +                exit(1);
391 +                                        /* initialize BMP header */
392 +        if (tmflags & TM_F_BW) {
393 +                hdr = BMPmappedHeader(xr, yr, 0, 256);
394 +                if (fnout != NULL)
395 +                        hdr->compr = BI_RLE8;
396 +        } else
397 +                hdr = BMPtruecolorHeader(xr, yr, 0);
398 +        if (hdr == NULL)
399 +                quiterr("cannot initialize BMP header");
400 +                                        /* open BMP output */
401 +        if (fnout != NULL)
402 +                wtr = BMPopenOutputFile(fnout, hdr);
403 +        else
404 +                wtr = BMPopenOutputStream(stdout, hdr);
405 +        if (wtr == NULL)
406 +                quiterr("cannot allocate writer structure");
407 +                                        /* write to BMP file */
408 +        while (wtr->yscan < yr) {
409 +                uby8    *scn = pa + xr*((tmflags & TM_F_BW) ? 1 : 3)*
410 +                                                (yr-1 - wtr->yscan);
411 +                if (tmflags & TM_F_BW)
412 +                        memcpy((void *)wtr->scanline, (void *)scn, xr);
413 +                else
414 +                        for (i = xr; i--; ) {
415 +                                wtr->scanline[3*i] = scn[3*i+BLU];
416 +                                wtr->scanline[3*i+1] = scn[3*i+GRN];
417 +                                wtr->scanline[3*i+2] = scn[3*i+RED];
418 +                        }
419 +                if ((i = BMPwriteScanline(wtr)) != BIR_OK)
420 +                        quiterr(BMPerrorMessage(i));
421 +        }
422 +                                        /* flush output */
423 +        if (fflush((FILE *)wtr->c_data) < 0)
424 +                quiterr("error writing BMP output");
425 +                                        /* clean up */
426 +        if (fnin != NULL)
427 +                fclose(fp);
428 +        free((void *)pa);
429 +        BMPcloseOutput(wtr);
430   }

Diff Legend

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