1c4762a1bSJed Brown static char help[] = "Tests MatView()/MatLoad() with binary viewers for AIJ matrices.\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown #include <petscviewer.h> 5c4762a1bSJed Brown 6c4762a1bSJed Brown #include <petsc/private/hashtable.h> 7c4762a1bSJed Brown static PetscReal MakeValue(PetscInt i,PetscInt j,PetscInt M) 8c4762a1bSJed Brown { 9c4762a1bSJed Brown PetscHash_t h = PetscHashCombine(PetscHashInt(i),PetscHashInt(j)); 10c4762a1bSJed Brown return (PetscReal) ((h % 5 == 0) ? (1 + i + j*M) : 0); 11c4762a1bSJed Brown } 12c4762a1bSJed Brown 13c4762a1bSJed Brown static PetscErrorCode CheckValuesAIJ(Mat A) 14c4762a1bSJed Brown { 15c4762a1bSJed Brown PetscInt M,N,rstart,rend,i,j; 16c4762a1bSJed Brown PetscReal v,w; 17c4762a1bSJed Brown PetscScalar val; 18c4762a1bSJed Brown 19c4762a1bSJed Brown PetscFunctionBegin; 20*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetSize(A,&M,&N)); 21*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(A,&rstart,&rend)); 22c4762a1bSJed Brown for (i=rstart; i<rend; i++) { 23c4762a1bSJed Brown for (j=0; j<N; j++) { 24*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetValue(A,i,j,&val)); 25c4762a1bSJed Brown v = MakeValue(i,j,M); w = PetscRealPart(val); 262c71b3e2SJacob Faibussowitsch PetscCheckFalse(PetscAbsReal(v-w) > 0,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix entry (%" PetscInt_FMT ",%" PetscInt_FMT ") should be %g, got %g",i,j,(double)v,(double)w); 27c4762a1bSJed Brown } 28c4762a1bSJed Brown } 29c4762a1bSJed Brown PetscFunctionReturn(0); 30c4762a1bSJed Brown } 31c4762a1bSJed Brown 32c4762a1bSJed Brown int main(int argc,char **args) 33c4762a1bSJed Brown { 34c4762a1bSJed Brown Mat A; 35c4762a1bSJed Brown PetscInt M = 11,N = 13; 36c4762a1bSJed Brown PetscInt rstart,rend,i,j; 37c4762a1bSJed Brown PetscErrorCode ierr; 38c4762a1bSJed Brown PetscViewer view; 39c4762a1bSJed Brown 40c4762a1bSJed Brown ierr = PetscInitialize(&argc,&args,NULL,help);if (ierr) return ierr; 41c4762a1bSJed Brown /* 42c4762a1bSJed Brown Create a parallel AIJ matrix shared by all processors 43c4762a1bSJed Brown */ 44c4762a1bSJed Brown ierr = MatCreateAIJ(PETSC_COMM_WORLD, 45c4762a1bSJed Brown PETSC_DECIDE,PETSC_DECIDE, 46c4762a1bSJed Brown M,N, 47c4762a1bSJed Brown PETSC_DECIDE,NULL, 48c4762a1bSJed Brown PETSC_DECIDE,NULL, 49c4762a1bSJed Brown &A);CHKERRQ(ierr); 50c4762a1bSJed Brown 51c4762a1bSJed Brown /* 52c4762a1bSJed Brown Set values into the matrix 53c4762a1bSJed Brown */ 54*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatGetOwnershipRange(A,&rstart,&rend)); 55c4762a1bSJed Brown for (i=rstart; i<rend; i++) { 56c4762a1bSJed Brown for (j=0; j<N; j++) { 57c4762a1bSJed Brown PetscReal v = MakeValue(i,j,M); 58c4762a1bSJed Brown if (PetscAbsReal(v) > 0) { 59*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetValue(A,i,j,v,INSERT_VALUES)); 60c4762a1bSJed Brown } 61c4762a1bSJed Brown } 62c4762a1bSJed Brown } 63*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 64*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 65*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatViewFromOptions(A,NULL,"-mat_base_view")); 66c4762a1bSJed Brown 67c4762a1bSJed Brown /* 68c4762a1bSJed Brown Store the binary matrix to a file 69c4762a1bSJed Brown */ 70*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "matrix.dat", FILE_MODE_WRITE, &view)); 71c4762a1bSJed Brown for (i=0; i<3; i++) { 72*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatView(A,view)); 73c4762a1bSJed Brown } 74*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDestroy(&view)); 75*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&A)); 76c4762a1bSJed Brown 77c4762a1bSJed Brown /* 78c4762a1bSJed Brown Now reload the matrix and check its values 79c4762a1bSJed Brown */ 80*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"matrix.dat",FILE_MODE_READ,&view)); 81*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 82*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(A,MATAIJ)); 83c4762a1bSJed Brown for (i=0; i<3; i++) { 84*5f80ce2aSJacob Faibussowitsch if (i > 0) CHKERRQ(MatZeroEntries(A)); 85*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatLoad(A,view)); 86*5f80ce2aSJacob Faibussowitsch CHKERRQ(CheckValuesAIJ(A)); 87c4762a1bSJed Brown } 88*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDestroy(&view)); 89*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatViewFromOptions(A,NULL,"-mat_load_view")); 90*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&A)); 91c4762a1bSJed Brown 92c4762a1bSJed Brown /* 93c4762a1bSJed Brown Reload in SEQAIJ matrix and check its values 94c4762a1bSJed Brown */ 95*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_SELF,"matrix.dat",FILE_MODE_READ,&view)); 96*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(PETSC_COMM_SELF,&A)); 97*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(A,MATSEQAIJ)); 98c4762a1bSJed Brown for (i=0; i<3; i++) { 99*5f80ce2aSJacob Faibussowitsch if (i > 0) CHKERRQ(MatZeroEntries(A)); 100*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatLoad(A,view)); 101*5f80ce2aSJacob Faibussowitsch CHKERRQ(CheckValuesAIJ(A)); 102c4762a1bSJed Brown } 103*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDestroy(&view)); 104*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&A)); 105c4762a1bSJed Brown 106c4762a1bSJed Brown /* 107c4762a1bSJed Brown Reload in MPIAIJ matrix and check its values 108c4762a1bSJed Brown */ 109*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"matrix.dat",FILE_MODE_READ,&view)); 110*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 111*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatSetType(A,MATMPIAIJ)); 112c4762a1bSJed Brown for (i=0; i<3; i++) { 113*5f80ce2aSJacob Faibussowitsch if (i > 0) CHKERRQ(MatZeroEntries(A)); 114*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatLoad(A,view)); 115*5f80ce2aSJacob Faibussowitsch CHKERRQ(CheckValuesAIJ(A)); 116c4762a1bSJed Brown } 117*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerDestroy(&view)); 118*5f80ce2aSJacob Faibussowitsch CHKERRQ(MatDestroy(&A)); 119c4762a1bSJed Brown 120c4762a1bSJed Brown ierr = PetscFinalize(); 121c4762a1bSJed Brown return ierr; 122c4762a1bSJed Brown } 123c4762a1bSJed Brown 124c4762a1bSJed Brown /*TEST 125c4762a1bSJed Brown 126c4762a1bSJed Brown testset: 127c4762a1bSJed Brown args: -viewer_binary_mpiio 0 128c4762a1bSJed Brown output_file: output/ex44.out 129c4762a1bSJed Brown test: 130c4762a1bSJed Brown suffix: stdio_1 131c4762a1bSJed Brown nsize: 1 132c4762a1bSJed Brown test: 133c4762a1bSJed Brown suffix: stdio_2 134c4762a1bSJed Brown nsize: 2 135c4762a1bSJed Brown test: 136c4762a1bSJed Brown suffix: stdio_3 137c4762a1bSJed Brown nsize: 3 138c4762a1bSJed Brown test: 139c4762a1bSJed Brown suffix: stdio_4 140c4762a1bSJed Brown nsize: 4 141c4762a1bSJed Brown test: 142c4762a1bSJed Brown suffix: stdio_15 143c4762a1bSJed Brown nsize: 15 144c4762a1bSJed Brown 145c4762a1bSJed Brown testset: 146c4762a1bSJed Brown requires: mpiio 147c4762a1bSJed Brown args: -viewer_binary_mpiio 1 148c4762a1bSJed Brown output_file: output/ex44.out 149c4762a1bSJed Brown test: 150c4762a1bSJed Brown suffix: mpiio_1 151c4762a1bSJed Brown nsize: 1 152c4762a1bSJed Brown test: 153c4762a1bSJed Brown suffix: mpiio_2 154c4762a1bSJed Brown nsize: 2 155c4762a1bSJed Brown test: 156c4762a1bSJed Brown suffix: mpiio_3 157c4762a1bSJed Brown nsize: 3 158c4762a1bSJed Brown test: 159c4762a1bSJed Brown suffix: mpiio_4 160c4762a1bSJed Brown nsize: 4 161c4762a1bSJed Brown test: 162c4762a1bSJed Brown suffix: mpiio_15 163c4762a1bSJed Brown nsize: 15 164c4762a1bSJed Brown 165c4762a1bSJed Brown TEST*/ 166