xref: /petsc/src/mat/tests/ex65.c (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
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   PetscErrorCode ierr;
10c4762a1bSJed Brown   PetscInt       m = 100,n = 11,js[11],i,j,cnt;
11c4762a1bSJed Brown   PetscScalar    values[11];
12c4762a1bSJed Brown   PetscViewer    view;
13c4762a1bSJed Brown 
14c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
15*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreateSeqAIJ(PETSC_COMM_WORLD,m,n,20,0,&A));
16c4762a1bSJed Brown 
17c4762a1bSJed Brown   for (i=0; i<n; i++) values[i] = (PetscReal)i;
18c4762a1bSJed Brown 
19c4762a1bSJed Brown   for (i=0; i<m; i++) {
20c4762a1bSJed Brown     cnt = 0;
21c4762a1bSJed Brown     if (i % 2) {
22c4762a1bSJed Brown       for (j=0; j<n; j += 2) {
23c4762a1bSJed Brown         js[cnt++] = j;
24c4762a1bSJed Brown       }
25c4762a1bSJed Brown     } else {
26c4762a1bSJed Brown       ;
27c4762a1bSJed Brown     }
28*5f80ce2aSJacob Faibussowitsch     CHKERRQ(MatSetValues(A,1,&i,cnt,js,values,INSERT_VALUES));
29c4762a1bSJed Brown   }
30*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY));
31*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY));
32c4762a1bSJed Brown 
33*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,"rect",FILE_MODE_WRITE,&view));
34*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatView(A,view));
35*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDestroy(&view));
36c4762a1bSJed Brown 
37*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&A));
38c4762a1bSJed Brown 
39c4762a1bSJed Brown   ierr = PetscFinalize();
40c4762a1bSJed Brown   return ierr;
41c4762a1bSJed Brown }
42c4762a1bSJed Brown 
43c4762a1bSJed Brown /*TEST
44c4762a1bSJed Brown 
45c4762a1bSJed Brown    test:
46c4762a1bSJed Brown 
47c4762a1bSJed Brown TEST*/
48