1 2 static char help[] = "Tests DMCreateMatrix for DMComposite.\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 ISLocalToGlobalMapping *ltog,ltogs; 14 PetscMPIInt size; 15 DM packer; 16 DM da,dmred; 17 Mat M; 18 PetscInt i; 19 20 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 21 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 22 23 ierr = DMCompositeCreate(PETSC_COMM_WORLD,&packer);CHKERRQ(ierr); 24 25 ierr = DMRedundantCreate(PETSC_COMM_WORLD,0,5,&dmred);CHKERRQ(ierr); 26 ierr = DMCompositeAddDM(packer,dmred);CHKERRQ(ierr); 27 ierr = DMGetLocalToGlobalMapping(dmred,<ogs);CHKERRQ(ierr); 28 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of dmred\n");CHKERRQ(ierr); 29 ierr = ISLocalToGlobalMappingView(ltogs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 30 ierr = DMDestroy(&dmred);CHKERRQ(ierr); 31 32 ierr = DMDACreate2d(PETSC_COMM_WORLD,DM_BOUNDARY_MIRROR,DM_BOUNDARY_MIRROR,DMDA_STENCIL_STAR,3,3,PETSC_DECIDE,PETSC_DECIDE,2,1,NULL,NULL,&da);CHKERRQ(ierr); 33 ierr = DMSetFromOptions(da);CHKERRQ(ierr); 34 ierr = DMSetUp(da);CHKERRQ(ierr); 35 ierr = DMCompositeAddDM(packer,da);CHKERRQ(ierr); 36 ierr = DMGetLocalToGlobalMapping(da,<ogs);CHKERRQ(ierr); 37 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of da\n");CHKERRQ(ierr); 38 ierr = ISLocalToGlobalMappingView(ltogs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 39 ierr = DMDestroy(&da);CHKERRQ(ierr); 40 41 ierr = DMSetMatType(packer,MATNEST);CHKERRQ(ierr); 42 ierr = DMSetFromOptions(packer);CHKERRQ(ierr); 43 ierr = DMCreateMatrix(packer,&M);CHKERRQ(ierr); 44 ierr = MatView(M,NULL);CHKERRQ(ierr); 45 ierr = MatDestroy(&M);CHKERRQ(ierr); 46 47 /* get the global numbering for each subvector element */ 48 ierr = DMCompositeGetISLocalToGlobalMappings(packer,<og);CHKERRQ(ierr); 49 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of dmred vector\n");CHKERRQ(ierr); 50 ierr = ISLocalToGlobalMappingView(ltog[0],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 51 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Local to global mapping of da vector\n");CHKERRQ(ierr); 52 ierr = ISLocalToGlobalMappingView(ltog[1],PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 53 for (i=0; i<2; i++) {ierr = ISLocalToGlobalMappingDestroy(<og[i]);CHKERRQ(ierr);} 54 55 ierr = PetscFree(ltog);CHKERRQ(ierr); 56 ierr = DMDestroy(&packer);CHKERRQ(ierr); 57 ierr = PetscFinalize(); 58 return ierr; 59 } 60 61 /*TEST 62 63 test: 64 suffix: composite_nest_l2g 65 nsize: {{1 2}separate output} 66 67 TEST*/ 68