pzelmat.h

Go to the documentation of this file.

#ifndef ELMATHPP
#define ELMATHPP


#include "pzmatrix.h"
#include "pzblock.h"
#include "pzconnect.h"
#include "pzfmatrix.h"
#include "pzmanvector.h"
#include "pzstack.h"
#include "TPZOneShapeRestraint.h"




struct TPZElementMatrix {
                
    enum MType{Unknown = 0, EF = 1, EK = 2};
                
                MType fType;
                
                TPZCompMesh * fMesh;
                
                TPZStack<int64_t> fConnect;
                TPZFNMatrix<1000, STATE> fMat;
                TPZBlock<STATE> fBlock;
                TPZStack<int64_t> fConstrConnect;
                //TPZFNMatrix<1000> fConstrMat;
                TPZFNMatrix<1000, STATE> fConstrMat;
                //TPZBlock<REAL> fConstrBlock;
                TPZBlock<STATE> fConstrBlock;
                
                TPZManVector<int64_t> fDestinationIndex, fSourceIndex;
    
    std::list<TPZOneShapeRestraint> fOneRestraints;
                
                void Reset(TPZCompMesh *mesh = NULL, MType type=Unknown)
                {
      fMesh = mesh;
      fType = type;
      fConnect.Resize(0);
      fMat.Resize(0,0);
      fBlock.SetNBlocks(0);
      fConstrConnect.Resize(0);
      fConstrMat.Resize(0,0);
      fConstrBlock.SetNBlocks(0);
                }
                
                TPZElementMatrix(TPZCompMesh *mesh, MType type) : fType(type), fMesh(mesh), fConnect(), fMat(0,0), fBlock(),  fConstrConnect(), fConstrMat(0,0), fConstrBlock()
    {
        fBlock.SetMatrix(&fMat);
        fConstrBlock.SetMatrix(&fConstrMat);
    }

    TPZElementMatrix() : fType(Unknown), fMesh(NULL), fConnect(), fMat(0,0), fBlock(&fMat), fConstrConnect(), 
      fConstrMat(0,0), fConstrBlock(&fConstrMat)
    {}
                
    TPZElementMatrix(const TPZElementMatrix &copy);
                
                ~TPZElementMatrix(){
                }
                
                int NConnects(){
                                return fConnect.NElements();
                }
                
                int64_t ConnectIndex(int i)const{
                                return fConnect[i];
                }
                
                void Print(std::ostream &out);
                
                void SetMatrixSize(short NumBli, short NumBlj, short BlSizei, short BlSizej);
                
                void SetMatrixMinSize(short NumBli, short NumBlj, short BlMinSizei, short BlMinSizej);
                
                void ComputeDestinationIndices();
    
    void PermuteGather(TPZVec<int64_t> &permute);
                
                
                bool HasDependency();
                
                void ApplyConstraints();
    
    void ApplyOneShapeConstraints(int constraintindex);
    
    void BuildDependencyOrder(TPZVec<int64_t> &connectlist, TPZVec<int> &DependenceOrder, TPZCompMesh &mesh);
                
};

#endif