1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
"""This module provides an extensive list of predefined finite element
families. Users or more likely, form compilers, may register new
elements by calling the function register_element."""
# Copyright (C) 2008-2011 Martin Sandve Alnes and Anders Logg
#
# This file is part of UFL.
#
# UFL is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# UFL is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with UFL. If not, see <http://www.gnu.org/licenses/>.
#
# Modified by Marie E. Rognes <meg@simula.no>, 2010
#
# First added: 2008-03-03
# Last changed: 2011-06-06
from ufl.assertions import ufl_assert
from ufl.feec import FEEC_aliases
# List of valid elements
ufl_elements = {}
# Aliases: aliases[name] (...) -> (standard_name, ...)
aliases = {}
# Function for registering new elements
def register_element(family, short_name, value_rank, degree_range, domains):
"Register new finite element family"
ufl_assert(family not in ufl_elements, 'Finite element \"%s\" has already been registered.' % family)
ufl_elements[family] = (family, short_name, value_rank, degree_range, domains)
ufl_elements[short_name] = (family, short_name, value_rank, degree_range, domains)
def register_alias(alias, to):
aliases[alias] = to
def show_elements():
print "Showing all registered elements:"
for k in sorted(ufl_elements.keys()):
(family, short_name, value_rank, degree_range, domains) = ufl_elements[k]
print
print "Finite Element Family: %s, %s" % (repr(family), repr(short_name))
print "Value rank: ", value_rank
print "Degree range: ", degree_range
print "Defined on domains:" , domains
# TODO: Any more of these elements valid for generic nD cells?
# Standard elements
register_element("Argyris", "ARG", 0, (1, None),
("triangle", "tetrahedron"))
register_element("Arnold-Winther", "AW", 0, None,
("triangle",))
register_element("Brezzi-Douglas-Fortin-Marini", "BDFM", 1, (1, None),
("triangle", "tetrahedron"))
register_element("Brezzi-Douglas-Marini", "BDM", 1, (1, None),
("triangle", "tetrahedron"))
register_element("Crouzeix-Raviart", "CR", 0, (1, 1),
("triangle", "tetrahedron"))
register_element("Discontinuous Lagrange", "DG", 0, (0, None),
("cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
register_element("Hermite", "HER", 0, None,
("triangle", "tetrahedron"))
register_element("Lagrange", "CG", 0, (1, None),
("cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
register_element("Mardal-Tai-Winther", "MTW", 0, None,
("triangle",))
register_element("Morley", "MOR", 0, None,
("triangle",))
register_element("Nedelec 1st kind H(curl)", "N1curl", 1, (1, None),
("triangle", "tetrahedron"))
register_element("Nedelec 2nd kind H(curl)", "N2curl", 1, (1, None),
("triangle", "tetrahedron"))
register_element("Raviart-Thomas", "RT", 1, (1, None),
("triangle", "tetrahedron"))
# Special elements
register_element("Boundary Quadrature", "BQ", 0, (0, None),
(None, "cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
register_element("Bubble", "B", 0, (2, None),
("interval", "triangle", "tetrahedron"))
register_element("Quadrature", "Q", 0, (0, None),
(None, "cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
register_element("Real", "R", 0, (0, 0),
(None, "cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
register_element("Undefined", "U", 0, (0, None),
("cell1D", "cell2D", "cell3D",
"interval", "triangle", "tetrahedron",
"quadrilateral", "hexahedron"))
# Let Nedelec H(div) elements be aliases to BDMs/RTs
register_alias("Nedelec 1st kind H(div)",
lambda family, cell, order, degree: ("Raviart-Thomas", cell, order))
register_alias("N1div",
lambda family, cell, order, degree: ("Raviart-Thomas", cell, order))
register_alias("Nedelec 2nd kind H(div)",
lambda family, cell, order, degree: ("Brezzi-Douglas-Marini", cell, order))
register_alias("N2div",
lambda family, cell, order, degree: ("Brezzi-Douglas-Marini", cell, order))
# Use P Lambda/P- Lambda aliases
register_alias("P Lambda", lambda *foo: FEEC_aliases(*foo))
register_alias("P- Lambda", lambda *foo: FEEC_aliases(*foo))
|