From 6a918f0827a6e8806f77e39e0348b1a2abed71f2 Mon Sep 17 00:00:00 2001 From: "Claudius \"keldu\" Holeksa" Date: Fri, 19 Dec 2025 14:11:57 +0100 Subject: Rewriting large portion for simpler approach --- lib/core/c++/flatten.hpp | 44 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 lib/core/c++/flatten.hpp (limited to 'lib/core/c++/flatten.hpp') diff --git a/lib/core/c++/flatten.hpp b/lib/core/c++/flatten.hpp new file mode 100644 index 0000000..164bb77 --- /dev/null +++ b/lib/core/c++/flatten.hpp @@ -0,0 +1,44 @@ +#pragma once + +#include +#include + +namespace kel { +namespace lbm { +namespace sch { +using namespace saw::schema; +} + +template +struct flatten_index { +public: + template + static constexpr saw::data stride(const saw::data>& meta) { + if constexpr (i > 0u){ + return stride(meta) * meta.at({i-1u}); + } + + return 1u; + } + +private: + /// 2,3,4 => 2,6,24 + /// i + j * 2 + k * 3*2 + /// 1 + 2 * 2 + 3 * 3*2 = 1+4+18 = 23 + template + static void apply_i(saw::data& flat_ind, const saw::data>& index, const saw::data>& meta){ + if constexpr ( D > i ) { + flat_ind = flat_ind + index.at({i}) * stride(meta); + apply_i(flat_ind,index,meta); + } + } +public: + static saw::data apply(const saw::data>& index, const saw::data>& meta){ + saw::data flat_ind{0u}; + + apply_i<0u>(flat_ind, index, meta); + return flat_ind; + } +}; +} +} -- cgit v1.2.3