xref: /petsc/src/mat/tests/ex13.c (revision 327415f76d85372a4417cf1aaa14db707d4d6c04)
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*327415f7SBarry Smith   PetscFunctionBeginUser;
149566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc,&args,(char*)0,help));
159566063dSJacob Faibussowitsch   PetscCall(MatCreateSeqAIJ(PETSC_COMM_SELF,m*n,m*n,5,NULL,&C));
16c4762a1bSJed Brown   /* create the matrix for the five point stencil, YET AGAIN*/
17c4762a1bSJed Brown   for (i=0; i<m; i++) {
18c4762a1bSJed Brown     for (j=0; j<n; j++) {
19c4762a1bSJed Brown       v = -1.0;  Ii = j + n*i;
209566063dSJacob Faibussowitsch       if (i>0)   {J = Ii - n; PetscCall(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
219566063dSJacob Faibussowitsch       if (i<m-1) {J = Ii + n; PetscCall(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
229566063dSJacob Faibussowitsch       if (j>0)   {J = Ii - 1; PetscCall(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
239566063dSJacob Faibussowitsch       if (j<n-1) {J = Ii + 1; PetscCall(MatSetValues(C,1,&Ii,1,&J,&v,INSERT_VALUES));}
249566063dSJacob Faibussowitsch       v = 4.0; PetscCall(MatSetValues(C,1,&Ii,1,&Ii,&v,INSERT_VALUES));
25c4762a1bSJed Brown     }
26c4762a1bSJed Brown   }
279566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY));
289566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY));
29c4762a1bSJed Brown 
309566063dSJacob Faibussowitsch   PetscCall(MatConvert(C,MATSAME,MAT_INITIAL_MATRIX,&A));
31c4762a1bSJed Brown 
329566063dSJacob Faibussowitsch   PetscCall(MatGetOrdering(A,MATORDERINGND,&perm,&iperm));
339566063dSJacob Faibussowitsch   PetscCall(ISView(perm,PETSC_VIEWER_STDOUT_SELF));
349566063dSJacob Faibussowitsch   PetscCall(ISView(iperm,PETSC_VIEWER_STDOUT_SELF));
359566063dSJacob Faibussowitsch   PetscCall(MatView(A,PETSC_VIEWER_STDOUT_SELF));
36c4762a1bSJed Brown 
379566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&perm));
389566063dSJacob Faibussowitsch   PetscCall(ISDestroy(&iperm));
399566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&C));
409566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&A));
419566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
42b122ec5aSJacob Faibussowitsch   return 0;
43c4762a1bSJed Brown }
44c4762a1bSJed Brown 
45c4762a1bSJed Brown /*TEST
46c4762a1bSJed Brown 
47c4762a1bSJed Brown    test:
488cc725e6SPierre Jolivet       filter: grep -v " MPI process"
49c4762a1bSJed Brown 
50c4762a1bSJed Brown TEST*/
51