xref: /petsc/src/mat/tests/ex13.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Tests copying and ordering uniprocessor row-based sparse matrices.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscmat.h>
5c4762a1bSJed Brown 
6c4762a1bSJed Brown int main(int argc,char **args)
7c4762a1bSJed Brown {
8c4762a1bSJed Brown   Mat            C,A;
9c4762a1bSJed Brown   PetscInt       i,j,m = 5,n = 5,Ii,J;
10c4762a1bSJed Brown   PetscScalar    v;
11c4762a1bSJed Brown   IS             perm,iperm;
12c4762a1bSJed Brown 
13*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help));
145f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreateSeqAIJ(PETSC_COMM_SELF,m*n,m*n,5,NULL,&C));
15c4762a1bSJed Brown   /* create the matrix for the five point stencil, YET AGAIN*/
16c4762a1bSJed Brown   for (i=0; i<m; i++) {
17c4762a1bSJed Brown     for (j=0; j<n; j++) {
18c4762a1bSJed Brown       v = -1.0;  Ii = j + n*i;
195f80ce2aSJacob Faibussowitsch       if (i>0)   {J = Ii - n; CHKERRQ(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
205f80ce2aSJacob Faibussowitsch       if (i<m-1) {J = Ii + n; CHKERRQ(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
215f80ce2aSJacob Faibussowitsch       if (j>0)   {J = Ii - 1; CHKERRQ(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
225f80ce2aSJacob Faibussowitsch       if (j<n-1) {J = Ii + 1; CHKERRQ(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
235f80ce2aSJacob Faibussowitsch       v = 4.0; CHKERRQ(MatSetValues(C,1,&Ii,1,&Ii,&v,INSERT_VALUES));
24c4762a1bSJed Brown     }
25c4762a1bSJed Brown   }
265f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY));
275f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY));
28c4762a1bSJed Brown 
295f80ce2aSJacob Faibussowitsch   CHKERRQ(MatConvert(C,MATSAME,MAT_INITIAL_MATRIX,&A));
30c4762a1bSJed Brown 
315f80ce2aSJacob Faibussowitsch   CHKERRQ(MatGetOrdering(A,MATORDERINGND,&perm,&iperm));
325f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(perm,PETSC_VIEWER_STDOUT_SELF));
335f80ce2aSJacob Faibussowitsch   CHKERRQ(ISView(iperm,PETSC_VIEWER_STDOUT_SELF));
345f80ce2aSJacob Faibussowitsch   CHKERRQ(MatView(A,PETSC_VIEWER_STDOUT_SELF));
35c4762a1bSJed Brown 
365f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&perm));
375f80ce2aSJacob Faibussowitsch   CHKERRQ(ISDestroy(&iperm));
385f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&C));
395f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&A));
40*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
41*b122ec5aSJacob Faibussowitsch   return 0;
42c4762a1bSJed Brown }
43c4762a1bSJed Brown 
44c4762a1bSJed Brown /*TEST
45c4762a1bSJed Brown 
46c4762a1bSJed Brown    test:
47c4762a1bSJed Brown       filter: grep -v "MPI processes"
48c4762a1bSJed Brown 
49c4762a1bSJed Brown TEST*/
50