otbLabelImageToLabelMapWithAdjacencyFilter.h

Go to the documentation of this file.

/*
 * Copyright (C) 1999-2011 Insight Software Consortium
 * 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 otbLabelImageToLabelMapWithAdjacencyFilter_h
#define otbLabelImageToLabelMapWithAdjacencyFilter_h
 
#include "itkImageToImageFilter.h"
#include "otbLabelMapWithAdjacency.h"
#include "itkLabelObject.h"
 
namespace otb
{
 
template <class TInputImage, class TOutputImage = otb::LabelMapWithAdjacency<itk::LabelObject<typename TInputImage::PixelType, TInputImage::ImageDimension>>>
class ITK_EXPORT LabelImageToLabelMapWithAdjacencyFilter : public itk::ImageToImageFilter<TInputImage, TOutputImage>
{
public:
  typedef LabelImageToLabelMapWithAdjacencyFilter Self;
  typedef itk::ImageToImageFilter<TInputImage, TOutputImage> Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;
 
  typedef TInputImage                           InputImageType;
  typedef TOutputImage                          OutputImageType;
  typedef typename InputImageType::Pointer      InputImagePointer;
  typedef typename InputImageType::ConstPointer InputImageConstPointer;
  typedef typename InputImageType::RegionType   InputImageRegionType;
  typedef typename InputImageType::PixelType    InputImagePixelType;
  typedef typename InputImageType::IndexType    IndexType;
 
  typedef typename OutputImageType::Pointer                     OutputImagePointer;
  typedef typename OutputImageType::ConstPointer                OutputImageConstPointer;
  typedef typename OutputImageType::RegionType                  OutputImageRegionType;
  typedef typename OutputImageType::PixelType                   OutputImagePixelType;
  typedef typename OutputImageType::LabelObjectType             LabelObjectType;
  typedef typename OutputImageType::AdjacencyMapType            AdjacencyMapType;
  typedef typename OutputImageType::AdjacentLabelsContainerType AdjacentLabelsContainerType;
  typedef typename OutputImageType::LabelType                   LabelType;
 
  typedef typename LabelObjectType::ConstLineIterator ConstLineIteratorType;
 
  itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
  itkStaticConstMacro(OutputImageDimension, unsigned int, TOutputImage::ImageDimension);
 
  itkNewMacro(Self);
 
  itkTypeMacro(LabelImageToLabelMapWithAdjacencyFilter, ImageToImageFilter);
 
  itkSetMacro(BackgroundValue, OutputImagePixelType);
  itkGetConstMacro(BackgroundValue, OutputImagePixelType);
 
protected:
  LabelImageToLabelMapWithAdjacencyFilter();
 
  ~LabelImageToLabelMapWithAdjacencyFilter() override{};
 
  void PrintSelf(std::ostream& os, itk::Indent indent) const override;
 
  // class to store a RLE
  class RLE
  {
  public:
    // Location
    IndexType where;
    // Length
    long unsigned int length;
    // Label
    unsigned long int label;
    // Constructor
    RLE(const IndexType& w, long unsigned int l, unsigned long int la)
    {
      where  = w;
      length = l;
      label  = la;
    }
 
  }; // end class RLE
 
  // RLE Vector
  typedef std::vector<RLE> RLEVectorType;
 
  void GenerateInputRequestedRegion() override;
 
  void EnlargeOutputRequestedRegion(itk::DataObject* itkNotUsed(output)) override;
 
  void BeforeThreadedGenerateData() override;
 
  void ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) override;
 
  void AfterThreadedGenerateData() override;
 
  void AddAdjacency(LabelType label1, LabelType label2, itk::ThreadIdType threadId);
 
  void ParseLine(const RLEVectorType& line, itk::ThreadIdType threadId);
 
  void ParseConsecutiveLines(const RLEVectorType& line1, const RLEVectorType& line2, itk::ThreadIdType threadId);
 
 
private:
  LabelImageToLabelMapWithAdjacencyFilter(const Self&) = delete;
  void operator=(const Self&) = delete;
 
  OutputImagePixelType m_BackgroundValue;
 
  typename std::vector<OutputImagePointer> m_TemporaryImages;
  typename std::vector<AdjacencyMapType>   m_TemporaryAdjacencyMaps;
 
}; // end of class
 
} // end namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbLabelImageToLabelMapWithAdjacencyFilter.hxx"
#endif
 
#endif