1 static char help[] = "Tests PetscOptionsGetScalar(), PetscOptionsScalarArray() for complex numbers\n";
2
3 #include <petscsys.h>
4
main(int argc,char ** argv)5 int main(int argc, char **argv)
6 {
7 PetscInt n, i;
8 PetscScalar a, array[10];
9 PetscReal rarray[10];
10
11 PetscFunctionBeginUser;
12 PetscCall(PetscInitialize(&argc, &argv, NULL, help));
13 PetscCall(PetscOptionsGetScalar(NULL, NULL, "-a", &a, NULL));
14 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Scalar a = %g + %gi\n", (double)PetscRealPart(a), (double)PetscImaginaryPart(a)));
15
16 PetscOptionsBegin(PETSC_COMM_WORLD, NULL, "test options", NULL);
17 n = 10; /* max num of input values */
18 PetscCall(PetscOptionsRealArray("-rarray", "Input a real array", "ex14.c", rarray, &n, NULL));
19 if (n) {
20 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Real rarray of length %" PetscInt_FMT "\n", n));
21 for (i = 0; i < n; i++) PetscCall(PetscPrintf(PETSC_COMM_SELF, " %g,\n", (double)rarray[i]));
22 }
23
24 n = 10; /* max num of input values */
25 PetscCall(PetscOptionsScalarArray("-array", "Input a scalar array", "ex14.c", array, &n, NULL));
26 if (n) {
27 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Scalar rarray of length %" PetscInt_FMT "\n", n));
28 for (i = 0; i < n; i++) {
29 if (PetscImaginaryPart(array[i]) < 0.0) {
30 PetscCall(PetscPrintf(PETSC_COMM_SELF, " %g - %gi\n", (double)PetscRealPart(array[i]), (double)PetscAbsReal(PetscImaginaryPart(array[i]))));
31 } else {
32 PetscCall(PetscPrintf(PETSC_COMM_SELF, " %g + %gi\n", (double)PetscRealPart(array[i]), (double)PetscImaginaryPart(array[i])));
33 }
34 }
35 }
36 PetscOptionsEnd();
37 PetscCall(PetscFinalize());
38 return 0;
39 }
40
41 /*TEST
42
43 test:
44 requires: complex
45 args: -array 1.0,-2-3i,4.5+6.2i,4.5,6.8+4i,i,-i,-1.2i -rarray 1,2,3 -a 1.5+2.1i
46
47 TEST*/
48