otbImageListToVectorImageFilter.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 otbImageListToVectorImageFilter_hxx
#define otbImageListToVectorImageFilter_hxx
 
#include "otbImageListToVectorImageFilter.h"
#include "itkImageRegionIterator.h"
#include <vector>
#include "otbMacro.h"
#include "itkProgressReporter.h"
#include "otbImageMetadata.h"
 
namespace otb
{
template <class TImageList, class TVectorImage>
void ImageListToVectorImageFilter<TImageList, TVectorImage>::GenerateOutputInformation(void)
{
  if (this->GetOutput())
  {
    if (this->GetInput()->Size() > 0)
    {
      this->GetOutput()->CopyInformation(this->GetInput()->GetNthElement(0));
      this->GetOutput()->SetNumberOfComponentsPerPixel(this->GetInput()->Size());
      this->GetOutput()->SetLargestPossibleRegion(this->GetInput()->GetNthElement(0)->GetLargestPossibleRegion());
 
      // Copy band specific metadata from the inputs to the output
      ImageMetadata::ImageMetadataBandsType bandsImd;
      InputImageListPointerType                  inputPtr    = this->GetInput();
      typename InputImageListType::ConstIterator inputListIt = inputPtr->Begin();
      
      while (inputListIt != inputPtr->End())
      {
        const auto & imd = inputListIt.Get()->GetImageMetadata();
        if (imd.Bands.size())
        {
          bandsImd.push_back(imd.Bands[0]);
        }
        else
        {
          bandsImd.push_back({});
        }
        ++inputListIt;
      }
      
      this->GetOutput()->SetBandImageMetadata(bandsImd);
    }
  }
}
template <class TImageList, class TVectorImage>
void ImageListToVectorImageFilter<TImageList, TVectorImage>::GenerateInputRequestedRegion(void)
{
  InputImageListPointerType                  inputPtr    = this->GetInput();
  typename InputImageListType::ConstIterator inputListIt = inputPtr->Begin();
  while (inputListIt != inputPtr->End())
  {
    inputListIt.Get()->SetRequestedRegion(this->GetOutput()->GetRequestedRegion());
    ++inputListIt;
  }
}
 
template <class TImageList, class TVectorImage>
void ImageListToVectorImageFilter<TImageList, TVectorImage>::GenerateData(void)
{
 
  InputImageListPointerType    inputPtr  = this->GetInput();
  OutputVectorImagePointerType outputPtr = this->GetOutput();
 
  itk::ProgressReporter progress(this, 0, outputPtr->GetRequestedRegion().GetNumberOfPixels());
 
  // Output image initializations
  typename OutputVectorImageType::PixelType black;
  black.SetSize(inputPtr->Size());
  black.Fill(0);
  outputPtr->SetBufferedRegion(outputPtr->GetRequestedRegion());
  outputPtr->Allocate();
  outputPtr->FillBuffer(black);
 
  // defines input and output iterators
  typedef itk::ImageRegionConstIterator<InputImageType>   InputIteratorType;
  typedef itk::ImageRegionIterator<OutputVectorImageType> OutputIteratorType;
 
  typename InputImageListType::ConstIterator inputListIt = inputPtr->Begin();
 
  // defines a vector of input iterators
  typedef std::vector<InputIteratorType> InputIteratorListType;
  InputIteratorListType                  inputIteratorList;
 
  // fills the vector of input iterators
  for (; inputListIt != inputPtr->End(); ++inputListIt)
  {
    inputIteratorList.push_back(InputIteratorType(inputListIt.Get(), outputPtr->GetRequestedRegion()));
    inputIteratorList.back().GoToBegin();
  }
 
  // walk through the output image
  OutputIteratorType outputIt(outputPtr, outputPtr->GetRequestedRegion());
  outputIt.GoToBegin();
 
  while (!outputIt.IsAtEnd())
  {
    typename OutputVectorImageType::PixelType pixel   = outputIt.Get();
    unsigned int                              counter = 0;
    // for each input iterator, fill the right component
    for (typename InputIteratorListType::iterator it = inputIteratorList.begin(); it != inputIteratorList.end(); ++it)
    {
      if (!it->IsAtEnd())
      {
        pixel[counter] = static_cast<typename OutputVectorImageType::InternalPixelType>(it->Get());
        ++(*it);
        ++counter;
      }
    }
    outputIt.Set(pixel);
    progress.CompletedPixel();
    ++outputIt;
  }
}