xref: /petsc/src/mat/tests/ex199.c (revision b122ec5aa1bd4469eb4e0673542fb7de3f411254)
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   PetscViewer    viewer;
10c4762a1bSJed Brown   char           file[128];
11c4762a1bSJed Brown   PetscBool      flg;
12c4762a1bSJed Brown   MatColoring    ctx;
13c4762a1bSJed Brown   ISColoring     coloring;
14c4762a1bSJed Brown   PetscMPIInt    size;
15c4762a1bSJed Brown 
16*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help));
175f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
18c4762a1bSJed Brown 
195f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg));
2028b400f6SJacob Faibussowitsch   PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must use -f filename to load sparse matrix");
215f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&viewer));
225f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PETSC_COMM_WORLD,&C));
235f80ce2aSJacob Faibussowitsch   CHKERRQ(MatLoad(C,viewer));
245f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDestroy(&viewer));
25c4762a1bSJed Brown 
265f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringCreate(C,&ctx));
275f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringSetFromOptions(ctx));
285f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringApply(ctx,&coloring));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringTest(ctx,coloring));
30c4762a1bSJed Brown   if (size == 1) {
31c4762a1bSJed Brown     /* jp, power and greedy have bug -- need to be fixed */
325f80ce2aSJacob Faibussowitsch     CHKERRQ(MatISColoringTest(C,coloring));
33c4762a1bSJed Brown   }
34c4762a1bSJed Brown 
35c4762a1bSJed Brown   /* Free data structures */
365f80ce2aSJacob Faibussowitsch   CHKERRQ(ISColoringDestroy(&coloring));
375f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringDestroy(&ctx));
385f80ce2aSJacob Faibussowitsch   CHKERRQ(MatDestroy(&C));
39*b122ec5aSJacob Faibussowitsch   CHKERRQ(PetscFinalize());
40*b122ec5aSJacob Faibussowitsch   return 0;
41c4762a1bSJed Brown }
42c4762a1bSJed Brown 
43c4762a1bSJed Brown /*TEST
44c4762a1bSJed Brown 
45c4762a1bSJed Brown    test:
46c4762a1bSJed Brown       nsize: {{3}}
47dfd57a17SPierre Jolivet       requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES)
48c4762a1bSJed Brown       args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{ jp power natural greedy}} -mat_coloring_distance {{ 1 2}}
49c4762a1bSJed Brown 
50c4762a1bSJed Brown    test:
51c4762a1bSJed Brown       suffix: 2
52c4762a1bSJed Brown       nsize: {{1 2}}
53dfd57a17SPierre Jolivet       requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES)
54c4762a1bSJed Brown       args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{  sl lf id }} -mat_coloring_distance 2
55c4762a1bSJed Brown       output_file: output/ex199_1.out
56c4762a1bSJed Brown 
57c4762a1bSJed Brown TEST*/
58