vectorelement.py

来自「finite element library for mathematic ma」· Python 代码 · 共 20 行

"Factory function for generating vector elements as mixed elements"

__author__ = "Anders Logg ([e-mail])"
__date__ = "2007-03-20 -- 2007-03-29"
__copyright__ = "Copyright (C) 2007 Anders Logg"
__license__  = "GNU GPL version 3 or any later version"

from finiteelement import *
from mixedelement import *

def VectorElement(family, shape, degree, num_sub_elements=None):
    "Create vector finite element as the repeated tensor product of a given element"

    # Choose default number of sub elements if number is not specified by user
    if not num_sub_elements:
        num_sub_elements = shape_to_dim[string_to_shape[shape]]

    # Create mixed element by repeating the "scalar" element
    return MixedElement([FiniteElement(family, shape, degree) for i in range(num_sub_elements)])