1 /// @file
2 /// Test creation, action, and destruction for mass matrix operator with multigrid level, non-tensor basis and interpolation basis generation
3 /// \test Test creation, action, and destruction for mass matrix operator with multigrid level, non-tensor basis and interpolation basis generation
4 #include <ceed.h>
5 #include <math.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8
9 #include "t502-operator.h"
10
main(int argc,char ** argv)11 int main(int argc, char **argv) {
12 Ceed ceed;
13 CeedElemRestriction elem_restriction_x, elem_restriction_q_data, elem_restriction_u_coarse, elem_restriction_u_fine;
14 CeedBasis basis_x, basis_coarse, basis_fine;
15 CeedQFunction qf_setup, qf_mass;
16 CeedOperator op_setup, op_mass_coarse, op_mass_fine, op_prolong, op_restrict;
17 CeedVector q_data, x, u_coarse, u_fine, v_coarse, v_fine, p_mult_fine;
18 CeedInt num_elem = 15, p_coarse = 3, p_fine = 5, q = 8, num_comp = 2;
19 CeedInt num_dofs_x = num_elem + 1, num_dofs_u_c = num_elem * (p_coarse - 1) + 1, num_dofs_u_f = num_elem * (p_fine - 1) + 1;
20 CeedInt ind_u_coarse[num_elem * p_coarse], ind_u_fine[num_elem * p_fine], ind_x[num_elem * 2];
21
22 CeedInit(argv[1], &ceed);
23
24 CeedVectorCreate(ceed, num_dofs_x, &x);
25 {
26 CeedScalar x_array[num_dofs_x];
27
28 for (CeedInt i = 0; i < num_dofs_x; i++) x_array[i] = (CeedScalar)i / (num_dofs_x - 1);
29 CeedVectorSetArray(x, CEED_MEM_HOST, CEED_COPY_VALUES, x_array);
30 }
31 CeedVectorCreate(ceed, num_comp * num_dofs_u_f, &p_mult_fine);
32 CeedVectorCreate(ceed, num_comp * num_dofs_u_c, &u_coarse);
33 CeedVectorCreate(ceed, num_comp * num_dofs_u_f, &u_fine);
34 CeedVectorCreate(ceed, num_comp * num_dofs_u_c, &v_coarse);
35 CeedVectorCreate(ceed, num_comp * num_dofs_u_f, &v_fine);
36 CeedVectorCreate(ceed, num_elem * q, &q_data);
37
38 // Restrictions
39 for (CeedInt i = 0; i < num_elem; i++) {
40 ind_x[2 * i + 0] = i;
41 ind_x[2 * i + 1] = i + 1;
42 }
43 CeedElemRestrictionCreate(ceed, num_elem, 2, 1, 1, num_dofs_x, CEED_MEM_HOST, CEED_USE_POINTER, ind_x, &elem_restriction_x);
44
45 for (CeedInt i = 0; i < num_elem; i++) {
46 for (CeedInt j = 0; j < p_coarse; j++) {
47 ind_u_coarse[p_coarse * i + j] = i * (p_coarse - 1) + j;
48 }
49 }
50 CeedElemRestrictionCreate(ceed, num_elem, p_coarse, num_comp, num_dofs_u_c, num_comp * num_dofs_u_c, CEED_MEM_HOST, CEED_USE_POINTER, ind_u_coarse,
51 &elem_restriction_u_coarse);
52
53 for (CeedInt i = 0; i < num_elem; i++) {
54 for (CeedInt j = 0; j < p_fine; j++) {
55 ind_u_fine[p_fine * i + j] = i * (p_fine - 1) + j;
56 }
57 }
58 CeedElemRestrictionCreate(ceed, num_elem, p_fine, num_comp, num_dofs_u_f, num_comp * num_dofs_u_f, CEED_MEM_HOST, CEED_USE_POINTER, ind_u_fine,
59 &elem_restriction_u_fine);
60
61 CeedInt strides_q_data[3] = {1, q, q};
62 CeedElemRestrictionCreateStrided(ceed, num_elem, q, 1, q * num_elem, strides_q_data, &elem_restriction_q_data);
63
64 // Bases
65 CeedBasisCreateTensorH1Lagrange(ceed, 1, 1, 2, q, CEED_GAUSS, &basis_x);
66 {
67 CeedBasis basis_temporary;
68
69 CeedBasisCreateTensorH1Lagrange(ceed, 1, num_comp, p_coarse, q, CEED_GAUSS, &basis_temporary);
70 const CeedScalar *interp, *grad, *q_ref, *q_weight;
71 CeedBasisGetInterp1D(basis_temporary, &interp);
72 CeedBasisGetGrad1D(basis_temporary, &grad);
73 CeedBasisGetQRef(basis_temporary, &q_ref);
74 CeedBasisGetQWeights(basis_temporary, &q_weight);
75 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_LINE, num_comp, p_coarse, q, interp, grad, q_ref, q_weight, &basis_coarse);
76 CeedBasisDestroy(&basis_temporary);
77 CeedBasisCreateTensorH1Lagrange(ceed, 1, num_comp, p_fine, q, CEED_GAUSS, &basis_temporary);
78 CeedBasisGetInterp1D(basis_temporary, &interp);
79 CeedBasisGetGrad1D(basis_temporary, &grad);
80 CeedBasisGetQRef(basis_temporary, &q_ref);
81 CeedBasisGetQWeights(basis_temporary, &q_weight);
82 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_LINE, num_comp, p_fine, q, interp, grad, q_ref, q_weight, &basis_fine);
83 CeedBasisDestroy(&basis_temporary);
84 }
85
86 // QFunctions
87 CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup);
88 CeedQFunctionAddInput(qf_setup, "weight", 1, CEED_EVAL_WEIGHT);
89 CeedQFunctionAddInput(qf_setup, "dx", 1 * 1, CEED_EVAL_GRAD);
90 CeedQFunctionAddOutput(qf_setup, "q data", 1, CEED_EVAL_NONE);
91
92 CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass);
93 CeedQFunctionAddInput(qf_mass, "q data", 1, CEED_EVAL_NONE);
94 CeedQFunctionAddInput(qf_mass, "u", num_comp, CEED_EVAL_INTERP);
95 CeedQFunctionAddOutput(qf_mass, "v", num_comp, CEED_EVAL_INTERP);
96
97 // Operators
98 CeedOperatorCreate(ceed, qf_setup, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE, &op_setup);
99 CeedOperatorSetField(op_setup, "weight", CEED_ELEMRESTRICTION_NONE, basis_x, CEED_VECTOR_NONE);
100 CeedOperatorSetField(op_setup, "dx", elem_restriction_x, basis_x, CEED_VECTOR_ACTIVE);
101 CeedOperatorSetField(op_setup, "q data", elem_restriction_q_data, CEED_BASIS_NONE, CEED_VECTOR_ACTIVE);
102
103 CeedOperatorCreate(ceed, qf_mass, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE, &op_mass_fine);
104 CeedOperatorSetField(op_mass_fine, "q data", elem_restriction_q_data, CEED_BASIS_NONE, q_data);
105 CeedOperatorSetField(op_mass_fine, "u", elem_restriction_u_fine, basis_fine, CEED_VECTOR_ACTIVE);
106 CeedOperatorSetField(op_mass_fine, "v", elem_restriction_u_fine, basis_fine, CEED_VECTOR_ACTIVE);
107
108 CeedOperatorApply(op_setup, x, q_data, CEED_REQUEST_IMMEDIATE);
109
110 // Create multigrid level
111 CeedVectorSetValue(p_mult_fine, 1.0);
112 CeedOperatorMultigridLevelCreate(op_mass_fine, p_mult_fine, elem_restriction_u_coarse, basis_coarse, &op_mass_coarse, &op_prolong, &op_restrict);
113
114 // Coarse problem
115 CeedVectorSetValue(u_coarse, 1.0);
116 CeedOperatorApply(op_mass_coarse, u_coarse, v_coarse, CEED_REQUEST_IMMEDIATE);
117
118 // Check output
119 {
120 const CeedScalar *v_array;
121 CeedScalar sum = 0.;
122
123 CeedVectorGetArrayRead(v_coarse, CEED_MEM_HOST, &v_array);
124 for (CeedInt i = 0; i < num_comp * num_dofs_u_c; i++) {
125 sum += v_array[i];
126 }
127 CeedVectorRestoreArrayRead(v_coarse, &v_array);
128 if (fabs(sum - 2.) > 1000. * CEED_EPSILON) printf("Computed Area Coarse Grid: %f != True Area: 2.0\n", sum);
129 }
130
131 // Prolong coarse u
132 CeedOperatorApply(op_prolong, u_coarse, u_fine, CEED_REQUEST_IMMEDIATE);
133
134 // Fine problem
135 CeedOperatorApply(op_mass_fine, u_fine, v_fine, CEED_REQUEST_IMMEDIATE);
136
137 // Check output
138 {
139 const CeedScalar *v_array;
140 CeedScalar sum = 0.;
141
142 CeedVectorGetArrayRead(v_fine, CEED_MEM_HOST, &v_array);
143 for (CeedInt i = 0; i < num_comp * num_dofs_u_f; i++) {
144 sum += v_array[i];
145 }
146 CeedVectorRestoreArrayRead(v_fine, &v_array);
147 if (fabs(sum - 2.) > 1000. * CEED_EPSILON) printf("Computed Area Fine Grid: %f != True Area: 2.0\n", sum);
148 }
149
150 // Restrict state to coarse grid
151 CeedOperatorApply(op_restrict, v_fine, v_coarse, CEED_REQUEST_IMMEDIATE);
152
153 // Check output
154 {
155 const CeedScalar *v_array;
156 CeedScalar sum = 0.;
157
158 CeedVectorGetArrayRead(v_coarse, CEED_MEM_HOST, &v_array);
159 for (CeedInt i = 0; i < num_comp * num_dofs_u_c; i++) {
160 sum += v_array[i];
161 }
162 CeedVectorRestoreArrayRead(v_coarse, &v_array);
163 if (fabs(sum - 2.) > 1000. * CEED_EPSILON) printf("Computed Area Coarse Grid: %f != True Area: 2.0\n", sum);
164 }
165
166 // Cleanup
167 CeedVectorDestroy(&x);
168 CeedVectorDestroy(&u_coarse);
169 CeedVectorDestroy(&u_fine);
170 CeedVectorDestroy(&v_coarse);
171 CeedVectorDestroy(&v_fine);
172 CeedVectorDestroy(&p_mult_fine);
173 CeedVectorDestroy(&q_data);
174 CeedElemRestrictionDestroy(&elem_restriction_u_coarse);
175 CeedElemRestrictionDestroy(&elem_restriction_u_fine);
176 CeedElemRestrictionDestroy(&elem_restriction_x);
177 CeedElemRestrictionDestroy(&elem_restriction_q_data);
178 CeedBasisDestroy(&basis_coarse);
179 CeedBasisDestroy(&basis_fine);
180 CeedBasisDestroy(&basis_x);
181 CeedQFunctionDestroy(&qf_setup);
182 CeedQFunctionDestroy(&qf_mass);
183 CeedOperatorDestroy(&op_setup);
184 CeedOperatorDestroy(&op_mass_coarse);
185 CeedOperatorDestroy(&op_mass_fine);
186 CeedOperatorDestroy(&op_prolong);
187 CeedOperatorDestroy(&op_restrict);
188 CeedDestroy(&ceed);
189 return 0;
190 }
191