source/blender/compositor/operations/COM_MixSaturationOperation.cpp

   1 /*
   2  * Copyright 2011, Blender Foundation.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (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 Foundation,
  16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  * Contributor:
  19  *              Jeroen Bakker
  20  *              Monique Dewanchand
  21  */
  22
  23 #include "COM_MixSaturationOperation.h"
  24
  25 extern "C" {
  26         #include "BLI_math.h"
  27 }
  28
  29 MixSaturationOperation::MixSaturationOperation() : MixBaseOperation()
  30 {
  31         /* pass */
  32 }
  33
  34 void MixSaturationOperation::executePixel(float output[4], float x, float y, PixelSampler sampler)
  35 {
  36         float inputColor1[4];
  37         float inputColor2[4];
  38         float value;
  39
  40         this->m_inputValueOperation->read(&value, x, y, sampler);
  41         this->m_inputColor1Operation->read(&inputColor1[0], x, y, sampler);
  42         this->m_inputColor2Operation->read(&inputColor2[0], x, y, sampler);
  43
  44         if (this->useValueAlphaMultiply()) {
  45                 value *= inputColor2[3];
  46         }
  47         float valuem = 1.0f - value;
  48
  49         float rH, rS, rV;
  50         rgb_to_hsv(inputColor1[0], inputColor1[1], inputColor1[2], &rH, &rS, &rV);
  51         if (rS != 0.0f) {
  52                 float colH, colS, colV;
  53                 rgb_to_hsv(inputColor2[0], inputColor2[1], inputColor2[2], &colH, &colS, &colV);
  54                 hsv_to_rgb(rH, (valuem * rS + value * colS), rV, &output[0], &output[1], &output[2]);
  55         } else {
  56                 copy_v3_v3(output, inputColor1);
  57         }
  58
  59         output[3] = inputColor1[3];
  60
  61         clampIfNeeded(output);
  62 }