1da81f932SPierre Jolivet static char help[] = "Tests the different MatColoring implementations and ISColoringTestValid() \n\
2da81f932SPierre Jolivet Modified from the code contributed by Ali Berk Kahraman. \n\n";
3c4762a1bSJed Brown #include <petscmat.h>
4c4762a1bSJed Brown
FormJacobian(Mat A)5d71ae5a4SJacob Faibussowitsch PetscErrorCode FormJacobian(Mat A)
6d71ae5a4SJacob Faibussowitsch {
7c4762a1bSJed Brown PetscInt M, ownbegin, ownend, i, j;
8c4762a1bSJed Brown PetscScalar dummy = 0.0;
9c4762a1bSJed Brown
10c4762a1bSJed Brown PetscFunctionBeginUser;
119566063dSJacob Faibussowitsch PetscCall(MatGetSize(A, &M, NULL));
129566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRange(A, &ownbegin, &ownend));
13c4762a1bSJed Brown
14c4762a1bSJed Brown for (i = ownbegin; i < ownend; i++) {
15c4762a1bSJed Brown for (j = i - 3; j < i + 3; j++) {
1648a46eb9SPierre Jolivet if (j >= 0 && j < M) PetscCall(MatSetValues(A, 1, &i, 1, &j, &dummy, INSERT_VALUES));
17c4762a1bSJed Brown }
18c4762a1bSJed Brown }
199566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
209566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS);
22c4762a1bSJed Brown }
23c4762a1bSJed Brown
main(int argc,char * argv[])24d71ae5a4SJacob Faibussowitsch int main(int argc, char *argv[])
25d71ae5a4SJacob Faibussowitsch {
26c4762a1bSJed Brown Mat J;
27c4762a1bSJed Brown PetscMPIInt size;
28c4762a1bSJed Brown PetscInt M = 8;
29c4762a1bSJed Brown ISColoring iscoloring;
30c4762a1bSJed Brown MatColoring coloring;
31c4762a1bSJed Brown
32327415f7SBarry Smith PetscFunctionBeginUser;
33*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help));
349566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
35c4762a1bSJed Brown
369566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &J));
379566063dSJacob Faibussowitsch PetscCall(MatSetSizes(J, PETSC_DECIDE, PETSC_DECIDE, M, M));
389566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J));
399566063dSJacob Faibussowitsch PetscCall(MatSetUp(J));
40c4762a1bSJed Brown
419566063dSJacob Faibussowitsch PetscCall(FormJacobian(J));
429566063dSJacob Faibussowitsch PetscCall(MatView(J, PETSC_VIEWER_STDOUT_WORLD));
43c4762a1bSJed Brown
44c4762a1bSJed Brown /*
45c4762a1bSJed Brown Color the matrix, i.e. determine groups of columns that share no common
46c4762a1bSJed Brown rows. These columns in the Jacobian can all be computed simultaneously.
47c4762a1bSJed Brown */
489566063dSJacob Faibussowitsch PetscCall(MatColoringCreate(J, &coloring));
499566063dSJacob Faibussowitsch PetscCall(MatColoringSetType(coloring, MATCOLORINGGREEDY));
509566063dSJacob Faibussowitsch PetscCall(MatColoringSetFromOptions(coloring));
519566063dSJacob Faibussowitsch PetscCall(MatColoringApply(coloring, &iscoloring));
52c4762a1bSJed Brown
5348a46eb9SPierre Jolivet if (size == 1) PetscCall(MatISColoringTest(J, iscoloring));
54c4762a1bSJed Brown
559566063dSJacob Faibussowitsch PetscCall(ISColoringDestroy(&iscoloring));
569566063dSJacob Faibussowitsch PetscCall(MatColoringDestroy(&coloring));
579566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J));
589566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
59b122ec5aSJacob Faibussowitsch return 0;
60c4762a1bSJed Brown }
61c4762a1bSJed Brown
62c4762a1bSJed Brown /*TEST
63c4762a1bSJed Brown
64c4762a1bSJed Brown test:
65c4762a1bSJed Brown suffix: sl
66dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES)
67c4762a1bSJed Brown args: -mat_coloring_type sl
68c4762a1bSJed Brown output_file: output/ex24_1.out
69c4762a1bSJed Brown
70c4762a1bSJed Brown test:
71c4762a1bSJed Brown suffix: lf
72dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES)
73c4762a1bSJed Brown args: -mat_coloring_type lf
74c4762a1bSJed Brown output_file: output/ex24_1.out
75c4762a1bSJed Brown
76c4762a1bSJed Brown test:
77c4762a1bSJed Brown suffix: id
78dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES)
79c4762a1bSJed Brown args: -mat_coloring_type id
80c4762a1bSJed Brown output_file: output/ex24_1.out
81c4762a1bSJed Brown
82c4762a1bSJed Brown TEST*/
83