1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Tests using PetscViewerGetSubViewer() recursively\n\n"; 3c4762a1bSJed Brown 4c4762a1bSJed Brown /*T 5c4762a1bSJed Brown Concepts: viewers 6c4762a1bSJed Brown Processors: n 7c4762a1bSJed Brown T*/ 8c4762a1bSJed Brown #include <petscsys.h> 9c4762a1bSJed Brown #include <petscviewer.h> 10c4762a1bSJed Brown 11c4762a1bSJed Brown int main(int argc,char **argv) 12c4762a1bSJed Brown { 13c4762a1bSJed Brown PetscViewer viewer,subviewer,subsubviewer; 14c4762a1bSJed Brown PetscViewerFormat format; 15c4762a1bSJed Brown PetscBool flg; 16c4762a1bSJed Brown PetscSubcomm psubcomm,psubsubcomm; 17c4762a1bSJed Brown MPI_Comm comm,subcomm,subsubcomm; 18c4762a1bSJed Brown PetscMPIInt size; 19c4762a1bSJed Brown 20c4762a1bSJed Brown /* 21c4762a1bSJed Brown Every PETSc routine should begin with the PetscInitialize() routine. 22c4762a1bSJed Brown argc, argv - These command line arguments are taken to extract the options 23c4762a1bSJed Brown supplied to PETSc and options supplied to MPI. 24c4762a1bSJed Brown help - When PETSc executable is invoked with the option -help, 25c4762a1bSJed Brown it prints the various options that can be applied at 26c4762a1bSJed Brown runtime. The user can use the "help" variable place 27c4762a1bSJed Brown additional help messages in this printout. 28c4762a1bSJed Brown */ 299566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc,&argv,(char*)0,help)); 30c4762a1bSJed Brown comm = PETSC_COMM_WORLD; 319566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(comm,&size)); 32*be096a46SBarry Smith PetscCheck(size >= 4,PETSC_COMM_WORLD,PETSC_ERR_WRONG_MPI_SIZE,"Must run with at least 4 MPI processes"); 339566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetViewer(comm,NULL,NULL,"-viewer",&viewer,&format,&flg)); 3428b400f6SJacob Faibussowitsch PetscCheck(viewer,PETSC_COMM_WORLD,PETSC_ERR_SUP,"Must use -viewer option"); 35c4762a1bSJed Brown 369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Print called on original full viewer %d\n",PetscGlobalRank)); 37c4762a1bSJed Brown 389566063dSJacob Faibussowitsch PetscCall(PetscSubcommCreate(comm,&psubcomm)); 399566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetNumber(psubcomm,2)); 409566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetType(psubcomm,PETSC_SUBCOMM_CONTIGUOUS)); 41c4762a1bSJed Brown /* enable runtime switch of psubcomm type, e.g., '-psubcomm_type interlaced */ 429566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetFromOptions(psubcomm)); 43c4762a1bSJed Brown subcomm = PetscSubcommChild(psubcomm); 44c4762a1bSJed Brown 459566063dSJacob Faibussowitsch PetscCall(PetscViewerGetSubViewer(viewer,subcomm,&subviewer)); 46c4762a1bSJed Brown 479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(subviewer," Print called on sub viewers %d\n",PetscGlobalRank)); 48c4762a1bSJed Brown 499566063dSJacob Faibussowitsch PetscCall(PetscSubcommCreate(subcomm,&psubsubcomm)); 509566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetNumber(psubsubcomm,2)); 519566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetType(psubsubcomm,PETSC_SUBCOMM_CONTIGUOUS)); 52c4762a1bSJed Brown /* enable runtime switch of psubcomm type, e.g., '-psubcomm_type interlaced */ 539566063dSJacob Faibussowitsch PetscCall(PetscSubcommSetFromOptions(psubsubcomm)); 54c4762a1bSJed Brown subsubcomm = PetscSubcommChild(psubsubcomm); 55c4762a1bSJed Brown 569566063dSJacob Faibussowitsch PetscCall(PetscViewerGetSubViewer(subviewer,subsubcomm,&subsubviewer)); 57c4762a1bSJed Brown 589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(subsubviewer," Print called on sub sub viewers %d\n",PetscGlobalRank)); 59c4762a1bSJed Brown 609566063dSJacob Faibussowitsch PetscCall(PetscViewerRestoreSubViewer(subviewer,subsubcomm,&subsubviewer)); 619566063dSJacob Faibussowitsch PetscCall(PetscViewerRestoreSubViewer(viewer,subcomm,&subviewer)); 62c4762a1bSJed Brown 639566063dSJacob Faibussowitsch PetscCall(PetscSubcommDestroy(&psubsubcomm)); 649566063dSJacob Faibussowitsch PetscCall(PetscSubcommDestroy(&psubcomm)); 659566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer)); 669566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 67b122ec5aSJacob Faibussowitsch return 0; 68c4762a1bSJed Brown } 69c4762a1bSJed Brown 70c4762a1bSJed Brown /*TEST 71c4762a1bSJed Brown 72c4762a1bSJed Brown test: 73c4762a1bSJed Brown nsize: 4 74c4762a1bSJed Brown args: -viewer 75c4762a1bSJed Brown 76c4762a1bSJed Brown TEST*/ 77