1c4762a1bSJed Brown static char help[] = "Second Order TVD Finite Volume Example.\n"; 2c4762a1bSJed Brown /*F 3c4762a1bSJed Brown 4c4762a1bSJed Brown We use a second order TVD finite volume method to evolve a system of PDEs. Our simple upwinded residual evaluation loops 5c4762a1bSJed Brown over all mesh faces and uses a Riemann solver to produce the flux given the face geometry and cell values, 6c4762a1bSJed Brown \begin{equation} 7c4762a1bSJed Brown f_i = \mathrm{riemann}(\mathrm{phys}, p_\mathrm{centroid}, \hat n, x^L, x^R) 8c4762a1bSJed Brown \end{equation} 9c4762a1bSJed Brown and then update the cell values given the cell volume. 10c4762a1bSJed Brown \begin{eqnarray} 11c4762a1bSJed Brown f^L_i &-=& \frac{f_i}{vol^L} \\ 12c4762a1bSJed Brown f^R_i &+=& \frac{f_i}{vol^R} 13c4762a1bSJed Brown \end{eqnarray} 14c4762a1bSJed Brown 15c4762a1bSJed Brown As an example, we can consider the shallow water wave equation, 16c4762a1bSJed Brown \begin{eqnarray} 17c4762a1bSJed Brown h_t + \nabla\cdot \left( uh \right) &=& 0 \\ 18c4762a1bSJed Brown (uh)_t + \nabla\cdot \left( u\otimes uh + \frac{g h^2}{2} I \right) &=& 0 19c4762a1bSJed Brown \end{eqnarray} 20c4762a1bSJed Brown where $h$ is wave height, $u$ is wave velocity, and $g$ is the acceleration due to gravity. 21c4762a1bSJed Brown 22c4762a1bSJed Brown A representative Riemann solver for the shallow water equations is given in the PhysicsRiemann_SW() function, 23c4762a1bSJed Brown \begin{eqnarray} 24c4762a1bSJed Brown f^{L,R}_h &=& uh^{L,R} \cdot \hat n \\ 25c4762a1bSJed Brown f^{L,R}_{uh} &=& \frac{f^{L,R}_h}{h^{L,R}} uh^{L,R} + g (h^{L,R})^2 \hat n \\ 26c4762a1bSJed Brown c^{L,R} &=& \sqrt{g h^{L,R}} \\ 27c4762a1bSJed Brown s &=& \max\left( \left|\frac{uh^L \cdot \hat n}{h^L}\right| + c^L, \left|\frac{uh^R \cdot \hat n}{h^R}\right| + c^R \right) \\ 28c4762a1bSJed Brown f_i &=& \frac{A_\mathrm{face}}{2} \left( f^L_i + f^R_i + s \left( x^L_i - x^R_i \right) \right) 29c4762a1bSJed Brown \end{eqnarray} 30c4762a1bSJed Brown where $c$ is the local gravity wave speed and $f_i$ is a Rusanov flux. 31c4762a1bSJed Brown 32c4762a1bSJed Brown The more sophisticated residual evaluation in RHSFunctionLocal_LS() uses a least-squares fit to a quadratic polynomial 33c4762a1bSJed Brown over a neighborhood of the given element. 34c4762a1bSJed Brown 35c4762a1bSJed Brown The mesh is read in from an ExodusII file, usually generated by Cubit. 36c4762a1bSJed Brown F*/ 37c4762a1bSJed Brown #include <petscdmplex.h> 38c4762a1bSJed Brown #include <petscdmforest.h> 39c4762a1bSJed Brown #include <petscds.h> 40c4762a1bSJed Brown #include <petscts.h> 41c4762a1bSJed Brown 42c4762a1bSJed Brown #define DIM 2 /* Geometric dimension */ 43c4762a1bSJed Brown 448d2c18caSMukkund Sunjii static PetscFunctionList PhysicsList, PhysicsRiemannList_SW; 45c4762a1bSJed Brown 46c4762a1bSJed Brown /* Represents continuum physical equations. */ 47c4762a1bSJed Brown typedef struct _n_Physics *Physics; 48c4762a1bSJed Brown 49c4762a1bSJed Brown /* Physical model includes boundary conditions, initial conditions, and functionals of interest. It is 50c4762a1bSJed Brown * discretization-independent, but its members depend on the scenario being solved. */ 51c4762a1bSJed Brown typedef struct _n_Model *Model; 52c4762a1bSJed Brown 53c4762a1bSJed Brown /* 'User' implements a discretization of a continuous model. */ 54c4762a1bSJed Brown typedef struct _n_User *User; 55c4762a1bSJed Brown typedef PetscErrorCode (*SolutionFunction)(Model, PetscReal, const PetscReal *, PetscScalar *, void *); 5645480ffeSMatthew G. Knepley typedef PetscErrorCode (*SetUpBCFunction)(DM, PetscDS, Physics); 57c4762a1bSJed Brown typedef PetscErrorCode (*FunctionalFunction)(Model, PetscReal, const PetscReal *, const PetscScalar *, PetscReal *, void *); 58c4762a1bSJed Brown typedef PetscErrorCode (*SetupFields)(Physics, PetscSection); 59c4762a1bSJed Brown static PetscErrorCode ModelSolutionSetDefault(Model, SolutionFunction, void *); 60c4762a1bSJed Brown static PetscErrorCode ModelFunctionalRegister(Model, const char *, PetscInt *, FunctionalFunction, void *); 61c4762a1bSJed Brown static PetscErrorCode OutputVTK(DM, const char *, PetscViewer *); 62c4762a1bSJed Brown 63c4762a1bSJed Brown struct FieldDescription { 64c4762a1bSJed Brown const char *name; 65c4762a1bSJed Brown PetscInt dof; 66c4762a1bSJed Brown }; 67c4762a1bSJed Brown 68c4762a1bSJed Brown typedef struct _n_FunctionalLink *FunctionalLink; 69c4762a1bSJed Brown struct _n_FunctionalLink { 70c4762a1bSJed Brown char *name; 71c4762a1bSJed Brown FunctionalFunction func; 72c4762a1bSJed Brown void *ctx; 73c4762a1bSJed Brown PetscInt offset; 74c4762a1bSJed Brown FunctionalLink next; 75c4762a1bSJed Brown }; 76c4762a1bSJed Brown 77c4762a1bSJed Brown struct _n_Physics { 78c4762a1bSJed Brown PetscRiemannFunc riemann; 79c4762a1bSJed Brown PetscInt dof; /* number of degrees of freedom per cell */ 80c4762a1bSJed Brown PetscReal maxspeed; /* kludge to pick initial time step, need to add monitoring and step control */ 81c4762a1bSJed Brown void *data; 82c4762a1bSJed Brown PetscInt nfields; 83c4762a1bSJed Brown const struct FieldDescription *field_desc; 84c4762a1bSJed Brown }; 85c4762a1bSJed Brown 86c4762a1bSJed Brown struct _n_Model { 87c4762a1bSJed Brown MPI_Comm comm; /* Does not do collective communicaton, but some error conditions can be collective */ 88c4762a1bSJed Brown Physics physics; 89c4762a1bSJed Brown FunctionalLink functionalRegistry; 90c4762a1bSJed Brown PetscInt maxComputed; 91c4762a1bSJed Brown PetscInt numMonitored; 92c4762a1bSJed Brown FunctionalLink *functionalMonitored; 93c4762a1bSJed Brown PetscInt numCall; 94c4762a1bSJed Brown FunctionalLink *functionalCall; 95c4762a1bSJed Brown SolutionFunction solution; 96c4762a1bSJed Brown SetUpBCFunction setupbc; 97c4762a1bSJed Brown void *solutionctx; 98c4762a1bSJed Brown PetscReal maxspeed; /* estimate of global maximum speed (for CFL calculation) */ 99c4762a1bSJed Brown PetscReal bounds[2 * DIM]; 100c4762a1bSJed Brown PetscErrorCode (*errorIndicator)(PetscInt, PetscReal, PetscInt, const PetscScalar[], const PetscScalar[], PetscReal *, void *); 101c4762a1bSJed Brown void *errorCtx; 102c4762a1bSJed Brown }; 103c4762a1bSJed Brown 104c4762a1bSJed Brown struct _n_User { 105c4762a1bSJed Brown PetscInt vtkInterval; /* For monitor */ 106c4762a1bSJed Brown char outputBasename[PETSC_MAX_PATH_LEN]; /* Basename for output files */ 107c4762a1bSJed Brown PetscInt monitorStepOffset; 108c4762a1bSJed Brown Model model; 109c4762a1bSJed Brown PetscBool vtkmon; 110c4762a1bSJed Brown }; 111c4762a1bSJed Brown 1129371c9d4SSatish Balay static inline PetscReal DotDIMReal(const PetscReal *x, const PetscReal *y) { 113c4762a1bSJed Brown PetscInt i; 114c4762a1bSJed Brown PetscReal prod = 0.0; 115c4762a1bSJed Brown 116c4762a1bSJed Brown for (i = 0; i < DIM; i++) prod += x[i] * y[i]; 117c4762a1bSJed Brown return prod; 118c4762a1bSJed Brown } 1199371c9d4SSatish Balay static inline PetscReal NormDIM(const PetscReal *x) { 1209371c9d4SSatish Balay return PetscSqrtReal(PetscAbsReal(DotDIMReal(x, x))); 1219371c9d4SSatish Balay } 122c4762a1bSJed Brown 1239371c9d4SSatish Balay static inline PetscReal Dot2Real(const PetscReal *x, const PetscReal *y) { 1249371c9d4SSatish Balay return x[0] * y[0] + x[1] * y[1]; 1259371c9d4SSatish Balay } 1269371c9d4SSatish Balay static inline PetscReal Norm2Real(const PetscReal *x) { 1279371c9d4SSatish Balay return PetscSqrtReal(PetscAbsReal(Dot2Real(x, x))); 1289371c9d4SSatish Balay } 1299371c9d4SSatish Balay static inline void Normalize2Real(PetscReal *x) { 1309371c9d4SSatish Balay PetscReal a = 1. / Norm2Real(x); 1319371c9d4SSatish Balay x[0] *= a; 1329371c9d4SSatish Balay x[1] *= a; 1339371c9d4SSatish Balay } 1349371c9d4SSatish Balay static inline void Waxpy2Real(PetscReal a, const PetscReal *x, const PetscReal *y, PetscReal *w) { 1359371c9d4SSatish Balay w[0] = a * x[0] + y[0]; 1369371c9d4SSatish Balay w[1] = a * x[1] + y[1]; 1379371c9d4SSatish Balay } 1389371c9d4SSatish Balay static inline void Scale2Real(PetscReal a, const PetscReal *x, PetscReal *y) { 1399371c9d4SSatish Balay y[0] = a * x[0]; 1409371c9d4SSatish Balay y[1] = a * x[1]; 1419371c9d4SSatish Balay } 142c4762a1bSJed Brown 143c4762a1bSJed Brown /******************* Advect ********************/ 1449371c9d4SSatish Balay typedef enum { 1459371c9d4SSatish Balay ADVECT_SOL_TILTED, 1469371c9d4SSatish Balay ADVECT_SOL_BUMP, 1479371c9d4SSatish Balay ADVECT_SOL_BUMP_CAVITY 1489371c9d4SSatish Balay } AdvectSolType; 149c4762a1bSJed Brown static const char *const AdvectSolTypes[] = {"TILTED", "BUMP", "BUMP_CAVITY", "AdvectSolType", "ADVECT_SOL_", 0}; 1509371c9d4SSatish Balay typedef enum { 1519371c9d4SSatish Balay ADVECT_SOL_BUMP_CONE, 1529371c9d4SSatish Balay ADVECT_SOL_BUMP_COS 1539371c9d4SSatish Balay } AdvectSolBumpType; 154c4762a1bSJed Brown static const char *const AdvectSolBumpTypes[] = {"CONE", "COS", "AdvectSolBumpType", "ADVECT_SOL_BUMP_", 0}; 155c4762a1bSJed Brown 156c4762a1bSJed Brown typedef struct { 157c4762a1bSJed Brown PetscReal wind[DIM]; 158c4762a1bSJed Brown } Physics_Advect_Tilted; 159c4762a1bSJed Brown typedef struct { 160c4762a1bSJed Brown PetscReal center[DIM]; 161c4762a1bSJed Brown PetscReal radius; 162c4762a1bSJed Brown AdvectSolBumpType type; 163c4762a1bSJed Brown } Physics_Advect_Bump; 164c4762a1bSJed Brown 165c4762a1bSJed Brown typedef struct { 166c4762a1bSJed Brown PetscReal inflowState; 167c4762a1bSJed Brown AdvectSolType soltype; 1689371c9d4SSatish Balay union 1699371c9d4SSatish Balay { 170c4762a1bSJed Brown Physics_Advect_Tilted tilted; 171c4762a1bSJed Brown Physics_Advect_Bump bump; 172c4762a1bSJed Brown } sol; 173c4762a1bSJed Brown struct { 174c4762a1bSJed Brown PetscInt Solution; 175c4762a1bSJed Brown PetscInt Error; 176c4762a1bSJed Brown } functional; 177c4762a1bSJed Brown } Physics_Advect; 178c4762a1bSJed Brown 1799371c9d4SSatish Balay static const struct FieldDescription PhysicsFields_Advect[] = { 1809371c9d4SSatish Balay {"U", 1}, 1819371c9d4SSatish Balay {NULL, 0} 1829371c9d4SSatish Balay }; 183c4762a1bSJed Brown 1849371c9d4SSatish Balay static PetscErrorCode PhysicsBoundary_Advect_Inflow(PetscReal time, const PetscReal *c, const PetscReal *n, const PetscScalar *xI, PetscScalar *xG, void *ctx) { 185c4762a1bSJed Brown Physics phys = (Physics)ctx; 186c4762a1bSJed Brown Physics_Advect *advect = (Physics_Advect *)phys->data; 187c4762a1bSJed Brown 188c4762a1bSJed Brown PetscFunctionBeginUser; 189c4762a1bSJed Brown xG[0] = advect->inflowState; 190c4762a1bSJed Brown PetscFunctionReturn(0); 191c4762a1bSJed Brown } 192c4762a1bSJed Brown 1939371c9d4SSatish Balay static PetscErrorCode PhysicsBoundary_Advect_Outflow(PetscReal time, const PetscReal *c, const PetscReal *n, const PetscScalar *xI, PetscScalar *xG, void *ctx) { 194c4762a1bSJed Brown PetscFunctionBeginUser; 195c4762a1bSJed Brown xG[0] = xI[0]; 196c4762a1bSJed Brown PetscFunctionReturn(0); 197c4762a1bSJed Brown } 198c4762a1bSJed Brown 1999371c9d4SSatish Balay static void PhysicsRiemann_Advect(PetscInt dim, PetscInt Nf, const PetscReal *qp, const PetscReal *n, const PetscScalar *xL, const PetscScalar *xR, PetscInt numConstants, const PetscScalar constants[], PetscScalar *flux, Physics phys) { 200c4762a1bSJed Brown Physics_Advect *advect = (Physics_Advect *)phys->data; 201c4762a1bSJed Brown PetscReal wind[DIM], wn; 202c4762a1bSJed Brown 203c4762a1bSJed Brown switch (advect->soltype) { 204c4762a1bSJed Brown case ADVECT_SOL_TILTED: { 205c4762a1bSJed Brown Physics_Advect_Tilted *tilted = &advect->sol.tilted; 206c4762a1bSJed Brown wind[0] = tilted->wind[0]; 207c4762a1bSJed Brown wind[1] = tilted->wind[1]; 208c4762a1bSJed Brown } break; 209c4762a1bSJed Brown case ADVECT_SOL_BUMP: 210c4762a1bSJed Brown wind[0] = -qp[1]; 211c4762a1bSJed Brown wind[1] = qp[0]; 212c4762a1bSJed Brown break; 2139371c9d4SSatish Balay case ADVECT_SOL_BUMP_CAVITY: { 214c4762a1bSJed Brown PetscInt i; 215c4762a1bSJed Brown PetscReal comp2[3] = {0., 0., 0.}, rad2; 216c4762a1bSJed Brown 217c4762a1bSJed Brown rad2 = 0.; 218c4762a1bSJed Brown for (i = 0; i < dim; i++) { 219c4762a1bSJed Brown comp2[i] = qp[i] * qp[i]; 220c4762a1bSJed Brown rad2 += comp2[i]; 221c4762a1bSJed Brown } 222c4762a1bSJed Brown 223c4762a1bSJed Brown wind[0] = -qp[1]; 224c4762a1bSJed Brown wind[1] = qp[0]; 225c4762a1bSJed Brown if (rad2 > 1.) { 226c4762a1bSJed Brown PetscInt maxI = 0; 227c4762a1bSJed Brown PetscReal maxComp2 = comp2[0]; 228c4762a1bSJed Brown 229c4762a1bSJed Brown for (i = 1; i < dim; i++) { 230c4762a1bSJed Brown if (comp2[i] > maxComp2) { 231c4762a1bSJed Brown maxI = i; 232c4762a1bSJed Brown maxComp2 = comp2[i]; 233c4762a1bSJed Brown } 234c4762a1bSJed Brown } 235c4762a1bSJed Brown wind[maxI] = 0.; 236c4762a1bSJed Brown } 2379371c9d4SSatish Balay } break; 2389371c9d4SSatish Balay default: { 239c4762a1bSJed Brown PetscInt i; 240c4762a1bSJed Brown for (i = 0; i < DIM; ++i) wind[i] = 0.0; 241c4762a1bSJed Brown } 24298921bdaSJacob Faibussowitsch /* default: SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for solution type %s",AdvectSolBumpTypes[advect->soltype]); */ 243c4762a1bSJed Brown } 244c4762a1bSJed Brown wn = Dot2Real(wind, n); 245c4762a1bSJed Brown flux[0] = (wn > 0 ? xL[0] : xR[0]) * wn; 246c4762a1bSJed Brown } 247c4762a1bSJed Brown 2489371c9d4SSatish Balay static PetscErrorCode PhysicsSolution_Advect(Model mod, PetscReal time, const PetscReal *x, PetscScalar *u, void *ctx) { 249c4762a1bSJed Brown Physics phys = (Physics)ctx; 250c4762a1bSJed Brown Physics_Advect *advect = (Physics_Advect *)phys->data; 251c4762a1bSJed Brown 252c4762a1bSJed Brown PetscFunctionBeginUser; 253c4762a1bSJed Brown switch (advect->soltype) { 254c4762a1bSJed Brown case ADVECT_SOL_TILTED: { 255c4762a1bSJed Brown PetscReal x0[DIM]; 256c4762a1bSJed Brown Physics_Advect_Tilted *tilted = &advect->sol.tilted; 257c4762a1bSJed Brown Waxpy2Real(-time, tilted->wind, x, x0); 258c4762a1bSJed Brown if (x0[1] > 0) u[0] = 1. * x[0] + 3. * x[1]; 259c4762a1bSJed Brown else u[0] = advect->inflowState; 260c4762a1bSJed Brown } break; 261c4762a1bSJed Brown case ADVECT_SOL_BUMP_CAVITY: 262c4762a1bSJed Brown case ADVECT_SOL_BUMP: { 263c4762a1bSJed Brown Physics_Advect_Bump *bump = &advect->sol.bump; 264c4762a1bSJed Brown PetscReal x0[DIM], v[DIM], r, cost, sint; 265c4762a1bSJed Brown cost = PetscCosReal(time); 266c4762a1bSJed Brown sint = PetscSinReal(time); 267c4762a1bSJed Brown x0[0] = cost * x[0] + sint * x[1]; 268c4762a1bSJed Brown x0[1] = -sint * x[0] + cost * x[1]; 269c4762a1bSJed Brown Waxpy2Real(-1, bump->center, x0, v); 270c4762a1bSJed Brown r = Norm2Real(v); 271c4762a1bSJed Brown switch (bump->type) { 2729371c9d4SSatish Balay case ADVECT_SOL_BUMP_CONE: u[0] = PetscMax(1 - r / bump->radius, 0); break; 2739371c9d4SSatish Balay case ADVECT_SOL_BUMP_COS: u[0] = 0.5 + 0.5 * PetscCosReal(PetscMin(r / bump->radius, 1) * PETSC_PI); break; 274c4762a1bSJed Brown } 275c4762a1bSJed Brown } break; 276c4762a1bSJed Brown default: SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unknown solution type"); 277c4762a1bSJed Brown } 278c4762a1bSJed Brown PetscFunctionReturn(0); 279c4762a1bSJed Brown } 280c4762a1bSJed Brown 2819371c9d4SSatish Balay static PetscErrorCode PhysicsFunctional_Advect(Model mod, PetscReal time, const PetscReal *x, const PetscScalar *y, PetscReal *f, void *ctx) { 282c4762a1bSJed Brown Physics phys = (Physics)ctx; 283c4762a1bSJed Brown Physics_Advect *advect = (Physics_Advect *)phys->data; 284391faea4SSatish Balay PetscScalar yexact[1] = {0.0}; 285c4762a1bSJed Brown 286c4762a1bSJed Brown PetscFunctionBeginUser; 2879566063dSJacob Faibussowitsch PetscCall(PhysicsSolution_Advect(mod, time, x, yexact, phys)); 288c4762a1bSJed Brown f[advect->functional.Solution] = PetscRealPart(y[0]); 289c4762a1bSJed Brown f[advect->functional.Error] = PetscAbsScalar(y[0] - yexact[0]); 290c4762a1bSJed Brown PetscFunctionReturn(0); 291c4762a1bSJed Brown } 292c4762a1bSJed Brown 2939371c9d4SSatish Balay static PetscErrorCode SetUpBC_Advect(DM dm, PetscDS prob, Physics phys) { 294c4762a1bSJed Brown const PetscInt inflowids[] = {100, 200, 300}, outflowids[] = {101}; 29545480ffeSMatthew G. Knepley DMLabel label; 296c4762a1bSJed Brown 297c4762a1bSJed Brown PetscFunctionBeginUser; 298c4762a1bSJed Brown /* Register "canned" boundary conditions and defaults for where to apply. */ 2999566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "Face Sets", &label)); 300dd39110bSPierre Jolivet PetscCall(PetscDSAddBoundary(prob, DM_BC_NATURAL_RIEMANN, "inflow", label, PETSC_STATIC_ARRAY_LENGTH(inflowids), inflowids, 0, 0, NULL, (void (*)(void))PhysicsBoundary_Advect_Inflow, NULL, phys, NULL)); 301dd39110bSPierre Jolivet PetscCall(PetscDSAddBoundary(prob, DM_BC_NATURAL_RIEMANN, "outflow", label, PETSC_STATIC_ARRAY_LENGTH(outflowids), outflowids, 0, 0, NULL, (void (*)(void))PhysicsBoundary_Advect_Outflow, NULL, phys, NULL)); 302c4762a1bSJed Brown PetscFunctionReturn(0); 303c4762a1bSJed Brown } 304c4762a1bSJed Brown 3059371c9d4SSatish Balay static PetscErrorCode PhysicsCreate_Advect(Model mod, Physics phys, PetscOptionItems *PetscOptionsObject) { 306c4762a1bSJed Brown Physics_Advect *advect; 307c4762a1bSJed Brown 308c4762a1bSJed Brown PetscFunctionBeginUser; 309c4762a1bSJed Brown phys->field_desc = PhysicsFields_Advect; 310c4762a1bSJed Brown phys->riemann = (PetscRiemannFunc)PhysicsRiemann_Advect; 3119566063dSJacob Faibussowitsch PetscCall(PetscNew(&advect)); 312c4762a1bSJed Brown phys->data = advect; 313c4762a1bSJed Brown mod->setupbc = SetUpBC_Advect; 314c4762a1bSJed Brown 315d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "Advect options"); 316c4762a1bSJed Brown { 317c4762a1bSJed Brown PetscInt two = 2, dof = 1; 318c4762a1bSJed Brown advect->soltype = ADVECT_SOL_TILTED; 3199566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-advect_sol_type", "solution type", "", AdvectSolTypes, (PetscEnum)advect->soltype, (PetscEnum *)&advect->soltype, NULL)); 320c4762a1bSJed Brown switch (advect->soltype) { 321c4762a1bSJed Brown case ADVECT_SOL_TILTED: { 322c4762a1bSJed Brown Physics_Advect_Tilted *tilted = &advect->sol.tilted; 323c4762a1bSJed Brown two = 2; 324c4762a1bSJed Brown tilted->wind[0] = 0.0; 325c4762a1bSJed Brown tilted->wind[1] = 1.0; 3269566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-advect_tilted_wind", "background wind vx,vy", "", tilted->wind, &two, NULL)); 327c4762a1bSJed Brown advect->inflowState = -2.0; 3289566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-advect_tilted_inflow", "Inflow state", "", &advect->inflowState, &dof, NULL)); 329c4762a1bSJed Brown phys->maxspeed = Norm2Real(tilted->wind); 330c4762a1bSJed Brown } break; 331c4762a1bSJed Brown case ADVECT_SOL_BUMP_CAVITY: 332c4762a1bSJed Brown case ADVECT_SOL_BUMP: { 333c4762a1bSJed Brown Physics_Advect_Bump *bump = &advect->sol.bump; 334c4762a1bSJed Brown two = 2; 335c4762a1bSJed Brown bump->center[0] = 2.; 336c4762a1bSJed Brown bump->center[1] = 0.; 3379566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-advect_bump_center", "location of center of bump x,y", "", bump->center, &two, NULL)); 338c4762a1bSJed Brown bump->radius = 0.9; 3399566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-advect_bump_radius", "radius of bump", "", bump->radius, &bump->radius, NULL)); 340c4762a1bSJed Brown bump->type = ADVECT_SOL_BUMP_CONE; 3419566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-advect_bump_type", "type of bump", "", AdvectSolBumpTypes, (PetscEnum)bump->type, (PetscEnum *)&bump->type, NULL)); 342c4762a1bSJed Brown phys->maxspeed = 3.; /* radius of mesh, kludge */ 343c4762a1bSJed Brown } break; 344c4762a1bSJed Brown } 345c4762a1bSJed Brown } 346d0609cedSBarry Smith PetscOptionsHeadEnd(); 347c4762a1bSJed Brown /* Initial/transient solution with default boundary conditions */ 3489566063dSJacob Faibussowitsch PetscCall(ModelSolutionSetDefault(mod, PhysicsSolution_Advect, phys)); 349c4762a1bSJed Brown /* Register "canned" functionals */ 3509566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Solution", &advect->functional.Solution, PhysicsFunctional_Advect, phys)); 3519566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Error", &advect->functional.Error, PhysicsFunctional_Advect, phys)); 352c4762a1bSJed Brown PetscFunctionReturn(0); 353c4762a1bSJed Brown } 354c4762a1bSJed Brown 355c4762a1bSJed Brown /******************* Shallow Water ********************/ 356c4762a1bSJed Brown typedef struct { 357c4762a1bSJed Brown PetscReal gravity; 358c4762a1bSJed Brown PetscReal boundaryHeight; 359c4762a1bSJed Brown struct { 360c4762a1bSJed Brown PetscInt Height; 361c4762a1bSJed Brown PetscInt Speed; 362c4762a1bSJed Brown PetscInt Energy; 363c4762a1bSJed Brown } functional; 364c4762a1bSJed Brown } Physics_SW; 365c4762a1bSJed Brown typedef struct { 366c4762a1bSJed Brown PetscReal h; 367c4762a1bSJed Brown PetscReal uh[DIM]; 368c4762a1bSJed Brown } SWNode; 3699371c9d4SSatish Balay typedef union 3709371c9d4SSatish Balay { 371c4762a1bSJed Brown SWNode swnode; 372c4762a1bSJed Brown PetscReal vals[DIM + 1]; 373c4762a1bSJed Brown } SWNodeUnion; 374c4762a1bSJed Brown 3759371c9d4SSatish Balay static const struct FieldDescription PhysicsFields_SW[] = { 3769371c9d4SSatish Balay {"Height", 1 }, 3779371c9d4SSatish Balay {"Momentum", DIM}, 3789371c9d4SSatish Balay {NULL, 0 } 3799371c9d4SSatish Balay }; 380c4762a1bSJed Brown 381c4762a1bSJed Brown /* 382c4762a1bSJed Brown * h_t + div(uh) = 0 383c4762a1bSJed Brown * (uh)_t + div (u\otimes uh + g h^2 / 2 I) = 0 384c4762a1bSJed Brown * 385c4762a1bSJed Brown * */ 3869371c9d4SSatish Balay static PetscErrorCode SWFlux(Physics phys, const PetscReal *n, const SWNode *x, SWNode *f) { 387c4762a1bSJed Brown Physics_SW *sw = (Physics_SW *)phys->data; 388c4762a1bSJed Brown PetscReal uhn, u[DIM]; 389c4762a1bSJed Brown PetscInt i; 390c4762a1bSJed Brown 391c4762a1bSJed Brown PetscFunctionBeginUser; 392c4762a1bSJed Brown Scale2Real(1. / x->h, x->uh, u); 393c4762a1bSJed Brown uhn = x->uh[0] * n[0] + x->uh[1] * n[1]; 394c4762a1bSJed Brown f->h = uhn; 395c4762a1bSJed Brown for (i = 0; i < DIM; i++) f->uh[i] = u[i] * uhn + sw->gravity * PetscSqr(x->h) * n[i]; 396c4762a1bSJed Brown PetscFunctionReturn(0); 397c4762a1bSJed Brown } 398c4762a1bSJed Brown 3999371c9d4SSatish Balay static PetscErrorCode PhysicsBoundary_SW_Wall(PetscReal time, const PetscReal *c, const PetscReal *n, const PetscScalar *xI, PetscScalar *xG, void *ctx) { 400c4762a1bSJed Brown PetscFunctionBeginUser; 401c4762a1bSJed Brown xG[0] = xI[0]; 402c4762a1bSJed Brown xG[1] = -xI[1]; 403c4762a1bSJed Brown xG[2] = -xI[2]; 404c4762a1bSJed Brown PetscFunctionReturn(0); 405c4762a1bSJed Brown } 406c4762a1bSJed Brown 4079371c9d4SSatish Balay static void PhysicsRiemann_SW_HLL(PetscInt dim, PetscInt Nf, const PetscReal *qp, const PetscReal *n, const PetscScalar *xL, const PetscScalar *xR, PetscInt numConstants, const PetscScalar constants[], PetscScalar *flux, Physics phys) { 4088d2c18caSMukkund Sunjii Physics_SW *sw = (Physics_SW *)phys->data; 4098d2c18caSMukkund Sunjii PetscReal aL, aR; 4108d2c18caSMukkund Sunjii PetscReal nn[DIM]; 4118d2c18caSMukkund Sunjii #if !defined(PETSC_USE_COMPLEX) 4128d2c18caSMukkund Sunjii const SWNode *uL = (const SWNode *)xL, *uR = (const SWNode *)xR; 4138d2c18caSMukkund Sunjii #else 4148d2c18caSMukkund Sunjii SWNodeUnion uLreal, uRreal; 4158d2c18caSMukkund Sunjii const SWNode *uL = &uLreal.swnode; 4168d2c18caSMukkund Sunjii const SWNode *uR = &uRreal.swnode; 4178d2c18caSMukkund Sunjii #endif 4188d2c18caSMukkund Sunjii SWNodeUnion fL, fR; 4198d2c18caSMukkund Sunjii PetscInt i; 4208d2c18caSMukkund Sunjii PetscReal zero = 0.; 4218d2c18caSMukkund Sunjii 4228d2c18caSMukkund Sunjii #if defined(PETSC_USE_COMPLEX) 4239371c9d4SSatish Balay uLreal.swnode.h = 0; 4249371c9d4SSatish Balay uRreal.swnode.h = 0; 4258d2c18caSMukkund Sunjii for (i = 0; i < 1 + dim; i++) uLreal.vals[i] = PetscRealPart(xL[i]); 4268d2c18caSMukkund Sunjii for (i = 0; i < 1 + dim; i++) uRreal.vals[i] = PetscRealPart(xR[i]); 4278d2c18caSMukkund Sunjii #endif 4288d2c18caSMukkund Sunjii if (uL->h <= 0 || uR->h <= 0) { 4298d2c18caSMukkund Sunjii for (i = 0; i < 1 + dim; i++) flux[i] = zero; 4308d2c18caSMukkund Sunjii return; 4318d2c18caSMukkund Sunjii } /* SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Reconstructed thickness is negative"); */ 4328d2c18caSMukkund Sunjii nn[0] = n[0]; 4338d2c18caSMukkund Sunjii nn[1] = n[1]; 4348d2c18caSMukkund Sunjii Normalize2Real(nn); 4358d2c18caSMukkund Sunjii SWFlux(phys, nn, uL, &(fL.swnode)); 4368d2c18caSMukkund Sunjii SWFlux(phys, nn, uR, &(fR.swnode)); 4378d2c18caSMukkund Sunjii /* gravity wave speed */ 4388d2c18caSMukkund Sunjii aL = PetscSqrtReal(sw->gravity * uL->h); 4398d2c18caSMukkund Sunjii aR = PetscSqrtReal(sw->gravity * uR->h); 4408d2c18caSMukkund Sunjii // Defining u_tilda and v_tilda as u and v 4418d2c18caSMukkund Sunjii PetscReal u_L, u_R; 4428d2c18caSMukkund Sunjii u_L = Dot2Real(uL->uh, nn) / uL->h; 4438d2c18caSMukkund Sunjii u_R = Dot2Real(uR->uh, nn) / uR->h; 4448d2c18caSMukkund Sunjii PetscReal sL, sR; 4458d2c18caSMukkund Sunjii sL = PetscMin(u_L - aL, u_R - aR); 4468d2c18caSMukkund Sunjii sR = PetscMax(u_L + aL, u_R + aR); 4478d2c18caSMukkund Sunjii if (sL > zero) { 4489371c9d4SSatish Balay for (i = 0; i < dim + 1; i++) { flux[i] = fL.vals[i] * Norm2Real(n); } 4498d2c18caSMukkund Sunjii } else if (sR < zero) { 4509371c9d4SSatish Balay for (i = 0; i < dim + 1; i++) { flux[i] = fR.vals[i] * Norm2Real(n); } 4518d2c18caSMukkund Sunjii } else { 4529371c9d4SSatish Balay for (i = 0; i < dim + 1; i++) { flux[i] = ((sR * fL.vals[i] - sL * fR.vals[i] + sR * sL * (xR[i] - xL[i])) / (sR - sL)) * Norm2Real(n); } 4538d2c18caSMukkund Sunjii } 4548d2c18caSMukkund Sunjii } 4558d2c18caSMukkund Sunjii 4569371c9d4SSatish Balay static void PhysicsRiemann_SW_Rusanov(PetscInt dim, PetscInt Nf, const PetscReal *qp, const PetscReal *n, const PetscScalar *xL, const PetscScalar *xR, PetscInt numConstants, const PetscScalar constants[], PetscScalar *flux, Physics phys) { 457c4762a1bSJed Brown Physics_SW *sw = (Physics_SW *)phys->data; 458c4762a1bSJed Brown PetscReal cL, cR, speed; 459c4762a1bSJed Brown PetscReal nn[DIM]; 460c4762a1bSJed Brown #if !defined(PETSC_USE_COMPLEX) 461c4762a1bSJed Brown const SWNode *uL = (const SWNode *)xL, *uR = (const SWNode *)xR; 462c4762a1bSJed Brown #else 463c4762a1bSJed Brown SWNodeUnion uLreal, uRreal; 464c4762a1bSJed Brown const SWNode *uL = &uLreal.swnode; 465c4762a1bSJed Brown const SWNode *uR = &uRreal.swnode; 466c4762a1bSJed Brown #endif 467c4762a1bSJed Brown SWNodeUnion fL, fR; 468c4762a1bSJed Brown PetscInt i; 469c4762a1bSJed Brown PetscReal zero = 0.; 470c4762a1bSJed Brown 471c4762a1bSJed Brown #if defined(PETSC_USE_COMPLEX) 4729371c9d4SSatish Balay uLreal.swnode.h = 0; 4739371c9d4SSatish Balay uRreal.swnode.h = 0; 474c4762a1bSJed Brown for (i = 0; i < 1 + dim; i++) uLreal.vals[i] = PetscRealPart(xL[i]); 475c4762a1bSJed Brown for (i = 0; i < 1 + dim; i++) uRreal.vals[i] = PetscRealPart(xR[i]); 476c4762a1bSJed Brown #endif 4779371c9d4SSatish Balay if (uL->h < 0 || uR->h < 0) { 4789371c9d4SSatish Balay for (i = 0; i < 1 + dim; i++) flux[i] = zero / zero; 4799371c9d4SSatish Balay return; 4809371c9d4SSatish Balay } /* reconstructed thickness is negative */ 481c4762a1bSJed Brown nn[0] = n[0]; 482c4762a1bSJed Brown nn[1] = n[1]; 483c4762a1bSJed Brown Normalize2Real(nn); 484c4762a1bSJed Brown SWFlux(phys, nn, uL, &(fL.swnode)); 485c4762a1bSJed Brown SWFlux(phys, nn, uR, &(fR.swnode)); 486c4762a1bSJed Brown cL = PetscSqrtReal(sw->gravity * uL->h); 487c4762a1bSJed Brown cR = PetscSqrtReal(sw->gravity * uR->h); /* gravity wave speed */ 488c4762a1bSJed Brown speed = PetscMax(PetscAbsReal(Dot2Real(uL->uh, nn) / uL->h) + cL, PetscAbsReal(Dot2Real(uR->uh, nn) / uR->h) + cR); 489c4762a1bSJed Brown for (i = 0; i < 1 + dim; i++) flux[i] = (0.5 * (fL.vals[i] + fR.vals[i]) + 0.5 * speed * (xL[i] - xR[i])) * Norm2Real(n); 490c4762a1bSJed Brown } 491c4762a1bSJed Brown 4929371c9d4SSatish Balay static PetscErrorCode PhysicsSolution_SW(Model mod, PetscReal time, const PetscReal *x, PetscScalar *u, void *ctx) { 493c4762a1bSJed Brown PetscReal dx[2], r, sigma; 494c4762a1bSJed Brown 495c4762a1bSJed Brown PetscFunctionBeginUser; 49654c59aa7SJacob Faibussowitsch PetscCheck(time == 0.0, mod->comm, PETSC_ERR_SUP, "No solution known for time %g", (double)time); 497c4762a1bSJed Brown dx[0] = x[0] - 1.5; 498c4762a1bSJed Brown dx[1] = x[1] - 1.0; 499c4762a1bSJed Brown r = Norm2Real(dx); 500c4762a1bSJed Brown sigma = 0.5; 501c4762a1bSJed Brown u[0] = 1 + 2 * PetscExpReal(-PetscSqr(r) / (2 * PetscSqr(sigma))); 502c4762a1bSJed Brown u[1] = 0.0; 503c4762a1bSJed Brown u[2] = 0.0; 504c4762a1bSJed Brown PetscFunctionReturn(0); 505c4762a1bSJed Brown } 506c4762a1bSJed Brown 5079371c9d4SSatish Balay static PetscErrorCode PhysicsFunctional_SW(Model mod, PetscReal time, const PetscReal *coord, const PetscScalar *xx, PetscReal *f, void *ctx) { 508c4762a1bSJed Brown Physics phys = (Physics)ctx; 509c4762a1bSJed Brown Physics_SW *sw = (Physics_SW *)phys->data; 510c4762a1bSJed Brown const SWNode *x = (const SWNode *)xx; 511c4762a1bSJed Brown PetscReal u[2]; 512c4762a1bSJed Brown PetscReal h; 513c4762a1bSJed Brown 514c4762a1bSJed Brown PetscFunctionBeginUser; 515c4762a1bSJed Brown h = x->h; 516c4762a1bSJed Brown Scale2Real(1. / x->h, x->uh, u); 517c4762a1bSJed Brown f[sw->functional.Height] = h; 518c4762a1bSJed Brown f[sw->functional.Speed] = Norm2Real(u) + PetscSqrtReal(sw->gravity * h); 519c4762a1bSJed Brown f[sw->functional.Energy] = 0.5 * (Dot2Real(x->uh, u) + sw->gravity * PetscSqr(h)); 520c4762a1bSJed Brown PetscFunctionReturn(0); 521c4762a1bSJed Brown } 522c4762a1bSJed Brown 5239371c9d4SSatish Balay static PetscErrorCode SetUpBC_SW(DM dm, PetscDS prob, Physics phys) { 524c4762a1bSJed Brown const PetscInt wallids[] = {100, 101, 200, 300}; 52545480ffeSMatthew G. Knepley DMLabel label; 52645480ffeSMatthew G. Knepley 527c4762a1bSJed Brown PetscFunctionBeginUser; 5289566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "Face Sets", &label)); 529dd39110bSPierre Jolivet PetscCall(PetscDSAddBoundary(prob, DM_BC_NATURAL_RIEMANN, "wall", label, PETSC_STATIC_ARRAY_LENGTH(wallids), wallids, 0, 0, NULL, (void (*)(void))PhysicsBoundary_SW_Wall, NULL, phys, NULL)); 530c4762a1bSJed Brown PetscFunctionReturn(0); 531c4762a1bSJed Brown } 532c4762a1bSJed Brown 5339371c9d4SSatish Balay static PetscErrorCode PhysicsCreate_SW(Model mod, Physics phys, PetscOptionItems *PetscOptionsObject) { 534c4762a1bSJed Brown Physics_SW *sw; 5358d2c18caSMukkund Sunjii char sw_riemann[64] = "rusanov"; 536c4762a1bSJed Brown 537c4762a1bSJed Brown PetscFunctionBeginUser; 538c4762a1bSJed Brown phys->field_desc = PhysicsFields_SW; 5399566063dSJacob Faibussowitsch PetscCall(PetscNew(&sw)); 540c4762a1bSJed Brown phys->data = sw; 541c4762a1bSJed Brown mod->setupbc = SetUpBC_SW; 542c4762a1bSJed Brown 5438d2c18caSMukkund Sunjii PetscFunctionListAdd(&PhysicsRiemannList_SW, "rusanov", PhysicsRiemann_SW_Rusanov); 5448d2c18caSMukkund Sunjii PetscFunctionListAdd(&PhysicsRiemannList_SW, "hll", PhysicsRiemann_SW_HLL); 5458d2c18caSMukkund Sunjii 546d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "SW options"); 547c4762a1bSJed Brown { 5488d2c18caSMukkund Sunjii void (*PhysicsRiemann_SW)(PetscInt, PetscInt, const PetscReal *, const PetscReal *, const PetscScalar *, const PetscScalar *, PetscInt, const PetscScalar, PetscScalar *, Physics); 549c4762a1bSJed Brown sw->gravity = 1.0; 5509566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-sw_gravity", "Gravitational constant", "", sw->gravity, &sw->gravity, NULL)); 5519566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-sw_riemann", "Riemann solver", "", PhysicsRiemannList_SW, sw_riemann, sw_riemann, sizeof sw_riemann, NULL)); 5529566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(PhysicsRiemannList_SW, sw_riemann, &PhysicsRiemann_SW)); 5538d2c18caSMukkund Sunjii phys->riemann = (PetscRiemannFunc)PhysicsRiemann_SW; 554c4762a1bSJed Brown } 555d0609cedSBarry Smith PetscOptionsHeadEnd(); 556c4762a1bSJed Brown phys->maxspeed = PetscSqrtReal(2.0 * sw->gravity); /* Mach 1 for depth of 2 */ 557c4762a1bSJed Brown 5589566063dSJacob Faibussowitsch PetscCall(ModelSolutionSetDefault(mod, PhysicsSolution_SW, phys)); 5599566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Height", &sw->functional.Height, PhysicsFunctional_SW, phys)); 5609566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Speed", &sw->functional.Speed, PhysicsFunctional_SW, phys)); 5619566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Energy", &sw->functional.Energy, PhysicsFunctional_SW, phys)); 562c4762a1bSJed Brown 563c4762a1bSJed Brown PetscFunctionReturn(0); 564c4762a1bSJed Brown } 565c4762a1bSJed Brown 566c4762a1bSJed Brown /******************* Euler Density Shock (EULER_IV_SHOCK,EULER_SS_SHOCK) ********************/ 567c4762a1bSJed Brown /* An initial-value and self-similar solutions of the compressible Euler equations */ 568c4762a1bSJed Brown /* Ravi Samtaney and D. I. Pullin */ 569c4762a1bSJed Brown /* Phys. Fluids 8, 2650 (1996); http://dx.doi.org/10.1063/1.869050 */ 5709371c9d4SSatish Balay typedef enum { 5719371c9d4SSatish Balay EULER_PAR_GAMMA, 5729371c9d4SSatish Balay EULER_PAR_RHOR, 5739371c9d4SSatish Balay EULER_PAR_AMACH, 5749371c9d4SSatish Balay EULER_PAR_ITANA, 5759371c9d4SSatish Balay EULER_PAR_SIZE 5769371c9d4SSatish Balay } EulerParamIdx; 5779371c9d4SSatish Balay typedef enum { 5789371c9d4SSatish Balay EULER_IV_SHOCK, 5799371c9d4SSatish Balay EULER_SS_SHOCK, 5809371c9d4SSatish Balay EULER_SHOCK_TUBE, 5819371c9d4SSatish Balay EULER_LINEAR_WAVE 5829371c9d4SSatish Balay } EulerType; 583c4762a1bSJed Brown typedef struct { 584c4762a1bSJed Brown PetscReal r; 585c4762a1bSJed Brown PetscReal ru[DIM]; 586c4762a1bSJed Brown PetscReal E; 587c4762a1bSJed Brown } EulerNode; 5889371c9d4SSatish Balay typedef union 5899371c9d4SSatish Balay { 590c4762a1bSJed Brown EulerNode eulernode; 591c4762a1bSJed Brown PetscReal vals[DIM + 2]; 592c4762a1bSJed Brown } EulerNodeUnion; 593c4762a1bSJed Brown typedef PetscErrorCode (*EquationOfState)(const PetscReal *, const EulerNode *, PetscReal *); 594c4762a1bSJed Brown typedef struct { 595c4762a1bSJed Brown EulerType type; 596c4762a1bSJed Brown PetscReal pars[EULER_PAR_SIZE]; 597c4762a1bSJed Brown EquationOfState sound; 598c4762a1bSJed Brown struct { 599c4762a1bSJed Brown PetscInt Density; 600c4762a1bSJed Brown PetscInt Momentum; 601c4762a1bSJed Brown PetscInt Energy; 602c4762a1bSJed Brown PetscInt Pressure; 603c4762a1bSJed Brown PetscInt Speed; 604c4762a1bSJed Brown } monitor; 605c4762a1bSJed Brown } Physics_Euler; 606c4762a1bSJed Brown 6079371c9d4SSatish Balay static const struct FieldDescription PhysicsFields_Euler[] = { 6089371c9d4SSatish Balay {"Density", 1 }, 6099371c9d4SSatish Balay {"Momentum", DIM}, 6109371c9d4SSatish Balay {"Energy", 1 }, 6119371c9d4SSatish Balay {NULL, 0 } 6129371c9d4SSatish Balay }; 613c4762a1bSJed Brown 614c4762a1bSJed Brown /* initial condition */ 615c4762a1bSJed Brown int initLinearWave(EulerNode *ux, const PetscReal gamma, const PetscReal coord[], const PetscReal Lx); 6169371c9d4SSatish Balay static PetscErrorCode PhysicsSolution_Euler(Model mod, PetscReal time, const PetscReal *x, PetscScalar *u, void *ctx) { 617c4762a1bSJed Brown PetscInt i; 618c4762a1bSJed Brown Physics phys = (Physics)ctx; 619c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 620c4762a1bSJed Brown EulerNode *uu = (EulerNode *)u; 621c4762a1bSJed Brown PetscReal p0, gamma, c; 622c4762a1bSJed Brown PetscFunctionBeginUser; 62354c59aa7SJacob Faibussowitsch PetscCheck(time == 0.0, mod->comm, PETSC_ERR_SUP, "No solution known for time %g", (double)time); 624c4762a1bSJed Brown 625c4762a1bSJed Brown for (i = 0; i < DIM; i++) uu->ru[i] = 0.0; /* zero out initial velocity */ 626c4762a1bSJed Brown /* set E and rho */ 627c4762a1bSJed Brown gamma = eu->pars[EULER_PAR_GAMMA]; 628c4762a1bSJed Brown 629c4762a1bSJed Brown if (eu->type == EULER_IV_SHOCK || eu->type == EULER_SS_SHOCK) { 630c4762a1bSJed Brown /******************* Euler Density Shock ********************/ 631c4762a1bSJed Brown /* On initial-value and self-similar solutions of the compressible Euler equations */ 632c4762a1bSJed Brown /* Ravi Samtaney and D. I. Pullin */ 633c4762a1bSJed Brown /* Phys. Fluids 8, 2650 (1996); http://dx.doi.org/10.1063/1.869050 */ 634c4762a1bSJed Brown /* initial conditions 1: left of shock, 0: left of discontinuity 2: right of discontinuity, */ 635c4762a1bSJed Brown p0 = 1.; 636c4762a1bSJed Brown if (x[0] < 0.0 + x[1] * eu->pars[EULER_PAR_ITANA]) { 637c4762a1bSJed Brown if (x[0] < mod->bounds[0] * 0.5) { /* left of shock (1) */ 638c4762a1bSJed Brown PetscReal amach, rho, press, gas1, p1; 639c4762a1bSJed Brown amach = eu->pars[EULER_PAR_AMACH]; 640c4762a1bSJed Brown rho = 1.; 641c4762a1bSJed Brown press = p0; 642c4762a1bSJed Brown p1 = press * (1.0 + 2.0 * gamma / (gamma + 1.0) * (amach * amach - 1.0)); 643c4762a1bSJed Brown gas1 = (gamma - 1.0) / (gamma + 1.0); 644c4762a1bSJed Brown uu->r = rho * (p1 / press + gas1) / (gas1 * p1 / press + 1.0); 645c4762a1bSJed Brown uu->ru[0] = ((uu->r - rho) * PetscSqrtReal(gamma * press / rho) * amach); 646c4762a1bSJed Brown uu->E = p1 / (gamma - 1.0) + .5 / uu->r * uu->ru[0] * uu->ru[0]; 6479371c9d4SSatish Balay } else { /* left of discontinuity (0) */ 648c4762a1bSJed Brown uu->r = 1.; /* rho = 1 */ 649c4762a1bSJed Brown uu->E = p0 / (gamma - 1.0); 650c4762a1bSJed Brown } 6519371c9d4SSatish Balay } else { /* right of discontinuity (2) */ 652c4762a1bSJed Brown uu->r = eu->pars[EULER_PAR_RHOR]; 653c4762a1bSJed Brown uu->E = p0 / (gamma - 1.0); 654c4762a1bSJed Brown } 6559371c9d4SSatish Balay } else if (eu->type == EULER_SHOCK_TUBE) { 656c4762a1bSJed Brown /* For (x<x0) set (rho,u,p)=(8,0,10) and for (x>x0) set (rho,u,p)=(1,0,1). Choose x0 to the midpoint of the domain in the x-direction. */ 657c4762a1bSJed Brown if (x[0] < 0.0) { 658c4762a1bSJed Brown uu->r = 8.; 659c4762a1bSJed Brown uu->E = 10. / (gamma - 1.); 6609371c9d4SSatish Balay } else { 661c4762a1bSJed Brown uu->r = 1.; 662c4762a1bSJed Brown uu->E = 1. / (gamma - 1.); 663c4762a1bSJed Brown } 6649371c9d4SSatish Balay } else if (eu->type == EULER_LINEAR_WAVE) { 665c4762a1bSJed Brown initLinearWave(uu, gamma, x, mod->bounds[1] - mod->bounds[0]); 6669371c9d4SSatish Balay } else SETERRQ(mod->comm, PETSC_ERR_SUP, "Unknown type %d", eu->type); 667c4762a1bSJed Brown 668c4762a1bSJed Brown /* set phys->maxspeed: (mod->maxspeed = phys->maxspeed) in main; */ 669c4762a1bSJed Brown eu->sound(&gamma, uu, &c); 670c4762a1bSJed Brown c = (uu->ru[0] / uu->r) + c; 671c4762a1bSJed Brown if (c > phys->maxspeed) phys->maxspeed = c; 672c4762a1bSJed Brown 673c4762a1bSJed Brown PetscFunctionReturn(0); 674c4762a1bSJed Brown } 675c4762a1bSJed Brown 6769371c9d4SSatish Balay static PetscErrorCode Pressure_PG(const PetscReal gamma, const EulerNode *x, PetscReal *p) { 677c4762a1bSJed Brown PetscReal ru2; 678c4762a1bSJed Brown 679c4762a1bSJed Brown PetscFunctionBeginUser; 680c4762a1bSJed Brown ru2 = DotDIMReal(x->ru, x->ru); 681c4762a1bSJed Brown (*p) = (x->E - 0.5 * ru2 / x->r) * (gamma - 1.0); /* (E - rho V^2/2)(gamma-1) = e rho (gamma-1) */ 682c4762a1bSJed Brown PetscFunctionReturn(0); 683c4762a1bSJed Brown } 684c4762a1bSJed Brown 6859371c9d4SSatish Balay static PetscErrorCode SpeedOfSound_PG(const PetscReal *gamma, const EulerNode *x, PetscReal *c) { 686c4762a1bSJed Brown PetscReal p; 687c4762a1bSJed Brown 688c4762a1bSJed Brown PetscFunctionBeginUser; 689c4762a1bSJed Brown Pressure_PG(*gamma, x, &p); 69054c59aa7SJacob Faibussowitsch PetscCheck(p >= 0., PETSC_COMM_WORLD, PETSC_ERR_SUP, "negative pressure time %g -- NEED TO FIX!!!!!!", (double)p); 691c4762a1bSJed Brown /* pars[EULER_PAR_GAMMA] = heat capacity ratio */ 692c4762a1bSJed Brown (*c) = PetscSqrtReal(*gamma * p / x->r); 693c4762a1bSJed Brown PetscFunctionReturn(0); 694c4762a1bSJed Brown } 695c4762a1bSJed Brown 696c4762a1bSJed Brown /* 697c4762a1bSJed Brown * x = (rho,rho*(u_1),...,rho*e)^T 698c4762a1bSJed Brown * x_t+div(f_1(x))+...+div(f_DIM(x)) = 0 699c4762a1bSJed Brown * 700c4762a1bSJed Brown * f_i(x) = u_i*x+(0,0,...,p,...,p*u_i)^T 701c4762a1bSJed Brown * 702c4762a1bSJed Brown */ 7039371c9d4SSatish Balay static PetscErrorCode EulerFlux(Physics phys, const PetscReal *n, const EulerNode *x, EulerNode *f) { 704c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 705c4762a1bSJed Brown PetscReal nu, p; 706c4762a1bSJed Brown PetscInt i; 707c4762a1bSJed Brown 708c4762a1bSJed Brown PetscFunctionBeginUser; 709c4762a1bSJed Brown Pressure_PG(eu->pars[EULER_PAR_GAMMA], x, &p); 710c4762a1bSJed Brown nu = DotDIMReal(x->ru, n); 711c4762a1bSJed Brown f->r = nu; /* A rho u */ 712c4762a1bSJed Brown nu /= x->r; /* A u */ 713c4762a1bSJed Brown for (i = 0; i < DIM; i++) f->ru[i] = nu * x->ru[i] + n[i] * p; /* r u^2 + p */ 714c4762a1bSJed Brown f->E = nu * (x->E + p); /* u(e+p) */ 715c4762a1bSJed Brown PetscFunctionReturn(0); 716c4762a1bSJed Brown } 717c4762a1bSJed Brown 718c4762a1bSJed Brown /* PetscReal* => EulerNode* conversion */ 7199371c9d4SSatish Balay static PetscErrorCode PhysicsBoundary_Euler_Wall(PetscReal time, const PetscReal *c, const PetscReal *n, const PetscScalar *a_xI, PetscScalar *a_xG, void *ctx) { 720c4762a1bSJed Brown PetscInt i; 721c4762a1bSJed Brown const EulerNode *xI = (const EulerNode *)a_xI; 722c4762a1bSJed Brown EulerNode *xG = (EulerNode *)a_xG; 723c4762a1bSJed Brown Physics phys = (Physics)ctx; 724c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 725c4762a1bSJed Brown PetscFunctionBeginUser; 726c4762a1bSJed Brown xG->r = xI->r; /* ghost cell density - same */ 727c4762a1bSJed Brown xG->E = xI->E; /* ghost cell energy - same */ 728c4762a1bSJed Brown if (n[1] != 0.) { /* top and bottom */ 729c4762a1bSJed Brown xG->ru[0] = xI->ru[0]; /* copy tang to wall */ 730c4762a1bSJed Brown xG->ru[1] = -xI->ru[1]; /* reflect perp to t/b wall */ 7319371c9d4SSatish Balay } else { /* sides */ 732c4762a1bSJed Brown for (i = 0; i < DIM; i++) xG->ru[i] = xI->ru[i]; /* copy */ 733c4762a1bSJed Brown } 734c4762a1bSJed Brown if (eu->type == EULER_LINEAR_WAVE) { /* debug */ 735c4762a1bSJed Brown #if 0 73663a3b9bcSJacob Faibussowitsch PetscPrintf(PETSC_COMM_WORLD,"%s coord=%g,%g\n",PETSC_FUNCTION_NAME,(double)c[0],(double)c[1]); 737c4762a1bSJed Brown #endif 738c4762a1bSJed Brown } 739c4762a1bSJed Brown PetscFunctionReturn(0); 740c4762a1bSJed Brown } 741c4762a1bSJed Brown int godunovflux(const PetscScalar *ul, const PetscScalar *ur, PetscScalar *flux, const PetscReal *nn, const int *ndim, const PetscReal *gamma); 742c4762a1bSJed Brown /* PetscReal* => EulerNode* conversion */ 7439371c9d4SSatish Balay static void PhysicsRiemann_Euler_Godunov(PetscInt dim, PetscInt Nf, const PetscReal *qp, const PetscReal *n, const PetscScalar *xL, const PetscScalar *xR, PetscInt numConstants, const PetscScalar constants[], PetscScalar *flux, Physics phys) { 744c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 745c4762a1bSJed Brown PetscReal cL, cR, speed, velL, velR, nn[DIM], s2; 746c4762a1bSJed Brown PetscInt i; 747c4762a1bSJed Brown PetscErrorCode ierr; 748c4762a1bSJed Brown 749d0609cedSBarry Smith PetscFunctionBeginUser; 750c4762a1bSJed Brown for (i = 0, s2 = 0.; i < DIM; i++) { 751c4762a1bSJed Brown nn[i] = n[i]; 752c4762a1bSJed Brown s2 += nn[i] * nn[i]; 753c4762a1bSJed Brown } 754c4762a1bSJed Brown s2 = PetscSqrtReal(s2); /* |n|_2 = sum(n^2)^1/2 */ 755c4762a1bSJed Brown for (i = 0.; i < DIM; i++) nn[i] /= s2; 756c4762a1bSJed Brown if (0) { /* Rusanov */ 757c4762a1bSJed Brown const EulerNode *uL = (const EulerNode *)xL, *uR = (const EulerNode *)xR; 758c4762a1bSJed Brown EulerNodeUnion fL, fR; 759c4762a1bSJed Brown EulerFlux(phys, nn, uL, &(fL.eulernode)); 760c4762a1bSJed Brown EulerFlux(phys, nn, uR, &(fR.eulernode)); 7619371c9d4SSatish Balay ierr = eu->sound(&eu->pars[EULER_PAR_GAMMA], uL, &cL); 7629371c9d4SSatish Balay if (ierr) exit(13); 7639371c9d4SSatish Balay ierr = eu->sound(&eu->pars[EULER_PAR_GAMMA], uR, &cR); 7649371c9d4SSatish Balay if (ierr) exit(14); 765c4762a1bSJed Brown velL = DotDIMReal(uL->ru, nn) / uL->r; 766c4762a1bSJed Brown velR = DotDIMReal(uR->ru, nn) / uR->r; 767c4762a1bSJed Brown speed = PetscMax(velR + cR, velL + cL); 768c4762a1bSJed Brown for (i = 0; i < 2 + dim; i++) flux[i] = 0.5 * ((fL.vals[i] + fR.vals[i]) + speed * (xL[i] - xR[i])) * s2; 7699371c9d4SSatish Balay } else { 770c4762a1bSJed Brown int dim = DIM; 771c4762a1bSJed Brown /* int iwave = */ 772c4762a1bSJed Brown godunovflux(xL, xR, flux, nn, &dim, &eu->pars[EULER_PAR_GAMMA]); 773c4762a1bSJed Brown for (i = 0; i < 2 + dim; i++) flux[i] *= s2; 774c4762a1bSJed Brown } 775c4762a1bSJed Brown PetscFunctionReturnVoid(); 776c4762a1bSJed Brown } 777c4762a1bSJed Brown 7789371c9d4SSatish Balay static PetscErrorCode PhysicsFunctional_Euler(Model mod, PetscReal time, const PetscReal *coord, const PetscScalar *xx, PetscReal *f, void *ctx) { 779c4762a1bSJed Brown Physics phys = (Physics)ctx; 780c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 781c4762a1bSJed Brown const EulerNode *x = (const EulerNode *)xx; 782c4762a1bSJed Brown PetscReal p; 783c4762a1bSJed Brown 784c4762a1bSJed Brown PetscFunctionBeginUser; 785c4762a1bSJed Brown f[eu->monitor.Density] = x->r; 786c4762a1bSJed Brown f[eu->monitor.Momentum] = NormDIM(x->ru); 787c4762a1bSJed Brown f[eu->monitor.Energy] = x->E; 788c4762a1bSJed Brown f[eu->monitor.Speed] = NormDIM(x->ru) / x->r; 789c4762a1bSJed Brown Pressure_PG(eu->pars[EULER_PAR_GAMMA], x, &p); 790c4762a1bSJed Brown f[eu->monitor.Pressure] = p; 791c4762a1bSJed Brown PetscFunctionReturn(0); 792c4762a1bSJed Brown } 793c4762a1bSJed Brown 7949371c9d4SSatish Balay static PetscErrorCode SetUpBC_Euler(DM dm, PetscDS prob, Physics phys) { 795c4762a1bSJed Brown Physics_Euler *eu = (Physics_Euler *)phys->data; 79645480ffeSMatthew G. Knepley DMLabel label; 79745480ffeSMatthew G. Knepley 798362febeeSStefano Zampini PetscFunctionBeginUser; 7999566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "Face Sets", &label)); 800c4762a1bSJed Brown if (eu->type == EULER_LINEAR_WAVE) { 801c4762a1bSJed Brown const PetscInt wallids[] = {100, 101}; 802dd39110bSPierre Jolivet PetscCall(PetscDSAddBoundary(prob, DM_BC_NATURAL_RIEMANN, "wall", label, PETSC_STATIC_ARRAY_LENGTH(wallids), wallids, 0, 0, NULL, (void (*)(void))PhysicsBoundary_Euler_Wall, NULL, phys, NULL)); 8039371c9d4SSatish Balay } else { 804c4762a1bSJed Brown const PetscInt wallids[] = {100, 101, 200, 300}; 805dd39110bSPierre Jolivet PetscCall(PetscDSAddBoundary(prob, DM_BC_NATURAL_RIEMANN, "wall", label, PETSC_STATIC_ARRAY_LENGTH(wallids), wallids, 0, 0, NULL, (void (*)(void))PhysicsBoundary_Euler_Wall, NULL, phys, NULL)); 806c4762a1bSJed Brown } 807c4762a1bSJed Brown PetscFunctionReturn(0); 808c4762a1bSJed Brown } 809c4762a1bSJed Brown 8109371c9d4SSatish Balay static PetscErrorCode PhysicsCreate_Euler(Model mod, Physics phys, PetscOptionItems *PetscOptionsObject) { 811c4762a1bSJed Brown Physics_Euler *eu; 812c4762a1bSJed Brown 813c4762a1bSJed Brown PetscFunctionBeginUser; 814c4762a1bSJed Brown phys->field_desc = PhysicsFields_Euler; 815c4762a1bSJed Brown phys->riemann = (PetscRiemannFunc)PhysicsRiemann_Euler_Godunov; 8169566063dSJacob Faibussowitsch PetscCall(PetscNew(&eu)); 817c4762a1bSJed Brown phys->data = eu; 818c4762a1bSJed Brown mod->setupbc = SetUpBC_Euler; 819d0609cedSBarry Smith PetscOptionsHeadBegin(PetscOptionsObject, "Euler options"); 820c4762a1bSJed Brown { 821c4762a1bSJed Brown PetscReal alpha; 822c4762a1bSJed Brown char type[64] = "linear_wave"; 823c4762a1bSJed Brown PetscBool is; 824c4762a1bSJed Brown eu->pars[EULER_PAR_GAMMA] = 1.4; 825c4762a1bSJed Brown eu->pars[EULER_PAR_AMACH] = 2.02; 826c4762a1bSJed Brown eu->pars[EULER_PAR_RHOR] = 3.0; 827c4762a1bSJed Brown eu->pars[EULER_PAR_ITANA] = 0.57735026918963; /* angle of Euler self similar (SS) shock */ 8289566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-eu_gamma", "Heat capacity ratio", "", eu->pars[EULER_PAR_GAMMA], &eu->pars[EULER_PAR_GAMMA], NULL)); 8299566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-eu_amach", "Shock speed (Mach)", "", eu->pars[EULER_PAR_AMACH], &eu->pars[EULER_PAR_AMACH], NULL)); 8309566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-eu_rho2", "Density right of discontinuity", "", eu->pars[EULER_PAR_RHOR], &eu->pars[EULER_PAR_RHOR], NULL)); 831c4762a1bSJed Brown alpha = 60.; 8329566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-eu_alpha", "Angle of discontinuity", "", alpha, &alpha, NULL)); 83363a3b9bcSJacob Faibussowitsch PetscCheck(alpha > 0. && alpha <= 90., PETSC_COMM_WORLD, PETSC_ERR_SUP, "Alpha bust be > 0 and <= 90 (%g)", (double)alpha); 834c4762a1bSJed Brown eu->pars[EULER_PAR_ITANA] = 1. / PetscTanReal(alpha * PETSC_PI / 180.0); 8359566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-eu_type", "Type of Euler test", "", type, type, sizeof(type), NULL)); 8369566063dSJacob Faibussowitsch PetscCall(PetscStrcmp(type, "linear_wave", &is)); 837c4762a1bSJed Brown if (is) { 83830602db0SMatthew G. Knepley /* Remember this should be periodic */ 839c4762a1bSJed Brown eu->type = EULER_LINEAR_WAVE; 8409566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%s set Euler type: %s\n", PETSC_FUNCTION_NAME, "linear_wave")); 8419371c9d4SSatish Balay } else { 84254c59aa7SJacob Faibussowitsch PetscCheck(DIM == 2, PETSC_COMM_WORLD, PETSC_ERR_SUP, "DIM must be 2 unless linear wave test %s", type); 8439566063dSJacob Faibussowitsch PetscCall(PetscStrcmp(type, "iv_shock", &is)); 844c4762a1bSJed Brown if (is) { 845c4762a1bSJed Brown eu->type = EULER_IV_SHOCK; 8469566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%s set Euler type: %s\n", PETSC_FUNCTION_NAME, "iv_shock")); 8479371c9d4SSatish Balay } else { 8489566063dSJacob Faibussowitsch PetscCall(PetscStrcmp(type, "ss_shock", &is)); 849c4762a1bSJed Brown if (is) { 850c4762a1bSJed Brown eu->type = EULER_SS_SHOCK; 8519566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%s set Euler type: %s\n", PETSC_FUNCTION_NAME, "ss_shock")); 8529371c9d4SSatish Balay } else { 8539566063dSJacob Faibussowitsch PetscCall(PetscStrcmp(type, "shock_tube", &is)); 854c4762a1bSJed Brown if (is) eu->type = EULER_SHOCK_TUBE; 85598921bdaSJacob Faibussowitsch else SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_SUP, "Unknown Euler type %s", type); 8569566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%s set Euler type: %s\n", PETSC_FUNCTION_NAME, "shock_tube")); 857c4762a1bSJed Brown } 858c4762a1bSJed Brown } 859c4762a1bSJed Brown } 860c4762a1bSJed Brown } 861d0609cedSBarry Smith PetscOptionsHeadEnd(); 862c4762a1bSJed Brown eu->sound = SpeedOfSound_PG; 863c4762a1bSJed Brown phys->maxspeed = 0.; /* will get set in solution */ 8649566063dSJacob Faibussowitsch PetscCall(ModelSolutionSetDefault(mod, PhysicsSolution_Euler, phys)); 8659566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Speed", &eu->monitor.Speed, PhysicsFunctional_Euler, phys)); 8669566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Energy", &eu->monitor.Energy, PhysicsFunctional_Euler, phys)); 8679566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Density", &eu->monitor.Density, PhysicsFunctional_Euler, phys)); 8689566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Momentum", &eu->monitor.Momentum, PhysicsFunctional_Euler, phys)); 8699566063dSJacob Faibussowitsch PetscCall(ModelFunctionalRegister(mod, "Pressure", &eu->monitor.Pressure, PhysicsFunctional_Euler, phys)); 870c4762a1bSJed Brown 871c4762a1bSJed Brown PetscFunctionReturn(0); 872c4762a1bSJed Brown } 873c4762a1bSJed Brown 8749371c9d4SSatish Balay static PetscErrorCode ErrorIndicator_Simple(PetscInt dim, PetscReal volume, PetscInt numComps, const PetscScalar u[], const PetscScalar grad[], PetscReal *error, void *ctx) { 875c4762a1bSJed Brown PetscReal err = 0.; 876c4762a1bSJed Brown PetscInt i, j; 877c4762a1bSJed Brown 878c4762a1bSJed Brown PetscFunctionBeginUser; 879c4762a1bSJed Brown for (i = 0; i < numComps; i++) { 8809371c9d4SSatish Balay for (j = 0; j < dim; j++) { err += PetscSqr(PetscRealPart(grad[i * dim + j])); } 881c4762a1bSJed Brown } 882c4762a1bSJed Brown *error = volume * err; 883c4762a1bSJed Brown PetscFunctionReturn(0); 884c4762a1bSJed Brown } 885c4762a1bSJed Brown 8869371c9d4SSatish Balay PetscErrorCode CreatePartitionVec(DM dm, DM *dmCell, Vec *partition) { 887c4762a1bSJed Brown PetscSF sfPoint; 888c4762a1bSJed Brown PetscSection coordSection; 889c4762a1bSJed Brown Vec coordinates; 890c4762a1bSJed Brown PetscSection sectionCell; 891c4762a1bSJed Brown PetscScalar *part; 892c4762a1bSJed Brown PetscInt cStart, cEnd, c; 893c4762a1bSJed Brown PetscMPIInt rank; 894c4762a1bSJed Brown 895c4762a1bSJed Brown PetscFunctionBeginUser; 8969566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 8979566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &coordinates)); 8989566063dSJacob Faibussowitsch PetscCall(DMClone(dm, dmCell)); 8999566063dSJacob Faibussowitsch PetscCall(DMGetPointSF(dm, &sfPoint)); 9009566063dSJacob Faibussowitsch PetscCall(DMSetPointSF(*dmCell, sfPoint)); 9019566063dSJacob Faibussowitsch PetscCall(DMSetCoordinateSection(*dmCell, PETSC_DETERMINE, coordSection)); 9029566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(*dmCell, coordinates)); 9039566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank)); 9049566063dSJacob Faibussowitsch PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), §ionCell)); 9059566063dSJacob Faibussowitsch PetscCall(DMPlexGetHeightStratum(*dmCell, 0, &cStart, &cEnd)); 9069566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(sectionCell, cStart, cEnd)); 907*48a46eb9SPierre Jolivet for (c = cStart; c < cEnd; ++c) PetscCall(PetscSectionSetDof(sectionCell, c, 1)); 9089566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(sectionCell)); 9099566063dSJacob Faibussowitsch PetscCall(DMSetLocalSection(*dmCell, sectionCell)); 9109566063dSJacob Faibussowitsch PetscCall(PetscSectionDestroy(§ionCell)); 9119566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(*dmCell, partition)); 9129566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)*partition, "partition")); 9139566063dSJacob Faibussowitsch PetscCall(VecGetArray(*partition, &part)); 914c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 915c4762a1bSJed Brown PetscScalar *p; 916c4762a1bSJed Brown 9179566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRef(*dmCell, c, part, &p)); 918c4762a1bSJed Brown p[0] = rank; 919c4762a1bSJed Brown } 9209566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(*partition, &part)); 921c4762a1bSJed Brown PetscFunctionReturn(0); 922c4762a1bSJed Brown } 923c4762a1bSJed Brown 9249371c9d4SSatish Balay PetscErrorCode CreateMassMatrix(DM dm, Vec *massMatrix, User user) { 9253e9753d6SMatthew G. Knepley DM plex, dmMass, dmFace, dmCell, dmCoord; 926c4762a1bSJed Brown PetscSection coordSection; 927c4762a1bSJed Brown Vec coordinates, facegeom, cellgeom; 928c4762a1bSJed Brown PetscSection sectionMass; 929c4762a1bSJed Brown PetscScalar *m; 930c4762a1bSJed Brown const PetscScalar *fgeom, *cgeom, *coords; 931c4762a1bSJed Brown PetscInt vStart, vEnd, v; 932c4762a1bSJed Brown 933c4762a1bSJed Brown PetscFunctionBeginUser; 9349566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 9359566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateSection(dm, &coordSection)); 9369566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &coordinates)); 9379566063dSJacob Faibussowitsch PetscCall(DMClone(dm, &dmMass)); 9389566063dSJacob Faibussowitsch PetscCall(DMSetCoordinateSection(dmMass, PETSC_DETERMINE, coordSection)); 9399566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(dmMass, coordinates)); 9409566063dSJacob Faibussowitsch PetscCall(PetscSectionCreate(PetscObjectComm((PetscObject)dm), §ionMass)); 9419566063dSJacob Faibussowitsch PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd)); 9429566063dSJacob Faibussowitsch PetscCall(PetscSectionSetChart(sectionMass, vStart, vEnd)); 943c4762a1bSJed Brown for (v = vStart; v < vEnd; ++v) { 944c4762a1bSJed Brown PetscInt numFaces; 945c4762a1bSJed Brown 9469566063dSJacob Faibussowitsch PetscCall(DMPlexGetSupportSize(dmMass, v, &numFaces)); 9479566063dSJacob Faibussowitsch PetscCall(PetscSectionSetDof(sectionMass, v, numFaces * numFaces)); 948c4762a1bSJed Brown } 9499566063dSJacob Faibussowitsch PetscCall(PetscSectionSetUp(sectionMass)); 9509566063dSJacob Faibussowitsch PetscCall(DMSetLocalSection(dmMass, sectionMass)); 9519566063dSJacob Faibussowitsch PetscCall(PetscSectionDestroy(§ionMass)); 9529566063dSJacob Faibussowitsch PetscCall(DMGetLocalVector(dmMass, massMatrix)); 9539566063dSJacob Faibussowitsch PetscCall(VecGetArray(*massMatrix, &m)); 9549566063dSJacob Faibussowitsch PetscCall(DMPlexGetGeometryFVM(plex, &facegeom, &cellgeom, NULL)); 9559566063dSJacob Faibussowitsch PetscCall(VecGetDM(facegeom, &dmFace)); 9569566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(facegeom, &fgeom)); 9579566063dSJacob Faibussowitsch PetscCall(VecGetDM(cellgeom, &dmCell)); 9589566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(cellgeom, &cgeom)); 9599566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDM(dm, &dmCoord)); 9609566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(coordinates, &coords)); 961c4762a1bSJed Brown for (v = vStart; v < vEnd; ++v) { 962c4762a1bSJed Brown const PetscInt *faces; 963c4762a1bSJed Brown PetscFVFaceGeom *fgA, *fgB, *cg; 964c4762a1bSJed Brown PetscScalar *vertex; 965c4762a1bSJed Brown PetscInt numFaces, sides[2], f, g; 966c4762a1bSJed Brown 9679566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(dmCoord, v, coords, &vertex)); 9689566063dSJacob Faibussowitsch PetscCall(DMPlexGetSupportSize(dmMass, v, &numFaces)); 9699566063dSJacob Faibussowitsch PetscCall(DMPlexGetSupport(dmMass, v, &faces)); 970c4762a1bSJed Brown for (f = 0; f < numFaces; ++f) { 971c4762a1bSJed Brown sides[0] = faces[f]; 9729566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(dmFace, faces[f], fgeom, &fgA)); 973c4762a1bSJed Brown for (g = 0; g < numFaces; ++g) { 974c4762a1bSJed Brown const PetscInt *cells = NULL; 975c4762a1bSJed Brown PetscReal area = 0.0; 976c4762a1bSJed Brown PetscInt numCells; 977c4762a1bSJed Brown 978c4762a1bSJed Brown sides[1] = faces[g]; 9799566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(dmFace, faces[g], fgeom, &fgB)); 9809566063dSJacob Faibussowitsch PetscCall(DMPlexGetJoin(dmMass, 2, sides, &numCells, &cells)); 98154c59aa7SJacob Faibussowitsch PetscCheck(numCells == 1, PETSC_COMM_SELF, PETSC_ERR_LIB, "Invalid join for faces"); 9829566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(dmCell, cells[0], cgeom, &cg)); 983c4762a1bSJed Brown area += PetscAbsScalar((vertex[0] - cg->centroid[0]) * (fgA->centroid[1] - cg->centroid[1]) - (vertex[1] - cg->centroid[1]) * (fgA->centroid[0] - cg->centroid[0])); 984c4762a1bSJed Brown area += PetscAbsScalar((vertex[0] - cg->centroid[0]) * (fgB->centroid[1] - cg->centroid[1]) - (vertex[1] - cg->centroid[1]) * (fgB->centroid[0] - cg->centroid[0])); 985c4762a1bSJed Brown m[f * numFaces + g] = Dot2Real(fgA->normal, fgB->normal) * area * 0.5; 9869566063dSJacob Faibussowitsch PetscCall(DMPlexRestoreJoin(dmMass, 2, sides, &numCells, &cells)); 987c4762a1bSJed Brown } 988c4762a1bSJed Brown } 989c4762a1bSJed Brown } 9909566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(facegeom, &fgeom)); 9919566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(cellgeom, &cgeom)); 9929566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(coordinates, &coords)); 9939566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(*massMatrix, &m)); 9949566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dmMass)); 9959566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 996c4762a1bSJed Brown PetscFunctionReturn(0); 997c4762a1bSJed Brown } 998c4762a1bSJed Brown 999c4762a1bSJed Brown /* Behavior will be different for multi-physics or when using non-default boundary conditions */ 10009371c9d4SSatish Balay static PetscErrorCode ModelSolutionSetDefault(Model mod, SolutionFunction func, void *ctx) { 1001c4762a1bSJed Brown PetscFunctionBeginUser; 1002c4762a1bSJed Brown mod->solution = func; 1003c4762a1bSJed Brown mod->solutionctx = ctx; 1004c4762a1bSJed Brown PetscFunctionReturn(0); 1005c4762a1bSJed Brown } 1006c4762a1bSJed Brown 10079371c9d4SSatish Balay static PetscErrorCode ModelFunctionalRegister(Model mod, const char *name, PetscInt *offset, FunctionalFunction func, void *ctx) { 1008c4762a1bSJed Brown FunctionalLink link, *ptr; 1009c4762a1bSJed Brown PetscInt lastoffset = -1; 1010c4762a1bSJed Brown 1011c4762a1bSJed Brown PetscFunctionBeginUser; 1012c4762a1bSJed Brown for (ptr = &mod->functionalRegistry; *ptr; ptr = &(*ptr)->next) lastoffset = (*ptr)->offset; 10139566063dSJacob Faibussowitsch PetscCall(PetscNew(&link)); 10149566063dSJacob Faibussowitsch PetscCall(PetscStrallocpy(name, &link->name)); 1015c4762a1bSJed Brown link->offset = lastoffset + 1; 1016c4762a1bSJed Brown link->func = func; 1017c4762a1bSJed Brown link->ctx = ctx; 1018c4762a1bSJed Brown link->next = NULL; 1019c4762a1bSJed Brown *ptr = link; 1020c4762a1bSJed Brown *offset = link->offset; 1021c4762a1bSJed Brown PetscFunctionReturn(0); 1022c4762a1bSJed Brown } 1023c4762a1bSJed Brown 10249371c9d4SSatish Balay static PetscErrorCode ModelFunctionalSetFromOptions(Model mod, PetscOptionItems *PetscOptionsObject) { 1025c4762a1bSJed Brown PetscInt i, j; 1026c4762a1bSJed Brown FunctionalLink link; 1027c4762a1bSJed Brown char *names[256]; 1028c4762a1bSJed Brown 1029c4762a1bSJed Brown PetscFunctionBeginUser; 1030dd39110bSPierre Jolivet mod->numMonitored = PETSC_STATIC_ARRAY_LENGTH(names); 10319566063dSJacob Faibussowitsch PetscCall(PetscOptionsStringArray("-monitor", "list of functionals to monitor", "", names, &mod->numMonitored, NULL)); 1032c4762a1bSJed Brown /* Create list of functionals that will be computed somehow */ 10339566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mod->numMonitored, &mod->functionalMonitored)); 1034c4762a1bSJed Brown /* Create index of calls that we will have to make to compute these functionals (over-allocation in general). */ 10359566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(mod->numMonitored, &mod->functionalCall)); 1036c4762a1bSJed Brown mod->numCall = 0; 1037c4762a1bSJed Brown for (i = 0; i < mod->numMonitored; i++) { 1038c4762a1bSJed Brown for (link = mod->functionalRegistry; link; link = link->next) { 1039c4762a1bSJed Brown PetscBool match; 10409566063dSJacob Faibussowitsch PetscCall(PetscStrcasecmp(names[i], link->name, &match)); 1041c4762a1bSJed Brown if (match) break; 1042c4762a1bSJed Brown } 104354c59aa7SJacob Faibussowitsch PetscCheck(link, mod->comm, PETSC_ERR_USER, "No known functional '%s'", names[i]); 1044c4762a1bSJed Brown mod->functionalMonitored[i] = link; 1045c4762a1bSJed Brown for (j = 0; j < i; j++) { 1046c4762a1bSJed Brown if (mod->functionalCall[j]->func == link->func && mod->functionalCall[j]->ctx == link->ctx) goto next_name; 1047c4762a1bSJed Brown } 1048c4762a1bSJed Brown mod->functionalCall[mod->numCall++] = link; /* Just points to the first link using the result. There may be more results. */ 1049c4762a1bSJed Brown next_name: 10509566063dSJacob Faibussowitsch PetscCall(PetscFree(names[i])); 1051c4762a1bSJed Brown } 1052c4762a1bSJed Brown 1053c4762a1bSJed Brown /* Find out the maximum index of any functional computed by a function we will be calling (even if we are not using it) */ 1054c4762a1bSJed Brown mod->maxComputed = -1; 1055c4762a1bSJed Brown for (link = mod->functionalRegistry; link; link = link->next) { 1056c4762a1bSJed Brown for (i = 0; i < mod->numCall; i++) { 1057c4762a1bSJed Brown FunctionalLink call = mod->functionalCall[i]; 10589371c9d4SSatish Balay if (link->func == call->func && link->ctx == call->ctx) { mod->maxComputed = PetscMax(mod->maxComputed, link->offset); } 1059c4762a1bSJed Brown } 1060c4762a1bSJed Brown } 1061c4762a1bSJed Brown PetscFunctionReturn(0); 1062c4762a1bSJed Brown } 1063c4762a1bSJed Brown 10649371c9d4SSatish Balay static PetscErrorCode FunctionalLinkDestroy(FunctionalLink *link) { 1065c4762a1bSJed Brown FunctionalLink l, next; 1066c4762a1bSJed Brown 1067c4762a1bSJed Brown PetscFunctionBeginUser; 1068c4762a1bSJed Brown if (!link) PetscFunctionReturn(0); 1069c4762a1bSJed Brown l = *link; 1070c4762a1bSJed Brown *link = NULL; 1071c4762a1bSJed Brown for (; l; l = next) { 1072c4762a1bSJed Brown next = l->next; 10739566063dSJacob Faibussowitsch PetscCall(PetscFree(l->name)); 10749566063dSJacob Faibussowitsch PetscCall(PetscFree(l)); 1075c4762a1bSJed Brown } 1076c4762a1bSJed Brown PetscFunctionReturn(0); 1077c4762a1bSJed Brown } 1078c4762a1bSJed Brown 1079c4762a1bSJed Brown /* put the solution callback into a functional callback */ 10809371c9d4SSatish Balay static PetscErrorCode SolutionFunctional(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *modctx) { 1081c4762a1bSJed Brown Model mod; 10827510d9b0SBarry Smith PetscFunctionBeginUser; 1083c4762a1bSJed Brown mod = (Model)modctx; 10849566063dSJacob Faibussowitsch PetscCall((*mod->solution)(mod, time, x, u, mod->solutionctx)); 1085c4762a1bSJed Brown PetscFunctionReturn(0); 1086c4762a1bSJed Brown } 1087c4762a1bSJed Brown 10889371c9d4SSatish Balay PetscErrorCode SetInitialCondition(DM dm, Vec X, User user) { 1089c4762a1bSJed Brown PetscErrorCode (*func[1])(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx); 1090c4762a1bSJed Brown void *ctx[1]; 1091c4762a1bSJed Brown Model mod = user->model; 1092c4762a1bSJed Brown 1093c4762a1bSJed Brown PetscFunctionBeginUser; 1094c4762a1bSJed Brown func[0] = SolutionFunctional; 1095c4762a1bSJed Brown ctx[0] = (void *)mod; 10969566063dSJacob Faibussowitsch PetscCall(DMProjectFunction(dm, 0.0, func, ctx, INSERT_ALL_VALUES, X)); 1097c4762a1bSJed Brown PetscFunctionReturn(0); 1098c4762a1bSJed Brown } 1099c4762a1bSJed Brown 11009371c9d4SSatish Balay static PetscErrorCode OutputVTK(DM dm, const char *filename, PetscViewer *viewer) { 1101c4762a1bSJed Brown PetscFunctionBeginUser; 11029566063dSJacob Faibussowitsch PetscCall(PetscViewerCreate(PetscObjectComm((PetscObject)dm), viewer)); 11039566063dSJacob Faibussowitsch PetscCall(PetscViewerSetType(*viewer, PETSCVIEWERVTK)); 11049566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetName(*viewer, filename)); 1105c4762a1bSJed Brown PetscFunctionReturn(0); 1106c4762a1bSJed Brown } 1107c4762a1bSJed Brown 11089371c9d4SSatish Balay static PetscErrorCode MonitorVTK(TS ts, PetscInt stepnum, PetscReal time, Vec X, void *ctx) { 1109c4762a1bSJed Brown User user = (User)ctx; 11103e9753d6SMatthew G. Knepley DM dm, plex; 1111c4762a1bSJed Brown PetscViewer viewer; 1112c4762a1bSJed Brown char filename[PETSC_MAX_PATH_LEN], *ftable = NULL; 1113c4762a1bSJed Brown PetscReal xnorm; 1114c4762a1bSJed Brown 1115c4762a1bSJed Brown PetscFunctionBeginUser; 11169566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)X, "u")); 11179566063dSJacob Faibussowitsch PetscCall(VecGetDM(X, &dm)); 11189566063dSJacob Faibussowitsch PetscCall(VecNorm(X, NORM_INFINITY, &xnorm)); 1119c4762a1bSJed Brown 11209371c9d4SSatish Balay if (stepnum >= 0) { stepnum += user->monitorStepOffset; } 1121c4762a1bSJed Brown if (stepnum >= 0) { /* No summary for final time */ 1122c4762a1bSJed Brown Model mod = user->model; 11233e9753d6SMatthew G. Knepley Vec cellgeom; 1124c4762a1bSJed Brown PetscInt c, cStart, cEnd, fcount, i; 1125c4762a1bSJed Brown size_t ftableused, ftablealloc; 1126c4762a1bSJed Brown const PetscScalar *cgeom, *x; 1127c4762a1bSJed Brown DM dmCell; 1128c4762a1bSJed Brown DMLabel vtkLabel; 1129c4762a1bSJed Brown PetscReal *fmin, *fmax, *fintegral, *ftmp; 11303e9753d6SMatthew G. Knepley 11319566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 11329566063dSJacob Faibussowitsch PetscCall(DMPlexGetGeometryFVM(plex, NULL, &cellgeom, NULL)); 1133c4762a1bSJed Brown fcount = mod->maxComputed + 1; 11349566063dSJacob Faibussowitsch PetscCall(PetscMalloc4(fcount, &fmin, fcount, &fmax, fcount, &fintegral, fcount, &ftmp)); 1135c4762a1bSJed Brown for (i = 0; i < fcount; i++) { 1136c4762a1bSJed Brown fmin[i] = PETSC_MAX_REAL; 1137c4762a1bSJed Brown fmax[i] = PETSC_MIN_REAL; 1138c4762a1bSJed Brown fintegral[i] = 0; 1139c4762a1bSJed Brown } 11409566063dSJacob Faibussowitsch PetscCall(VecGetDM(cellgeom, &dmCell)); 11419566063dSJacob Faibussowitsch PetscCall(DMPlexGetSimplexOrBoxCells(dmCell, 0, &cStart, &cEnd)); 11429566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(cellgeom, &cgeom)); 11439566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(X, &x)); 11449566063dSJacob Faibussowitsch PetscCall(DMGetLabel(dm, "vtk", &vtkLabel)); 1145c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 1146c4762a1bSJed Brown PetscFVCellGeom *cg; 1147c4762a1bSJed Brown const PetscScalar *cx = NULL; 1148c4762a1bSJed Brown PetscInt vtkVal = 0; 1149c4762a1bSJed Brown 1150c4762a1bSJed Brown /* not that these two routines as currently implemented work for any dm with a 1151c4762a1bSJed Brown * localSection/globalSection */ 11529566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(dmCell, c, cgeom, &cg)); 11539566063dSJacob Faibussowitsch PetscCall(DMPlexPointGlobalRead(dm, c, x, &cx)); 11549566063dSJacob Faibussowitsch if (vtkLabel) PetscCall(DMLabelGetValue(vtkLabel, c, &vtkVal)); 1155c4762a1bSJed Brown if (!vtkVal || !cx) continue; /* ghost, or not a global cell */ 1156c4762a1bSJed Brown for (i = 0; i < mod->numCall; i++) { 1157c4762a1bSJed Brown FunctionalLink flink = mod->functionalCall[i]; 11589566063dSJacob Faibussowitsch PetscCall((*flink->func)(mod, time, cg->centroid, cx, ftmp, flink->ctx)); 1159c4762a1bSJed Brown } 1160c4762a1bSJed Brown for (i = 0; i < fcount; i++) { 1161c4762a1bSJed Brown fmin[i] = PetscMin(fmin[i], ftmp[i]); 1162c4762a1bSJed Brown fmax[i] = PetscMax(fmax[i], ftmp[i]); 1163c4762a1bSJed Brown fintegral[i] += cg->volume * ftmp[i]; 1164c4762a1bSJed Brown } 1165c4762a1bSJed Brown } 11669566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(cellgeom, &cgeom)); 11679566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(X, &x)); 11689566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 11699566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(MPI_IN_PLACE, fmin, fcount, MPIU_REAL, MPIU_MIN, PetscObjectComm((PetscObject)ts))); 11709566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(MPI_IN_PLACE, fmax, fcount, MPIU_REAL, MPIU_MAX, PetscObjectComm((PetscObject)ts))); 11719566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(MPI_IN_PLACE, fintegral, fcount, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)ts))); 1172c4762a1bSJed Brown 1173c4762a1bSJed Brown ftablealloc = fcount * 100; 1174c4762a1bSJed Brown ftableused = 0; 11759566063dSJacob Faibussowitsch PetscCall(PetscMalloc1(ftablealloc, &ftable)); 1176c4762a1bSJed Brown for (i = 0; i < mod->numMonitored; i++) { 1177c4762a1bSJed Brown size_t countused; 1178c4762a1bSJed Brown char buffer[256], *p; 1179c4762a1bSJed Brown FunctionalLink flink = mod->functionalMonitored[i]; 1180c4762a1bSJed Brown PetscInt id = flink->offset; 1181c4762a1bSJed Brown if (i % 3) { 11829566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(buffer, " ", 2)); 1183c4762a1bSJed Brown p = buffer + 2; 1184c4762a1bSJed Brown } else if (i) { 1185c4762a1bSJed Brown char newline[] = "\n"; 11869566063dSJacob Faibussowitsch PetscCall(PetscMemcpy(buffer, newline, sizeof(newline) - 1)); 1187c4762a1bSJed Brown p = buffer + sizeof(newline) - 1; 1188c4762a1bSJed Brown } else { 1189c4762a1bSJed Brown p = buffer; 1190c4762a1bSJed Brown } 11919566063dSJacob Faibussowitsch PetscCall(PetscSNPrintfCount(p, sizeof buffer - (p - buffer), "%12s [%10.7g,%10.7g] int %10.7g", &countused, flink->name, (double)fmin[id], (double)fmax[id], (double)fintegral[id])); 1192c4762a1bSJed Brown countused--; 1193c4762a1bSJed Brown countused += p - buffer; 1194c4762a1bSJed Brown if (countused > ftablealloc - ftableused - 1) { /* reallocate */ 1195c4762a1bSJed Brown char *ftablenew; 1196c4762a1bSJed Brown ftablealloc = 2 * ftablealloc + countused; 11979566063dSJacob Faibussowitsch PetscCall(PetscMalloc(ftablealloc, &ftablenew)); 11989566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(ftablenew, ftable, ftableused)); 11999566063dSJacob Faibussowitsch PetscCall(PetscFree(ftable)); 1200c4762a1bSJed Brown ftable = ftablenew; 1201c4762a1bSJed Brown } 12029566063dSJacob Faibussowitsch PetscCall(PetscArraycpy(ftable + ftableused, buffer, countused)); 1203c4762a1bSJed Brown ftableused += countused; 1204c4762a1bSJed Brown ftable[ftableused] = 0; 1205c4762a1bSJed Brown } 12069566063dSJacob Faibussowitsch PetscCall(PetscFree4(fmin, fmax, fintegral, ftmp)); 1207c4762a1bSJed Brown 120863a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(PetscObjectComm((PetscObject)ts), "% 3" PetscInt_FMT " time %8.4g |x| %8.4g %s\n", stepnum, (double)time, (double)xnorm, ftable ? ftable : "")); 12099566063dSJacob Faibussowitsch PetscCall(PetscFree(ftable)); 1210c4762a1bSJed Brown } 1211c4762a1bSJed Brown if (user->vtkInterval < 1) PetscFunctionReturn(0); 1212c4762a1bSJed Brown if ((stepnum == -1) ^ (stepnum % user->vtkInterval == 0)) { 1213c4762a1bSJed Brown if (stepnum == -1) { /* Final time is not multiple of normal time interval, write it anyway */ 12149566063dSJacob Faibussowitsch PetscCall(TSGetStepNumber(ts, &stepnum)); 1215c4762a1bSJed Brown } 121663a3b9bcSJacob Faibussowitsch PetscCall(PetscSNPrintf(filename, sizeof filename, "%s-%03" PetscInt_FMT ".vtu", user->outputBasename, stepnum)); 12179566063dSJacob Faibussowitsch PetscCall(OutputVTK(dm, filename, &viewer)); 12189566063dSJacob Faibussowitsch PetscCall(VecView(X, viewer)); 12199566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 1220c4762a1bSJed Brown } 1221c4762a1bSJed Brown PetscFunctionReturn(0); 1222c4762a1bSJed Brown } 1223c4762a1bSJed Brown 12249371c9d4SSatish Balay static PetscErrorCode initializeTS(DM dm, User user, TS *ts) { 12257510d9b0SBarry Smith PetscFunctionBeginUser; 12269566063dSJacob Faibussowitsch PetscCall(TSCreate(PetscObjectComm((PetscObject)dm), ts)); 12279566063dSJacob Faibussowitsch PetscCall(TSSetType(*ts, TSSSP)); 12289566063dSJacob Faibussowitsch PetscCall(TSSetDM(*ts, dm)); 12291baa6e33SBarry Smith if (user->vtkmon) PetscCall(TSMonitorSet(*ts, MonitorVTK, user, NULL)); 12309566063dSJacob Faibussowitsch PetscCall(DMTSSetBoundaryLocal(dm, DMPlexTSComputeBoundary, user)); 12319566063dSJacob Faibussowitsch PetscCall(DMTSSetRHSFunctionLocal(dm, DMPlexTSComputeRHSFunctionFVM, user)); 12329566063dSJacob Faibussowitsch PetscCall(TSSetMaxTime(*ts, 2.0)); 12339566063dSJacob Faibussowitsch PetscCall(TSSetExactFinalTime(*ts, TS_EXACTFINALTIME_STEPOVER)); 1234c4762a1bSJed Brown PetscFunctionReturn(0); 1235c4762a1bSJed Brown } 1236c4762a1bSJed Brown 12379371c9d4SSatish Balay static PetscErrorCode adaptToleranceFVM(PetscFV fvm, TS ts, Vec sol, VecTagger refineTag, VecTagger coarsenTag, User user, TS *tsNew, Vec *solNew) { 1238c4762a1bSJed Brown DM dm, gradDM, plex, cellDM, adaptedDM = NULL; 1239c4762a1bSJed Brown Vec cellGeom, faceGeom; 1240c4762a1bSJed Brown PetscBool isForest, computeGradient; 1241c4762a1bSJed Brown Vec grad, locGrad, locX, errVec; 1242c4762a1bSJed Brown PetscInt cStart, cEnd, c, dim, nRefine, nCoarsen; 1243c4762a1bSJed Brown PetscReal minMaxInd[2] = {PETSC_MAX_REAL, PETSC_MIN_REAL}, minMaxIndGlobal[2], minInd, maxInd, time; 1244c4762a1bSJed Brown PetscScalar *errArray; 1245c4762a1bSJed Brown const PetscScalar *pointVals; 1246c4762a1bSJed Brown const PetscScalar *pointGrads; 1247c4762a1bSJed Brown const PetscScalar *pointGeom; 1248c4762a1bSJed Brown DMLabel adaptLabel = NULL; 1249c4762a1bSJed Brown IS refineIS, coarsenIS; 1250c4762a1bSJed Brown 12517510d9b0SBarry Smith PetscFunctionBeginUser; 12529566063dSJacob Faibussowitsch PetscCall(TSGetTime(ts, &time)); 12539566063dSJacob Faibussowitsch PetscCall(VecGetDM(sol, &dm)); 12549566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 12559566063dSJacob Faibussowitsch PetscCall(PetscFVGetComputeGradients(fvm, &computeGradient)); 12569566063dSJacob Faibussowitsch PetscCall(PetscFVSetComputeGradients(fvm, PETSC_TRUE)); 12579566063dSJacob Faibussowitsch PetscCall(DMIsForest(dm, &isForest)); 12589566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 12599566063dSJacob Faibussowitsch PetscCall(DMPlexGetDataFVM(plex, fvm, &cellGeom, &faceGeom, &gradDM)); 12609566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(plex, &locX)); 12619566063dSJacob Faibussowitsch PetscCall(DMPlexInsertBoundaryValues(plex, PETSC_TRUE, locX, 0.0, faceGeom, cellGeom, NULL)); 12629566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalBegin(plex, sol, INSERT_VALUES, locX)); 12639566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalEnd(plex, sol, INSERT_VALUES, locX)); 12649566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(gradDM, &grad)); 12659566063dSJacob Faibussowitsch PetscCall(DMPlexReconstructGradientsFVM(plex, locX, grad)); 12669566063dSJacob Faibussowitsch PetscCall(DMCreateLocalVector(gradDM, &locGrad)); 12679566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalBegin(gradDM, grad, INSERT_VALUES, locGrad)); 12689566063dSJacob Faibussowitsch PetscCall(DMGlobalToLocalEnd(gradDM, grad, INSERT_VALUES, locGrad)); 12699566063dSJacob Faibussowitsch PetscCall(VecDestroy(&grad)); 12709566063dSJacob Faibussowitsch PetscCall(DMPlexGetSimplexOrBoxCells(plex, 0, &cStart, &cEnd)); 12719566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(locGrad, &pointGrads)); 12729566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(cellGeom, &pointGeom)); 12739566063dSJacob Faibussowitsch PetscCall(VecGetArrayRead(locX, &pointVals)); 12749566063dSJacob Faibussowitsch PetscCall(VecGetDM(cellGeom, &cellDM)); 12759566063dSJacob Faibussowitsch PetscCall(DMLabelCreate(PETSC_COMM_SELF, "adapt", &adaptLabel)); 12769566063dSJacob Faibussowitsch PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)plex), cEnd - cStart, PETSC_DETERMINE, &errVec)); 12779566063dSJacob Faibussowitsch PetscCall(VecSetUp(errVec)); 12789566063dSJacob Faibussowitsch PetscCall(VecGetArray(errVec, &errArray)); 1279c4762a1bSJed Brown for (c = cStart; c < cEnd; c++) { 1280c4762a1bSJed Brown PetscReal errInd = 0.; 1281c4762a1bSJed Brown PetscScalar *pointGrad; 1282c4762a1bSJed Brown PetscScalar *pointVal; 1283c4762a1bSJed Brown PetscFVCellGeom *cg; 1284c4762a1bSJed Brown 12859566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(gradDM, c, pointGrads, &pointGrad)); 12869566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(cellDM, c, pointGeom, &cg)); 12879566063dSJacob Faibussowitsch PetscCall(DMPlexPointLocalRead(plex, c, pointVals, &pointVal)); 1288c4762a1bSJed Brown 12899566063dSJacob Faibussowitsch PetscCall((user->model->errorIndicator)(dim, cg->volume, user->model->physics->dof, pointVal, pointGrad, &errInd, user->model->errorCtx)); 1290c4762a1bSJed Brown errArray[c - cStart] = errInd; 1291c4762a1bSJed Brown minMaxInd[0] = PetscMin(minMaxInd[0], errInd); 1292c4762a1bSJed Brown minMaxInd[1] = PetscMax(minMaxInd[1], errInd); 1293c4762a1bSJed Brown } 12949566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(errVec, &errArray)); 12959566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(locX, &pointVals)); 12969566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(cellGeom, &pointGeom)); 12979566063dSJacob Faibussowitsch PetscCall(VecRestoreArrayRead(locGrad, &pointGrads)); 12989566063dSJacob Faibussowitsch PetscCall(VecDestroy(&locGrad)); 12999566063dSJacob Faibussowitsch PetscCall(VecDestroy(&locX)); 13009566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 1301c4762a1bSJed Brown 13029566063dSJacob Faibussowitsch PetscCall(VecTaggerComputeIS(refineTag, errVec, &refineIS, NULL)); 13039566063dSJacob Faibussowitsch PetscCall(VecTaggerComputeIS(coarsenTag, errVec, &coarsenIS, NULL)); 13049566063dSJacob Faibussowitsch PetscCall(ISGetSize(refineIS, &nRefine)); 13059566063dSJacob Faibussowitsch PetscCall(ISGetSize(coarsenIS, &nCoarsen)); 13069566063dSJacob Faibussowitsch if (nRefine) PetscCall(DMLabelSetStratumIS(adaptLabel, DM_ADAPT_REFINE, refineIS)); 13079566063dSJacob Faibussowitsch if (nCoarsen) PetscCall(DMLabelSetStratumIS(adaptLabel, DM_ADAPT_COARSEN, coarsenIS)); 13089566063dSJacob Faibussowitsch PetscCall(ISDestroy(&coarsenIS)); 13099566063dSJacob Faibussowitsch PetscCall(ISDestroy(&refineIS)); 13109566063dSJacob Faibussowitsch PetscCall(VecDestroy(&errVec)); 1311c4762a1bSJed Brown 13129566063dSJacob Faibussowitsch PetscCall(PetscFVSetComputeGradients(fvm, computeGradient)); 1313c4762a1bSJed Brown minMaxInd[1] = -minMaxInd[1]; 13149566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(minMaxInd, minMaxIndGlobal, 2, MPIU_REAL, MPI_MIN, PetscObjectComm((PetscObject)dm))); 1315c4762a1bSJed Brown minInd = minMaxIndGlobal[0]; 1316c4762a1bSJed Brown maxInd = -minMaxIndGlobal[1]; 131763a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(ts, "error indicator range (%E, %E)\n", (double)minInd, (double)maxInd)); 1318c4762a1bSJed Brown if (nRefine || nCoarsen) { /* at least one cell is over the refinement threshold */ 13199566063dSJacob Faibussowitsch PetscCall(DMAdaptLabel(dm, adaptLabel, &adaptedDM)); 1320c4762a1bSJed Brown } 13219566063dSJacob Faibussowitsch PetscCall(DMLabelDestroy(&adaptLabel)); 1322c4762a1bSJed Brown if (adaptedDM) { 132363a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(ts, "Adapted mesh, marking %" PetscInt_FMT " cells for refinement, and %" PetscInt_FMT " cells for coarsening\n", nRefine, nCoarsen)); 13249566063dSJacob Faibussowitsch if (tsNew) PetscCall(initializeTS(adaptedDM, user, tsNew)); 1325c4762a1bSJed Brown if (solNew) { 13269566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(adaptedDM, solNew)); 13279566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)*solNew, "solution")); 13289566063dSJacob Faibussowitsch PetscCall(DMForestTransferVec(dm, sol, adaptedDM, *solNew, PETSC_TRUE, time)); 1329c4762a1bSJed Brown } 13309566063dSJacob Faibussowitsch if (isForest) PetscCall(DMForestSetAdaptivityForest(adaptedDM, NULL)); /* clear internal references to the previous dm */ 13319566063dSJacob Faibussowitsch PetscCall(DMDestroy(&adaptedDM)); 1332c4762a1bSJed Brown } else { 1333c4762a1bSJed Brown if (tsNew) *tsNew = NULL; 1334c4762a1bSJed Brown if (solNew) *solNew = NULL; 1335c4762a1bSJed Brown } 1336c4762a1bSJed Brown PetscFunctionReturn(0); 1337c4762a1bSJed Brown } 1338c4762a1bSJed Brown 13399371c9d4SSatish Balay int main(int argc, char **argv) { 1340c4762a1bSJed Brown MPI_Comm comm; 1341c4762a1bSJed Brown PetscDS prob; 1342c4762a1bSJed Brown PetscFV fvm; 1343c4762a1bSJed Brown PetscLimiter limiter = NULL, noneLimiter = NULL; 1344c4762a1bSJed Brown User user; 1345c4762a1bSJed Brown Model mod; 1346c4762a1bSJed Brown Physics phys; 13473e9753d6SMatthew G. Knepley DM dm, plex; 1348c4762a1bSJed Brown PetscReal ftime, cfl, dt, minRadius; 1349c4762a1bSJed Brown PetscInt dim, nsteps; 1350c4762a1bSJed Brown TS ts; 1351c4762a1bSJed Brown TSConvergedReason reason; 1352c4762a1bSJed Brown Vec X; 1353c4762a1bSJed Brown PetscViewer viewer; 1354b5a892a1SMatthew G. Knepley PetscBool vtkCellGeom, useAMR; 135530602db0SMatthew G. Knepley PetscInt adaptInterval; 1356c4762a1bSJed Brown char physname[256] = "advect"; 1357c4762a1bSJed Brown VecTagger refineTag = NULL, coarsenTag = NULL; 1358c4762a1bSJed Brown 1359327415f7SBarry Smith PetscFunctionBeginUser; 13609566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 1361c4762a1bSJed Brown comm = PETSC_COMM_WORLD; 1362c4762a1bSJed Brown 13639566063dSJacob Faibussowitsch PetscCall(PetscNew(&user)); 13649566063dSJacob Faibussowitsch PetscCall(PetscNew(&user->model)); 13659566063dSJacob Faibussowitsch PetscCall(PetscNew(&user->model->physics)); 1366c4762a1bSJed Brown mod = user->model; 1367c4762a1bSJed Brown phys = mod->physics; 1368c4762a1bSJed Brown mod->comm = comm; 1369c4762a1bSJed Brown useAMR = PETSC_FALSE; 1370c4762a1bSJed Brown adaptInterval = 1; 1371c4762a1bSJed Brown 1372c4762a1bSJed Brown /* Register physical models to be available on the command line */ 13739566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&PhysicsList, "advect", PhysicsCreate_Advect)); 13749566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&PhysicsList, "sw", PhysicsCreate_SW)); 13759566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&PhysicsList, "euler", PhysicsCreate_Euler)); 1376c4762a1bSJed Brown 1377d0609cedSBarry Smith PetscOptionsBegin(comm, NULL, "Unstructured Finite Volume Mesh Options", ""); 1378c4762a1bSJed Brown { 1379c4762a1bSJed Brown cfl = 0.9 * 4; /* default SSPRKS2 with s=5 stages is stable for CFL number s-1 */ 13809566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-ufv_cfl", "CFL number per step", "", cfl, &cfl, NULL)); 1381c4762a1bSJed Brown user->vtkInterval = 1; 13829566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-ufv_vtk_interval", "VTK output interval (0 to disable)", "", user->vtkInterval, &user->vtkInterval, NULL)); 1383c4762a1bSJed Brown user->vtkmon = PETSC_TRUE; 13849566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-ufv_vtk_monitor", "Use VTKMonitor routine", "", user->vtkmon, &user->vtkmon, NULL)); 1385c4762a1bSJed Brown vtkCellGeom = PETSC_FALSE; 13869566063dSJacob Faibussowitsch PetscCall(PetscStrcpy(user->outputBasename, "ex11")); 13879566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-ufv_vtk_basename", "VTK output basename", "", user->outputBasename, user->outputBasename, sizeof(user->outputBasename), NULL)); 13889566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-ufv_vtk_cellgeom", "Write cell geometry (for debugging)", "", vtkCellGeom, &vtkCellGeom, NULL)); 13899566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-ufv_use_amr", "use local adaptive mesh refinement", "", useAMR, &useAMR, NULL)); 13909566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-ufv_adapt_interval", "time steps between AMR", "", adaptInterval, &adaptInterval, NULL)); 1391c4762a1bSJed Brown } 1392d0609cedSBarry Smith PetscOptionsEnd(); 1393c4762a1bSJed Brown 1394c4762a1bSJed Brown if (useAMR) { 1395c4762a1bSJed Brown VecTaggerBox refineBox, coarsenBox; 1396c4762a1bSJed Brown 1397c4762a1bSJed Brown refineBox.min = refineBox.max = PETSC_MAX_REAL; 1398c4762a1bSJed Brown coarsenBox.min = coarsenBox.max = PETSC_MIN_REAL; 1399c4762a1bSJed Brown 14009566063dSJacob Faibussowitsch PetscCall(VecTaggerCreate(comm, &refineTag)); 14019566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)refineTag, "refine_")); 14029566063dSJacob Faibussowitsch PetscCall(VecTaggerSetType(refineTag, VECTAGGERABSOLUTE)); 14039566063dSJacob Faibussowitsch PetscCall(VecTaggerAbsoluteSetBox(refineTag, &refineBox)); 14049566063dSJacob Faibussowitsch PetscCall(VecTaggerSetFromOptions(refineTag)); 14059566063dSJacob Faibussowitsch PetscCall(VecTaggerSetUp(refineTag)); 14069566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)refineTag, NULL, "-tag_view")); 1407c4762a1bSJed Brown 14089566063dSJacob Faibussowitsch PetscCall(VecTaggerCreate(comm, &coarsenTag)); 14099566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)coarsenTag, "coarsen_")); 14109566063dSJacob Faibussowitsch PetscCall(VecTaggerSetType(coarsenTag, VECTAGGERABSOLUTE)); 14119566063dSJacob Faibussowitsch PetscCall(VecTaggerAbsoluteSetBox(coarsenTag, &coarsenBox)); 14129566063dSJacob Faibussowitsch PetscCall(VecTaggerSetFromOptions(coarsenTag)); 14139566063dSJacob Faibussowitsch PetscCall(VecTaggerSetUp(coarsenTag)); 14149566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)coarsenTag, NULL, "-tag_view")); 1415c4762a1bSJed Brown } 1416c4762a1bSJed Brown 1417d0609cedSBarry Smith PetscOptionsBegin(comm, NULL, "Unstructured Finite Volume Physics Options", ""); 1418c4762a1bSJed Brown { 1419c4762a1bSJed Brown PetscErrorCode (*physcreate)(Model, Physics, PetscOptionItems *); 14209566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-physics", "Physics module to solve", "", PhysicsList, physname, physname, sizeof physname, NULL)); 14219566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(PhysicsList, physname, &physcreate)); 14229566063dSJacob Faibussowitsch PetscCall(PetscMemzero(phys, sizeof(struct _n_Physics))); 14239566063dSJacob Faibussowitsch PetscCall((*physcreate)(mod, phys, PetscOptionsObject)); 1424c4762a1bSJed Brown /* Count number of fields and dofs */ 1425c4762a1bSJed Brown for (phys->nfields = 0, phys->dof = 0; phys->field_desc[phys->nfields].name; phys->nfields++) phys->dof += phys->field_desc[phys->nfields].dof; 142654c59aa7SJacob Faibussowitsch PetscCheck(phys->dof > 0, comm, PETSC_ERR_ARG_WRONGSTATE, "Physics '%s' did not set dof", physname); 14279566063dSJacob Faibussowitsch PetscCall(ModelFunctionalSetFromOptions(mod, PetscOptionsObject)); 1428c4762a1bSJed Brown } 1429d0609cedSBarry Smith PetscOptionsEnd(); 1430c4762a1bSJed Brown 1431c4762a1bSJed Brown /* Create mesh */ 1432c4762a1bSJed Brown { 143330602db0SMatthew G. Knepley PetscInt i; 143430602db0SMatthew G. Knepley 14359566063dSJacob Faibussowitsch PetscCall(DMCreate(comm, &dm)); 14369566063dSJacob Faibussowitsch PetscCall(DMSetType(dm, DMPLEX)); 14379566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(dm)); 14389371c9d4SSatish Balay for (i = 0; i < DIM; i++) { 14399371c9d4SSatish Balay mod->bounds[2 * i] = 0.; 14409371c9d4SSatish Balay mod->bounds[2 * i + 1] = 1.; 14419371c9d4SSatish Balay }; 1442c4762a1bSJed Brown dim = DIM; 144330602db0SMatthew G. Knepley { /* a null name means just do a hex box */ 144430602db0SMatthew G. Knepley PetscInt cells[3] = {1, 1, 1}, n = 3; 144530602db0SMatthew G. Knepley PetscBool flg2, skew = PETSC_FALSE; 1446c4762a1bSJed Brown PetscInt nret2 = 2 * DIM; 1447d0609cedSBarry Smith PetscOptionsBegin(comm, NULL, "Rectangular mesh options", ""); 14489566063dSJacob Faibussowitsch PetscCall(PetscOptionsRealArray("-grid_bounds", "bounds of the mesh in each direction (i.e., x_min,x_max,y_min,y_max", "", mod->bounds, &nret2, &flg2)); 14499566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-grid_skew_60", "Skew grid for 60 degree shock mesh", "", skew, &skew, NULL)); 14509566063dSJacob Faibussowitsch PetscCall(PetscOptionsIntArray("-dm_plex_box_faces", "Number of faces along each dimension", "", cells, &n, NULL)); 1451d0609cedSBarry Smith PetscOptionsEnd(); 145230602db0SMatthew G. Knepley /* TODO Rewrite this with Mark, and remove grid_bounds at that time */ 1453c4762a1bSJed Brown if (flg2) { 1454c4762a1bSJed Brown PetscInt dimEmbed, i; 1455c4762a1bSJed Brown PetscInt nCoords; 1456c4762a1bSJed Brown PetscScalar *coords; 1457c4762a1bSJed Brown Vec coordinates; 1458c4762a1bSJed Brown 14599566063dSJacob Faibussowitsch PetscCall(DMGetCoordinatesLocal(dm, &coordinates)); 14609566063dSJacob Faibussowitsch PetscCall(DMGetCoordinateDim(dm, &dimEmbed)); 14619566063dSJacob Faibussowitsch PetscCall(VecGetLocalSize(coordinates, &nCoords)); 146254c59aa7SJacob Faibussowitsch PetscCheck(!(nCoords % dimEmbed), PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Coordinate vector the wrong size"); 14639566063dSJacob Faibussowitsch PetscCall(VecGetArray(coordinates, &coords)); 1464c4762a1bSJed Brown for (i = 0; i < nCoords; i += dimEmbed) { 1465c4762a1bSJed Brown PetscInt j; 1466c4762a1bSJed Brown 1467c4762a1bSJed Brown PetscScalar *coord = &coords[i]; 1468c4762a1bSJed Brown for (j = 0; j < dimEmbed; j++) { 1469c4762a1bSJed Brown coord[j] = mod->bounds[2 * j] + coord[j] * (mod->bounds[2 * j + 1] - mod->bounds[2 * j]); 1470c4762a1bSJed Brown if (dim == 2 && cells[1] == 1 && j == 0 && skew) { 1471c4762a1bSJed Brown if (cells[0] == 2 && i == 8) { 1472c4762a1bSJed Brown coord[j] = .57735026918963; /* hack to get 60 deg skewed mesh */ 147330602db0SMatthew G. Knepley } else if (cells[0] == 3) { 1474c4762a1bSJed Brown if (i == 2 || i == 10) coord[j] = mod->bounds[1] / 4.; 1475c4762a1bSJed Brown else if (i == 4) coord[j] = mod->bounds[1] / 2.; 1476c4762a1bSJed Brown else if (i == 12) coord[j] = 1.57735026918963 * mod->bounds[1] / 2.; 1477c4762a1bSJed Brown } 1478c4762a1bSJed Brown } 1479c4762a1bSJed Brown } 1480c4762a1bSJed Brown } 14819566063dSJacob Faibussowitsch PetscCall(VecRestoreArray(coordinates, &coords)); 14829566063dSJacob Faibussowitsch PetscCall(DMSetCoordinatesLocal(dm, coordinates)); 1483c4762a1bSJed Brown } 1484c4762a1bSJed Brown } 1485c4762a1bSJed Brown } 14869566063dSJacob Faibussowitsch PetscCall(DMViewFromOptions(dm, NULL, "-orig_dm_view")); 14879566063dSJacob Faibussowitsch PetscCall(DMGetDimension(dm, &dim)); 1488c4762a1bSJed Brown 1489c4762a1bSJed Brown /* set up BCs, functions, tags */ 14909566063dSJacob Faibussowitsch PetscCall(DMCreateLabel(dm, "Face Sets")); 1491c4762a1bSJed Brown mod->errorIndicator = ErrorIndicator_Simple; 1492c4762a1bSJed Brown 1493c4762a1bSJed Brown { 1494c4762a1bSJed Brown DM gdm; 1495c4762a1bSJed Brown 14969566063dSJacob Faibussowitsch PetscCall(DMPlexConstructGhostCells(dm, NULL, NULL, &gdm)); 14979566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 1498c4762a1bSJed Brown dm = gdm; 14999566063dSJacob Faibussowitsch PetscCall(DMViewFromOptions(dm, NULL, "-dm_view")); 1500c4762a1bSJed Brown } 1501c4762a1bSJed Brown 15029566063dSJacob Faibussowitsch PetscCall(PetscFVCreate(comm, &fvm)); 15039566063dSJacob Faibussowitsch PetscCall(PetscFVSetFromOptions(fvm)); 15049566063dSJacob Faibussowitsch PetscCall(PetscFVSetNumComponents(fvm, phys->dof)); 15059566063dSJacob Faibussowitsch PetscCall(PetscFVSetSpatialDimension(fvm, dim)); 15069566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)fvm, "")); 1507c4762a1bSJed Brown { 1508c4762a1bSJed Brown PetscInt f, dof; 1509c4762a1bSJed Brown for (f = 0, dof = 0; f < phys->nfields; f++) { 1510c4762a1bSJed Brown PetscInt newDof = phys->field_desc[f].dof; 1511c4762a1bSJed Brown 1512c4762a1bSJed Brown if (newDof == 1) { 15139566063dSJacob Faibussowitsch PetscCall(PetscFVSetComponentName(fvm, dof, phys->field_desc[f].name)); 15149371c9d4SSatish Balay } else { 1515c4762a1bSJed Brown PetscInt j; 1516c4762a1bSJed Brown 1517c4762a1bSJed Brown for (j = 0; j < newDof; j++) { 1518c4762a1bSJed Brown char compName[256] = "Unknown"; 1519c4762a1bSJed Brown 152063a3b9bcSJacob Faibussowitsch PetscCall(PetscSNPrintf(compName, sizeof(compName), "%s_%" PetscInt_FMT, phys->field_desc[f].name, j)); 15219566063dSJacob Faibussowitsch PetscCall(PetscFVSetComponentName(fvm, dof + j, compName)); 1522c4762a1bSJed Brown } 1523c4762a1bSJed Brown } 1524c4762a1bSJed Brown dof += newDof; 1525c4762a1bSJed Brown } 1526c4762a1bSJed Brown } 1527c4762a1bSJed Brown /* FV is now structured with one field having all physics as components */ 15289566063dSJacob Faibussowitsch PetscCall(DMAddField(dm, NULL, (PetscObject)fvm)); 15299566063dSJacob Faibussowitsch PetscCall(DMCreateDS(dm)); 15309566063dSJacob Faibussowitsch PetscCall(DMGetDS(dm, &prob)); 15319566063dSJacob Faibussowitsch PetscCall(PetscDSSetRiemannSolver(prob, 0, user->model->physics->riemann)); 15329566063dSJacob Faibussowitsch PetscCall(PetscDSSetContext(prob, 0, user->model->physics)); 15339566063dSJacob Faibussowitsch PetscCall((*mod->setupbc)(dm, prob, phys)); 15349566063dSJacob Faibussowitsch PetscCall(PetscDSSetFromOptions(prob)); 1535c4762a1bSJed Brown { 1536c4762a1bSJed Brown char convType[256]; 1537c4762a1bSJed Brown PetscBool flg; 1538c4762a1bSJed Brown 1539d0609cedSBarry Smith PetscOptionsBegin(comm, "", "Mesh conversion options", "DMPLEX"); 15409566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-dm_type", "Convert DMPlex to another format", "ex12", DMList, DMPLEX, convType, 256, &flg)); 1541d0609cedSBarry Smith PetscOptionsEnd(); 1542c4762a1bSJed Brown if (flg) { 1543c4762a1bSJed Brown DM dmConv; 1544c4762a1bSJed Brown 15459566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, convType, &dmConv)); 1546c4762a1bSJed Brown if (dmConv) { 15479566063dSJacob Faibussowitsch PetscCall(DMViewFromOptions(dmConv, NULL, "-dm_conv_view")); 15489566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 1549c4762a1bSJed Brown dm = dmConv; 15509566063dSJacob Faibussowitsch PetscCall(DMSetFromOptions(dm)); 1551c4762a1bSJed Brown } 1552c4762a1bSJed Brown } 1553c4762a1bSJed Brown } 1554c4762a1bSJed Brown 15559566063dSJacob Faibussowitsch PetscCall(initializeTS(dm, user, &ts)); 1556c4762a1bSJed Brown 15579566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(dm, &X)); 15589566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)X, "solution")); 15599566063dSJacob Faibussowitsch PetscCall(SetInitialCondition(dm, X, user)); 1560c4762a1bSJed Brown if (useAMR) { 1561c4762a1bSJed Brown PetscInt adaptIter; 1562c4762a1bSJed Brown 1563c4762a1bSJed Brown /* use no limiting when reconstructing gradients for adaptivity */ 15649566063dSJacob Faibussowitsch PetscCall(PetscFVGetLimiter(fvm, &limiter)); 15659566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)limiter)); 15669566063dSJacob Faibussowitsch PetscCall(PetscLimiterCreate(PetscObjectComm((PetscObject)fvm), &noneLimiter)); 15679566063dSJacob Faibussowitsch PetscCall(PetscLimiterSetType(noneLimiter, PETSCLIMITERNONE)); 1568c4762a1bSJed Brown 15699566063dSJacob Faibussowitsch PetscCall(PetscFVSetLimiter(fvm, noneLimiter)); 1570c4762a1bSJed Brown for (adaptIter = 0;; ++adaptIter) { 1571c4762a1bSJed Brown PetscLogDouble bytes; 1572c4762a1bSJed Brown TS tsNew = NULL; 1573c4762a1bSJed Brown 15749566063dSJacob Faibussowitsch PetscCall(PetscMemoryGetCurrentUsage(&bytes)); 157563a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(ts, "refinement loop %" PetscInt_FMT ": memory used %g\n", adaptIter, (double)bytes)); 15769566063dSJacob Faibussowitsch PetscCall(DMViewFromOptions(dm, NULL, "-initial_dm_view")); 15779566063dSJacob Faibussowitsch PetscCall(VecViewFromOptions(X, NULL, "-initial_vec_view")); 1578c4762a1bSJed Brown #if 0 1579c4762a1bSJed Brown if (viewInitial) { 1580c4762a1bSJed Brown PetscViewer viewer; 1581c4762a1bSJed Brown char buf[256]; 1582c4762a1bSJed Brown PetscBool isHDF5, isVTK; 1583c4762a1bSJed Brown 15849566063dSJacob Faibussowitsch PetscCall(PetscViewerCreate(comm,&viewer)); 15859566063dSJacob Faibussowitsch PetscCall(PetscViewerSetType(viewer,PETSCVIEWERVTK)); 15869566063dSJacob Faibussowitsch PetscCall(PetscViewerSetOptionsPrefix(viewer,"initial_")); 15879566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions(viewer)); 15889566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&isHDF5)); 15899566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERVTK,&isVTK)); 1590c4762a1bSJed Brown if (isHDF5) { 159163a3b9bcSJacob Faibussowitsch PetscCall(PetscSNPrintf(buf, 256, "ex11-initial-%" PetscInt_FMT ".h5", adaptIter)); 1592c4762a1bSJed Brown } else if (isVTK) { 159363a3b9bcSJacob Faibussowitsch PetscCall(PetscSNPrintf(buf, 256, "ex11-initial-%" PetscInt_FMT ".vtu", adaptIter)); 15949566063dSJacob Faibussowitsch PetscCall(PetscViewerPushFormat(viewer,PETSC_VIEWER_VTK_VTU)); 1595c4762a1bSJed Brown } 15969566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetMode(viewer,FILE_MODE_WRITE)); 15979566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetName(viewer,buf)); 1598c4762a1bSJed Brown if (isHDF5) { 15999566063dSJacob Faibussowitsch PetscCall(DMView(dm,viewer)); 16009566063dSJacob Faibussowitsch PetscCall(PetscViewerFileSetMode(viewer,FILE_MODE_UPDATE)); 1601c4762a1bSJed Brown } 16029566063dSJacob Faibussowitsch PetscCall(VecView(X,viewer)); 16039566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 1604c4762a1bSJed Brown } 1605c4762a1bSJed Brown #endif 1606c4762a1bSJed Brown 16079566063dSJacob Faibussowitsch PetscCall(adaptToleranceFVM(fvm, ts, X, refineTag, coarsenTag, user, &tsNew, NULL)); 1608c4762a1bSJed Brown if (!tsNew) { 1609c4762a1bSJed Brown break; 1610c4762a1bSJed Brown } else { 16119566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 16129566063dSJacob Faibussowitsch PetscCall(VecDestroy(&X)); 16139566063dSJacob Faibussowitsch PetscCall(TSDestroy(&ts)); 1614c4762a1bSJed Brown ts = tsNew; 16159566063dSJacob Faibussowitsch PetscCall(TSGetDM(ts, &dm)); 16169566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)dm)); 16179566063dSJacob Faibussowitsch PetscCall(DMCreateGlobalVector(dm, &X)); 16189566063dSJacob Faibussowitsch PetscCall(PetscObjectSetName((PetscObject)X, "solution")); 16199566063dSJacob Faibussowitsch PetscCall(SetInitialCondition(dm, X, user)); 1620c4762a1bSJed Brown } 1621c4762a1bSJed Brown } 1622c4762a1bSJed Brown /* restore original limiter */ 16239566063dSJacob Faibussowitsch PetscCall(PetscFVSetLimiter(fvm, limiter)); 1624c4762a1bSJed Brown } 1625c4762a1bSJed Brown 16269566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 1627c4762a1bSJed Brown if (vtkCellGeom) { 1628c4762a1bSJed Brown DM dmCell; 1629c4762a1bSJed Brown Vec cellgeom, partition; 1630c4762a1bSJed Brown 16319566063dSJacob Faibussowitsch PetscCall(DMPlexGetGeometryFVM(plex, NULL, &cellgeom, NULL)); 16329566063dSJacob Faibussowitsch PetscCall(OutputVTK(dm, "ex11-cellgeom.vtk", &viewer)); 16339566063dSJacob Faibussowitsch PetscCall(VecView(cellgeom, viewer)); 16349566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 16359566063dSJacob Faibussowitsch PetscCall(CreatePartitionVec(dm, &dmCell, &partition)); 16369566063dSJacob Faibussowitsch PetscCall(OutputVTK(dmCell, "ex11-partition.vtk", &viewer)); 16379566063dSJacob Faibussowitsch PetscCall(VecView(partition, viewer)); 16389566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 16399566063dSJacob Faibussowitsch PetscCall(VecDestroy(&partition)); 16409566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dmCell)); 1641c4762a1bSJed Brown } 1642c4762a1bSJed Brown /* collect max maxspeed from all processes -- todo */ 16439566063dSJacob Faibussowitsch PetscCall(DMPlexGetGeometryFVM(plex, NULL, NULL, &minRadius)); 16449566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 16459566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(&phys->maxspeed, &mod->maxspeed, 1, MPIU_REAL, MPIU_MAX, PetscObjectComm((PetscObject)ts))); 164654c59aa7SJacob Faibussowitsch PetscCheck(mod->maxspeed > 0, comm, PETSC_ERR_ARG_WRONGSTATE, "Physics '%s' did not set maxspeed", physname); 1647c4762a1bSJed Brown dt = cfl * minRadius / mod->maxspeed; 16489566063dSJacob Faibussowitsch PetscCall(TSSetTimeStep(ts, dt)); 16499566063dSJacob Faibussowitsch PetscCall(TSSetFromOptions(ts)); 1650c4762a1bSJed Brown if (!useAMR) { 16519566063dSJacob Faibussowitsch PetscCall(TSSolve(ts, X)); 16529566063dSJacob Faibussowitsch PetscCall(TSGetSolveTime(ts, &ftime)); 16539566063dSJacob Faibussowitsch PetscCall(TSGetStepNumber(ts, &nsteps)); 1654c4762a1bSJed Brown } else { 1655c4762a1bSJed Brown PetscReal finalTime; 1656c4762a1bSJed Brown PetscInt adaptIter; 1657c4762a1bSJed Brown TS tsNew = NULL; 1658c4762a1bSJed Brown Vec solNew = NULL; 1659c4762a1bSJed Brown 16609566063dSJacob Faibussowitsch PetscCall(TSGetMaxTime(ts, &finalTime)); 16619566063dSJacob Faibussowitsch PetscCall(TSSetMaxSteps(ts, adaptInterval)); 16629566063dSJacob Faibussowitsch PetscCall(TSSolve(ts, X)); 16639566063dSJacob Faibussowitsch PetscCall(TSGetSolveTime(ts, &ftime)); 16649566063dSJacob Faibussowitsch PetscCall(TSGetStepNumber(ts, &nsteps)); 1665c4762a1bSJed Brown for (adaptIter = 0; ftime < finalTime; adaptIter++) { 1666c4762a1bSJed Brown PetscLogDouble bytes; 1667c4762a1bSJed Brown 16689566063dSJacob Faibussowitsch PetscCall(PetscMemoryGetCurrentUsage(&bytes)); 166963a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(ts, "AMR time step loop %" PetscInt_FMT ": memory used %g\n", adaptIter, bytes)); 16709566063dSJacob Faibussowitsch PetscCall(PetscFVSetLimiter(fvm, noneLimiter)); 16719566063dSJacob Faibussowitsch PetscCall(adaptToleranceFVM(fvm, ts, X, refineTag, coarsenTag, user, &tsNew, &solNew)); 16729566063dSJacob Faibussowitsch PetscCall(PetscFVSetLimiter(fvm, limiter)); 1673c4762a1bSJed Brown if (tsNew) { 16749566063dSJacob Faibussowitsch PetscCall(PetscInfo(ts, "AMR used\n")); 16759566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 16769566063dSJacob Faibussowitsch PetscCall(VecDestroy(&X)); 16779566063dSJacob Faibussowitsch PetscCall(TSDestroy(&ts)); 1678c4762a1bSJed Brown ts = tsNew; 1679c4762a1bSJed Brown X = solNew; 16809566063dSJacob Faibussowitsch PetscCall(TSSetFromOptions(ts)); 16819566063dSJacob Faibussowitsch PetscCall(VecGetDM(X, &dm)); 16829566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)dm)); 16839566063dSJacob Faibussowitsch PetscCall(DMConvert(dm, DMPLEX, &plex)); 16849566063dSJacob Faibussowitsch PetscCall(DMPlexGetGeometryFVM(dm, NULL, NULL, &minRadius)); 16859566063dSJacob Faibussowitsch PetscCall(DMDestroy(&plex)); 16869566063dSJacob Faibussowitsch PetscCallMPI(MPI_Allreduce(&phys->maxspeed, &mod->maxspeed, 1, MPIU_REAL, MPIU_MAX, PetscObjectComm((PetscObject)ts))); 168754c59aa7SJacob Faibussowitsch PetscCheck(mod->maxspeed > 0, comm, PETSC_ERR_ARG_WRONGSTATE, "Physics '%s' did not set maxspeed", physname); 1688c4762a1bSJed Brown dt = cfl * minRadius / mod->maxspeed; 16899566063dSJacob Faibussowitsch PetscCall(TSSetStepNumber(ts, nsteps)); 16909566063dSJacob Faibussowitsch PetscCall(TSSetTime(ts, ftime)); 16919566063dSJacob Faibussowitsch PetscCall(TSSetTimeStep(ts, dt)); 1692c4762a1bSJed Brown } else { 16939566063dSJacob Faibussowitsch PetscCall(PetscInfo(ts, "AMR not used\n")); 1694c4762a1bSJed Brown } 1695c4762a1bSJed Brown user->monitorStepOffset = nsteps; 16969566063dSJacob Faibussowitsch PetscCall(TSSetMaxSteps(ts, nsteps + adaptInterval)); 16979566063dSJacob Faibussowitsch PetscCall(TSSolve(ts, X)); 16989566063dSJacob Faibussowitsch PetscCall(TSGetSolveTime(ts, &ftime)); 16999566063dSJacob Faibussowitsch PetscCall(TSGetStepNumber(ts, &nsteps)); 1700c4762a1bSJed Brown } 1701c4762a1bSJed Brown } 17029566063dSJacob Faibussowitsch PetscCall(TSGetConvergedReason(ts, &reason)); 170363a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%s at time %g after %" PetscInt_FMT " steps\n", TSConvergedReasons[reason], (double)ftime, nsteps)); 17049566063dSJacob Faibussowitsch PetscCall(TSDestroy(&ts)); 1705c4762a1bSJed Brown 17069566063dSJacob Faibussowitsch PetscCall(VecTaggerDestroy(&refineTag)); 17079566063dSJacob Faibussowitsch PetscCall(VecTaggerDestroy(&coarsenTag)); 17089566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&PhysicsList)); 17099566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&PhysicsRiemannList_SW)); 17109566063dSJacob Faibussowitsch PetscCall(FunctionalLinkDestroy(&user->model->functionalRegistry)); 17119566063dSJacob Faibussowitsch PetscCall(PetscFree(user->model->functionalMonitored)); 17129566063dSJacob Faibussowitsch PetscCall(PetscFree(user->model->functionalCall)); 17139566063dSJacob Faibussowitsch PetscCall(PetscFree(user->model->physics->data)); 17149566063dSJacob Faibussowitsch PetscCall(PetscFree(user->model->physics)); 17159566063dSJacob Faibussowitsch PetscCall(PetscFree(user->model)); 17169566063dSJacob Faibussowitsch PetscCall(PetscFree(user)); 17179566063dSJacob Faibussowitsch PetscCall(VecDestroy(&X)); 17189566063dSJacob Faibussowitsch PetscCall(PetscLimiterDestroy(&limiter)); 17199566063dSJacob Faibussowitsch PetscCall(PetscLimiterDestroy(&noneLimiter)); 17209566063dSJacob Faibussowitsch PetscCall(PetscFVDestroy(&fvm)); 17219566063dSJacob Faibussowitsch PetscCall(DMDestroy(&dm)); 17229566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 1723b122ec5aSJacob Faibussowitsch return 0; 1724c4762a1bSJed Brown } 1725c4762a1bSJed Brown 1726c4762a1bSJed Brown /* Godunov fluxs */ 17279371c9d4SSatish Balay PetscScalar cvmgp_(PetscScalar *a, PetscScalar *b, PetscScalar *test) { 1728c4762a1bSJed Brown /* System generated locals */ 1729c4762a1bSJed Brown PetscScalar ret_val; 1730c4762a1bSJed Brown 17319371c9d4SSatish Balay if (PetscRealPart(*test) > 0.) { goto L10; } 1732c4762a1bSJed Brown ret_val = *b; 1733c4762a1bSJed Brown return ret_val; 1734c4762a1bSJed Brown L10: 1735c4762a1bSJed Brown ret_val = *a; 1736c4762a1bSJed Brown return ret_val; 1737c4762a1bSJed Brown } /* cvmgp_ */ 1738c4762a1bSJed Brown 17399371c9d4SSatish Balay PetscScalar cvmgm_(PetscScalar *a, PetscScalar *b, PetscScalar *test) { 1740c4762a1bSJed Brown /* System generated locals */ 1741c4762a1bSJed Brown PetscScalar ret_val; 1742c4762a1bSJed Brown 17439371c9d4SSatish Balay if (PetscRealPart(*test) < 0.) { goto L10; } 1744c4762a1bSJed Brown ret_val = *b; 1745c4762a1bSJed Brown return ret_val; 1746c4762a1bSJed Brown L10: 1747c4762a1bSJed Brown ret_val = *a; 1748c4762a1bSJed Brown return ret_val; 1749c4762a1bSJed Brown } /* cvmgm_ */ 1750c4762a1bSJed Brown 17519371c9d4SSatish Balay int riem1mdt(PetscScalar *gaml, PetscScalar *gamr, PetscScalar *rl, PetscScalar *pl, PetscScalar *uxl, PetscScalar *rr, PetscScalar *pr, PetscScalar *uxr, PetscScalar *rstarl, PetscScalar *rstarr, PetscScalar *pstar, PetscScalar *ustar) { 1752c4762a1bSJed Brown /* Initialized data */ 1753c4762a1bSJed Brown 1754c4762a1bSJed Brown static PetscScalar smallp = 1e-8; 1755c4762a1bSJed Brown 1756c4762a1bSJed Brown /* System generated locals */ 1757c4762a1bSJed Brown int i__1; 1758c4762a1bSJed Brown PetscScalar d__1, d__2; 1759c4762a1bSJed Brown 1760c4762a1bSJed Brown /* Local variables */ 1761c4762a1bSJed Brown static int i0; 1762c4762a1bSJed Brown static PetscScalar cl, cr, wl, zl, wr, zr, pst, durl, skpr1, skpr2; 1763c4762a1bSJed Brown static int iwave; 1764c4762a1bSJed Brown static PetscScalar gascl4, gascr4, cstarl, dpstar, cstarr; 1765c4762a1bSJed Brown /* static PetscScalar csqrl, csqrr, gascl1, gascl2, gascl3, gascr1, gascr2, gascr3; */ 1766c4762a1bSJed Brown static int iterno; 1767c4762a1bSJed Brown static PetscScalar ustarl, ustarr, rarepr1, rarepr2; 1768c4762a1bSJed Brown 1769c4762a1bSJed Brown /* gascl1 = *gaml - 1.; */ 1770c4762a1bSJed Brown /* gascl2 = (*gaml + 1.) * .5; */ 1771c4762a1bSJed Brown /* gascl3 = gascl2 / *gaml; */ 1772c4762a1bSJed Brown gascl4 = 1. / (*gaml - 1.); 1773c4762a1bSJed Brown 1774c4762a1bSJed Brown /* gascr1 = *gamr - 1.; */ 1775c4762a1bSJed Brown /* gascr2 = (*gamr + 1.) * .5; */ 1776c4762a1bSJed Brown /* gascr3 = gascr2 / *gamr; */ 1777c4762a1bSJed Brown gascr4 = 1. / (*gamr - 1.); 1778c4762a1bSJed Brown iterno = 10; 1779c4762a1bSJed Brown /* find pstar: */ 1780c4762a1bSJed Brown cl = PetscSqrtScalar(*gaml * *pl / *rl); 1781c4762a1bSJed Brown cr = PetscSqrtScalar(*gamr * *pr / *rr); 1782c4762a1bSJed Brown wl = *rl * cl; 1783c4762a1bSJed Brown wr = *rr * cr; 1784c4762a1bSJed Brown /* csqrl = wl * wl; */ 1785c4762a1bSJed Brown /* csqrr = wr * wr; */ 1786c4762a1bSJed Brown *pstar = (wl * *pr + wr * *pl) / (wl + wr); 1787c4762a1bSJed Brown *pstar = PetscMax(PetscRealPart(*pstar), PetscRealPart(smallp)); 1788c4762a1bSJed Brown pst = *pl / *pr; 1789c4762a1bSJed Brown skpr1 = cr * (pst - 1.) * PetscSqrtScalar(2. / (*gamr * (*gamr - 1. + (*gamr + 1.) * pst))); 1790c4762a1bSJed Brown d__1 = (*gamr - 1.) / (*gamr * 2.); 1791c4762a1bSJed Brown rarepr2 = gascr4 * 2. * cr * (1. - PetscPowScalar(pst, d__1)); 1792c4762a1bSJed Brown pst = *pr / *pl; 1793c4762a1bSJed Brown skpr2 = cl * (pst - 1.) * PetscSqrtScalar(2. / (*gaml * (*gaml - 1. + (*gaml + 1.) * pst))); 1794c4762a1bSJed Brown d__1 = (*gaml - 1.) / (*gaml * 2.); 1795c4762a1bSJed Brown rarepr1 = gascl4 * 2. * cl * (1. - PetscPowScalar(pst, d__1)); 1796c4762a1bSJed Brown durl = *uxr - *uxl; 1797c4762a1bSJed Brown if (PetscRealPart(*pr) < PetscRealPart(*pl)) { 1798c4762a1bSJed Brown if (PetscRealPart(durl) >= PetscRealPart(rarepr1)) { 1799c4762a1bSJed Brown iwave = 100; 1800c4762a1bSJed Brown } else if (PetscRealPart(durl) <= PetscRealPart(-skpr1)) { 1801c4762a1bSJed Brown iwave = 300; 1802c4762a1bSJed Brown } else { 1803c4762a1bSJed Brown iwave = 400; 1804c4762a1bSJed Brown } 1805c4762a1bSJed Brown } else { 1806c4762a1bSJed Brown if (PetscRealPart(durl) >= PetscRealPart(rarepr2)) { 1807c4762a1bSJed Brown iwave = 100; 1808c4762a1bSJed Brown } else if (PetscRealPart(durl) <= PetscRealPart(-skpr2)) { 1809c4762a1bSJed Brown iwave = 300; 1810c4762a1bSJed Brown } else { 1811c4762a1bSJed Brown iwave = 200; 1812c4762a1bSJed Brown } 1813c4762a1bSJed Brown } 1814c4762a1bSJed Brown if (iwave == 100) { 1815c4762a1bSJed Brown /* 1-wave: rarefaction wave, 3-wave: rarefaction wave */ 1816c4762a1bSJed Brown /* case (100) */ 1817c4762a1bSJed Brown i__1 = iterno; 1818c4762a1bSJed Brown for (i0 = 1; i0 <= i__1; ++i0) { 1819c4762a1bSJed Brown d__1 = *pstar / *pl; 1820c4762a1bSJed Brown d__2 = 1. / *gaml; 1821c4762a1bSJed Brown *rstarl = *rl * PetscPowScalar(d__1, d__2); 1822c4762a1bSJed Brown cstarl = PetscSqrtScalar(*gaml * *pstar / *rstarl); 1823c4762a1bSJed Brown ustarl = *uxl - gascl4 * 2. * (cstarl - cl); 1824c4762a1bSJed Brown zl = *rstarl * cstarl; 1825c4762a1bSJed Brown d__1 = *pstar / *pr; 1826c4762a1bSJed Brown d__2 = 1. / *gamr; 1827c4762a1bSJed Brown *rstarr = *rr * PetscPowScalar(d__1, d__2); 1828c4762a1bSJed Brown cstarr = PetscSqrtScalar(*gamr * *pstar / *rstarr); 1829c4762a1bSJed Brown ustarr = *uxr + gascr4 * 2. * (cstarr - cr); 1830c4762a1bSJed Brown zr = *rstarr * cstarr; 1831c4762a1bSJed Brown dpstar = zl * zr * (ustarr - ustarl) / (zl + zr); 1832c4762a1bSJed Brown *pstar -= dpstar; 1833c4762a1bSJed Brown *pstar = PetscMax(PetscRealPart(*pstar), PetscRealPart(smallp)); 1834c4762a1bSJed Brown if (PetscAbsScalar(dpstar) / PetscRealPart(*pstar) <= 1e-8) { 1835c4762a1bSJed Brown #if 0 1836c4762a1bSJed Brown break; 1837c4762a1bSJed Brown #endif 1838c4762a1bSJed Brown } 1839c4762a1bSJed Brown } 1840c4762a1bSJed Brown /* 1-wave: shock wave, 3-wave: rarefaction wave */ 1841c4762a1bSJed Brown } else if (iwave == 200) { 1842c4762a1bSJed Brown /* case (200) */ 1843c4762a1bSJed Brown i__1 = iterno; 1844c4762a1bSJed Brown for (i0 = 1; i0 <= i__1; ++i0) { 1845c4762a1bSJed Brown pst = *pstar / *pl; 1846c4762a1bSJed Brown ustarl = *uxl - (pst - 1.) * cl * PetscSqrtScalar(2. / (*gaml * (*gaml - 1. + (*gaml + 1.) * pst))); 1847c4762a1bSJed Brown zl = *pl / cl * PetscSqrtScalar(*gaml * 2. * (*gaml - 1. + (*gaml + 1.) * pst)) * (*gaml - 1. + (*gaml + 1.) * pst) / (*gaml * 3. - 1. + (*gaml + 1.) * pst); 1848c4762a1bSJed Brown d__1 = *pstar / *pr; 1849c4762a1bSJed Brown d__2 = 1. / *gamr; 1850c4762a1bSJed Brown *rstarr = *rr * PetscPowScalar(d__1, d__2); 1851c4762a1bSJed Brown cstarr = PetscSqrtScalar(*gamr * *pstar / *rstarr); 1852c4762a1bSJed Brown zr = *rstarr * cstarr; 1853c4762a1bSJed Brown ustarr = *uxr + gascr4 * 2. * (cstarr - cr); 1854c4762a1bSJed Brown dpstar = zl * zr * (ustarr - ustarl) / (zl + zr); 1855c4762a1bSJed Brown *pstar -= dpstar; 1856c4762a1bSJed Brown *pstar = PetscMax(PetscRealPart(*pstar), PetscRealPart(smallp)); 1857c4762a1bSJed Brown if (PetscAbsScalar(dpstar) / PetscRealPart(*pstar) <= 1e-8) { 1858c4762a1bSJed Brown #if 0 1859c4762a1bSJed Brown break; 1860c4762a1bSJed Brown #endif 1861c4762a1bSJed Brown } 1862c4762a1bSJed Brown } 1863c4762a1bSJed Brown /* 1-wave: shock wave, 3-wave: shock */ 1864c4762a1bSJed Brown } else if (iwave == 300) { 1865c4762a1bSJed Brown /* case (300) */ 1866c4762a1bSJed Brown i__1 = iterno; 1867c4762a1bSJed Brown for (i0 = 1; i0 <= i__1; ++i0) { 1868c4762a1bSJed Brown pst = *pstar / *pl; 1869c4762a1bSJed Brown ustarl = *uxl - (pst - 1.) * cl * PetscSqrtScalar(2. / (*gaml * (*gaml - 1. + (*gaml + 1.) * pst))); 1870c4762a1bSJed Brown zl = *pl / cl * PetscSqrtScalar(*gaml * 2. * (*gaml - 1. + (*gaml + 1.) * pst)) * (*gaml - 1. + (*gaml + 1.) * pst) / (*gaml * 3. - 1. + (*gaml + 1.) * pst); 1871c4762a1bSJed Brown pst = *pstar / *pr; 1872c4762a1bSJed Brown ustarr = *uxr + (pst - 1.) * cr * PetscSqrtScalar(2. / (*gamr * (*gamr - 1. + (*gamr + 1.) * pst))); 1873c4762a1bSJed Brown zr = *pr / cr * PetscSqrtScalar(*gamr * 2. * (*gamr - 1. + (*gamr + 1.) * pst)) * (*gamr - 1. + (*gamr + 1.) * pst) / (*gamr * 3. - 1. + (*gamr + 1.) * pst); 1874c4762a1bSJed Brown dpstar = zl * zr * (ustarr - ustarl) / (zl + zr); 1875c4762a1bSJed Brown *pstar -= dpstar; 1876c4762a1bSJed Brown *pstar = PetscMax(PetscRealPart(*pstar), PetscRealPart(smallp)); 1877c4762a1bSJed Brown if (PetscAbsScalar(dpstar) / PetscRealPart(*pstar) <= 1e-8) { 1878c4762a1bSJed Brown #if 0 1879c4762a1bSJed Brown break; 1880c4762a1bSJed Brown #endif 1881c4762a1bSJed Brown } 1882c4762a1bSJed Brown } 1883c4762a1bSJed Brown /* 1-wave: rarefaction wave, 3-wave: shock */ 1884c4762a1bSJed Brown } else if (iwave == 400) { 1885c4762a1bSJed Brown /* case (400) */ 1886c4762a1bSJed Brown i__1 = iterno; 1887c4762a1bSJed Brown for (i0 = 1; i0 <= i__1; ++i0) { 1888c4762a1bSJed Brown d__1 = *pstar / *pl; 1889c4762a1bSJed Brown d__2 = 1. / *gaml; 1890c4762a1bSJed Brown *rstarl = *rl * PetscPowScalar(d__1, d__2); 1891c4762a1bSJed Brown cstarl = PetscSqrtScalar(*gaml * *pstar / *rstarl); 1892c4762a1bSJed Brown ustarl = *uxl - gascl4 * 2. * (cstarl - cl); 1893c4762a1bSJed Brown zl = *rstarl * cstarl; 1894c4762a1bSJed Brown pst = *pstar / *pr; 1895c4762a1bSJed Brown ustarr = *uxr + (pst - 1.) * cr * PetscSqrtScalar(2. / (*gamr * (*gamr - 1. + (*gamr + 1.) * pst))); 1896c4762a1bSJed Brown zr = *pr / cr * PetscSqrtScalar(*gamr * 2. * (*gamr - 1. + (*gamr + 1.) * pst)) * (*gamr - 1. + (*gamr + 1.) * pst) / (*gamr * 3. - 1. + (*gamr + 1.) * pst); 1897c4762a1bSJed Brown dpstar = zl * zr * (ustarr - ustarl) / (zl + zr); 1898c4762a1bSJed Brown *pstar -= dpstar; 1899c4762a1bSJed Brown *pstar = PetscMax(PetscRealPart(*pstar), PetscRealPart(smallp)); 1900c4762a1bSJed Brown if (PetscAbsScalar(dpstar) / PetscRealPart(*pstar) <= 1e-8) { 1901c4762a1bSJed Brown #if 0 1902c4762a1bSJed Brown break; 1903c4762a1bSJed Brown #endif 1904c4762a1bSJed Brown } 1905c4762a1bSJed Brown } 1906c4762a1bSJed Brown } 1907c4762a1bSJed Brown 1908c4762a1bSJed Brown *ustar = (zl * ustarr + zr * ustarl) / (zl + zr); 1909c4762a1bSJed Brown if (PetscRealPart(*pstar) > PetscRealPart(*pl)) { 1910c4762a1bSJed Brown pst = *pstar / *pl; 19119371c9d4SSatish Balay *rstarl = ((*gaml + 1.) * pst + *gaml - 1.) / ((*gaml - 1.) * pst + *gaml + 1.) * *rl; 1912c4762a1bSJed Brown } 1913c4762a1bSJed Brown if (PetscRealPart(*pstar) > PetscRealPart(*pr)) { 1914c4762a1bSJed Brown pst = *pstar / *pr; 19159371c9d4SSatish Balay *rstarr = ((*gamr + 1.) * pst + *gamr - 1.) / ((*gamr - 1.) * pst + *gamr + 1.) * *rr; 1916c4762a1bSJed Brown } 1917c4762a1bSJed Brown return iwave; 1918c4762a1bSJed Brown } 1919c4762a1bSJed Brown 19209371c9d4SSatish Balay PetscScalar sign(PetscScalar x) { 1921c4762a1bSJed Brown if (PetscRealPart(x) > 0) return 1.0; 1922c4762a1bSJed Brown if (PetscRealPart(x) < 0) return -1.0; 1923c4762a1bSJed Brown return 0.0; 1924c4762a1bSJed Brown } 1925c4762a1bSJed Brown /* Riemann Solver */ 1926c4762a1bSJed Brown /* -------------------------------------------------------------------- */ 19279371c9d4SSatish Balay int riemannsolver(PetscScalar *xcen, PetscScalar *xp, PetscScalar *dtt, PetscScalar *rl, PetscScalar *uxl, PetscScalar *pl, PetscScalar *utl, PetscScalar *ubl, PetscScalar *gaml, PetscScalar *rho1l, PetscScalar *rr, PetscScalar *uxr, PetscScalar *pr, PetscScalar *utr, PetscScalar *ubr, PetscScalar *gamr, PetscScalar *rho1r, PetscScalar *rx, PetscScalar *uxm, PetscScalar *px, PetscScalar *utx, PetscScalar *ubx, PetscScalar *gam, PetscScalar *rho1) { 1928c4762a1bSJed Brown /* System generated locals */ 1929c4762a1bSJed Brown PetscScalar d__1, d__2; 1930c4762a1bSJed Brown 1931c4762a1bSJed Brown /* Local variables */ 1932c4762a1bSJed Brown static PetscScalar s, c0, p0, r0, u0, w0, x0, x2, ri, cx, sgn0, wsp0, gasc1, gasc2, gasc3, gasc4; 1933c4762a1bSJed Brown static PetscScalar cstar, pstar, rstar, ustar, xstar, wspst, ushock, streng, rstarl, rstarr, rstars; 1934c4762a1bSJed Brown int iwave; 1935c4762a1bSJed Brown 1936c4762a1bSJed Brown if (*rl == *rr && *pr == *pl && *uxl == *uxr && *gaml == *gamr) { 1937c4762a1bSJed Brown *rx = *rl; 1938c4762a1bSJed Brown *px = *pl; 1939c4762a1bSJed Brown *uxm = *uxl; 1940c4762a1bSJed Brown *gam = *gaml; 1941c4762a1bSJed Brown x2 = *xcen + *uxm * *dtt; 1942c4762a1bSJed Brown 1943c4762a1bSJed Brown if (PetscRealPart(*xp) >= PetscRealPart(x2)) { 1944c4762a1bSJed Brown *utx = *utr; 1945c4762a1bSJed Brown *ubx = *ubr; 1946c4762a1bSJed Brown *rho1 = *rho1r; 1947c4762a1bSJed Brown } else { 1948c4762a1bSJed Brown *utx = *utl; 1949c4762a1bSJed Brown *ubx = *ubl; 1950c4762a1bSJed Brown *rho1 = *rho1l; 1951c4762a1bSJed Brown } 1952c4762a1bSJed Brown return 0; 1953c4762a1bSJed Brown } 1954c4762a1bSJed Brown iwave = riem1mdt(gaml, gamr, rl, pl, uxl, rr, pr, uxr, &rstarl, &rstarr, &pstar, &ustar); 1955c4762a1bSJed Brown 1956c4762a1bSJed Brown x2 = *xcen + ustar * *dtt; 1957c4762a1bSJed Brown d__1 = *xp - x2; 1958c4762a1bSJed Brown sgn0 = sign(d__1); 1959c4762a1bSJed Brown /* x is in 3-wave if sgn0 = 1 */ 1960c4762a1bSJed Brown /* x is in 1-wave if sgn0 = -1 */ 1961c4762a1bSJed Brown r0 = cvmgm_(rl, rr, &sgn0); 1962c4762a1bSJed Brown p0 = cvmgm_(pl, pr, &sgn0); 1963c4762a1bSJed Brown u0 = cvmgm_(uxl, uxr, &sgn0); 1964c4762a1bSJed Brown *gam = cvmgm_(gaml, gamr, &sgn0); 1965c4762a1bSJed Brown gasc1 = *gam - 1.; 1966c4762a1bSJed Brown gasc2 = (*gam + 1.) * .5; 1967c4762a1bSJed Brown gasc3 = gasc2 / *gam; 1968c4762a1bSJed Brown gasc4 = 1. / (*gam - 1.); 1969c4762a1bSJed Brown c0 = PetscSqrtScalar(*gam * p0 / r0); 1970c4762a1bSJed Brown streng = pstar - p0; 1971c4762a1bSJed Brown w0 = *gam * r0 * p0 * (gasc3 * streng / p0 + 1.); 1972c4762a1bSJed Brown rstars = r0 / (1. - r0 * streng / w0); 1973c4762a1bSJed Brown d__1 = p0 / pstar; 1974c4762a1bSJed Brown d__2 = -1. / *gam; 1975c4762a1bSJed Brown rstarr = r0 * PetscPowScalar(d__1, d__2); 1976c4762a1bSJed Brown rstar = cvmgm_(&rstarr, &rstars, &streng); 1977c4762a1bSJed Brown w0 = PetscSqrtScalar(w0); 1978c4762a1bSJed Brown cstar = PetscSqrtScalar(*gam * pstar / rstar); 1979c4762a1bSJed Brown wsp0 = u0 + sgn0 * c0; 1980c4762a1bSJed Brown wspst = ustar + sgn0 * cstar; 1981c4762a1bSJed Brown ushock = ustar + sgn0 * w0 / rstar; 1982c4762a1bSJed Brown wspst = cvmgp_(&ushock, &wspst, &streng); 1983c4762a1bSJed Brown wsp0 = cvmgp_(&ushock, &wsp0, &streng); 1984c4762a1bSJed Brown x0 = *xcen + wsp0 * *dtt; 1985c4762a1bSJed Brown xstar = *xcen + wspst * *dtt; 1986c4762a1bSJed Brown /* using gas formula to evaluate rarefaction wave */ 1987c4762a1bSJed Brown /* ri : reiman invariant */ 1988c4762a1bSJed Brown ri = u0 - sgn0 * 2. * gasc4 * c0; 1989c4762a1bSJed Brown cx = sgn0 * .5 * gasc1 / gasc2 * ((*xp - *xcen) / *dtt - ri); 1990c4762a1bSJed Brown *uxm = ri + sgn0 * 2. * gasc4 * cx; 1991c4762a1bSJed Brown s = p0 / PetscPowScalar(r0, *gam); 1992c4762a1bSJed Brown d__1 = cx * cx / (*gam * s); 1993c4762a1bSJed Brown *rx = PetscPowScalar(d__1, gasc4); 1994c4762a1bSJed Brown *px = cx * cx * *rx / *gam; 1995c4762a1bSJed Brown d__1 = sgn0 * (x0 - *xp); 1996c4762a1bSJed Brown *rx = cvmgp_(rx, &r0, &d__1); 1997c4762a1bSJed Brown d__1 = sgn0 * (x0 - *xp); 1998c4762a1bSJed Brown *px = cvmgp_(px, &p0, &d__1); 1999c4762a1bSJed Brown d__1 = sgn0 * (x0 - *xp); 2000c4762a1bSJed Brown *uxm = cvmgp_(uxm, &u0, &d__1); 2001c4762a1bSJed Brown d__1 = sgn0 * (xstar - *xp); 2002c4762a1bSJed Brown *rx = cvmgm_(rx, &rstar, &d__1); 2003c4762a1bSJed Brown d__1 = sgn0 * (xstar - *xp); 2004c4762a1bSJed Brown *px = cvmgm_(px, &pstar, &d__1); 2005c4762a1bSJed Brown d__1 = sgn0 * (xstar - *xp); 2006c4762a1bSJed Brown *uxm = cvmgm_(uxm, &ustar, &d__1); 2007c4762a1bSJed Brown if (PetscRealPart(*xp) >= PetscRealPart(x2)) { 2008c4762a1bSJed Brown *utx = *utr; 2009c4762a1bSJed Brown *ubx = *ubr; 2010c4762a1bSJed Brown *rho1 = *rho1r; 2011c4762a1bSJed Brown } else { 2012c4762a1bSJed Brown *utx = *utl; 2013c4762a1bSJed Brown *ubx = *ubl; 2014c4762a1bSJed Brown *rho1 = *rho1l; 2015c4762a1bSJed Brown } 2016c4762a1bSJed Brown return iwave; 2017c4762a1bSJed Brown } 20189371c9d4SSatish Balay int godunovflux(const PetscScalar *ul, const PetscScalar *ur, PetscScalar *flux, const PetscReal *nn, const int *ndim, const PetscReal *gamma) { 2019c4762a1bSJed Brown /* System generated locals */ 2020c4762a1bSJed Brown int i__1, iwave; 2021c4762a1bSJed Brown PetscScalar d__1, d__2, d__3; 2022c4762a1bSJed Brown 2023c4762a1bSJed Brown /* Local variables */ 2024c4762a1bSJed Brown static int k; 20259371c9d4SSatish Balay static PetscScalar bn[3], fn, ft, tg[3], pl, rl, pm, pr, rr, xp, ubl, ubm, ubr, dtt, unm, tmp, utl, utm, uxl, utr, uxr, gaml, gamm, gamr, xcen, rhom, rho1l, rho1m, rho1r; 2026c4762a1bSJed Brown 2027c4762a1bSJed Brown /* Function Body */ 2028c4762a1bSJed Brown xcen = 0.; 2029c4762a1bSJed Brown xp = 0.; 2030c4762a1bSJed Brown i__1 = *ndim; 2031c4762a1bSJed Brown for (k = 1; k <= i__1; ++k) { 2032c4762a1bSJed Brown tg[k - 1] = 0.; 2033c4762a1bSJed Brown bn[k - 1] = 0.; 2034c4762a1bSJed Brown } 2035c4762a1bSJed Brown dtt = 1.; 2036c4762a1bSJed Brown if (*ndim == 3) { 2037a795f3e7SBarry Smith if (nn[0] == 0. && nn[1] == 0.) { 2038c4762a1bSJed Brown tg[0] = 1.; 2039c4762a1bSJed Brown } else { 2040a795f3e7SBarry Smith tg[0] = -nn[1]; 2041a795f3e7SBarry Smith tg[1] = nn[0]; 2042c4762a1bSJed Brown } 2043c4762a1bSJed Brown /* tmp=dsqrt(tg(1)**2+tg(2)**2) */ 2044c4762a1bSJed Brown /* tg=tg/tmp */ 2045a795f3e7SBarry Smith bn[0] = -nn[2] * tg[1]; 2046a795f3e7SBarry Smith bn[1] = nn[2] * tg[0]; 2047a795f3e7SBarry Smith bn[2] = nn[0] * tg[1] - nn[1] * tg[0]; 2048c4762a1bSJed Brown /* Computing 2nd power */ 2049c4762a1bSJed Brown d__1 = bn[0]; 2050c4762a1bSJed Brown /* Computing 2nd power */ 2051c4762a1bSJed Brown d__2 = bn[1]; 2052c4762a1bSJed Brown /* Computing 2nd power */ 2053c4762a1bSJed Brown d__3 = bn[2]; 2054c4762a1bSJed Brown tmp = PetscSqrtScalar(d__1 * d__1 + d__2 * d__2 + d__3 * d__3); 2055c4762a1bSJed Brown i__1 = *ndim; 20569371c9d4SSatish Balay for (k = 1; k <= i__1; ++k) { bn[k - 1] /= tmp; } 2057c4762a1bSJed Brown } else if (*ndim == 2) { 2058a795f3e7SBarry Smith tg[0] = -nn[1]; 2059a795f3e7SBarry Smith tg[1] = nn[0]; 2060c4762a1bSJed Brown /* tmp=dsqrt(tg(1)**2+tg(2)**2) */ 2061c4762a1bSJed Brown /* tg=tg/tmp */ 2062c4762a1bSJed Brown bn[0] = 0.; 2063c4762a1bSJed Brown bn[1] = 0.; 2064c4762a1bSJed Brown bn[2] = 1.; 2065c4762a1bSJed Brown } 2066a795f3e7SBarry Smith rl = ul[0]; 2067a795f3e7SBarry Smith rr = ur[0]; 2068c4762a1bSJed Brown uxl = 0.; 2069c4762a1bSJed Brown uxr = 0.; 2070c4762a1bSJed Brown utl = 0.; 2071c4762a1bSJed Brown utr = 0.; 2072c4762a1bSJed Brown ubl = 0.; 2073c4762a1bSJed Brown ubr = 0.; 2074c4762a1bSJed Brown i__1 = *ndim; 2075c4762a1bSJed Brown for (k = 1; k <= i__1; ++k) { 2076a795f3e7SBarry Smith uxl += ul[k] * nn[k - 1]; 2077a795f3e7SBarry Smith uxr += ur[k] * nn[k - 1]; 2078a795f3e7SBarry Smith utl += ul[k] * tg[k - 1]; 2079a795f3e7SBarry Smith utr += ur[k] * tg[k - 1]; 2080a795f3e7SBarry Smith ubl += ul[k] * bn[k - 1]; 2081a795f3e7SBarry Smith ubr += ur[k] * bn[k - 1]; 2082c4762a1bSJed Brown } 2083c4762a1bSJed Brown uxl /= rl; 2084c4762a1bSJed Brown uxr /= rr; 2085c4762a1bSJed Brown utl /= rl; 2086c4762a1bSJed Brown utr /= rr; 2087c4762a1bSJed Brown ubl /= rl; 2088c4762a1bSJed Brown ubr /= rr; 2089c4762a1bSJed Brown 2090c4762a1bSJed Brown gaml = *gamma; 2091c4762a1bSJed Brown gamr = *gamma; 2092c4762a1bSJed Brown /* Computing 2nd power */ 2093c4762a1bSJed Brown d__1 = uxl; 2094c4762a1bSJed Brown /* Computing 2nd power */ 2095c4762a1bSJed Brown d__2 = utl; 2096c4762a1bSJed Brown /* Computing 2nd power */ 2097c4762a1bSJed Brown d__3 = ubl; 2098a795f3e7SBarry Smith pl = (*gamma - 1.) * (ul[*ndim + 1] - rl * .5 * (d__1 * d__1 + d__2 * d__2 + d__3 * d__3)); 2099c4762a1bSJed Brown /* Computing 2nd power */ 2100c4762a1bSJed Brown d__1 = uxr; 2101c4762a1bSJed Brown /* Computing 2nd power */ 2102c4762a1bSJed Brown d__2 = utr; 2103c4762a1bSJed Brown /* Computing 2nd power */ 2104c4762a1bSJed Brown d__3 = ubr; 2105a795f3e7SBarry Smith pr = (*gamma - 1.) * (ur[*ndim + 1] - rr * .5 * (d__1 * d__1 + d__2 * d__2 + d__3 * d__3)); 2106c4762a1bSJed Brown rho1l = rl; 2107c4762a1bSJed Brown rho1r = rr; 2108c4762a1bSJed Brown 21099371c9d4SSatish Balay iwave = riemannsolver(&xcen, &xp, &dtt, &rl, &uxl, &pl, &utl, &ubl, &gaml, &rho1l, &rr, &uxr, &pr, &utr, &ubr, &gamr, &rho1r, &rhom, &unm, &pm, &utm, &ubm, &gamm, &rho1m); 2110c4762a1bSJed Brown 2111a795f3e7SBarry Smith flux[0] = rhom * unm; 2112c4762a1bSJed Brown fn = rhom * unm * unm + pm; 2113c4762a1bSJed Brown ft = rhom * unm * utm; 2114c4762a1bSJed Brown /* flux(2)=fn*nn(1)+ft*nn(2) */ 2115c4762a1bSJed Brown /* flux(3)=fn*tg(1)+ft*tg(2) */ 2116a795f3e7SBarry Smith flux[1] = fn * nn[0] + ft * tg[0]; 2117a795f3e7SBarry Smith flux[2] = fn * nn[1] + ft * tg[1]; 2118c4762a1bSJed Brown /* flux(2)=rhom*unm*(unm)+pm */ 2119c4762a1bSJed Brown /* flux(3)=rhom*(unm)*utm */ 21209371c9d4SSatish Balay if (*ndim == 3) { flux[3] = rhom * unm * ubm; } 2121a795f3e7SBarry Smith flux[*ndim + 1] = (rhom * .5 * (unm * unm + utm * utm + ubm * ubm) + gamm / (gamm - 1.) * pm) * unm; 2122c4762a1bSJed Brown return iwave; 2123c4762a1bSJed Brown } /* godunovflux_ */ 2124c4762a1bSJed Brown 2125c4762a1bSJed Brown /* Subroutine to set up the initial conditions for the */ 2126c4762a1bSJed Brown /* Shock Interface interaction or linear wave (Ravi Samtaney,Mark Adams). */ 2127c4762a1bSJed Brown /* ----------------------------------------------------------------------- */ 21289371c9d4SSatish Balay int projecteqstate(PetscReal wc[], const PetscReal ueq[], PetscReal lv[][3]) { 2129c4762a1bSJed Brown int j, k; 2130c4762a1bSJed Brown /* Wc=matmul(lv,Ueq) 3 vars */ 2131c4762a1bSJed Brown for (k = 0; k < 3; ++k) { 2132c4762a1bSJed Brown wc[k] = 0.; 21339371c9d4SSatish Balay for (j = 0; j < 3; ++j) { wc[k] += lv[k][j] * ueq[j]; } 2134c4762a1bSJed Brown } 2135c4762a1bSJed Brown return 0; 2136c4762a1bSJed Brown } 2137c4762a1bSJed Brown /* ----------------------------------------------------------------------- */ 21389371c9d4SSatish Balay int projecttoprim(PetscReal v[], const PetscReal wc[], PetscReal rv[][3]) { 2139c4762a1bSJed Brown int k, j; 2140c4762a1bSJed Brown /* V=matmul(rv,WC) 3 vars */ 2141c4762a1bSJed Brown for (k = 0; k < 3; ++k) { 2142c4762a1bSJed Brown v[k] = 0.; 21439371c9d4SSatish Balay for (j = 0; j < 3; ++j) { v[k] += rv[k][j] * wc[j]; } 2144c4762a1bSJed Brown } 2145c4762a1bSJed Brown return 0; 2146c4762a1bSJed Brown } 2147c4762a1bSJed Brown /* ---------------------------------------------------------------------- */ 21489371c9d4SSatish Balay int eigenvectors(PetscReal rv[][3], PetscReal lv[][3], const PetscReal ueq[], PetscReal gamma) { 2149c4762a1bSJed Brown int j, k; 2150c4762a1bSJed Brown PetscReal rho, csnd, p0; 2151c4762a1bSJed Brown /* PetscScalar u; */ 2152c4762a1bSJed Brown 21539371c9d4SSatish Balay for (k = 0; k < 3; ++k) 21549371c9d4SSatish Balay for (j = 0; j < 3; ++j) { 21559371c9d4SSatish Balay lv[k][j] = 0.; 21569371c9d4SSatish Balay rv[k][j] = 0.; 21579371c9d4SSatish Balay } 2158c4762a1bSJed Brown rho = ueq[0]; 2159c4762a1bSJed Brown /* u = ueq[1]; */ 2160c4762a1bSJed Brown p0 = ueq[2]; 2161c4762a1bSJed Brown csnd = PetscSqrtReal(gamma * p0 / rho); 2162c4762a1bSJed Brown lv[0][1] = rho * .5; 2163c4762a1bSJed Brown lv[0][2] = -.5 / csnd; 2164c4762a1bSJed Brown lv[1][0] = csnd; 2165c4762a1bSJed Brown lv[1][2] = -1. / csnd; 2166c4762a1bSJed Brown lv[2][1] = rho * .5; 2167c4762a1bSJed Brown lv[2][2] = .5 / csnd; 2168c4762a1bSJed Brown rv[0][0] = -1. / csnd; 2169c4762a1bSJed Brown rv[1][0] = 1. / rho; 2170c4762a1bSJed Brown rv[2][0] = -csnd; 2171c4762a1bSJed Brown rv[0][1] = 1. / csnd; 2172c4762a1bSJed Brown rv[0][2] = 1. / csnd; 2173c4762a1bSJed Brown rv[1][2] = 1. / rho; 2174c4762a1bSJed Brown rv[2][2] = csnd; 2175c4762a1bSJed Brown return 0; 2176c4762a1bSJed Brown } 2177c4762a1bSJed Brown 21789371c9d4SSatish Balay int initLinearWave(EulerNode *ux, const PetscReal gamma, const PetscReal coord[], const PetscReal Lx) { 2179c4762a1bSJed Brown PetscReal p0, u0, wcp[3], wc[3]; 2180c4762a1bSJed Brown PetscReal lv[3][3]; 2181c4762a1bSJed Brown PetscReal vp[3]; 2182c4762a1bSJed Brown PetscReal rv[3][3]; 2183c4762a1bSJed Brown PetscReal eps, ueq[3], rho0, twopi; 2184c4762a1bSJed Brown 2185c4762a1bSJed Brown /* Function Body */ 2186c4762a1bSJed Brown twopi = 2. * PETSC_PI; 2187c4762a1bSJed Brown eps = 1e-4; /* perturbation */ 2188c4762a1bSJed Brown rho0 = 1e3; /* density of water */ 2189c4762a1bSJed Brown p0 = 101325.; /* init pressure of 1 atm (?) */ 2190c4762a1bSJed Brown u0 = 0.; 2191c4762a1bSJed Brown ueq[0] = rho0; 2192c4762a1bSJed Brown ueq[1] = u0; 2193c4762a1bSJed Brown ueq[2] = p0; 2194c4762a1bSJed Brown /* Project initial state to characteristic variables */ 2195c4762a1bSJed Brown eigenvectors(rv, lv, ueq, gamma); 2196c4762a1bSJed Brown projecteqstate(wc, ueq, lv); 2197c4762a1bSJed Brown wcp[0] = wc[0]; 2198c4762a1bSJed Brown wcp[1] = wc[1]; 2199c4762a1bSJed Brown wcp[2] = wc[2] + eps * PetscCosReal(coord[0] * 2. * twopi / Lx); 2200c4762a1bSJed Brown projecttoprim(vp, wcp, rv); 2201c4762a1bSJed Brown ux->r = vp[0]; /* density */ 2202c4762a1bSJed Brown ux->ru[0] = vp[0] * vp[1]; /* x momentum */ 2203c4762a1bSJed Brown ux->ru[1] = 0.; 2204c4762a1bSJed Brown #if defined DIM > 2 2205c4762a1bSJed Brown if (dim > 2) ux->ru[2] = 0.; 2206c4762a1bSJed Brown #endif 2207c4762a1bSJed Brown /* E = rho * e + rho * v^2/2 = p/(gam-1) + rho*v^2/2 */ 2208c4762a1bSJed Brown ux->E = vp[2] / (gamma - 1.) + 0.5 * vp[0] * vp[1] * vp[1]; 2209c4762a1bSJed Brown return 0; 2210c4762a1bSJed Brown } 2211c4762a1bSJed Brown 2212c4762a1bSJed Brown /*TEST 2213c4762a1bSJed Brown 221430602db0SMatthew G. Knepley testset: 221530602db0SMatthew G. Knepley args: -dm_plex_adj_cone -dm_plex_adj_closure 0 221630602db0SMatthew G. Knepley 221730602db0SMatthew G. Knepley test: 221830602db0SMatthew G. Knepley suffix: adv_2d_tri_0 221930602db0SMatthew G. Knepley requires: triangle 222030602db0SMatthew G. Knepley TODO: how did this ever get in main when there is no support for this 222130602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -simplex -dm_refine 3 -dm_plex_faces 1,1 -dm_plex_separate_marker -bc_inflow 1,2,4 -bc_outflow 3 222230602db0SMatthew G. Knepley 222330602db0SMatthew G. Knepley test: 222430602db0SMatthew G. Knepley suffix: adv_2d_tri_1 222530602db0SMatthew G. Knepley requires: triangle 222630602db0SMatthew G. Knepley TODO: how did this ever get in main when there is no support for this 222730602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -simplex -dm_refine 5 -dm_plex_faces 1,1 -dm_plex_separate_marker -grid_bounds -0.5,0.5,-0.5,0.5 -bc_inflow 1,2,4 -bc_outflow 3 -advect_sol_type bump -advect_bump_center 0.25,0 -advect_bump_radius 0.1 222830602db0SMatthew G. Knepley 222930602db0SMatthew G. Knepley test: 223030602db0SMatthew G. Knepley suffix: tut_1 223130602db0SMatthew G. Knepley requires: exodusii 223230602db0SMatthew G. Knepley nsize: 1 223330602db0SMatthew G. Knepley args: -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside.exo 223430602db0SMatthew G. Knepley 223530602db0SMatthew G. Knepley test: 223630602db0SMatthew G. Knepley suffix: tut_2 223730602db0SMatthew G. Knepley requires: exodusii 223830602db0SMatthew G. Knepley nsize: 1 223930602db0SMatthew G. Knepley args: -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside.exo -ts_type rosw 224030602db0SMatthew G. Knepley 224130602db0SMatthew G. Knepley test: 224230602db0SMatthew G. Knepley suffix: tut_3 224330602db0SMatthew G. Knepley requires: exodusii 224430602db0SMatthew G. Knepley nsize: 4 2245e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/annulus-20.exo -monitor Error -advect_sol_type bump -petscfv_type leastsquares -petsclimiter_type sin 224630602db0SMatthew G. Knepley 224730602db0SMatthew G. Knepley test: 224830602db0SMatthew G. Knepley suffix: tut_4 224930602db0SMatthew G. Knepley requires: exodusii 225030602db0SMatthew G. Knepley nsize: 4 2251e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/annulus-20.exo -physics sw -monitor Height,Energy -petscfv_type leastsquares -petsclimiter_type minmod 225230602db0SMatthew G. Knepley 225330602db0SMatthew G. Knepley testset: 225430602db0SMatthew G. Knepley args: -dm_plex_adj_cone -dm_plex_adj_closure 0 -dm_plex_simplex 0 -dm_plex_box_faces 1,1,1 225530602db0SMatthew G. Knepley 2256c4762a1bSJed Brown # 2D Advection 0-10 2257c4762a1bSJed Brown test: 2258c4762a1bSJed Brown suffix: 0 2259c4762a1bSJed Brown requires: exodusii 226030602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside.exo 2261c4762a1bSJed Brown 2262c4762a1bSJed Brown test: 2263c4762a1bSJed Brown suffix: 1 2264c4762a1bSJed Brown requires: exodusii 226530602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad-15.exo 2266c4762a1bSJed Brown 2267c4762a1bSJed Brown test: 2268c4762a1bSJed Brown suffix: 2 2269c4762a1bSJed Brown requires: exodusii 2270c4762a1bSJed Brown nsize: 2 2271e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside.exo 2272c4762a1bSJed Brown 2273c4762a1bSJed Brown test: 2274c4762a1bSJed Brown suffix: 3 2275c4762a1bSJed Brown requires: exodusii 2276c4762a1bSJed Brown nsize: 2 2277e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad-15.exo 2278c4762a1bSJed Brown 2279c4762a1bSJed Brown test: 2280c4762a1bSJed Brown suffix: 4 2281c4762a1bSJed Brown requires: exodusii 2282c4762a1bSJed Brown nsize: 8 2283e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad.exo 2284c4762a1bSJed Brown 2285c4762a1bSJed Brown test: 2286c4762a1bSJed Brown suffix: 5 2287c4762a1bSJed Brown requires: exodusii 228830602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside.exo -ts_type rosw -ts_adapt_reject_safety 1 2289c4762a1bSJed Brown 2290c4762a1bSJed Brown test: 2291c4762a1bSJed Brown suffix: 7 2292c4762a1bSJed Brown requires: exodusii 229330602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad-15.exo -dm_refine 1 2294c4762a1bSJed Brown 2295c4762a1bSJed Brown test: 2296c4762a1bSJed Brown suffix: 8 2297c4762a1bSJed Brown requires: exodusii 2298c4762a1bSJed Brown nsize: 2 2299e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad-15.exo -dm_refine 1 2300c4762a1bSJed Brown 2301c4762a1bSJed Brown test: 2302c4762a1bSJed Brown suffix: 9 2303c4762a1bSJed Brown requires: exodusii 2304c4762a1bSJed Brown nsize: 8 2305e600fa54SMatthew G. Knepley args: -dm_distribute_overlap 1 -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad-15.exo -dm_refine 1 2306c4762a1bSJed Brown 2307c4762a1bSJed Brown test: 2308c4762a1bSJed Brown suffix: 10 2309c4762a1bSJed Brown requires: exodusii 231030602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/sevenside-quad.exo 2311c4762a1bSJed Brown 2312c4762a1bSJed Brown # 2D Shallow water 2313c4762a1bSJed Brown test: 2314c4762a1bSJed Brown suffix: sw_0 2315c4762a1bSJed Brown requires: exodusii 231630602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/annulus-20.exo -bc_wall 100,101 -physics sw -ufv_cfl 5 -petscfv_type leastsquares -petsclimiter_type sin -ts_max_time 1 -ts_ssp_type rks2 -ts_ssp_nstages 10 -monitor height,energy 2317c4762a1bSJed Brown 23188d2c18caSMukkund Sunjii test: 23198d2c18caSMukkund Sunjii suffix: sw_hll 232030602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -bc_wall 1,2,3,4 -physics sw -ufv_cfl 3 -petscfv_type leastsquares -petsclimiter_type sin -ts_max_steps 5 -ts_ssp_type rks2 -ts_ssp_nstages 10 -monitor height,energy -grid_bounds 0,5,0,5 -dm_plex_box_faces 25,25 -sw_riemann hll 23218d2c18caSMukkund Sunjii 2322c4762a1bSJed Brown # 2D Advection: p4est 2323c4762a1bSJed Brown test: 2324c4762a1bSJed Brown suffix: p4est_advec_2d 2325c4762a1bSJed Brown requires: p4est 232630602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_type p4est -dm_forest_minimum_refinement 1 -dm_forest_initial_refinement 2 -dm_p4est_refine_pattern hash -dm_forest_maximum_refinement 5 2327c4762a1bSJed Brown 2328c4762a1bSJed Brown # Advection in a box 2329c4762a1bSJed Brown test: 2330c4762a1bSJed Brown suffix: adv_2d_quad_0 2331c4762a1bSJed Brown args: -ufv_vtk_interval 0 -dm_refine 3 -dm_plex_separate_marker -bc_inflow 1,2,4 -bc_outflow 3 2332c4762a1bSJed Brown 2333c4762a1bSJed Brown test: 2334c4762a1bSJed Brown suffix: adv_2d_quad_1 2335c4762a1bSJed Brown args: -ufv_vtk_interval 0 -dm_refine 3 -dm_plex_separate_marker -grid_bounds -0.5,0.5,-0.5,0.5 -bc_inflow 1,2,4 -bc_outflow 3 -advect_sol_type bump -advect_bump_center 0.25,0 -advect_bump_radius 0.1 2336c4762a1bSJed Brown timeoutfactor: 3 2337c4762a1bSJed Brown 2338c4762a1bSJed Brown test: 2339c4762a1bSJed Brown suffix: adv_2d_quad_p4est_0 2340c4762a1bSJed Brown requires: p4est 2341c4762a1bSJed Brown args: -ufv_vtk_interval 0 -dm_refine 5 -dm_type p4est -dm_plex_separate_marker -bc_inflow 1,2,4 -bc_outflow 3 2342c4762a1bSJed Brown 2343c4762a1bSJed Brown test: 2344c4762a1bSJed Brown suffix: adv_2d_quad_p4est_1 2345c4762a1bSJed Brown requires: p4est 2346c4762a1bSJed Brown args: -ufv_vtk_interval 0 -dm_refine 5 -dm_type p4est -dm_plex_separate_marker -grid_bounds -0.5,0.5,-0.5,0.5 -bc_inflow 1,2,4 -bc_outflow 3 -advect_sol_type bump -advect_bump_center 0.25,0 -advect_bump_radius 0.1 2347c4762a1bSJed Brown timeoutfactor: 3 2348c4762a1bSJed Brown 2349c4762a1bSJed Brown test: 2350c4762a1bSJed Brown suffix: adv_2d_quad_p4est_adapt_0 2351c4762a1bSJed Brown requires: p4est !__float128 #broken for quad precision 2352c4762a1bSJed Brown args: -ufv_vtk_interval 0 -dm_refine 3 -dm_type p4est -dm_plex_separate_marker -grid_bounds -0.5,0.5,-0.5,0.5 -bc_inflow 1,2,4 -bc_outflow 3 -advect_sol_type bump -advect_bump_center 0.25,0 -advect_bump_radius 0.1 -ufv_use_amr -refine_vec_tagger_box 0.005,inf -coarsen_vec_tagger_box 0,1.e-5 -petscfv_type leastsquares -ts_max_time 0.01 2353c4762a1bSJed Brown timeoutfactor: 3 2354c4762a1bSJed Brown 2355c4762a1bSJed Brown test: 2356c4762a1bSJed Brown suffix: adv_0 2357c4762a1bSJed Brown requires: exodusii 235830602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/blockcylinder-50.exo -bc_inflow 100,101,200 -bc_outflow 201 2359c4762a1bSJed Brown 2360c4762a1bSJed Brown test: 2361c4762a1bSJed Brown suffix: shock_0 2362c4762a1bSJed Brown requires: p4est !single !complex 236330602db0SMatthew G. Knepley args: -dm_plex_box_faces 2,1 -grid_bounds -1,1.,0.,1 -grid_skew_60 \ 236430602db0SMatthew G. Knepley -dm_type p4est -dm_forest_partition_overlap 1 -dm_forest_maximum_refinement 6 -dm_forest_minimum_refinement 2 -dm_forest_initial_refinement 2 \ 236530602db0SMatthew G. Knepley -ufv_use_amr -refine_vec_tagger_box 0.5,inf -coarsen_vec_tagger_box 0,1.e-2 -refine_tag_view -coarsen_tag_view \ 236630602db0SMatthew G. Knepley -bc_wall 1,2,3,4 -physics euler -eu_type iv_shock -ufv_cfl 10 -eu_alpha 60. -eu_gamma 1.4 -eu_amach 2.02 -eu_rho2 3. \ 236730602db0SMatthew G. Knepley -petscfv_type leastsquares -petsclimiter_type minmod -petscfv_compute_gradients 0 \ 236830602db0SMatthew G. Knepley -ts_max_time 0.5 -ts_ssp_type rks2 -ts_ssp_nstages 10 \ 236930602db0SMatthew G. Knepley -ufv_vtk_basename ${wPETSC_DIR}/ex11 -ufv_vtk_interval 0 -monitor density,energy 2370c4762a1bSJed Brown timeoutfactor: 3 2371c4762a1bSJed Brown 2372c4762a1bSJed Brown # Test GLVis visualization of PetscFV fields 2373c4762a1bSJed Brown test: 2374c4762a1bSJed Brown suffix: glvis_adv_2d_tet 237530602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -ufv_vtk_monitor 0 \ 237630602db0SMatthew G. Knepley -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/square_periodic.msh -dm_plex_gmsh_periodic 0 \ 237730602db0SMatthew G. Knepley -ts_monitor_solution glvis: -ts_max_steps 0 2378c4762a1bSJed Brown 2379c4762a1bSJed Brown test: 2380c4762a1bSJed Brown suffix: glvis_adv_2d_quad 238130602db0SMatthew G. Knepley args: -ufv_vtk_interval 0 -ufv_vtk_monitor 0 -bc_inflow 1,2,4 -bc_outflow 3 \ 238230602db0SMatthew G. Knepley -dm_refine 5 -dm_plex_separate_marker \ 238330602db0SMatthew G. Knepley -ts_monitor_solution glvis: -ts_max_steps 0 2384c4762a1bSJed Brown 2385c4762a1bSJed Brown TEST*/ 2386