TPZRenumbering.h

Go to the documentation of this file.

#ifndef TPZRENUMBERING_H
#define TPZRENUMBERING_H

#include "pzvec.h"
#include "pzmanvector.h"
#include <set>
#include "TPZSavable.h"

class TPZCompMesh;

class TPZRenumbering : public TPZSavable {
public:
                
                int fHDivPermute;
                
                TPZRenumbering() : fNElements(0), fNNodes(0)
                {
                }
                
                TPZRenumbering(int64_t NElements, int64_t NNodes);
                
                virtual ~TPZRenumbering()
                {
                }
                int ClassId() const override;
        void Read(TPZStream &buf, void *context) override;
        void Write(TPZStream &buf, int withclassid) const override;

                void SetElementsNodes(int64_t NElements, int64_t NNodes)
                {
                                this->fNElements = NElements;
                                this->fNNodes = NNodes;
                }
                void SetElementGraph(TPZVec<int64_t> &elgraph, TPZVec<int64_t> &elgraphindex);
                
                void SetNodeWeights(TPZVec<int> &weights)
                {
                                fNodeWeights = weights;
                }
                
                virtual void ClearDataStructures();
                
                virtual void Resequence(TPZVec<int64_t> &perm, TPZVec<int64_t> &iperm)
                {
                                /*if(fHDivPermute)
                                 {
                                 AdjustHDivPermutation(perm, iperm);
                                 }*/
                                std::cout << "Resequence not implemented\n";
                                DebugStop();
                }
                
                
                void ConvertGraph(TPZVec<int64_t> &elgraph, TPZVec<int64_t> &elgraphindex, TPZManVector<int64_t> &nodegraph, TPZManVector<int64_t> &nodegraphindex);
                
                void ConvertToElementoToElementGraph(TPZVec<int64_t> &elgraph, TPZVec<int64_t> &elgraphindex, TPZVec<int64_t> &eltotelgraph, TPZVec<int> &eltoelweight, TPZVec<int64_t> &eltoelgraphindex);

                void NodeToElGraph(TPZVec<int64_t> &elgraph, TPZVec<int64_t> &elgraphindex, TPZVec<int64_t> &nodetoelgraph, TPZVec<int64_t> &nodetoelgraphindex);
                
                int64_t ColorNodes(TPZVec<int64_t> &nodegraph, TPZVec<int64_t> &nodegraphindex, TPZVec<int> &family, TPZVec<int> &colors);
                
        static int64_t ColorElements(const TPZCompMesh *cmesh, const TPZVec<int64_t> &elementIndices, TPZVec<int64_t> &elementColors);
                
                void Print(TPZVec<int64_t> &grapho, TPZVec<int64_t> &graphoindex, const char *name = 0, std::ostream &out = std::cout);
                
                void CornerEqs(unsigned int mincorners, int64_t nelconsider, std::set<int> &eligible, std::set<int> &cornernodes);
                
protected:
                int64_t fNElements;
                
                int64_t fNNodes;
        
                TPZVec<int> fNodeWeights;

                TPZVec<int64_t> fElementGraph;

                TPZVec<int64_t> fElementGraphIndex;
                
};

class TPZCompMesh;
void ResequenceByGeometry(TPZCompMesh *cmesh,const TPZVec<REAL> &normal);

#endif //TPZRENUMBERING_H