1 2 static char help[] = "Tests DMComposite routines.\n\n"; 3 4 #include <petscdmredundant.h> 5 #include <petscdm.h> 6 #include <petscdmda.h> 7 #include <petscdmcomposite.h> 8 #include <petscpf.h> 9 10 int main(int argc,char **argv) 11 { 12 PetscErrorCode ierr; 13 PetscInt nredundant1 = 5,nredundant2 = 2,i; 14 ISLocalToGlobalMapping *ltog; 15 PetscMPIInt rank,size; 16 DM packer; 17 Vec global,local1,local2,redundant1,redundant2; 18 PF pf; 19 DM da1,da2,dmred1,dmred2; 20 PetscScalar *redundant1a,*redundant2a; 21 PetscViewer sviewer; 22 PetscBool gather_add = PETSC_FALSE; 23 24 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 25 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); 26 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 27 28 ierr = PetscOptionsGetBool(NULL,NULL,"-gather_add",&gather_add,NULL);CHKERRQ(ierr); 29 30 ierr = DMCompositeCreate(PETSC_COMM_WORLD,&packer);CHKERRQ(ierr); 31 32 ierr = DMRedundantCreate(PETSC_COMM_WORLD,0,nredundant1,&dmred1);CHKERRQ(ierr); 33 ierr = DMCreateLocalVector(dmred1,&redundant1);CHKERRQ(ierr); 34 ierr = DMCompositeAddDM(packer,dmred1);CHKERRQ(ierr); 35 36 ierr = DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,8,1,1,NULL,&da1);CHKERRQ(ierr); 37 ierr = DMSetFromOptions(da1);CHKERRQ(ierr); 38 ierr = DMSetUp(da1);CHKERRQ(ierr); 39 ierr = DMCreateLocalVector(da1,&local1);CHKERRQ(ierr); 40 ierr = DMCompositeAddDM(packer,da1);CHKERRQ(ierr); 41 42 ierr = DMRedundantCreate(PETSC_COMM_WORLD,1%size,nredundant2,&dmred2);CHKERRQ(ierr); 43 ierr = DMCreateLocalVector(dmred2,&redundant2);CHKERRQ(ierr); 44 ierr = DMCompositeAddDM(packer,dmred2);CHKERRQ(ierr); 45 46 ierr = DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_NONE,6,1,1,NULL,&da2);CHKERRQ(ierr); 47 ierr = DMSetFromOptions(da2);CHKERRQ(ierr); 48 ierr = DMSetUp(da2);CHKERRQ(ierr); 49 ierr = DMCreateLocalVector(da2,&local2);CHKERRQ(ierr); 50 ierr = DMCompositeAddDM(packer,da2);CHKERRQ(ierr); 51 52 ierr = DMCreateGlobalVector(packer,&global);CHKERRQ(ierr); 53 ierr = PFCreate(PETSC_COMM_WORLD,1,1,&pf);CHKERRQ(ierr); 54 ierr = PFSetType(pf,PFIDENTITY,NULL);CHKERRQ(ierr); 55 ierr = PFApplyVec(pf,NULL,global);CHKERRQ(ierr); 56 ierr = PFDestroy(&pf);CHKERRQ(ierr); 57 ierr = VecView(global,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 58 59 ierr = DMCompositeScatter(packer,global,redundant1,local1,redundant2,local2);CHKERRQ(ierr); 60 ierr = PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 61 ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"[%d] My part of redundant1 vector\n",rank);CHKERRQ(ierr); 62 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 63 ierr = VecView(redundant1,sviewer);CHKERRQ(ierr); 64 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 65 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 66 ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"[%d] My part of da1 vector\n",rank);CHKERRQ(ierr); 67 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 68 ierr = VecView(local1,sviewer);CHKERRQ(ierr); 69 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 70 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 71 ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"[%d] My part of redundant2 vector\n",rank);CHKERRQ(ierr); 72 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 73 ierr = VecView(redundant2,sviewer);CHKERRQ(ierr); 74 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 75 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 76 ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"[%d] My part of da2 vector\n",rank);CHKERRQ(ierr); 77 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 78 ierr = VecView(local2,sviewer);CHKERRQ(ierr); 79 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 80 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 81 ierr = PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 82 83 ierr = VecGetArray(redundant1,&redundant1a);CHKERRQ(ierr); 84 ierr = VecGetArray(redundant2,&redundant2a);CHKERRQ(ierr); 85 for (i=0; i<nredundant1; i++) redundant1a[i] = (rank+2)*i; 86 for (i=0; i<nredundant2; i++) redundant2a[i] = (rank+10)*i; 87 ierr = VecRestoreArray(redundant1,&redundant1a);CHKERRQ(ierr); 88 ierr = VecRestoreArray(redundant2,&redundant2a);CHKERRQ(ierr); 89 90 ierr = DMCompositeGather(packer,gather_add ? ADD_VALUES : INSERT_VALUES,global,redundant1,local1,redundant2,local2);CHKERRQ(ierr); 91 ierr = VecView(global,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 92 93 /* get the global numbering for each subvector element */ 94 ierr = DMCompositeGetISLocalToGlobalMappings(packer,<og);CHKERRQ(ierr); 95 96 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of redundant1 vector\n");CHKERRQ(ierr); 97 ierr = ISLocalToGlobalMappingView(ltog[0],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 98 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of local1 vector\n");CHKERRQ(ierr); 99 ierr = ISLocalToGlobalMappingView(ltog[1],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 100 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of redundant2 vector\n");CHKERRQ(ierr); 101 ierr = ISLocalToGlobalMappingView(ltog[2],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 102 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of local2 vector\n");CHKERRQ(ierr); 103 ierr = ISLocalToGlobalMappingView(ltog[3],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 104 105 for (i=0; i<4; i++) {ierr = ISLocalToGlobalMappingDestroy(<og[i]);CHKERRQ(ierr);} 106 ierr = PetscFree(ltog);CHKERRQ(ierr); 107 108 ierr = DMDestroy(&da1);CHKERRQ(ierr); 109 ierr = DMDestroy(&dmred1);CHKERRQ(ierr); 110 ierr = DMDestroy(&dmred2);CHKERRQ(ierr); 111 ierr = DMDestroy(&da2);CHKERRQ(ierr); 112 ierr = VecDestroy(&redundant1);CHKERRQ(ierr); 113 ierr = VecDestroy(&redundant2);CHKERRQ(ierr); 114 ierr = VecDestroy(&local1);CHKERRQ(ierr); 115 ierr = VecDestroy(&local2);CHKERRQ(ierr); 116 ierr = VecDestroy(&global);CHKERRQ(ierr); 117 ierr = DMDestroy(&packer);CHKERRQ(ierr); 118 ierr = PetscFinalize(); 119 return ierr; 120 } 121 122 /*TEST 123 124 build: 125 requires: !complex 126 127 test: 128 nsize: 3 129 130 test: 131 suffix: 2 132 nsize: 3 133 args: -gather_add 134 135 TEST*/ 136