1c4762a1bSJed Brown static char help[] = "Tests the use of MatZeroRows() for uniprocessor matrices.\n\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscmat.h>
4c4762a1bSJed Brown
main(int argc,char ** args)5d71ae5a4SJacob Faibussowitsch int main(int argc, char **args)
6d71ae5a4SJacob Faibussowitsch {
7c4762a1bSJed Brown Mat C;
8c4762a1bSJed Brown PetscInt i, j, m = 5, n = 5, Ii, J;
9c4762a1bSJed Brown PetscScalar v, five = 5.0;
10c4762a1bSJed Brown IS isrow;
11c4762a1bSJed Brown PetscBool keepnonzeropattern;
12c4762a1bSJed Brown
13327415f7SBarry Smith PetscFunctionBeginUser;
14*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &args, NULL, help));
15c4762a1bSJed Brown /* create the matrix for the five point stencil, YET AGAIN*/
169566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_SELF, &C));
179566063dSJacob Faibussowitsch PetscCall(MatSetSizes(C, PETSC_DECIDE, PETSC_DECIDE, m * n, m * n));
189566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(C));
199566063dSJacob Faibussowitsch PetscCall(MatSetUp(C));
20c4762a1bSJed Brown for (i = 0; i < m; i++) {
21c4762a1bSJed Brown for (j = 0; j < n; j++) {
229371c9d4SSatish Balay v = -1.0;
239371c9d4SSatish Balay Ii = j + n * i;
249371c9d4SSatish Balay if (i > 0) {
259371c9d4SSatish Balay J = Ii - n;
269371c9d4SSatish Balay PetscCall(MatSetValues(C, 1, &Ii, 1, &J, &v, INSERT_VALUES));
279371c9d4SSatish Balay }
289371c9d4SSatish Balay if (i < m - 1) {
299371c9d4SSatish Balay J = Ii + n;
309371c9d4SSatish Balay PetscCall(MatSetValues(C, 1, &Ii, 1, &J, &v, INSERT_VALUES));
319371c9d4SSatish Balay }
329371c9d4SSatish Balay if (j > 0) {
339371c9d4SSatish Balay J = Ii - 1;
349371c9d4SSatish Balay PetscCall(MatSetValues(C, 1, &Ii, 1, &J, &v, INSERT_VALUES));
359371c9d4SSatish Balay }
369371c9d4SSatish Balay if (j < n - 1) {
379371c9d4SSatish Balay J = Ii + 1;
389371c9d4SSatish Balay PetscCall(MatSetValues(C, 1, &Ii, 1, &J, &v, INSERT_VALUES));
399371c9d4SSatish Balay }
409371c9d4SSatish Balay v = 4.0;
419371c9d4SSatish Balay PetscCall(MatSetValues(C, 1, &Ii, 1, &Ii, &v, INSERT_VALUES));
42c4762a1bSJed Brown }
43c4762a1bSJed Brown }
449566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(C, MAT_FINAL_ASSEMBLY));
459566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(C, MAT_FINAL_ASSEMBLY));
46c4762a1bSJed Brown
479566063dSJacob Faibussowitsch PetscCall(ISCreateStride(PETSC_COMM_SELF, (m * n) / 2, 0, 2, &isrow));
48c4762a1bSJed Brown
499566063dSJacob Faibussowitsch PetscCall(PetscOptionsHasName(NULL, NULL, "-keep_nonzero_pattern", &keepnonzeropattern));
501baa6e33SBarry Smith if (keepnonzeropattern) PetscCall(MatSetOption(C, MAT_KEEP_NONZERO_PATTERN, PETSC_TRUE));
51c4762a1bSJed Brown
529566063dSJacob Faibussowitsch PetscCall(MatZeroRowsIS(C, isrow, five, 0, 0));
53c4762a1bSJed Brown
549566063dSJacob Faibussowitsch PetscCall(MatView(C, PETSC_VIEWER_STDOUT_SELF));
55c4762a1bSJed Brown
569566063dSJacob Faibussowitsch PetscCall(ISDestroy(&isrow));
579566063dSJacob Faibussowitsch PetscCall(MatDestroy(&C));
589566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
59b122ec5aSJacob Faibussowitsch return 0;
60c4762a1bSJed Brown }
61c4762a1bSJed Brown
62c4762a1bSJed Brown /*TEST
63c4762a1bSJed Brown
64c4762a1bSJed Brown test:
65c4762a1bSJed Brown
66c4762a1bSJed Brown test:
67c4762a1bSJed Brown suffix: 2
68c4762a1bSJed Brown args: -mat_type seqbaij -mat_block_size 5
69c4762a1bSJed Brown
70c4762a1bSJed Brown test:
71c4762a1bSJed Brown suffix: 3
72c4762a1bSJed Brown args: -keep_nonzero_pattern
73c4762a1bSJed Brown
74c4762a1bSJed Brown test:
75c4762a1bSJed Brown suffix: 4
76c4762a1bSJed Brown args: -keep_nonzero_pattern -mat_type seqbaij -mat_block_size 5
77c4762a1bSJed Brown
78c4762a1bSJed Brown TEST*/
79