pybind

{
    "configurations": [
        {
            "name": "Linux",
            "includePath": [
                "${workspaceFolder}/**",
                "/usr/include",
                "/usr/local/include",
                "/usr/include/eigen3",
                "/usr/include/python3.12"
            ],
            "defines": [],
            "compilerPath": "/usr/bin/gcc",
            "cStandard": "c17",
            "cppStandard": "c++17",
            "intelliSenseMode": "linux-gcc-arm64"
        }
    ],
    "version": 4
}

.
|-- CMakeLists.txt


cmake_minimum_required(VERSION 3.12)

project(example VERSION 1.0.0 LANGUAGES C CXX)

find_package(GEOS 3.12 REQUIRED)
find_package(Python 3.12 COMPONENTS Interpreter Development REQUIRED)
find_package(pybind11 CONFIG REQUIRED)
find_package (GeographicLib REQUIRED)
find_package(Eigen3 REQUIRED NO_MODULE)

message("GeographicLib_INCLUDE_DIRS:" ${GeographicLib_INCLUDE_DIRS})
message("GeographicLib_LIBRARIES:" ${GeographicLib_LIBRARIES})
message("GEOS_LIBRARIES:" ${GEOS_LIBRARIES})
message("GEOS_INCLUDE_DIR:" ${GEOS_INCLUDE_DIR})

file(GLOB SRC_LIST1 "*.cpp")
file(GLOB SRC_LIST2 "src/*.cpp")

include_directories(${GEOS_INCLUDE_DIR}) 
include_directories(${GeographicLib_INCLUDE_DIRS})

include_directories(
    /usr/local/include
    /usr/include/eigen3
    ${CMAKE_CURRENT_SOURCE_DIR}/include
)

link_directories(
    /usr/local/lib
)

pybind11_add_module(example ${SRC_LIST1} ${SRC_LIST2})


target_link_libraries(example 
    PRIVATE GEOS::geos
    ${GeographicLib_LIBRARIES}
    )

add_subdirectory(app)


|-- apt.install

wget https://github.com/geographiclib/geographiclib/archive/refs/tags/v2.4.tar.gz
cd geographiclib
mkdir build && cd build
cmake .. #无报错
make #无报错
make install #无报错

===================================================================================================================

apt install libeigen3-dev

===================================================================================================================


===================================================================================================================


|-- build
|-- app
| |-- CMakeLists.txt


# cmake_minimum_required(VERSION 3.12)

# project(example VERSION 1.0.0 LANGUAGES C CXX)

# find_package(GEOS 3.12 REQUIRED)
# find_package(pybind11 REQUIRED)
# find_package (GeographicLib REQUIRED)
# find_package(Eigen3 REQUIRED NO_MODULE)

# message("GeographicLib_INCLUDE_DIRS:" ${GeographicLib_INCLUDE_DIRS})
# message("GeographicLib_LIBRARIES:" ${GeographicLib_LIBRARIES})
# message("GEOS_LIBRARIES:" ${GEOS_LIBRARIES})
# message("GEOS_INCLUDE_DIR:" ${GEOS_INCLUDE_DIR})


# include_directories(${GEOS_INCLUDE_DIR}) 
# include_directories(${GeographicLib_INCLUDE_DIRS})

# include_directories(
#     /usr/local/include
#     /usr/include/eigen3
#     ${CMAKE_CURRENT_SOURCE_DIR}/../include
# )

# link_directories(
#     /usr/local/lib
# )

# # 将链接模式设为静态
# # set(CMAKE_EXE_LINKER_FLAGS "-static")
# # set(CMAKE_EXE_LINKER_FLAGS "-static -static-libgcc -static-libstdc++")


add_executable(example_exe main.cpp)

target_link_libraries(example_exe 
    PRIVATE GEOS::geos
    ${GeographicLib_LIBRARIES}
)


| |-- build
| `-- main.cpp

#include <iostream>
#include <geoutil.hpp>

int main()
{
    Eigen::Vector2d ba(1, 1);
    Eigen::Vector2d bc(1,0);
    float angle = GeoUtil::getVectorAngle(ba, bc);
    std::cout << angle << std::endl;
}

|-- include
| |-- a.h

#ifndef A_H_
#define A_H_

#include <iostream>
#include <string>

int add(int i, int j);
std::string lineInterpolatePoint(const std::string wkt, float index);
std::string getIntersect();

#endif

| |-- duckdb_udf.h

#pragma once

#include <string>

namespace DuckdbUDF
{
    float geomZDiff(const std::string &geomWkt);
    float lineStringLength2d(const std::string &geomWkt);
    float line3PointMinAngle(const std::string &geomWkt);
}

| `-- geoutil.hpp

#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <geos/io/WKTReader.h>
#include <GeographicLib/Geodesic.hpp>
#include <Eigen/Dense>

