otbStatisticsAttributesLabelMapFilter.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 otbStatisticsAttributesLabelMapFilter_h
#define otbStatisticsAttributesLabelMapFilter_h
 
#include "otbLabelMapFeaturesFunctorImageFilter.h"
#include "itkMatrix.h"
#include "itkVector.h"
#include <string>
 
namespace otb
{
namespace Functor
{
template <class TLabelObject, class TFeatureImage>
class StatisticsAttributesLabelObjectFunctor
{
public:
  // Self typedef
  typedef StatisticsAttributesLabelObjectFunctor Self;
 
  // Matrix typedef
  typedef typename itk::Matrix<double, TFeatureImage::ImageDimension, TFeatureImage::ImageDimension> MatrixType;
  // Vector typedef
  typedef typename itk::Vector<double, TFeatureImage::ImageDimension> VectorType;
 
  typedef typename TFeatureImage::PixelType FeatureType;
 
  typedef TLabelObject LabelObjectType;
 
  typedef typename LabelObjectType::ConstLineIterator ConstLineIteratorType;
 
  StatisticsAttributesLabelObjectFunctor();
 
  virtual ~StatisticsAttributesLabelObjectFunctor();
 
  bool operator!=(const Self& self);
  bool operator==(const Self& self);
 
  inline void operator()(LabelObjectType* lo) const;
 
  void SetFeatureName(const std::string& name);
 
  const std::string& GetFeatureName() const;
 
  void SetFeatureImage(const TFeatureImage* img);
 
  const TFeatureImage* GetFeatureImage() const;
 
  void SetReducedAttributeSet(bool flag);
 
  bool GetReducedAttributeSet() const;
 
private:
  // The name of the feature
  std::string m_FeatureName;
 
  // The feature image
  typename TFeatureImage::ConstPointer m_FeatureImage;
 
  // True to compute only a reduced attribute set
  bool m_ReducedAttributeSet;
};
} // End namespace Functor
 
template <class TImage, class TFeatureImage>
class ITK_EXPORT StatisticsAttributesLabelMapFilter
    : public LabelMapFeaturesFunctorImageFilter<TImage,
                                                typename Functor::StatisticsAttributesLabelObjectFunctor<typename TImage::LabelObjectType, TFeatureImage>>
{
public:
  typedef TImage                              ImageType;
  typedef typename ImageType::LabelObjectType LabelObjectType;
  typedef TFeatureImage                       FeatureImageType;
 
  typedef Functor::StatisticsAttributesLabelObjectFunctor<LabelObjectType, FeatureImageType> FunctorType;
 
  typedef StatisticsAttributesLabelMapFilter Self;
  typedef LabelMapFeaturesFunctorImageFilter<ImageType, FunctorType> Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;
 
  itkStaticConstMacro(ImageDimension, unsigned int, TImage::ImageDimension);
 
  itkNewMacro(Self);
 
  itkTypeMacro(StatisticsAttributesLabelMapFilter, LabelMapFeaturesFunctorImageFilter);
 
  void SetFeatureImage(const TFeatureImage* input);
 
  const FeatureImageType* GetFeatureImage() const;
 
  void SetInput1(const TImage* input);
 
  const TImage* GetInput1() const;
 
  void SetInput2(const TFeatureImage* input);
 
  const TFeatureImage* GetInput2() const;
 
  void SetFeatureName(const std::string& name);
 
  const std::string& GetFeatureName() const;
 
  void SetReducedAttributeSet(bool flag);
 
  bool GetReducedAttributeSet() const;
 
  itkBooleanMacro(ReducedAttributeSet);
 
protected:
  StatisticsAttributesLabelMapFilter();
 
  ~StatisticsAttributesLabelMapFilter() override;
 
  void BeforeThreadedGenerateData() override;
 
  void PrintSelf(std::ostream& os, itk::Indent indent) const override;
 
private:
  StatisticsAttributesLabelMapFilter(const Self&) = delete;
  void operator=(const Self&) = delete;
}; // end of class
 
} // end namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbStatisticsAttributesLabelMapFilter.hxx"
#endif
 
#endif