otbFillSegmentationHoles.cxx

/*
 * Author: Raffaele GAETANO
 *
 * Affiliation: CIRAD, UMR TETIS
 */

#include "otbWrapperApplication.h"
#include "otbWrapperApplicationFactory.h"

#include "otbStreamingStatisticsImageFilter.h"
#include "itkConnectedComponentImageFilter.h"
#include "otbBandMathImageFilter.h"

#include <string>

namespace otb
{
namespace Wrapper
{
class FillSegmentationHoles : public Application
{
public:
  typedef FillSegmentationHoles         Self;
  typedef Application                   Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;

  typedef UInt32ImageType               LabelImageType;

  typedef itk::ConnectedComponentImageFilter<LabelImageType, LabelImageType> CCLFIlterType;
  typedef otb::BandMathImageFilter<LabelImageType> BandMathFilterType;
  typedef otb::StreamingStatisticsImageFilter<LabelImageType> StatsFilterType;

  itkNewMacro(Self);

  itkTypeMacro(Segmentation, otb::Application);

private:
  void DoInit() override
  {
    SetName("FillSegmentationHoles");
    SetDescription("This application performs ...");

    SetDocLongDescription(
        "..."
        "...");
    SetDocLimitations("This application is proposed as part of the LSGRM Remote Module from Remi Cresson / Pierre Lassalle.");
    SetDocAuthors("Raffaele Gaetano");

    AddDocTag(Tags::Segmentation);
    AddDocTag("LSGRM");

    AddParameter(ParameterType_InputImage, "in", "Input segmentation");
    SetParameterDescription("in", "The input segmentation label image.");
    
    AddParameter(ParameterType_InputImage, "mask", "Input validity mask");
    SetParameterDescription("mask",
                            "Optional mask to indicate which pixels are valid (any non-zero value).");
    MandatoryOff("mask");
    DisableParameter("mask");
    
    AddParameter(ParameterType_OutputImage, "out", "Output segmentation");
    SetParameterDescription("out",
                            "The output segmentation with filled holes.");

    AddRAMParameter();

    // Doc example parameter settings
    SetDocExampleParameterValue("in", "seg.tif");
    SetDocExampleParameterValue("out", "seg_filled.tif");