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