1 2 static char help[] = "Tests MatSeq(B)AIJSetColumnIndices().\n\n"; 3 4 #include <petscmat.h> 5 6 /* 7 Generate the following matrix: 8 9 1 0 3 10 1 2 3 11 0 0 3 12 */ 13 int main(int argc,char **args) 14 { 15 Mat A; 16 PetscScalar v; 17 PetscErrorCode ierr; 18 PetscInt i,j,rowlens[] = {2,3,1},cols[] = {0,2,0,1,2,2}; 19 PetscBool flg; 20 21 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 22 ierr = PetscOptionsHasName(NULL,NULL,"-baij",&flg);CHKERRQ(ierr); 23 if (flg) { 24 ierr = MatCreateSeqBAIJ(PETSC_COMM_WORLD,1,3,3,0,rowlens,&A);CHKERRQ(ierr); 25 ierr = MatSeqBAIJSetColumnIndices(A,cols);CHKERRQ(ierr); 26 } else { 27 ierr = MatCreateSeqAIJ(PETSC_COMM_WORLD,3,3,0,rowlens,&A);CHKERRQ(ierr); 28 ierr = MatSeqAIJSetColumnIndices(A,cols);CHKERRQ(ierr); 29 } 30 31 i = 0; j = 0; v = 1.0; 32 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 33 i = 0; j = 2; v = 3.0; 34 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 35 36 i = 1; j = 0; v = 1.0; 37 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 38 i = 1; j = 1; v = 2.0; 39 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 40 i = 1; j = 2; v = 3.0; 41 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 42 43 i = 2; j = 2; v = 3.0; 44 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 45 46 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 47 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 48 ierr = MatView(A,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 49 50 ierr = MatDestroy(&A);CHKERRQ(ierr); 51 ierr = PetscFinalize(); 52 return ierr; 53 } 54 55 56 /*TEST 57 58 test: 59 60 test: 61 suffix: 2 62 args: -baij 63 64 TEST*/ 65