from osgeo import ogr
import matplotlib.pyplot as plt
from ospybook.vectorplotter import VectorPlotter
# 一、构建几何类型:点
point = ogr.Geometry(ogr.wkbPoint)
p1=point.AddPoint(3, 3) # 创建点
x, y = point.GetX(), point.GetY()
# 画图
vp = VectorPlotter(True) # 调用VectorPlotter类
vp.plot(point, 'bo') # 画出蓝色圆点
p2=point.AddPoint(5, 5) # 添加第二个点
vp.plot(point, 'rs') # 画出红色方点
plt.show()
plt.pause(2) #画图窗口显示2秒后自动关闭
# 二、构建线矢量
# 构建几何类型:线
line=ogr.Geometry(ogr.wkbLineString)
line.AddPoint(3,5)
line.AddPoint(5,3)
line.AddPoint(5,5)
# line.AddPoint(3,5)
# 画图
# 调用VectorPlotter类
vp = VectorPlotter(True)
vp.plot(line, 'r-')
plt.show()
plt.pause(2) #画图窗口显示2秒后自动关闭
# 三、构建面矢量
# 由线构建多边形
ring = ogr.Geometry(ogr.wkbLinearRing) # 构建几何类型:线
ring.AddPoint(1, 1) # 添加点01
ring.AddPoint(1, 3) # 添加点02
ring.AddPoint(3, 5) # 添加点03
ring.AddPoint(5, 5) # 添加点04
yard = ogr.Geometry(ogr.wkbPolygon) # 构建几何类型:多边形面
yard.AddGeometry(ring)
yard.CloseRings()
# 画图
vp = VectorPlotter(True) # 调用VectorPlotter类
vp.plot(yard, fill=False, edgecolor='green')
ring = yard.GetGeometryRef(0) # 再次创建线,循环构建
for i in range(ring.GetPointCount()):
ring.SetPoint(i, ring.GetX(i)-3, ring.GetY(i))
vp.plot(yard, fill=False, ec='red', linestyle='dashed')
plt.show()
plt.pause(2) # 画图窗口显示2秒后自动关闭
结果: