otbImageToPointSetFilter.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 otbImageToPointSetFilter_h
#define otbImageToPointSetFilter_h
 
#include "otbPointSetSource.h"
#include "otbRAMDrivenAdaptativeStreamingManager.h"
 
// TODO: This class needs a refactoring to become a persistent filter
 
namespace otb
{
 
template <class TInputImage, class TOutputPointSet>
class ITK_EXPORT ImageToPointSetFilter : public PointSetSource<TOutputPointSet>
{
public:
  typedef ImageToPointSetFilter           Self;
  typedef PointSetSource<TOutputPointSet> Superclass;
  typedef itk::SmartPointer<Self>         Pointer;
  typedef itk::SmartPointer<const Self>   ConstPointer;
 
  itkTypeMacro(ImageToPointSetFilter, PointSetSource);
 
  typedef TInputImage                           InputImageType;
  typedef typename InputImageType::Pointer      InputImagePointer;
  typedef typename InputImageType::ConstPointer InputImageConstPointer;
  typedef typename InputImageType::RegionType   InputImageRegionType;
  typedef typename InputImageType::PixelType    InputImagePixelType;
  itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
 
  typedef typename Superclass::OutputPointSetType     OutputPointSetType;
  typedef typename Superclass::OutputPointSetPointer  OutputPointSetPointer;
  typedef typename Superclass::PointsContainerType    PointsContainerType;
  typedef typename Superclass::PointDataContainerType PointDataContainerType;
  typedef itk::ProcessObject                          ProcessObjectType;
 
  using Superclass::SetInput;
  void SetInput(unsigned int idx, const InputImageType* input);
  void SetInput(const InputImageType* input);
 
  const InputImageType* GetInput(unsigned int idx);
  const InputImageType* GetInput();
 
  void GenerateOutputInformation(void) override;
 
protected:
  ImageToPointSetFilter();
  ~ImageToPointSetFilter() override
  {
  }
  void PrintSelf(std::ostream& os, itk::Indent indent) const override;
 
  void GenerateData(void) override;
 
  typedef std::vector<typename OutputPointSetType::PointsContainer::Pointer>    OutputPointsContainerForThreadType;
  typedef std::vector<typename OutputPointSetType::PointDataContainer::Pointer> OutputPointDataContainerForThreadType;
 
  virtual void BeforeThreadedGenerateData();
 
  virtual void AfterThreadedGenerateData();
 
  virtual int SplitRequestedRegion(int i, int num, InputImageRegionType& splitRegion);
 
  virtual void ThreadedGenerateData(const InputImageRegionType& inputRegionForThread, itk::ThreadIdType threadId);
 
  static ITK_THREAD_RETURN_TYPE ThreaderCallback(void* arg);
 
  struct ThreadStruct
  {
    Pointer Filter;
  };
 
  OutputPointsContainerForThreadType    m_PointsContainerPerThread;
  OutputPointDataContainerForThreadType m_PointDataContainerPerThread;
 
  typedef RAMDrivenAdaptativeStreamingManager<InputImageType> StreamingManagerType;
  typedef typename StreamingManagerType::Pointer              StreamingManagerPointer;
 
  StreamingManagerPointer m_StreamingManager;
 
private:
  ImageToPointSetFilter(const ImageToPointSetFilter&) = delete;
  void operator=(const ImageToPointSetFilter&) = delete;
};
 
} // end namespace otb
 
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbImageToPointSetFilter.hxx"
#endif
 
#endif