src/modified_gamma_transfer_function.cc

   1 /*
   2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of libdcp.
   5
   6     libdcp is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     libdcp is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with libdcp.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21 #include "modified_gamma_transfer_function.h"
  22 #include <cmath>
  23
  24 using std::pow;
  25 using boost::shared_ptr;
  26 using boost::dynamic_pointer_cast;
  27 using namespace dcp;
  28
  29 ModifiedGammaTransferFunction::ModifiedGammaTransferFunction (double power, double threshold, double A, double B)
  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, bool inverse) 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         shared_ptr<const ModifiedGammaTransferFunction> o = dynamic_pointer_cast<const ModifiedGammaTransferFunction> (other);
  71         if (!o) {
  72                 return false;
  73         }
  74
  75         return (
  76                 fabs (_power - o->_power) < epsilon &&
  77                 fabs (_threshold - o->_threshold) < epsilon &&
  78                 fabs (_A - o->_A) < epsilon &&
  79                 fabs (_B - o->_B) < epsilon
  80                 );
  81 }