meta/power/coefficients.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
// SPDX-FileCopyrightInfo: Copyright © DUNE Project contributors, see file LICENSE.md in module root
// SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception

#ifndef DUNE_LOCALFUNCTIONS_META_POWER_COEFFICIENTS_HH
#define DUNE_LOCALFUNCTIONS_META_POWER_COEFFICIENTS_HH

#include <cstddef>
#include <vector>

#include <dune/localfunctions/common/localkey.hh>

namespace Dune {


  class PowerCoefficients {
    std::vector<LocalKey> keys;

  public:

    template<class Backend>
    PowerCoefficients(const Backend &backend, std::size_t power) :
      keys(backend.size()*power)
    {
      for(std::size_t i = 0; i < backend.size(); ++i) {
        const LocalKey &k = backend.localKey(i);
        for(std::size_t d = 0; d < power; ++d)
          keys[i+d*backend.size()] =
            LocalKey(k.subEntity(), k.codim(), power*k.index() + d);
      }
    }
    inline std::size_t size() const { return keys.size(); }

    inline const LocalKey& localKey(std::size_t i) const { return keys[i]; }
  };

} // namespace Dune

#endif // DUNE_LOCALFUNCTIONS_META_POWER_COEFFICIENTS_HH