1 // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at 2 // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights 3 // reserved. See files LICENSE and NOTICE for details. 4 // 5 // This file is part of CEED, a collection of benchmarks, miniapps, software 6 // libraries and APIs for efficient high-order finite element and spectral 7 // element discretizations for exascale applications. For more information and 8 // source code availability see http://github.com/ceed. 9 // 10 // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 11 // a collaborative effort of two U.S. Department of Energy organizations (Office 12 // of Science and the National Nuclear Security Administration) responsible for 13 // the planning and preparation of a capable exascale ecosystem, including 14 // software, applications, hardware, advanced system engineering and early 15 // testbed platforms, in support of the nation's exascale computing imperative. 16 17 /// @file 18 /// Geometric factors (3D) for Navier-Stokes example using PETSc 19 20 #ifndef setup_geo_h 21 #define setup_geo_h 22 23 #include <math.h> 24 #include <ceed.h> 25 26 // ***************************************************************************** 27 // This QFunction sets up the geometric factors required for integration and 28 // coordinate transformations 29 // 30 // Reference (parent) coordinates: X 31 // Physical (current) coordinates: x 32 // Change of coordinate matrix: dxdX_{i,j} = x_{i,j} (indicial notation) 33 // Inverse of change of coordinate matrix: dXdx_{i,j} = (detJ^-1) * X_{i,j} 34 // 35 // All quadrature data is stored in 10 field vector of quadrature data. 36 // 37 // We require the determinant of the Jacobian to properly compute integrals of 38 // the form: int( v u ) 39 // 40 // Determinant of Jacobian: 41 // detJ = J11*A11 + J21*A12 + J31*A13 42 // Jij = Jacobian entry ij 43 // Aij = Adjoint ij 44 // 45 // Stored: w detJ 46 // in q_data[0] 47 // 48 // We require the transpose of the inverse of the Jacobian to properly compute 49 // integrals of the form: int( gradv u ) 50 // 51 // Inverse of Jacobian: 52 // dXdx_i,j = Aij / detJ 53 // 54 // Stored: Aij / detJ 55 // in q_data[1:9] as 56 // (detJ^-1) * [A11 A12 A13] 57 // [A21 A22 A23] 58 // [A31 A32 A33] 59 // 60 // ***************************************************************************** 61 CEED_QFUNCTION(Setup)(void *ctx, CeedInt Q, 62 const CeedScalar *const *in, CeedScalar *const *out) { 63 // *INDENT-OFF* 64 // Inputs 65 const CeedScalar (*J)[3][CEED_Q_VLA] = (const CeedScalar(*)[3][CEED_Q_VLA])in[0], 66 (*w) = in[1]; 67 68 // Outputs 69 CeedScalar (*q_data)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 70 // *INDENT-ON* 71 72 CeedPragmaSIMD 73 // Quadrature Point Loop 74 for (CeedInt i=0; i<Q; i++) { 75 // Setup 76 const CeedScalar J11 = J[0][0][i]; 77 const CeedScalar J21 = J[0][1][i]; 78 const CeedScalar J31 = J[0][2][i]; 79 const CeedScalar J12 = J[1][0][i]; 80 const CeedScalar J22 = J[1][1][i]; 81 const CeedScalar J32 = J[1][2][i]; 82 const CeedScalar J13 = J[2][0][i]; 83 const CeedScalar J23 = J[2][1][i]; 84 const CeedScalar J33 = J[2][2][i]; 85 const CeedScalar A11 = J22*J33 - J23*J32; 86 const CeedScalar A12 = J13*J32 - J12*J33; 87 const CeedScalar A13 = J12*J23 - J13*J22; 88 const CeedScalar A21 = J23*J31 - J21*J33; 89 const CeedScalar A22 = J11*J33 - J13*J31; 90 const CeedScalar A23 = J13*J21 - J11*J23; 91 const CeedScalar A31 = J21*J32 - J22*J31; 92 const CeedScalar A32 = J12*J31 - J11*J32; 93 const CeedScalar A33 = J11*J22 - J12*J21; 94 const CeedScalar detJ = J11*A11 + J21*A12 + J31*A13; 95 96 // Qdata 97 // -- Interp-to-Interp q_data 98 q_data[0][i] = w[i] * detJ; 99 // -- Interp-to-Grad q_data 100 // Inverse of change of coordinate matrix: X_i,j 101 q_data[1][i] = A11 / detJ; 102 q_data[2][i] = A12 / detJ; 103 q_data[3][i] = A13 / detJ; 104 q_data[4][i] = A21 / detJ; 105 q_data[5][i] = A22 / detJ; 106 q_data[6][i] = A23 / detJ; 107 q_data[7][i] = A31 / detJ; 108 q_data[8][i] = A32 / detJ; 109 q_data[9][i] = A33 / detJ; 110 111 } // End of Quadrature Point Loop 112 113 // Return 114 return 0; 115 } 116 117 // ***************************************************************************** 118 // This QFunction sets up the geometric factor required for integration when 119 // reference coordinates are in 2D and the physical coordinates are in 3D 120 // 121 // Reference (parent) 2D coordinates: X 122 // Physical (current) 3D coordinates: x 123 // Change of coordinate matrix: 124 // dxdX_{i,j} = dx_i/dX_j (indicial notation) [3 * 2] 125 // 126 // (J1,J2,J3) is given by the cross product of the columns of dxdX_{i,j} 127 // 128 // detJb is the magnitude of (J1,J2,J3) 129 // 130 // All quadrature data is stored in 4 field vector of quadrature data. 131 // 132 // We require the determinant of the Jacobian to properly compute integrals of 133 // the form: int( u v ) 134 // 135 // Stored: w detJb 136 // in q_data_sur[0] 137 // 138 // Normal vector = (J1,J2,J3) / detJb 139 // 140 // Stored: (J1,J2,J3) / detJb 141 // in q_data_sur[1:3] as 142 // (detJb^-1) * [ J1 ] 143 // [ J2 ] 144 // [ J3 ] 145 // 146 // ***************************************************************************** 147 CEED_QFUNCTION(SetupBoundary)(void *ctx, CeedInt Q, 148 const CeedScalar *const *in, CeedScalar *const *out) { 149 // *INDENT-OFF* 150 // Inputs 151 const CeedScalar (*J)[3][CEED_Q_VLA] = (const CeedScalar(*)[3][CEED_Q_VLA])in[0], 152 (*w) = in[1]; 153 // Outputs 154 CeedScalar (*q_data_sur)[CEED_Q_VLA] = (CeedScalar(*)[CEED_Q_VLA])out[0]; 155 156 CeedPragmaSIMD 157 // Quadrature Point Loop 158 for (CeedInt i=0; i<Q; i++) { 159 // Setup 160 const CeedScalar dxdX[3][2] = {{J[0][0][i], 161 J[1][0][i]}, 162 {J[0][1][i], 163 J[1][1][i]}, 164 {J[0][2][i], 165 J[1][2][i]} 166 }; 167 // *INDENT-ON* 168 // J1, J2, and J3 are given by the cross product of the columns of dxdX 169 const CeedScalar J1 = dxdX[1][0]*dxdX[2][1] - dxdX[2][0]*dxdX[1][1]; 170 const CeedScalar J2 = dxdX[2][0]*dxdX[0][1] - dxdX[0][0]*dxdX[2][1]; 171 const CeedScalar J3 = dxdX[0][0]*dxdX[1][1] - dxdX[1][0]*dxdX[0][1]; 172 173 const CeedScalar detJb = sqrt(J1*J1 + J2*J2 + J3*J3); 174 175 // q_data_sur 176 // -- Interp-to-Interp q_data_sur 177 q_data_sur[0][i] = w[i] * detJb; 178 q_data_sur[1][i] = J1 / detJb; 179 q_data_sur[2][i] = J2 / detJb; 180 q_data_sur[3][i] = J3 / detJb; 181 182 } // End of Quadrature Point Loop 183 184 // Return 185 return 0; 186 } 187 188 // ***************************************************************************** 189 190 #endif // setup_geo_h 191