otbHooverInstanceFilter.h

Go to the documentation of this file.

/*
 * 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 otbHooverInstanceFilter_h
#define otbHooverInstanceFilter_h
 
#include <set>
#include "itkInPlaceLabelMapFilter.h"
#include "itkVariableSizeMatrix.h"
#include "itkVariableLengthVector.h"
#include <string>
 
namespace otb
{
template <class TLabelMap>
class ITK_EXPORT HooverInstanceFilter : public itk::InPlaceLabelMapFilter<TLabelMap>
{
public:
  typedef HooverInstanceFilter                  Self;
  typedef itk::InPlaceLabelMapFilter<TLabelMap> Superclass;
  typedef itk::SmartPointer<Self>               Pointer;
  typedef itk::SmartPointer<const Self>         ConstPointer;
 
  itkNewMacro(Self);
 
  itkTypeMacro(HooverInstanceFilter, InPlaceLabelMapFilter);
 
  typedef TLabelMap                                     LabelMapType;
  typedef typename LabelMapType::Pointer                LabelMapPointer;
  typedef typename LabelMapType::ConstIterator          ConstIteratorType;
  typedef typename LabelMapType::Iterator               IteratorType;
  typedef typename LabelMapType::LabelObjectType        LabelObjectType;
  typedef typename LabelObjectType::AttributeType       AttributeType;
  typedef typename LabelObjectType::AttributesValueType AttributesValueType;
  typedef typename LabelMapType::LabelVectorType        LabelVectorType;
  typedef typename LabelMapType::RegionType             ImageRegionType;
 
  // typedef typename LabelObjectType::LineContainerType   LineContainerType;
  typedef typename LabelObjectType::IndexType IndexType;
  typedef typename LabelObjectType::LabelType LabelType;
 
  typedef unsigned long                            CoefficientType;
  typedef itk::VariableSizeMatrix<CoefficientType> MatrixType;
 
  typedef itk::VariableLengthVector<CoefficientType> CardinalVector;
  typedef std::set<CoefficientType>                  RegionSetType;
  typedef std::vector<LabelObjectType*>              ObjectVectorType;
 
  void SetGroundTruthLabelMap(const LabelMapType* gt);
  void SetMachineSegmentationLabelMap(const LabelMapType* ms);
 
  const LabelMapType* GetGroundTruthLabelMap();
  LabelMapType*       GetMachineSegmentationLabelMap();
 
  LabelMapType* GetOutputGroundTruthLabelMap();
  LabelMapType* GetOutputMachineSegmentationLabelMap();
 
  itkSetMacro(HooverMatrix, MatrixType);
  itkGetMacro(HooverMatrix, MatrixType);
 
  itkSetMacro(Threshold, double);
  itkGetMacro(Threshold, double);
 
  itkSetMacro(UseExtendedAttributes, bool);
  itkGetMacro(UseExtendedAttributes, bool);
 
  itkGetMacro(MeanRC, AttributesValueType);
  itkGetMacro(MeanRF, AttributesValueType);
  itkGetMacro(MeanRA, AttributesValueType);
  itkGetMacro(MeanRM, AttributesValueType);
  itkGetMacro(MeanRN, AttributesValueType);
 
  enum AttributeTypes
  {
    ATTRIBUTE_CD = 100,
    ATTRIBUTE_OS,
    ATTRIBUTE_US,
    ATTRIBUTE_M,
    ATTRIBUTE_N,
    ATTRIBUTE_RC,
    ATTRIBUTE_RF,
    ATTRIBUTE_RA,
    ATTRIBUTE_RM,
    ATTRIBUTE_RN
  };
 
  static std::string GetNameFromAttribute(const AttributeType& a)
  {
    std::string name;
    switch (a)
    {
    case ATTRIBUTE_CD:
      name = "HooverInstance_Ext_CD";
      break;
    case ATTRIBUTE_OS:
      name = "HooverInstance_Ext_OS";
      break;
    case ATTRIBUTE_US:
      name = "HooverInstance_Ext_US";
      break;
    case ATTRIBUTE_M:
      name = "HooverInstance_Ext_M";
      break;
    case ATTRIBUTE_N:
      name = "HooverInstance_Ext_N";
      break;
    case ATTRIBUTE_RC:
      name = "HooverInstance_RC";
      break;
    case ATTRIBUTE_RF:
      name = "HooverInstance_RF";
      break;
    case ATTRIBUTE_RA:
      name = "HooverInstance_RA";
      break;
    case ATTRIBUTE_RM:
      name = "HooverInstance_RM";
      break;
    case ATTRIBUTE_RN:
      name = "HooverInstance_RN";
      break;
    }
    return name;
  }
 
  static AttributeType GetAttributeFromName(const std::string& name)
  {
    if (name == "HooverInstance_Ext_CD")
    {
      return ATTRIBUTE_CD;
    }
    else if (name == "HooverInstance_Ext_OS")
    {
      return ATTRIBUTE_OS;
    }
    else if (name == "HooverInstance_Ext_US")
    {
      return ATTRIBUTE_US;
    }
    else if (name == "HooverInstance_Ext_M")
    {
      return ATTRIBUTE_M;
    }
    else if (name == "HooverInstance_Ext_N")
    {
      return ATTRIBUTE_N;
    }
    else if (name == "HooverInstance_RC")
    {
      return ATTRIBUTE_RC;
    }
    else if (name == "HooverInstance_RF")
    {
      return ATTRIBUTE_RF;
    }
    else if (name == "HooverInstance_RA")
    {
      return ATTRIBUTE_RA;
    }
    else if (name == "HooverInstance_RM")
    {
      return ATTRIBUTE_RM;
    }
    else if (name == "HooverInstance_RN")
    {
      return ATTRIBUTE_RN;
    }
  }
 
protected:
  HooverInstanceFilter();
  ~HooverInstanceFilter() override{};
 
  void AllocateOutputs() override;
 
  void ReleaseInputs() override;
 
  void ThreadedProcessLabelObject(LabelObjectType* labelObject) override;
 
  void BeforeThreadedGenerateData() override;
 
  void AfterThreadedGenerateData() override;
 
private:
  unsigned long m_NumberOfRegionsGT;
 
  unsigned long m_NumberOfRegionsMS;
 
  LabelVectorType m_LabelsGT;
 
  MatrixType m_HooverMatrix;
 
  CardinalVector m_CardRegGT;
 
  CardinalVector m_CardRegMS;
 
  double m_Threshold;
 
  bool m_UseExtendedAttributes;
 
  AttributesValueType m_MeanRC;
 
  AttributesValueType m_MeanRF;
 
  AttributesValueType m_MeanRA;
 
  AttributesValueType m_MeanRM;
 
  AttributesValueType m_MeanRN;
};
}
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbHooverInstanceFilter.hxx"
#endif
 
#endif