DofContainer.h

Go to the documentation of this file.

#pragma once

#include <map>
#include <vector>
#include "nuto/mechanics/dofs/DofType.h"

namespace NuTo
{
template <typename T>
class DofContainer
{
public:
    virtual ~DofContainer() = default;

    //          an existing value
    //          an newly default constructed value
    T& operator[](DofType dofType)
    {
        return mData[dofType];
    }

    //          require T to be default constructable.
    T& At(DofType dofType)
    {
        return mData.at(dofType);
    }

    const T& operator[](DofType dofType) const
    {
        return mData.at(dofType);
    }

    void Insert(DofType dofType, T t)
    {
        auto it = mData.emplace(dofType, t); // it = pair<iterator, bool>
        if (not it.second)
            throw Exception(__PRETTY_FUNCTION__,
                            "Insert failed. Container already contains an entry for " + dofType.GetName() + ".");
    }

    bool Has(DofType dofType) const
    {
        return mData.find(dofType) != mData.end();
    }

    auto begin() const
    {
        return mData.begin();
    }

    auto end() const
    {
        return mData.end();
    }

protected:
    std::map<DofType, T, CompareDofType> mData;
};
} /* NuTo */