1c4762a1bSJed Brown static char help[] = "Time dependent Navier-Stokes problem in 2d and 3d with finite elements.\n\ 2c4762a1bSJed Brown We solve the Navier-Stokes in a rectangular\n\ 3c4762a1bSJed Brown domain, using a parallel unstructured mesh (DMPLEX) to discretize it.\n\ 4c4762a1bSJed Brown This example supports discretized auxiliary fields (Re) as well as\n\ 5c4762a1bSJed Brown multilevel nonlinear solvers.\n\ 6c4762a1bSJed Brown Contributed by: Julian Andrej <juan@tf.uni-kiel.de>\n\n\n"; 7c4762a1bSJed Brown 8c4762a1bSJed Brown #include <petscdmplex.h> 9c4762a1bSJed Brown #include <petscsnes.h> 10c4762a1bSJed Brown #include <petscts.h> 11c4762a1bSJed Brown #include <petscds.h> 12c4762a1bSJed Brown 13c4762a1bSJed Brown /* 14c4762a1bSJed Brown Navier-Stokes equation: 15c4762a1bSJed Brown 16c4762a1bSJed Brown du/dt + u . grad u - \Delta u - grad p = f 17c4762a1bSJed Brown div u = 0 18c4762a1bSJed Brown */ 19c4762a1bSJed Brown 20c4762a1bSJed Brown typedef struct { 21c4762a1bSJed Brown PetscInt mms; 22c4762a1bSJed Brown } AppCtx; 23c4762a1bSJed Brown 24c4762a1bSJed Brown #define REYN 400.0 25c4762a1bSJed Brown 26c4762a1bSJed Brown /* MMS1 27c4762a1bSJed Brown 28c4762a1bSJed Brown u = t + x^2 + y^2; 29c4762a1bSJed Brown v = t + 2*x^2 - 2*x*y; 30c4762a1bSJed Brown p = x + y - 1; 31c4762a1bSJed Brown 32c4762a1bSJed Brown f_x = -2*t*(x + y) + 2*x*y^2 - 4*x^2*y - 2*x^3 + 4.0/Re - 1.0 33c4762a1bSJed Brown f_y = -2*t*x + 2*y^3 - 4*x*y^2 - 2*x^2*y + 4.0/Re - 1.0 34c4762a1bSJed Brown 35c4762a1bSJed Brown so that 36c4762a1bSJed Brown 37c4762a1bSJed Brown u_t + u \cdot \nabla u - 1/Re \Delta u + \nabla p + f = <1, 1> + <t (2x + 2y) + 2x^3 + 4x^2y - 2xy^2, t 2x + 2x^2y + 4xy^2 - 2y^3> - 1/Re <4, 4> + <1, 1> 38c4762a1bSJed Brown + <-t (2x + 2y) + 2xy^2 - 4x^2y - 2x^3 + 4/Re - 1, -2xt + 2y^3 - 4xy^2 - 2x^2y + 4/Re - 1> = 0 39c4762a1bSJed Brown \nabla \cdot u = 2x - 2x = 0 40c4762a1bSJed Brown 41c4762a1bSJed Brown where 42c4762a1bSJed Brown 43c4762a1bSJed Brown <u, v> . <<u_x, v_x>, <u_y, v_y>> = <u u_x + v u_y, u v_x + v v_y> 44c4762a1bSJed Brown */ 45c4762a1bSJed Brown PetscErrorCode mms1_u_2d(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx) 46c4762a1bSJed Brown { 47c4762a1bSJed Brown u[0] = time + x[0]*x[0] + x[1]*x[1]; 48c4762a1bSJed Brown u[1] = time + 2.0*x[0]*x[0] - 2.0*x[0]*x[1]; 49c4762a1bSJed Brown return 0; 50c4762a1bSJed Brown } 51c4762a1bSJed Brown 52c4762a1bSJed Brown PetscErrorCode mms1_p_2d(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *p, void *ctx) 53c4762a1bSJed Brown { 54c4762a1bSJed Brown *p = x[0] + x[1] - 1.0; 55c4762a1bSJed Brown return 0; 56c4762a1bSJed Brown } 57c4762a1bSJed Brown 58c4762a1bSJed Brown /* MMS 2*/ 59c4762a1bSJed Brown 60c4762a1bSJed Brown static PetscErrorCode mms2_u_2d(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *u, void *ctx) 61c4762a1bSJed Brown { 62c4762a1bSJed Brown u[0] = PetscSinReal(time + x[0])*PetscSinReal(time + x[1]); 63c4762a1bSJed Brown u[1] = PetscCosReal(time + x[0])*PetscCosReal(time + x[1]); 64c4762a1bSJed Brown return 0; 65c4762a1bSJed Brown } 66c4762a1bSJed Brown 67c4762a1bSJed Brown static PetscErrorCode mms2_p_2d(PetscInt dim, PetscReal time, const PetscReal x[], PetscInt Nf, PetscScalar *p, void *ctx) 68c4762a1bSJed Brown { 69c4762a1bSJed Brown *p = PetscSinReal(time + x[0] - x[1]); 70c4762a1bSJed Brown return 0; 71c4762a1bSJed Brown } 72c4762a1bSJed Brown 73c4762a1bSJed Brown static void f0_mms1_u(PetscInt dim, PetscInt Nf, PetscInt NfAux, 74c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 75c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 76c4762a1bSJed Brown PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]) 77c4762a1bSJed Brown { 78c4762a1bSJed Brown const PetscReal Re = REYN; 79c4762a1bSJed Brown const PetscInt Ncomp = dim; 80c4762a1bSJed Brown PetscInt c, d; 81c4762a1bSJed Brown 82c4762a1bSJed Brown for (c = 0; c < Ncomp; ++c) { 83c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 84c4762a1bSJed Brown f0[c] += u[d] * u_x[c*dim+d]; 85c4762a1bSJed Brown } 86c4762a1bSJed Brown } 87c4762a1bSJed Brown f0[0] += u_t[0]; 88c4762a1bSJed Brown f0[1] += u_t[1]; 89c4762a1bSJed Brown 90c4762a1bSJed Brown f0[0] += -2.0*t*(x[0] + x[1]) + 2.0*x[0]*x[1]*x[1] - 4.0*x[0]*x[0]*x[1] - 2.0*x[0]*x[0]*x[0] + 4.0/Re - 1.0; 91c4762a1bSJed Brown f0[1] += -2.0*t*x[0] + 2.0*x[1]*x[1]*x[1] - 4.0*x[0]*x[1]*x[1] - 2.0*x[0]*x[0]*x[1] + 4.0/Re - 1.0; 92c4762a1bSJed Brown } 93c4762a1bSJed Brown 94c4762a1bSJed Brown static void f0_mms2_u(PetscInt dim, PetscInt Nf, PetscInt NfAux, 95c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 96c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 97c4762a1bSJed Brown PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]) 98c4762a1bSJed Brown { 99c4762a1bSJed Brown const PetscReal Re = REYN; 100c4762a1bSJed Brown const PetscInt Ncomp = dim; 101c4762a1bSJed Brown PetscInt c, d; 102c4762a1bSJed Brown 103c4762a1bSJed Brown for (c = 0; c < Ncomp; ++c) { 104c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 105c4762a1bSJed Brown f0[c] += u[d] * u_x[c*dim+d]; 106c4762a1bSJed Brown } 107c4762a1bSJed Brown } 108c4762a1bSJed Brown f0[0] += u_t[0]; 109c4762a1bSJed Brown f0[1] += u_t[1]; 110c4762a1bSJed Brown 111c4762a1bSJed Brown f0[0] -= ( Re*((1.0L/2.0L)*PetscSinReal(2*t + 2*x[0]) + PetscSinReal(2*t + x[0] + x[1]) + PetscCosReal(t + x[0] - x[1])) + 2.0*PetscSinReal(t + x[0])*PetscSinReal(t + x[1]))/Re; 112c4762a1bSJed Brown f0[1] -= (-Re*((1.0L/2.0L)*PetscSinReal(2*t + 2*x[1]) + PetscSinReal(2*t + x[0] + x[1]) + PetscCosReal(t + x[0] - x[1])) + 2.0*PetscCosReal(t + x[0])*PetscCosReal(t + x[1]))/Re; 113c4762a1bSJed Brown } 114c4762a1bSJed Brown 115c4762a1bSJed Brown static void f1_u(PetscInt dim, PetscInt Nf, PetscInt NfAux, 116c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 117c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 118c4762a1bSJed Brown PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]) 119c4762a1bSJed Brown { 120c4762a1bSJed Brown const PetscReal Re = REYN; 121c4762a1bSJed Brown const PetscInt Ncomp = dim; 122c4762a1bSJed Brown PetscInt comp, d; 123c4762a1bSJed Brown 124c4762a1bSJed Brown for (comp = 0; comp < Ncomp; ++comp) { 125c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 126c4762a1bSJed Brown f1[comp*dim+d] = 1.0/Re * u_x[comp*dim+d]; 127c4762a1bSJed Brown } 128c4762a1bSJed Brown f1[comp*dim+comp] -= u[Ncomp]; 129c4762a1bSJed Brown } 130c4762a1bSJed Brown } 131c4762a1bSJed Brown 132c4762a1bSJed Brown static void f0_p(PetscInt dim, PetscInt Nf, PetscInt NfAux, 133c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 134c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 135c4762a1bSJed Brown PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f0[]) 136c4762a1bSJed Brown { 137c4762a1bSJed Brown PetscInt d; 138c4762a1bSJed Brown for (d = 0, f0[0] = 0.0; d < dim; ++d) f0[0] += u_x[d*dim+d]; 139c4762a1bSJed Brown } 140c4762a1bSJed Brown 141c4762a1bSJed Brown static void f1_p(PetscInt dim, PetscInt Nf, PetscInt NfAux, 142c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 143c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 144c4762a1bSJed Brown PetscReal t, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar f1[]) 145c4762a1bSJed Brown { 146c4762a1bSJed Brown PetscInt d; 147c4762a1bSJed Brown for (d = 0; d < dim; ++d) f1[d] = 0.0; 148c4762a1bSJed Brown } 149c4762a1bSJed Brown 150c4762a1bSJed Brown /* 151c4762a1bSJed Brown (psi_i, u_j grad_j u_i) ==> (\psi_i, \phi_j grad_j u_i) 152c4762a1bSJed Brown */ 153c4762a1bSJed Brown static void g0_uu(PetscInt dim, PetscInt Nf, PetscInt NfAux, 154c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 155c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 156c4762a1bSJed Brown PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g0[]) 157c4762a1bSJed Brown { 158c4762a1bSJed Brown PetscInt NcI = dim, NcJ = dim; 159c4762a1bSJed Brown PetscInt fc, gc; 160c4762a1bSJed Brown PetscInt d; 161c4762a1bSJed Brown 162c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 163c4762a1bSJed Brown g0[d*dim+d] = u_tShift; 164c4762a1bSJed Brown } 165c4762a1bSJed Brown 166c4762a1bSJed Brown for (fc = 0; fc < NcI; ++fc) { 167c4762a1bSJed Brown for (gc = 0; gc < NcJ; ++gc) { 168c4762a1bSJed Brown g0[fc*NcJ+gc] += u_x[fc*NcJ+gc]; 169c4762a1bSJed Brown } 170c4762a1bSJed Brown } 171c4762a1bSJed Brown } 172c4762a1bSJed Brown 173c4762a1bSJed Brown /* 174c4762a1bSJed Brown (psi_i, u_j grad_j u_i) ==> (\psi_i, \u_j grad_j \phi_i) 175c4762a1bSJed Brown */ 176c4762a1bSJed Brown static void g1_uu(PetscInt dim, PetscInt Nf, PetscInt NfAux, 177c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 178c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 179c4762a1bSJed Brown PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]) 180c4762a1bSJed Brown { 181c4762a1bSJed Brown PetscInt NcI = dim; 182c4762a1bSJed Brown PetscInt NcJ = dim; 183c4762a1bSJed Brown PetscInt fc, gc, dg; 184c4762a1bSJed Brown for (fc = 0; fc < NcI; ++fc) { 185c4762a1bSJed Brown for (gc = 0; gc < NcJ; ++gc) { 186c4762a1bSJed Brown for (dg = 0; dg < dim; ++dg) { 187c4762a1bSJed Brown /* kronecker delta */ 188c4762a1bSJed Brown if (fc == gc) { 189c4762a1bSJed Brown g1[(fc*NcJ+gc)*dim+dg] += u[dg]; 190c4762a1bSJed Brown } 191c4762a1bSJed Brown } 192c4762a1bSJed Brown } 193c4762a1bSJed Brown } 194c4762a1bSJed Brown } 195c4762a1bSJed Brown 196c4762a1bSJed Brown /* < q, \nabla\cdot u > 197c4762a1bSJed Brown NcompI = 1, NcompJ = dim */ 198c4762a1bSJed Brown static void g1_pu(PetscInt dim, PetscInt Nf, PetscInt NfAux, 199c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 200c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 201c4762a1bSJed Brown PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g1[]) 202c4762a1bSJed Brown { 203c4762a1bSJed Brown PetscInt d; 204c4762a1bSJed Brown for (d = 0; d < dim; ++d) g1[d*dim+d] = 1.0; /* \frac{\partial\phi^{u_d}}{\partial x_d} */ 205c4762a1bSJed Brown } 206c4762a1bSJed Brown 207c4762a1bSJed Brown /* -< \nabla\cdot v, p > 208c4762a1bSJed Brown NcompI = dim, NcompJ = 1 */ 209c4762a1bSJed Brown static void g2_up(PetscInt dim, PetscInt Nf, PetscInt NfAux, 210c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 211c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 212c4762a1bSJed Brown PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g2[]) 213c4762a1bSJed Brown { 214c4762a1bSJed Brown PetscInt d; 215c4762a1bSJed Brown for (d = 0; d < dim; ++d) g2[d*dim+d] = -1.0; /* \frac{\partial\psi^{u_d}}{\partial x_d} */ 216c4762a1bSJed Brown } 217c4762a1bSJed Brown 218c4762a1bSJed Brown /* < \nabla v, \nabla u + {\nabla u}^T > 219c4762a1bSJed Brown This just gives \nabla u, give the perdiagonal for the transpose */ 220c4762a1bSJed Brown static void g3_uu(PetscInt dim, PetscInt Nf, PetscInt NfAux, 221c4762a1bSJed Brown const PetscInt uOff[], const PetscInt uOff_x[], const PetscScalar u[], const PetscScalar u_t[], const PetscScalar u_x[], 222c4762a1bSJed Brown const PetscInt aOff[], const PetscInt aOff_x[], const PetscScalar a[], const PetscScalar a_t[], const PetscScalar a_x[], 223c4762a1bSJed Brown PetscReal t, PetscReal u_tShift, const PetscReal x[], PetscInt numConstants, const PetscScalar constants[], PetscScalar g3[]) 224c4762a1bSJed Brown { 225c4762a1bSJed Brown const PetscReal Re = REYN; 226c4762a1bSJed Brown const PetscInt Ncomp = dim; 227c4762a1bSJed Brown PetscInt compI, d; 228c4762a1bSJed Brown 229c4762a1bSJed Brown for (compI = 0; compI < Ncomp; ++compI) { 230c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 231c4762a1bSJed Brown g3[((compI*Ncomp+compI)*dim+d)*dim+d] = 1.0/Re; 232c4762a1bSJed Brown } 233c4762a1bSJed Brown } 234c4762a1bSJed Brown } 235c4762a1bSJed Brown 236c4762a1bSJed Brown static PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options) 237c4762a1bSJed Brown { 238c4762a1bSJed Brown PetscErrorCode ierr; 239c4762a1bSJed Brown 240c4762a1bSJed Brown PetscFunctionBeginUser; 241c4762a1bSJed Brown options->mms = 1; 242c4762a1bSJed Brown 243c4762a1bSJed Brown ierr = PetscOptionsBegin(comm, "", "Navier-Stokes Equation Options", "DMPLEX");CHKERRQ(ierr); 244c4762a1bSJed Brown ierr = PetscOptionsInt("-mms", "The manufactured solution to use", "ex46.c", options->mms, &options->mms, NULL);CHKERRQ(ierr); 245c4762a1bSJed Brown ierr = PetscOptionsEnd();CHKERRQ(ierr); 246c4762a1bSJed Brown PetscFunctionReturn(0); 247c4762a1bSJed Brown } 248c4762a1bSJed Brown 249c4762a1bSJed Brown static PetscErrorCode CreateMesh(MPI_Comm comm, DM *dm, AppCtx *ctx) 250c4762a1bSJed Brown { 251c4762a1bSJed Brown PetscErrorCode ierr; 252c4762a1bSJed Brown 253c4762a1bSJed Brown PetscFunctionBeginUser; 25430602db0SMatthew G. Knepley ierr = DMCreate(comm, dm);CHKERRQ(ierr); 25530602db0SMatthew G. Knepley ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr); 256c4762a1bSJed Brown ierr = DMSetFromOptions(*dm);CHKERRQ(ierr); 257c4762a1bSJed Brown ierr = DMViewFromOptions(*dm, NULL, "-dm_view");CHKERRQ(ierr); 258c4762a1bSJed Brown PetscFunctionReturn(0); 259c4762a1bSJed Brown } 260c4762a1bSJed Brown 261c4762a1bSJed Brown static PetscErrorCode SetupProblem(DM dm, AppCtx *ctx) 262c4762a1bSJed Brown { 26345480ffeSMatthew G. Knepley PetscDS ds; 26445480ffeSMatthew G. Knepley DMLabel label; 265c4762a1bSJed Brown const PetscInt id = 1; 26630602db0SMatthew G. Knepley PetscInt dim; 267c4762a1bSJed Brown PetscErrorCode ierr; 268c4762a1bSJed Brown 269c4762a1bSJed Brown PetscFunctionBeginUser; 27030602db0SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 27145480ffeSMatthew G. Knepley ierr = DMGetDS(dm, &ds);CHKERRQ(ierr); 27230602db0SMatthew G. Knepley ierr = DMGetLabel(dm, "marker", &label);CHKERRQ(ierr); 27330602db0SMatthew G. Knepley switch (dim) { 27430602db0SMatthew G. Knepley case 2: 275c4762a1bSJed Brown switch (ctx->mms) { 276c4762a1bSJed Brown case 1: 27730602db0SMatthew G. Knepley ierr = PetscDSSetResidual(ds, 0, f0_mms1_u, f1_u);CHKERRQ(ierr); 27845480ffeSMatthew G. Knepley ierr = PetscDSSetResidual(ds, 1, f0_p, f1_p);CHKERRQ(ierr); 27945480ffeSMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 0, 0, g0_uu, g1_uu, NULL, g3_uu);CHKERRQ(ierr); 28045480ffeSMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 0, 1, NULL, NULL, g2_up, NULL);CHKERRQ(ierr); 28145480ffeSMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 1, 0, NULL, g1_pu, NULL, NULL);CHKERRQ(ierr); 28230602db0SMatthew G. Knepley ierr = PetscDSSetExactSolution(ds, 0, mms1_u_2d, ctx);CHKERRQ(ierr); 28330602db0SMatthew G. Knepley ierr = PetscDSSetExactSolution(ds, 1, mms1_p_2d, ctx);CHKERRQ(ierr); 28430602db0SMatthew G. Knepley ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", label, 1, &id, 0, 0, NULL, (void (*)(void)) mms1_u_2d, NULL, ctx, NULL);CHKERRQ(ierr); 285c4762a1bSJed Brown break; 286c4762a1bSJed Brown case 2: 28730602db0SMatthew G. Knepley ierr = PetscDSSetResidual(ds, 0, f0_mms2_u, f1_u);CHKERRQ(ierr); 28830602db0SMatthew G. Knepley ierr = PetscDSSetResidual(ds, 1, f0_p, f1_p);CHKERRQ(ierr); 28930602db0SMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 0, 0, g0_uu, g1_uu, NULL, g3_uu);CHKERRQ(ierr); 29030602db0SMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 0, 1, NULL, NULL, g2_up, NULL);CHKERRQ(ierr); 29130602db0SMatthew G. Knepley ierr = PetscDSSetJacobian(ds, 1, 0, NULL, g1_pu, NULL, NULL);CHKERRQ(ierr); 29230602db0SMatthew G. Knepley ierr = PetscDSSetExactSolution(ds, 0, mms2_u_2d, ctx);CHKERRQ(ierr); 29330602db0SMatthew G. Knepley ierr = PetscDSSetExactSolution(ds, 1, mms2_p_2d, ctx);CHKERRQ(ierr); 29430602db0SMatthew G. Knepley ierr = DMAddBoundary(dm, DM_BC_ESSENTIAL, "wall", label, 1, &id, 0, 0, NULL, (void (*)(void)) mms2_u_2d, NULL, ctx, NULL);CHKERRQ(ierr); 295c4762a1bSJed Brown break; 296*98921bdaSJacob Faibussowitsch default: SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid MMS %D", ctx->mms); 297c4762a1bSJed Brown } 298c4762a1bSJed Brown break; 299*98921bdaSJacob Faibussowitsch default: SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid dimension %D", dim); 300c4762a1bSJed Brown } 301c4762a1bSJed Brown PetscFunctionReturn(0); 302c4762a1bSJed Brown } 303c4762a1bSJed Brown 304c4762a1bSJed Brown static PetscErrorCode SetupDiscretization(DM dm, AppCtx *ctx) 305c4762a1bSJed Brown { 306c4762a1bSJed Brown MPI_Comm comm; 30730602db0SMatthew G. Knepley DM cdm = dm; 30830602db0SMatthew G. Knepley PetscFE fe[2]; 30930602db0SMatthew G. Knepley PetscInt dim; 31030602db0SMatthew G. Knepley PetscBool simplex; 311c4762a1bSJed Brown PetscErrorCode ierr; 312c4762a1bSJed Brown 313c4762a1bSJed Brown PetscFunctionBeginUser; 314c4762a1bSJed Brown ierr = PetscObjectGetComm((PetscObject) dm, &comm);CHKERRQ(ierr); 31530602db0SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 31630602db0SMatthew G. Knepley ierr = DMPlexIsSimplex(dm, &simplex);CHKERRQ(ierr); 31730602db0SMatthew G. Knepley ierr = PetscFECreateDefault(comm, dim, dim, simplex, "vel_", PETSC_DEFAULT, &fe[0]);CHKERRQ(ierr); 318c4762a1bSJed Brown ierr = PetscObjectSetName((PetscObject) fe[0], "velocity");CHKERRQ(ierr); 31930602db0SMatthew G. Knepley ierr = PetscFECreateDefault(comm, dim, 1, simplex, "pres_", PETSC_DEFAULT, &fe[1]);CHKERRQ(ierr); 320c4762a1bSJed Brown ierr = PetscFECopyQuadrature(fe[0], fe[1]);CHKERRQ(ierr); 321c4762a1bSJed Brown ierr = PetscObjectSetName((PetscObject) fe[1], "pressure");CHKERRQ(ierr); 322c4762a1bSJed Brown /* Set discretization and boundary conditions for each mesh */ 323c4762a1bSJed Brown ierr = DMSetField(dm, 0, NULL, (PetscObject) fe[0]);CHKERRQ(ierr); 324c4762a1bSJed Brown ierr = DMSetField(dm, 1, NULL, (PetscObject) fe[1]);CHKERRQ(ierr); 325c4762a1bSJed Brown ierr = DMCreateDS(dm);CHKERRQ(ierr); 326c4762a1bSJed Brown ierr = SetupProblem(dm, ctx);CHKERRQ(ierr); 327c4762a1bSJed Brown while (cdm) { 328c4762a1bSJed Brown PetscObject pressure; 329c4762a1bSJed Brown MatNullSpace nsp; 330c4762a1bSJed Brown 331c4762a1bSJed Brown ierr = DMGetField(cdm, 1, NULL, &pressure);CHKERRQ(ierr); 332c4762a1bSJed Brown ierr = MatNullSpaceCreate(PetscObjectComm(pressure), PETSC_TRUE, 0, NULL, &nsp);CHKERRQ(ierr); 333c4762a1bSJed Brown ierr = PetscObjectCompose(pressure, "nullspace", (PetscObject) nsp);CHKERRQ(ierr); 334c4762a1bSJed Brown ierr = MatNullSpaceDestroy(&nsp);CHKERRQ(ierr); 335c4762a1bSJed Brown 336408cafa0SMatthew G. Knepley ierr = DMCopyDisc(dm, cdm);CHKERRQ(ierr); 337c4762a1bSJed Brown ierr = DMGetCoarseDM(cdm, &cdm);CHKERRQ(ierr); 338c4762a1bSJed Brown } 339c4762a1bSJed Brown ierr = PetscFEDestroy(&fe[0]);CHKERRQ(ierr); 340c4762a1bSJed Brown ierr = PetscFEDestroy(&fe[1]);CHKERRQ(ierr); 341c4762a1bSJed Brown PetscFunctionReturn(0); 342c4762a1bSJed Brown } 343c4762a1bSJed Brown 344c4762a1bSJed Brown static PetscErrorCode MonitorError(TS ts, PetscInt step, PetscReal crtime, Vec u, void *ctx) 345c4762a1bSJed Brown { 34630602db0SMatthew G. Knepley PetscSimplePointFunc funcs[2]; 34730602db0SMatthew G. Knepley void *ctxs[2]; 348c4762a1bSJed Brown DM dm; 34930602db0SMatthew G. Knepley PetscDS ds; 350c4762a1bSJed Brown PetscReal ferrors[2]; 351c4762a1bSJed Brown PetscErrorCode ierr; 352c4762a1bSJed Brown 353c4762a1bSJed Brown PetscFunctionBeginUser; 354c4762a1bSJed Brown ierr = TSGetDM(ts, &dm);CHKERRQ(ierr); 35530602db0SMatthew G. Knepley ierr = DMGetDS(dm, &ds);CHKERRQ(ierr); 35630602db0SMatthew G. Knepley ierr = PetscDSGetExactSolution(ds, 0, &funcs[0], &ctxs[0]);CHKERRQ(ierr); 35730602db0SMatthew G. Knepley ierr = PetscDSGetExactSolution(ds, 1, &funcs[1], &ctxs[1]);CHKERRQ(ierr); 35830602db0SMatthew G. Knepley ierr = DMComputeL2FieldDiff(dm, crtime, funcs, ctxs, u, ferrors);CHKERRQ(ierr); 359c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD, "Timestep: %04d time = %-8.4g \t L_2 Error: [%2.3g, %2.3g]\n", (int) step, (double) crtime, (double) ferrors[0], (double) ferrors[1]);CHKERRQ(ierr); 360c4762a1bSJed Brown PetscFunctionReturn(0); 361c4762a1bSJed Brown } 362c4762a1bSJed Brown 363c4762a1bSJed Brown int main(int argc, char **argv) 364c4762a1bSJed Brown { 365c4762a1bSJed Brown AppCtx ctx; 366c4762a1bSJed Brown DM dm; 367c4762a1bSJed Brown TS ts; 368c4762a1bSJed Brown Vec u, r; 369c4762a1bSJed Brown PetscErrorCode ierr; 370c4762a1bSJed Brown 371c4762a1bSJed Brown ierr = PetscInitialize(&argc, &argv, NULL, help);if (ierr) return ierr; 372c4762a1bSJed Brown ierr = ProcessOptions(PETSC_COMM_WORLD, &ctx);CHKERRQ(ierr); 373c4762a1bSJed Brown ierr = CreateMesh(PETSC_COMM_WORLD, &dm, &ctx);CHKERRQ(ierr); 374c4762a1bSJed Brown ierr = DMSetApplicationContext(dm, &ctx);CHKERRQ(ierr); 375c4762a1bSJed Brown ierr = SetupDiscretization(dm, &ctx);CHKERRQ(ierr); 376c4762a1bSJed Brown ierr = DMPlexCreateClosureIndex(dm, NULL);CHKERRQ(ierr); 377c4762a1bSJed Brown 378c4762a1bSJed Brown ierr = DMCreateGlobalVector(dm, &u);CHKERRQ(ierr); 379c4762a1bSJed Brown ierr = VecDuplicate(u, &r);CHKERRQ(ierr); 380c4762a1bSJed Brown 381c4762a1bSJed Brown ierr = TSCreate(PETSC_COMM_WORLD, &ts);CHKERRQ(ierr); 382c4762a1bSJed Brown ierr = TSMonitorSet(ts, MonitorError, &ctx, NULL);CHKERRQ(ierr); 383c4762a1bSJed Brown ierr = TSSetDM(ts, dm);CHKERRQ(ierr); 384c4762a1bSJed Brown ierr = DMTSSetBoundaryLocal(dm, DMPlexTSComputeBoundary, &ctx);CHKERRQ(ierr); 385c4762a1bSJed Brown ierr = DMTSSetIFunctionLocal(dm, DMPlexTSComputeIFunctionFEM, &ctx);CHKERRQ(ierr); 386c4762a1bSJed Brown ierr = DMTSSetIJacobianLocal(dm, DMPlexTSComputeIJacobianFEM, &ctx);CHKERRQ(ierr); 387c4762a1bSJed Brown ierr = TSSetExactFinalTime(ts, TS_EXACTFINALTIME_STEPOVER);CHKERRQ(ierr); 388c4762a1bSJed Brown ierr = TSSetFromOptions(ts);CHKERRQ(ierr); 38930602db0SMatthew G. Knepley ierr = DMTSCheckFromOptions(ts, u);CHKERRQ(ierr); 390c4762a1bSJed Brown 39130602db0SMatthew G. Knepley { 39230602db0SMatthew G. Knepley PetscSimplePointFunc funcs[2]; 39330602db0SMatthew G. Knepley void *ctxs[2]; 39430602db0SMatthew G. Knepley PetscDS ds; 39530602db0SMatthew G. Knepley 39630602db0SMatthew G. Knepley ierr = DMGetDS(dm, &ds);CHKERRQ(ierr); 39730602db0SMatthew G. Knepley ierr = PetscDSGetExactSolution(ds, 0, &funcs[0], &ctxs[0]);CHKERRQ(ierr); 39830602db0SMatthew G. Knepley ierr = PetscDSGetExactSolution(ds, 1, &funcs[1], &ctxs[1]);CHKERRQ(ierr); 39930602db0SMatthew G. Knepley ierr = DMProjectFunction(dm, 0.0, funcs, ctxs, INSERT_ALL_VALUES, u);CHKERRQ(ierr); 40030602db0SMatthew G. Knepley } 401c4762a1bSJed Brown ierr = TSSolve(ts, u);CHKERRQ(ierr); 402c4762a1bSJed Brown ierr = VecViewFromOptions(u, NULL, "-sol_vec_view");CHKERRQ(ierr); 403c4762a1bSJed Brown 404c4762a1bSJed Brown ierr = VecDestroy(&u);CHKERRQ(ierr); 405c4762a1bSJed Brown ierr = VecDestroy(&r);CHKERRQ(ierr); 406c4762a1bSJed Brown ierr = TSDestroy(&ts);CHKERRQ(ierr); 407c4762a1bSJed Brown ierr = DMDestroy(&dm);CHKERRQ(ierr); 408c4762a1bSJed Brown ierr = PetscFinalize(); 409c4762a1bSJed Brown return ierr; 410c4762a1bSJed Brown } 411c4762a1bSJed Brown 412c4762a1bSJed Brown /*TEST 413c4762a1bSJed Brown 414c4762a1bSJed Brown # Full solves 415c4762a1bSJed Brown test: 416c4762a1bSJed Brown suffix: 2d_p2p1_r1 417c4762a1bSJed Brown requires: !single triangle 418c4762a1bSJed Brown filter: sed -e "s~ATOL~RTOL~g" -e "s~ABS~RELATIVE~g" 41930602db0SMatthew G. Knepley args: -dm_refine 1 -vel_petscspace_degree 2 -pres_petscspace_degree 1 \ 42030602db0SMatthew G. Knepley -ts_type beuler -ts_max_steps 10 -ts_dt 0.1 -ts_monitor -dmts_check \ 42130602db0SMatthew G. Knepley -snes_monitor_short -snes_converged_reason \ 42230602db0SMatthew G. Knepley -ksp_monitor_short -ksp_converged_reason \ 42330602db0SMatthew G. Knepley -pc_type fieldsplit -pc_fieldsplit_type schur -pc_fieldsplit_schur_fact_type full \ 42430602db0SMatthew G. Knepley -fieldsplit_velocity_pc_type lu \ 42530602db0SMatthew G. Knepley -fieldsplit_pressure_ksp_rtol 1.0e-10 -fieldsplit_pressure_pc_type jacobi 42630602db0SMatthew G. Knepley 427c4762a1bSJed Brown test: 428c4762a1bSJed Brown suffix: 2d_q2q1_r1 429c4762a1bSJed Brown requires: !single 430c4762a1bSJed Brown filter: sed -e "s~ATOL~RTOL~g" -e "s~ABS~RELATIVE~g" -e "s~ 0\]~ 0.0\]~g" 43130602db0SMatthew G. Knepley args: -dm_plex_simplex 0 -dm_refine 1 -vel_petscspace_degree 2 -pres_petscspace_degree 1 \ 43230602db0SMatthew G. Knepley -ts_type beuler -ts_max_steps 10 -ts_dt 0.1 -ts_monitor -dmts_check \ 43330602db0SMatthew G. Knepley -snes_monitor_short -snes_converged_reason \ 43430602db0SMatthew G. Knepley -ksp_monitor_short -ksp_converged_reason \ 43530602db0SMatthew G. Knepley -pc_type fieldsplit -pc_fieldsplit_type schur -pc_fieldsplit_schur_fact_type full \ 43630602db0SMatthew G. Knepley -fieldsplit_velocity_pc_type lu \ 43730602db0SMatthew G. Knepley -fieldsplit_pressure_ksp_rtol 1.0e-10 -fieldsplit_pressure_pc_type jacobi 438c4762a1bSJed Brown 439c4762a1bSJed Brown TEST*/ 440