SimpleAssembler.h

Go to the documentation of this file.

#pragma once

#include "nuto/base/Group.h"
#include "nuto/mechanics/cell/CellInterface.h"
#include "nuto/mechanics/dofs/DofNumbering.h"
#include "nuto/mechanics/dofs/DofVector.h"
#include "nuto/mechanics/dofs/DofMatrixSparse.h"

namespace NuTo
{
class SimpleAssembler
{
public:
    SimpleAssembler() = default;
    SimpleAssembler(DofInfo dofInfo);

    DofVector<double> BuildVector(const Group<CellInterface>& cells, std::vector<DofType> dofTypes,
                                  CellInterface::VectorFunction f) const;

    DofMatrixSparse<double> BuildMatrix(const Group<CellInterface>& cells, std::vector<DofType> dofTypes,
                                        CellInterface::MatrixFunction f) const;

    DofVector<double> BuildDiagonallyLumpedMatrix(const Group<CellInterface>& cells, std::vector<DofType> dofTypes,
                                                  CellInterface::MatrixFunction f) const;

    void SetDofInfo(DofInfo dofInfo);

private:
    DofVector<double> ProperlyResizedVector(std::vector<DofType> dofTypes) const;
    DofMatrixSparse<double> ProperlyResizedMatrix(std::vector<DofType> dofTypes) const;

    DofInfo mDofInfo;

    void ThrowOnZeroDofNumbering(std::vector<DofType> dofTypes) const;
};
} /* NuTo */