1 2 static char help[] = "Creates a matrix, inserts some values, and tests MatCreateSubMatrices() and MatZeroEntries().\n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **argv) 7 { 8 Mat mat,submat,submat1,*submatrices; 9 PetscInt m = 10,n = 10,i = 4,tmp,rstart,rend; 10 PetscErrorCode ierr; 11 IS irow,icol; 12 PetscScalar value = 1.0; 13 PetscViewer sviewer; 14 PetscBool allA = PETSC_FALSE; 15 16 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 17 ierr = PetscViewerPushFormat(PETSC_VIEWER_STDOUT_WORLD,PETSC_VIEWER_ASCII_COMMON);CHKERRQ(ierr); 18 ierr = PetscViewerPushFormat(PETSC_VIEWER_STDOUT_SELF,PETSC_VIEWER_ASCII_COMMON);CHKERRQ(ierr); 19 20 ierr = MatCreate(PETSC_COMM_WORLD,&mat);CHKERRQ(ierr); 21 ierr = MatSetSizes(mat,PETSC_DECIDE,PETSC_DECIDE,m,n);CHKERRQ(ierr); 22 ierr = MatSetFromOptions(mat);CHKERRQ(ierr); 23 ierr = MatSetUp(mat);CHKERRQ(ierr); 24 ierr = MatGetOwnershipRange(mat,&rstart,&rend);CHKERRQ(ierr); 25 for (i=rstart; i<rend; i++) { 26 value = (PetscReal)i+1; tmp = i % 5; 27 ierr = MatSetValues(mat,1,&tmp,1,&i,&value,INSERT_VALUES);CHKERRQ(ierr); 28 } 29 ierr = MatAssemblyBegin(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 30 ierr = MatAssemblyEnd(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 31 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Original matrix\n");CHKERRQ(ierr); 32 ierr = MatView(mat,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 33 34 /* Test MatCreateSubMatrix_XXX_All(), i.e., submatrix = A */ 35 ierr = PetscOptionsGetBool(NULL,NULL,"-test_all",&allA,NULL);CHKERRQ(ierr); 36 if (allA) { 37 ierr = ISCreateStride(PETSC_COMM_SELF,m,0,1,&irow);CHKERRQ(ierr); 38 ierr = ISCreateStride(PETSC_COMM_SELF,n,0,1,&icol);CHKERRQ(ierr); 39 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_INITIAL_MATRIX,&submatrices);CHKERRQ(ierr); 40 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_REUSE_MATRIX,&submatrices);CHKERRQ(ierr); 41 submat = *submatrices; 42 43 /* sviewer will cause the submatrices (one per processor) to be printed in the correct order */ 44 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nSubmatrices with all\n");CHKERRQ(ierr); 45 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"--------------------\n");CHKERRQ(ierr); 46 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 47 ierr = MatView(submat,sviewer);CHKERRQ(ierr); 48 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 49 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 50 51 ierr = ISDestroy(&irow);CHKERRQ(ierr); 52 ierr = ISDestroy(&icol);CHKERRQ(ierr); 53 54 /* test getting a reference on a submat */ 55 ierr = PetscObjectReference((PetscObject)submat);CHKERRQ(ierr); 56 ierr = MatDestroySubMatrices(1,&submatrices);CHKERRQ(ierr); 57 ierr = MatDestroy(&submat);CHKERRQ(ierr); 58 } 59 60 /* Form submatrix with rows 2-4 and columns 4-8 */ 61 ierr = ISCreateStride(PETSC_COMM_SELF,3,2,1,&irow);CHKERRQ(ierr); 62 ierr = ISCreateStride(PETSC_COMM_SELF,5,4,1,&icol);CHKERRQ(ierr); 63 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_INITIAL_MATRIX,&submatrices);CHKERRQ(ierr); 64 submat = *submatrices; 65 66 /* Test reuse submatrices */ 67 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_REUSE_MATRIX,&submatrices);CHKERRQ(ierr); 68 69 /* sviewer will cause the submatrices (one per processor) to be printed in the correct order */ 70 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nSubmatrices\n");CHKERRQ(ierr); 71 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 72 ierr = MatView(submat,sviewer);CHKERRQ(ierr); 73 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 74 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 75 ierr = PetscObjectReference((PetscObject)submat);CHKERRQ(ierr); 76 ierr = MatDestroySubMatrices(1,&submatrices);CHKERRQ(ierr); 77 ierr = MatDestroy(&submat);CHKERRQ(ierr); 78 79 /* Form submatrix with rows 2-4 and all columns */ 80 ierr = ISDestroy(&icol);CHKERRQ(ierr); 81 ierr = ISCreateStride(PETSC_COMM_SELF,10,0,1,&icol);CHKERRQ(ierr); 82 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_INITIAL_MATRIX,&submatrices);CHKERRQ(ierr); 83 ierr = MatCreateSubMatrices(mat,1,&irow,&icol,MAT_REUSE_MATRIX,&submatrices);CHKERRQ(ierr); 84 submat = *submatrices; 85 86 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"\nSubmatrices with allcolumns\n");CHKERRQ(ierr); 87 ierr = PetscViewerGetSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 88 ierr = MatView(submat,sviewer);CHKERRQ(ierr); 89 ierr = PetscViewerRestoreSubViewer(PETSC_VIEWER_STDOUT_WORLD,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 90 ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 91 92 /* Test MatDuplicate */ 93 ierr = MatDuplicate(submat,MAT_COPY_VALUES,&submat1);CHKERRQ(ierr); 94 ierr = MatDestroy(&submat1);CHKERRQ(ierr); 95 96 /* Zero the original matrix */ 97 ierr = PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_WORLD,"Original zeroed matrix\n");CHKERRQ(ierr); 98 ierr = MatZeroEntries(mat);CHKERRQ(ierr); 99 ierr = MatView(mat,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 100 101 ierr = ISDestroy(&irow);CHKERRQ(ierr); 102 ierr = ISDestroy(&icol);CHKERRQ(ierr); 103 ierr = PetscObjectReference((PetscObject)submat);CHKERRQ(ierr); 104 ierr = MatDestroySubMatrices(1,&submatrices);CHKERRQ(ierr); 105 ierr = MatDestroy(&submat);CHKERRQ(ierr); 106 ierr = MatDestroy(&mat);CHKERRQ(ierr); 107 ierr = PetscFinalize(); 108 return ierr; 109 } 110 111 /*TEST 112 113 test: 114 args: -mat_type aij 115 116 test: 117 suffix: 2 118 args: -mat_type dense 119 120 test: 121 suffix: 3 122 nsize: 3 123 args: -mat_type aij 124 125 test: 126 suffix: 4 127 nsize: 3 128 args: -mat_type dense 129 130 test: 131 suffix: 5 132 nsize: 3 133 args: -mat_type aij -test_all 134 135 TEST*/ 136