otbWaterSqrtSpectralAngleImageFilter.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 otbWaterSqrtSpectralAngleImageFilter_h
#define otbWaterSqrtSpectralAngleImageFilter_h
 
#include "otbSqrtSpectralAngleFunctor.h"
#include "itkUnaryFunctorImageFilter.h"
#include "otbRadiometricIndex.h"
 
namespace otb
{
 
namespace Functor
{
template <class TInput, class TOutput>
class WaterSqrtSpectralAngleFunctor : public RadiometricIndex<TInput, TOutput>
{
public:
  using typename RadiometricIndex<TInput, TOutput>::PixelType;
 
  WaterSqrtSpectralAngleFunctor() 
        : RadiometricIndex<TInput, TOutput>({CommonBandNames::BLUE, CommonBandNames::GREEN, CommonBandNames::RED, CommonBandNames::NIR} )
  {
    // Set default reference water value
    m_ReferencePixel.SetSize(4);
    m_ReferencePixel[0] = 136.0;
    m_ReferencePixel[1] = 132.0;
    m_ReferencePixel[2] = 47.0;
    m_ReferencePixel[3] = 24.0;
    m_RefNorm = m_ReferencePixel.GetNorm();
  }
  
  virtual ~WaterSqrtSpectralAngleFunctor() = default;
 
  void SetReferenceWaterPixel(PixelType ref)
  {
    assert(m_ReferencePixel.Size() == 4);
    m_ReferencePixel[0] = this->Value(CommonBandNames::BLUE, ref);
    m_ReferencePixel[1] = this->Value(CommonBandNames::GREEN, ref);
    m_ReferencePixel[2] = this->Value(CommonBandNames::RED, ref);
    m_ReferencePixel[3] = this->Value(CommonBandNames::NIR, ref);
    m_RefNorm = m_ReferencePixel.GetNorm();
  }
 
  // Binary operator
  inline TOutput operator()(PixelType const & inPix) const override
  {
    PixelType pix(4);
    pix[0] = this->Value(CommonBandNames::BLUE, inPix);
    pix[1] = this->Value(CommonBandNames::GREEN, inPix);
    pix[2] = this->Value(CommonBandNames::RED, inPix);
    pix[3] = this->Value(CommonBandNames::NIR, inPix);
 
    return std::sqrt(SpectralAngleDetails::ComputeSpectralAngle<PixelType, PixelType, TOutput>
                                                                  (pix, pix.GetNorm(),
                                                                   m_ReferencePixel, m_RefNorm));
  }
 
  void SetBlueChannel(unsigned int channel)
  {
    this->SetBandIndex(CommonBandNames::BLUE, channel + 1);
  }
  unsigned int GetBlueChannel() const
  {
    return this->GetBandIndex(CommonBandNames::BLUE) - 1;
  }
  void SetGreenChannel(unsigned int channel)
  {
    this->SetBandIndex(CommonBandNames::GREEN, channel + 1);
  }
  unsigned int GetGreenChannel() const
  {
    return this->GetBandIndex(CommonBandNames::GREEN) - 1;
  }
  void SetRedChannel(unsigned int channel)
  {
    this->SetBandIndex(CommonBandNames::RED, channel + 1);
  }
  unsigned int GetRedChannel() const
  {
     return this->GetBandIndex(CommonBandNames::RED) - 1;
  }
  void SetNIRChannel(unsigned int channel)
  {
    this->SetBandIndex(CommonBandNames::NIR, channel + 1);
  }
  unsigned int GetNIRChannel() const
  {
    return this->GetBandIndex(CommonBandNames::NIR) - 1;
  }
 
private:
 
  PixelType m_ReferencePixel;
  double m_RefNorm;
};
} // End namespace Functor
 
 
template <class TInputVectorImage, class TOutputImage,
          class TFunction = Functor::WaterSqrtSpectralAngleFunctor<double, typename TOutputImage::PixelType>>
class ITK_EXPORT WaterSqrtSpectralAngleImageFilter : public itk::UnaryFunctorImageFilter<TInputVectorImage, TOutputImage, TFunction>
{
public:
  typedef WaterSqrtSpectralAngleImageFilter Self;
  typedef itk::UnaryFunctorImageFilter<TInputVectorImage, TOutputImage, TFunction> Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;
 
  itkNewMacro(Self);
 
  itkTypeMacro(WaterSqrtSpectralAngleImageFilter, itk::UnaryFunctorImageFilter);
 
  typedef TInputVectorImage                        InputVectorImageType;
  typedef typename InputVectorImageType::PixelType InputVectorPixelType;
  typedef TOutputImage                             OutputImageType;
 
protected:
  WaterSqrtSpectralAngleImageFilter()
  {
  }
  ~WaterSqrtSpectralAngleImageFilter() override
  {
  }
 
private:
  WaterSqrtSpectralAngleImageFilter(Self&) = delete;
  void operator=(const Self&) = delete;
};
} // end namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#endif
 
#endif