openclaw把GIS的直线边弧段# -*- coding: utf-8 -*-"""线段转曲线工具将 2 点直线转换为曲线（圆弧）"""import arcpyimport mathimport os# 输入输出input_fc = r"C:\Users\yl\Documents\ArcGIS\Projects\MyProject17\MyProject17.gdb\line"output_fc = r"C:\Users\yl\Documents\ArcGIS\Projects\MyProject17\MyProject17.gdb\k"# 曲线参数CURVE_HEIGHT = 200 # 曲线拱高（米），正值向上拱，负值向下拱print("=" * 60)print("线段转曲线工具")print("=" * 60)print("输入：%s" % input_fc)print("输出：%s" % output_fc)print("曲线拱高：%.1f 米" % CURVE_HEIGHT)print()# 设置环境arcpy.env.overwriteOutput = True# 获取输入要素类信息desc = arcpy.Describe(input_fc)spatial_ref = desc.spatialReferenceprint("坐标系：%s" % spatial_ref.name)# 创建输出要素类print("\n创建输出要素类...")arcpy.management.CreateFeatureclass( os.path.dirname(output_fc), os.path.basename(output_fc),"POLYLINE", spatial_reference=spatial_ref, has_m="DISABLED", has_z="DISABLED")print("[OK] 输出要素类已创建")# 复制字段print("\n复制字段结构...")input_fields = [f for f in arcpy.ListFields(input_fc) if f.name not in ['Shape', 'OID', 'FID']]for field in input_fields:if field.type not in ['OID', 'Geometry']: arcpy.management.AddField(output_fc, field.name, field.type, field.precision, field.scale, field.length)print("[OK] 字段已复制")# 转换线段到曲线def line_to_curve(start_pt, end_pt, height=CURVE_HEIGHT):"""将直线转换为圆弧曲线 参数: start_pt: 起点 (arcpy.Point) end_pt: 终点 (arcpy.Point) height: 拱高（米） 返回: arcpy.Polyline: 曲线几何 """ # 计算中点mid_x = (start_pt.X + end_pt.X) / 2mid_y = (start_pt.Y + end_pt.Y) / 2# 计算线段长度和方向dx = end_pt.X - start_pt.X dy = end_pt.Y - start_pt.Y length = math.sqrt(dx * dx + dy * dy)if length == 0:return None# 计算垂直方向的单位向量perp_x = -dy / length perp_y = dx / length# 计算控制点（中点垂直偏移）control_x = mid_x + perp_x * height control_y = mid_y + perp_y * height# 创建圆弧（使用三个点：起点、控制点、终点）# ArcGIS 中使用 densify 或 true curve # 这里使用多段线近似圆弧# 生成圆弧上的点（使用二次贝塞尔曲线公式）num_points = 20 # 圆弧分段数points = arcpy.Array()for i in range(num_points + 1): t = i / num_points# 二次贝塞尔曲线公式# B(t) = (1-t)²*P0 + 2(1-t)*t*P1 + t²*P2x = (1-t)**2 * start_pt.X + 2*(1-t)*t * control_x + t**2 * end_pt.X y = (1-t)**2 * start_pt.Y + 2*(1-t)*t * control_y + t**2 * end_pt.Y pt = arcpy.Point(x, y) points.add(pt)# 创建折线polyline = arcpy.Polyline(points, spatial_ref)return polyline# 处理要素print("\n开始转换...")insert_cursor = arcpy.da.InsertCursor(output_fc, ["SHAPE@"] + [f.name for f in input_fields])count_total = 0count_success = 0with arcpy.da.SearchCursor(input_fc, ["SHAPE@", "OID@"] + [f.name for f in input_fields]) as search_cursor:for row in search_cursor: count_total += 1geometry = row[0] oid = row[1] attributes = row[2:]try:# 处理每个部分new_parts = []for part in geometry: points = [pt for pt in part if pt]if len(points) < 2:continue# 将每对相邻点转换为曲线curve_points = arcpy.Array()for i in range(len(points) - 1): start_pt = points[i] end_pt = points[i + 1]# 转换为曲线curve = line_to_curve(start_pt, end_pt)if curve:# 添加曲线上的点（排除第一个点，避免重复）for j, pt in enumerate(curve.getPart(0)):if j > 0 or i == 0: # 第一段包含起点curve_points.add(pt)if curve_points.count > 1: new_parts.append(curve_points)if new_parts:# 创建新的折线if len(new_parts) == 1: new_geometry = arcpy.Polyline(new_parts[0], spatial_ref)else: new_geometry = arcpy.Polyline(arcpy.Array(new_parts), spatial_ref)# 插入新要素insert_cursor.insertRow([new_geometry] + list(attributes)) count_success += 1if count_success % 10 == 0: print(" 已处理 %d 个要素..." % count_success)except Exception as e: print("[警告] 要素 %d 处理失败：%s" % (oid, str(e)))del insert_cursorprint()print("=" * 60)print("处理完成!")print("共处理 %d 个要素，成功 %d 个" % (count_total, count_success))print("输出：%s" % output_fc)print("=" * 60)
夜雨聆风
