TPZBuildSBFem.h

Go to the documentation of this file.

//
//  TPZBuildSBFem.hpp
//  PZ
//
//  Created by Philippe Devloo on 06/01/17.
//
//

#ifndef TPZBuildSBFem_hpp
#define TPZBuildSBFem_hpp

#include <stdio.h>
#include "pzmanvector.h"
#include "pzgmesh.h"
#include "pzcmesh.h"
#include "tpzautopointer.h"
#include <map>

class TPZBuildSBFem
{
    TPZAutoPointer<TPZGeoMesh> fGMesh;
    
    std::map<int,int> fMatIdTranslation;
    
    int fSkeletonMatId;
    
    TPZManVector<int64_t> fElementPartition;
    
    TPZManVector<int64_t> fPartitionCenterNode;
    
public:
    
    TPZBuildSBFem(TPZAutoPointer<TPZGeoMesh> gmesh, int skeletonmatid, std::map<int,int> &matidtranslation) : fGMesh(gmesh), fMatIdTranslation(matidtranslation), fSkeletonMatId(skeletonmatid)
    {
        fElementPartition.resize(fGMesh->NElements());
        fElementPartition.Fill(-1);
    }
    
    void SetMatIdTranslation(const std::map<int,int> &matidtranslation)
    {
        fMatIdTranslation = matidtranslation;
    }
    
    void StandardConfiguration();

    void StandardConfiguration(TPZVec<int64_t> &elementindices);
    
    void Configure(TPZVec<int64_t> &scalingcenters);
    
    void AddPartition(TPZVec<int64_t> &elids, int64_t centernodeindex);
    
    void SetPartitions(TPZVec<int64_t> &gelpartitionids, TPZVec<int64_t> &partition_nodeindices)
    {
#ifdef PZDEBUG
        if(gelpartitionids.size() != fGMesh->NElements())
        {
            DebugStop();
        }
#endif
        fElementPartition = gelpartitionids;
        fPartitionCenterNode = partition_nodeindices;
    }
    
    void BuildComputationMesh(TPZCompMesh &cmesh);
    
    void BuildComputationMesh(TPZCompMesh &cmesh, const std::set<int> &volmatids, const std::set<int> &boundmatids);
    
    void BuildComputationalMeshFromSkeleton(TPZCompMesh &cmesh);
    
    void DivideSkeleton(int nref);

    void DivideSkeleton(int nref, const std::set<int> &volmatids);

private:
    void AddSkeletonElements();
    
    void CreateElementCenterNodes(TPZVec<int64_t> &elindices);
    
    void CreateVolumetricElements(TPZCompMesh &cmesh);
    
    void CreateVolumetricElementsFromSkeleton(TPZCompMesh &cmesh);
    
    void CreateVolumetricElements(TPZCompMesh &cmesh, const std::set<int> &matids);
    
    void CreateElementGroups(TPZCompMesh &cmesh);
};

#endif /* TPZBuildSBFem_hpp */