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 (float power, float threshold, float A, float B)
  29         : _power (power)
  30         , _threshold (threshold)
  31         , _A (A)
  32         , _B (B)
  33 {
  34
  35 }
  36
  37 float *
  38 ModifiedGammaTransferFunction::make_lut (int bit_depth) const
  39 {
  40         int const bit_length = pow (2, bit_depth);
  41         float* lut = new float[int(std::pow(2.0f, bit_depth))];
  42         for (int i = 0; i < bit_length; ++i) {
  43                 float const p = static_cast<float> (i) / (bit_length - 1);
  44                 if (p > _threshold) {
  45                         lut[i] = pow ((p + _A) / (1 + _A), _power);
  46                 } else {
  47                         lut[i] = p / _B;
  48                 }
  49         }
  50
  51         return lut;
  52 }
  53
  54 bool
  55 ModifiedGammaTransferFunction::about_equal (shared_ptr<const TransferFunction> other, float epsilon) const
  56 {
  57         shared_ptr<const ModifiedGammaTransferFunction> o = dynamic_pointer_cast<const ModifiedGammaTransferFunction> (other);
  58         if (!o) {
  59                 return false;
  60         }
  61
  62         return (
  63                 fabs (_power - o->_power) < epsilon &&
  64                 fabs (_threshold - o->_threshold) < epsilon &&
  65                 fabs (_A - o->_A) < epsilon &&
  66                 fabs (_B - o->_B) < epsilon
  67                 );
  68 }