def _generate_edge_dofs(self, cell, degree, offset, variant, quad_deg):
"""Generate degrees of freedoms (dofs) for entities of
codimension 1 (edges)."""
# (degree+1) tangential component point evaluation degrees of
# freedom per entity of codimension 1 (edges)
dofs = []
ids = {}
if variant == "integral":
edge = cell.construct_subelement(1)
Q = quadrature.make_quadrature(edge, quad_deg)
Pq = polynomial_set.ONPolynomialSet(edge, degree)
Pq_at_qpts = Pq.tabulate(Q.get_points())[tuple([0]*(1))]
for e in range(len(cell.get_topology()[1])):
for i in range(Pq_at_qpts.shape[0]):
phi = Pq_at_qpts[i, :]
dofs.append(functional.IntegralMomentOfEdgeTangentEvaluation(cell, Q, phi, e))
jj = Pq_at_qpts.shape[0] * e
ids[e] = list(range(offset + jj, offset + jj + Pq_at_qpts.shape[0]))
elif variant == "point":
for edge in range(len(cell.get_topology()[1])):
# Create points for evaluation of tangential components
points = cell.make_points(1, edge, degree + 2)
# A tangential component evaluation for each point
dofs += [Tangent(cell, edge, point) for point in points]
# Associate these dofs with this edge
i = len(points) * edge
ids[edge] = list(range(offset + i, offset + i + len(points)))
return (dofs, ids)
def _generate_edge_dofs(self, cell, degree, offset):
"""Generate degrees of freedoms (dofs) for entities of
codimension 1 (edges)."""
# (degree+1) tangential component point evaluation degrees of
# freedom per entity of codimension 1 (edges)
dofs = []
ids = {}
for edge in range(len(cell.get_topology()[1])):
# Create points for evaluation of tangential components
points = cell.make_points(1, edge, degree + 2)
# A tangential component evaluation for each point
dofs += [Tangent(cell, edge, point) for point in points]
# Associate these dofs with this edge
i = len(points) * edge
ids[edge] = list(range(offset + i, offset + i + len(points)))
return (dofs, ids)