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 6d71ae5a4SJacob Faibussowitsch PetscErrorCode FormJacobian(Mat A) 7d71ae5a4SJacob Faibussowitsch { 8c4762a1bSJed Brown PetscInt M, ownbegin, ownend, i, j; 9c4762a1bSJed Brown PetscScalar dummy = 0.0; 10c4762a1bSJed Brown 11c4762a1bSJed Brown PetscFunctionBeginUser; 129566063dSJacob Faibussowitsch PetscCall(MatGetSize(A, &M, NULL)); 139566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRange(A, &ownbegin, &ownend)); 14c4762a1bSJed Brown 15c4762a1bSJed Brown for (i = ownbegin; i < ownend; i++) { 16c4762a1bSJed Brown for (j = i - 3; j < i + 3; j++) { 1748a46eb9SPierre Jolivet if (j >= 0 && j < M) PetscCall(MatSetValues(A, 1, &i, 1, &j, &dummy, INSERT_VALUES)); 18c4762a1bSJed Brown } 19c4762a1bSJed Brown } 209566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 219566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 22*3ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23c4762a1bSJed Brown } 24c4762a1bSJed Brown 25d71ae5a4SJacob Faibussowitsch int main(int argc, char *argv[]) 26d71ae5a4SJacob Faibussowitsch { 27c4762a1bSJed Brown Mat J; 28c4762a1bSJed Brown PetscMPIInt size; 29c4762a1bSJed Brown PetscInt M = 8; 30c4762a1bSJed Brown ISColoring iscoloring; 31c4762a1bSJed Brown MatColoring coloring; 32c4762a1bSJed Brown 33327415f7SBarry Smith PetscFunctionBeginUser; 349566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, (char *)0, help)); 359566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 36c4762a1bSJed Brown 379566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &J)); 389566063dSJacob Faibussowitsch PetscCall(MatSetSizes(J, PETSC_DECIDE, PETSC_DECIDE, M, M)); 399566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 409566063dSJacob Faibussowitsch PetscCall(MatSetUp(J)); 41c4762a1bSJed Brown 429566063dSJacob Faibussowitsch PetscCall(FormJacobian(J)); 439566063dSJacob Faibussowitsch PetscCall(MatView(J, PETSC_VIEWER_STDOUT_WORLD)); 44c4762a1bSJed Brown 45c4762a1bSJed Brown /* 46c4762a1bSJed Brown Color the matrix, i.e. determine groups of columns that share no common 47c4762a1bSJed Brown rows. These columns in the Jacobian can all be computed simultaneously. 48c4762a1bSJed Brown */ 499566063dSJacob Faibussowitsch PetscCall(MatColoringCreate(J, &coloring)); 509566063dSJacob Faibussowitsch PetscCall(MatColoringSetType(coloring, MATCOLORINGGREEDY)); 519566063dSJacob Faibussowitsch PetscCall(MatColoringSetFromOptions(coloring)); 529566063dSJacob Faibussowitsch PetscCall(MatColoringApply(coloring, &iscoloring)); 53c4762a1bSJed Brown 5448a46eb9SPierre Jolivet if (size == 1) PetscCall(MatISColoringTest(J, iscoloring)); 55c4762a1bSJed Brown 569566063dSJacob Faibussowitsch PetscCall(ISColoringDestroy(&iscoloring)); 579566063dSJacob Faibussowitsch PetscCall(MatColoringDestroy(&coloring)); 589566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 599566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 60b122ec5aSJacob Faibussowitsch return 0; 61c4762a1bSJed Brown } 62c4762a1bSJed Brown 63c4762a1bSJed Brown /*TEST 64c4762a1bSJed Brown 65c4762a1bSJed Brown test: 66c4762a1bSJed Brown suffix: sl 67dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 68c4762a1bSJed Brown args: -mat_coloring_type sl 69c4762a1bSJed Brown output_file: output/ex24_1.out 70c4762a1bSJed Brown 71c4762a1bSJed Brown test: 72c4762a1bSJed Brown suffix: lf 73dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 74c4762a1bSJed Brown args: -mat_coloring_type lf 75c4762a1bSJed Brown output_file: output/ex24_1.out 76c4762a1bSJed Brown 77c4762a1bSJed Brown test: 78c4762a1bSJed Brown suffix: id 79dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 80c4762a1bSJed Brown args: -mat_coloring_type id 81c4762a1bSJed Brown output_file: output/ex24_1.out 82c4762a1bSJed Brown 83c4762a1bSJed Brown TEST*/ 84