Actual source code: ex2.c
1: static char help[] = "Tests 1D cell-based discretization tools.\n\n";
3: #include <petscdt.h>
5: int main(int argc, char **argv)
6: {
7: PetscInt i, j, degrees[1000], ndegrees, nsrc_points, ntarget_points;
8: PetscReal src_points[1000], target_points[1000], *R;
9: PetscBool flg;
11: PetscFunctionBeginUser;
12: PetscCall(PetscInitialize(&argc, &argv, NULL, help));
13: PetscOptionsBegin(PETSC_COMM_WORLD, NULL, "Discretization tools test options", NULL);
14: {
15: ndegrees = 1000;
16: degrees[0] = 1;
17: degrees[1] = 2;
18: degrees[2] = 3;
19: PetscCall(PetscOptionsIntArray("-degrees", "list of max degrees to evaluate", "", degrees, &ndegrees, &flg));
20: if (!flg) ndegrees = 3;
22: nsrc_points = 1000;
23: src_points[0] = -1.;
24: src_points[1] = 0.;
25: src_points[2] = 1.;
26: PetscCall(PetscOptionsRealArray("-src_points", "list of points defining intervals on which to integrate", "", src_points, &nsrc_points, &flg));
27: if (!flg) nsrc_points = 3;
29: ntarget_points = 1000;
30: target_points[0] = -1.;
31: target_points[1] = 0.;
32: target_points[2] = 1.;
33: PetscCall(PetscOptionsRealArray("-target_points", "list of points defining intervals on which to integrate", "", target_points, &ntarget_points, &flg));
34: if (!flg) ntarget_points = 3;
35: }
36: PetscOptionsEnd();
38: PetscCall(PetscMalloc1((nsrc_points - 1) * (ntarget_points - 1), &R));
39: for (i = 0; i < ndegrees; i++) {
40: PetscCall(PetscDTReconstructPoly(degrees[i], nsrc_points - 1, src_points, ntarget_points - 1, target_points, R));
41: for (j = 0; j < (ntarget_points - 1) * (nsrc_points - 1); j++) { /* Truncate to zero for nicer output */
42: if (PetscAbs(R[j]) < 10 * PETSC_MACHINE_EPSILON) R[j] = 0;
43: }
44: for (j = 0; j < ntarget_points - 1; j++) {
45: PetscCall(PetscPrintf(PETSC_COMM_WORLD, "Degree %" PetscInt_FMT " target interval (%g,%g)\n", degrees[i], (double)target_points[j], (double)target_points[j + 1]));
46: PetscCall(PetscRealView(nsrc_points - 1, R + j * (nsrc_points - 1), PETSC_VIEWER_STDOUT_WORLD));
47: }
48: }
49: PetscCall(PetscFree(R));
50: PetscCall(PetscFinalize());
51: return 0;
52: }
54: /*TEST
55: test:
56: suffix: 1
57: args: -degrees 1,2,3 -target_points -0.3,0,.2 -src_points -1,-.3,0,.2,1
58: TEST*/