File: subspacebasis.py

package info (click to toggle)
dune-functions 2.10.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,544 kB
  • sloc: cpp: 14,241; python: 661; makefile: 3
file content (37 lines) | stat: -rw-r--r-- 1,673 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# SPDX-FileCopyrightText: Copyright © DUNE Project contributors, see file AUTHORS.md
# SPDX-License-Identifier: LicenseRef-GPL-2.0-only-with-DUNE-exception OR LGPL-3.0-or-later

from dune.generator.generator import SimpleGenerator
from dune.common.hashit import hashIt


# A subspace basis for some descendent node, can be obtained
# by passing the indices of the corresponding tree path
# as individual arguments. Passing a tree path at once is
# currently not supported.
#
# Netsed calls to subspaceBasis(subspaceBasis(basis, i), j)
# will create a nested subspace basis. This works fine, but
# is slightly inconsistent with C++ where this is automatically
# resolved to subspaceBasis(basis, i, j).
def subspaceBasis(basis, *args):
    includes = []
    includes += list(basis.cppIncludes)
    includes += ["dune/functions/functionspacebases/subspacebasis.hh"]
    includes += ["dune/python/functions/subspacebasis.hh"]

    # In contrast to C++, the generated wrapper of SubspaceBasis
    # is only constructed from the root basis and no tree path
    # is passed. This is possible, because we only use fully
    # static tree path types that can be default constructed.
    prefixPathTypeName = "Dune::TypeTree::HybridTreePath<"
    prefixPathTypeName += ",".join("Dune::index_constant<"+str(i)+">" for i in args)
    prefixPathTypeName += ">"

    typeName = "Dune::Functions::SubspaceBasis< " + basis.cppTypeName+ "," +  prefixPathTypeName + " >"

    generator = SimpleGenerator("SubspaceBasis", "Dune::Python")

    moduleName = "subspaceBasis_" + hashIt(typeName)
    module = generator.load(includes, typeName, moduleName)
    return module.SubspaceBasis(basis)