readMeshFromFile.cpp

Go to the documentation of this file.

 
#include "readMeshFromFile.h"
 
#ifdef USE_PETSC
#include <petsc.h>
#endif
 
#include <boost/algorithm/string/erase.hpp>
 
#include "BaseLib/FileTools.h"
#include "BaseLib/Logging.h"
#include "BaseLib/MPI.h"
#include "BaseLib/StringTools.h"
#include "MeshLib/IO/Legacy/MeshIO.h"
#include "MeshLib/IO/VtkIO/VtuInterface.h"
#include "MeshLib/Mesh.h"
 
#ifdef USE_PETSC
#include "MeshLib/IO/MPI_IO/NodePartitionedMeshReader.h"
#include "MeshLib/NodePartitionedMesh.h"
#endif
 
namespace
{
MeshLib::Mesh* readMeshFromFileSerial(const std::string& file_name,
                                      bool const compute_element_neighbors)
{
    if (BaseLib::hasFileExtension(".msh", file_name))
    {
        MeshLib::IO::Legacy::MeshIO meshIO;
        return meshIO.loadMeshFromFile(file_name);
    }
 
    if (BaseLib::hasFileExtension(".vtu", file_name))
    {
        return MeshLib::IO::VtuInterface::readVTUFile(
            file_name, compute_element_neighbors);
    }
 
    if (BaseLib::hasFileExtension(".vtk", file_name))
    {
        return MeshLib::IO::VtuInterface::readVTKFile(
            file_name, compute_element_neighbors);
    }
 
    ERR("readMeshFromFile(): Unknown mesh file format in file {:s}.",
        file_name);
    return nullptr;
}
}  // namespace
 
namespace MeshLib
{
namespace IO
{
MeshLib::Mesh* readMeshFromFile(const std::string& file_name,
                                bool const compute_element_neighbors)
{
#ifdef USE_PETSC
    BaseLib::MPI::Mpi mpi;
    if (mpi.size > 1)
    {
        MeshLib::IO::NodePartitionedMeshReader read_pmesh(mpi.communicator);
        const std::string file_name_base =
            BaseLib::dropFileExtension(file_name);
        return read_pmesh.read(file_name_base);
    }
    if (mpi.size == 1)
    {
        std::unique_ptr<Mesh> mesh{
            readMeshFromFileSerial(file_name, compute_element_neighbors)};
 
        if (!mesh)
        {
            return nullptr;
        }
 
        return new MeshLib::NodePartitionedMesh(*mesh);
    }
    return nullptr;
#endif
    return readMeshFromFileSerial(file_name, compute_element_neighbors);
}
 
}  // end namespace IO
}  // end namespace MeshLib