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