#
##
## This file is part of pyFormex 1.0.7 (Mon Jun 17 12:20:39 CEST 2019)
## pyFormex is a tool for generating, manipulating and transforming 3D
## geometrical models by sequences of mathematical operations.
## Home page: http://pyformex.org
## Project page: http://savannah.nongnu.org/projects/pyformex/
## Copyright 2004-2019 (C) Benedict Verhegghe (benedict.verhegghe@ugent.be)
## Distributed under the GNU General Public License version 3 or later.
##
## This program is free software: you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation, either version 3 of the License, or
## (at your option) any later version.
##
## This program 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 General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program. If not, see http://www.gnu.org/licenses/.
##
#
## This program 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 General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program. If not, see http://www.gnu.org/licenses/.
##
"""Field data in pyFormex.
This module defines the Field class, which can be used to describe
scalar and vectorial field data over a geometrical domain.
"""
from __future__ import absolute_import, division, print_function
from pyformex import utils
import pyformex.arraytools as at
##############################################################
[docs]class Field(object):
"""Scalar or vectorial field data defined over a geometric domain.
A scalar data field is a quantity having exactly one value in each
point of some geometric domain. A vectorial field is a quantity having
exactly `nval` values at each point, where `nval >= 1`.
A vectorial field can also be considered as a collection of `nval`
scalar fields: as far as the Field class is concerned, there is no
relation between the `nval` components of a vectorial field.
The definition of a field is always tied to some geometric domain.
Currently pyFormex allows fields to be defined on Formex and Mesh
type geometries.
Fields should only be defined on geometries whose topology does not
change anymore. This means that for Formex type, the shape of the
`coords` attribute should not be changed, and for Mesh type, the
shape of `coords` and the full contents of `elems` should not be
changed. It is therefore best ot only add field data to geometry
objects that will not be changed in place.
Nearly all methods in pyFormex return a copy of the object, and the
copy currently looses all the fields defined on the parent.
In future however, selected transformations may inherit fields from
the parent.
While Field class instances are usually created automatically by the
:func:`Geometry.addField` method of some Geometry,
it is possible to create Field
instances yourself and manage the objects like you want. The Fields
stored inside Geometry objects have some special features though,
like being exported to a PGF file together with the geometry.
A Field is defined by the following parameters:
- `geometry`: Geometry instance: describes the geometrical domain
over which the field is defined. Currently this has to be a
:class:`Formex` or a :class:`Mesh`.
- `fldtype`: string: one of the following predefined field types:
- 'node': the field data are specified at the nodes of the geometry;
- 'elemc': the field data are constant per element;
- 'elemn': the field data vary over the element and are specified at
the nodes of the elements;
- 'elemg': the field data are specified at a number of points of
the elements, from which they can be inter- or extrapolated;
The actually available field types depend on the type of the
`geometry` object. :class:`Formex` type has only 'elemc' and 'elemn'.
`Mesh` currently has 'node', 'elemc' and 'elemn'.
- `data`: an array with the field values defined at the specified
points. The required shape of the array depends on `fldtype`:
- 'node': ( nnodes, ) or ( nnodes, nval )
- 'elemc': ( nelems, ) or ( nelems, nval )
- 'elemn': ( nelems, nplex ) or (nelems, nplex, nval )
- 'elemg': ( nelems, ngp ) or (nelems, ngp, nval )
- `fldname`: string: the name used to identify the field. Fields
stored in a Geometry object can be retrieved using this name. See
:func:`Geometry.getField`. If no name is specified, one is generated.
Example:
>>> from pyformex.formex import Formex
>>> M = Formex('4:0123').replic(2).toMesh()
>>> print(M.coords)
[[ 0. 0. 0.]
[ 0. 1. 0.]
[ 1. 0. 0.]
[ 1. 1. 0.]
[ 2. 0. 0.]
[ 2. 1. 0.]]
>>> print(M.elems)
[[0 2 3 1]
[2 4 5 3]]
>>> d = M.coords.distanceFromPlane([0.,0.,0.],[1.,0.,0.])
>>> f1 = Field(M,'node',d)
>>> print(f1)
Field 'field-0', type 'node', shape (6,), nnodes=6, nelems=2, nplex=4
[ 0. 0. 1. 1. 2. 2.]
>>> f2 = f1.convert('elemn')
>>> print(f2)
Field 'field-1', type 'elemn', shape (2, 4), nnodes=6, nelems=2, nplex=4
[[ 0. 1. 1. 0.]
[ 1. 2. 2. 1.]]
>>> f3 = f2.convert('elemc')
>>> print(f3)
Field 'field-2', type 'elemc', shape (2,), nnodes=6, nelems=2, nplex=4
[ 0.5 1.5]
>>> d1 = M.coords.distanceFromPlane([0.,0.,0.],[0.,1.,0.])
>>> f4 = Field(M,'node',at.column_stack([d,d1]))
>>> print(f4)
Field 'field-3', type 'node', shape (6, 2), nnodes=6, nelems=2, nplex=4
[[ 0. 0.]
[ 0. 1.]
[ 1. 0.]
[ 1. 1.]
[ 2. 0.]
[ 2. 1.]]
>>> f5 = f4.convert('elemn')
>>> print(f5)
Field 'field-4', type 'elemn', shape (2, 4, 2), nnodes=6, nelems=2, nplex=4
[[[ 0. 0.]
[ 1. 0.]
[ 1. 1.]
[ 0. 1.]]
<BLANKLINE>
[[ 1. 0.]
[ 2. 0.]
[ 2. 1.]
[ 1. 1.]]]
>>> f6 = f5.convert('elemc')
>>> print(f6)
Field 'field-5', type 'elemc', shape (2, 2), nnodes=6, nelems=2, nplex=4
[[ 0.5 0.5]
[ 1.5 0.5]]
>>> print(f3.convert('elemn'))
Field 'field-6', type 'elemn', shape (2, 4), nnodes=6, nelems=2, nplex=4
[[ 0.5 0.5 0.5 0.5]
[ 1.5 1.5 1.5 1.5]]
>>> print(f3.convert('node'))
Field 'field-8', type 'node', shape (6, 1), nnodes=6, nelems=2, nplex=4
[[ 0.5]
[ 0.5]
[ 1. ]
[ 1. ]
[ 1.5]
[ 1.5]]
>>> print(f6.convert('elemn'))
Field 'field-9', type 'elemn', shape (2, 4, 2), nnodes=6, nelems=2, nplex=4
[[[ 0.5 0.5]
[ 0.5 0.5]
[ 0.5 0.5]
[ 0.5 0.5]]
<BLANKLINE>
[[ 1.5 0.5]
[ 1.5 0.5]
[ 1.5 0.5]
[ 1.5 0.5]]]
>>> print(f6.convert('node'))
Field 'field-11', type 'node', shape (6, 2), nnodes=6, nelems=2, nplex=4
[[ 0.5 0.5]
[ 0.5 0.5]
[ 1. 0.5]
[ 1. 0.5]
[ 1.5 0.5]
[ 1.5 0.5]]
"""
_autoname = utils.autoName('Field')
def __init__(self, geometry, fldtype, data, fldname=None):
"""Initialize a Field.
"""
if not hasattr(
geometry, 'fieldtypes'
) or fldtype not in geometry.fieldtypes:
raise ValueError(
"Can not add field of type '%s' to a %s" %
(fldtype, geometry.__class__.__name__)
)
if fldtype == 'node':
datashape = (geometry.nnodes(), -1)
elif fldtype == 'elemc':
datashape = (geometry.nelems(), -1)
elif fldtype == 'elemn':
datashape = (geometry.nelems(), geometry.nplex(), -1)
elif fldtype == 'elemg':
datashape = (geometry.nelems(), -1, -1)
scalar = len(data.shape) < len(datashape)
if scalar:
datashape = datashape[:-1]
data = at.checkArray(data, shape=datashape)
if fldname is None:
fldname = next(Field._autoname)
# All data seem OK, store them
self.geometry = geometry
self.fldname = fldname
self.fldtype = fldtype
self.data = data
self.scalar = scalar
[docs] def comp(self, i):
"""Return the data component i of a vectorial Field.
Parameters:
- `i`: int: component number of a vectorial Field. If the Field is a
scalar one, any value will return the full scalar data.
Returns an array representing the scalar data over the
Geometry.
"""
if self.scalar:
return self.data
else:
return self.data[:, i]
[docs] def convert(self, totype, toname=None):
"""Convert a Field to another type.
Parameters:
- `totype`: string: target field type. Can be any of the available
field types. See :class:`Field` class. If the target type is equal
to the source type, a copy of the original Field will result.
This may or may not be a shallow copy.
- `toname`: string: the name of the target field. If not specified,
a autoname is generated.
Returns a Field of type `totype`.
"""
data = None
if totype == self.fldtype:
data = self.data
elif totype in self.geometry.fieldtypes:
if self.fldtype == 'node':
if totype == 'elemn':
data = self.data[self.geometry.elems]
elif totype == 'elemc':
return self.convert('elemn').convert('elemc', toname)
elif self.fldtype == 'elemn':
if totype == 'elemc':
data = self.data.mean(axis=1)
elif totype == 'node':
data = at.nodalAvg(
self.data, self.geometry.elems, self.geometry.nnodes()
)
elif self.fldtype == 'elemc':
if totype == 'elemn':
data = at.multiplex(
self.data, self.geometry.nplex(), axis=1
)
elif totype == 'node':
return self.convert('elemn').convert('node', toname)
if data is None:
raise ValueError(
"Can not convert %s field data from '%s' to '%s'" %
(self.geometry.__class__.__name__, self.fldtype, totype)
)
return Field(self.geometry, totype, data, toname)
def __str__(self):
s = "Field '%s', type '%s', shape %s, nnodes=%s, nelems=%s, nplex=%s\n" % \
(self.fldname, self.fldtype, self.data.shape, self.geometry.nnodes(),
self.geometry.nelems(), self.geometry.nplex(),)
return s + str(self.data)
# End
