otbWrapperCastImage.h

Go to the documentation of this file.

/*
 * Copyright (C) 2005-2020 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 otbWrapperCastimage_h
#define otbWrapperCastimage_h


#include "otbClampImageFilter.h"

#include "OTBApplicationEngineExport.h"

#include "otbWrapperTypes.h"


namespace otb
{
namespace Wrapper
{
namespace details
{

template <typename TOutputImage, typename TInputImage>
struct OTBApplicationEngine_EXPORT_TEMPLATE CastImage
{
  using InputClampImageFilter = ClampImageFilter<TInputImage, DoubleVectorImageType>;

  using OutputClampImageFilter = ClampImageFilter<DoubleVectorImageType, TOutputImage>;


  CastImage(TInputImage* in) : icif(InputClampImageFilter::New()), ocif(OutputClampImageFilter::New()), out(ocif->GetOutput())
  {
    assert(in);

    icif->SetInput(in);

    ocif->SetInput(icif->GetOutput());
  }

  typename InputClampImageFilter::Pointer icif;

  typename OutputClampImageFilter::Pointer ocif;

  TOutputImage* out;
};


template <typename TOutputImage>
struct OTBApplicationEngine_EXPORT_TEMPLATE CastImage<TOutputImage, DoubleVectorImageType>
{
  using OutputClampImageFilter = ClampImageFilter<DoubleVectorImageType, TOutputImage>;


  CastImage(DoubleVectorImageType* in) : ocif(OutputClampImageFilter::New()), out(ocif->GetOutput())
  {
    assert(in);

    ocif->SetInput(in);
  }

  itk::ProcessObject::Pointer icif;

  typename OutputClampImageFilter::Pointer ocif;

  TOutputImage* out;
};


template <typename T>
struct OTBApplicationEngine_EXPORT_TEMPLATE CastImage<T, T>
{
  CastImage(T* in) : out(in)
  {
    assert(in);
  }

  itk::ProcessObject::Pointer icif;
  itk::ProcessObject::Pointer ocif;
  T*                          out;
};


template <>
struct OTBApplicationEngine_EXPORT_TEMPLATE CastImage<DoubleVectorImageType, DoubleVectorImageType>
{
  CastImage(DoubleVectorImageType* in) : out(in)
  {
    assert(in);
  }

  itk::ProcessObject::Pointer icif;
  itk::ProcessObject::Pointer ocif;
  DoubleVectorImageType*      out;
};

} // namespace details.

} // namespace Wrapper

} // namespace otb

#endif