1 2 static char help[] = "Tests using PetscViewerGetSubViewer() recursively\n\n"; 3 4 /*T 5 Concepts: viewers 6 Processors: n 7 T*/ 8 #include <petscsys.h> 9 #include <petscviewer.h> 10 11 int main(int argc,char **argv) 12 { 13 PetscErrorCode ierr; 14 PetscViewer viewer,subviewer,subsubviewer; 15 PetscViewerFormat format; 16 PetscBool flg; 17 PetscSubcomm psubcomm,psubsubcomm; 18 MPI_Comm comm,subcomm,subsubcomm; 19 PetscMPIInt size; 20 21 /* 22 Every PETSc routine should begin with the PetscInitialize() routine. 23 argc, argv - These command line arguments are taken to extract the options 24 supplied to PETSc and options supplied to MPI. 25 help - When PETSc executable is invoked with the option -help, 26 it prints the various options that can be applied at 27 runtime. The user can use the "help" variable place 28 additional help messages in this printout. 29 */ 30 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 31 comm = PETSC_COMM_WORLD; 32 ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr); 33 PetscCheckFalse(size < 4,PETSC_COMM_WORLD,PETSC_ERR_SUP,"Must run with at least 4 MPI processes"); 34 ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-viewer",&viewer,&format,&flg);CHKERRQ(ierr); 35 PetscCheckFalse(!viewer,PETSC_COMM_WORLD,PETSC_ERR_SUP,"Must use -viewer option"); 36 37 ierr = PetscViewerASCIIPrintf(viewer,"Print called on original full viewer %d\n",PetscGlobalRank);CHKERRQ(ierr); 38 39 ierr = PetscSubcommCreate(comm,&psubcomm);CHKERRQ(ierr); 40 ierr = PetscSubcommSetNumber(psubcomm,2);CHKERRQ(ierr); 41 ierr = PetscSubcommSetType(psubcomm,PETSC_SUBCOMM_CONTIGUOUS);CHKERRQ(ierr); 42 /* enable runtime switch of psubcomm type, e.g., '-psubcomm_type interlaced */ 43 ierr = PetscSubcommSetFromOptions(psubcomm);CHKERRQ(ierr); 44 subcomm = PetscSubcommChild(psubcomm); 45 46 ierr = PetscViewerGetSubViewer(viewer,subcomm,&subviewer);CHKERRQ(ierr); 47 48 ierr = PetscViewerASCIIPrintf(subviewer," Print called on sub viewers %d\n",PetscGlobalRank);CHKERRQ(ierr); 49 50 ierr = PetscSubcommCreate(subcomm,&psubsubcomm);CHKERRQ(ierr); 51 ierr = PetscSubcommSetNumber(psubsubcomm,2);CHKERRQ(ierr); 52 ierr = PetscSubcommSetType(psubsubcomm,PETSC_SUBCOMM_CONTIGUOUS);CHKERRQ(ierr); 53 /* enable runtime switch of psubcomm type, e.g., '-psubcomm_type interlaced */ 54 ierr = PetscSubcommSetFromOptions(psubsubcomm);CHKERRQ(ierr); 55 subsubcomm = PetscSubcommChild(psubsubcomm); 56 57 ierr = PetscViewerGetSubViewer(subviewer,subsubcomm,&subsubviewer);CHKERRQ(ierr); 58 59 ierr = PetscViewerASCIIPrintf(subsubviewer," Print called on sub sub viewers %d\n",PetscGlobalRank);CHKERRQ(ierr); 60 61 ierr = PetscViewerRestoreSubViewer(subviewer,subsubcomm,&subsubviewer);CHKERRQ(ierr); 62 ierr = PetscViewerRestoreSubViewer(viewer,subcomm,&subviewer);CHKERRQ(ierr); 63 64 ierr = PetscSubcommDestroy(&psubsubcomm);CHKERRQ(ierr); 65 ierr = PetscSubcommDestroy(&psubcomm);CHKERRQ(ierr); 66 ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 67 ierr = PetscFinalize(); 68 return ierr; 69 } 70 71 /*TEST 72 73 test: 74 nsize: 4 75 args: -viewer 76 77 TEST*/ 78