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
65
66
67
68
69
70
71
72
73
74
75
76
|
/*
* Copyright (c) 2008-2018 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/
*
* MRtrix3 is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty
* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
*
* For more details, see http://www.mrtrix.org/
*/
#ifndef __fixel_loop_h__
#define __fixel_loop_h__
#include "formats/mrtrix_utils.h"
namespace MR
{
namespace Fixel {
namespace {
struct set_offset { NOMEMALIGN
FORCE_INLINE set_offset (uint32_t offset) : offset (offset) { }
template <class DataType>
FORCE_INLINE void operator() (DataType& data) { data.index(0) = offset; }
uint32_t offset;
};
struct inc_fixel { NOMEMALIGN
template <class DataType>
FORCE_INLINE void operator() (DataType& data) { ++data.index(0); }
};
}
struct LoopFixelsInVoxel { NOMEMALIGN
const size_t num_fixels;
const uint32_t offset;
template <class... DataType>
struct Run { NOMEMALIGN
const size_t num_fixels;
const uint32_t offset;
uint32_t fixel_index;
const std::tuple<DataType&...> data;
FORCE_INLINE Run (const size_t num_fixels, const uint32_t offset, const std::tuple<DataType&...>& data) :
num_fixels (num_fixels), offset (offset), fixel_index (0), data (data) {
apply (set_offset (offset), data);
}
FORCE_INLINE operator bool() const { return fixel_index < num_fixels; }
FORCE_INLINE void operator++() { apply (inc_fixel (), data); fixel_index++; }
FORCE_INLINE void operator++(int) const { operator++(); }
};
template <class... DataType>
FORCE_INLINE Run<DataType...> operator() (DataType&... data) const { return { num_fixels, offset, std::tie (data...) }; }
};
template <class IndexType>
FORCE_INLINE LoopFixelsInVoxel
Loop (IndexType& index) {
index.index(3) = 0;
size_t num_fixels = index.value();
index.index(3) = 1;
uint32_t offset = index.value();
return { num_fixels, offset };
}
}
}
#endif
|
