tpzpairstructmatrix.h

Go to the documentation of this file.

#ifndef PAIRSTRUCTMATRIX
#define PAIRSTRUCTMATRIX

#include "pzcmesh.h"
#include "pzvec.h"
#include "tpzautopointer.h"
#include "pzmatrix.h"
#include "TPZGuiInterface.h"
#include "pzelmat.h"
#include "TPZSemaphore.h"
#include "pzstrmatrix.h"

//class TPZStructMatrix;

class TPZPairStructMatrix
{
                TPZVec<int> fPermuteScatter;
    TPZStructMatrix fStrMatrix;
                
                void PermuteScatter(TPZVec<int> &index);
                void PermuteScatter(TPZVec<int64_t> &index);
                
public:
                
                static int gNumThreads;
                
                TPZPairStructMatrix(TPZCompMesh *mesh, TPZVec<int> &permutescatter);
    
    ~TPZPairStructMatrix()
    {
        
    }
                
                void SetNumThreads(int numthreads)
                {
                                fStrMatrix.SetNumThreads(numthreads);
                }
                
                void SetMaterialIds(const std::set<int> &materialids);
                
                void Assemble(TPZMatrix<STATE> *first, TPZMatrix<STATE> *second, TPZFMatrix<STATE> &rhs);
                void TBBAssemble(TPZMatrix<STATE> *first, 
                     TPZMatrix<STATE> *second, TPZFMatrix<STATE> &rhs);

                void SerialAssemble(TPZMatrix<STATE> *first, TPZMatrix<STATE> *second, TPZFMatrix<STATE> &rhs);
                
                void MultiThread_Assemble(TPZMatrix<STATE> *first, TPZMatrix<STATE> *second, TPZFMatrix<STATE> &rhs);
                
                struct ThreadData
                {
                                ThreadData(TPZStructMatrix *strmatrix,TPZMatrix<STATE> &mat1, TPZMatrix<STATE> &mat2, TPZFMatrix<STATE> &rhs);
                                ~ThreadData();
                                TPZStructMatrix *fStrMatrix;
                                pthread_mutex_t fAccessElement;
                                TPZSemaphore fAssembly1;
                                TPZSemaphore fAssembly2;
                                TPZMatrix<STATE> *fGlobMatrix1;
                                TPZMatrix<STATE> *fGlobMatrix2;
                                TPZFMatrix<STATE> *fGlobRhs;
                                TPZVec<int> fPermuteScatter;
                                std::map<int, std::pair< TPZAutoPointer<TPZElementMatrix>, TPZAutoPointer<TPZElementMatrix> > > fSubmitted1;
                                std::map<int, TPZAutoPointer<TPZElementMatrix> > fSubmitted2;
                                std::set<int> fProcessed1;
                                std::set<int> fProcessed2;
                                int fNextElement;
                                int NextElement();
                                void ComputedElementMatrix(int iel, TPZAutoPointer<TPZElementMatrix> &ek, TPZAutoPointer<TPZElementMatrix> &ef);
                                static void *ThreadWork(void *threaddata);
                                static void *ThreadAssembly1(void *threaddata);
                                static void *ThreadAssembly2(void *threaddata);
                                
                                bool ShouldCompute(int matid)
                                {
            return fStrMatrix->ShouldCompute(matid);
                                }
                                void PermuteScatter(TPZVec<int> &index);
                                void PermuteScatter(TPZVec<int64_t> &index);
                                
                };
                
};

#endif