ElementCollection.h

Go to the documentation of this file.

#pragma once
#include "nuto/mechanics/elements/ElementInterface.h"
#include "nuto/mechanics/dofs/DofContainer.h"
#include "nuto/mechanics/elements/ElementFem.h"
#include "nuto/mechanics/elements/ElementIga.h"

namespace NuTo
{
//                           coordinate element and the dof elements.
//                        b) allows storing elements (implementations of ElementInterface) by value
//                           in the class ElementCollectionImpl.
class ElementCollection
{
public:
    virtual ~ElementCollection() = default;
    virtual const ElementInterface& CoordinateElement() const = 0;
    virtual const ElementInterface& DofElement(DofType) const = 0;
    virtual const Shape& GetShape() const = 0;
};

template <typename TElement>
class ElementCollectionImpl : public ElementCollection
{
public:
    static_assert(std::is_base_of<ElementInterface, TElement>::value,
                  "TElement must be a descendant of ElementInterface");

    ElementCollectionImpl(TElement coordinateElement)
        : mCoordinateElement(coordinateElement)
        , mShape(coordinateElement.GetShape())
    {
    }

    void AddDofElement(DofType dofType, TElement dofElement)
    {
        mDofElements.Insert(dofType, dofElement);
        // The alternative implementation with
        //        mDofElements[dofType] = dofElement
        // will fail. The operator[] must be able default construct a new TElement, if it does not exist. It would then
        // return a reference to it and dofElement can be copied/moved into it. Our elements are not default
        // constructable (may require nodes, interpolations, ...). Thus, use Insert here, that copies/moves the entity
        // into the DofContainer without a temporary, default constructed TElement.
    }

    const TElement& CoordinateElement() const override
    {
        return mCoordinateElement;
    }

    TElement& CoordinateElement()
    {
        return mCoordinateElement;
    }

    const TElement& DofElement(DofType dofType) const override
    {
        return mDofElements[dofType];
    }

    TElement& DofElement(DofType dofType)
    {
        return mDofElements.At(dofType);
    }

    bool Has(DofType dof) const
    {
        return mDofElements.Has(dof);
    }

    const Shape& GetShape() const override
    {
        return mShape;
    }

private:
    TElement mCoordinateElement;
    DofContainer<TElement> mDofElements;
    const Shape& mShape;
};

using ElementCollectionFem = ElementCollectionImpl<NuTo::ElementFem>;

template <int TDimParameter>
using ElementCollectionIga = ElementCollectionImpl<NuTo::ElementIga<TDimParameter>>;

} /* NuTo */