xref: /petsc/src/mat/tests/ex65.c (revision 9371c9d470a9602b6d10a8bf50c9b2280a79e45a)
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 
6*9371c9d4SSatish Balay int main(int argc, char **args) {
7c4762a1bSJed Brown   Mat         A;
8c4762a1bSJed Brown   PetscInt    m = 100, n = 11, js[11], i, j, cnt;
9c4762a1bSJed Brown   PetscScalar values[11];
10c4762a1bSJed Brown   PetscViewer view;
11c4762a1bSJed Brown 
12327415f7SBarry Smith   PetscFunctionBeginUser;
139566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
149566063dSJacob Faibussowitsch   PetscCall(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) {
21*9371c9d4SSatish Balay       for (j = 0; j < n; j += 2) { js[cnt++] = j; }
22c4762a1bSJed Brown     } else {
23c4762a1bSJed Brown       ;
24c4762a1bSJed Brown     }
259566063dSJacob Faibussowitsch     PetscCall(MatSetValues(A, 1, &i, cnt, js, values, INSERT_VALUES));
26c4762a1bSJed Brown   }
279566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
289566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
29c4762a1bSJed Brown 
309566063dSJacob Faibussowitsch   PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "rect", FILE_MODE_WRITE, &view));
319566063dSJacob Faibussowitsch   PetscCall(MatView(A, view));
329566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&view));
33c4762a1bSJed Brown 
349566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&A));
35c4762a1bSJed Brown 
369566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
37b122ec5aSJacob Faibussowitsch   return 0;
38c4762a1bSJed Brown }
39c4762a1bSJed Brown 
40c4762a1bSJed Brown /*TEST
41c4762a1bSJed Brown 
42c4762a1bSJed Brown    test:
43c4762a1bSJed Brown 
44c4762a1bSJed Brown TEST*/
45