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
ProcessOptions(MPI_Comm comm,AppCtx * options)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
TestSetup(DMLabel label,AppCtx * user)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
TestLookup(DMLabel label,AppCtx * user)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
TestClear(DMLabel label,AppCtx * user)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
main(int argc,char ** argv)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