Homec4science
Diffusion GooseFEM (main) Browse

rGOOSEFEM/include/xGooseFEM/Element.hpp
6bb8640345c8main

Last Change
Raw File

Element.hpp
View Options

/* =================================================================================================
(c - GPLv3) T.W.J. de Geus (Tom) | tom@geus.me | www.geus.me | github.com/tdegeus/GooseFEM
================================================================================================= */
#ifndef XGOOSEFEM_ELEMENT_CPP
#define XGOOSEFEM_ELEMENT_CPP
// -------------------------------------------------------------------------------------------------
#include "Element.h"
// ======================================= GooseFEM::Element =======================================
namespace xGooseFEM {
namespace Element {
// -------------------------------------------------------------------------------------------------
inline xt::xtensor<double,3> asElementVector(const xt::xtensor<size_t,2> &conn, const xt::xtensor<double,2> &nodevec)
{
// extract dimensions
size_t nelem = conn .shape()[0];
size_t nne = conn .shape()[1];
size_t ndim = nodevec.shape()[1];
// allocate output: nodal vectors stored per element
xt::xtensor<double,3> elemvec = xt::empty<double>({nelem, nne, ndim});
// read from nodal vectors
#pragma omp parallel for
for ( size_t e = 0 ; e < nelem ; ++e )
for ( size_t m = 0 ; m < nne ; ++m )
for ( size_t i = 0 ; i < ndim ; ++i )
elemvec(e,m,i) = nodevec(conn(e,m),i);
return elemvec;
}
// -------------------------------------------------------------------------------------------------
inline xt::xtensor<double,2> assembleNodeVector(const xt::xtensor<size_t,2> &conn, const xt::xtensor<double,3> &elemvec)
{
// extract dimensions
size_t nelem = conn.shape()[0]; // number of elements
size_t nne = conn.shape()[1]; // number of nodes per element
size_t ndim = elemvec.shape()[2]; // number of dimensions
size_t nnode = xt::amax(conn)[0]+1; // number of nodes
// check input
assert( elemvec.shape()[0] == nelem );
assert( elemvec.shape()[1] == nne );
// zero-initialize output: nodal vectors
xt::xtensor<double,2> nodevec = xt::zeros<double>({nnode, ndim});
// assemble from nodal vectors stored per element
for ( size_t e = 0 ; e < nelem ; ++e )
for ( size_t m = 0 ; m < nne ; ++m )
for ( size_t i = 0 ; i < ndim ; ++i )
nodevec(conn(e,m),i) += elemvec(e,m,i);
return nodevec;
}
// -------------------------------------------------------------------------------------------------
}} // namespace ...
// =================================================================================================
#endif
c4science · Help