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

Comparing ray/src/common/tonemap.c (file contents):
Revision 3.22 by greg, Thu Mar 23 22:01:43 2006 UTC vs.
Revision 3.51 by greg, Tue May 4 21:50:54 2021 UTC

# Line 12 | Line 12 | static const char      RCSid[] = "$Id$";
12   #include "copyright.h"
13  
14   #include        <stdio.h>
15 + #include        <stdlib.h>
16   #include        <math.h>
17 + #include        <string.h>
18   #include        "tmprivat.h"
19   #include        "tmerrmsg.h"
20  
# Line 22 | Line 24 | static const char      RCSid[] = "$Id$";
24   struct tmPackage        *tmPkg[TM_MAXPKG];
25   int     tmNumPkgs = 0;                  /* number of registered packages */
26  
27 +                                        /* luminance->brightness lookup */
28 + static TMbright         *tmFloat2BrtLUT = NULL;
29  
30 + #define tmCvLumLUfp(pf) tmFloat2BrtLUT[*(int32 *)(pf) >> 15]
31 +
32 +
33   TMstruct *
34   tmInit(                                 /* initialize new tone mapping */
35   int     flags,
# Line 48 | Line 55 | double gamval
55                  tmnew->clf[GRN] = rgb2xyzmat[1][1];
56                  tmnew->clf[BLU] = rgb2xyzmat[1][2];
57          } else {
58 <                comprgb2xyzWBmat(cmat, tmnew->monpri=monpri);
58 >                comprgb2xyzmat(cmat, tmnew->monpri=monpri);
59                  tmnew->clf[RED] = cmat[1][0];
60                  tmnew->clf[GRN] = cmat[1][1];
61                  tmnew->clf[BLU] = cmat[1][2];
# Line 60 | Line 67 | double gamval
67                  tmnew->mongam = gamval;
68                                                  /* set color divisors */
69          for (i = 0; i < 3; i++)
70 <                tmnew->cdiv[i] = 256.*pow(tmnew->clf[i], 1./tmnew->mongam);
70 >                tmnew->cdiv[i] = TM_BRES*pow(tmnew->clf[i], 1./tmnew->mongam);
71  
72                                                  /* set input transform */
73          tmnew->inppri = tmnew->monpri;
# Line 111 | Line 118 | MEM_PTR        dat
118                          tms->clf[CIEX] = tms->clf[CIEZ] = 0.;
119                          tms->clf[CIEY] = 1.;
120                  } else {
121 <                        comprgb2xyzWBmat(tms->cmat, tms->monpri);
121 >                        comprgb2xyzmat(tms->cmat, tms->monpri);
122                          tms->clf[RED] = tms->cmat[1][0];
123                          tms->clf[GRN] = tms->cmat[1][1];
124                          tms->clf[BLU] = tms->cmat[1][2];
125                  }
126 <                tms->cmat[0][0] = tms->cmat[1][1] = tms->cmat[2][2] =
120 <                                tms->inpsf;
126 >                tms->cmat[0][0] = tms->cmat[1][1] = tms->cmat[2][2] = 1.;
127                  tms->cmat[0][1] = tms->cmat[0][2] = tms->cmat[1][0] =
128                  tms->cmat[1][2] = tms->cmat[2][0] = tms->cmat[2][1] = 0.;
129  
130 <        } else if (tms->inppri == TM_XYZPRIM)   /* input is XYZ */
130 >        } else if (tms->inppri == TM_XYZPRIM) {         /* input is XYZ */
131                  compxyz2rgbWBmat(tms->cmat, tms->monpri);
132  
133 <        else {                                  /* input is RGB */
133 >        } else {                                        /* input is RGB */
134                  if (tms->inppri != tms->monpri &&
135                                  PRIMEQ(tms->inppri, tms->monpri))
136                          tms->inppri = tms->monpri;      /* no xform */
137 <                comprgb2rgbWBmat(tms->cmat, tms->inppri, tms->monpri);
137 >                if (!comprgb2rgbWBmat(tms->cmat, tms->inppri, tms->monpri))
138 >                        returnErr(TM_E_ILLEGAL);
139          }
140          for (i = 0; i < 3; i++)
141                  for (j = 0; j < 3; j++)
142                          tms->cmat[i][j] *= tms->inpsf;
143                                                  /* set color divisors */
144          for (i = 0; i < 3; i++)
145 <                if (tms->clf[i] > .001)
146 <                        tms->cdiv[i] =
140 <                                256.*pow(tms->clf[i], 1./tms->mongam);
141 <                else
142 <                        tms->cdiv[i] = 1;
145 >                tms->cdiv[i] = TM_BRES*pow(tms->clf[i] < .001 ? .001 :
146 >                                                tms->clf[i], 1./tms->mongam);
147                                                  /* notify packages */
148          for (i = tmNumPkgs; i--; )
149                  if (tms->pd[i] != NULL && tmPkg[i]->NewSpace != NULL)
# Line 149 | Line 153 | MEM_PTR        dat
153  
154  
155   void
156 < tmClearHisto(                   /* clear current histogram */
156 > tmClearHisto(                           /* clear current histogram */
157   TMstruct        *tms
158   )
159   {
160          if (tms == NULL || tms->histo == NULL)
161                  return;
162 <        free((MEM_PTR)tms->histo);
162 >        free(tms->histo);
163 >        tms->hbrmin = 10; tms->hbrmax = -10;
164          tms->histo = NULL;
165   }
166  
167  
168 + TMbright
169 + tmCvLuminance(                          /* convert a single luminance */
170 + double  lum
171 + )
172 + {
173 +        double  d;
174 +
175 + #ifdef isfinite
176 +        if (!isfinite(lum) || lum <= TM_NOLUM)
177 + #else
178 +        if (lum <= TM_NOLUM)
179 + #endif
180 +                return(TM_NOBRT);
181 +        d = TM_BRTSCALE*log(lum);
182 +        return((TMbright)(d + .5 - (d < 0.)));
183 + }
184 +
185 +
186   int
187 + tmCvLums(                               /* convert luminances using lookup */
188 + TMbright        *ls,
189 + float           *scan,
190 + int             len
191 + )
192 + {
193 +        if (tmFloat2BrtLUT == NULL) {   /* initialize lookup table */
194 +                int32   i;
195 +                tmFloat2BrtLUT = (TMbright *)malloc(sizeof(TMbright)*0x10000);
196 +                if (tmFloat2BrtLUT == NULL)
197 +                        return(TM_E_NOMEM);
198 +                for (i = 0; i < 0x10000; i++) {
199 +                        int32   l = (i<<1 | 1) << 14;
200 + #ifndef isfinite
201 +                        if ((l & 0x7f800000) == 0x7f800000)
202 +                                tmFloat2BrtLUT[i] = TM_NOBRT;
203 +                        else
204 + #endif
205 +                        tmFloat2BrtLUT[i] = tmCvLuminance(*(float *)&l);
206 +                }
207 +        }
208 +        if (len <= 0)
209 +                return(TM_E_OK);
210 +        if ((ls == NULL) | (scan == NULL))
211 +                return(TM_E_ILLEGAL);
212 +        while (len--) {
213 +                if (*scan <= TM_NOLUM) {
214 +                        *ls++ = TM_NOBRT;
215 +                        ++scan;
216 +                        continue;
217 +                }
218 +                *ls++ = tmCvLumLUfp(scan++);
219 +        }
220 +        return(TM_E_OK);
221 + }
222 +
223 +
224 + int
225 + tmCvGrays(                              /* convert float gray values */
226 + TMstruct        *tms,
227 + TMbright        *ls,
228 + float           *scan,
229 + int             len
230 + )
231 + {
232 +        static const char funcName[] = "tmCvGrays";
233 +        int     i;
234 +
235 +        if (tms == NULL)
236 +                returnErr(TM_E_TMINVAL);
237 +        if ((ls == NULL) | (scan == NULL) | (len < 0))
238 +                returnErr(TM_E_ILLEGAL);
239 +        if (tmFloat2BrtLUT == NULL)                     /* initialize */
240 +                tmCvLums(NULL, NULL, 0);
241 +        for (i = len; i--; ) {
242 +                float   lum = tms->inpsf * scan[i];
243 +                if (lum <= TM_NOLUM)
244 +                        ls[i] = TM_NOBRT;
245 +                else
246 +                        ls[i] = tmCvLumLUfp(&lum);
247 +        }
248 +        returnOK;
249 + }
250 +
251 +
252 + int
253   tmCvColors(                             /* convert float colors */
254   TMstruct        *tms,
255   TMbright        *ls,
256 < BYTE    *cs,
256 > uby8    *cs,
257   COLOR   *scan,
258   int     len
259   )
260   {
261          static const char funcName[] = "tmCvColors";
262 <        static COLOR    csmall = {.5*MINLUM, .5*MINLUM, .5*MINLUM};
174 <        static BYTE     gamtab[1024];
262 >        static uby8     gamtab[1024];
263          static double   curgam = .0;
264          COLOR   cmon;
265 <        double  lum, slum;
178 <        double  d;
265 >        float   lum, slum, d;
266          int     i;
267  
268          if (tms == NULL)
269                  returnErr(TM_E_TMINVAL);
270          if ((ls == NULL) | (scan == NULL) | (len < 0))
271                  returnErr(TM_E_ILLEGAL);
272 +        if (tmFloat2BrtLUT == NULL)                     /* initialize */
273 +                tmCvLums(NULL, NULL, 0);
274          if (cs != TM_NOCHROM && fabs(tms->mongam - curgam) > .02) {
275                  curgam = tms->mongam;                   /* (re)build table */
276                  for (i = 1024; i--; )
# Line 195 | Line 284 | int    len
284                          cmon[GRN] = tms->inpsf*scan[i][GRN];
285                          cmon[BLU] = tms->inpsf*scan[i][BLU];
286                  }
287 < #ifdef isnan
288 <                if (isnan(cmon[RED]) || isinf(cmon[RED])) cmon[RED] = .0f;
289 <                if (isnan(cmon[GRN]) || isinf(cmon[GRN])) cmon[GRN] = .0f;
290 <                if (isnan(cmon[BLU]) || isinf(cmon[BLU])) cmon[BLU] = .0f;
287 > #ifdef isfinite
288 >                if (!isfinite(cmon[RED]) || cmon[RED] < .0f) cmon[RED] = .0f;
289 >                if (!isfinite(cmon[GRN]) || cmon[GRN] < .0f) cmon[GRN] = .0f;
290 >                if (!isfinite(cmon[BLU]) || cmon[BLU] < .0f) cmon[BLU] = .0f;
291 > #else
292 >                if (cmon[RED] < .0f) cmon[RED] = .0f;
293 >                if (cmon[GRN] < .0f) cmon[GRN] = .0f;
294 >                if (cmon[BLU] < .0f) cmon[BLU] = .0f;
295   #endif
296                                                          /* world luminance */
297                  lum =   tms->clf[RED]*cmon[RED] +
298                          tms->clf[GRN]*cmon[GRN] +
299                          tms->clf[BLU]*cmon[BLU] ;
300 <                                                        /* check range */
301 <                if (clipgamut(cmon, lum, CGAMUT_LOWER, csmall, cwhite))
302 <                        lum =   tms->clf[RED]*cmon[RED] +
303 <                                tms->clf[GRN]*cmon[GRN] +
304 <                                tms->clf[BLU]*cmon[BLU] ;
212 <                if (lum < MINLUM) {
213 <                        ls[i] = MINBRT-1;               /* bogus value */
214 <                        lum = MINLUM;
215 <                } else {
216 <                        d = TM_BRTSCALE*log(lum);       /* encode it */
217 <                        ls[i] = d>0. ? (int)(d+.5) : (int)(d-.5);
218 <                }
300 >                if (lum <= TM_NOLUM) {                  /* convert brightness */
301 >                        lum = cmon[RED] = cmon[GRN] = cmon[BLU] = TM_NOLUM;
302 >                        ls[i] = TM_NOBRT;
303 >                } else
304 >                        ls[i] = tmCvLumLUfp(&lum);
305                  if (cs == TM_NOCHROM)                   /* no color? */
306                          continue;
307                  if (tms->flags & TM_F_MESOPIC && lum < LMESUPPER) {
308                          slum = scotlum(cmon);           /* mesopic adj. */
309 <                        if (lum < LMESLOWER)
309 >                        if (lum < LMESLOWER) {
310                                  cmon[RED] = cmon[GRN] = cmon[BLU] = slum;
311 <                        else {
311 >                        } else {
312                                  d = (lum - LMESLOWER)/(LMESUPPER - LMESLOWER);
313                                  if (tms->flags & TM_F_BW)
314                                          cmon[RED] = cmon[GRN] =
315                                                          cmon[BLU] = d*lum;
316                                  else
317                                          scalecolor(cmon, d);
318 <                                d = (1.-d)*slum;
318 >                                d = (1.f-d)*slum;
319                                  cmon[RED] += d;
320                                  cmon[GRN] += d;
321                                  cmon[BLU] += d;
# Line 238 | Line 324 | int    len
324                          cmon[RED] = cmon[GRN] = cmon[BLU] = lum;
325                  }
326                  d = tms->clf[RED]*cmon[RED]/lum;
327 <                cs[3*i  ] = d>=.999 ? 255 : gamtab[(int)(1024.*d)];
327 >                cs[3*i  ] = d>=.999f ? 255 : gamtab[(int)(1024.f*d)];
328                  d = tms->clf[GRN]*cmon[GRN]/lum;
329 <                cs[3*i+1] = d>=.999 ? 255 : gamtab[(int)(1024.*d)];
329 >                cs[3*i+1] = d>=.999f ? 255 : gamtab[(int)(1024.f*d)];
330                  d = tms->clf[BLU]*cmon[BLU]/lum;
331 <                cs[3*i+2] = d>=.999 ? 255 : gamtab[(int)(1024.*d)];
331 >                cs[3*i+2] = d>=.999f ? 255 : gamtab[(int)(1024.f*d)];
332          }
333          returnOK;
334   }
335  
336  
251 – TMbright
252 – tmCvLuminance(                          /* convert a single luminance */
253 – double  lum
254 – )
255 – {
256 –        double  d;
257 –
258 –        if (lum <= TM_NOLUM)
259 –                return(TM_NOBRT);
260 –        d = TM_BRTSCALE*log(lum);
261 –        if (d > 0.)
262 –                return((TMbright)(d+.5));
263 –        return((TMbright)(d-.5));
264 – }
265 –
266 –
337   int
268 – tmCvGrays(                              /* convert float gray values */
269 – TMstruct        *tms,
270 – TMbright        *ls,
271 – float   *scan,
272 – int     len
273 – )
274 – {
275 –        static const char funcName[] = "tmCvGrays";
276 –        double  d;
277 –        int     i;
278 –
279 –        if (tms == NULL)
280 –                returnErr(TM_E_TMINVAL);
281 –        if ((ls == NULL) | (scan == NULL) | (len < 0))
282 –                returnErr(TM_E_ILLEGAL);
283 –        for (i = len; i--; )
284 –                if (scan[i] <= TM_NOLUM) {
285 –                        ls[i] = TM_NOBRT;               /* bogus value */
286 –                } else {
287 –                        d = TM_BRTSCALE*log(scan[i]);   /* encode it */
288 –                        ls[i] = d>0. ? (int)(d+.5) : (int)(d-.5);
289 –                }
290 –        returnOK;
291 – }
292 –
293 –
294 – int
338   tmAddHisto(                             /* add values to histogram */
339   TMstruct        *tms,
340   TMbright        *ls,
# Line 301 | Line 344 | int    wt
344   {
345          static const char funcName[] = "tmAddHisto";
346          int     oldorig=0, oldlen, horig, hlen;
347 <        int     i, j;
347 >        int     i;
348  
349          if (tms == NULL)
350                  returnErr(TM_E_TMINVAL);
# Line 318 | Line 361 | int    wt
361                  tms->hbrmin = tms->hbrmax = ls[i];
362                  oldlen = 0;
363          } else {
364 <                oldorig = (tms->hbrmin-MINBRT)/HISTEP;
365 <                oldlen = (tms->hbrmax-MINBRT)/HISTEP + 1 - oldorig;
364 >                oldorig = HISTI(tms->hbrmin);
365 >                oldlen = HISTI(tms->hbrmax) + 1 - oldorig;
366          }
367          for (i = len; i--; ) {
368 <                if ((j = ls[i]) < MINBRT)
368 >                if (ls[i] < MINBRT)
369                          continue;
370 <                if (j < tms->hbrmin)
371 <                        tms->hbrmin = j;
372 <                else if (j > tms->hbrmax)
373 <                        tms->hbrmax = j;
370 >                if (ls[i] < tms->hbrmin)
371 >                        tms->hbrmin = ls[i];
372 >                else if (ls[i] > tms->hbrmax)
373 >                        tms->hbrmax = ls[i];
374          }
375 <        horig = (tms->hbrmin-MINBRT)/HISTEP;
376 <        hlen = (tms->hbrmax-MINBRT)/HISTEP + 1 - horig;
375 >        horig = HISTI(tms->hbrmin);
376 >        hlen = HISTI(tms->hbrmax) + 1 - horig;
377          if (hlen > oldlen) {                    /* (re)allocate histogram */
378 <                int     *newhist = (int *)calloc(hlen, sizeof(int));
378 >                HIST_TYP  *newhist = (HIST_TYP *)calloc(hlen, sizeof(HIST_TYP));
379                  if (newhist == NULL)
380                          returnErr(TM_E_NOMEM);
381                  if (oldlen) {                   /* copy and free old */
382 <                        for (i = oldlen, j = i+oldorig-horig; i; )
383 <                                newhist[--j] = tms->histo[--i];
384 <                        free((MEM_PTR)tms->histo);
382 >                        memcpy(newhist+(oldorig-horig),
383 >                                        tms->histo, oldlen*sizeof(HIST_TYP));
384 >                        free(tms->histo);
385                  }
386                  tms->histo = newhist;
387          }
# Line 346 | Line 389 | int    wt
389                  returnOK;
390          for (i = len; i--; )                    /* add in new counts */
391                  if (ls[i] >= MINBRT)
392 <                        tms->histo[ (ls[i]-MINBRT)/HISTEP - horig ] += wt;
392 >                        tms->histo[ HISTI(ls[i]) - horig ] += wt;
393          returnOK;
394   }
395  
# Line 374 | Line 417 | double La
417   }
418  
419  
377 – static int
378 – tmNewMap(                       /* allocate new tone-mapping array */
379 – TMstruct        *tms
380 – )
381 – {
382 –        if (tms->lumap != NULL && (tms->mbrmax - tms->mbrmin) !=
383 –                                        (tms->hbrmax - tms->hbrmin)) {
384 –                free((MEM_PTR)tms->lumap);
385 –                tms->lumap = NULL;
386 –        }
387 –        tms->mbrmin = tms->hbrmin;
388 –        tms->mbrmax = tms->hbrmax;
389 –        if (tms->mbrmin > tms->mbrmax)
390 –                return 0;
391 –        if (tms->lumap == NULL)
392 –                tms->lumap = (unsigned short *)malloc(sizeof(unsigned short)*
393 –                                        (tms->mbrmax-tms->mbrmin+1));
394 –        return(tms->lumap != NULL);
395 – }
396 –
397 –
420   int
421   tmFixedMapping(                 /* compute fixed, linear tone-mapping */
422   TMstruct        *tms,
423   double  expmult,
424 < double  gamval
424 > double  gamval,
425 > double  Lddyn
426   )
427   {
428          static const char funcName[] = "tmFixedMapping";
429 <        double          d;
430 <        int     i;
429 >        const int       maxV = (1L<<(8*sizeof(TMAP_TYP))) - 1;
430 >        double          minD;
431 >        int             i;
432          
433          if (!tmNewMap(tms))
434                  returnErr(TM_E_NOMEM);
435 <        if (expmult <= .0)
436 <                expmult = 1.;
437 <        if (gamval < MINGAM)
438 <                gamval = tms->mongam;
439 <        d = log(expmult/tms->inpsf);
440 <        for (i = tms->mbrmax-tms->mbrmin+1; i--; )
441 <                tms->lumap[i] = 256. * exp(
442 <                        ( d + (tms->mbrmin+i)*(1./TM_BRTSCALE) )
443 <                        / gamval );
435 >                                        /* check arguments */
436 >        if (expmult <= .0) expmult = 1.;
437 >        if (gamval < MINGAM) gamval = tms->mongam;
438 >        if (Lddyn < MINLDDYN) Lddyn = DEFLDDYN;
439 >        minD = 1./Lddyn;
440 >        for (i = tms->mbrmax-tms->mbrmin+1; i--; ) {
441 >                double  d;
442 >                d = expmult/tms->inpsf * tmLuminance(tms->mbrmin + i);
443 >                if (d >= 2.*minD)
444 >                        d -= minD;
445 >                else                    /* soft black crushing */
446 >                        d *= d/(4.*minD);
447 >                d /= 1. - minD;
448 >                d = TM_BRES*pow(d, 1./gamval);
449 >                tms->lumap[i] = (d > maxV) ? maxV : (int)d;
450 >        }
451          returnOK;
452   }
453  
# Line 430 | Line 461 | double Ldmax
461   )
462   {
463          static const char funcName[] = "tmComputeMapping";
464 <        int     *histo;
464 >        HIST_TYP        *histo;
465          float   *cumf;
466 <        int     brt0, histlen, threshold, ceiling, trimmings;
467 <        double  logLddyn, Ldmin, Ldavg, Lwavg, Tr, Lw, Ld;
468 <        int32   histot;
466 >        int     brt0, histlen;
467 >        HIST_TYP        threshold, ceiling, trimmings, histot;
468 >        double  logLddyn, Ldmin, Lwavg, Tr, Lw, Ld;
469          double  sum;
470          double  d;
471          int     i, j;
# Line 448 | Line 479 | double Ldmax
479                                          /* compute handy values */
480          Ldmin = Ldmax/Lddyn;
481          logLddyn = log(Lddyn);
482 <        Ldavg = sqrt(Ldmax*Ldmin);
483 <        i = (tms->hbrmin-MINBRT)/HISTEP;
484 <        brt0 = MINBRT + HISTEP/2 + i*HISTEP;
454 <        histlen = (tms->hbrmax-MINBRT)/HISTEP + 1 - i;
482 >        i = HISTI(tms->hbrmin);
483 >        brt0 = HISTV(i);
484 >        histlen = HISTI(tms->hbrmax) + 1 - i;
485                                          /* histogram total and mean */
486          histot = 0; sum = 0;
487          j = brt0 + histlen*HISTEP;
488          for (i = histlen; i--; ) {
489                  histot += tms->histo[i];
490 <                sum += (j -= HISTEP) * tms->histo[i];
490 >                sum += (double)(j -= HISTEP) * tms->histo[i];
491          }
492 <        threshold = histot*0.005 + .5;
463 <        if (threshold < 4)
492 >        if (!histot)
493                  returnErr(TM_E_TMFAIL);
494 +        threshold = histot/40 + 1;
495          Lwavg = tmLuminance( (double)sum / histot );
466 –                                        /* allocate space for mapping */
467 –        if (!tmNewMap(tms))
468 –                returnErr(TM_E_NOMEM);
496                                          /* use linear tone mapping? */
497 <        if (tms->flags & TM_F_LINEAR)
497 >        if (tms->flags & TM_F_LINEAR || threshold < 4 ||
498 >                        tms->hbrmax - tms->hbrmin < TM_BRTSCALE*logLddyn)
499                  goto linearmap;
500                                          /* clamp histogram */
501 <        histo = (int *)malloc(histlen*sizeof(int));
501 >        histo = (HIST_TYP *)malloc(histlen*sizeof(HIST_TYP));
502          cumf = (float *)malloc((histlen+2)*sizeof(float));
503          if ((histo == NULL) | (cumf == NULL))
504                  returnErr(TM_E_NOMEM);
505          cumf[histlen+1] = 1.;           /* guard for assignment code */
506 <        for (i = histlen; i--; )        /* make malleable copy */
507 <                histo[i] = tms->histo[i];
506 >                                        /* make malleable copy */
507 >        memcpy(histo, tms->histo, histlen*sizeof(HIST_TYP));
508          do {                            /* iterate to solution */
509                  sum = 0;                /* cumulative probability */
510                  for (i = 0; i < histlen; i++) {
511 <                        cumf[i] = (double)sum/histot;
512 <                        sum += histo[i];
511 >                        cumf[i] = sum/histot;
512 >                        sum += (double)histo[i];
513                  }
514                  cumf[histlen] = 1.;
515                  Tr = histot * (double)(tms->hbrmax - tms->hbrmin) /
516 <                        ((double)histlen*TM_BRTSCALE) / logLddyn;
516 >                                ((double)TM_BRTSCALE*histlen*logLddyn);
517                  ceiling = Tr + 1.;
518                  trimmings = 0;          /* clip to envelope */
519                  for (i = histlen; i--; ) {
# Line 503 | Line 531 | double Ldmax
531                  }
532                                          /* check if we're out of data */
533                  if ((histot -= trimmings) <= threshold) {
534 <                        free((MEM_PTR)histo);
535 <                        free((MEM_PTR)cumf);
534 >                        free(histo);
535 >                        free(cumf);
536                          goto linearmap;
537                  }
538 <        } while (trimmings > threshold);
538 >        } while (40*trimmings > histot);
539 >                                        /* allocate space for mapping */
540 >        if (!tmNewMap(tms))
541 >                returnErr(TM_E_NOMEM);
542                                          /* assign tone-mapping */
543          for (i = tms->mbrmax-tms->mbrmin+1; i--; ) {
544                  j = d = (double)i/(tms->mbrmax-tms->mbrmin)*histlen;
545                  d -= (double)j;
546                  Ld = Ldmin*exp(logLddyn*((1.-d)*cumf[j]+d*cumf[j+1]));
547                  d = (Ld - Ldmin)/(Ldmax - Ldmin);
548 <                tms->lumap[i] = 256.*pow(d, 1./gamval);
548 >                tms->lumap[i] = TM_BRES*pow(d, 1./gamval);
549          }
550 <        free((MEM_PTR)histo);           /* clean up and return */
551 <        free((MEM_PTR)cumf);
550 >        free(histo);                    /* clean up and return */
551 >        free(cumf);
552          returnOK;
553   linearmap:                              /* linear tone-mapping */
554          if (tms->flags & TM_F_HCONTR)
555 <                d = htcontrs(Ldavg) / htcontrs(Lwavg);
555 >                d = htcontrs(sqrt(Ldmax*Ldmin)) / htcontrs(Lwavg);
556          else
557 <                d = Ldavg / Lwavg;
558 <        return(tmFixedMapping(tms, tms->inpsf*d/Ldmax, gamval));
557 >                d = Ldmax / tmLuminance(tms->hbrmax);
558 >        return(tmFixedMapping(tms, tms->inpsf*d/Ldmax, gamval, Lddyn));
559   }
560  
561  
562   int
563   tmMapPixels(                    /* apply tone-mapping to pixel(s) */
564   TMstruct        *tms,
565 < BYTE    *ps,
565 > uby8    *ps,
566   TMbright        *ls,
567 < BYTE    *cs,
567 > uby8    *cs,
568   int     len
569   )
570   {
571          static const char funcName[] = "tmMapPixels";
572 <        int32   li, pv;
572 >        TMbright        lv;
573 >        TMAP_TYP        li;
574 >        int             pv;
575  
576          if (tms == NULL || tms->lumap == NULL)
577                  returnErr(TM_E_TMINVAL);
578          if ((ps == NULL) | (ls == NULL) | (len < 0))
579                  returnErr(TM_E_ILLEGAL);
580          while (len--) {
581 <                if ((li = *ls++) < tms->mbrmin) {
581 >                if ((lv = *ls++) < tms->mbrmin) {
582                          li = 0;
583                  } else {
584 <                        if (li > tms->mbrmax)
585 <                                li = tms->mbrmax;
586 <                        li = tms->lumap[li - tms->mbrmin];
584 >                        if (lv > tms->mbrmax)
585 >                                lv = tms->mbrmax;
586 >                        li = tms->lumap[lv - tms->mbrmin];
587                  }
588 <                if (cs == TM_NOCHROM)
589 <                        *ps++ = li>255 ? 255 : li;
590 <                else {
588 >                if (cs == TM_NOCHROM) {
589 > #if !(TM_BRES & 0xff)
590 >                        *ps++ = li>=TM_BRES ? 255 : li/(TM_BRES>>8);
591 > #else
592 >                        *ps++ = li>=TM_BRES ? 255 : (li<<8)/TM_BRES;
593 > #endif
594 >                } else {
595                          pv = *cs++ * li / tms->cdiv[RED];
596                          *ps++ = pv>255 ? 255 : pv;
597                          pv = *cs++ * li / tms->cdiv[GRN];
# Line 567 | Line 604 | int    len
604   }
605  
606  
570 –
571 –
607   TMstruct *
608 < tmDup(                          /* duplicate top tone mapping */
608 > tmDup(                          /* duplicate tone mapping */
609   TMstruct        *tms
610   )
611   {
# Line 585 | Line 620 | TMstruct       *tms
620                  return(NULL);
621          *tmnew = *tms;          /* copy everything */
622          if (tmnew->histo != NULL) {     /* duplicate histogram */
623 <                len = (tmnew->hbrmax-MINBRT)/HISTEP + 1 -
624 <                                (tmnew->hbrmin-MINBRT)/HISTEP;
590 <                tmnew->histo = (int *)malloc(len*sizeof(int));
623 >                len = HISTI(tmnew->hbrmax) + 1 - HISTI(tmnew->hbrmin);
624 >                tmnew->histo = (HIST_TYP *)malloc(len*sizeof(HIST_TYP));
625                  if (tmnew->histo != NULL)
626 <                        for (i = len; i--; )
593 <                                tmnew->histo[i] = tms->histo[i];
626 >                        memcpy(tmnew->histo, tms->histo, len*sizeof(HIST_TYP));
627          }
628          if (tmnew->lumap != NULL) {     /* duplicate luminance mapping */
629                  len = tmnew->mbrmax-tmnew->mbrmin+1;
630 <                tmnew->lumap = (unsigned short *)malloc(
598 <                                                len*sizeof(unsigned short) );
630 >                tmnew->lumap = (TMAP_TYP *)malloc(len*sizeof(TMAP_TYP));
631                  if (tmnew->lumap != NULL)
632 <                        for (i = len; i--; )
601 <                                tmnew->lumap[i] = tms->lumap[i];
632 >                        memcpy(tmnew->lumap, tms->lumap, len*sizeof(TMAP_TYP));
633          }
634                                          /* clear package data */
635          for (i = tmNumPkgs; i--; )
# Line 618 | Line 649 | TMstruct       *tms;
649                  return;
650                                          /* free tables */
651          if (tms->histo != NULL)
652 <                free((MEM_PTR)tms->histo);
652 >                free(tms->histo);
653          if (tms->lumap != NULL)
654 <                free((MEM_PTR)tms->lumap);
654 >                free(tms->lumap);
655                                          /* free private data */
656          for (i = tmNumPkgs; i--; )
657                  if (tms->pd[i] != NULL)
658                          (*tmPkg[i]->Free)(tms->pd[i]);
659 <        free((MEM_PTR)tms);             /* free basic structure */
659 >        free(tms);                      /* free basic structure */
660   }
661  
662   /******************** Shared but Private library routines *********************/
663  
664 < BYTE    tmMesofact[BMESUPPER-BMESLOWER];
664 > uby8    tmMesofact[BMESUPPER-BMESLOWER];
665  
666   void
667   tmMkMesofact()                          /* build mesopic lookup factor table */
# Line 644 | Line 675 | tmMkMesofact()                         /* build mesopic lookup factor table
675                  tmMesofact[i-BMESLOWER] = 256. *
676                                  (tmLuminance(i) - LMESLOWER) /
677                                  (LMESUPPER - LMESLOWER);
678 + }
679 +
680 +
681 + int
682 + tmNewMap(                       /* allocate new tone-mapping array */
683 + TMstruct        *tms
684 + )
685 + {
686 +        if (tms->lumap != NULL && (tms->mbrmax - tms->mbrmin) !=
687 +                                        (tms->hbrmax - tms->hbrmin)) {
688 +                free(tms->lumap);
689 +                tms->lumap = NULL;
690 +        }
691 +        tms->mbrmin = tms->hbrmin;
692 +        tms->mbrmax = tms->hbrmax;
693 +        if (tms->mbrmin > tms->mbrmax)
694 +                return(0);
695 +        if (tms->lumap == NULL)
696 +                tms->lumap = (TMAP_TYP *)calloc(tms->mbrmax-tms->mbrmin+1,
697 +                                                sizeof(TMAP_TYP));
698 +        else
699 +                memset(tms->lumap, 0, (tms->mbrmax-tms->mbrmin+1)*sizeof(TMAP_TYP));
700 +
701 +        return(tms->lumap != NULL);
702   }
703  
704  

Diff Legend

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