pzstrmatrixst.h

Go to the documentation of this file.

#ifndef TPZStructMatrixST_H
#define TPZStructMatrixST_H

#include <set>
#include <map>
#include <semaphore.h>
#include "pzvec.h"
#include "tpzautopointer.h"
#include "pzcmesh.h"
#include "pzelmat.h"
#include "TPZSemaphore.h"
#include "TPZEquationFilter.h"
#include "TPZGuiInterface.h"
#include "pzmatrix.h"
#include "pzfmatrix.h"

class TPZStructMatrixST;
#include "TPZStructMatrixBase.h"


class TPZStructMatrixST : public TPZStructMatrixBase {
    
public:
    
    TPZStructMatrixST(): TPZStructMatrixBase(){}
    
    TPZStructMatrixST(TPZCompMesh *);
    
    TPZStructMatrixST(TPZAutoPointer<TPZCompMesh> cmesh);
    
    TPZStructMatrixST(const TPZStructMatrixST &copy);
    
    virtual ~TPZStructMatrixST(){};
        
    virtual TPZMatrix<STATE> * Create() override;
    
    virtual TPZStructMatrixST * Clone() override;

    virtual TPZMatrix<STATE> * CreateAssemble(TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface) override;
    
    virtual TPZMatrix<STATE> * CreateAssemble(TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface,
                                              unsigned numthreads_assemble, unsigned numthreads_decompose) {
        SetNumThreads(numthreads_assemble);
        return CreateAssemble(rhs, guiInterface);
    }
    
    virtual void Assemble(TPZMatrix<STATE> &mat, TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface) override;
    
    virtual void Assemble(TPZMatrix<STATE> &mat, TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface,
                          unsigned numthreads_assemble, unsigned numthreads_decompose) {
        std::cout << "Nothing to do." << std::endl;
    }
    
    virtual void Assemble(TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface) override;
    
    public:
int ClassId() const override;
    void Read(TPZStream &buf, void *context) override;
    void Write(TPZStream &buf, int withclassid) const override;

    
protected:
    
    virtual void OnlyAssemble(TPZMatrix<STATE> *mat, TPZFMatrix<STATE> *rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);
    
    virtual void OnlyAssemble(TPZFMatrix<STATE> *rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);
    
    virtual void ExecuteAssemble(TPZMatrix<STATE> *fGlobMatrix, TPZFMatrix<STATE> *fGlobRhs, TPZAutoPointer<TPZGuiInterface> guiInterface);
    
public:
    
    void SetEquationRange(int64_t mineq, int64_t maxeq) override
    {
        fEquationFilter.Reset();
        fEquationFilter.SetMinMaxEq(mineq, maxeq);
    }
    
    virtual bool HasRange() const override
    {
        return fEquationFilter.IsActive();
    }
    
    TPZEquationFilter &EquationFilter() override
    {
        return fEquationFilter;
    }
    
    int64_t NReducedEquations() const override
    {
        return fEquationFilter.NActiveEquations();
    }
    
    TPZCompMesh *Mesh() const override
    {
        return fMesh;
    }
    
    virtual void FilterEquations(TPZVec<int64_t> &origindex, TPZVec<int64_t> &destindex) const override;
    
    void SetMaterialIds(const std::set<int> &materialids) override;
    
    bool ShouldCompute(int matid) const override
    {
        const size_t size = fMaterialIds.size();
        return size == 0 || fMaterialIds.find(matid) != fMaterialIds.end();
    }
    const std::set<int> &MaterialIds() override
    {
        return fMaterialIds;
    }
    
protected:
    
    TPZCompMesh * fMesh;
    TPZAutoPointer<TPZCompMesh> fCompMesh;
    TPZEquationFilter fEquationFilter;
    
protected:
    
    std::set<int> fMaterialIds;
    
    int fNumThreads;
};

#endif