1c4762a1bSJed Brown static char help[] = "Solves the ordinary differential equations (IVPs) using explicit and implicit time-integration methods.\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown /* 4c4762a1bSJed Brown 5c4762a1bSJed Brown Concepts: TS 6c4762a1bSJed Brown Useful command line parameters: 7c4762a1bSJed Brown -problem <hull1972a1>: choose which problem to solve (see references 8c4762a1bSJed Brown for complete listing of problems). 9c4762a1bSJed Brown -ts_type <euler>: specify time-integrator 10c4762a1bSJed Brown -ts_adapt_type <basic>: specify time-step adapting (none,basic,advanced) 11c4762a1bSJed Brown -refinement_levels <1>: number of refinement levels for convergence analysis 12c4762a1bSJed Brown -refinement_factor <2.0>: factor to refine time step size by for convergence analysis 13c4762a1bSJed Brown -dt <0.01>: specify time step (initial time step for convergence analysis) 14c4762a1bSJed Brown 15c4762a1bSJed Brown */ 16c4762a1bSJed Brown 17c4762a1bSJed Brown /* 18c4762a1bSJed Brown List of cases and their names in the code:- 19c4762a1bSJed Brown From Hull, T.E., Enright, W.H., Fellen, B.M., and Sedgwick, A.E., 20c4762a1bSJed Brown "Comparing Numerical Methods for Ordinary Differential 21c4762a1bSJed Brown Equations", SIAM J. Numer. Anal., 9(4), 1972, pp. 603 - 635 22c4762a1bSJed Brown A1 -> "hull1972a1" (exact solution available) 23c4762a1bSJed Brown A2 -> "hull1972a2" (exact solution available) 24c4762a1bSJed Brown A3 -> "hull1972a3" (exact solution available) 25c4762a1bSJed Brown A4 -> "hull1972a4" (exact solution available) 26c4762a1bSJed Brown A5 -> "hull1972a5" 27c4762a1bSJed Brown B1 -> "hull1972b1" 28c4762a1bSJed Brown B2 -> "hull1972b2" 29c4762a1bSJed Brown B3 -> "hull1972b3" 30c4762a1bSJed Brown B4 -> "hull1972b4" 31c4762a1bSJed Brown B5 -> "hull1972b5" 32c4762a1bSJed Brown C1 -> "hull1972c1" 33c4762a1bSJed Brown C2 -> "hull1972c2" 34c4762a1bSJed Brown C3 -> "hull1972c3" 35c4762a1bSJed Brown C4 -> "hull1972c4" 36c4762a1bSJed Brown 37c4762a1bSJed Brown From Constantinescu, E. "Estimating Global Errors in Time Stepping" ArXiv e-prints, 38c4762a1bSJed Brown https://arxiv.org/abs/1503.05166, 2016 39c4762a1bSJed Brown 40c4762a1bSJed Brown Kulikov2013I -> "kulik2013i" 41c4762a1bSJed Brown 42c4762a1bSJed Brown */ 43c4762a1bSJed Brown 44c4762a1bSJed Brown #include <petscts.h> 45c4762a1bSJed Brown 46c4762a1bSJed Brown /* Function declarations */ 47c4762a1bSJed Brown PetscErrorCode (*RHSFunction) (TS,PetscReal,Vec,Vec,void*); 48c4762a1bSJed Brown PetscErrorCode (*RHSJacobian) (TS,PetscReal,Vec,Mat,Mat,void*); 49c4762a1bSJed Brown PetscErrorCode (*IFunction) (TS,PetscReal,Vec,Vec,Vec,void*); 50c4762a1bSJed Brown PetscErrorCode (*IJacobian) (TS,PetscReal,Vec,Vec,PetscReal,Mat,Mat,void*); 51c4762a1bSJed Brown 52c4762a1bSJed Brown /* Returns the size of the system of equations depending on problem specification */ 53c4762a1bSJed Brown PetscInt GetSize(const char *p) 54c4762a1bSJed Brown { 55c4762a1bSJed Brown PetscFunctionBegin; 56c4762a1bSJed Brown if ((!strcmp(p,"hull1972a1")) 57c4762a1bSJed Brown ||(!strcmp(p,"hull1972a2")) 58c4762a1bSJed Brown ||(!strcmp(p,"hull1972a3")) 59c4762a1bSJed Brown ||(!strcmp(p,"hull1972a4")) 60c4762a1bSJed Brown ||(!strcmp(p,"hull1972a5"))) PetscFunctionReturn(1); 61c4762a1bSJed Brown else if (!strcmp(p,"hull1972b1")) PetscFunctionReturn(2); 62c4762a1bSJed Brown else if ((!strcmp(p,"hull1972b2")) 63c4762a1bSJed Brown ||(!strcmp(p,"hull1972b3")) 64c4762a1bSJed Brown ||(!strcmp(p,"hull1972b4")) 65c4762a1bSJed Brown ||(!strcmp(p,"hull1972b5"))) PetscFunctionReturn(3); 66c4762a1bSJed Brown else if ((!strcmp(p,"kulik2013i"))) PetscFunctionReturn(4); 67c4762a1bSJed Brown else if ((!strcmp(p,"hull1972c1")) 68c4762a1bSJed Brown ||(!strcmp(p,"hull1972c2")) 69c4762a1bSJed Brown ||(!strcmp(p,"hull1972c3"))) PetscFunctionReturn(10); 70c4762a1bSJed Brown else if (!strcmp(p,"hull1972c4")) PetscFunctionReturn(51); 71c4762a1bSJed Brown else PetscFunctionReturn(-1); 72c4762a1bSJed Brown } 73c4762a1bSJed Brown 74c4762a1bSJed Brown /****************************************************************/ 75c4762a1bSJed Brown 76c4762a1bSJed Brown /* Problem specific functions */ 77c4762a1bSJed Brown 78c4762a1bSJed Brown /* Hull, 1972, Problem A1 */ 79c4762a1bSJed Brown 80c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec F, void *s) 81c4762a1bSJed Brown { 82c4762a1bSJed Brown PetscErrorCode ierr; 83c4762a1bSJed Brown PetscScalar *f; 84c4762a1bSJed Brown const PetscScalar *y; 85c4762a1bSJed Brown 86c4762a1bSJed Brown PetscFunctionBegin; 87c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 88c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 89c4762a1bSJed Brown f[0] = -y[0]; 90c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 91c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 92c4762a1bSJed Brown PetscFunctionReturn(0); 93c4762a1bSJed Brown } 94c4762a1bSJed Brown 95c4762a1bSJed Brown PetscErrorCode RHSJacobian_Hull1972A1(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 96c4762a1bSJed Brown { 97c4762a1bSJed Brown PetscErrorCode ierr; 98c4762a1bSJed Brown const PetscScalar *y; 99c4762a1bSJed Brown PetscInt row = 0,col = 0; 100c4762a1bSJed Brown PetscScalar value = -1.0; 101c4762a1bSJed Brown 102c4762a1bSJed Brown PetscFunctionBegin; 103c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 104c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 105c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 106c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 107c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 108c4762a1bSJed Brown PetscFunctionReturn(0); 109c4762a1bSJed Brown } 110c4762a1bSJed Brown 111c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 112c4762a1bSJed Brown { 113c4762a1bSJed Brown PetscErrorCode ierr; 114c4762a1bSJed Brown const PetscScalar *y; 115c4762a1bSJed Brown PetscScalar *f; 116c4762a1bSJed Brown 117c4762a1bSJed Brown PetscFunctionBegin; 118c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 119c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 120c4762a1bSJed Brown f[0] = -y[0]; 121c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 122c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 123c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 124c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 125c4762a1bSJed Brown PetscFunctionReturn(0); 126c4762a1bSJed Brown } 127c4762a1bSJed Brown 128c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 129c4762a1bSJed Brown { 130c4762a1bSJed Brown PetscErrorCode ierr; 131c4762a1bSJed Brown const PetscScalar *y; 132c4762a1bSJed Brown PetscInt row = 0,col = 0; 133c4762a1bSJed Brown PetscScalar value = a - 1.0; 134c4762a1bSJed Brown 135c4762a1bSJed Brown PetscFunctionBegin; 136c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 137c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 138c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 139c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 140c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 141c4762a1bSJed Brown PetscFunctionReturn(0); 142c4762a1bSJed Brown } 143c4762a1bSJed Brown 144c4762a1bSJed Brown /* Hull, 1972, Problem A2 */ 145c4762a1bSJed Brown 146c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec F, void *s) 147c4762a1bSJed Brown { 148c4762a1bSJed Brown PetscErrorCode ierr; 149c4762a1bSJed Brown const PetscScalar *y; 150c4762a1bSJed Brown PetscScalar *f; 151c4762a1bSJed Brown 152c4762a1bSJed Brown PetscFunctionBegin; 153c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 154c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 155c4762a1bSJed Brown f[0] = -0.5*y[0]*y[0]*y[0]; 156c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 157c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 158c4762a1bSJed Brown PetscFunctionReturn(0); 159c4762a1bSJed Brown } 160c4762a1bSJed Brown 161c4762a1bSJed Brown PetscErrorCode RHSJacobian_Hull1972A2(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 162c4762a1bSJed Brown { 163c4762a1bSJed Brown PetscErrorCode ierr; 164c4762a1bSJed Brown const PetscScalar *y; 165c4762a1bSJed Brown PetscInt row = 0,col = 0; 166c4762a1bSJed Brown PetscScalar value; 167c4762a1bSJed Brown 168c4762a1bSJed Brown PetscFunctionBegin; 169c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 170c4762a1bSJed Brown value = -0.5*3.0*y[0]*y[0]; 171c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 172c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 173c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 174c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 175c4762a1bSJed Brown PetscFunctionReturn(0); 176c4762a1bSJed Brown } 177c4762a1bSJed Brown 178c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 179c4762a1bSJed Brown { 180c4762a1bSJed Brown PetscErrorCode ierr; 181c4762a1bSJed Brown PetscScalar *f; 182c4762a1bSJed Brown const PetscScalar *y; 183c4762a1bSJed Brown 184c4762a1bSJed Brown PetscFunctionBegin; 185c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 186c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 187c4762a1bSJed Brown f[0] = -0.5*y[0]*y[0]*y[0]; 188c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 189c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 190c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 191c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 192c4762a1bSJed Brown PetscFunctionReturn(0); 193c4762a1bSJed Brown } 194c4762a1bSJed Brown 195c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 196c4762a1bSJed Brown { 197c4762a1bSJed Brown PetscErrorCode ierr; 198c4762a1bSJed Brown const PetscScalar *y; 199c4762a1bSJed Brown PetscInt row = 0,col = 0; 200c4762a1bSJed Brown PetscScalar value; 201c4762a1bSJed Brown 202c4762a1bSJed Brown PetscFunctionBegin; 203c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 204c4762a1bSJed Brown value = a + 0.5*3.0*y[0]*y[0]; 205c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 206c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 207c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 208c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 209c4762a1bSJed Brown PetscFunctionReturn(0); 210c4762a1bSJed Brown } 211c4762a1bSJed Brown 212c4762a1bSJed Brown /* Hull, 1972, Problem A3 */ 213c4762a1bSJed Brown 214c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec F, void *s) 215c4762a1bSJed Brown { 216c4762a1bSJed Brown PetscErrorCode ierr; 217c4762a1bSJed Brown const PetscScalar *y; 218c4762a1bSJed Brown PetscScalar *f; 219c4762a1bSJed Brown 220c4762a1bSJed Brown PetscFunctionBegin; 221c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 222c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 223c4762a1bSJed Brown f[0] = y[0]*PetscCosReal(t); 224c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 225c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 226c4762a1bSJed Brown PetscFunctionReturn(0); 227c4762a1bSJed Brown } 228c4762a1bSJed Brown 229c4762a1bSJed Brown PetscErrorCode RHSJacobian_Hull1972A3(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 230c4762a1bSJed Brown { 231c4762a1bSJed Brown PetscErrorCode ierr; 232c4762a1bSJed Brown const PetscScalar *y; 233c4762a1bSJed Brown PetscInt row = 0,col = 0; 234c4762a1bSJed Brown PetscScalar value = PetscCosReal(t); 235c4762a1bSJed Brown 236c4762a1bSJed Brown PetscFunctionBegin; 237c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 238c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 239c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 240c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 241c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 242c4762a1bSJed Brown PetscFunctionReturn(0); 243c4762a1bSJed Brown } 244c4762a1bSJed Brown 245c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 246c4762a1bSJed Brown { 247c4762a1bSJed Brown PetscErrorCode ierr; 248c4762a1bSJed Brown PetscScalar *f; 249c4762a1bSJed Brown const PetscScalar *y; 250c4762a1bSJed Brown 251c4762a1bSJed Brown PetscFunctionBegin; 252c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 253c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 254c4762a1bSJed Brown f[0] = y[0]*PetscCosReal(t); 255c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 256c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 257c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 258c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 259c4762a1bSJed Brown PetscFunctionReturn(0); 260c4762a1bSJed Brown } 261c4762a1bSJed Brown 262c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 263c4762a1bSJed Brown { 264c4762a1bSJed Brown PetscErrorCode ierr; 265c4762a1bSJed Brown const PetscScalar *y; 266c4762a1bSJed Brown PetscInt row = 0,col = 0; 267c4762a1bSJed Brown PetscScalar value = a - PetscCosReal(t); 268c4762a1bSJed Brown 269c4762a1bSJed Brown PetscFunctionBegin; 270c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 271c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 272c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 273c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 274c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 275c4762a1bSJed Brown PetscFunctionReturn(0); 276c4762a1bSJed Brown } 277c4762a1bSJed Brown 278c4762a1bSJed Brown /* Hull, 1972, Problem A4 */ 279c4762a1bSJed Brown 280c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec F, void *s) 281c4762a1bSJed Brown { 282c4762a1bSJed Brown PetscErrorCode ierr; 283c4762a1bSJed Brown PetscScalar *f; 284c4762a1bSJed Brown const PetscScalar *y; 285c4762a1bSJed Brown 286c4762a1bSJed Brown PetscFunctionBegin; 287c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 288c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 289c4762a1bSJed Brown f[0] = (0.25*y[0])*(1.0-0.05*y[0]); 290c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 291c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 292c4762a1bSJed Brown PetscFunctionReturn(0); 293c4762a1bSJed Brown } 294c4762a1bSJed Brown 295c4762a1bSJed Brown PetscErrorCode RHSJacobian_Hull1972A4(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 296c4762a1bSJed Brown { 297c4762a1bSJed Brown PetscErrorCode ierr; 298c4762a1bSJed Brown const PetscScalar *y; 299c4762a1bSJed Brown PetscInt row = 0,col = 0; 300c4762a1bSJed Brown PetscScalar value; 301c4762a1bSJed Brown 302c4762a1bSJed Brown PetscFunctionBegin; 303c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 304c4762a1bSJed Brown value = 0.25*(1.0-0.05*y[0]) - (0.25*y[0])*0.05; 305c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 306c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 307c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 308c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 309c4762a1bSJed Brown PetscFunctionReturn(0); 310c4762a1bSJed Brown } 311c4762a1bSJed Brown 312c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 313c4762a1bSJed Brown { 314c4762a1bSJed Brown PetscErrorCode ierr; 315c4762a1bSJed Brown PetscScalar *f; 316c4762a1bSJed Brown const PetscScalar *y; 317c4762a1bSJed Brown 318c4762a1bSJed Brown PetscFunctionBegin; 319c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 320c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 321c4762a1bSJed Brown f[0] = (0.25*y[0])*(1.0-0.05*y[0]); 322c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 323c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 324c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 325c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 326c4762a1bSJed Brown PetscFunctionReturn(0); 327c4762a1bSJed Brown } 328c4762a1bSJed Brown 329c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 330c4762a1bSJed Brown { 331c4762a1bSJed Brown PetscErrorCode ierr; 332c4762a1bSJed Brown const PetscScalar *y; 333c4762a1bSJed Brown PetscInt row = 0,col = 0; 334c4762a1bSJed Brown PetscScalar value; 335c4762a1bSJed Brown 336c4762a1bSJed Brown PetscFunctionBegin; 337c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 338c4762a1bSJed Brown value = a - 0.25*(1.0-0.05*y[0]) + (0.25*y[0])*0.05; 339c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 340c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 341c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 342c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 343c4762a1bSJed Brown PetscFunctionReturn(0); 344c4762a1bSJed Brown } 345c4762a1bSJed Brown 346c4762a1bSJed Brown /* Hull, 1972, Problem A5 */ 347c4762a1bSJed Brown 348c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec F, void *s) 349c4762a1bSJed Brown { 350c4762a1bSJed Brown PetscErrorCode ierr; 351c4762a1bSJed Brown PetscScalar *f; 352c4762a1bSJed Brown const PetscScalar *y; 353c4762a1bSJed Brown 354c4762a1bSJed Brown PetscFunctionBegin; 355c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 356c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 357c4762a1bSJed Brown f[0] = (y[0]-t)/(y[0]+t); 358c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 359c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 360c4762a1bSJed Brown PetscFunctionReturn(0); 361c4762a1bSJed Brown } 362c4762a1bSJed Brown 363c4762a1bSJed Brown PetscErrorCode RHSJacobian_Hull1972A5(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 364c4762a1bSJed Brown { 365c4762a1bSJed Brown PetscErrorCode ierr; 366c4762a1bSJed Brown const PetscScalar *y; 367c4762a1bSJed Brown PetscInt row = 0,col = 0; 368c4762a1bSJed Brown PetscScalar value; 369c4762a1bSJed Brown 370c4762a1bSJed Brown PetscFunctionBegin; 371c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 372c4762a1bSJed Brown value = 2*t/((t+y[0])*(t+y[0])); 373c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 374c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 375c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 376c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 377c4762a1bSJed Brown PetscFunctionReturn(0); 378c4762a1bSJed Brown } 379c4762a1bSJed Brown 380c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 381c4762a1bSJed Brown { 382c4762a1bSJed Brown PetscErrorCode ierr; 383c4762a1bSJed Brown PetscScalar *f; 384c4762a1bSJed Brown const PetscScalar *y; 385c4762a1bSJed Brown 386c4762a1bSJed Brown PetscFunctionBegin; 387c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 388c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 389c4762a1bSJed Brown f[0] = (y[0]-t)/(y[0]+t); 390c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 391c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 392c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 393c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 394c4762a1bSJed Brown PetscFunctionReturn(0); 395c4762a1bSJed Brown } 396c4762a1bSJed Brown 397c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 398c4762a1bSJed Brown { 399c4762a1bSJed Brown PetscErrorCode ierr; 400c4762a1bSJed Brown const PetscScalar *y; 401c4762a1bSJed Brown PetscInt row = 0,col = 0; 402c4762a1bSJed Brown PetscScalar value; 403c4762a1bSJed Brown 404c4762a1bSJed Brown PetscFunctionBegin; 405c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 406c4762a1bSJed Brown value = a - 2*t/((t+y[0])*(t+y[0])); 407c4762a1bSJed Brown ierr = MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);CHKERRQ(ierr); 408c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 409c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 410c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 411c4762a1bSJed Brown PetscFunctionReturn(0); 412c4762a1bSJed Brown } 413c4762a1bSJed Brown 414c4762a1bSJed Brown /* Hull, 1972, Problem B1 */ 415c4762a1bSJed Brown 416c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec F, void *s) 417c4762a1bSJed Brown { 418c4762a1bSJed Brown PetscErrorCode ierr; 419c4762a1bSJed Brown PetscScalar *f; 420c4762a1bSJed Brown const PetscScalar *y; 421c4762a1bSJed Brown 422c4762a1bSJed Brown PetscFunctionBegin; 423c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 424c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 425c4762a1bSJed Brown f[0] = 2.0*(y[0] - y[0]*y[1]); 426c4762a1bSJed Brown f[1] = -(y[1]-y[0]*y[1]); 427c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 428c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 429c4762a1bSJed Brown PetscFunctionReturn(0); 430c4762a1bSJed Brown } 431c4762a1bSJed Brown 432c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 433c4762a1bSJed Brown { 434c4762a1bSJed Brown PetscErrorCode ierr; 435c4762a1bSJed Brown PetscScalar *f; 436c4762a1bSJed Brown const PetscScalar *y; 437c4762a1bSJed Brown 438c4762a1bSJed Brown PetscFunctionBegin; 439c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 440c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 441c4762a1bSJed Brown f[0] = 2.0*(y[0] - y[0]*y[1]); 442c4762a1bSJed Brown f[1] = -(y[1]-y[0]*y[1]); 443c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 444c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 445c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 446c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 447c4762a1bSJed Brown PetscFunctionReturn(0); 448c4762a1bSJed Brown } 449c4762a1bSJed Brown 450c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 451c4762a1bSJed Brown { 452c4762a1bSJed Brown PetscErrorCode ierr; 453c4762a1bSJed Brown const PetscScalar *y; 454c4762a1bSJed Brown PetscInt row[2] = {0,1}; 455c4762a1bSJed Brown PetscScalar value[2][2]; 456c4762a1bSJed Brown 457c4762a1bSJed Brown PetscFunctionBegin; 458c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 459c4762a1bSJed Brown value[0][0] = a - 2.0*(1.0-y[1]); value[0][1] = 2.0*y[0]; 460c4762a1bSJed Brown value[1][0] = -y[1]; value[1][1] = a + 1.0 - y[0]; 461c4762a1bSJed Brown ierr = MatSetValues(A,2,&row[0],2,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 462c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 463c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 464c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 465c4762a1bSJed Brown PetscFunctionReturn(0); 466c4762a1bSJed Brown } 467c4762a1bSJed Brown 468c4762a1bSJed Brown /* Hull, 1972, Problem B2 */ 469c4762a1bSJed Brown 470c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec F, void *s) 471c4762a1bSJed Brown { 472c4762a1bSJed Brown PetscErrorCode ierr; 473c4762a1bSJed Brown PetscScalar *f; 474c4762a1bSJed Brown const PetscScalar *y; 475c4762a1bSJed Brown 476c4762a1bSJed Brown PetscFunctionBegin; 477c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 478c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 479c4762a1bSJed Brown f[0] = -y[0] + y[1]; 480c4762a1bSJed Brown f[1] = y[0] - 2.0*y[1] + y[2]; 481c4762a1bSJed Brown f[2] = y[1] - y[2]; 482c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 483c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 484c4762a1bSJed Brown PetscFunctionReturn(0); 485c4762a1bSJed Brown } 486c4762a1bSJed Brown 487c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 488c4762a1bSJed Brown { 489c4762a1bSJed Brown PetscErrorCode ierr; 490c4762a1bSJed Brown PetscScalar *f; 491c4762a1bSJed Brown const PetscScalar *y; 492c4762a1bSJed Brown 493c4762a1bSJed Brown PetscFunctionBegin; 494c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 495c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 496c4762a1bSJed Brown f[0] = -y[0] + y[1]; 497c4762a1bSJed Brown f[1] = y[0] - 2.0*y[1] + y[2]; 498c4762a1bSJed Brown f[2] = y[1] - y[2]; 499c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 500c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 501c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 502c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 503c4762a1bSJed Brown PetscFunctionReturn(0); 504c4762a1bSJed Brown } 505c4762a1bSJed Brown 506c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 507c4762a1bSJed Brown { 508c4762a1bSJed Brown PetscErrorCode ierr; 509c4762a1bSJed Brown const PetscScalar *y; 510c4762a1bSJed Brown PetscInt row[3] = {0,1,2}; 511c4762a1bSJed Brown PetscScalar value[3][3]; 512c4762a1bSJed Brown 513c4762a1bSJed Brown PetscFunctionBegin; 514c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 515c4762a1bSJed Brown value[0][0] = a + 1.0; value[0][1] = -1.0; value[0][2] = 0; 516c4762a1bSJed Brown value[1][0] = -1.0; value[1][1] = a + 2.0; value[1][2] = -1.0; 517c4762a1bSJed Brown value[2][0] = 0; value[2][1] = -1.0; value[2][2] = a + 1.0; 518c4762a1bSJed Brown ierr = MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 519c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 520c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 521c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 522c4762a1bSJed Brown PetscFunctionReturn(0); 523c4762a1bSJed Brown } 524c4762a1bSJed Brown 525c4762a1bSJed Brown /* Hull, 1972, Problem B3 */ 526c4762a1bSJed Brown 527c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec F, void *s) 528c4762a1bSJed Brown { 529c4762a1bSJed Brown PetscErrorCode ierr; 530c4762a1bSJed Brown PetscScalar *f; 531c4762a1bSJed Brown const PetscScalar *y; 532c4762a1bSJed Brown 533c4762a1bSJed Brown PetscFunctionBegin; 534c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 535c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 536c4762a1bSJed Brown f[0] = -y[0]; 537c4762a1bSJed Brown f[1] = y[0] - y[1]*y[1]; 538c4762a1bSJed Brown f[2] = y[1]*y[1]; 539c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 540c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 541c4762a1bSJed Brown PetscFunctionReturn(0); 542c4762a1bSJed Brown } 543c4762a1bSJed Brown 544c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 545c4762a1bSJed Brown { 546c4762a1bSJed Brown PetscErrorCode ierr; 547c4762a1bSJed Brown PetscScalar *f; 548c4762a1bSJed Brown const PetscScalar *y; 549c4762a1bSJed Brown 550c4762a1bSJed Brown PetscFunctionBegin; 551c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 552c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 553c4762a1bSJed Brown f[0] = -y[0]; 554c4762a1bSJed Brown f[1] = y[0] - y[1]*y[1]; 555c4762a1bSJed Brown f[2] = y[1]*y[1]; 556c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 557c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 558c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 559c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 560c4762a1bSJed Brown PetscFunctionReturn(0); 561c4762a1bSJed Brown } 562c4762a1bSJed Brown 563c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 564c4762a1bSJed Brown { 565c4762a1bSJed Brown PetscErrorCode ierr; 566c4762a1bSJed Brown const PetscScalar *y; 567c4762a1bSJed Brown PetscInt row[3] = {0,1,2}; 568c4762a1bSJed Brown PetscScalar value[3][3]; 569c4762a1bSJed Brown 570c4762a1bSJed Brown PetscFunctionBegin; 571c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 572c4762a1bSJed Brown value[0][0] = a + 1.0; value[0][1] = 0; value[0][2] = 0; 573c4762a1bSJed Brown value[1][0] = -1.0; value[1][1] = a + 2.0*y[1]; value[1][2] = 0; 574c4762a1bSJed Brown value[2][0] = 0; value[2][1] = -2.0*y[1]; value[2][2] = a; 575c4762a1bSJed Brown ierr = MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 576c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 577c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 578c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 579c4762a1bSJed Brown PetscFunctionReturn(0); 580c4762a1bSJed Brown } 581c4762a1bSJed Brown 582c4762a1bSJed Brown /* Hull, 1972, Problem B4 */ 583c4762a1bSJed Brown 584c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec F, void *s) 585c4762a1bSJed Brown { 586c4762a1bSJed Brown PetscErrorCode ierr; 587c4762a1bSJed Brown PetscScalar *f; 588c4762a1bSJed Brown const PetscScalar *y; 589c4762a1bSJed Brown 590c4762a1bSJed Brown PetscFunctionBegin; 591c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 592c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 593c4762a1bSJed Brown f[0] = -y[1] - y[0]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 594c4762a1bSJed Brown f[1] = y[0] - y[1]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 595c4762a1bSJed Brown f[2] = y[0]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 596c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 597c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 598c4762a1bSJed Brown PetscFunctionReturn(0); 599c4762a1bSJed Brown } 600c4762a1bSJed Brown 601c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 602c4762a1bSJed Brown { 603c4762a1bSJed Brown PetscErrorCode ierr; 604c4762a1bSJed Brown PetscScalar *f; 605c4762a1bSJed Brown const PetscScalar *y; 606c4762a1bSJed Brown 607c4762a1bSJed Brown PetscFunctionBegin; 608c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 609c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 610c4762a1bSJed Brown f[0] = -y[1] - y[0]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 611c4762a1bSJed Brown f[1] = y[0] - y[1]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 612c4762a1bSJed Brown f[2] = y[0]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]); 613c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 614c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 615c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 616c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 617c4762a1bSJed Brown PetscFunctionReturn(0); 618c4762a1bSJed Brown } 619c4762a1bSJed Brown 620c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 621c4762a1bSJed Brown { 622c4762a1bSJed Brown PetscErrorCode ierr; 623c4762a1bSJed Brown const PetscScalar *y; 624c4762a1bSJed Brown PetscInt row[3] = {0,1,2}; 625c4762a1bSJed Brown PetscScalar value[3][3],fac,fac2; 626c4762a1bSJed Brown 627c4762a1bSJed Brown PetscFunctionBegin; 628c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 629c4762a1bSJed Brown fac = PetscPowScalar(y[0]*y[0]+y[1]*y[1],-1.5); 630c4762a1bSJed Brown fac2 = PetscPowScalar(y[0]*y[0]+y[1]*y[1],-0.5); 631c4762a1bSJed Brown value[0][0] = a + (y[1]*y[1]*y[2])*fac; 632c4762a1bSJed Brown value[0][1] = 1.0 - (y[0]*y[1]*y[2])*fac; 633c4762a1bSJed Brown value[0][2] = y[0]*fac2; 634c4762a1bSJed Brown value[1][0] = -1.0 - y[0]*y[1]*y[2]*fac; 635c4762a1bSJed Brown value[1][1] = a + y[0]*y[0]*y[2]*fac; 636c4762a1bSJed Brown value[1][2] = y[1]*fac2; 637c4762a1bSJed Brown value[2][0] = -y[1]*y[1]*fac; 638c4762a1bSJed Brown value[2][1] = y[0]*y[1]*fac; 639c4762a1bSJed Brown value[2][2] = a; 640c4762a1bSJed Brown ierr = MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 641c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 642c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 643c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 644c4762a1bSJed Brown PetscFunctionReturn(0); 645c4762a1bSJed Brown } 646c4762a1bSJed Brown 647c4762a1bSJed Brown /* Hull, 1972, Problem B5 */ 648c4762a1bSJed Brown 649c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec F, void *s) 650c4762a1bSJed Brown { 651c4762a1bSJed Brown PetscErrorCode ierr; 652c4762a1bSJed Brown PetscScalar *f; 653c4762a1bSJed Brown const PetscScalar *y; 654c4762a1bSJed Brown 655c4762a1bSJed Brown PetscFunctionBegin; 656c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 657c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 658c4762a1bSJed Brown f[0] = y[1]*y[2]; 659c4762a1bSJed Brown f[1] = -y[0]*y[2]; 660c4762a1bSJed Brown f[2] = -0.51*y[0]*y[1]; 661c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 662c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 663c4762a1bSJed Brown PetscFunctionReturn(0); 664c4762a1bSJed Brown } 665c4762a1bSJed Brown 666c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 667c4762a1bSJed Brown { 668c4762a1bSJed Brown PetscErrorCode ierr; 669c4762a1bSJed Brown PetscScalar *f; 670c4762a1bSJed Brown const PetscScalar *y; 671c4762a1bSJed Brown 672c4762a1bSJed Brown PetscFunctionBegin; 673c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 674c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 675c4762a1bSJed Brown f[0] = y[1]*y[2]; 676c4762a1bSJed Brown f[1] = -y[0]*y[2]; 677c4762a1bSJed Brown f[2] = -0.51*y[0]*y[1]; 678c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 679c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 680c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 681c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 682c4762a1bSJed Brown PetscFunctionReturn(0); 683c4762a1bSJed Brown } 684c4762a1bSJed Brown 685c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 686c4762a1bSJed Brown { 687c4762a1bSJed Brown PetscErrorCode ierr; 688c4762a1bSJed Brown const PetscScalar *y; 689c4762a1bSJed Brown PetscInt row[3] = {0,1,2}; 690c4762a1bSJed Brown PetscScalar value[3][3]; 691c4762a1bSJed Brown 692c4762a1bSJed Brown PetscFunctionBegin; 693c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 694c4762a1bSJed Brown value[0][0] = a; value[0][1] = -y[2]; value[0][2] = -y[1]; 695c4762a1bSJed Brown value[1][0] = y[2]; value[1][1] = a; value[1][2] = y[0]; 696c4762a1bSJed Brown value[2][0] = 0.51*y[1]; value[2][1] = 0.51*y[0]; value[2][2] = a; 697c4762a1bSJed Brown ierr = MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 698c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 699c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 700c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 701c4762a1bSJed Brown PetscFunctionReturn(0); 702c4762a1bSJed Brown } 703c4762a1bSJed Brown 704c4762a1bSJed Brown 705c4762a1bSJed Brown /* Kulikov, 2013, Problem I */ 706c4762a1bSJed Brown 707c4762a1bSJed Brown PetscErrorCode RHSFunction_Kulikov2013I(TS ts, PetscReal t, Vec Y, Vec F, void *s) 708c4762a1bSJed Brown { 709c4762a1bSJed Brown PetscErrorCode ierr; 710c4762a1bSJed Brown PetscScalar *f; 711c4762a1bSJed Brown const PetscScalar *y; 712c4762a1bSJed Brown 713c4762a1bSJed Brown PetscFunctionBegin; 714c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 715c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 716c4762a1bSJed Brown f[0] = 2.*t*PetscPowScalar(y[1],1./5.)*y[3]; 717c4762a1bSJed Brown f[1] = 10.*t*y[3]*PetscExpScalar(5.0*(y[2]-1.)); 718c4762a1bSJed Brown f[2] = 2.*t*y[3]; 719c4762a1bSJed Brown f[3] = -2.*t*PetscLogScalar(y[0]); 720c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 721c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 722c4762a1bSJed Brown PetscFunctionReturn(0); 723c4762a1bSJed Brown } 724c4762a1bSJed Brown 725c4762a1bSJed Brown PetscErrorCode RHSJacobian_Kulikov2013I(TS ts, PetscReal t, Vec Y, Mat A, Mat B, void *s) 726c4762a1bSJed Brown { 727c4762a1bSJed Brown PetscErrorCode ierr; 728c4762a1bSJed Brown const PetscScalar *y; 729c4762a1bSJed Brown PetscInt row[4] = {0,1,2,3}; 730c4762a1bSJed Brown PetscScalar value[4][4]; 731c4762a1bSJed Brown PetscScalar m1,m2; 732c4762a1bSJed Brown PetscFunctionBegin; 733c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 734c4762a1bSJed Brown m1=(2.*t*y[3])/(5.*PetscPowScalar(y[1],4./5.)); 735c4762a1bSJed Brown m2=2.*t*PetscPowScalar(y[1],1./5.); 736c4762a1bSJed Brown value[0][0] = 0. ; value[0][1] = m1; value[0][2] = 0.; value[0][3] = m2; 737c4762a1bSJed Brown m1=50.*t*y[3]*PetscExpScalar(5.0*(y[2]-1.)); 738c4762a1bSJed Brown m2=10.*t*PetscExpScalar(5.0*(y[2]-1.)); 739c4762a1bSJed Brown value[1][0] = 0.; value[1][1] = 0. ; value[1][2] = m1; value[1][3] = m2; 740c4762a1bSJed Brown value[2][0] = 0.; value[2][1] = 0.; value[2][2] = 0.; value[2][3] = 2*t; 741c4762a1bSJed Brown value[3][0] = -2.*t/y[0];value[3][1] = 0.; value[3][2] = 0.; value[3][3] = 0.; 742c4762a1bSJed Brown ierr = MatSetValues(A,4,&row[0],4,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 743c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 744c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 745c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 746c4762a1bSJed Brown PetscFunctionReturn(0); 747c4762a1bSJed Brown } 748c4762a1bSJed Brown 749c4762a1bSJed Brown PetscErrorCode IFunction_Kulikov2013I(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 750c4762a1bSJed Brown { 751c4762a1bSJed Brown PetscErrorCode ierr; 752c4762a1bSJed Brown PetscScalar *f; 753c4762a1bSJed Brown const PetscScalar *y; 754c4762a1bSJed Brown 755c4762a1bSJed Brown PetscFunctionBegin; 756c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 757c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 758c4762a1bSJed Brown f[0] = 2.*t*PetscPowScalar(y[1],1./5.)*y[3]; 759c4762a1bSJed Brown f[1] = 10.*t*y[3]*PetscExpScalar(5.0*(y[2]-1.)); 760c4762a1bSJed Brown f[2] = 2.*t*y[3]; 761c4762a1bSJed Brown f[3] = -2.*t*PetscLogScalar(y[0]); 762c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 763c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 764c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 765c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 766c4762a1bSJed Brown PetscFunctionReturn(0); 767c4762a1bSJed Brown } 768c4762a1bSJed Brown 769c4762a1bSJed Brown PetscErrorCode IJacobian_Kulikov2013I(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 770c4762a1bSJed Brown { 771c4762a1bSJed Brown PetscErrorCode ierr; 772c4762a1bSJed Brown const PetscScalar *y; 773c4762a1bSJed Brown PetscInt row[4] = {0,1,2,3}; 774c4762a1bSJed Brown PetscScalar value[4][4]; 775c4762a1bSJed Brown PetscScalar m1,m2; 776c4762a1bSJed Brown 777c4762a1bSJed Brown PetscFunctionBegin; 778c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 779c4762a1bSJed Brown m1=(2.*t*y[3])/(5.*PetscPowScalar(y[1],4./5.)); 780c4762a1bSJed Brown m2=2.*t*PetscPowScalar(y[1],1./5.); 781c4762a1bSJed Brown value[0][0] = a ; value[0][1] = m1; value[0][2] = 0.; value[0][3] = m2; 782c4762a1bSJed Brown m1=50.*t*y[3]*PetscExpScalar(5.0*(y[2]-1.)); 783c4762a1bSJed Brown m2=10.*t*PetscExpScalar(5.0*(y[2]-1.)); 784c4762a1bSJed Brown value[1][0] = 0.; value[1][1] = a ; value[1][2] = m1; value[1][3] = m2; 785c4762a1bSJed Brown value[2][0] = 0.; value[2][1] = 0.; value[2][2] = a; value[2][3] = 2*t; 786c4762a1bSJed Brown value[3][0] = -2.*t/y[0];value[3][1] = 0.; value[3][2] = 0.; value[3][3] = a; 787c4762a1bSJed Brown ierr = MatSetValues(A,4,&row[0],4,&row[0],&value[0][0],INSERT_VALUES);CHKERRQ(ierr); 788c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 789c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 790c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 791c4762a1bSJed Brown PetscFunctionReturn(0); 792c4762a1bSJed Brown } 793c4762a1bSJed Brown 794c4762a1bSJed Brown 795c4762a1bSJed Brown /* Hull, 1972, Problem C1 */ 796c4762a1bSJed Brown 797c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec F, void *s) 798c4762a1bSJed Brown { 799c4762a1bSJed Brown PetscErrorCode ierr; 800c4762a1bSJed Brown PetscScalar *f; 801c4762a1bSJed Brown const PetscScalar *y; 802c4762a1bSJed Brown PetscInt N,i; 803c4762a1bSJed Brown 804c4762a1bSJed Brown PetscFunctionBegin; 805c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 806c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 807c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 808c4762a1bSJed Brown f[0] = -y[0]; 809c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 810c4762a1bSJed Brown f[i] = y[i-1] - y[i]; 811c4762a1bSJed Brown } 812c4762a1bSJed Brown f[N-1] = y[N-2]; 813c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 814c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 815c4762a1bSJed Brown PetscFunctionReturn(0); 816c4762a1bSJed Brown } 817c4762a1bSJed Brown 818c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 819c4762a1bSJed Brown { 820c4762a1bSJed Brown PetscErrorCode ierr; 821c4762a1bSJed Brown PetscScalar *f; 822c4762a1bSJed Brown const PetscScalar *y; 823c4762a1bSJed Brown PetscInt N,i; 824c4762a1bSJed Brown 825c4762a1bSJed Brown PetscFunctionBegin; 826c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 827c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 828c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 829c4762a1bSJed Brown f[0] = -y[0]; 830c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 831c4762a1bSJed Brown f[i] = y[i-1] - y[i]; 832c4762a1bSJed Brown } 833c4762a1bSJed Brown f[N-1] = y[N-2]; 834c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 835c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 836c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 837c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 838c4762a1bSJed Brown PetscFunctionReturn(0); 839c4762a1bSJed Brown } 840c4762a1bSJed Brown 841c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 842c4762a1bSJed Brown { 843c4762a1bSJed Brown PetscErrorCode ierr; 844c4762a1bSJed Brown const PetscScalar *y; 845c4762a1bSJed Brown PetscInt N,i,col[2]; 846c4762a1bSJed Brown PetscScalar value[2]; 847c4762a1bSJed Brown 848c4762a1bSJed Brown PetscFunctionBegin; 849c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 850c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 851c4762a1bSJed Brown i = 0; 852c4762a1bSJed Brown value[0] = a+1; col[0] = 0; 853c4762a1bSJed Brown value[1] = 0; col[1] = 1; 854c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 855c4762a1bSJed Brown for (i = 0; i < N; i++) { 856c4762a1bSJed Brown value[0] = -1; col[0] = i-1; 857c4762a1bSJed Brown value[1] = a+1; col[1] = i; 858c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 859c4762a1bSJed Brown } 860c4762a1bSJed Brown i = N-1; 861c4762a1bSJed Brown value[0] = -1; col[0] = N-2; 862c4762a1bSJed Brown value[1] = a; col[1] = N-1; 863c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 864c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 865c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 866c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 867c4762a1bSJed Brown PetscFunctionReturn(0); 868c4762a1bSJed Brown } 869c4762a1bSJed Brown 870c4762a1bSJed Brown /* Hull, 1972, Problem C2 */ 871c4762a1bSJed Brown 872c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec F, void *s) 873c4762a1bSJed Brown { 874c4762a1bSJed Brown PetscErrorCode ierr; 875c4762a1bSJed Brown const PetscScalar *y; 876c4762a1bSJed Brown PetscScalar *f; 877c4762a1bSJed Brown PetscInt N,i; 878c4762a1bSJed Brown 879c4762a1bSJed Brown PetscFunctionBegin; 880c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 881c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 882c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 883c4762a1bSJed Brown f[0] = -y[0]; 884c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 885c4762a1bSJed Brown f[i] = (PetscReal)i*y[i-1] - (PetscReal)(i+1)*y[i]; 886c4762a1bSJed Brown } 887c4762a1bSJed Brown f[N-1] = (PetscReal)(N-1)*y[N-2]; 888c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 889c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 890c4762a1bSJed Brown PetscFunctionReturn(0); 891c4762a1bSJed Brown } 892c4762a1bSJed Brown 893c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 894c4762a1bSJed Brown { 895c4762a1bSJed Brown PetscErrorCode ierr; 896c4762a1bSJed Brown PetscScalar *f; 897c4762a1bSJed Brown const PetscScalar *y; 898c4762a1bSJed Brown PetscInt N,i; 899c4762a1bSJed Brown 900c4762a1bSJed Brown PetscFunctionBegin; 901c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 902c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 903c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 904c4762a1bSJed Brown f[0] = -y[0]; 905c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 906c4762a1bSJed Brown f[i] = (PetscReal)i*y[i-1] - (PetscReal)(i+1)*y[i]; 907c4762a1bSJed Brown } 908c4762a1bSJed Brown f[N-1] = (PetscReal)(N-1)*y[N-2]; 909c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 910c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 911c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 912c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 913c4762a1bSJed Brown PetscFunctionReturn(0); 914c4762a1bSJed Brown } 915c4762a1bSJed Brown 916c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 917c4762a1bSJed Brown { 918c4762a1bSJed Brown PetscErrorCode ierr; 919c4762a1bSJed Brown const PetscScalar *y; 920c4762a1bSJed Brown PetscInt N,i,col[2]; 921c4762a1bSJed Brown PetscScalar value[2]; 922c4762a1bSJed Brown 923c4762a1bSJed Brown PetscFunctionBegin; 924c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 925c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 926c4762a1bSJed Brown i = 0; 927c4762a1bSJed Brown value[0] = a+1; col[0] = 0; 928c4762a1bSJed Brown value[1] = 0; col[1] = 1; 929c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 930c4762a1bSJed Brown for (i = 0; i < N; i++) { 931c4762a1bSJed Brown value[0] = -(PetscReal) i; col[0] = i-1; 932c4762a1bSJed Brown value[1] = a+(PetscReal)(i+1); col[1] = i; 933c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 934c4762a1bSJed Brown } 935c4762a1bSJed Brown i = N-1; 936c4762a1bSJed Brown value[0] = -(PetscReal) (N-1); col[0] = N-2; 937c4762a1bSJed Brown value[1] = a; col[1] = N-1; 938c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);CHKERRQ(ierr); 939c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 940c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 941c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 942c4762a1bSJed Brown PetscFunctionReturn(0); 943c4762a1bSJed Brown } 944c4762a1bSJed Brown 945c4762a1bSJed Brown /* Hull, 1972, Problem C3 and C4 */ 946c4762a1bSJed Brown 947c4762a1bSJed Brown PetscErrorCode RHSFunction_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec F, void *s) 948c4762a1bSJed Brown { 949c4762a1bSJed Brown PetscErrorCode ierr; 950c4762a1bSJed Brown PetscScalar *f; 951c4762a1bSJed Brown const PetscScalar *y; 952c4762a1bSJed Brown PetscInt N,i; 953c4762a1bSJed Brown 954c4762a1bSJed Brown PetscFunctionBegin; 955c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 956c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 957c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 958c4762a1bSJed Brown f[0] = -2.0*y[0] + y[1]; 959c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 960c4762a1bSJed Brown f[i] = y[i-1] - 2.0*y[i] + y[i+1]; 961c4762a1bSJed Brown } 962c4762a1bSJed Brown f[N-1] = y[N-2] - 2.0*y[N-1]; 963c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 964c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 965c4762a1bSJed Brown PetscFunctionReturn(0); 966c4762a1bSJed Brown } 967c4762a1bSJed Brown 968c4762a1bSJed Brown PetscErrorCode IFunction_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s) 969c4762a1bSJed Brown { 970c4762a1bSJed Brown PetscErrorCode ierr; 971c4762a1bSJed Brown PetscScalar *f; 972c4762a1bSJed Brown const PetscScalar *y; 973c4762a1bSJed Brown PetscInt N,i; 974c4762a1bSJed Brown 975c4762a1bSJed Brown PetscFunctionBegin; 976c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 977c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 978c4762a1bSJed Brown ierr = VecGetArray(F,&f);CHKERRQ(ierr); 979c4762a1bSJed Brown f[0] = -2.0*y[0] + y[1]; 980c4762a1bSJed Brown for (i = 1; i < N-1; i++) { 981c4762a1bSJed Brown f[i] = y[i-1] - 2.0*y[i] + y[i+1]; 982c4762a1bSJed Brown } 983c4762a1bSJed Brown f[N-1] = y[N-2] - 2.0*y[N-1]; 984c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 985c4762a1bSJed Brown ierr = VecRestoreArray(F,&f);CHKERRQ(ierr); 986c4762a1bSJed Brown /* Left hand side = ydot - f(y) */ 987c4762a1bSJed Brown ierr = VecAYPX(F,-1.0,Ydot);CHKERRQ(ierr); 988c4762a1bSJed Brown PetscFunctionReturn(0); 989c4762a1bSJed Brown } 990c4762a1bSJed Brown 991c4762a1bSJed Brown PetscErrorCode IJacobian_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s) 992c4762a1bSJed Brown { 993c4762a1bSJed Brown PetscErrorCode ierr; 994c4762a1bSJed Brown const PetscScalar *y; 995c4762a1bSJed Brown PetscScalar value[3]; 996c4762a1bSJed Brown PetscInt N,i,col[3]; 997c4762a1bSJed Brown 998c4762a1bSJed Brown PetscFunctionBegin; 999c4762a1bSJed Brown ierr = VecGetSize (Y,&N);CHKERRQ(ierr); 1000c4762a1bSJed Brown ierr = VecGetArrayRead(Y,&y);CHKERRQ(ierr); 1001c4762a1bSJed Brown for (i = 0; i < N; i++) { 1002c4762a1bSJed Brown if (i == 0) { 1003c4762a1bSJed Brown value[0] = a+2; col[0] = i; 1004c4762a1bSJed Brown value[1] = -1; col[1] = i+1; 1005c4762a1bSJed Brown value[2] = 0; col[2] = i+2; 1006c4762a1bSJed Brown } else if (i == N-1) { 1007c4762a1bSJed Brown value[0] = 0; col[0] = i-2; 1008c4762a1bSJed Brown value[1] = -1; col[1] = i-1; 1009c4762a1bSJed Brown value[2] = a+2; col[2] = i; 1010c4762a1bSJed Brown } else { 1011c4762a1bSJed Brown value[0] = -1; col[0] = i-1; 1012c4762a1bSJed Brown value[1] = a+2; col[1] = i; 1013c4762a1bSJed Brown value[2] = -1; col[2] = i+1; 1014c4762a1bSJed Brown } 1015c4762a1bSJed Brown ierr = MatSetValues(A,1,&i,3,col,value,INSERT_VALUES);CHKERRQ(ierr); 1016c4762a1bSJed Brown } 1017c4762a1bSJed Brown ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1018c4762a1bSJed Brown ierr = MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1019c4762a1bSJed Brown ierr = VecRestoreArrayRead(Y,&y);CHKERRQ(ierr); 1020c4762a1bSJed Brown PetscFunctionReturn(0); 1021c4762a1bSJed Brown } 1022c4762a1bSJed Brown 1023c4762a1bSJed Brown /***************************************************************************/ 1024c4762a1bSJed Brown 1025c4762a1bSJed Brown /* Sets the initial solution for the IVP and sets up the function pointers*/ 1026c4762a1bSJed Brown PetscErrorCode Initialize(Vec Y, void* s) 1027c4762a1bSJed Brown { 1028c4762a1bSJed Brown PetscErrorCode ierr; 1029c4762a1bSJed Brown char *p = (char*) s; 1030c4762a1bSJed Brown PetscScalar *y; 1031c4762a1bSJed Brown PetscReal t0; 1032c4762a1bSJed Brown PetscInt N = GetSize((const char *)s); 1033c4762a1bSJed Brown PetscBool flg; 1034c4762a1bSJed Brown 1035c4762a1bSJed Brown PetscFunctionBegin; 1036c4762a1bSJed Brown VecZeroEntries(Y); 1037c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1038c4762a1bSJed Brown if (!strcmp(p,"hull1972a1")) { 1039c4762a1bSJed Brown y[0] = 1.0; 1040c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972A1; 1041c4762a1bSJed Brown RHSJacobian = RHSJacobian_Hull1972A1; 1042c4762a1bSJed Brown IFunction = IFunction_Hull1972A1; 1043c4762a1bSJed Brown IJacobian = IJacobian_Hull1972A1; 1044c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a2")) { 1045c4762a1bSJed Brown y[0] = 1.0; 1046c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972A2; 1047c4762a1bSJed Brown RHSJacobian = RHSJacobian_Hull1972A2; 1048c4762a1bSJed Brown IFunction = IFunction_Hull1972A2; 1049c4762a1bSJed Brown IJacobian = IJacobian_Hull1972A2; 1050c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a3")) { 1051c4762a1bSJed Brown y[0] = 1.0; 1052c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972A3; 1053c4762a1bSJed Brown RHSJacobian = RHSJacobian_Hull1972A3; 1054c4762a1bSJed Brown IFunction = IFunction_Hull1972A3; 1055c4762a1bSJed Brown IJacobian = IJacobian_Hull1972A3; 1056c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a4")) { 1057c4762a1bSJed Brown y[0] = 1.0; 1058c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972A4; 1059c4762a1bSJed Brown RHSJacobian = RHSJacobian_Hull1972A4; 1060c4762a1bSJed Brown IFunction = IFunction_Hull1972A4; 1061c4762a1bSJed Brown IJacobian = IJacobian_Hull1972A4; 1062c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a5")) { 1063c4762a1bSJed Brown y[0] = 4.0; 1064c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972A5; 1065c4762a1bSJed Brown RHSJacobian = RHSJacobian_Hull1972A5; 1066c4762a1bSJed Brown IFunction = IFunction_Hull1972A5; 1067c4762a1bSJed Brown IJacobian = IJacobian_Hull1972A5; 1068c4762a1bSJed Brown } else if (!strcmp(p,"hull1972b1")) { 1069c4762a1bSJed Brown y[0] = 1.0; 1070c4762a1bSJed Brown y[1] = 3.0; 1071c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972B1; 1072c4762a1bSJed Brown IFunction = IFunction_Hull1972B1; 1073c4762a1bSJed Brown IJacobian = IJacobian_Hull1972B1; 1074c4762a1bSJed Brown } else if (!strcmp(p,"hull1972b2")) { 1075c4762a1bSJed Brown y[0] = 2.0; 1076c4762a1bSJed Brown y[1] = 0.0; 1077c4762a1bSJed Brown y[2] = 1.0; 1078c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972B2; 1079c4762a1bSJed Brown IFunction = IFunction_Hull1972B2; 1080c4762a1bSJed Brown IJacobian = IJacobian_Hull1972B2; 1081c4762a1bSJed Brown } else if (!strcmp(p,"hull1972b3")) { 1082c4762a1bSJed Brown y[0] = 1.0; 1083c4762a1bSJed Brown y[1] = 0.0; 1084c4762a1bSJed Brown y[2] = 0.0; 1085c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972B3; 1086c4762a1bSJed Brown IFunction = IFunction_Hull1972B3; 1087c4762a1bSJed Brown IJacobian = IJacobian_Hull1972B3; 1088c4762a1bSJed Brown } else if (!strcmp(p,"hull1972b4")) { 1089c4762a1bSJed Brown y[0] = 3.0; 1090c4762a1bSJed Brown y[1] = 0.0; 1091c4762a1bSJed Brown y[2] = 0.0; 1092c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972B4; 1093c4762a1bSJed Brown IFunction = IFunction_Hull1972B4; 1094c4762a1bSJed Brown IJacobian = IJacobian_Hull1972B4; 1095c4762a1bSJed Brown } else if (!strcmp(p,"hull1972b5")) { 1096c4762a1bSJed Brown y[0] = 0.0; 1097c4762a1bSJed Brown y[1] = 1.0; 1098c4762a1bSJed Brown y[2] = 1.0; 1099c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972B5; 1100c4762a1bSJed Brown IFunction = IFunction_Hull1972B5; 1101c4762a1bSJed Brown IJacobian = IJacobian_Hull1972B5; 1102c4762a1bSJed Brown } else if (!strcmp(p,"kulik2013i")) { 1103c4762a1bSJed Brown t0=0.; 1104c4762a1bSJed Brown y[0] = PetscExpReal(PetscSinReal(t0*t0)); 1105c4762a1bSJed Brown y[1] = PetscExpReal(5.*PetscSinReal(t0*t0)); 1106c4762a1bSJed Brown y[2] = PetscSinReal(t0*t0)+1.0; 1107c4762a1bSJed Brown y[3] = PetscCosReal(t0*t0); 1108c4762a1bSJed Brown RHSFunction = RHSFunction_Kulikov2013I; 1109c4762a1bSJed Brown RHSJacobian = RHSJacobian_Kulikov2013I; 1110c4762a1bSJed Brown IFunction = IFunction_Kulikov2013I; 1111c4762a1bSJed Brown IJacobian = IJacobian_Kulikov2013I; 1112c4762a1bSJed Brown } else if (!strcmp(p,"hull1972c1")) { 1113c4762a1bSJed Brown y[0] = 1.0; 1114c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972C1; 1115c4762a1bSJed Brown IFunction = IFunction_Hull1972C1; 1116c4762a1bSJed Brown IJacobian = IJacobian_Hull1972C1; 1117c4762a1bSJed Brown } else if (!strcmp(p,"hull1972c2")) { 1118c4762a1bSJed Brown y[0] = 1.0; 1119c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972C2; 1120c4762a1bSJed Brown IFunction = IFunction_Hull1972C2; 1121c4762a1bSJed Brown IJacobian = IJacobian_Hull1972C2; 1122c4762a1bSJed Brown } else if ((!strcmp(p,"hull1972c3")) 1123c4762a1bSJed Brown ||(!strcmp(p,"hull1972c4"))){ 1124c4762a1bSJed Brown y[0] = 1.0; 1125c4762a1bSJed Brown RHSFunction = RHSFunction_Hull1972C34; 1126c4762a1bSJed Brown IFunction = IFunction_Hull1972C34; 1127c4762a1bSJed Brown IJacobian = IJacobian_Hull1972C34; 1128c4762a1bSJed Brown } 1129c4762a1bSJed Brown ierr = PetscOptionsGetScalarArray(NULL,NULL,"-yinit",y,&N,&flg);CHKERRQ(ierr); 1130c4762a1bSJed Brown if ((N != GetSize((const char*)s)) && flg) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_SIZ,"Number of initial values %D does not match problem size %D.\n",N,GetSize((const char*)s)); 1131c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1132c4762a1bSJed Brown PetscFunctionReturn(0); 1133c4762a1bSJed Brown } 1134c4762a1bSJed Brown 1135c4762a1bSJed Brown /* Calculates the exact solution to problems that have one */ 1136c4762a1bSJed Brown PetscErrorCode ExactSolution(Vec Y, void* s, PetscReal t, PetscBool *flag) 1137c4762a1bSJed Brown { 1138c4762a1bSJed Brown PetscErrorCode ierr; 1139c4762a1bSJed Brown char *p = (char*) s; 1140c4762a1bSJed Brown PetscScalar *y; 1141c4762a1bSJed Brown 1142c4762a1bSJed Brown PetscFunctionBegin; 1143c4762a1bSJed Brown if (!strcmp(p,"hull1972a1")) { 1144c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1145c4762a1bSJed Brown y[0] = PetscExpReal(-t); 1146c4762a1bSJed Brown *flag = PETSC_TRUE; 1147c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1148c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a2")) { 1149c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1150c4762a1bSJed Brown y[0] = 1.0/PetscSqrtReal(t+1); 1151c4762a1bSJed Brown *flag = PETSC_TRUE; 1152c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1153c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a3")) { 1154c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1155c4762a1bSJed Brown y[0] = PetscExpReal(PetscSinReal(t)); 1156c4762a1bSJed Brown *flag = PETSC_TRUE; 1157c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1158c4762a1bSJed Brown } else if (!strcmp(p,"hull1972a4")) { 1159c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1160c4762a1bSJed Brown y[0] = 20.0/(1+19.0*PetscExpReal(-t/4.0)); 1161c4762a1bSJed Brown *flag = PETSC_TRUE; 1162c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1163c4762a1bSJed Brown } else if (!strcmp(p,"kulik2013i")) { 1164c4762a1bSJed Brown ierr = VecGetArray(Y,&y);CHKERRQ(ierr); 1165c4762a1bSJed Brown y[0] = PetscExpReal(PetscSinReal(t*t)); 1166c4762a1bSJed Brown y[1] = PetscExpReal(5.*PetscSinReal(t*t)); 1167c4762a1bSJed Brown y[2] = PetscSinReal(t*t)+1.0; 1168c4762a1bSJed Brown y[3] = PetscCosReal(t*t); 1169c4762a1bSJed Brown *flag = PETSC_TRUE; 1170c4762a1bSJed Brown ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr); 1171c4762a1bSJed Brown } else { 1172c4762a1bSJed Brown ierr = VecSet(Y,0);CHKERRQ(ierr); 1173c4762a1bSJed Brown *flag = PETSC_FALSE; 1174c4762a1bSJed Brown } 1175c4762a1bSJed Brown PetscFunctionReturn(0); 1176c4762a1bSJed Brown } 1177c4762a1bSJed Brown 1178c4762a1bSJed Brown /* Solves the specified ODE and computes the error if exact solution is available */ 1179c4762a1bSJed Brown PetscErrorCode SolveODE(char* ptype, PetscReal dt, PetscReal tfinal, PetscInt maxiter, PetscReal *error, PetscBool *exact_flag) 1180c4762a1bSJed Brown { 1181c4762a1bSJed Brown PetscErrorCode ierr; /* Error code */ 1182c4762a1bSJed Brown TS ts; /* time-integrator */ 1183c4762a1bSJed Brown Vec Y; /* Solution vector */ 1184c4762a1bSJed Brown Vec Yex; /* Exact solution */ 1185c4762a1bSJed Brown PetscInt N; /* Size of the system of equations */ 1186c4762a1bSJed Brown TSType time_scheme; /* Type of time-integration scheme */ 1187c4762a1bSJed Brown Mat Jac = NULL; /* Jacobian matrix */ 1188c4762a1bSJed Brown Vec Yerr; /* Auxiliary solution vector */ 1189c4762a1bSJed Brown PetscReal err_norm; /* Estimated error norm */ 1190c4762a1bSJed Brown PetscReal final_time; /* Actual final time from the integrator */ 1191c4762a1bSJed Brown 1192c4762a1bSJed Brown PetscFunctionBegin; 1193c4762a1bSJed Brown N = GetSize((const char *)&ptype[0]); 1194c4762a1bSJed Brown if (N < 0) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_SIZ,"Illegal problem specification.\n"); 1195c4762a1bSJed Brown ierr = VecCreate(PETSC_COMM_WORLD,&Y);CHKERRQ(ierr); 1196c4762a1bSJed Brown ierr = VecSetSizes(Y,N,PETSC_DECIDE);CHKERRQ(ierr); 1197c4762a1bSJed Brown ierr = VecSetUp(Y);CHKERRQ(ierr); 1198c4762a1bSJed Brown ierr = VecSet(Y,0);CHKERRQ(ierr); 1199c4762a1bSJed Brown 1200c4762a1bSJed Brown /* Initialize the problem */ 1201c4762a1bSJed Brown ierr = Initialize(Y,&ptype[0]);CHKERRQ(ierr); 1202c4762a1bSJed Brown 1203c4762a1bSJed Brown /* Create and initialize the time-integrator */ 1204c4762a1bSJed Brown ierr = TSCreate(PETSC_COMM_WORLD,&ts);CHKERRQ(ierr); 1205c4762a1bSJed Brown /* Default time integration options */ 1206c4762a1bSJed Brown ierr = TSSetType(ts,TSRK);CHKERRQ(ierr); 1207c4762a1bSJed Brown ierr = TSSetMaxSteps(ts,maxiter);CHKERRQ(ierr); 1208c4762a1bSJed Brown ierr = TSSetMaxTime(ts,tfinal);CHKERRQ(ierr); 1209c4762a1bSJed Brown ierr = TSSetTimeStep(ts,dt);CHKERRQ(ierr); 1210c4762a1bSJed Brown ierr = TSSetExactFinalTime(ts,TS_EXACTFINALTIME_MATCHSTEP);CHKERRQ(ierr); 1211c4762a1bSJed Brown /* Read command line options for time integration */ 1212c4762a1bSJed Brown ierr = TSSetFromOptions(ts);CHKERRQ(ierr); 1213c4762a1bSJed Brown /* Set solution vector */ 1214c4762a1bSJed Brown ierr = TSSetSolution(ts,Y);CHKERRQ(ierr); 1215c4762a1bSJed Brown /* Specify left/right-hand side functions */ 1216c4762a1bSJed Brown ierr = TSGetType(ts,&time_scheme);CHKERRQ(ierr); 1217c4762a1bSJed Brown 1218c4762a1bSJed Brown if ((!strcmp(time_scheme,TSEULER)) || (!strcmp(time_scheme,TSRK)) || (!strcmp(time_scheme,TSSSP) || (!strcmp(time_scheme,TSGLEE)))) { 1219c4762a1bSJed Brown /* Explicit time-integration -> specify right-hand side function ydot = f(y) */ 1220c4762a1bSJed Brown ierr = TSSetRHSFunction(ts,NULL,RHSFunction,&ptype[0]);CHKERRQ(ierr); 1221c4762a1bSJed Brown ierr = MatCreate(PETSC_COMM_WORLD,&Jac);CHKERRQ(ierr); 1222c4762a1bSJed Brown ierr = MatSetSizes(Jac,PETSC_DECIDE,PETSC_DECIDE,N,N);CHKERRQ(ierr); 1223c4762a1bSJed Brown ierr = MatSetFromOptions(Jac);CHKERRQ(ierr); 1224c4762a1bSJed Brown ierr = MatSetUp(Jac);CHKERRQ(ierr); 1225c4762a1bSJed Brown ierr = TSSetRHSJacobian(ts,Jac,Jac,RHSJacobian,&ptype[0]);CHKERRQ(ierr); 1226c4762a1bSJed Brown } else if ((!strcmp(time_scheme,TSTHETA)) || (!strcmp(time_scheme,TSBEULER)) || (!strcmp(time_scheme,TSCN)) || (!strcmp(time_scheme,TSALPHA)) || (!strcmp(time_scheme,TSARKIMEX))) { 1227c4762a1bSJed Brown /* Implicit time-integration -> specify left-hand side function ydot-f(y) = 0 */ 1228c4762a1bSJed Brown /* and its Jacobian function */ 1229c4762a1bSJed Brown ierr = TSSetIFunction(ts,NULL,IFunction,&ptype[0]);CHKERRQ(ierr); 1230c4762a1bSJed Brown ierr = MatCreate(PETSC_COMM_WORLD,&Jac);CHKERRQ(ierr); 1231c4762a1bSJed Brown ierr = MatSetSizes(Jac,PETSC_DECIDE,PETSC_DECIDE,N,N);CHKERRQ(ierr); 1232c4762a1bSJed Brown ierr = MatSetFromOptions(Jac);CHKERRQ(ierr); 1233c4762a1bSJed Brown ierr = MatSetUp(Jac);CHKERRQ(ierr); 1234c4762a1bSJed Brown ierr = TSSetIJacobian(ts,Jac,Jac,IJacobian,&ptype[0]);CHKERRQ(ierr); 1235c4762a1bSJed Brown } 1236c4762a1bSJed Brown 1237c4762a1bSJed Brown /* Solve */ 1238c4762a1bSJed Brown ierr = TSSolve(ts,Y);CHKERRQ(ierr); 1239c4762a1bSJed Brown ierr = TSGetTime(ts,&final_time);CHKERRQ(ierr); 1240c4762a1bSJed Brown 1241c4762a1bSJed Brown /* Get the estimated error, if available */ 1242c4762a1bSJed Brown ierr = VecDuplicate(Y,&Yerr);CHKERRQ(ierr); 1243c4762a1bSJed Brown ierr = VecZeroEntries(Yerr);CHKERRQ(ierr); 1244c4762a1bSJed Brown ierr = TSGetTimeError(ts,0,&Yerr);CHKERRQ(ierr); 1245c4762a1bSJed Brown ierr = VecNorm(Yerr,NORM_2,&err_norm);CHKERRQ(ierr); 1246c4762a1bSJed Brown ierr = VecDestroy(&Yerr);CHKERRQ(ierr); 1247c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Estimated Error = %E.\n",err_norm);CHKERRQ(ierr); 1248c4762a1bSJed Brown 1249c4762a1bSJed Brown /* Exact solution */ 1250c4762a1bSJed Brown ierr = VecDuplicate(Y,&Yex);CHKERRQ(ierr); 1251c4762a1bSJed Brown if (PetscAbsScalar(final_time-tfinal)>2.*PETSC_MACHINE_EPSILON) { 1252c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Note: There is a difference between the prescribed final time %g and the actual final time, %g.\n",(double)tfinal,(double)final_time);CHKERRQ(ierr); 1253c4762a1bSJed Brown } 1254c4762a1bSJed Brown ierr = ExactSolution(Yex,&ptype[0],final_time,exact_flag);CHKERRQ(ierr); 1255c4762a1bSJed Brown 1256c4762a1bSJed Brown /* Calculate Error */ 1257c4762a1bSJed Brown ierr = VecAYPX(Yex,-1.0,Y);CHKERRQ(ierr); 1258c4762a1bSJed Brown ierr = VecNorm(Yex,NORM_2,error);CHKERRQ(ierr); 1259c4762a1bSJed Brown *error = PetscSqrtReal(((*error)*(*error))/N); 1260c4762a1bSJed Brown 1261c4762a1bSJed Brown /* Clean up and finalize */ 1262c4762a1bSJed Brown ierr = MatDestroy(&Jac);CHKERRQ(ierr); 1263c4762a1bSJed Brown ierr = TSDestroy(&ts);CHKERRQ(ierr); 1264c4762a1bSJed Brown ierr = VecDestroy(&Yex);CHKERRQ(ierr); 1265c4762a1bSJed Brown ierr = VecDestroy(&Y);CHKERRQ(ierr); 1266c4762a1bSJed Brown 1267c4762a1bSJed Brown PetscFunctionReturn(0); 1268c4762a1bSJed Brown } 1269c4762a1bSJed Brown 1270c4762a1bSJed Brown int main(int argc, char **argv) 1271c4762a1bSJed Brown { 1272c4762a1bSJed Brown PetscErrorCode ierr; /* Error code */ 1273c4762a1bSJed Brown char ptype[256] = "hull1972a1"; /* Problem specification */ 1274c4762a1bSJed Brown PetscInt n_refine = 1; /* Number of refinement levels for convergence analysis */ 1275c4762a1bSJed Brown PetscReal refine_fac = 2.0; /* Refinement factor for dt */ 1276c4762a1bSJed Brown PetscReal dt_initial = 0.01; /* Initial default value of dt */ 1277c4762a1bSJed Brown PetscReal dt; 1278c4762a1bSJed Brown PetscReal tfinal = 20.0; /* Final time for the time-integration */ 1279c4762a1bSJed Brown PetscInt maxiter = 100000; /* Maximum number of time-integration iterations */ 1280c4762a1bSJed Brown PetscReal *error; /* Array to store the errors for convergence analysis */ 1281c4762a1bSJed Brown PetscMPIInt size; /* No of processors */ 1282c4762a1bSJed Brown PetscBool flag; /* Flag denoting availability of exact solution */ 1283c4762a1bSJed Brown PetscInt r; 1284c4762a1bSJed Brown 1285c4762a1bSJed Brown /* Initialize program */ 1286c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 1287c4762a1bSJed Brown 1288c4762a1bSJed Brown /* Check if running with only 1 proc */ 1289*ffc4695bSBarry Smith ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 1290c4762a1bSJed Brown if (size>1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Only for sequential runs"); 1291c4762a1bSJed Brown 129276280437SVaclav Hapla ierr = PetscOptionsBegin(PETSC_COMM_WORLD,NULL,"ex31",NULL);CHKERRQ(ierr); 1293c4762a1bSJed Brown ierr = PetscOptionsString("-problem","Problem specification","<hull1972a1>",ptype,ptype,sizeof(ptype),NULL);CHKERRQ(ierr); 1294c4762a1bSJed Brown ierr = PetscOptionsInt("-refinement_levels","Number of refinement levels for convergence analysis","<1>",n_refine,&n_refine,NULL);CHKERRQ(ierr); 1295c4762a1bSJed Brown ierr = PetscOptionsReal("-refinement_factor","Refinement factor for dt","<2.0>",refine_fac,&refine_fac,NULL);CHKERRQ(ierr); 1296c4762a1bSJed Brown ierr = PetscOptionsReal("-dt","Time step size (for convergence analysis, initial time step)","<0.01>",dt_initial,&dt_initial,NULL);CHKERRQ(ierr); 1297c4762a1bSJed Brown ierr = PetscOptionsReal("-final_time","Final time for the time-integration","<20.0>",tfinal,&tfinal,NULL);CHKERRQ(ierr); 1298c4762a1bSJed Brown ierr = PetscOptionsEnd();CHKERRQ(ierr); 1299c4762a1bSJed Brown 1300c4762a1bSJed Brown ierr = PetscMalloc1(n_refine,&error);CHKERRQ(ierr); 1301c4762a1bSJed Brown for (r = 0,dt = dt_initial; r < n_refine; r++) { 1302c4762a1bSJed Brown error[r] = 0; 1303c4762a1bSJed Brown if (r > 0) dt /= refine_fac; 1304c4762a1bSJed Brown 1305c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Solving ODE \"%s\" with dt %f, final time %f and system size %D.\n",ptype,(double)dt,(double)tfinal,GetSize(&ptype[0]));CHKERRQ(ierr); 1306c4762a1bSJed Brown ierr = SolveODE(&ptype[0],dt,tfinal,maxiter,&error[r],&flag);CHKERRQ(ierr); 1307c4762a1bSJed Brown if (flag) { 1308c4762a1bSJed Brown /* If exact solution available for the specified ODE */ 1309c4762a1bSJed Brown if (r > 0) { 1310c4762a1bSJed Brown PetscReal conv_rate = (PetscLogReal(error[r]) - PetscLogReal(error[r-1])) / (-PetscLogReal(refine_fac)); 1311c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Error = %E,\tConvergence rate = %f.\n",(double)error[r],(double)conv_rate);CHKERRQ(ierr); 1312c4762a1bSJed Brown } else { 1313c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Error = %E.\n",error[r]);CHKERRQ(ierr); 1314c4762a1bSJed Brown } 1315c4762a1bSJed Brown } 1316c4762a1bSJed Brown } 1317c4762a1bSJed Brown ierr = PetscFree(error);CHKERRQ(ierr); 1318c4762a1bSJed Brown ierr = PetscFinalize(); 1319c4762a1bSJed Brown return ierr; 1320c4762a1bSJed Brown } 1321c4762a1bSJed Brown 1322c4762a1bSJed Brown /*TEST 1323c4762a1bSJed Brown 1324c4762a1bSJed Brown test: 1325c4762a1bSJed Brown suffix: 2 1326c4762a1bSJed Brown args: -ts_type glee -final_time 5 -ts_adapt_type none 1327c4762a1bSJed Brown timeoutfactor: 3 1328c4762a1bSJed Brown requires: !single 1329c4762a1bSJed Brown 1330c4762a1bSJed Brown test: 1331c4762a1bSJed Brown suffix: 3 1332c4762a1bSJed Brown args: -ts_type glee -final_time 5 -ts_adapt_type glee -ts_adapt_monitor -ts_max_steps 50 -problem hull1972a3 -ts_adapt_glee_use_local 1 1333c4762a1bSJed Brown timeoutfactor: 3 1334c4762a1bSJed Brown requires: !single 1335c4762a1bSJed Brown 1336c4762a1bSJed Brown test: 1337c4762a1bSJed Brown suffix: 4 1338c4762a1bSJed Brown args: -ts_type glee -final_time 5 -ts_adapt_type glee -ts_adapt_monitor -ts_max_steps 50 -problem hull1972a3 -ts_max_reject 100 -ts_adapt_glee_use_local 0 1339c4762a1bSJed Brown timeoutfactor: 3 1340c4762a1bSJed Brown requires: !single !__float128 1341c4762a1bSJed Brown 1342c4762a1bSJed Brown TEST*/ 1343