1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Tests the different MatColoring implementatons and ISColoringTestValid() \n\ 3c4762a1bSJed Brown Modifed from the code contributed by Ali Berk Kahraman. \n\n"; 4c4762a1bSJed Brown #include <petscmat.h> 5c4762a1bSJed Brown 69371c9d4SSatish Balay PetscErrorCode FormJacobian(Mat A) { 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++) { 16*48a46eb9SPierre 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)); 21c4762a1bSJed Brown PetscFunctionReturn(0); 22c4762a1bSJed Brown } 23c4762a1bSJed Brown 249371c9d4SSatish Balay int main(int argc, char *argv[]) { 25c4762a1bSJed Brown Mat J; 26c4762a1bSJed Brown PetscMPIInt size; 27c4762a1bSJed Brown PetscInt M = 8; 28c4762a1bSJed Brown ISColoring iscoloring; 29c4762a1bSJed Brown MatColoring coloring; 30c4762a1bSJed Brown 31327415f7SBarry Smith PetscFunctionBeginUser; 329566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 339566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 34c4762a1bSJed Brown 359566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &J)); 369566063dSJacob Faibussowitsch PetscCall(MatSetSizes(J, PETSC_DECIDE, PETSC_DECIDE, M, M)); 379566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 389566063dSJacob Faibussowitsch PetscCall(MatSetUp(J)); 39c4762a1bSJed Brown 409566063dSJacob Faibussowitsch PetscCall(FormJacobian(J)); 419566063dSJacob Faibussowitsch PetscCall(MatView(J, PETSC_VIEWER_STDOUT_WORLD)); 42c4762a1bSJed Brown 43c4762a1bSJed Brown /* 44c4762a1bSJed Brown Color the matrix, i.e. determine groups of columns that share no common 45c4762a1bSJed Brown rows. These columns in the Jacobian can all be computed simultaneously. 46c4762a1bSJed Brown */ 479566063dSJacob Faibussowitsch PetscCall(MatColoringCreate(J, &coloring)); 489566063dSJacob Faibussowitsch PetscCall(MatColoringSetType(coloring, MATCOLORINGGREEDY)); 499566063dSJacob Faibussowitsch PetscCall(MatColoringSetFromOptions(coloring)); 509566063dSJacob Faibussowitsch PetscCall(MatColoringApply(coloring, &iscoloring)); 51c4762a1bSJed Brown 52*48a46eb9SPierre Jolivet if (size == 1) PetscCall(MatISColoringTest(J, iscoloring)); 53c4762a1bSJed Brown 549566063dSJacob Faibussowitsch PetscCall(ISColoringDestroy(&iscoloring)); 559566063dSJacob Faibussowitsch PetscCall(MatColoringDestroy(&coloring)); 569566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 579566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 58b122ec5aSJacob Faibussowitsch return 0; 59c4762a1bSJed Brown } 60c4762a1bSJed Brown 61c4762a1bSJed Brown /*TEST 62c4762a1bSJed Brown 63c4762a1bSJed Brown test: 64c4762a1bSJed Brown suffix: sl 65dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 66c4762a1bSJed Brown args: -mat_coloring_type sl 67c4762a1bSJed Brown output_file: output/ex24_1.out 68c4762a1bSJed Brown 69c4762a1bSJed Brown test: 70c4762a1bSJed Brown suffix: lf 71dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 72c4762a1bSJed Brown args: -mat_coloring_type lf 73c4762a1bSJed Brown output_file: output/ex24_1.out 74c4762a1bSJed Brown 75c4762a1bSJed Brown test: 76c4762a1bSJed Brown suffix: id 77dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 78c4762a1bSJed Brown args: -mat_coloring_type id 79c4762a1bSJed Brown output_file: output/ex24_1.out 80c4762a1bSJed Brown 81c4762a1bSJed Brown TEST*/ 82