enumerate_slice.hpp

Go to the documentation of this file.

// Copyright (c) 2025--present, The Simons Foundation
// This file is part of TRIQS/modest and is licensed under the terms of GPLv3 or later.
// SPDX-License-Identifier: GPL-3.0-or-later
// See LICENSE in the root of this distribution for details.
 
#pragma once
#include <nda/nda.hpp>
#include <triqs/utility/generator.hpp>
 
using itertools::range;
 
namespace triqs {
  inline generator<std::pair<long, nda::range>> enumerated_sub_slices(auto sub_div) {
    long m_start = 0, n = 0;
    for (auto p : sub_div) {
      co_yield {n, {m_start, m_start + p}};
      m_start += p;
      ++n;
    }
  }
} // namespace triqs