1 /// @file
2 /// Test creation, action, and destruction for mass matrix operator
3 /// \test Test creation, action, and destruction for mass matrix operator
4 #include "t510-operator.h"
5
6 #include <ceed.h>
7 #include <math.h>
8 #include <stdio.h>
9 #include <stdlib.h>
10
11 #include "t320-basis.h"
12
main(int argc,char ** argv)13 int main(int argc, char **argv) {
14 Ceed ceed;
15 CeedElemRestriction elem_restriction_x, elem_restriction_u, elem_restriction_q_data;
16 CeedBasis basis_x, basis_u;
17 CeedQFunction qf_setup, qf_mass;
18 CeedOperator op_setup, op_mass;
19 CeedVector q_data, x, u, v;
20 CeedInt num_elem = 12, dim = 2, p = 6, q = 4;
21 CeedInt nx = 3, ny = 2;
22 CeedInt row, col, offset;
23 CeedInt num_dofs = (nx * 2 + 1) * (ny * 2 + 1), num_qpts = num_elem * q;
24 CeedInt ind_x[num_elem * p];
25 CeedScalar q_ref[dim * q], q_weight[q];
26 CeedScalar interp[p * q], grad[dim * p * q];
27
28 CeedInit(argv[1], &ceed);
29
30 CeedVectorCreate(ceed, dim * num_dofs, &x);
31 {
32 CeedScalar x_array[dim * num_dofs];
33
34 for (CeedInt i = 0; i < num_dofs; i++) {
35 x_array[i] = (1. / (nx * 2)) * (CeedScalar)(i % (nx * 2 + 1));
36 x_array[i + num_dofs] = (1. / (ny * 2)) * (CeedScalar)(i / (nx * 2 + 1));
37 }
38 CeedVectorSetArray(x, CEED_MEM_HOST, CEED_COPY_VALUES, x_array);
39 }
40 CeedVectorCreate(ceed, num_qpts, &q_data);
41 CeedVectorCreate(ceed, num_dofs, &u);
42 CeedVectorCreate(ceed, num_dofs, &v);
43
44 // Restrictions
45 for (CeedInt i = 0; i < num_elem / 2; i++) {
46 col = i % nx;
47 row = i / nx;
48 offset = col * 2 + row * (nx * 2 + 1) * 2;
49
50 ind_x[i * 2 * p + 0] = 2 + offset;
51 ind_x[i * 2 * p + 1] = 9 + offset;
52 ind_x[i * 2 * p + 2] = 16 + offset;
53 ind_x[i * 2 * p + 3] = 1 + offset;
54 ind_x[i * 2 * p + 4] = 8 + offset;
55 ind_x[i * 2 * p + 5] = 0 + offset;
56
57 ind_x[i * 2 * p + 6] = 14 + offset;
58 ind_x[i * 2 * p + 7] = 7 + offset;
59 ind_x[i * 2 * p + 8] = 0 + offset;
60 ind_x[i * 2 * p + 9] = 15 + offset;
61 ind_x[i * 2 * p + 10] = 8 + offset;
62 ind_x[i * 2 * p + 11] = 16 + offset;
63 }
64 CeedElemRestrictionCreate(ceed, num_elem, p, dim, num_dofs, dim * num_dofs, CEED_MEM_HOST, CEED_USE_POINTER, ind_x, &elem_restriction_x);
65 CeedElemRestrictionCreate(ceed, num_elem, p, 1, 1, num_dofs, CEED_MEM_HOST, CEED_USE_POINTER, ind_x, &elem_restriction_u);
66
67 CeedInt strides_q_data[3] = {1, q, q};
68 CeedElemRestrictionCreateStrided(ceed, num_elem, q, 1, num_qpts, strides_q_data, &elem_restriction_q_data);
69
70 // Bases
71 Build2DSimplex(q_ref, q_weight, interp, grad);
72 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_TRIANGLE, dim, p, q, interp, grad, q_ref, q_weight, &basis_x);
73
74 Build2DSimplex(q_ref, q_weight, interp, grad);
75 CeedBasisCreateH1(ceed, CEED_TOPOLOGY_TRIANGLE, 1, p, q, interp, grad, q_ref, q_weight, &basis_u);
76
77 // QFunctions
78 CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup);
79 CeedQFunctionAddInput(qf_setup, "weight", 1, CEED_EVAL_WEIGHT);
80 CeedQFunctionAddInput(qf_setup, "dx", dim * dim, CEED_EVAL_GRAD);
81 CeedQFunctionAddOutput(qf_setup, "rho", 1, CEED_EVAL_NONE);
82
83 CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass);
84 CeedQFunctionAddInput(qf_mass, "rho", 1, CEED_EVAL_NONE);
85 CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP);
86 CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP);
87
88 // Operators
89 CeedOperatorCreate(ceed, qf_setup, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE, &op_setup);
90 CeedOperatorSetField(op_setup, "weight", CEED_ELEMRESTRICTION_NONE, basis_x, CEED_VECTOR_NONE);
91 CeedOperatorSetField(op_setup, "dx", elem_restriction_x, basis_x, CEED_VECTOR_ACTIVE);
92 CeedOperatorSetField(op_setup, "rho", elem_restriction_q_data, CEED_BASIS_NONE, CEED_VECTOR_ACTIVE);
93
94 CeedOperatorCreate(ceed, qf_mass, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE, &op_mass);
95 CeedOperatorSetField(op_mass, "rho", elem_restriction_q_data, CEED_BASIS_NONE, q_data);
96 CeedOperatorSetField(op_mass, "u", elem_restriction_u, basis_u, CEED_VECTOR_ACTIVE);
97 CeedOperatorSetField(op_mass, "v", elem_restriction_u, basis_u, CEED_VECTOR_ACTIVE);
98
99 CeedOperatorApply(op_setup, x, q_data, CEED_REQUEST_IMMEDIATE);
100
101 CeedVectorSetValue(u, 0.0);
102 CeedOperatorApply(op_mass, u, v, CEED_REQUEST_IMMEDIATE);
103
104 // Check output
105 {
106 const CeedScalar *v_array;
107
108 CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array);
109 for (CeedInt i = 0; i < num_dofs; i++) {
110 if (fabs(v_array[i]) > 1e-14) printf("[%" CeedInt_FMT "] v %g != 0.0\n", i, v_array[i]);
111 }
112 CeedVectorRestoreArrayRead(v, &v_array);
113 }
114
115 CeedVectorDestroy(&x);
116 CeedVectorDestroy(&u);
117 CeedVectorDestroy(&v);
118 CeedVectorDestroy(&q_data);
119 CeedElemRestrictionDestroy(&elem_restriction_u);
120 CeedElemRestrictionDestroy(&elem_restriction_x);
121 CeedElemRestrictionDestroy(&elem_restriction_q_data);
122 CeedBasisDestroy(&basis_u);
123 CeedBasisDestroy(&basis_x);
124 CeedQFunctionDestroy(&qf_setup);
125 CeedQFunctionDestroy(&qf_mass);
126 CeedOperatorDestroy(&op_setup);
127 CeedOperatorDestroy(&op_mass);
128 CeedDestroy(&ceed);
129 return 0;
130 }
131