xref: /petsc/src/mat/tests/ex199.c (revision 28b400f66ebc7ae0049166a2294dfcd3df27e64b)
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;
185f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size));
19c4762a1bSJed Brown 
205f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg));
21*28b400f6SJacob Faibussowitsch   PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must use -f filename to load sparse matrix");
225f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&viewer));
235f80ce2aSJacob Faibussowitsch   CHKERRQ(MatCreate(PETSC_COMM_WORLD,&C));
245f80ce2aSJacob Faibussowitsch   CHKERRQ(MatLoad(C,viewer));
255f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscViewerDestroy(&viewer));
26c4762a1bSJed Brown 
275f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringCreate(C,&ctx));
285f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringSetFromOptions(ctx));
295f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringApply(ctx,&coloring));
305f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringTest(ctx,coloring));
31c4762a1bSJed Brown   if (size == 1) {
32c4762a1bSJed Brown     /* jp, power and greedy have bug -- need to be fixed */
335f80ce2aSJacob Faibussowitsch     CHKERRQ(MatISColoringTest(C,coloring));
34c4762a1bSJed Brown   }
35c4762a1bSJed Brown 
36c4762a1bSJed Brown   /* Free data structures */
375f80ce2aSJacob Faibussowitsch   CHKERRQ(ISColoringDestroy(&coloring));
385f80ce2aSJacob Faibussowitsch   CHKERRQ(MatColoringDestroy(&ctx));
395f80ce2aSJacob 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