otbLabelMapToVectorDataFilter.hxx

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 otbLabelMapToVectorDataFilter_hxx
#define otbLabelMapToVectorDataFilter_hxx
 
#include "otbLabelMapToVectorDataFilter.h"
 
namespace otb
{
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::LabelMapToVectorDataFilter()
{ /*
  m_BackgroundValue = NumericTraits<OutputImagePixelType>::NonpositiveMin(); */
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
void LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::SetInput(const InputLabelMapType* input)
{
  // Process object is not const-correct so the const_cast is required here
  this->itk::ProcessObject::SetNthInput(0, const_cast<InputLabelMapType*>(input));
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
void LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::SetInput(unsigned int idx, const InputLabelMapType* input)
{
  // Process object is not const-correct so the const_cast is required here
  this->itk::ProcessObject::SetNthInput(idx, const_cast<InputLabelMapType*>(input));
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
const typename LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::InputLabelMapType*
LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::GetInput(void)
{
  if (this->GetNumberOfInputs() < 1)
  {
    return nullptr;
  }
 
  return static_cast<const TLabelMap*>(this->itk::ProcessObject::GetInput(0));
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
const typename LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::InputLabelMapType*
LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::GetInput(unsigned int idx)
{
  return static_cast<const TLabelMap*>(this->itk::ProcessObject::GetInput(idx));
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
void LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::GenerateData()
{
  OutputVectorDataType*    output = this->GetOutput();
  const InputLabelMapType* input  = this->GetInput();
 
  FunctorType        functor;
  CorrectFunctorType correctFunctor;
 
  DataNodePointerType document = DataNodeType::New();
  DataNodePointerType folder1  = DataNodeType::New();
 
  document->SetNodeType(DOCUMENT);
  folder1->SetNodeType(FOLDER);
  DataNodePointerType root = output->GetDataTree()->GetRoot()->Get();
  output->GetDataTree()->Add(document, root);
  output->GetDataTree()->Add(folder1, document);
 
  // Lets begin by declaring the iterator for the objects in the image.
  ConstIteratorType it = ConstIteratorType(input);
 
  while (!it.IsAtEnd())
  {
    const LabelObjectType* labelObject = it.GetLabelObject();
 
    typename PolygonType::Pointer polygon = functor(const_cast<LabelObjectType*>(labelObject));
 
    PolygonPointerType  correctPolygon = correctFunctor(polygon);
    DataNodePointerType node           = DataNodeType::New();
    node->SetNodeType(otb::FEATURE_POLYGON);
 
    std::ostringstream oss;
    oss << labelObject->GetLabel();
    node->SetNodeId(oss.str());
    // TODO hole in the polygon are not handle yet by the functor
    node->SetPolygonExteriorRing(correctPolygon);
 
    std::map<std::string, std::string>                 fields = m_FieldsFunctor(const_cast<LabelObjectType*>(labelObject));
    std::map<std::string, std::string>::const_iterator it2;
    for (it2 = fields.begin(); it2 != fields.end(); ++it2)
    {
      node->SetFieldAsString(it2->first, it2->second);
    }
 
    output->GetDataTree()->Add(node, folder1);
    ++it;
  }
}
 
template <class TLabelMap, class TVectorData, class TFieldsFunctor>
void LabelMapToVectorDataFilter<TLabelMap, TVectorData, TFieldsFunctor>::PrintSelf(std::ostream& os, itk::Indent indent) const
{
  Superclass::PrintSelf(os, indent);
}
 
} // end namespace otb
#endif