Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
otbSuperSuperimpose.cxx 7.56 KiB
1 
/*=========================================================================
2
3 
  Copyright (c) Remi Cresson (IRSTEA). All rights reserved.
4
5
6 
     This software is distributed WITHOUT ANY WARRANTY; without even
7 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
8 
     PURPOSE.  See the above copyright notices for more information.
9
10 
=========================================================================*/
11 
#include "otbWrapperApplicationFactory.h"
12
13 
#include "otbStreamingResampleImageFilter.h"
14
15 
#include "otbBCOInterpolateImageFunction.h"
16 
#include "itkNearestNeighborInterpolateImageFunction.h"
17
18 
#include "otbRegionComparator.h"
19 
#include "otbMultiChannelExtractROI.h"
20
21 
namespace otb
22 
{
23
24 
enum
25 
{
26 
  Interpolator_BCO,
27 
  Interpolator_NNeighbor,
28 
  Interpolator_Linear
29 
};
30
31 
namespace Wrapper
32 
{
33
34 
class SuperSuperimpose : public Application
35 
{
36 
public:
37 
  /** Standard class typedefs. */
38 
  typedef SuperSuperimpose         Self;
39 
  typedef Application                   Superclass;
40 
  typedef itk::SmartPointer<Self>       Pointer;
41 
  typedef itk::SmartPointer<const Self> ConstPointer;
42
43 
  /** Standard macro */
44 
  itkNewMacro(Self);
45
46 
  itkTypeMacro(SuperSuperimpose, Application);
47
48 
  typedef itk::LinearInterpolateImageFunction<FloatVectorImageType, double>  LinInterpolatorType;
49 
  typedef itk::NearestNeighborInterpolateImageFunction<FloatVectorImageType, double> NNInterpolatorType;
50 
  typedef otb::BCOInterpolateImageFunction<FloatVectorImageType> BCOInterpolatorType;
51 
  typedef otb::MultiChannelExtractROI<FloatVectorImageType::InternalPixelType,FloatVectorImageType::InternalPixelType> ExtractROIFilterType;
52 
  typedef otb::StreamingResampleImageFilter<FloatVectorImageType, FloatVectorImageType> BasicResamplerType;
53
54 
private:
55 
  void DoInit() override
56 
  {
57 
    SetName("SuperSuperimpose");
58 
    SetDescription("Using available image metadata, project one image onto another one");
59
60 
    // Documentation
61 
    SetDocLongDescription("This application performs the projection of an image into the geometry of another one.");
62 
    SetDocLimitations("None");
63 
    SetDocAuthors("Remi Cresson");
64 
    SetDocSeeAlso(" ");
65
66 
    AddDocTag(Tags::Geometry);
67 
    AddDocTag("Superimposition");
68
69 
    AddParameter(ParameterType_InputImage,   "inr",   "Reference input");
70 
    SetParameterDescription("inr","The input reference image.");
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
AddParameter(ParameterType_InputImage, "inm", "The image to reproject"); SetParameterDescription("inm","The image to reproject into the geometry of the reference input."); AddParameter(ParameterType_OutputImage, "out", "Output image"); SetParameterDescription("out", "Output reprojected image."); AddParameter(ParameterType_Float, "ratio", "Ratio between input image pixel spacing and output image pixel spacing"); SetDefaultParameterFloat ("ratio", 1.0); AddParameter(ParameterType_Int, "padding", "Padding radius (in pixels of the reprojected image)"); SetDefaultParameterInt ("padding", 0); AddParameter(ParameterType_Float, "fv", "Fill Value"); SetParameterDescription("fv","Fill value for area outside the reprojected image"); SetDefaultParameterFloat("fv", 0.); MandatoryOff("fv"); // Interpolators AddParameter(ParameterType_Choice, "interpolator", "Interpolation"); SetParameterDescription("interpolator","This group of parameters allows defining how the input image will be interpolated during resampling."); AddChoice("interpolator.bco", "Bicubic interpolation"); SetParameterDescription("interpolator.bco", "Bicubic interpolation leads to very good image quality but is slow."); AddParameter(ParameterType_Radius, "interpolator.bco.radius", "Radius for bicubic interpolation"); SetParameterDescription("interpolator.bco.radius","This parameter allows controlling the size of the bicubic interpolation filter. If the target pixel size is higher than the input pixel size, increasing this parameter will reduce aliasing artifacts."); SetDefaultParameterInt("interpolator.bco.radius", 2); AddChoice("interpolator.nn", "Nearest Neighbor interpolation"); SetParameterDescription("interpolator.nn","Nearest neighbor interpolation leads to poor image quality, but it is very fast."); AddChoice("interpolator.linear", "Linear interpolation"); SetParameterDescription("interpolator.linear","Linear interpolation leads to average image quality but is quite fast"); AddRAMParameter(); // Doc example parameter settings SetDocExampleParameterValue("inr", "QB_Toulouse_Ortho_PAN.tif"); SetDocExampleParameterValue("inm", "QB_Toulouse_Ortho_XS.tif"); SetDocExampleParameterValue("out", "SuperSuperimposedXS_to_PAN.tif"); SetOfficialDocLink(); } void DoUpdateParameters() override { } void DoExecute() override { // Get the inputs FloatVectorImageType* refImage = GetParameterImage("inr"); FloatVectorImageType* movingImage = GetParameterImage("inm"); // Resample filter m_BasicResampler = BasicResamplerType::New(); // Get Interpolator switch ( GetParameterInt("interpolator") ) { case Interpolator_Linear: { LinInterpolatorType::Pointer interpolator = LinInterpolatorType::New(); m_BasicResampler->SetInterpolator(interpolator); } break; case Interpolator_NNeighbor:
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206
{ NNInterpolatorType::Pointer interpolator = NNInterpolatorType::New(); m_BasicResampler->SetInterpolator(interpolator); } break; case Interpolator_BCO: { BCOInterpolatorType::Pointer interpolator = BCOInterpolatorType::New(); interpolator->SetRadius(GetParameterInt("interpolator.bco.radius")); m_BasicResampler->SetInterpolator(interpolator); } break; } // Set up output image information FloatVectorImageType::SpacingType spacing = refImage->GetSignedSpacing(); FloatVectorImageType::IndexType start = refImage->GetLargestPossibleRegion().GetIndex(); FloatVectorImageType::SizeType size = refImage->GetLargestPossibleRegion().GetSize(); FloatVectorImageType::PointType origin = refImage->GetOrigin(); for (int i = 0 ; i < 2 ; i++) { spacing[i] *= GetParameterFloat("ratio"); size[i] /= GetParameterFloat("ratio"); size[i] += GetParameterInt("padding") * 2; origin[i] += 0.5*(spacing[i] - refImage->GetSignedSpacing()[i]) - GetParameterInt("padding") * spacing[i]; } FloatVectorImageType::PixelType defaultValue; itk::NumericTraits<FloatVectorImageType::PixelType>::SetLength(defaultValue, movingImage->GetNumberOfComponentsPerPixel()); defaultValue.Fill(GetParameterFloat("fv")); m_BasicResampler->SetInput(movingImage); m_BasicResampler->SetOutputOrigin(origin); m_BasicResampler->SetOutputSpacing(spacing); m_BasicResampler->SetOutputSize(size); m_BasicResampler->SetOutputStartIndex(start); m_BasicResampler->SetEdgePaddingValue(defaultValue); m_BasicResampler->UpdateOutputInformation(); // Keep only overlap otb::RegionComparator<FloatVectorImageType, FloatVectorImageType> comparator; comparator.SetImage1(m_BasicResampler->GetOutput()); comparator.SetImage2(movingImage); // Initialize ROI extract filters m_ExtractROI = ExtractROIFilterType::New(); m_ExtractROI->SetInput(m_BasicResampler->GetOutput()); m_ExtractROI->SetExtractionRegion(comparator.GetOverlapInImage1Indices()); // Set the output image SetParameterOutputImage("out", m_ExtractROI->GetOutput()); } BasicResamplerType::Pointer m_BasicResampler; ExtractROIFilterType::Pointer m_ExtractROI; }; } // end namespace Wrapper } // end namespace otb OTB_APPLICATION_EXPORT(otb::Wrapper::SuperSuperimpose)