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 TMaterialProjections,
          class TPhotonCounts,
          class TSpectrum,
          class TProjections =
            itk::Image<typename TMaterialProjections::PixelType::ValueType, TMaterialProjections::ImageDimension>>
class ITK_TEMPLATE_EXPORT WeidingerForwardModelImageFilter
  : public itk::ImageToImageFilter<TMaterialProjections, TMaterialProjections>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(WeidingerForwardModelImageFilter);

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

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(WeidingerForwardModelImageFilter);

  static constexpr unsigned int nBins = TPhotonCounts::PixelType::Dimension;
  static constexpr unsigned int nMaterials = TMaterialProjections::PixelType::Dimension;

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

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

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

  void
  SetInputMaterialProjections(const TMaterialProjections * materialProjections);
  void
  SetInputPhotonCounts(const TPhotonCounts * photonCounts);
  void
  SetInputSpectrum(const TSpectrum * spectrum);
  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 TMaterialProjections::ConstPointer
  GetInputMaterialProjections();
  typename TPhotonCounts::ConstPointer
  GetInputPhotonCounts();
  typename TSpectrum::ConstPointer
  GetInputSpectrum();
  typename TProjections::ConstPointer
  GetInputProjectionsOfOnes();

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


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

#endif