生成凹包

复制代码
import time
from struct import *
import subprocess
import fiona
import math
import numpy as np
import pylab as pl
from osgeo import ogr, gdal
import shapely.geometry as geometry
from shapely.geometry import polygon, multipolygon
from geojson import Feature, Point, FeatureCollection, dumps
from descartes import PolygonPatch
from shapely.ops import cascaded_union, polygonize
from scipy.spatial import Delaunay

#
input_shapefile = 'E:/test/pointNoRepeat.shp'
shapefile = fiona.open(input_shapefile)
points = [geometry.shape(point['geometry']) for point in shapefile]
x = [p.coords.xy[0] for p in points]
y = [p.coords.xy[1] for p in points]


'''
point_collection = geometry.MultiPoint(list(points))
point_collection.envelope

def plot_polygon(polygon):
    fig = pl.figure(figsize=(10,10))
    ax = fig.add_subplot(111)
    margin = .3
    x_min, y_min, x_max, y_max = polygon.bounds
    ax.set_xlim([x_min-margin, x_max+margin])
    ax.set_ylim([y_min-margin, y_max+margin])
    #print(str(x_min) + "  " + str(y_min) + "  " + str(x_max) + "  " + str(y_max))
    #print(polygon)
    patch = PolygonPatch(polygon, fc='#999999', ec='#000000', fill=True, zorder=-1)
    ax.add_patch(patch)
    return fig

plot_polygon(point_collection.envelope)
pl.plot(x,y,'o', color='#f16824')
'''


#
def alpha_shape(points, alpha):
    """
    Compute the alpha shape (concave hull) of a set of points.
    @param points: Iterable container of points.
    @param alpha: alpha value to influence the gooeyness of the border. Smaller numbers don't fall inward as much as larger numbers.
        Too large, and you lose everything!
    """
    if len(points) < 4:
        # When you have a triangle, there is no sense in computing an alpha shape.
        return geometry.MultiPoint(list(points)).convex_hull
    def add_edge(edges, edge_points, coords, i, j):
        """
        Add a line between the i-th and j-th points, if not in the list already
        """
        if (i, j) in edges or (j, i) in edges:
            # already added
            return
        edges.add( (i, j) )
        edge_points.append(coords[ [i, j] ])
    coords = np.array([point.coords[0] for point in points])
    #print(coords)
    tri = Delaunay(coords)
    edges = set()
    edge_points = []
    # loop over triangles:
    # ia, ib, ic = indices of corner points of the triangle
    for ia, ib, ic in tri.vertices:
        pa = coords[ia]
        pb = coords[ib]
        pc = coords[ic]
        # Lengths of sides of triangle
        a = math.sqrt((pa[0]-pb[0])**2 + (pa[1]-pb[1])**2)
        b = math.sqrt((pb[0]-pc[0])**2 + (pb[1]-pc[1])**2)
        c = math.sqrt((pc[0]-pa[0])**2 + (pc[1]-pa[1])**2)
        # Semiperimeter of triangle
        s = (a + b + c)/2.0
        # Area of triangle by Heron's formula
        area = math.sqrt(s*(s-a)*(s-b)*(s-c))
        if area <= 0:
            continue;
        circum_r = a*b*c/(4.0*area)
        # Here's the radius filter.
        #print circum_r
        if circum_r < 1.0/alpha:
            add_edge(edges, edge_points, coords, ia, ib)
            add_edge(edges, edge_points, coords, ib, ic)
            add_edge(edges, edge_points, coords, ic, ia)
    m = geometry.MultiLineString(edge_points)
    triangles = list(polygonize(m))
    #
    return cascaded_union(triangles), edge_points

#
concave_hull, edge_points = alpha_shape(points, alpha=0.38)
#
featCollection = FeatureCollection([])
featCollection["crs"] = {"type":"name","properties":{"name":"EPSG:2385"}}
if hasattr(concave_hull, "geoms"):
    polys = concave_hull.geoms
    fig = pl.figure(figsize=(10,10))
    ax = fig.add_subplot(111)
    margin = .3
    if len(concave_hull.bounds)>0:
        #
        x_min, y_min, x_max, y_max = concave_hull.bounds
        ax.set_xlim([x_min-margin, x_max+margin])
        ax.set_ylim([y_min-margin, y_max+margin])
        for poly in polys:
            patch = PolygonPatch(poly, fc='#999999', ec='#000000', fill=True, zorder=-1)
            ax.add_patch(patch)
            #
            geojs_geom = poly.__geo_interface__
            hull_poly = dict(type='Feature', properties=dict(id=1))
            hull_poly['geometry'] = geojs_geom
            #print(hull_poly)
            my_feature = Feature(geometry=geojs_geom)
            featCollection["features"].append(my_feature)
        print(featCollection)
        pl.plot(x,y,'o', color='#f16824')
    #
    #print(concave_hull.geoms)
else:
    fig = pl.figure(figsize=(10,10))
    ax = fig.add_subplot(111)
    margin = .3
    x_min, y_min, x_max, y_max = concave_hull.bounds
    ax.set_xlim([x_min-margin, x_max+margin])
    ax.set_ylim([y_min-margin, y_max+margin])
    patch = PolygonPatch(concave_hull, fc='#999999', ec='#000000', fill=True, zorder=-1)
    ax.add_patch(patch)
    pl.plot(x, y, 'o', color='#f16824')
    geojs_geom = concave_hull.__geo_interface__
    my_feature = Feature(geometry=geojs_geom)
    featCollection["features"].append(my_feature)
#
file_object = open('e:\\test\\result\\result.json', 'w')
file_object.write(dumps(featCollection))
file_object.close()
subprocess.call(["ogr2ogr", "-t_srs", "EPSG:2385", "-f", "ESRI Shapefile", "e:\\test\\result\\ConcaveEnvelope.shp", "e:\\test\\result\\result.json"])
pl.show()
复制代码

 

image

 

image

posted @   ParamousGIS  阅读(2130)  评论(1编辑  收藏  举报
编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· DeepSeek 开源周回顾「GitHub 热点速览」
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
点击右上角即可分享
微信分享提示