xref: /petsc/src/mat/tests/ex65.c (revision d71ae5a4db6382e7f06317b8d368875286fe9008)
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*d71ae5a4SJacob Faibussowitsch int main(int argc, char **args)
7*d71ae5a4SJacob Faibussowitsch {
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 
13327415f7SBarry Smith   PetscFunctionBeginUser;
149566063dSJacob Faibussowitsch   PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
159566063dSJacob Faibussowitsch   PetscCall(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) {
22ad540459SPierre Jolivet       for (j = 0; j < n; j += 2) js[cnt++] = j;
23c4762a1bSJed Brown     } else {
24c4762a1bSJed Brown       ;
25c4762a1bSJed Brown     }
269566063dSJacob Faibussowitsch     PetscCall(MatSetValues(A, 1, &i, cnt, js, values, INSERT_VALUES));
27c4762a1bSJed Brown   }
289566063dSJacob Faibussowitsch   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
299566063dSJacob Faibussowitsch   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
30c4762a1bSJed Brown 
319566063dSJacob Faibussowitsch   PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, "rect", FILE_MODE_WRITE, &view));
329566063dSJacob Faibussowitsch   PetscCall(MatView(A, view));
339566063dSJacob Faibussowitsch   PetscCall(PetscViewerDestroy(&view));
34c4762a1bSJed Brown 
359566063dSJacob Faibussowitsch   PetscCall(MatDestroy(&A));
36c4762a1bSJed Brown 
379566063dSJacob Faibussowitsch   PetscCall(PetscFinalize());
38b122ec5aSJacob Faibussowitsch   return 0;
39c4762a1bSJed Brown }
40c4762a1bSJed Brown 
41c4762a1bSJed Brown /*TEST
42c4762a1bSJed Brown 
43c4762a1bSJed Brown    test:
44c4762a1bSJed Brown 
45c4762a1bSJed Brown TEST*/
46