rtkWeidingerForwardModelImageFilter.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 rtkWeidingerForwardModelImageFilter_h
#define rtkWeidingerForwardModelImageFilter_h

#include "itkImageToImageFilter.h"
#include "rtkMacro.h"

#ifdef RTK_USE_CUDA
#  include <itkCudaImage.h>
#endif

namespace rtk
{
template <class TDecomposedProjections,
          class TMeasuredProjections,
          class TIncidentSpectrum,
          class TProjections =
            itk::Image<typename TDecomposedProjections::PixelType::ValueType, TDecomposedProjections::ImageDimension>>
class ITK_TEMPLATE_EXPORT WeidingerForwardModelImageFilter
  : public itk::ImageToImageFilter<TDecomposedProjections, TDecomposedProjections>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(WeidingerForwardModelImageFilter);

  using Self = WeidingerForwardModelImageFilter;
  using Superclass = itk::ImageToImageFilter<TDecomposedProjections, TDecomposedProjections>;
  using Pointer = itk::SmartPointer<Self>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(WeidingerForwardModelImageFilter);

  static constexpr unsigned int nBins = TMeasuredProjections::PixelType::Dimension;
  static constexpr unsigned int nMaterials = TDecomposedProjections::PixelType::Dimension;

  using dataType = typename TDecomposedProjections::PixelType::ValueType;

  using TOutputImage1 = TDecomposedProjections;
  using TPixelOutput2 = itk::Vector<dataType, nMaterials * nMaterials>;
#ifdef RTK_USE_CUDA
  using TOutputImage2 = itk::CudaImage<TPixelOutput2, TDecomposedProjections::ImageDimension>;
#else
  using TOutputImage2 = itk::Image<TPixelOutput2, TDecomposedProjections::ImageDimension>;
#endif

  TOutputImage1 *
  GetOutput1();
  TOutputImage2 *
  GetOutput2();

  void
  SetInputDecomposedProjections(const TDecomposedProjections * decomposedProjections);
  void
  SetInputMeasuredProjections(const TMeasuredProjections * measuredProjections);
  void
  SetInputIncidentSpectrum(const TIncidentSpectrum * incidentSpectrum);
#ifndef ITK_FUTURE_LEGACY_REMOVE
  void
  SetInputMaterialProjections(const TDecomposedProjections * decomposedProjections);
  void
  SetInputPhotonCounts(const TMeasuredProjections * measuredProjections);
  void
  SetInputSpectrum(const TIncidentSpectrum * incidentSpectrum);
#endif
  void
  SetInputProjectionsOfOnes(const TProjections * projectionsOfOnes);

  using BinnedDetectorResponseType = vnl_matrix<dataType>;
  using MaterialAttenuationsType = vnl_matrix<dataType>;

  itkGetConstReferenceMacro(BinnedDetectorResponse, BinnedDetectorResponseType);
  itkGetConstReferenceMacro(MaterialAttenuations, MaterialAttenuationsType);
  virtual void
  SetBinnedDetectorResponse(const BinnedDetectorResponseType & detResp);
  virtual void
  SetMaterialAttenuations(const MaterialAttenuationsType & matAtt);

protected:
  WeidingerForwardModelImageFilter();
  ~WeidingerForwardModelImageFilter() override = default;

  void
  GenerateInputRequestedRegion() override;
  void
  VerifyInputInformation() const override
  {}

  void
  DynamicThreadedGenerateData(const typename TOutputImage1::RegionType & outputRegionForThread) override;

  itk::ProcessObject::DataObjectPointer
  MakeOutput(itk::ProcessObject::DataObjectPointerArraySizeType idx) override;
  itk::ProcessObject::DataObjectPointer
  MakeOutput(const itk::ProcessObject::DataObjectIdentifierType &) override;

  typename TDecomposedProjections::ConstPointer
  GetInputDecomposedProjections();
  typename TMeasuredProjections::ConstPointer
  GetInputMeasuredProjections();
  typename TIncidentSpectrum::ConstPointer
  GetInputIncidentSpectrum();
#ifndef ITK_FUTURE_LEGACY_REMOVE
  typename TDecomposedProjections::ConstPointer
  GetInputMaterialProjections();
  typename TMeasuredProjections::ConstPointer
  GetInputPhotonCounts();
  typename TIncidentSpectrum::ConstPointer
  GetInputSpectrum();
#endif
  typename TProjections::ConstPointer
  GetInputProjectionsOfOnes();

  BinnedDetectorResponseType m_BinnedDetectorResponse;
  MaterialAttenuationsType   m_MaterialAttenuations;
  itk::SizeValueType         m_NumberOfProjectionsInIncidentSpectrum;
};
} // namespace rtk


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

#endif