namespace GeoUtil
{
    std::unique_ptr<geos::geom::LineString> getLineStringFromWkt(const std::string &wkt)
    {
        geos::io::WKTReader reader;
        std::unique_ptr<geos::geom::Geometry> geom = nullptr;
        try
        {
            geom = reader.read(wkt);
        }
        catch (const geos::util::GEOSException &e)
        {
            std::cerr << "Failed to read WKT geometry: " << e.what() << std::endl;
            return nullptr;
        }

        if (!geom || geom->getGeometryTypeId() != geos::geom::GEOS_LINESTRING)
        {
            std::cerr << "The WKT geometry is not a LineString." << std::endl;
            return nullptr;
        }

        return std::unique_ptr<geos::geom::LineString>(dynamic_cast<geos::geom::LineString *>(geom.release()));
    }
    std::vector<geos::geom::Coordinate> getCoordsFromWkt(const std::string &wkt)
    {
        std::vector<geos::geom::Coordinate> coords;
        geos::io::WKTReader reader;
        std::unique_ptr<geos::geom::Geometry> geom = nullptr;
        try
        {
            geom = reader.read(wkt);
        }
        catch (const geos::util::GEOSException &e)
        {
            std::cerr << "Failed to read WKT geometry: " << e.what() << std::endl;
            return coords;
        }
        if (geom)
        {
            std::unique_ptr<geos::geom::CoordinateSequence> coordSeq;
            if (geom->getGeometryTypeId() == geos::geom::GEOS_POINT)
            {
                std::unique_ptr<geos::geom::Point> point(dynamic_cast<geos::geom::Point *>(geom.release()));
                coordSeq = point->getCoordinates();
            }
            else if (geom->getGeometryTypeId() == geos::geom::GEOS_LINESTRING)
            {
                std::unique_ptr<geos::geom::LineString> lineString(dynamic_cast<geos::geom::LineString *>(geom.release()));
                coordSeq = lineString->getCoordinates();
            }
            else if (geom->getGeometryTypeId() == geos::geom::GEOS_POLYGON)
            {
                std::unique_ptr<geos::geom::Polygon> polygon(dynamic_cast<geos::geom::Polygon *>(geom.release()));
                coordSeq = polygon->getExteriorRing()->getCoordinates();
            }
            for (std::size_t i = 0; i < coordSeq->getSize(); ++i)
            {
                coords.push_back(coordSeq->getAt(i));
            }
        }
        return coords;
    }
    float getDistance2d(float lon1, float lat1, float lon2, float lat2)
    {
        const GeographicLib::Geodesic &geod = GeographicLib::Geodesic::WGS84();
        double distance;
        geod.Inverse(lat1, lon1, lat2, lon2, distance);
        return distance;
    }

    double getVectorAngle(const Eigen::Vector2d &ba, const Eigen::Vector2d &bc)
    {
        // 计算点积
        double dotVal = ba.dot(bc);

        // 计算向量的范数
        double normBa = ba.norm();
        double normBc = bc.norm();

        // 计算 cos(theta)
        double cosTheta = dotVal / (normBa * normBc);

        // 将 cos(theta) 限制在 [-1, 1] 范围内
        cosTheta = std::max(-1.0, std::min(1.0, cosTheta));

        // 计算角度(弧度)
        double angle = std::acos(cosTheta);

        // 将角度转换为度
        return angle * (180.0 / M_PI);
    }

}

|-- pybind.cpp

#include <pybind11/pybind11.h>
#include "a.h"
#include "duckdb_udf.h"

PYBIND11_MODULE(example, m)
{
      m.doc() = "pybind11 example plugin"; // optional module docstring

      m.def("add", &add, "A function which adds two numbers");
      m.def("add_lambda", [](int a, int b)
            { return a + b; }, "A function which adds two numbers");
      m.def("getIntersect", &getIntersect, "getIntersect");
      m.def("lineInterpolatePoint", &lineInterpolatePoint, "lineInterpolatePoint",
            pybind11::arg("line_wkt"), pybind11::arg("index") = 0.6);
      m.def("geomZDiff", &DuckdbUDF::geomZDiff, "geom_z_diff");
      m.def("lineStringLength2d", &DuckdbUDF::lineStringLength2d, "lineStringLength2d");
      m.def("line3PointMinAngle", &DuckdbUDF::line3PointMinAngle, "line3PointMinAngle");
}

