TPZStrMatrixGCTP.h

Go to the documentation of this file.

#ifndef TPZStructMatrixGCTP_H
#define TPZStructMatrixGCTP_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 TPZStructMatrixGCTP;
#include "TPZStructMatrixBase.h"

#ifdef USING_BOOST
#include <boost/atomic.hpp>
#endif

class TPZStructMatrixGCTP : public TPZStructMatrixBase {
public:

    TPZStructMatrixGCTP() : TPZStructMatrixBase() {
    }

    TPZStructMatrixGCTP(TPZCompMesh *);

    TPZStructMatrixGCTP(TPZAutoPointer<TPZCompMesh> cmesh);

    TPZStructMatrixGCTP(const TPZStructMatrixGCTP &copy);

    virtual ~TPZStructMatrixGCTP() {
    };

    virtual TPZMatrix<STATE> * Create() 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 TPZMatrix<STATE> * CreateAssemble(TPZFMatrix<STATE> &rhs, TPZAutoPointer<TPZGuiInterface> guiInterface) override;

    virtual TPZStructMatrixGCTP * Clone() override;

    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 Serial_Assemble(TPZMatrix<STATE> & mat, TPZFMatrix<STATE> & rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);

    virtual void Serial_Assemble(TPZFMatrix<STATE> & rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);

    virtual void MultiThread_Assemble(TPZFMatrix<STATE> & rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);

    virtual void MultiThread_Assemble(TPZMatrix<STATE> & mat, TPZFMatrix<STATE> & rhs, TPZAutoPointer<TPZGuiInterface> guiInterface);

public:

    static void OrderElement(TPZCompMesh *cmesh, TPZVec<int64_t> &ElementOrder);

    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:

    TPZManVector<int64_t> fElementOrder;
    TPZVec<int64_t> fElementColors;
    int64_t fNColors;
};

#endif