gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Templatize RankMap

Browse Source
This commit is contained in:
Alejandro Gallo 2022-01-27 20:41:08 +01:00
parent 61662e2717
commit 4543e712b3
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
atrip.org
View File

@ -549,6 +549,8 @@ namespace atrip {
#include <atrip/Slice.hpp> #include <atrip/Slice.hpp>
namespace atrip { namespace atrip {
template <typename F=double>
struct RankMap { struct RankMap {
std::vector<size_t> const lengths; std::vector<size_t> const lengths;
@ -561,7 +563,7 @@ namespace atrip {
1UL, std::multiplies<size_t>())) 1UL, std::multiplies<size_t>()))
{ assert(lengths.size() <= 2); } { assert(lengths.size() <= 2); }
size_t find(Slice::Location const& p) const noexcept { size_t find(typename Slice<F>::Location const& p) const noexcept {
return p.source * np + p.rank; return p.source * np + p.rank;
} }
@ -581,10 +583,10 @@ namespace atrip {
return source == nSources() && isPaddingRank(rank); return source == nSources() && isPaddingRank(rank);
} }
Slice::Location typename Slice<F>::Location
find(ABCTuple const& abc, Slice::Type sliceType) const noexcept { find(ABCTuple const& abc, typename Slice<F>::Type sliceType) const noexcept {
// tuple = {11, 8} when abc = {11, 8, 9} and sliceType = AB // tuple = {11, 8} when abc = {11, 8, 9} and sliceType = AB
const auto tuple = Slice::subtupleBySlice(abc, sliceType); const auto tuple = Slice<F>::subtupleBySlice(abc, sliceType);
const size_t index const size_t index
= tuple[0] = tuple[0]