tpzgensubstruct.h

Go to the documentation of this file.

#ifndef TPZGENSUBSTRUCT_H
#define TPZGENSUBSTRUCT_H

#include "tpzautopointer.h"
#include "pzcmesh.h"
#include "tpzdohrassembly.h"
#include "tpzdohrmatrix.h"

class TPZSubCompMesh;
template<class TVar>
class TPZDohrSubstruct;
template<class TVar>
class TPZDohrSubstructCondense;

class TPZGenSubStruct{
public:
                enum ETypemesh
                {
                                DistMaterial,RandomMat
                };
                
                ETypemesh fMatDist;
                
                TPZVec<REAL> fK;
                TPZGenSubStruct(int dimension, int numlevels, int substructlevel);
                
    ~TPZGenSubStruct();
    
#ifndef STATE_COMPLEX

    TPZAutoPointer<TPZCompMesh> GenerateMesh();
#endif
    
    void InitializeDohr(TPZAutoPointer<TPZMatrix<STATE> > dohr, TPZAutoPointer<TPZDohrAssembly<STATE> > assembly);
    void InitializeDohrCondense(TPZAutoPointer<TPZMatrix<STATE> > dohr, TPZAutoPointer<TPZDohrAssembly<STATE> > assembly);
                
                void ReorderInternalNodes(TPZSubCompMesh *sub, std::map<int,int> &globaltolocal,
                                                                                                                  TPZVec<int> &internalnodes);
                static void ReorderInternalNodes2(TPZSubCompMesh *sub,
                                                                                                                                                  TPZVec<int> &internalnodes, TPZVec<int64_t> &invpermute);
                
                static void ComputeInternalEquationPermutation(TPZSubCompMesh *sub,
                                                                                                                                                                                                   TPZVec<int> &scatterpermute, TPZVec<int> &gatherpermute);
    
private:
    int fDimension;
    int fNumLevels;
    int fSubstructLevel;
    
    TPZAutoPointer<TPZCompMesh> fCMesh;
    
    std::set<int> fCornerEqs;
    
    void UniformRefine();
    
public:
    void SubStructure();
private:
    void IdentifyCornerNodes();
    
    void IdentifyEqNumbers(TPZSubCompMesh *sub, TPZVec<std::pair<int,int> > &globaleq, std::map<int,int> &globinv);
                
    void IdentifyEqNumbers(TPZSubCompMesh *sub, TPZVec<int> &global, std::map<int,int> &globinv);
                
    void IdentifySubCornerEqs(std::map<int,int> &globaltolocal, TPZVec<int> &cornereqs,
                                                                                                                  TPZVec<int> &coarseindex);
                
                static    int64_t NInternalEq(TPZSubCompMesh *sub);
                
};

void InitializeMatrices(TPZSubCompMesh *sub, TPZAutoPointer<TPZDohrSubstruct<STATE> > substruct,  TPZDohrAssembly<STATE> &dohrassembly);

void InitializeMatrices(TPZSubCompMesh *sub, TPZAutoPointer<TPZDohrSubstructCondense<STATE> > substruct,  TPZDohrAssembly<STATE> &dohrassembly);

int64_t NSubMesh(TPZAutoPointer<TPZCompMesh> compmesh);

TPZSubCompMesh *SubMesh(TPZAutoPointer<TPZCompMesh> compmesh, int isub);

#endif