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