1 static char help[] = "Tests for DMLabel lookup\n\n"; 2 3 #include <petscdmplex.h> 4 5 typedef struct { 6 PetscInt debug; /* The debugging level */ 7 PetscInt pStart, pEnd; /* The label chart */ 8 PetscInt numStrata; /* The number of label strata */ 9 PetscReal fill; /* Percentage of label to fill */ 10 PetscInt size; /* The number of set values */ 11 } AppCtx; 12 13 PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options) 14 { 15 PetscFunctionBegin; 16 options->debug = 0; 17 options->pStart = 0; 18 options->pEnd = 1000; 19 options->numStrata = 5; 20 options->fill = 0.10; 21 22 PetscOptionsBegin(comm, "", "Meshing Problem Options", "DMPLEX"); 23 PetscCall(PetscOptionsBoundedInt("-debug", "The debugging level", "ex6.c", options->debug, &options->debug, NULL, 0)); 24 PetscCall(PetscOptionsBoundedInt("-num_strata", "The number of label values", "ex6.c", options->numStrata, &options->numStrata, NULL, 0)); 25 PetscCall(PetscOptionsBoundedInt("-pend", "The label point limit", "ex6.c", options->pEnd, &options->pEnd, NULL, 0)); 26 PetscCall(PetscOptionsReal("-fill", "The percentage of label chart to set", "ex6.c", options->fill, &options->fill, NULL)); 27 PetscOptionsEnd(); 28 PetscFunctionReturn(PETSC_SUCCESS); 29 } 30 31 PetscErrorCode TestSetup(DMLabel label, AppCtx *user) 32 { 33 PetscRandom r; 34 PetscInt n = (PetscInt)(user->fill * (user->pEnd - user->pStart)), i; 35 36 PetscFunctionBegin; 37 PetscCall(PetscRandomCreate(PETSC_COMM_SELF, &r)); 38 PetscCall(PetscRandomSetFromOptions(r)); /* -random_type <> */ 39 PetscCall(PetscRandomSetInterval(r, user->pStart, user->pEnd)); 40 PetscCall(PetscRandomSetSeed(r, 123456789L)); 41 PetscCall(PetscRandomSeed(r)); 42 user->size = 0; 43 for (i = 0; i < n; ++i) { 44 PetscReal p; 45 PetscInt val; 46 47 PetscCall(PetscRandomGetValueReal(r, &p)); 48 PetscCall(DMLabelGetValue(label, (PetscInt)p, &val)); 49 if (val < 0) { 50 ++user->size; 51 PetscCall(DMLabelSetValue(label, (PetscInt)p, i % user->numStrata)); 52 } 53 } 54 PetscCall(PetscRandomDestroy(&r)); 55 PetscCall(DMLabelCreateIndex(label, user->pStart, user->pEnd)); 56 PetscCall(PetscPrintf(PETSC_COMM_SELF, "Created label with chart [%" PetscInt_FMT ", %" PetscInt_FMT ") and set %" PetscInt_FMT " values\n", user->pStart, user->pEnd, user->size)); 57 PetscFunctionReturn(PETSC_SUCCESS); 58 } 59 60 PetscErrorCode TestLookup(DMLabel label, AppCtx *user) 61 { 62 const PetscInt pStart = user->pStart; 63 const PetscInt pEnd = user->pEnd; 64 PetscInt p, n = 0; 65 66 PetscFunctionBegin; 67 for (p = pStart; p < pEnd; ++p) { 68 PetscInt val; 69 PetscBool has; 70 71 PetscCall(DMLabelGetValue(label, p, &val)); 72 PetscCall(DMLabelHasPoint(label, p, &has)); 73 PetscCheck((val < 0 || has) && (val >= 0 || !has), PETSC_COMM_SELF, PETSC_ERR_PLIB, "Label value %" PetscInt_FMT " does not match contains check %" PetscInt_FMT " for point %" PetscInt_FMT, val, (PetscInt)has, p); 74 if (has) ++n; 75 } 76 PetscCheck(n == user->size, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of label points detected %" PetscInt_FMT " does not match number set %" PetscInt_FMT, n, user->size); 77 /* Also put in timing code */ 78 PetscFunctionReturn(PETSC_SUCCESS); 79 } 80 81 PetscErrorCode TestClear(DMLabel label, AppCtx *user) 82 { 83 PetscInt pStart = user->pStart, pEnd = user->pEnd, p; 84 PetscInt defaultValue; 85 86 PetscFunctionBegin; 87 PetscCall(DMLabelGetDefaultValue(label, &defaultValue)); 88 for (p = pStart; p < pEnd; p++) { 89 PetscInt val; 90 PetscBool hasPoint; 91 92 PetscCall(DMLabelGetValue(label, p, &val)); 93 if (val != defaultValue) PetscCall(DMLabelClearValue(label, p, val)); 94 PetscCall(DMLabelGetValue(label, p, &val)); 95 PetscCall(DMLabelHasPoint(label, p, &hasPoint)); 96 PetscCheck(val == defaultValue, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected default value %" PetscInt_FMT " after clearing point %" PetscInt_FMT ", got %" PetscInt_FMT, defaultValue, p, val); 97 PetscCheck(!hasPoint, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Label contains %" PetscInt_FMT " after clearing", p); 98 } 99 PetscFunctionReturn(PETSC_SUCCESS); 100 } 101 102 int main(int argc, char **argv) 103 { 104 DMLabel label; 105 AppCtx user; /* user-defined work context */ 106 107 PetscFunctionBeginUser; 108 PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 109 PetscCall(ProcessOptions(PETSC_COMM_WORLD, &user)); 110 PetscCall(DMLabelCreate(PETSC_COMM_SELF, "Test Label", &label)); 111 PetscCall(TestSetup(label, &user)); 112 PetscCall(TestLookup(label, &user)); 113 PetscCall(TestClear(label, &user)); 114 PetscCall(DMLabelDestroy(&label)); 115 PetscCall(PetscFinalize()); 116 return 0; 117 } 118 119 /*TEST 120 121 test: 122 suffix: 0 123 args: -malloc_dump 124 test: 125 suffix: 1 126 args: -malloc_dump -pend 10000 127 test: 128 suffix: 2 129 args: -malloc_dump -pend 10000 -fill 0.05 130 test: 131 suffix: 3 132 args: -malloc_dump -pend 10000 -fill 0.25 133 134 TEST*/ 135