rtkReorderProjectionsImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright RTK Consortium
 *
 *  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
 *
 *         https://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  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 rtkReorderProjectionsImageFilter_h
#define rtkReorderProjectionsImageFilter_h

#include <itkInPlaceImageFilter.h>
#include "rtkThreeDCircularProjectionGeometry.h"
#include "rtkConfiguration.h"

namespace rtk
{

template <class TInputImage, class TOutputImage = TInputImage>
class ITK_TEMPLATE_EXPORT ReorderProjectionsImageFilter : public itk::ImageToImageFilter<TInputImage, TOutputImage>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(ReorderProjectionsImageFilter);

  using Self = ReorderProjectionsImageFilter;

  using Superclass = itk::ImageToImageFilter<TInputImage, TOutputImage>;
  using Pointer = itk::SmartPointer<Self>;
  using ConstPointer = itk::SmartPointer<const Self>;

  using InputImageType = TInputImage;
  using OutputImageType = TOutputImage;
  using OutputImageRegionType = typename OutputImageType::RegionType;
  using PermutationType = enum { NONE = 0, SORT = 1, SHUFFLE = 2 };

  using GeometryType = ThreeDCircularProjectionGeometry;
  using GeometryPointer = GeometryType::Pointer;

  itkNewMacro(Self);

  itkTypeMacro(ReorderProjectionsImageFilter, ImageToImageFilter);

  itkGetModifiableObjectMacro(OutputGeometry, GeometryType);
  itkSetObjectMacro(InputGeometry, GeometryType);

  itkGetMacro(Permutation, PermutationType);
  itkSetMacro(Permutation, PermutationType);

  void
  SetInputSignal(const std::vector<double> signal);
  std::vector<double>
  GetOutputSignal();

protected:
  ReorderProjectionsImageFilter();

  ~ReorderProjectionsImageFilter() override = default;

  void
  VerifyPreconditions() ITKv5_CONST override;

  void
  GenerateData() override;

private:
  GeometryPointer m_InputGeometry;
  GeometryPointer m_OutputGeometry;

  std::vector<double> m_InputSignal;
  std::vector<double> m_OutputSignal;

  PermutationType m_Permutation;

}; // end of class

} // end namespace rtk

#ifndef ITK_MANUAL_INSTANTIATION
#  include "rtkReorderProjectionsImageFilter.hxx"
#endif

#endif