1c4762a1bSJed Brown static char help[] = "Tests PetscTreeProcess()";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscsys.h>
4c4762a1bSJed Brown
5c4762a1bSJed Brown /*
6c4762a1bSJed Brown 2 6
7c4762a1bSJed Brown 1 4
8c4762a1bSJed Brown 5
9c4762a1bSJed Brown */
main(int argc,char ** argv)10*d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv)
11*d71ae5a4SJacob Faibussowitsch {
12c4762a1bSJed Brown PetscInt n = 7, cnt = 0, i, j;
13c4762a1bSJed Brown PetscBool mask[] = {PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, PETSC_FALSE};
14c4762a1bSJed Brown PetscInt parentId[] = {-1, 2, 0, -1, 2, 1, 0};
15c4762a1bSJed Brown PetscInt Nlevels, *Level, *Levelcnt, *Idbylevel, *Column;
16c4762a1bSJed Brown
17327415f7SBarry Smith PetscFunctionBeginUser;
189566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, NULL, help));
199566063dSJacob Faibussowitsch PetscCall(PetscProcessTree(n, mask, parentId, &Nlevels, &Level, &Levelcnt, &Idbylevel, &Column));
20c4762a1bSJed Brown for (i = 0; i < n; i++) {
2148a46eb9SPierre Jolivet if (!mask[i]) PetscCall(PetscPrintf(PETSC_COMM_WORLD, " %" PetscInt_FMT " ", Level[i]));
22c4762a1bSJed Brown }
239566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nNumber of levels %" PetscInt_FMT "\n", Nlevels));
24c4762a1bSJed Brown for (i = 0; i < Nlevels; i++) {
259566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nLevel %" PetscInt_FMT " ", i));
2648a46eb9SPierre Jolivet for (j = 0; j < Levelcnt[i]; j++) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%" PetscInt_FMT " ", Idbylevel[cnt++]));
27c4762a1bSJed Brown }
289566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nColumn of each node"));
29c4762a1bSJed Brown for (i = 0; i < n; i++) {
3048a46eb9SPierre Jolivet if (!mask[i]) PetscCall(PetscPrintf(PETSC_COMM_WORLD, " %" PetscInt_FMT " ", Column[i]));
31c4762a1bSJed Brown }
329566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\n"));
339566063dSJacob Faibussowitsch PetscCall(PetscFree(Level));
349566063dSJacob Faibussowitsch PetscCall(PetscFree(Levelcnt));
359566063dSJacob Faibussowitsch PetscCall(PetscFree(Idbylevel));
369566063dSJacob Faibussowitsch PetscCall(PetscFree(Column));
379566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
38b122ec5aSJacob Faibussowitsch return 0;
39c4762a1bSJed Brown }
40c4762a1bSJed Brown
41c4762a1bSJed Brown /*TEST
42c4762a1bSJed Brown
43c4762a1bSJed Brown test:
44c4762a1bSJed Brown
45c4762a1bSJed Brown TEST*/
46