1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
/* Copyright (c) 2008-2022 the MRtrix3 contributors.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
* Covered Software is provided under this License on an "as is"
* basis, without warranty of any kind, either expressed, implied, or
* statutory, including, without limitation, warranties that the
* Covered Software is free of defects, merchantable, fit for a
* particular purpose or non-infringing.
* See the Mozilla Public License v. 2.0 for more details.
*
* For more details, see http://www.mrtrix.org/.
*/
#ifndef __image_adapter_gradient3D_h__
#define __image_adapter_gradient3D_h__
#include "adapter/gradient1D.h"
#include "transform.h"
namespace MR
{
namespace Adapter
{
template <class ImageType>
class Gradient3D :
public Gradient1D<ImageType>
{ MEMALIGN(Gradient3D<ImageType>)
public:
using value_type = Eigen::Matrix<typename ImageType::value_type,3,1>;
Gradient3D (const ImageType& parent,
bool wrt_scanner = false) :
Gradient1D<ImageType> (parent, wrt_scanner),
wrt_scanner (wrt_scanner),
transform (parent) {}
value_type value ()
{
value_type grad;
for (size_t i = 0; i < 3; ++i) {
Gradient1D<ImageType>::set_axis(i);
grad[i] = Gradient1D<ImageType>::value();
}
if (wrt_scanner)
grad = transform.image2scanner.linear().template cast<typename ImageType::value_type>() * grad;
return grad;
}
protected:
const bool wrt_scanner;
Transform transform;
};
}
}
#endif
|
