index utils: index depth/dimension

This commit is contained in:
Christian Zimmermann 2022-12-13 19:19:41 +01:00
parent 79535c496f
commit 6963dd82cd
2 changed files with 112 additions and 0 deletions

View file

@ -71,6 +71,112 @@ namespace CNORZX
} }
} }
template <class Index>
constexpr decltype(auto) getIndexDepth(const Index& ind)
{
static_assert(is_index<Index>::value, "got non-index type");
if constexpr(has_sub<Index>::value){
if constexpr(has_static_sub<Index>::value){
constexpr SizeT D = index_dim<Index>::value;
return iter<0,D>
( [&](auto i) { return getIndexDepth(*std::get<i>(ind.pack())); },
[](auto... e) {
if constexpr( is_integral_constant<decltype(e)>::value and ... ){
return std::integral_constant<SizeT,std::max({e...})> {};
}
else {
return std::max({e...});
}
} );
}
else {
SizeT o = 0;
if(ind.dim() == 1){
auto p = ind.pack();
if(p.size() == 0){
o = 1;
}
else {
o = getIndexDepth(*p[0]);
}
}
else {
for(SizeT i = 0; i != ind.dim(); ++i){
const SizeT e = getIndexDepth(*ind.pack()[i]);
if(e > o){
o = e;
}
}
}
return o;
}
}
else {
return std::integral_constant<SizeT,1> {};
}
}
template <class Index, typename IntT>
constexpr decltype(auto) getDimension(const Index& ind, IntT depth)
{
static_assert(is_index<Index>::value, "got non-index type");
if constexpr(std::is_same<std::integral_constant<SizeT,0>,IntT>::value) {
return std::integral_constant<SizeT,1> {}
}
else {
auto iminus = [](auto i) {
if constexpr( is_integral_constant<decltype(i)>::value ) {
return std::integral_constant<SizeT,i-1> {}
}
else {
return i-1;
}
};
if constexpr(has_sub<Index>::value){
if constexpr(has_static_sub<Index>::value and is_integral_constant<IntT>::value){
constexpr SizeT D = index_dim<Index>::value;
return iter<0,D>
( [&](auto i) { return getDimension(*std::get<i>(ind.pack()),
iminus(depth)); },
[](auto... e) {
if constexpr( is_integral_constant<decltype(e)>::value and ... ){
return std::integral_constant<SizeT,(e + ...)> {};
}
else {
return (e + ...);
}
} );
}
else {
SizeT o = 0;
if(depth == 0u){
o = 1;
}
else if(ind.dim() == 1){
auto p = ind.pack();
if(p.size() == 0){
o = 1;
}
else {
o = getDimension(*p[0], iminus(depth));
}
}
else {
for(SizeT i = 0; i != ind.dim(); ++i){
const SizeT e = getDimension(*ind.pack()[i], iminus(depth));
if(e > o){
o = e;
}
}
}
return o;
}
}
else {
return std::integral_constant<SizeT,1> {};
}
}
}
} }
#endif #endif

View file

@ -13,6 +13,12 @@ namespace CNORZX
template <class F> template <class F>
inline decltype(auto) indexPackZip(const Vector<XIndexPtr>& a, const Vector<XIndexPtr>& b, F&& f); inline decltype(auto) indexPackZip(const Vector<XIndexPtr>& a, const Vector<XIndexPtr>& b, F&& f);
template <class Index>
constexpr decltype(auto) getIndexDepth(const Index& ind);
template <class Index, typename IntT>
constexpr decltype(auto) getDimension(const Index& ind, IntT depth);
} }
#endif #endif