Equation.h

Go to the documentation of this file.

#pragma once
#include <vector>
#include "nuto/mechanics/constraints/Term.h"

namespace NuTo
{
namespace Constraint
{
typedef std::function<double(double)> RhsFunction;

class Equation
{
public:
    Equation(const NodeSimple& dependentNode, int dependentComponent, RhsFunction rhs)
        : mRhs(rhs)
        , mDependentTerm(dependentNode, dependentComponent, 1)
    {
    }

    void AddIndependentTerm(Term term)
    {
        mIndependentTerms.push_back(term);
    }

    double GetRhs(double time) const
    {
        return mRhs(time);
    }

    const Term& GetDependentTerm() const
    {
        return mDependentTerm;
    }

    const std::vector<Term>& GetIndependentTerms() const
    {
        return mIndependentTerms;
    }

    int GetDependentDofNumber() const
    {
        return mDependentTerm.GetConstrainedDofNumber();
    }

private:
    RhsFunction mRhs;

    Term mDependentTerm;

    std::vector<Term> mIndependentTerms;
};

} /* Constaint */
} /* NuTo */