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
|
// Copyright (c) 2020 Chris Richardson
// FEniCS Project
// SPDX-License-Identifier: MIT
#pragma once
namespace basix
{
/// 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; }
/// 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
|
