RTK  2.6.0
Reconstruction Toolkit
rtkBlockDiagonalMatrixVectorMultiplyImageFilter.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright RTK Consortium
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * https://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 #ifndef rtkBlockDiagonalMatrixVectorMultiplyImageFilter_h
19 #define rtkBlockDiagonalMatrixVectorMultiplyImageFilter_h
20 
21 #include "itkImageToImageFilter.h"
22 #include "rtkMacro.h"
23 
24 namespace rtk
25 {
34 template <class TVectorImage,
35  class TMatrixImage =
36  itk::Image<itk::Vector<typename TVectorImage::PixelType::ValueType,
37  TVectorImage::PixelType::Dimension * TVectorImage::PixelType::Dimension>,
38  TVectorImage::ImageDimension>>
40  : public itk::ImageToImageFilter<TVectorImage, TVectorImage>
41 {
42 public:
43  ITK_DISALLOW_COPY_AND_MOVE(BlockDiagonalMatrixVectorMultiplyImageFilter);
44 
49 
51  itkNewMacro(Self);
52 
54 #ifdef itkOverrideGetNameOfClassMacro
55  itkOverrideGetNameOfClassMacro(BlockDiagonalMatrixVectorMultiplyImageFilter);
56 #else
58 #endif
59 
60 
62  static constexpr unsigned int nChannels = TVectorImage::PixelType::Dimension;
63 
65  using dataType = typename TVectorImage::PixelType::ValueType;
66 
68  void
69  SetInput1(const TVectorImage * vector);
70  void
71  SetInput2(const TMatrixImage * matrix);
73 
74 protected:
77 
78  void
79  GenerateInputRequestedRegion() override;
80 
82  void
83  DynamicThreadedGenerateData(const typename TVectorImage::RegionType & outputRegionForThread) override;
84 
86  typename TVectorImage::ConstPointer
87  GetInput1();
88  typename TMatrixImage::ConstPointer
89  GetInput2();
90 };
91 } // namespace rtk
93 
94 
95 #ifndef ITK_MANUAL_INSTANTIATION
96 # include "rtkBlockDiagonalMatrixVectorMultiplyImageFilter.hxx"
97 #endif
98 
99 #endif