otbStereorectificationDisplacementFieldSource.h

Go to the documentation of this file.

/*
 * Copyright (C) 2005-2024 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 otbStereorectificationDisplacementFieldSource_h
#define otbStereorectificationDisplacementFieldSource_h
 
#include "itkImageSource.h"
#include "otbGenericRSTransform.h"
#include "itkImageRegionIteratorWithIndex.h"
#include "itkVector.h"
#include "otbMacro.h"
 
namespace otb
{
template <class TInputImage, class TOutputImage>
class ITK_EXPORT StereorectificationDisplacementFieldSource : public itk::ImageSource<TOutputImage>
{
public:
  typedef StereorectificationDisplacementFieldSource Self;
  typedef itk::ImageSource<TOutputImage>             Superclass;
  typedef itk::SmartPointer<Self>                    Pointer;
  typedef itk::SmartPointer<const Self>              ConstPointer;
 
  // Inputs and outputs relative typedefs
  typedef TInputImage                       InputImageType;
  typedef typename InputImageType::Pointer  InputImagePointerType;
  typedef TOutputImage                      OutputImageType;
  typedef typename OutputImageType::Pointer OutputImagePointerType;
 
  // Image related typedefs
  typedef typename OutputImageType::SizeType    SizeType;
  typedef typename OutputImageType::PointType   PointType;
  typedef typename OutputImageType::SpacingType SpacingType;
  typedef typename OutputImageType::RegionType  RegionType;
 
  // 3D RS transform
  // TODO: Allow tuning precision (i.e. double or float)
  typedef otb::GenericRSTransform<double, 3, 3> RSTransformType;
  typedef typename RSTransformType::Pointer RSTransformPointerType;
 
  // 3D points
  typedef typename RSTransformType::InputPointType TDPointType;
 
  itkNewMacro(Self);
 
  itkTypeMacro(StereorectificationDisplacementFieldSource, ImageSource);
 
  itkSetMacro(Scale, double);
  itkGetConstReferenceMacro(Scale, double);
 
  itkSetMacro(GridStep, double);
  itkGetMacro(GridStep, double);
 
  itkSetObjectMacro(LeftImage, InputImageType);
  itkGetObjectMacro(LeftImage, InputImageType);
 
  itkSetObjectMacro(RightImage, InputImageType);
  itkGetObjectMacro(RightImage, InputImageType);
 
  itkGetConstReferenceMacro(RectifiedImageSize, SizeType);
 
  itkGetConstReferenceMacro(MeanBaselineRatio, double);
 
  const OutputImageType* GetLeftDisplacementFieldOutput() const;
 
  OutputImageType* GetLeftDisplacementFieldOutput();
 
  const OutputImageType* GetRightDisplacementFieldOutput() const;
 
  OutputImageType* GetRightDisplacementFieldOutput();
 
  itkSetMacro(UseDEM, bool);
  itkGetMacro(UseDEM, bool);
  itkBooleanMacro(UseDEM);
 
protected:
  StereorectificationDisplacementFieldSource(void);
 
  ~StereorectificationDisplacementFieldSource(void) override{};
 
  void GenerateOutputInformation() override;
 
  void EnlargeOutputRequestedRegion(itk::DataObject* itkNotUsed(output)) override;
 
  void GenerateData() override;
 
  void PrintSelf(std::ostream& os, itk::Indent indent) const override;
 
private:
  StereorectificationDisplacementFieldSource(const Self&); // purposely
                                                           // not
                                                           // implemented
  void operator=(const Self&) = delete;
 
  double m_ElevationOffset;
 
  double m_Scale;
 
  double m_GridStep;
 
  InputImagePointerType m_LeftImage;
 
  InputImagePointerType m_RightImage;
 
  RSTransformPointerType m_LeftToRightTransform;
 
  RSTransformPointerType m_RightToLeftTransform;
 
  SizeType m_RectifiedImageSize;
 
  TDPointType m_OutputOriginInLeftImage;
 
  double m_MeanBaselineRatio;
 
  bool m_UseDEM;
};
 
} // End namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbStereorectificationDisplacementFieldSource.hxx"
#endif
 
#endif