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