SHPInterface.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#pragma once
 
#include <string>
#include <vector>
 
//ShapeLib includes
#include <shapefil.h>
 
 
namespace GeoLib {
    class GEOObjects;
    class Polyline;
}
 
namespace MeshLib {
    class Mesh;
    class Element;
}
 
 
namespace FileIO
{

class SHPInterface final
{
public:
    enum class OGSType
    {
        UNDEFINED   = 0,
        POINT       = 1,
        STATION     = 2,
        POLYLINE    = 3,
        POLYGON     = 4
    };

    explicit SHPInterface(GeoLib::GEOObjects& geoObjects)
        : _geoObjects(geoObjects)
    {
    }

    static bool readSHPInfo(const std::string& filename, int& shapeType,
                            int& numberOfEntities);

    void readSHPFile(const std::string& filename, OGSType choice,
                     const std::string& listName, std::string const& gmsh_path);

    static bool write2dMeshToSHP(const std::string &file_name, const MeshLib::Mesh &mesh);
 
private:
    void readPoints    (const SHPHandle &hSHP, int numberOfElements, std::string listName);

    void readStations  (const SHPHandle &hSHP, int numberOfElements, std::string listName);

    void readPolylines (const SHPHandle &hSHP, int numberOfElements, std::string listName);

    void readPolygons(const SHPHandle& hSHP, int numberOfElements,
                      const std::string& listName,
                      std::string const& gmsh_path);

    static SHPObject* createShapeObject(MeshLib::Element const& e,
                                        std::size_t const shp_record);
 
    GeoLib::GEOObjects& _geoObjects;
};
 
}  // namespace FileIO