flkit 0.1.0__py3-none-any.whl

flkit/__init__.py

@@ -0,0 +1,2 @@
+def hello() -> str:
+    return "Hello from flkit!"

flkit/alg/__init__.py

@@ -0,0 +1,68 @@
+"""
+算法库
+
+- c_next_point: 计算钻孔轨迹的下一个拐点坐标
+- get_up_low: 获取钻孔轨迹的上、下点坐标
+- finite_plane_distance: 计算点到有限平面的距离
+- inter: 插值算法
+"""
+
+from .c_track import c_next_point, comp_towards, comp_towards_batch
+from .get_up_low import get_up_low, linear_interpolation
+from .finite_plane_distance import comp_sliding_diff, finite_plane_distance
+from . import inter
+from .in_tin import (
+    is_point_in_triangle,
+    is_point_in_triangles,
+    point_to_triangle_distance,
+)
+from .point_in_polygon import PointInPolygon
+from .rotated_rect import RotatedRect
+from .index import list_index
+from .point_to_triangle_mesh import PointToTriangleMesh
+from .delaunay_triangulation import (
+    DelaunayTriangulation,
+    OptimizedDelaunayTriangulation,
+    TriangleLibTriangulation,
+    point_in_polygon,
+    bound_triangulation,
+    interpolate_delaunary_z,
+    convex_hull_boundary,
+)
+from .point_to_triangle import PointInTriangle
+from .point_to_triangle_rs import PointInTriangle as PointInTriangleRs
+from .point_in_volume import PointInVolume
+from .point_in_volume_rs import PointInVolume as PointInVolumeRs
+
+from .build_idx import build_idx
+
+
+__all__ = [
+    "c_next_point",
+    "comp_towards",
+    "comp_towards_batch",
+    "get_up_low",
+    "linear_interpolation",
+    "comp_sliding_diff",
+    "finite_plane_distance",
+    "inter",
+    "is_point_in_triangle",
+    "is_point_in_triangles",
+    "point_to_triangle_distance",
+    "PointInPolygon",
+    "RotatedRect",
+    "list_index",
+    "PointToTriangleMesh",
+    "DelaunayTriangulation",
+    "OptimizedDelaunayTriangulation",
+    "TriangleLibTriangulation",
+    "point_in_polygon",
+    "bound_triangulation",
+    "interpolate_delaunary_z",
+    "convex_hull_boundary",
+    "PointInTriangle",
+    "PointInTriangleRs",
+    "PointInVolume",
+    "PointInVolumeRs",
+    "build_idx",
+]

flkit/alg/build_idx.py

@@ -0,0 +1,9 @@
+import pandas as pd
+
+def build_idx(df: pd.DataFrame, idx_col: str, include_cols: list[str]) -> list:
+    """
+    构建索引
+    """
+    df[idx_col] = df[include_cols].astype(str).agg(":", sep="").tolist()
+    idxs = df[idx_col].unique().tolist()
+    return idxs

flkit/alg/c_track.py

@@ -0,0 +1,88 @@
+import math
+from typing import Any, Dict, List
+
+
+def c_next_point(start_point, length, inclination, azimuth):
+    """
+    计算钻孔轨迹的下一个拐点坐标
+
+    参数:
+        start_point: 起点坐标 (x, y, z) 元组
+        length: 钻孔长度 (必须为正数)
+        inclination: 倾角/倾斜角 (度，-90到90度，负值表示向上)
+        azimuth: 方位角 (度，0-360度)
+
+    返回:
+        下一个拐点的三维坐标 (x, y, z) 元组
+
+    坐标系约定:
+        - X轴：东西方向（东为正）
+        - Y轴：南北方向（北为正）
+        - Z轴：垂直方向（向下为正，符合钻孔深度增加的习惯）
+
+    倾角约定:
+        - 0度：水平
+        - 90度：垂直向下
+        - -90度：垂直向上
+    """
+    if length < 0:
+        raise ValueError(f"钻孔长度必须为正数{length}")
+
+    # 倾角范围检查（允许负值）
+    if not -90 <= inclination <= 90:
+        raise ValueError("倾角必须在-90到90度之间")
+
+    if not 0 <= azimuth <= 360:
+        raise ValueError("方位角必须在0-360度之间")
+
+    # 转换为弧度
+    inclination_rad = math.radians(inclination)
+    azimuth_rad = math.radians(azimuth)
+
+    # 计算坐标增量
+    # 水平分量（在XY平面上的投影）
+    horizontal_component = length * math.cos(inclination_rad)
+    # 垂直分量（Z方向）
+    vertical_component = length * math.sin(inclination_rad)
+
+    # 分解水平分量到X和Y方向
+    delta_x = horizontal_component * math.sin(azimuth_rad)
+    delta_y = horizontal_component * math.cos(azimuth_rad)
+    delta_z = vertical_component
+
+    # 计算新坐标
+    x = start_point[0] + delta_x
+    y = start_point[1] + delta_y
+    z = start_point[2] + delta_z
+
+    return (x, y, z)
+
+
+def comp_towards(start: dict, end: dict) -> float:
+    """计算从起点指向终点的方向角度（顺时针，以正北为0度）"""
+    if "x" not in start or "y" not in start or "x" not in end or "y" not in end:
+        raise ValueError("点必须包含x和y坐标")
+    dx = end["x"] - start["x"]
+    dy = end["y"] - start["y"]
+    angle = math.acos(dy / math.sqrt(dx**2 + dy**2))
+    return round(math.degrees(angle), 3)
+
+
+def comp_towards_batch(points: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+    """批量计算从起点指向终点的方向角度（顺时针，以正北为0度）"""
+    if len(points) < 2:
+        raise ValueError("点列表长度必须大于等于2")
+
+    blocks = []
+    for i in range(len(points) - 1):
+        block = {
+            "start": points[i],
+            "end": points[i + 1],
+        }
+        blocks.append(block)
+
+    for block in blocks:
+        angle = comp_towards(block["start"], block["end"])
+        block["angle"] = angle
+
+    return blocks