otbAlphaBlendingFunctor.h

Go to the documentation of this file.

/*
 * Copyright (C) 2005-2022 Centre National d'Etudes Spatiales (CNES)
 *
 * This file is part of Orfeo Toolbox
 *
 *     https://www.orfeo-toolbox.org/
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef otbAlphaBlendingFunctor_h
#define otbAlphaBlendingFunctor_h
 
#include "itkRGBAPixel.h"
namespace otb
{
namespace Functor
{
 
template <class TInputPixel1, class TInputPixel2, class TOutputPixel>
class ITK_EXPORT AlphaBlendingFunctorBase
{
public:
  AlphaBlendingFunctorBase() : m_Alpha(1.0)
  {
  }
  ~AlphaBlendingFunctorBase()
  {
  }
 
  typedef TInputPixel1 InputPixel1Type;
  typedef TInputPixel2 InputPixel2Type;
  typedef TOutputPixel OutputPixelType;
 
  virtual void SetAlpha(double a)
  {
    // Keep alpha between 0 and 1
    m_Alpha = a < 0.0 ? 0.0 : (a > 1.0 ? 1.0 : a);
  }
 
  virtual double GetAlpha() const
  {
    return m_Alpha;
  }
 
  virtual inline OutputPixelType operator()(InputPixel1Type input1, InputPixel2Type input2) const = 0;
 
protected:
  double m_Alpha;
};
 
template <class TInputPixel1, class TInputPixel2, class TOutputPixel>
class ITK_EXPORT AlphaBlendingFunctor : public AlphaBlendingFunctorBase<TInputPixel1, TInputPixel2, TOutputPixel>
{
public:
  AlphaBlendingFunctor()
  {
  }
  ~AlphaBlendingFunctor()
  {
  }
 
  typedef TInputPixel1 InputPixel1Type;
  typedef TInputPixel2 InputPixel2Type;
  typedef TOutputPixel OutputPixelType;
 
  inline OutputPixelType operator()(InputPixel1Type input1, InputPixel2Type input2) const
  {
    OutputPixelType resp;
    double          alpha = this->GetAlpha();
 
    resp = static_cast<OutputPixelType>(std::floor(0.5 + (1.0 - alpha) * static_cast<double>(input1) + alpha * static_cast<double>(input2)));
    return resp;
  }
 
protected:
  double m_Alpha;
};
 
template <class TInputInternalPixel1, class TInputInternalPixel2, class TOutputInternalPixel>
class ITK_EXPORT AlphaBlendingFunctor<itk::RGBAPixel<TInputInternalPixel1>, itk::RGBAPixel<TInputInternalPixel2>, itk::RGBAPixel<TOutputInternalPixel>>
    : public AlphaBlendingFunctorBase<itk::RGBAPixel<TInputInternalPixel1>, itk::RGBAPixel<TInputInternalPixel2>, itk::RGBAPixel<TOutputInternalPixel>>
{
public:
  AlphaBlendingFunctor()
  {
  }
  ~AlphaBlendingFunctor()
  {
  }
 
  typedef TInputInternalPixel1                    InternalInputPixel1Type;
  typedef itk::RGBAPixel<InternalInputPixel1Type> InputPixel1Type;
  typedef TInputInternalPixel2                    InternalInputPixel2Type;
  typedef itk::RGBAPixel<InternalInputPixel2Type> InputPixel2Type;
  typedef TOutputInternalPixel                    InternalOutputPixelType;
  typedef itk::RGBAPixel<InternalOutputPixelType> OutputPixelType;
 
  inline OutputPixelType operator()(InputPixel1Type input1, InputPixel2Type input2) const
  {
    OutputPixelType resp;
    resp.Fill(itk::NumericTraits<InternalOutputPixelType>::max());
    double alpha = static_cast<double>(input2.GetAlpha()) / 255.0 * this->GetAlpha();
 
    resp.SetRed(static_cast<InternalOutputPixelType>(
        std::floor(0.5 + (1.0 - alpha) * static_cast<double>(input1.GetRed()) + alpha * static_cast<double>(input2.GetRed()))));
    resp.SetGreen(static_cast<InternalOutputPixelType>(
        std::floor(0.5 + (1.0 - alpha) * static_cast<double>(input1.GetGreen()) + alpha * static_cast<double>(input2.GetGreen()))));
    resp.SetBlue(static_cast<InternalOutputPixelType>(
        std::floor(0.5 + (1.0 - alpha) * static_cast<double>(input1.GetBlue()) + alpha * static_cast<double>(input2.GetBlue()))));
    return resp;
  }
 
protected:
  double m_Alpha;
};
}
}
 
#endif