pzstrmatrixcs.h

Go to the documentation of this file.

#ifndef TPZStructMatrixCS_H
#define TPZStructMatrixCS_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 TPZStructMatrixCS;
#include "TPZStructMatrixBase.h"

#ifdef USING_TBB
#include <tbb/tbb.h>
#endif

//void UniformRefine(int num, TPZGeoMesh &m);

class TPZStructMatrixCS : public TPZStructMatrixBase{
    
public:
    
    TPZStructMatrixCS() : TPZStructMatrixBase(){}
    
    TPZStructMatrixCS(TPZCompMesh *);
    
    TPZStructMatrixCS(TPZAutoPointer<TPZCompMesh> cmesh);
    
    TPZStructMatrixCS(const TPZStructMatrixCS &copy);
    
    virtual ~TPZStructMatrixCS(){};
    
    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 TPZStructMatrixCS * 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);
    
protected:
    
    struct ThreadData
    {
        ThreadData(TPZStructMatrixCS *strmat,TPZMatrix<STATE> &mat, TPZFMatrix<STATE> &rhs, std::set<int> &MaterialIds, TPZAutoPointer<TPZGuiInterface> guiInterface);
        ThreadData(TPZStructMatrixCS *strmat, TPZFMatrix<STATE> &rhs, std::set<int> &MaterialIds, TPZAutoPointer<TPZGuiInterface> guiInterface);
        ~ThreadData();
        int64_t NextElement();
        static void *ThreadWork(void *threaddata);
        bool ShouldCompute(int matid)
        {
            return fStruct->ShouldCompute(matid);
        }
        
        TPZStructMatrixCS *fStruct;
        TPZAutoPointer<TPZGuiInterface> fGuiInterface;
        TPZMatrix<STATE> *fGlobMatrix;
        TPZFMatrix<STATE> *fGlobRhs;
        std::map<int, std::pair< TPZAutoPointer<TPZElementMatrix>, TPZAutoPointer<TPZElementMatrix> > > fSubmitted;
        int64_t fNextElement;
        pthread_mutex_t fAccessElement;
        pthread_mutex_t fAccessElementK;
        pthread_mutex_t fAccessElementF;
        TPZSemaphore fAssembly;
    };
    
#ifdef USING_TBB
    struct AssembleTask {
        ThreadData *data;
        AssembleTask(ThreadData *dt) : data(dt) {};
        void operator()(const tbb::blocked_range<size_t>& range) const;
    };
#endif
    
    friend struct ThreadData;
};

#endif