`-- src
|-- a.cpp


#include <iostream>
#include <geos/io/WKTReader.h>
#include <geos/io/WKTWriter.h>
#include <geos/linearref/LengthIndexOfPoint.h>
#include <geos/linearref/LengthIndexedLine.h>
#include <geos/geom/CoordinateSequence.h>
#include <geos/geom/Coordinate.h>
#include <string>
#include "a.h"

int add(int i, int j)
{
    return i + j;
}

std::string getIntersect()
{
    geos::geom::GeometryFactory::Ptr factory = geos::geom::GeometryFactory::create();
    geos::io::WKTReader reader(*factory);

    std::string wkt_a("POLYGON((0 0, 10 0, 10 10, 0 10, 0 0))");
    std::string wkt_b("POLYGON((5 5, 15 5, 15 15, 5 15, 5 5))");

    std::unique_ptr<geos::geom::Geometry> geom_a(reader.read(wkt_a));
    std::unique_ptr<geos::geom::Geometry> geom_b(reader.read(wkt_b));

    std::unique_ptr<geos::geom::Geometry> inter = geom_a->intersection(geom_b.get());

    geos::io::WKTWriter writer;
    writer.setTrim(true);
    std::string inter_wkt = writer.write(inter.get());
    return inter_wkt;
}
std::string lineInterpolatePoint(const std::string line_wkt, float index)
{
    geos::io::WKTReader wktReader;
    std::unique_ptr<geos::geom::Geometry> geom(wktReader.read(line_wkt));
    if (!geom)
    {
        std::cout << "wkt有错误" << std::endl;
        return "";
    }
    std::string geomType = geom->getGeometryType();
    if (geomType != "LineString")
    {
        std::cout << "wkt需要是linestring类型" << std::endl;
        return "";
    }
    std::unique_ptr<geos::geom::LineString> line(static_cast<geos::geom::LineString *>(geom.release()));
    geos::linearref::LengthIndexedLine lengthIndexedLine(line.release());
    geos::geom::Coordinate c = lengthIndexedLine.extractPoint(0.5);
    geos::geom::GeometryFactory::Ptr geomFactory = geos::geom::GeometryFactory::create();
    std::unique_ptr<geos::geom::Point> point = geomFactory->createPoint(c);
    // std::cout << point->getX() << '\t' << point->getY() << std::endl;
    geos::io::WKTWriter wktWriter;
    std::string pointWkt = wktWriter.write(point.get());
    return pointWkt;
}

`-- duckdb_udf.cpp
#include <iostream>
#include <string>
#include <geos/io/WKTReader.h>
#include "duckdb_udf.h"
#include "geoutil.hpp"

float DuckdbUDF::geomZDiff(const std::string &geomWkt)
{
    try
    {
        std::unique_ptr<geos::geom::LineString> lineString = GeoUtil::getLineStringFromWkt(geomWkt);
        if (lineString)
        {
            if (lineString->hasZ())
            {
                std::unique_ptr<geos::geom::CoordinateSequence> coordSeq = lineString->getCoordinates();
                const geos::geom::Coordinate &firstCoord = coordSeq->getAt(0);
                double maxZ = firstCoord.z;
                double minZ = firstCoord.z;
                for (std::size_t i = 1; i < coordSeq->getSize(); ++i)
                {
                    const geos::geom::Coordinate &coord = coordSeq->getAt(i);
                    if (coord.z > maxZ)
                    {
                        maxZ = coord.z;
                    }
                    if (coord.z < minZ)
                    {
                        minZ = coord.z;
                    }
                }
                return maxZ - minZ;
            }
        }
    }
    catch (const geos::util::GEOSException &e)
    {
        std::cerr << "GEOS Exception: " << e.what() << std::endl;
    }
    catch (const std::exception &e)
    {
        std::cerr << "Standard Exception: " << e.what() << std::endl;
    }

    return 0;
}

float DuckdbUDF::lineStringLength2d(const std::string &geomWkt)
{
    try
    {
        std::vector<geos::geom::Coordinate> coords = GeoUtil::getCoordsFromWkt(geomWkt);
        if (coords.size() > 0)
        {
            float distance = 0;
            for (int i = 1; i < coords.size(); ++i)
            {
                geos::geom::Coordinate coord0 = coords[i - 1];
                geos::geom::Coordinate coord1 = coords[i];
                distance += GeoUtil::getDistance2d(coord0.x, coord0.y, coord1.x, coord1.y);
            }
            return distance;
        }
    }
    catch (const std::exception &e)
    {
        std::cerr << "Standard Exception: " << e.what() << std::endl;
    }

    return 0;
}

float DuckdbUDF::line3PointMinAngle(const std::string &geomWkt)
{
    try
    {
        std::vector<geos::geom::Coordinate> coords = GeoUtil::getCoordsFromWkt(geomWkt);
        std::cout << geomWkt << std::endl;
        std::cout << coords.size() << std::endl;
        if (coords.size() > 2)
        {
            float minAngle = 180;
            for (int i = 0; i < coords.size() - 2; ++i)
            {
                geos::geom::Coordinate coord0 = coords[i];
                geos::geom::Coordinate coord1 = coords[i + 1];
                geos::geom::Coordinate coord2 = coords[i + 2];
                Eigen::Vector2d ba(coord0.x - coord1.x, coord0.y - coord1.y);
                Eigen::Vector2d bc(coord2.x - coord1.x, coord2.y - coord1.y);
                float angle = GeoUtil::getVectorAngle(ba, bc);
                if (angle < minAngle)
                {
                    minAngle = angle;
                }
            }
            return minAngle;
        }
    }
    catch (const std::exception &e)
    {
        std::cerr << "Standard Exception: " << e.what() << std::endl;
    }

    return 0;
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,843评论 6 502
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,538评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,187评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,264评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,289评论 6 390
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,231评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,116评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,945评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,367评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,581评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,754评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,458评论 5 344
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,068评论 3 327
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,692评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,842评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,797评论 2 369
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,654评论 2 354

推荐阅读更多精彩内容