xref: /honee/qfunctions/utils.h (revision 704b8bbec5160b50bdbeb899c12e13fec5e0b179)
1*704b8bbeSJames Wright // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2*704b8bbeSJames Wright // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3*704b8bbeSJames Wright //
4*704b8bbeSJames Wright // SPDX-License-Identifier: BSD-2-Clause
5*704b8bbeSJames Wright //
6*704b8bbeSJames Wright // This file is part of CEED:  http://github.com/ceed
7*704b8bbeSJames Wright 
8*704b8bbeSJames Wright #ifndef utils_h
9*704b8bbeSJames Wright #define utils_h
10*704b8bbeSJames Wright 
11*704b8bbeSJames Wright #include <math.h>
12*704b8bbeSJames Wright #include <ceed.h>
13*704b8bbeSJames Wright 
14*704b8bbeSJames Wright #ifndef M_PI
15*704b8bbeSJames Wright #define M_PI    3.14159265358979323846
16*704b8bbeSJames Wright #endif
17*704b8bbeSJames Wright 
18*704b8bbeSJames Wright CEED_QFUNCTION_HELPER CeedScalar Max(CeedScalar a, CeedScalar b) { return a < b ? b : a; }
19*704b8bbeSJames Wright CEED_QFUNCTION_HELPER CeedScalar Min(CeedScalar a, CeedScalar b) { return a < b ? a : b; }
20*704b8bbeSJames Wright 
21*704b8bbeSJames Wright CEED_QFUNCTION_HELPER CeedScalar Square(CeedScalar x) { return x*x; }
22*704b8bbeSJames Wright CEED_QFUNCTION_HELPER CeedScalar Cube(CeedScalar x) { return x*x*x; }
23*704b8bbeSJames Wright 
24*704b8bbeSJames Wright // @brief Dot product of 3 element vectors
25*704b8bbeSJames Wright CEED_QFUNCTION_HELPER CeedScalar Dot3(const CeedScalar u[3],
26*704b8bbeSJames Wright                                       const CeedScalar v[3]) {
27*704b8bbeSJames Wright   return u[0]*v[0] + u[1]*v[1] + u[2]*v[2];
28*704b8bbeSJames Wright }
29*704b8bbeSJames Wright 
30*704b8bbeSJames Wright // @brief Unpack Kelvin-Mandel notation symmetric tensor into full tensor
31*704b8bbeSJames Wright CEED_QFUNCTION_HELPER void KMUnpack(const CeedScalar v[6], CeedScalar A[3][3]) {
32*704b8bbeSJames Wright   const CeedScalar weight = 1 / sqrt(2.);
33*704b8bbeSJames Wright   A[0][0] = v[0];
34*704b8bbeSJames Wright   A[1][1] = v[1];
35*704b8bbeSJames Wright   A[2][2] = v[2];
36*704b8bbeSJames Wright   A[2][1] = A[1][2] = weight * v[3];
37*704b8bbeSJames Wright   A[2][0] = A[0][2] = weight * v[4];
38*704b8bbeSJames Wright   A[1][0] = A[0][1] = weight * v[5];
39*704b8bbeSJames Wright }
40*704b8bbeSJames Wright 
41*704b8bbeSJames Wright #endif // utils_h
42