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
|
# Copyright (C) 2023-2024 Matthew Scroggs and Garth N. Wells
#
# This file is part of Basix (https://www.fenicsproject.org)
#
# SPDX-License-Identifier: MIT
"""Utility funcitons."""
import typing
from basix._basixcpp import index as _index
def index(p: int, q: typing.Optional[int] = None, r: typing.Optional[int] = None) -> int:
"""Compute the indexing in a 1D, 2D or 3D simplex.
Args:
p: Index in x.
q: Index in y.
r: Index in z.
Returns:
Index in a flattened 1D array.
"""
if q is None:
assert r is None
return _index(p)
elif r is None:
return _index(p, q)
else:
return _index(p, q, r)
|
