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