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
|
// Copyright (c) 2020 Chris Richardson
// FEniCS Project
// SPDX-License-Identifier: MIT
#pragma once
/// @brief Indexing.
namespace basix::indexing
{
/// @brief Compute trivial indexing in a 1D array (for completeness).
/// @param p Index in x
/// @return 1D Index
constexpr int idx(int p) { return p; }
/// Compute indexing in a 2D triangular array compressed into a 1D
/// array. This can be used to find the index of a derivative returned
/// by FiniteElement::tabulate(). For instance to find d2N/dx2, use
/// `FiniteElement::tabulate(2, points)[idx(2, 0)];`
/// @param p Index in x
/// @param q Index in y
/// @return 1D Index
constexpr int idx(int p, int q) { return (p + q + 1) * (p + q) / 2 + q; }
/// @brief Compute indexing in a 3D tetrahedral array compressed into a
/// 1D array.
/// @param p Index in x.
/// @param q Index in y.
/// @param r Index in z.
/// @return 1D Index.
constexpr int idx(int p, int q, int r)
{
return (p + q + r) * (p + q + r + 1) * (p + q + r + 2) / 6
+ (q + r) * (q + r + 1) / 2 + r;
}
} // namespace basix::indexing
|
