CellData.h

Go to the documentation of this file.

#pragma once

#include "nuto/mechanics/dofs/DofContainer.h"
#include "nuto/mechanics/elements/ElementCollection.h"

namespace NuTo
{


class CellData
{
public:
    CellData(const ElementCollection& elements, int cellId)
        : mElements(elements)
        , mCellId(cellId)
    {
    }

    int GetCellId() const
    {
        return mCellId;
    }

    Eigen::VectorXd GetCoordinates() const
    {
        return mElements.CoordinateElement().ExtractNodeValues();
    }

    const Eigen::VectorXd& GetNodeValues(DofType dofType, int instance = 0) const
    {
        if (instance >= static_cast<int>(mNodeValues[dofType].size()))
            mNodeValues[dofType].resize(instance + 1);

        Eigen::VectorXd& nodeValues = mNodeValues[dofType][instance];
        if (nodeValues.size() == 0)
            nodeValues = mElements.DofElement(dofType).ExtractNodeValues(instance);

        return nodeValues;
    }

    const ElementCollection& Elements() const
    {
        return mElements;
    }

private:
    mutable DofContainer<std::vector<Eigen::VectorXd>> mNodeValues;
    const ElementCollection& mElements;
    int mCellId;
};
} /* NuTo */