src/modified_gamma_transfer_function.cc

   1 /*
   2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include "modified_gamma_transfer_function.h"
  21 #include <cmath>
  22
  23 using std::pow;
  24 using boost::shared_ptr;
  25 using boost::dynamic_pointer_cast;
  26 using namespace dcp;
  27
  28 ModifiedGammaTransferFunction::ModifiedGammaTransferFunction (bool inverse, double power, double threshold, double A, double B)
  29         : TransferFunction (inverse)
  30         , _power (power)
  31         , _threshold (threshold)
  32         , _A (A)
  33         , _B (B)
  34 {
  35
  36 }
  37
  38 double *
  39 ModifiedGammaTransferFunction::make_lut (int bit_depth) const
  40 {
  41         int const bit_length = int(std::pow(2.0f, bit_depth));
  42         double* lut = new double[bit_length];
  43         if (_inverse) {
  44                 double const threshold = _threshold / _B;
  45                 for (int i = 0; i < bit_length; ++i) {
  46                         double const p = static_cast<double> (i) / (bit_length - 1);
  47                         if (p > threshold) {
  48                                 lut[i] = (1 + _A) * pow (p, 1 / _power) - _A;
  49                         } else {
  50                                 lut[i] = p * _B;
  51                         }
  52                 }
  53         } else {
  54                 for (int i = 0; i < bit_length; ++i) {
  55                         double const p = static_cast<double> (i) / (bit_length - 1);
  56                         if (p > _threshold) {
  57                                 lut[i] = pow ((p + _A) / (1 + _A), _power);
  58                         } else {
  59                                 lut[i] = p / _B;
  60                         }
  61                 }
  62         }
  63
  64         return lut;
  65 }
  66
  67 bool
  68 ModifiedGammaTransferFunction::about_equal (shared_ptr<const TransferFunction> other, double epsilon) const
  69 {
  70         if (!TransferFunction::about_equal (other, epsilon)) {
  71                 return false;
  72         }
  73
  74         shared_ptr<const ModifiedGammaTransferFunction> o = dynamic_pointer_cast<const ModifiedGammaTransferFunction> (other);
  75         if (!o) {
  76                 return false;
  77         }
  78
  79         return (
  80                 fabs (_power - o->_power) < epsilon &&
  81                 fabs (_threshold - o->_threshold) < epsilon &&
  82                 fabs (_A - o->_A) < epsilon &&
  83                 fabs (_B - o->_B) < epsilon
  84                 );
  85 }