xref: /petsc/src/mat/tests/ex65.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Saves a rectangular sparse matrix to disk.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscmat.h>
5c4762a1bSJed Brown 
6c4762a1bSJed Brown int main(int argc,char **args)
7c4762a1bSJed Brown {
8c4762a1bSJed Brown   Mat            A;
9c4762a1bSJed Brown   PetscInt       m = 100,n = 11,js[11],i,j,cnt;
10c4762a1bSJed Brown   PetscScalar    values[11];
11c4762a1bSJed Brown   PetscViewer    view;
12c4762a1bSJed Brown 
13*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help));
145f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreateSeqAIJ(PETSC_COMM_WORLD,m,n,20,0,&A));
15c4762a1bSJed Brown 
16c4762a1bSJed Brown   for (i=0; i<n; i++) values[i] = (PetscReal)i;
17c4762a1bSJed Brown 
18c4762a1bSJed Brown   for (i=0; i<m; i++) {
19c4762a1bSJed Brown     cnt = 0;
20c4762a1bSJed Brown     if (i % 2) {
21c4762a1bSJed Brown       for (j=0; j<n; j += 2) {
22c4762a1bSJed Brown         js[cnt++] = j;
23c4762a1bSJed Brown       }
24c4762a1bSJed Brown     } else {
25c4762a1bSJed Brown       ;
26c4762a1bSJed Brown     }
275f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetValues(A,1,&i,cnt,js,values,INSERT_VALUES));
28c4762a1bSJed Brown   }
295f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY));
305f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY));
31c4762a1bSJed Brown 
325f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"rect",FILE_MODE_WRITE,&view));
335f80ce2aSJacob Faibussowitsch   CHKERRQ(MatView(A,view));
345f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDestroy(&view));
35c4762a1bSJed Brown 
365f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&A));
37c4762a1bSJed Brown 
38*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
39*b122ec5aSJacob Faibussowitsch   return 0;
40c4762a1bSJed Brown }
41c4762a1bSJed Brown 
42c4762a1bSJed Brown /*TEST
43c4762a1bSJed Brown 
44c4762a1bSJed Brown    test:
45c4762a1bSJed Brown 
46c4762a1bSJed Brown TEST*/
47