Page MenuHomec4science
Files F64045399

Vector.hpp
No OneTemporary
Actions

File Metadata

Created
Fri, May 24, 04:50

Vector.hpp
View Options

/**
Implementation of Vector.h
\file Vector.hpp
\copyright Copyright 2017. Tom de Geus. All rights reserved.
\license This project is released under the GNU Public License (GPLv3).
*/
#ifndef GOOSEFEM_VECTOR_HPP
#define GOOSEFEM_VECTOR_HPP
#include "Vector.h"
namespace GooseFEM {
inline Vector::Vector(const xt::xtensor<size_t, 2>& conn, const xt::xtensor<size_t, 2>& dofs)
: m_conn(conn), m_dofs(dofs)
{
m_nelem = m_conn.shape(0);
m_nne = m_conn.shape(1);
m_nnode = m_dofs.shape(0);
m_ndim = m_dofs.shape(1);
m_ndof = xt::amax(m_dofs)() + 1;
GOOSEFEM_ASSERT(xt::amax(m_conn)() + 1 <= m_nnode);
GOOSEFEM_ASSERT(m_ndof <= m_nnode * m_ndim);
}
inline size_t Vector::nelem() const
{
return m_nelem;
}
inline size_t Vector::nne() const
{
return m_nne;
}
inline size_t Vector::nnode() const
{
return m_nnode;
}
inline size_t Vector::ndim() const
{
return m_ndim;
}
inline size_t Vector::ndof() const
{
return m_ndof;
}
inline xt::xtensor<size_t, 2> Vector::dofs() const
{
return m_dofs;
}
inline void Vector::copy(
const xt::xtensor<double, 2>& nodevec_src, xt::xtensor<double, 2>& nodevec_dest) const
{
GOOSEFEM_ASSERT(xt::has_shape(nodevec_src, this->ShapeNodevec()));
GOOSEFEM_ASSERT(xt::has_shape(nodevec_dest, this->ShapeNodevec()));
xt::noalias(nodevec_dest) = nodevec_src;
}
inline void Vector::asDofs(
const xt::xtensor<double, 2>& nodevec, xt::xtensor<double, 1>& dofval) const
{
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
dofval.fill(0.0);
#pragma omp parallel for
for (size_t m = 0; m < m_nnode; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
dofval(m_dofs(m, i)) = nodevec(m, i);
}
}
}
inline void Vector::asDofs(
const xt::xtensor<double, 3>& elemvec, xt::xtensor<double, 1>& dofval) const
{
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
dofval.fill(0.0);
#pragma omp parallel for
for (size_t e = 0; e < m_nelem; ++e) {
for (size_t m = 0; m < m_nne; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
dofval(m_dofs(m_conn(e, m), i)) = elemvec(e, m, i);
}
}
}
}
inline void Vector::asNode(
const xt::xtensor<double, 1>& dofval, xt::xtensor<double, 2>& nodevec) const
{
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
#pragma omp parallel for
for (size_t m = 0; m < m_nnode; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
nodevec(m, i) = dofval(m_dofs(m, i));
}
}
}
inline void Vector::asNode(
const xt::xtensor<double, 3>& elemvec, xt::xtensor<double, 2>& nodevec) const
{
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
nodevec.fill(0.0);
#pragma omp parallel for
for (size_t e = 0; e < m_nelem; ++e) {
for (size_t m = 0; m < m_nne; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
nodevec(m_conn(e, m), i) = elemvec(e, m, i);
}
}
}
}
inline void Vector::asElement(
const xt::xtensor<double, 1>& dofval, xt::xtensor<double, 3>& elemvec) const
{
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
#pragma omp parallel for
for (size_t e = 0; e < m_nelem; ++e) {
for (size_t m = 0; m < m_nne; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
elemvec(e, m, i) = dofval(m_dofs(m_conn(e, m), i));
}
}
}
}
inline void Vector::asElement(
const xt::xtensor<double, 2>& nodevec, xt::xtensor<double, 3>& elemvec) const
{
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
#pragma omp parallel for
for (size_t e = 0; e < m_nelem; ++e) {
for (size_t m = 0; m < m_nne; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
elemvec(e, m, i) = nodevec(m_conn(e, m), i);
}
}
}
}
inline void Vector::assembleDofs(
const xt::xtensor<double, 2>& nodevec, xt::xtensor<double, 1>& dofval) const
{
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
dofval.fill(0.0);
for (size_t m = 0; m < m_nnode; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
dofval(m_dofs(m, i)) += nodevec(m, i);
}
}
}
inline void Vector::assembleDofs(
const xt::xtensor<double, 3>& elemvec, xt::xtensor<double, 1>& dofval) const
{
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
GOOSEFEM_ASSERT(dofval.size() == m_ndof);
dofval.fill(0.0);
for (size_t e = 0; e < m_nelem; ++e) {
for (size_t m = 0; m < m_nne; ++m) {
for (size_t i = 0; i < m_ndim; ++i) {
dofval(m_dofs(m_conn(e, m), i)) += elemvec(e, m, i);
}
}
}
}
inline void Vector::assembleNode(
const xt::xtensor<double, 3>& elemvec, xt::xtensor<double, 2>& nodevec) const
{
GOOSEFEM_ASSERT(xt::has_shape(elemvec, this->ShapeElemvec()));
GOOSEFEM_ASSERT(xt::has_shape(nodevec, this->ShapeNodevec()));
xt::xtensor<double, 1> dofval = this->AssembleDofs(elemvec);
this->asNode(dofval, nodevec);
}
inline xt::xtensor<double, 1> Vector::AsDofs(const xt::xtensor<double, 2>& nodevec) const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
this->asDofs(nodevec, dofval);
return dofval;
}
inline xt::xtensor<double, 1> Vector::AsDofs(const xt::xtensor<double, 3>& elemvec) const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
this->asDofs(elemvec, dofval);
return dofval;
}
inline xt::xtensor<double, 2> Vector::AsNode(const xt::xtensor<double, 1>& dofval) const
{
xt::xtensor<double, 2> nodevec = xt::empty<double>(this->ShapeNodevec());
this->asNode(dofval, nodevec);
return nodevec;
}
inline xt::xtensor<double, 2> Vector::AsNode(const xt::xtensor<double, 3>& elemvec) const
{
xt::xtensor<double, 2> nodevec = xt::empty<double>(this->ShapeNodevec());
this->asNode(elemvec, nodevec);
return nodevec;
}
inline xt::xtensor<double, 3> Vector::AsElement(const xt::xtensor<double, 1>& dofval) const
{
xt::xtensor<double, 3> elemvec = xt::empty<double>(this->ShapeElemvec());
this->asElement(dofval, elemvec);
return elemvec;
}
inline xt::xtensor<double, 3> Vector::AsElement(const xt::xtensor<double, 2>& nodevec) const
{
xt::xtensor<double, 3> elemvec = xt::empty<double>(this->ShapeElemvec());
this->asElement(nodevec, elemvec);
return elemvec;
}
inline xt::xtensor<double, 1> Vector::AssembleDofs(const xt::xtensor<double, 2>& nodevec) const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
this->assembleDofs(nodevec, dofval);
return dofval;
}
inline xt::xtensor<double, 1> Vector::AssembleDofs(const xt::xtensor<double, 3>& elemvec) const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
this->assembleDofs(elemvec, dofval);
return dofval;
}
inline xt::xtensor<double, 2> Vector::AssembleNode(const xt::xtensor<double, 3>& elemvec) const
{
xt::xtensor<double, 2> nodevec = xt::empty<double>(this->ShapeNodevec());
this->assembleNode(elemvec, nodevec);
return nodevec;
}
inline xt::xtensor<double, 2> Vector::Copy(
const xt::xtensor<double, 2>& nodevec_src, const xt::xtensor<double, 2>& nodevec_dest) const
{
xt::xtensor<double, 2> ret = nodevec_dest;
this->copy(nodevec_src, ret);
return ret;
}
inline std::array<size_t, 1> Vector::ShapeDofval() const
{
std::array<size_t, 1> shape;
shape[0] = m_ndof;
return shape;
}
inline std::array<size_t, 2> Vector::ShapeNodevec() const
{
std::array<size_t, 2> shape;
shape[0] = m_nnode;
shape[1] = m_ndim;
return shape;
}
inline std::array<size_t, 3> Vector::ShapeElemvec() const
{
std::array<size_t, 3> shape;
shape[0] = m_nelem;
shape[1] = m_nne;
shape[2] = m_ndim;
return shape;
}
inline std::array<size_t, 3> Vector::ShapeElemmat() const
{
std::array<size_t, 3> shape;
shape[0] = m_nelem;
shape[1] = m_nne * m_ndim;
shape[2] = m_nne * m_ndim;
return shape;
}
inline xt::xtensor<double, 1> Vector::AllocateDofval() const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
return dofval;
}
inline xt::xtensor<double, 2> Vector::AllocateNodevec() const
{
xt::xtensor<double, 2> nodevec = xt::empty<double>(this->ShapeNodevec());
return nodevec;
}
inline xt::xtensor<double, 3> Vector::AllocateElemvec() const
{
xt::xtensor<double, 3> elemvec = xt::empty<double>(this->ShapeElemvec());
return elemvec;
}
inline xt::xtensor<double, 3> Vector::AllocateElemmat() const
{
xt::xtensor<double, 3> elemmat = xt::empty<double>(this->ShapeElemmat());
return elemmat;
}
inline xt::xtensor<double, 1> Vector::AllocateDofval(double val) const
{
xt::xtensor<double, 1> dofval = xt::empty<double>(this->ShapeDofval());
dofval.fill(val);
return dofval;
}
inline xt::xtensor<double, 2> Vector::AllocateNodevec(double val) const
{
xt::xtensor<double, 2> nodevec = xt::empty<double>(this->ShapeNodevec());
nodevec.fill(val);
return nodevec;
}
inline xt::xtensor<double, 3> Vector::AllocateElemvec(double val) const
{
xt::xtensor<double, 3> elemvec = xt::empty<double>(this->ShapeElemvec());
elemvec.fill(val);
return elemvec;
}
inline xt::xtensor<double, 3> Vector::AllocateElemmat(double val) const
{
xt::xtensor<double, 3> elemmat = xt::empty<double>(this->ShapeElemmat());
elemmat.fill(val);
return elemmat;
}
} // namespace GooseFEM
#endif

Event Timeline

c4science · Help