TimeDependentProblem.h

Go to the documentation of this file.

#pragma once
#include "nuto/mechanics/cell/CellInterface.h"
#include "nuto/mechanics/cell/SimpleAssembler.h"
#include "nuto/base/Group.h"
#include "nuto/mechanics/dofs/DofVector.h"
#include "nuto/mechanics/dofs/DofMatrixSparse.h"

#include "nuto/mechanics/tools/NodalValueMerger.h"

namespace NuTo
{

class TimeDependentProblem
{
public:
    // let us call these "time dependent functions"
    using GradientFunction = std::function<DofVector<double>(const CellIpData&, double t, double dt)>;
    using HessianFunction = std::function<DofMatrix<double>(const CellIpData&, double t, double dt)>;
    using UpdateFunction = std::function<void(const CellIpData&, double t, double dt)>;

    TimeDependentProblem(MeshFem* rMesh);

    DofVector<double> RenumberDofs(Constraint::Constraints constraints, std::vector<DofType> dofTypes,
                                 DofVector<double> oldDofValues);

    void AddGradientFunction(Group<CellInterface> group, GradientFunction f);
    void AddHessian0Function(Group<CellInterface> group, HessianFunction f);
    void AddUpdateFunction(Group<CellInterface> group, UpdateFunction f);

    DofVector<double> Gradient(const DofVector<double>& dofValues, std::vector<DofType> dofs, double t, double dt);
    DofMatrixSparse<double> Hessian0(const DofVector<double>& dofValues, std::vector<DofType> dofs, double t, double dt);

    void UpdateHistory(const DofVector<double>& dofValues, std::vector<DofType> dofs, double t, double dt);

private:
    SimpleAssembler mAssembler;
    NodalValueMerger mMerger;

    using GradientPair = std::pair<Group<CellInterface>, GradientFunction>;
    using Hessian0Pair = std::pair<Group<CellInterface>, HessianFunction>;
    using UpdatePair = std::pair<Group<CellInterface>, UpdateFunction>;

    std::vector<GradientPair> mGradientFunctions;
    std::vector<Hessian0Pair> mHessian0Functions;
    std::vector<UpdatePair> mUpdateFunctions;


    /*
     *
     *
     *
     *
     *
     * We better hide these two beauties in the basement of this class
     *
     *
     *
     *
     *
     *
     *
     *
     */

public:
    template <typename TObject, typename TReturn>
    static auto Bind_dt(TObject& object, TReturn (TObject::*f)(const CellIpData&, double dt))
    {
        return [&object, f](const CellIpData& cellIpData, double, double dt) { return (object.*f)(cellIpData, dt); };
    }

    template <typename TObject, typename TReturn>
    static auto Bind_t(TObject& object, TReturn (TObject::*f)(const NuTo::CellIpData&, double t))
    {
        return [&object, f](const CellIpData& cellIpData, double t, double) { return (object.*f)(cellIpData, t); };
    }

    template <typename TObject, typename TReturn>
    static auto Bind(TObject& object, TReturn (TObject::*f)(const NuTo::CellIpData&))
    {
        return [&object, f](const CellIpData& cellIpData, double, double) { return (object.*f)(cellIpData); };
    }
};
} /* NuTo */