Actual source code: ex31.c
petsc-3.6.4 2016-04-12
1: static char help[] = "Solves the ordinary differential equations (IVPs) using explicit and implicit time-integration methods.\n";
3: /*
5: Concepts: TS
6: Useful command line parameters:
7: -problem <hull1972a1>: choose which problem to solve (see references
8: for complete listing of problems).
9: -ts_type <euler>: specify time-integrator
10: -ts_adapt_type <basic>: specify time-step adapting (none,basic,advanced)
11: -refinement_levels <1>: number of refinement levels for convergence analysis
12: -refinement_factor <2.0>: factor to refine time step size by for convergence analysis
13: -dt <0.01>: specify time step (initial time step for convergence analysis)
15: */
17: /*
18: List of cases and their names in the code:-
19: From Hull, T.E., Enright, W.H., Fellen, B.M., and Sedgwick, A.E.,
20: "Comparing Numerical Methods for Ordinary Differential
21: Equations", SIAM J. Numer. Anal., 9(4), 1972, pp. 603 - 635
22: A1 -> "hull1972a1" (exact solution available)
23: A2 -> "hull1972a2" (exact solution available)
24: A3 -> "hull1972a3" (exact solution available)
25: A4 -> "hull1972a4" (exact solution available)
26: A5 -> "hull1972a5"
27: B1 -> "hull1972b1"
28: B2 -> "hull1972b2"
29: B3 -> "hull1972b3"
30: B4 -> "hull1972b4"
31: B5 -> "hull1972b5"
32: C1 -> "hull1972c1"
33: C2 -> "hull1972c2"
34: C3 -> "hull1972c3"
35: C4 -> "hull1972c4"
36: */
38: #include <petscts.h>
39: #if defined(PETSC_HAVE_STRING_H)
40: #include <string.h> /* strcmp */
41: #endif
43: /* Function declarations */
44: PetscErrorCode (*RHSFunction) (TS,PetscReal,Vec,Vec,void*);
45: PetscErrorCode (*IFunction) (TS,PetscReal,Vec,Vec,Vec,void*);
46: PetscErrorCode (*IJacobian) (TS,PetscReal,Vec,Vec,PetscReal,Mat,Mat,void*);
50: /* Returns the size of the system of equations depending on problem specification */
51: PetscInt GetSize(const char *p)
52: {
54: if ((!strcmp(p,"hull1972a1"))
55: ||(!strcmp(p,"hull1972a2"))
56: ||(!strcmp(p,"hull1972a3"))
57: ||(!strcmp(p,"hull1972a4"))
58: ||(!strcmp(p,"hull1972a5")) ) PetscFunctionReturn(1);
59: else if (!strcmp(p,"hull1972b1") ) PetscFunctionReturn(2);
60: else if ((!strcmp(p,"hull1972b2"))
61: ||(!strcmp(p,"hull1972b3"))
62: ||(!strcmp(p,"hull1972b4"))
63: ||(!strcmp(p,"hull1972b5")) ) PetscFunctionReturn(3);
64: else if ((!strcmp(p,"hull1972c1"))
65: ||(!strcmp(p,"hull1972c2"))
66: ||(!strcmp(p,"hull1972c3")) ) PetscFunctionReturn(10);
67: else if (!strcmp(p,"hull1972c4") ) PetscFunctionReturn(51);
68: else PetscFunctionReturn(-1);
69: }
71: /****************************************************************/
73: /* Problem specific functions */
75: /* Hull, 1972, Problem A1 */
79: PetscErrorCode RHSFunction_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec F, void *s)
80: {
81: PetscErrorCode ierr;
82: PetscScalar *f;
83: const PetscScalar *y;
86: VecGetArrayRead(Y,&y);
87: VecGetArray(F,&f);
88: f[0] = -y[0];
89: VecRestoreArrayRead(Y,&y);
90: VecRestoreArray(F,&f);
91: return(0);
92: }
96: PetscErrorCode IFunction_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
97: {
98: PetscErrorCode ierr;
99: const PetscScalar *y;
100: PetscScalar *f;
103: VecGetArrayRead(Y,&y);
104: VecGetArray(F,&f);
105: f[0] = -y[0];
106: VecRestoreArrayRead(Y,&y);
107: VecRestoreArray(F,&f);
108: /* Left hand side = ydot - f(y) */
109: VecAYPX(F,-1.0,Ydot);
110: return(0);
111: }
115: PetscErrorCode IJacobian_Hull1972A1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
116: {
117: PetscErrorCode ierr;
118: const PetscScalar *y;
119: PetscInt row = 0,col = 0;
120: PetscScalar value = a - 1.0;
123: VecGetArrayRead(Y,&y);
124: MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);
125: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
126: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
127: VecRestoreArrayRead(Y,&y);
128: return(0);
129: }
131: /* Hull, 1972, Problem A2 */
135: PetscErrorCode RHSFunction_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec F, void *s)
136: {
137: PetscErrorCode ierr;
138: const PetscScalar *y;
139: PetscScalar *f;
142: VecGetArrayRead(Y,&y);
143: VecGetArray(F,&f);
144: f[0] = -0.5*y[0]*y[0]*y[0];
145: VecRestoreArrayRead(Y,&y);
146: VecRestoreArray(F,&f);
147: return(0);
148: }
152: PetscErrorCode IFunction_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
153: {
154: PetscErrorCode ierr;
155: PetscScalar *f;
156: const PetscScalar *y;
159: VecGetArrayRead(Y,&y);
160: VecGetArray(F,&f);
161: f[0] = -0.5*y[0]*y[0]*y[0];
162: VecRestoreArrayRead(Y,&y);
163: VecRestoreArray(F,&f);
164: /* Left hand side = ydot - f(y) */
165: VecAYPX(F,-1.0,Ydot);
166: return(0);
167: }
171: PetscErrorCode IJacobian_Hull1972A2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
172: {
173: PetscErrorCode ierr;
174: const PetscScalar *y;
175: PetscInt row = 0,col = 0;
176: PetscScalar value;
179: VecGetArrayRead(Y,&y);
180: value = a + 0.5*3.0*y[0]*y[0];
181: MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);
182: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
183: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
184: VecRestoreArrayRead(Y,&y);
185: return(0);
186: }
188: /* Hull, 1972, Problem A3 */
192: PetscErrorCode RHSFunction_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec F, void *s)
193: {
194: PetscErrorCode ierr;
195: const PetscScalar *y;
196: PetscScalar *f;
199: VecGetArrayRead(Y,&y);
200: VecGetArray(F,&f);
201: f[0] = y[0]*PetscCosReal(t);
202: VecRestoreArrayRead(Y,&y);
203: VecRestoreArray(F,&f);
204: return(0);
205: }
209: PetscErrorCode IFunction_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
210: {
211: PetscErrorCode ierr;
212: PetscScalar *f;
213: const PetscScalar *y;
216: VecGetArrayRead(Y,&y);
217: VecGetArray(F,&f);
218: f[0] = y[0]*PetscCosReal(t);
219: VecRestoreArrayRead(Y,&y);
220: VecRestoreArray(F,&f);
221: /* Left hand side = ydot - f(y) */
222: VecAYPX(F,-1.0,Ydot);
223: return(0);
224: }
228: PetscErrorCode IJacobian_Hull1972A3(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
229: {
230: PetscErrorCode ierr;
231: const PetscScalar *y;
232: PetscInt row = 0,col = 0;
233: PetscScalar value = a - PetscCosReal(t);
236: VecGetArrayRead(Y,&y);
237: MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);
238: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
239: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
240: VecRestoreArrayRead(Y,&y);
241: return(0);
242: }
244: /* Hull, 1972, Problem A4 */
248: PetscErrorCode RHSFunction_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec F, void *s)
249: {
250: PetscErrorCode ierr;
251: PetscScalar *f;
252: const PetscScalar *y;
255: VecGetArrayRead(Y,&y);
256: VecGetArray(F,&f);
257: f[0] = (0.25*y[0])*(1.0-0.05*y[0]);
258: VecRestoreArrayRead(Y,&y);
259: VecRestoreArray(F,&f);
260: return(0);
261: }
265: PetscErrorCode IFunction_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
266: {
267: PetscErrorCode ierr;
268: PetscScalar *f;
269: const PetscScalar *y;
272: VecGetArrayRead(Y,&y);
273: VecGetArray(F,&f);
274: f[0] = (0.25*y[0])*(1.0-0.05*y[0]);
275: VecRestoreArrayRead(Y,&y);
276: VecRestoreArray(F,&f);
277: /* Left hand side = ydot - f(y) */
278: VecAYPX(F,-1.0,Ydot);
279: return(0);
280: }
284: PetscErrorCode IJacobian_Hull1972A4(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
285: {
286: PetscErrorCode ierr;
287: const PetscScalar *y;
288: PetscInt row = 0,col = 0;
289: PetscScalar value;
292: VecGetArrayRead(Y,&y);
293: value = a - 0.25*(1.0-0.05*y[0]) + (0.25*y[0])*0.05;
294: MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);
295: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
296: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
297: VecRestoreArrayRead(Y,&y);
298: return(0);
299: }
301: /* Hull, 1972, Problem A5 */
305: PetscErrorCode RHSFunction_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec F, void *s)
306: {
307: PetscErrorCode ierr;
308: PetscScalar *f;
309: const PetscScalar *y;
312: VecGetArrayRead(Y,&y);
313: VecGetArray(F,&f);
314: f[0] = (y[0]-t)/(y[0]+t);
315: VecRestoreArrayRead(Y,&y);
316: VecRestoreArray(F,&f);
317: return(0);
318: }
322: PetscErrorCode IFunction_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
323: {
324: PetscErrorCode ierr;
325: PetscScalar *f;
326: const PetscScalar *y;
329: VecGetArrayRead(Y,&y);
330: VecGetArray(F,&f);
331: f[0] = (y[0]-t)/(y[0]+t);
332: VecRestoreArrayRead(Y,&y);
333: VecRestoreArray(F,&f);
334: /* Left hand side = ydot - f(y) */
335: VecAYPX(F,-1.0,Ydot);
336: return(0);
337: }
341: PetscErrorCode IJacobian_Hull1972A5(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
342: {
343: PetscErrorCode ierr;
344: const PetscScalar *y;
345: PetscInt row = 0,col = 0;
346: PetscScalar value;
349: VecGetArrayRead(Y,&y);
350: value = a - 2*t/((t+y[0])*(t+y[0]));
351: MatSetValues(A,1,&row,1,&col,&value,INSERT_VALUES);
352: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
353: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
354: VecRestoreArrayRead(Y,&y);
355: return(0);
356: }
358: /* Hull, 1972, Problem B1 */
362: PetscErrorCode RHSFunction_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec F, void *s)
363: {
364: PetscErrorCode ierr;
365: PetscScalar *f;
366: const PetscScalar *y;
369: VecGetArrayRead(Y,&y);
370: VecGetArray(F,&f);
371: f[0] = 2.0*(y[0] - y[0]*y[1]);
372: f[1] = -(y[1]-y[0]*y[1]);
373: VecRestoreArrayRead(Y,&y);
374: VecRestoreArray(F,&f);
375: return(0);
376: }
380: PetscErrorCode IFunction_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
381: {
382: PetscErrorCode ierr;
383: PetscScalar *f;
384: const PetscScalar *y;
387: VecGetArrayRead(Y,&y);
388: VecGetArray(F,&f);
389: f[0] = 2.0*(y[0] - y[0]*y[1]);
390: f[1] = -(y[1]-y[0]*y[1]);
391: VecRestoreArrayRead(Y,&y);
392: VecRestoreArray(F,&f);
393: /* Left hand side = ydot - f(y) */
394: VecAYPX(F,-1.0,Ydot);
395: return(0);
396: }
400: PetscErrorCode IJacobian_Hull1972B1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
401: {
402: PetscErrorCode ierr;
403: const PetscScalar *y;
404: PetscInt row[2] = {0,1};
405: PetscScalar value[2][2];
408: VecGetArrayRead(Y,&y);
409: value[0][0] = a - 2.0*(1.0-y[1]); value[0][1] = 2.0*y[0];
410: value[1][0] = -y[1]; value[1][1] = a + 1.0 - y[0];
411: MatSetValues(A,2,&row[0],2,&row[0],&value[0][0],INSERT_VALUES);
412: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
413: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
414: VecRestoreArrayRead(Y,&y);
415: return(0);
416: }
418: /* Hull, 1972, Problem B2 */
422: PetscErrorCode RHSFunction_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec F, void *s)
423: {
424: PetscErrorCode ierr;
425: PetscScalar *f;
426: const PetscScalar *y;
429: VecGetArrayRead(Y,&y);
430: VecGetArray(F,&f);
431: f[0] = -y[0] + y[1];
432: f[1] = y[0] - 2.0*y[1] + y[2];
433: f[2] = y[1] - y[2];
434: VecRestoreArrayRead(Y,&y);
435: VecRestoreArray(F,&f);
436: return(0);
437: }
441: PetscErrorCode IFunction_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
442: {
443: PetscErrorCode ierr;
444: PetscScalar *f;
445: const PetscScalar *y;
448: VecGetArrayRead(Y,&y);
449: VecGetArray(F,&f);
450: f[0] = -y[0] + y[1];
451: f[1] = y[0] - 2.0*y[1] + y[2];
452: f[2] = y[1] - y[2];
453: VecRestoreArrayRead(Y,&y);
454: VecRestoreArray(F,&f);
455: /* Left hand side = ydot - f(y) */
456: VecAYPX(F,-1.0,Ydot);
457: return(0);
458: }
462: PetscErrorCode IJacobian_Hull1972B2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
463: {
464: PetscErrorCode ierr;
465: const PetscScalar *y;
466: PetscInt row[3] = {0,1,2};
467: PetscScalar value[3][3];
470: VecGetArrayRead(Y,&y);
471: value[0][0] = a + 1.0; value[0][1] = -1.0; value[0][2] = 0;
472: value[1][0] = -1.0; value[1][1] = a + 2.0; value[1][2] = -1.0;
473: value[2][0] = 0; value[2][1] = -1.0; value[2][2] = a + 1.0;
474: MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);
475: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
476: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
477: VecRestoreArrayRead(Y,&y);
478: return(0);
479: }
481: /* Hull, 1972, Problem B3 */
485: PetscErrorCode RHSFunction_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec F, void *s)
486: {
487: PetscErrorCode ierr;
488: PetscScalar *f;
489: const PetscScalar *y;
492: VecGetArrayRead(Y,&y);
493: VecGetArray(F,&f);
494: f[0] = -y[0];
495: f[1] = y[0] - y[1]*y[1];
496: f[2] = y[1]*y[1];
497: VecRestoreArrayRead(Y,&y);
498: VecRestoreArray(F,&f);
499: return(0);
500: }
504: PetscErrorCode IFunction_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
505: {
506: PetscErrorCode ierr;
507: PetscScalar *f;
508: const PetscScalar *y;
511: VecGetArrayRead(Y,&y);
512: VecGetArray(F,&f);
513: f[0] = -y[0];
514: f[1] = y[0] - y[1]*y[1];
515: f[2] = y[1]*y[1];
516: VecRestoreArrayRead(Y,&y);
517: VecRestoreArray(F,&f);
518: /* Left hand side = ydot - f(y) */
519: VecAYPX(F,-1.0,Ydot);
520: return(0);
521: }
525: PetscErrorCode IJacobian_Hull1972B3(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
526: {
527: PetscErrorCode ierr;
528: const PetscScalar *y;
529: PetscInt row[3] = {0,1,2};
530: PetscScalar value[3][3];
533: VecGetArrayRead(Y,&y);
534: value[0][0] = a + 1.0; value[0][1] = 0; value[0][2] = 0;
535: value[1][0] = -1.0; value[1][1] = a + 2.0*y[1]; value[1][2] = 0;
536: value[2][0] = 0; value[2][1] = -2.0*y[1]; value[2][2] = a;
537: MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);
538: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
539: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
540: VecRestoreArrayRead(Y,&y);
541: return(0);
542: }
544: /* Hull, 1972, Problem B4 */
548: PetscErrorCode RHSFunction_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec F, void *s)
549: {
550: PetscErrorCode ierr;
551: PetscScalar *f;
552: const PetscScalar *y;
555: VecGetArrayRead(Y,&y);
556: VecGetArray(F,&f);
557: f[0] = -y[1] - y[0]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
558: f[1] = y[0] - y[1]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
559: f[2] = y[0]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
560: VecRestoreArrayRead(Y,&y);
561: VecRestoreArray(F,&f);
562: return(0);
563: }
567: PetscErrorCode IFunction_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
568: {
569: PetscErrorCode ierr;
570: PetscScalar *f;
571: const PetscScalar *y;
574: VecGetArrayRead(Y,&y);
575: VecGetArray(F,&f);
576: f[0] = -y[1] - y[0]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
577: f[1] = y[0] - y[1]*y[2]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
578: f[2] = y[0]/PetscSqrtScalar(y[0]*y[0]+y[1]*y[1]);
579: VecRestoreArrayRead(Y,&y);
580: VecRestoreArray(F,&f);
581: /* Left hand side = ydot - f(y) */
582: VecAYPX(F,-1.0,Ydot);
583: return(0);
584: }
588: PetscErrorCode IJacobian_Hull1972B4(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
589: {
590: PetscErrorCode ierr;
591: const PetscScalar *y;
592: PetscInt row[3] = {0,1,2};
593: PetscScalar value[3][3],fac,fac2;
596: VecGetArrayRead(Y,&y);
597: fac = PetscPowScalar(y[0]*y[0]+y[1]*y[1],-1.5);
598: fac2 = PetscPowScalar(y[0]*y[0]+y[1]*y[1],-0.5);
599: value[0][0] = a + (y[1]*y[1]*y[2])*fac;
600: value[0][1] = 1.0 - (y[0]*y[1]*y[2])*fac;
601: value[0][2] = y[0]*fac2;
602: value[1][0] = -1.0 - y[0]*y[1]*y[2]*fac;
603: value[1][1] = a + y[0]*y[0]*y[2]*fac;
604: value[1][2] = y[1]*fac2;
605: value[2][0] = -y[1]*y[1]*fac;
606: value[2][1] = y[0]*y[1]*fac;
607: value[2][2] = a;
608: MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);
609: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
610: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
611: VecRestoreArrayRead(Y,&y);
612: return(0);
613: }
615: /* Hull, 1972, Problem B5 */
619: PetscErrorCode RHSFunction_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec F, void *s)
620: {
621: PetscErrorCode ierr;
622: PetscScalar *f;
623: const PetscScalar *y;
626: VecGetArrayRead(Y,&y);
627: VecGetArray(F,&f);
628: f[0] = y[1]*y[2];
629: f[1] = -y[0]*y[2];
630: f[2] = -0.51*y[0]*y[1];
631: VecRestoreArrayRead(Y,&y);
632: VecRestoreArray(F,&f);
633: return(0);
634: }
638: PetscErrorCode IFunction_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
639: {
640: PetscErrorCode ierr;
641: PetscScalar *f;
642: const PetscScalar *y;
645: VecGetArrayRead(Y,&y);
646: VecGetArray(F,&f);
647: f[0] = y[1]*y[2];
648: f[1] = -y[0]*y[2];
649: f[2] = -0.51*y[0]*y[1];
650: VecRestoreArrayRead(Y,&y);
651: VecRestoreArray(F,&f);
652: /* Left hand side = ydot - f(y) */
653: VecAYPX(F,-1.0,Ydot);
654: return(0);
655: }
659: PetscErrorCode IJacobian_Hull1972B5(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
660: {
661: PetscErrorCode ierr;
662: const PetscScalar *y;
663: PetscInt row[3] = {0,1,2};
664: PetscScalar value[3][3];
667: VecGetArrayRead(Y,&y);
668: value[0][0] = a; value[0][1] = -y[2]; value[0][2] = -y[1];
669: value[1][0] = y[2]; value[1][1] = a; value[1][2] = y[0];
670: value[2][0] = 0.51*y[1]; value[2][1] = 0.51*y[0]; value[2][2] = a;
671: MatSetValues(A,3,&row[0],3,&row[0],&value[0][0],INSERT_VALUES);
672: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
673: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
674: VecRestoreArrayRead(Y,&y);
675: return(0);
676: }
678: /* Hull, 1972, Problem C1 */
682: PetscErrorCode RHSFunction_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec F, void *s)
683: {
684: PetscErrorCode ierr;
685: PetscScalar *f;
686: const PetscScalar *y;
687: PetscInt N,i;
690: VecGetSize (Y,&N);
691: VecGetArrayRead(Y,&y);
692: VecGetArray(F,&f);
693: f[0] = -y[0];
694: for (i = 1; i < N-1; i++) {
695: f[i] = y[i-1] - y[i];
696: }
697: f[N-1] = y[N-2];
698: VecRestoreArrayRead(Y,&y);
699: VecRestoreArray(F,&f);
700: return(0);
701: }
705: PetscErrorCode IFunction_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
706: {
707: PetscErrorCode ierr;
708: PetscScalar *f;
709: const PetscScalar *y;
710: PetscInt N,i;
713: VecGetSize (Y,&N);
714: VecGetArrayRead(Y,&y);
715: VecGetArray(F,&f);
716: f[0] = -y[0];
717: for (i = 1; i < N-1; i++) {
718: f[i] = y[i-1] - y[i];
719: }
720: f[N-1] = y[N-2];
721: VecRestoreArrayRead(Y,&y);
722: VecRestoreArray(F,&f);
723: /* Left hand side = ydot - f(y) */
724: VecAYPX(F,-1.0,Ydot);
725: return(0);
726: }
730: PetscErrorCode IJacobian_Hull1972C1(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
731: {
732: PetscErrorCode ierr;
733: const PetscScalar *y;
734: PetscInt N,i,col[2];
735: PetscScalar value[2];
738: VecGetSize (Y,&N);
739: VecGetArrayRead(Y,&y);
740: i = 0;
741: value[0] = a+1; col[0] = 0;
742: value[1] = 0; col[1] = 1;
743: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
744: for (i = 0; i < N; i++) {
745: value[0] = -1; col[0] = i-1;
746: value[1] = a+1; col[1] = i;
747: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
748: }
749: i = N-1;
750: value[0] = -1; col[0] = N-2;
751: value[1] = a; col[1] = N-1;
752: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
753: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
754: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
755: VecRestoreArrayRead(Y,&y);
756: return(0);
757: }
759: /* Hull, 1972, Problem C2 */
763: PetscErrorCode RHSFunction_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec F, void *s)
764: {
765: PetscErrorCode ierr;
766: const PetscScalar *y;
767: PetscScalar *f;
768: PetscInt N,i;
771: VecGetSize (Y,&N);
772: VecGetArrayRead(Y,&y);
773: VecGetArray(F,&f);
774: f[0] = -y[0];
775: for (i = 1; i < N-1; i++) {
776: f[i] = (PetscReal)i*y[i-1] - (PetscReal)(i+1)*y[i];
777: }
778: f[N-1] = (PetscReal)(N-1)*y[N-2];
779: VecRestoreArrayRead(Y,&y);
780: VecRestoreArray(F,&f);
781: return(0);
782: }
786: PetscErrorCode IFunction_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
787: {
788: PetscErrorCode ierr;
789: PetscScalar *f;
790: const PetscScalar *y;
791: PetscInt N,i;
794: VecGetSize (Y,&N);
795: VecGetArrayRead(Y,&y);
796: VecGetArray(F,&f);
797: f[0] = -y[0];
798: for (i = 1; i < N-1; i++) {
799: f[i] = (PetscReal)i*y[i-1] - (PetscReal)(i+1)*y[i];
800: }
801: f[N-1] = (PetscReal)(N-1)*y[N-2];
802: VecRestoreArrayRead(Y,&y);
803: VecRestoreArray(F,&f);
804: /* Left hand side = ydot - f(y) */
805: VecAYPX(F,-1.0,Ydot);
806: return(0);
807: }
811: PetscErrorCode IJacobian_Hull1972C2(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
812: {
813: PetscErrorCode ierr;
814: const PetscScalar *y;
815: PetscInt N,i,col[2];
816: PetscScalar value[2];
819: VecGetSize (Y,&N);
820: VecGetArrayRead(Y,&y);
821: i = 0;
822: value[0] = a+1; col[0] = 0;
823: value[1] = 0; col[1] = 1;
824: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
825: for (i = 0; i < N; i++) {
826: value[0] = -(PetscReal) i; col[0] = i-1;
827: value[1] = a+(PetscReal)(i+1); col[1] = i;
828: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
829: }
830: i = N-1;
831: value[0] = -(PetscReal) (N-1); col[0] = N-2;
832: value[1] = a; col[1] = N-1;
833: MatSetValues(A,1,&i,2,col,value,INSERT_VALUES);
834: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
835: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
836: VecRestoreArrayRead(Y,&y);
837: return(0);
838: }
840: /* Hull, 1972, Problem C3 and C4 */
844: PetscErrorCode RHSFunction_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec F, void *s)
845: {
846: PetscErrorCode ierr;
847: PetscScalar *f;
848: const PetscScalar *y;
849: PetscInt N,i;
852: VecGetSize (Y,&N);
853: VecGetArrayRead(Y,&y);
854: VecGetArray(F,&f);
855: f[0] = -2.0*y[0] + y[1];
856: for (i = 1; i < N-1; i++) {
857: f[i] = y[i-1] - 2.0*y[i] + y[i+1];
858: }
859: f[N-1] = y[N-2] - 2.0*y[N-1];
860: VecRestoreArrayRead(Y,&y);
861: VecRestoreArray(F,&f);
862: return(0);
863: }
867: PetscErrorCode IFunction_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec Ydot, Vec F, void *s)
868: {
869: PetscErrorCode ierr;
870: PetscScalar *f;
871: const PetscScalar *y;
872: PetscInt N,i;
875: VecGetSize (Y,&N);
876: VecGetArrayRead(Y,&y);
877: VecGetArray(F,&f);
878: f[0] = -2.0*y[0] + y[1];
879: for (i = 1; i < N-1; i++) {
880: f[i] = y[i-1] - 2.0*y[i] + y[i+1];
881: }
882: f[N-1] = y[N-2] - 2.0*y[N-1];
883: VecRestoreArrayRead(Y,&y);
884: VecRestoreArray(F,&f);
885: /* Left hand side = ydot - f(y) */
886: VecAYPX(F,-1.0,Ydot);
887: return(0);
888: }
892: PetscErrorCode IJacobian_Hull1972C34(TS ts, PetscReal t, Vec Y, Vec Ydot, PetscReal a, Mat A, Mat B, void *s)
893: {
894: PetscErrorCode ierr;
895: const PetscScalar *y;
896: PetscScalar value[3];
897: PetscInt N,i,col[3];
900: VecGetSize (Y,&N);
901: VecGetArrayRead(Y,&y);
902: for (i = 0; i < N; i++) {
903: if (i == 0) {
904: value[0] = a+2; col[0] = i;
905: value[1] = -1; col[1] = i+1;
906: value[2] = 0; col[2] = i+2;
907: } else if (i == N-1) {
908: value[0] = 0; col[0] = i-2;
909: value[1] = -1; col[1] = i-1;
910: value[2] = a+2; col[2] = i;
911: } else {
912: value[0] = -1; col[0] = i-1;
913: value[1] = a+2; col[1] = i;
914: value[2] = -1; col[2] = i+1;
915: }
916: MatSetValues(A,1,&i,3,col,value,INSERT_VALUES);
917: }
918: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
919: MatAssemblyEnd (A,MAT_FINAL_ASSEMBLY);
920: VecRestoreArrayRead(Y,&y);
921: return(0);
922: }
924: /***************************************************************************/
928: /* Sets the initial solution for the IVP and sets up the function pointers*/
929: PetscErrorCode Initialize(Vec Y, void* s)
930: {
932: char *p = (char*) s;
933: PetscScalar *y;
934: PetscInt N = GetSize((const char *)s);
935: PetscBool flg;
938: VecZeroEntries(Y);
939: VecGetArray(Y,&y);
940: if (!strcmp(p,"hull1972a1")) {
941: y[0] = 1.0;
942: RHSFunction = RHSFunction_Hull1972A1;
943: IFunction = IFunction_Hull1972A1;
944: IJacobian = IJacobian_Hull1972A1;
945: } else if (!strcmp(p,"hull1972a2")) {
946: y[0] = 1.0;
947: RHSFunction = RHSFunction_Hull1972A2;
948: IFunction = IFunction_Hull1972A2;
949: IJacobian = IJacobian_Hull1972A2;
950: } else if (!strcmp(p,"hull1972a3")) {
951: y[0] = 1.0;
952: RHSFunction = RHSFunction_Hull1972A3;
953: IFunction = IFunction_Hull1972A3;
954: IJacobian = IJacobian_Hull1972A3;
955: } else if (!strcmp(p,"hull1972a4")) {
956: y[0] = 1.0;
957: RHSFunction = RHSFunction_Hull1972A4;
958: IFunction = IFunction_Hull1972A4;
959: IJacobian = IJacobian_Hull1972A4;
960: } else if (!strcmp(p,"hull1972a5")) {
961: y[0] = 4.0;
962: RHSFunction = RHSFunction_Hull1972A5;
963: IFunction = IFunction_Hull1972A5;
964: IJacobian = IJacobian_Hull1972A5;
965: } else if (!strcmp(p,"hull1972b1")) {
966: y[0] = 1.0;
967: y[1] = 3.0;
968: RHSFunction = RHSFunction_Hull1972B1;
969: IFunction = IFunction_Hull1972B1;
970: IJacobian = IJacobian_Hull1972B1;
971: } else if (!strcmp(p,"hull1972b2")) {
972: y[0] = 2.0;
973: y[1] = 0.0;
974: y[2] = 1.0;
975: RHSFunction = RHSFunction_Hull1972B2;
976: IFunction = IFunction_Hull1972B2;
977: IJacobian = IJacobian_Hull1972B2;
978: } else if (!strcmp(p,"hull1972b3")) {
979: y[0] = 1.0;
980: y[1] = 0.0;
981: y[2] = 0.0;
982: RHSFunction = RHSFunction_Hull1972B3;
983: IFunction = IFunction_Hull1972B3;
984: IJacobian = IJacobian_Hull1972B3;
985: } else if (!strcmp(p,"hull1972b4")) {
986: y[0] = 3.0;
987: y[1] = 0.0;
988: y[2] = 0.0;
989: RHSFunction = RHSFunction_Hull1972B4;
990: IFunction = IFunction_Hull1972B4;
991: IJacobian = IJacobian_Hull1972B4;
992: } else if (!strcmp(p,"hull1972b5")) {
993: y[0] = 0.0;
994: y[1] = 1.0;
995: y[2] = 1.0;
996: RHSFunction = RHSFunction_Hull1972B5;
997: IFunction = IFunction_Hull1972B5;
998: IJacobian = IJacobian_Hull1972B5;
999: } else if (!strcmp(p,"hull1972c1")) {
1000: y[0] = 1.0;
1001: RHSFunction = RHSFunction_Hull1972C1;
1002: IFunction = IFunction_Hull1972C1;
1003: IJacobian = IJacobian_Hull1972C1;
1004: } else if (!strcmp(p,"hull1972c2")) {
1005: y[0] = 1.0;
1006: RHSFunction = RHSFunction_Hull1972C2;
1007: IFunction = IFunction_Hull1972C2;
1008: IJacobian = IJacobian_Hull1972C2;
1009: } else if ((!strcmp(p,"hull1972c3"))
1010: ||(!strcmp(p,"hull1972c4"))){
1011: y[0] = 1.0;
1012: RHSFunction = RHSFunction_Hull1972C34;
1013: IFunction = IFunction_Hull1972C34;
1014: IJacobian = IJacobian_Hull1972C34;
1015: }
1016: PetscOptionsGetScalarArray(NULL,"-yinit",y,&N,&flg);
1017: if ((N != GetSize((const char*)s)) && flg) {
1018: 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));
1019: }
1020: VecRestoreArray(Y,&y);
1021: return(0);
1022: }
1026: /* Calculates the exact solution to problems that have one */
1027: PetscErrorCode ExactSolution(Vec Y, void* s, PetscReal t, PetscBool *flag)
1028: {
1030: char *p = (char*) s;
1031: PetscScalar *y;
1034: if (!strcmp(p,"hull1972a1")) {
1035: VecGetArray(Y,&y);
1036: y[0] = PetscExpReal(-t);
1037: *flag = PETSC_TRUE;
1038: VecRestoreArray(Y,&y);
1039: } else if (!strcmp(p,"hull1972a2")) {
1040: VecGetArray(Y,&y);
1041: y[0] = 1.0/PetscSqrtReal(t+1);
1042: *flag = PETSC_TRUE;
1043: VecRestoreArray(Y,&y);
1044: } else if (!strcmp(p,"hull1972a3")) {
1045: VecGetArray(Y,&y);
1046: y[0] = PetscExpReal(PetscSinReal(t));
1047: *flag = PETSC_TRUE;
1048: VecRestoreArray(Y,&y);
1049: } else if (!strcmp(p,"hull1972a4")) {
1050: VecGetArray(Y,&y);
1051: y[0] = 20.0/(1+19.0*PetscExpReal(-t/4.0));
1052: *flag = PETSC_TRUE;
1053: VecRestoreArray(Y,&y);
1054: } else {
1055: VecSet(Y,0);
1056: *flag = PETSC_FALSE;
1057: }
1058: return(0);
1059: }
1063: /* Solves the specified ODE and computes the error if exact solution is available */
1064: PetscErrorCode SolveODE(char* ptype, PetscReal dt, PetscReal tfinal, PetscInt maxiter, PetscReal *error, PetscBool *exact_flag)
1065: {
1066: PetscErrorCode ierr; /* Error code */
1067: TS ts; /* time-integrator */
1068: Vec Y; /* Solution vector */
1069: Vec Yex; /* Exact solution */
1070: PetscInt N; /* Size of the system of equations */
1071: TSType time_scheme; /* Type of time-integration scheme */
1072: Mat Jac = NULL; /* Jacobian matrix */
1075: N = GetSize((const char *)&ptype[0]);
1076: if (N < 0) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_SIZ,"Illegal problem specification.\n");
1077: VecCreate(PETSC_COMM_WORLD,&Y);
1078: VecSetSizes(Y,N,PETSC_DECIDE);
1079: VecSetUp(Y);
1080: VecSet(Y,0);
1082: /* Initialize the problem */
1083: Initialize(Y,&ptype[0]);
1085: /* Create and initialize the time-integrator */
1086: TSCreate(PETSC_COMM_WORLD,&ts);
1087: /* Default time integration options */
1088: TSSetType(ts,TSEULER);
1089: TSSetDuration(ts,maxiter,tfinal);
1090: TSSetInitialTimeStep(ts,0.0,dt);
1091: TSSetExactFinalTime(ts,TS_EXACTFINALTIME_MATCHSTEP);
1092: /* Read command line options for time integration */
1093: TSSetFromOptions(ts);
1094: /* Set solution vector */
1095: TSSetSolution(ts,Y);
1096: /* Specify left/right-hand side functions */
1097: TSGetType(ts,&time_scheme);
1098: if ((!strcmp(time_scheme,TSEULER)) || (!strcmp(time_scheme,TSRK)) || (!strcmp(time_scheme,TSSSP))) {
1099: /* Explicit time-integration -> specify right-hand side function ydot = f(y) */
1100: TSSetRHSFunction(ts,NULL,RHSFunction,&ptype[0]);
1101: } else if ((!strcmp(time_scheme,TSBEULER)) || (!strcmp(time_scheme,TSARKIMEX))) {
1102: /* Implicit time-integration -> specify left-hand side function ydot-f(y) = 0 */
1103: /* and its Jacobian function */
1104: TSSetIFunction(ts,NULL,IFunction,&ptype[0]);
1105: MatCreate(PETSC_COMM_WORLD,&Jac);
1106: MatSetSizes(Jac,PETSC_DECIDE,PETSC_DECIDE,N,N);
1107: MatSetFromOptions(Jac);
1108: MatSetUp(Jac);
1109: TSSetIJacobian(ts,Jac,Jac,IJacobian,&ptype[0]);
1110: }
1112: /* Solve */
1113: TSSolve(ts,Y);
1115: /* Exact solution */
1116: VecDuplicate(Y,&Yex);
1117: ExactSolution(Yex,&ptype[0],tfinal,exact_flag);
1119: /* Calculate Error */
1120: VecAYPX(Yex,-1.0,Y);
1121: VecNorm(Yex,NORM_2,error);
1122: *error = PetscSqrtReal(((*error)*(*error))/N);
1124: /* Clean up and finalize */
1125: MatDestroy(&Jac);
1126: TSDestroy(&ts);
1127: VecDestroy(&Yex);
1128: VecDestroy(&Y);
1130: return(0);
1131: }
1135: int main(int argc, char **argv)
1136: {
1137: PetscErrorCode ierr; /* Error code */
1138: char ptype[256] = "hull1972a1"; /* Problem specification */
1139: PetscInt n_refine = 1; /* Number of refinement levels for convergence analysis */
1140: PetscReal refine_fac = 2.0; /* Refinement factor for dt */
1141: PetscReal dt_initial = 0.01; /* Initial default value of dt */
1142: PetscReal dt;
1143: PetscReal tfinal = 20.0; /* Final time for the time-integration */
1144: PetscInt maxiter = 100000; /* Maximum number of time-integration iterations */
1145: PetscReal *error; /* Array to store the errors for convergence analysis */
1146: PetscMPIInt size; /* No of processors */
1147: PetscBool flag; /* Flag denoting availability of exact solution */
1148: PetscInt r;
1150: /* Initialize program */
1151: PetscInitialize(&argc,&argv,(char*)0,help);
1153: /* Check if running with only 1 proc */
1154: MPI_Comm_size(PETSC_COMM_WORLD,&size);
1155: if (size>1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Only for sequential runs");
1157: PetscOptionsBegin(PETSC_COMM_WORLD,NULL,NULL,NULL);
1158: PetscOptionsString("-problem","Problem specification","<hull1972a1>",ptype,ptype,sizeof(ptype),NULL);
1159: PetscOptionsInt("-refinement_levels","Number of refinement levels for convergence analysis","<1>",n_refine,&n_refine,NULL);
1160: PetscOptionsReal("-refinement_factor","Refinement factor for dt","<2.0>",refine_fac,&refine_fac,NULL);
1161: PetscOptionsReal("-dt","Time step size (for convergence analysis, initial time step)","<0.01>",dt_initial,&dt_initial,NULL);
1162: PetscOptionsReal("-final_time","Final time for the time-integration","<20.0>",tfinal,&tfinal,NULL);
1163: PetscOptionsEnd();
1165: PetscMalloc1(n_refine,&error);
1166: for (r = 0,dt = dt_initial; r < n_refine; r++) {
1167: error[r] = 0;
1168: if (r > 0) dt /= refine_fac;
1170: 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]));
1171: SolveODE(&ptype[0],dt,tfinal,maxiter,&error[r],&flag);
1172: if (flag) {
1173: /* If exact solution available for the specified ODE */
1174: if (r > 0) {
1175: PetscReal conv_rate = (PetscLogReal(error[r]) - PetscLogReal(error[r-1])) / (-PetscLogReal(refine_fac));
1176: PetscPrintf(PETSC_COMM_WORLD,"Error = %E,\tConvergence rate = %f\n.",(double)error[r],(double)conv_rate);
1177: } else {
1178: PetscPrintf(PETSC_COMM_WORLD,"Error = %E.\n",error[r]);
1179: }
1180: }
1181: }
1182: PetscFree(error);
1184: /* Exit */
1185: PetscFinalize();
1186: return(0);
1187: }