1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Tests PetscTreeProcess()"; 3c4762a1bSJed Brown 4c4762a1bSJed Brown #include <petscsys.h> 5c4762a1bSJed Brown 6c4762a1bSJed Brown /* 7c4762a1bSJed Brown 2 6 8c4762a1bSJed Brown 1 4 9c4762a1bSJed Brown 5 10c4762a1bSJed Brown */ 11*d71ae5a4SJacob Faibussowitsch int main(int argc, char **argv) 12*d71ae5a4SJacob Faibussowitsch { 13c4762a1bSJed Brown PetscInt n = 7, cnt = 0, i, j; 14c4762a1bSJed Brown PetscBool mask[] = {PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, PETSC_TRUE, PETSC_FALSE, PETSC_FALSE, PETSC_FALSE}; 15c4762a1bSJed Brown PetscInt parentId[] = {-1, 2, 0, -1, 2, 1, 0}; 16c4762a1bSJed Brown PetscInt Nlevels, *Level, *Levelcnt, *Idbylevel, *Column; 17c4762a1bSJed Brown 18327415f7SBarry Smith PetscFunctionBeginUser; 199566063dSJacob Faibussowitsch PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 209566063dSJacob Faibussowitsch PetscCall(PetscProcessTree(n, mask, parentId, &Nlevels, &Level, &Levelcnt, &Idbylevel, &Column)); 21c4762a1bSJed Brown for (i = 0; i < n; i++) { 2248a46eb9SPierre Jolivet if (!mask[i]) PetscCall(PetscPrintf(PETSC_COMM_WORLD, " %" PetscInt_FMT " ", Level[i])); 23c4762a1bSJed Brown } 249566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nNumber of levels %" PetscInt_FMT "\n", Nlevels)); 25c4762a1bSJed Brown for (i = 0; i < Nlevels; i++) { 269566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nLevel %" PetscInt_FMT " ", i)); 2748a46eb9SPierre Jolivet for (j = 0; j < Levelcnt[i]; j++) PetscCall(PetscPrintf(PETSC_COMM_WORLD, "%" PetscInt_FMT " ", Idbylevel[cnt++])); 28c4762a1bSJed Brown } 299566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\nColumn of each node")); 30c4762a1bSJed Brown for (i = 0; i < n; i++) { 3148a46eb9SPierre Jolivet if (!mask[i]) PetscCall(PetscPrintf(PETSC_COMM_WORLD, " %" PetscInt_FMT " ", Column[i])); 32c4762a1bSJed Brown } 339566063dSJacob Faibussowitsch PetscCall(PetscPrintf(PETSC_COMM_WORLD, "\n")); 349566063dSJacob Faibussowitsch PetscCall(PetscFree(Level)); 359566063dSJacob Faibussowitsch PetscCall(PetscFree(Levelcnt)); 369566063dSJacob Faibussowitsch PetscCall(PetscFree(Idbylevel)); 379566063dSJacob Faibussowitsch PetscCall(PetscFree(Column)); 389566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 39b122ec5aSJacob Faibussowitsch return 0; 40c4762a1bSJed Brown } 41c4762a1bSJed Brown 42c4762a1bSJed Brown /*TEST 43c4762a1bSJed Brown 44c4762a1bSJed Brown test: 45c4762a1bSJed Brown 46c4762a1bSJed Brown TEST*/ 47