gather_modelpart_utility.h

Go to the documentation of this file.

//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Riccardo Rossi
//                   Michael Andre, https://github.com/msandre
//                   Jordi Cotela Dalmau
//                   Vicente Mataix Ferrandiz
//
 
#pragma once
 
// System includes
 
// External includes
 
// Project includes
#include "containers/variable.h"
#include "containers/array_1d.h"
#include "includes/model_part.h"
 
 
 
 
 
 
namespace Kratos
{
class KRATOS_API(KRATOS_MPI_CORE) GatherModelPartUtility
{
public:
 
    using NodesContainerType = ModelPart::NodesContainerType;
 
    using ElementsContainerType = ModelPart::ElementsContainerType;
 
    using ConditionsContainerType = ModelPart::ConditionsContainerType;
 
    KRATOS_CLASS_POINTER_DEFINITION(GatherModelPartUtility);
 
 
    GatherModelPartUtility(
        const int GatherRank,
        ModelPart& rOriginModelPart,
        ModelPart& rDestinationModelPart
        ) : GatherModelPartUtility(GatherRank, rOriginModelPart, 0, rDestinationModelPart)
    {
        // Nothing to do here
    }
 
    GatherModelPartUtility(
        const int GatherRank,
        ModelPart& rOriginModelPart,
        const int MeshId,
        ModelPart& rDestinationModelPart
        );
 
 
 
    void GatherOnMaster();
 
    template <class TDataType>
    void GatherOnMaster(const Variable<TDataType>& rVariable);
 
    template <class TDataType>
    void ScatterFromMaster(const Variable<TDataType>& rVariable);
 
    static void GatherEntitiesFromOtherPartitions(
        ModelPart& rModelPart,
        const std::map<int, std::vector<std::size_t>>& rNodesToBring,
        const std::map<int, std::vector<std::size_t>>& rElementsToBring,
        const std::map<int, std::vector<std::size_t>>& rConditionsToBring,
        const bool CallExecuteAfterBringingEntities = true,
        const int EchoLevel = 0
        );
 
    static void GatherNodesFromOtherPartitions(
        ModelPart& rModelPart,
        const std::map<int, std::vector<std::size_t>>& rNodesToBring,
        const bool CallExecuteAfterBringingEntities = true,
        const int EchoLevel = 0
        );
 
    static void GatherElementsFromOtherPartitions(
        ModelPart& rModelPart,
        const std::map<int, std::vector<std::size_t>>& rElementsToBring,
        const bool CallExecuteAfterBringingEntities = true,
        const int EchoLevel = 0
        );
 
    static void GatherConditionsFromOtherPartitions(
        ModelPart& rModelPart,
        const std::map<int, std::vector<std::size_t>>& rConditionsToBring,
        const bool CallExecuteAfterBringingEntities = true,
        const int EchoLevel = 0
        );
 
 
 
 
    std::string Info() const;
 
    void PrintInfo(std::ostream& rOStream) const;
 
    void PrintData(std::ostream& rOStream) const;
 
 
private:
 
 
    ModelPart& mrModelPart; 
    int mGatherRank;        
    
 
 
    template <class TObjectType>
    static void GatherEntityFromOtherPartitions(
        ModelPart& rModelPart,
        const std::map<int, std::vector<std::size_t>>& rEntitiesToBring,
        const int EchoLevel = 0
        );
 
 
 
 
 
    GatherModelPartUtility& operator=(GatherModelPartUtility const& rOther) = delete;
 
 
    GatherModelPartUtility(GatherModelPartUtility const& rOther) = delete;
 
 
}; // Class GatherModelPartUtility
 
 
extern template void GatherModelPartUtility::GatherOnMaster(const Variable<double>&);
extern template void GatherModelPartUtility::GatherOnMaster(const Variable<array_1d<double, 3>>&);
extern template void GatherModelPartUtility::ScatterFromMaster(const Variable<double>&);
extern template void GatherModelPartUtility::ScatterFromMaster(const Variable<array_1d<double, 3>>&);
 
 
inline std::istream & operator >>(std::istream& rIStream,
                                  GatherModelPartUtility& rThis)
{
    return rIStream;
}
 
inline std::ostream & operator <<(std::ostream& rOStream,
                                  const GatherModelPartUtility& rThis)
{
    rThis.PrintInfo(rOStream);
    rOStream << std::endl;
    rThis.PrintData(rOStream);
 
    return rOStream;
}
 
} // namespace Kratos.