xref: /petsc/src/mat/tests/ex199.c (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
1c4762a1bSJed Brown 
2c4762a1bSJed Brown static char help[] = "Tests the different MatColoring implementatons.\n\n";
3c4762a1bSJed Brown 
4c4762a1bSJed Brown #include <petscmat.h>
5c4762a1bSJed Brown 
6c4762a1bSJed Brown int main(int argc,char **args)
7c4762a1bSJed Brown {
8c4762a1bSJed Brown   Mat            C;
9c4762a1bSJed Brown   PetscErrorCode ierr;
10c4762a1bSJed Brown   PetscViewer    viewer;
11c4762a1bSJed Brown   char           file[128];
12c4762a1bSJed Brown   PetscBool      flg;
13c4762a1bSJed Brown   MatColoring    ctx;
14c4762a1bSJed Brown   ISColoring     coloring;
15c4762a1bSJed Brown   PetscMPIInt    size;
16c4762a1bSJed Brown 
17c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
18*5f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
19c4762a1bSJed Brown 
20*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg));
212c71b3e2SJacob Faibussowitsch   PetscCheckFalse(!flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must use -f filename to load sparse matrix");
22*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&viewer));
23*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PETSC_COMM_WORLD,&C));
24*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatLoad(C,viewer));
25*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDestroy(&viewer));
26c4762a1bSJed Brown 
27*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringCreate(C,&ctx));
28*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringSetFromOptions(ctx));
29*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringApply(ctx,&coloring));
30*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringTest(ctx,coloring));
31c4762a1bSJed Brown   if (size == 1) {
32c4762a1bSJed Brown     /* jp, power and greedy have bug -- need to be fixed */
33*5f80ce2aSJacob Faibussowitsch     CHKERRQ(MatISColoringTest(C,coloring));
34c4762a1bSJed Brown   }
35c4762a1bSJed Brown 
36c4762a1bSJed Brown   /* Free data structures */
37*5f80ce2aSJacob Faibussowitsch   CHKERRQ(ISColoringDestroy(&coloring));
38*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringDestroy(&ctx));
39*5f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&C));
40c4762a1bSJed Brown   ierr = PetscFinalize();
41c4762a1bSJed Brown   return ierr;
42c4762a1bSJed Brown }
43c4762a1bSJed Brown 
44c4762a1bSJed Brown /*TEST
45c4762a1bSJed Brown 
46c4762a1bSJed Brown    test:
47c4762a1bSJed Brown       nsize: {{3}}
48dfd57a17SPierre Jolivet       requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES)
49c4762a1bSJed Brown       args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{ jp power natural greedy}} -mat_coloring_distance {{ 1 2}}
50c4762a1bSJed Brown 
51c4762a1bSJed Brown    test:
52c4762a1bSJed Brown       suffix: 2
53c4762a1bSJed Brown       nsize: {{1 2}}
54dfd57a17SPierre Jolivet       requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES)
55c4762a1bSJed Brown       args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{  sl lf id }} -mat_coloring_distance 2
56c4762a1bSJed Brown       output_file: output/ex199_1.out
57c4762a1bSJed Brown 
58c4762a1bSJed Brown TEST*/
59