Assembler.h

Go to the documentation of this file.

#pragma once
#include "nuto/mechanics/dofs/DofContainer.h"
#include "nuto/mechanics/dofs/DofVector.h"
#include "nuto/mechanics/dofs/DofMatrix.h"
#include "nuto/mechanics/dofs/DofMatrixSparse.h"

namespace NuTo
{

namespace Assembler
{
void Add(DofVector<double>& rResult, const DofVector<double>& v, const DofVector<int>& numbering,
         std::vector<DofType> dofTypes = {});

void Add(DofMatrixSparse<double>& rSparseMatrix, const DofMatrix<double>& m, const DofVector<int>& numbering,
         std::vector<DofType> dofTypes = {});


void Add(DofMatrixContainer<std::vector<Eigen::Triplet<double>>>& rTriplets, const DofMatrix<double>& m,
         const DofVector<int>& numbering, std::vector<DofType> dofTypes = {});
}

class VectorAssembler
{
public:
    VectorAssembler(DofContainer<int> sizes = {});

    void Resize(DofContainer<int> sizes);

    void Add(const DofVector<double>& v, const DofVector<int>& numbering, std::vector<DofType> dofTypes = {});

    void SetZero();

    const DofVector<double>& Get() const;

private:
    DofVector<double> mVector;
};


class MatrixAssembler
{
public:
    MatrixAssembler(DofContainer<int> sizes = {});

    void Resize(DofContainer<int> sizes);

    void Add(const DofMatrix<double>& m, const DofVector<int>& numbering, std::vector<DofType> dofTypes = {});

    void SetZero();

    void Finish();

    const DofMatrixSparse<double>& Get() const;

private:
    DofMatrixContainer<std::vector<Eigen::Triplet<double>>> mTriplets;
    DofMatrixSparse<double> mMatrix;
    bool mFinished = false;
};


} /* NuTo */